feat(http): assign customHeaders to the map directly (projectdiscov…

…ery#5445) also add skip expr if header key is "Host" Signed-off-by: Dwi Siswanto <[email protected]>
alban-stourbe-wmx · Jul 26, 2024 · 6d325a4 · 6d325a4
1 parent 33dbb51
commit 6d325a4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/protocols/http/request.go b/pkg/protocols/http/request.go
@@ -1083,10 +1083,15 @@ func (request *Request) setCustomHeaders(req *generatedRequest) {
 			req.rawRequest.Headers[k] = v
 		} else {
 			kk, vv := strings.TrimSpace(k), strings.TrimSpace(v)
-			req.request.Header.Set(kk, vv)
+			// NOTE(dwisiswant0): Do we really not need to convert it first into
+			// lowercase?
 			if kk == "Host" {
 				req.request.Host = vv
+
+				continue
 			}
+
+			req.request.Header[kk] = []string{vv}
 		}
 	}
 }