Fix connection leaks.

Author: moriyoshi

main.go

@@ -95,12 +95,14 @@ func (ctx *DevProxy) bidiTunnel(connA net.Conn, connB net.Conn) {
 	wg := &sync.WaitGroup{}
 	wg.Add(1)
 	go func() {
+		defer wg.Done()
 		ctx.unidiTunnel(connA, connB)
 		connA.SetDeadline(time.Unix(0, 0))
 		connB.SetDeadline(time.Unix(0, 0))
 	}()
 	wg.Add(1)
 	go func() {
+		defer wg.Done()
 		ctx.unidiTunnel(connB, connA)
 		connA.SetDeadline(time.Unix(0, 0))
 		connB.SetDeadline(time.Unix(0, 0))

server.go

@@ -363,6 +363,7 @@ func (proxyCtx *OurProxyCtx) DoRequest(req *http.Request, respW http.ResponseWri
 }
 
 func (proxyCtx *OurProxyCtx) HandleConnect(r *http.Request, proxyClient net.Conn) {
+	defer proxyClient.Close()
 	var targetHostPort string
 	{
 		pair := splitHostPort(r.URL.Host)