add removal of proxy header

xvzc · Jan 14, 2022 · cb6a593 · cb6a593
1 parent 9d5cd8f
commit cb6a593
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/packet/http.go b/packet/http.go
@@ -83,6 +83,28 @@ func (p *HttpPacket) IsConnectMethod() bool {
 	return p.Method() == "CONNECT"
 }
 
+func (p *HttpPacket) RemoveProxyHeader() {
+	s := string(p.raw)
+
+	lines := strings.Split(s, "\n")
+	for i := 0; i < len(lines); i++ {
+		if strings.HasPrefix(lines[i], "Proxy-Connection") {
+			lines[i] = ""
+		}
+	}
+
+	result := ""
+	for i := 0; i < len(lines); i++ {
+		if lines[i] == "" {
+			continue
+		}
+
+		result += lines[i] + "\n"
+	}
+
+	p.raw = []byte(result)
+}
+
 func parse(raw []byte) (string, string, string) {
 	var firstLine string
 	for i := 0; i < len(raw); i++ {

diff --git a/proxy/proxy.go b/proxy/proxy.go
@@ -48,13 +48,16 @@ func (p *Proxy) Start() {
 			log.Debug("Client sent data: ", len(b))
 
 			pkt := packet.NewHttpPacket(b)
-			log.Debug("New request: \n\n" + string(pkt.Raw()))
 
 			if !pkt.IsValidMethod() {
 				log.Println("Unsupported method: ", pkt.Method())
 				return
 			}
 
+			pkt.RemoveProxyHeader()
+
+			log.Debug("New request: \n\n" + string(pkt.Raw()))
+
 			if pkt.IsConnectMethod() {
 				log.Debug("[HTTPS] Start")
 				go conn.HandleHttps(pkt)