fix closing scheme

client/main.go

@@ -71,10 +71,11 @@ func handleClient(mux generic.Mux, p1 net.Conn, ctrl *generic.CopyControl, quiet
 				}
 			}
 		}
+		p1.Close()
+		p2.Close()
 	}
 
 	go streamCopy(p1, p2)
-	// errors reading from p1(net.Conn) can close p2 while return
 	streamCopy(p2, p1)
 }
 

server/main.go

@@ -138,10 +138,11 @@ func handleClient(p1 io.ReadWriteCloser, p2 net.Conn, ctrl *generic.CopyControl,
 				}
 			}
 		}
+		p1.Close()
+		p2.Close()
 	}
 
 	go streamCopy(p2, p1)
-	// errors reading from p2(net.Conn) can close p1 while return
 	streamCopy(p1, p2)
 }