diff --git a/server/service/image.go b/server/service/image.go index 8c249a8..437b8fb 100644 --- a/server/service/image.go +++ b/server/service/image.go @@ -59,7 +59,7 @@ func CreateImage(uid uint, data []byte) (uint, error) { if len(data) == 0 { 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") } @@ -92,6 +92,11 @@ func CreateImage(uid uint, data []byte) (uint, error) { 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() if err := os.WriteFile(imageDir+filename, data, 0644); err != nil { return 0, errors.New("failed to save image file")