diff --git a/main.go b/main.go index d87af73..791facb 100644 --- a/main.go +++ b/main.go @@ -12,8 +12,9 @@ import ( ) const ( - port = ":8080" - filesDir = "./files" + port = ":8080" + filesDir = "./files" + maxFileSize = 20 * 1024 * 1024 // 20MiB ) var url string = os.Getenv("URL") @@ -41,9 +42,11 @@ func uploadHandler(w http.ResponseWriter, r *http.Request) { return } + r.Body = http.MaxBytesReader(w, r.Body, maxFileSize) + file, _, err := r.FormFile("file") if err != nil { - http.Error(w, "Error retrieving the file", http.StatusBadRequest) + http.Error(w, "Error retrieving the file or file size exceeds limit", http.StatusBadRequest) return } defer file.Close()