Added KeepHeader #306:

elazarl · Jul 10, 2020 · 96f1a10 · 96f1a10
1 parent 49ad98f
commit 96f1a10
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/go.mod b/go.mod
@@ -1,3 +1,5 @@
 module github.com/elazarl/goproxy
 
+go 1.14
+
 require github.com/elazarl/goproxy/ext v0.0.0-20190711103511-473e67f1d7d2
diff --git a/proxy.go b/proxy.go
@@ -30,6 +30,7 @@ type ProxyHttpServer struct {
 	// if nil Tr.Dial will be used
 	ConnectDial func(network string, addr string) (net.Conn, error)
 	CertStore   CertStorage
+	KeepHeader  bool
 }
 
 var hasPort = regexp.MustCompile(`:\d+$`)
@@ -118,7 +119,9 @@ func (proxy *ProxyHttpServer) ServeHTTP(w http.ResponseWriter, r *http.Request)
 				proxy.serveWebsocket(ctx, w, r)
 			}
 
-			removeProxyHeaders(ctx, r)
+			if !proxy.KeepHeader {
+				removeProxyHeaders(ctx, r)
+			}
 			resp, err = ctx.RoundTrip(r)
 			if err != nil {
 				ctx.Error = err