Merge pull request #179 from ddosify/perf_fix/body_size_check

req body store limit
getanteon · Apr 18, 2023 · 2a004b5 · 2a004b5
2 parents 6877608 + 8926938
commit 2a004b5
Showing 1 changed file with 14 additions and 9 deletions.
diff --git a/core/scenario/requester/http.go b/core/scenario/requester/http.go
@@ -232,16 +232,21 @@ func (h *HttpRequester) Send(client *http.Client, envs map[string]interface{}) (
 	}
 
 	if httpReq.Body != nil {
-		if h.constantBodyReader != nil {
-			// copy req body for debug
-			sectionReader := io.NewSectionReader(h.constantBodyReader, 0, int64(len(h.packet.Payload)))
-			copiedReqBody = make([]byte, len(h.packet.Payload))
-			sectionReader.Read(copiedReqBody)
+		if int64(len(h.packet.Payload)) > 300000 {
+			// Don't store req bodies bigger than 300KB
+			copiedReqBody = []byte("too long body")
 		} else {
-			buf := bytes.Buffer{}
-			io.Copy(&buf, httpReq.Body)
-			copiedReqBody = buf.Bytes()
-			httpReq.Body = io.NopCloser(bytes.NewReader(buf.Bytes()))
+			if h.constantBodyReader != nil {
+				// copy req body for debug
+				sectionReader := io.NewSectionReader(h.constantBodyReader, 0, int64(len(h.packet.Payload)))
+				copiedReqBody = make([]byte, len(h.packet.Payload))
+				sectionReader.Read(copiedReqBody)
+			} else {
+				buf := bytes.Buffer{}
+				io.Copy(&buf, httpReq.Body)
+				copiedReqBody = buf.Bytes()
+				httpReq.Body = io.NopCloser(bytes.NewReader(buf.Bytes()))
+			}
 		}
 	}