File tree Expand file tree Collapse file tree 1 file changed +14
-4
lines changed Expand file tree Collapse file tree 1 file changed +14
-4
lines changed Original file line number Diff line number Diff line change @@ -240,15 +240,25 @@ func (m *Module) inspectorPreRequest(req *http.Request) (inspin2 RPCMsgIn2, out
240
240
return
241
241
}
242
242
243
- // set any request headers
244
243
if out .RequestID != "" {
245
- req .Header .Add ("X-Sigsci-Requestid" , out .RequestID )
244
+ req .Header .Set ("X-Sigsci-Requestid" , out .RequestID )
245
+ } else {
246
+ req .Header .Del ("X-Sigsci-Requestid" )
246
247
}
247
248
248
249
wafresponse := out .WAFResponse
249
- req .Header .Add ("X-Sigsci-Agentresponse" , strconv .Itoa (int (wafresponse )))
250
+ req .Header .Set ("X-Sigsci-Agentresponse" , strconv .Itoa (int (wafresponse )))
251
+
252
+ // Add request headers from the WAF response to the request
253
+ req .Header .Del ("X-Sigsci-Tags" )
254
+ req .Header .Del ("X-Sigsci-Redirect" )
250
255
for _ , kv := range out .RequestHeaders {
251
- req .Header .Add (kv [0 ], kv [1 ])
256
+ // For X-Sigsci-* headers, use Set to override, but for custom headers, use Add to append
257
+ if strings .HasPrefix (http .CanonicalHeaderKey (kv [0 ]), "X-Sigsci-" ) {
258
+ req .Header .Set (kv [0 ], kv [1 ])
259
+ } else {
260
+ req .Header .Add (kv [0 ], kv [1 ])
261
+ }
252
262
}
253
263
254
264
inspin2 = RPCMsgIn2 {
You can’t perform that action at this time.
0 commit comments