Fix: http request keepAlive with right http header

zpzgone · Mar 3, 2019 · 23bb01a · 23bb01a
1 parent 0011c7a
commit 23bb01a
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/tunnel/connection.go b/tunnel/connection.go
@@ -25,12 +25,9 @@ func (t *Tunnel) handleHTTP(request *adapters.HTTPAdapter, outbound net.Conn) {
 	conn := newTrafficTrack(outbound, t.traffic)
 	req := request.R
 	host := req.Host
-	keepalive := true
 
 	for {
-		if strings.ToLower(req.Header.Get("Connection")) == "close" {
-			keepalive = false
-		}
+		keepAlive := strings.TrimSpace(strings.ToLower(req.Header.Get("Proxy-Connection"))) == "keep-alive"
 
 		req.Header.Set("Connection", "close")
 		req.RequestURI = ""
@@ -58,7 +55,7 @@ func (t *Tunnel) handleHTTP(request *adapters.HTTPAdapter, outbound net.Conn) {
 			break
 		}
 
-		if !keepalive {
+		if !keepAlive {
 			break
 		}