martian: manually handle closing req and res body

saucelabs · Nov 16, 2023 · 9934b88 · 9934b88
1 parent bc745c4
commit 9934b88
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/internal/martian/proxy.go b/internal/martian/proxy.go
@@ -859,6 +859,15 @@ func (p *Proxy) handle(ctx *Context, conn net.Conn, brw *bufio.ReadWriter) error
 		log.Errorf(req.Context(), "got error while flushing response back to client: %v", err)
 	}
 
+	if err := req.Body.Close(); err != nil {
+		log.Errorf(req.Context(), "failed to close request body: %v", err)
+		closing = errClose
+	}
+	if err := res.Body.Close(); err != nil {
+		log.Errorf(req.Context(), "failed to close response body: %v", err)
+		closing = errClose
+	}
+
 	if p.CloseAfterReply {
 		closing = errClose
 	}