adamf/web_server
1
1
Fork 0
Code Issues 5 Pull Requests Actions Packages Projects Releases Wiki Activity

Add error handling

Browse Source
This commit is contained in:
Adam French
2026-03-09 16:41:38 +00:00
parent 41d6cf0dac
commit d03f9668ad
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backend/handlers/handle_message_upload.go
View File

@@ -18,8 +18,8 @@ var allowedExtensions = map[string]bool{
} }
var extensionToMIMEPrefix = map[string]string{ var extensionToMIMEPrefix = map[string]string{
".jpg": "image/", ".jpeg": "image/", ".png": "image/png", ".gif": "image/gif", ".webp": "image/webp", ".jpg": "image/", ".jpeg": "image/", ".png": "image/", ".gif": "image/", ".webp": "image/",
".mp4": "video/", ".webm": "video/webm", ".mp3": "audio/", ".ogg": "audio/", ".mp4": "video/", ".webm": "video/",
".pdf": "application/pdf", ".txt": "text/", ".pdf": "application/pdf", ".txt": "text/",
} }
@@ -49,8 +49,12 @@ func (store *Store) UploadMessageFile(ctx *gin.Context) {
return return
} }
buf := make([]byte, 512) buf := make([]byte, 512)
n, _ := f.Read(buf) n, err := f.Read(buf)
f.Close() f.Close()
if err != nil && n == 0 {
ctx.JSON(http.StatusBadRequest, gin.H{"error": "failed to read file content"})
return
}
detectedType := http.DetectContentType(buf[:n]) detectedType := http.DetectContentType(buf[:n])
expectedPrefix, ok := extensionToMIMEPrefix[ext] expectedPrefix, ok := extensionToMIMEPrefix[ext]