nyne/nysoure
1
0
Fork 0
mirror of https://github.com/wgh136/nysoure.git synced 2025-09-27 12:17:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Increase maximum image data size limit to 8MB and add error handling for oversized images after conversion

Browse Source
This commit is contained in:
nyne
2025-05-18 17:31:59 +08:00
parent 4c0166e364
commit db34b32d63
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
server/service/image.go
View File

@@ -59,7 +59,7 @@ func CreateImage(uid uint, data []byte) (uint, error) {
if len(data) == 0 { if len(data) == 0 {
return 0, model.NewRequestError("Image data is empty") return 0, model.NewRequestError("Image data is empty")
} else if len(data) > 1024*1024*5 { } else if len(data) > 1024*1024*8 {
return 0, model.NewRequestError("Image data is too large") return 0, model.NewRequestError("Image data is too large")
} }
@@ -92,6 +92,11 @@ func CreateImage(uid uint, data []byte) (uint, error) {
contentType = "image/webp" contentType = "image/webp"
} }
// If the image is still too large after conversion, return an error
if len(data) > 1024*1024*4 {
return 0, model.NewRequestError("Image data is too large after conversion")
}
filename := uuid.New().String() filename := uuid.New().String()
if err := os.WriteFile(imageDir+filename, data, 0644); err != nil { if err := os.WriteFile(imageDir+filename, data, 0644); err != nil {
return 0, errors.New("failed to save image file") return 0, errors.New("failed to save image file")