iridium/neko-custom
Archived
2
0
Fork 0
Code Pull Requests Activity

remove any temporary files associated with a Form, fixes #347.

Browse Source
This commit is contained in:
Miroslav Šedivý 2023-12-26 15:49:16 +01:00
parent a03b29ba01
commit 792b1ac111
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
server/internal/http/http.go
View File

@ -166,7 +166,13 @@ func New(conf *config.Server, webSocketHandler types.WebSocketHandler, desktop t
return return
} }
r.ParseMultipartForm(32 << 20) err = r.ParseMultipartForm(32 << 20)
if err != nil || r.MultipartForm == nil {
logger.Warn().Err(err).Msg("failed to parse multipart form")
http.Error(w, "error parsing form", http.StatusBadRequest)
return
}
for _, formheader := range r.MultipartForm.File["files"] { for _, formheader := range r.MultipartForm.File["files"] {
filePath := webSocketHandler.FileTransferPath(formheader.Filename) filePath := webSocketHandler.FileTransferPath(formheader.Filename)
@ -187,6 +193,11 @@ func New(conf *config.Server, webSocketHandler types.WebSocketHandler, desktop t
io.Copy(f, formfile) io.Copy(f, formfile)
} }
err = r.MultipartForm.RemoveAll()
if err != nil {
logger.Warn().Err(err).Msg("failed to remove multipart form")
}
}) })
} }