Changeset 5 in code

Timestamp:

Oct 11, 2021, 3:14:13 PM (4 years ago)

Author:

dev

Message:

Limit writing to Content-Length header

File:

• trunk/partage.go (modified) (3 diffs)

trunk/partage.go

@@ -33 +33 @@
 }
 
-func writefile(f *os.File, s io.ReadCloser) int64 {
+func writefile(f *os.File, s io.ReadCloser, contentlength int64) int64 {
         buffer := make([]byte, 4096)
         eof := false
@@ -51 +51 @@
                 /* ensure we don't write more than expected */
                 r := int64(n)
-                if sz+r > conf.maxsize {
-                        r = conf.maxsize - sz
+                if sz+r > contentlength {
+                        r = contentlength - sz
                         eof = true
                 }
@@ -120 +120 @@
                 defer f.Close()
 
-                if writefile(f, r.Body) < 0 {
+                if writefile(f, r.Body, r.ContentLength) < 0 {
                         w.WriteHeader(http.StatusInternalServerError)
                         return