mirror of
https://github.com/wgh136/nysoure.git
synced 2025-09-27 04:17:23 +00:00
Increase maximum image data size limit to 8MB and add error handling for oversized images after conversion
This commit is contained in:
@@ -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")
|
||||
|
Reference in New Issue
Block a user