Fix memory leak

sower-proxy · Nov 28, 2018 · b772ed1 · b772ed1
1 parent 1222fdf
commit b772ed1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/conf/sower.toml b/conf/sower.toml
@@ -20,6 +20,7 @@ blocklist=[
     "img1.blogblog.com",
     "*.golang.org", # golang
     "go.googlesource.com",
+	"gist.github.com",
     "*.wikipedia.org", # wikipeida
 	"*.wikimedia.org",
 	"*.wikisource.org",

diff --git a/proxy/client.go b/proxy/client.go
@@ -57,11 +57,14 @@ func openStream(conn net.Conn, sess quic.Session, reDialCh chan<- net.Conn) bool
 		select {
 		case okCh <- struct{}{}:
 		default:
+			close(okCh)
+			return
 		}
 		close(okCh)
 
 		conn.(*net.TCPConn).SetKeepAlive(true)
 		relay(&streamConn{stream, sess}, conn)
+		conn.Close()
 	}()
 
 	select {