add defer connection close

Author: davidqhr

client/server.go

@@ -21,7 +21,7 @@ func proxyToServer(client *Client, serverAddr string) {
 	eConn := utils.NewEncryptedConn(remoteConn, client.Password)
 	// remoteConn, err := net.Dial("tcp", "localhost:8112")
 
-	defer eConn.Conn.Close()
+	defer eConn.Close()
 
 	go utils.Copy(eConn, client.Conn)
 	utils.Copy(client.Conn, eConn)
@@ -44,7 +44,6 @@ func handleClient(client *Client, serverAddr string) {
 	}
 
 	method := chooseAuthMethod(methods)
-
 	err = client.SetAuthMethod(method)
 
 	if err != nil {

server/server.go

@@ -7,6 +7,8 @@ import (
 )
 
 func handleConn(eConn *utils.EncryptedConn) {
+	defer eConn.Close()
+
 	buf := utils.Pool33K.Get()
 	defer utils.Pool33K.Put(buf)
 

utils/io.go

@@ -6,8 +6,10 @@ import (
 	"time"
 )
 
-// this code is copy from https://golang.org/src/io/io.go
-// add timeout and buffer pool support
+// this code is copied from https://golang.org/src/io/io.go
+// with some additions
+// 1. timeout
+// 2. buffer pool
 
 func Copy(dst net.Conn, src net.Conn) (written int64, err error) {
 	timeoutDuration := 15 * time.Second