Profouzors
/
ProfouzorsLinx
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update upload.go

This commit is contained in:
ZizzyDizzyMC 2021-02-21 21:57:19 -05:00
parent 93ec408815
commit 416a0b1dcc
1 changed files with 21 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
upload.go
View File

@ -55,8 +55,13 @@ func uploadPostHandler(c web.C, w http.ResponseWriter, r *http.Request) {
badRequestHandler(c, w, r, RespAUTO, "") badRequestHandler(c, w, r, RespAUTO, "")
return return
} }
if r.Header.Get("Test-Header") == "test" { if len(r.Header.Get("Content-Length")) > 0 {
oopsHandler(c, w, r, RespHTML, "You killed it") i, err := strconv.ParseInt(r.Header.Get("Content-Length"), 10, 64)
if err == nil {
if i > Config.maxSize {
oopsHandler(c, w, r, RespHTML, "Could not upload file: File too large")
}
}
} }
upReq := UploadRequest{} upReq := UploadRequest{}
@ -125,8 +130,13 @@ func uploadPostHandler(c web.C, w http.ResponseWriter, r *http.Request) {
} }
func uploadPutHandler(c web.C, w http.ResponseWriter, r *http.Request) { func uploadPutHandler(c web.C, w http.ResponseWriter, r *http.Request) {
if r.Header.Get("Test-Header") == "test" { if len(r.Header.Get("Content-Length")) > 0 {
oopsHandler(c, w, r, RespHTML, "You killed it") i, err := strconv.ParseInt(r.Header.Get("Content-Length"), 10, 64)
if err == nil {
if i > Config.maxSize {
oopsHandler(c, w, r, RespHTML, "Could not upload file: File too large")
}
}
} }
upReq := UploadRequest{} upReq := UploadRequest{}
uploadHeaderProcess(r, &upReq) uploadHeaderProcess(r, &upReq)
@ -236,8 +246,13 @@ func uploadHeaderProcess(r *http.Request, upReq *UploadRequest) {
if r.Header.Get("Linx-Randomize") == "yes" { if r.Header.Get("Linx-Randomize") == "yes" {
upReq.randomBarename = true upReq.randomBarename = true
} }
if r.Header.Get("Test-Header") == "test" { if len(r.Header.Get("Content-Length")) > 0 {
return i, err := strconv.ParseInt(r.Header.Get("Content-Length"), 10, 64)
if err == nil {
if i > Config.maxSize {
oopsHandler(c, w, r, RespHTML, "Could not upload file: File too large")
}
}
} }
upReq.deleteKey = r.Header.Get("Linx-Delete-Key") upReq.deleteKey = r.Header.Get("Linx-Delete-Key")
upReq.accessKey = r.Header.Get(accessKeyHeaderName) upReq.accessKey = r.Header.Get(accessKeyHeaderName)