fix memory leak

koalabearguo · koalabearguo · commit f8c27f7cbced · 2021-09-02T09:58:43.000-04:00
diff --git a/proxy.go b/proxy.go
@@ -171,12 +171,17 @@ func (prx *proxy) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
 			tlscon.Close()
 			return
 		}
-		go prx.IOCopy(loConn, tlscon)
-		prx.IOCopy(tlscon, loConn)
+
+		go prx.IOCopy(tlscon, loConn)
+		prx.IOCopy(loConn, tlscon)
 		err = tlscon.Close()
 		if err != nil {
 			log.Println(err)
 		}
+		err = loConn.Close()
+		if err != nil {
+			log.Println(err)
+		}
 		return
 
 	}

Original file line number	Diff line number	Diff line change
`@@ -171,12 +171,17 @@ func (prx proxy) ServeHTTP(rw http.ResponseWriter, req http.Request) {`
`171`	`171`	`tlscon.Close()`
`172`	`172`	`return`
`173`	`173`	`}`
`174`		`- go prx.IOCopy(loConn, tlscon)`
`175`		`- prx.IOCopy(tlscon, loConn)`
	`174`	`+`
	`175`	`+ go prx.IOCopy(tlscon, loConn)`
	`176`	`+ prx.IOCopy(loConn, tlscon)`
`176`	`177`	`err = tlscon.Close()`
`177`	`178`	`if err != nil {`
`178`	`179`	`log.Println(err)`
`179`	`180`	`}`
	`181`	`+ err = loConn.Close()`
	`182`	`+ if err != nil {`
	`183`	`+ log.Println(err)`
	`184`	`+ }`
`180`	`185`	`return`
`181`	`186`
`182`	`187`	`}`