From 4012ce84734f86b8365689e867389a6daa5e9411 Mon Sep 17 00:00:00 2001
From: dev <dev@d6811dac-2434-b64a-9ddc-f563ab233461>
Date: Mon, 11 Oct 2021 15:14:13 +0000
Subject: [PATCH] Limit writing to Content-Length header

git-svn-id: file:///srv/svn/repo/marisa/trunk@5 d6811dac-2434-b64a-9ddc-f563ab233461
---
 partage.go | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/partage.go b/partage.go
index 9163b0e..d1fe91d 100644
--- a/partage.go
+++ b/partage.go
@@ -32,7 +32,7 @@ func contenttype(f *os.File) string {
 	return mime
 }
 
-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
 	sz := int64(0)
@@ -50,8 +50,8 @@ func writefile(f *os.File, s io.ReadCloser) int64 {
 
 		/* 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
 		}
 
@@ -119,7 +119,7 @@ func parse(w http.ResponseWriter, r *http.Request) {
 		}
 		defer f.Close()
 
-		if writefile(f, r.Body) < 0 {
+		if writefile(f, r.Body, r.ContentLength) < 0 {
 			w.WriteHeader(http.StatusInternalServerError)
 			return
 		}