From 8e568b64510078ede7f994b8e50841333b6d218d Mon Sep 17 00:00:00 2001
From: "rui.zheng" <rui.zheng@tcl.com>
Date: Wed, 10 Jun 2015 15:36:17 +0800
Subject: [PATCH] close tunnel properly

---
 client.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/client.go b/client.go
index 64f00e7..315efe1 100644
--- a/client.go
+++ b/client.go
@@ -210,6 +210,7 @@ func handleSocks5(conn net.Conn) {
 		log.Println(err)
 		return
 	}
+	defer sconn.Close()
 
 	switch req.Cmd {
 	case gosocks5.CmdConnect, gosocks5.CmdBind:
@@ -326,6 +327,7 @@ func handleHttp(req *http.Request, conn net.Conn) {
 		log.Println(err)
 		return
 	}
+	defer sconn.Close()
 
 	if err := r.Write(sconn); err != nil {
 		return
@@ -366,6 +368,7 @@ func handleShadow(conn net.Conn) {
 		log.Println(err)
 		return
 	}
+	defer sconn.Close()
 
 	req := gosocks5.NewRequest(gosocks5.CmdConnect, addr)
 	if err := req.Write(sconn); err != nil {