From 710c83a16e1063600a38dc049cc5baf2fca55a8b Mon Sep 17 00:00:00 2001 From: nyne Date: Mon, 26 May 2025 20:56:45 +0800 Subject: [PATCH] Reject file uploads exceeding maximum size limit and delete rejected files --- server/service/file.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/server/service/file.go b/server/service/file.go index af0a522..0a1cc89 100644 --- a/server/service/file.go +++ b/server/service/file.go @@ -511,6 +511,12 @@ func CreateServerDownloadTask(uid uint, url, filename, description string, resou return nil, model.NewInternalServerError("failed to create file in db") } + if contentLength+getUploadingSize() > config.MaxUploadingSize() { + log.Info("A new downloading file is rejected due to max uploading size limit") + _ = dao.DeleteFile(file.UUID) + return nil, model.NewRequestError("server is busy, please try again later") + } + updateUploadingSize(contentLength) go func() {