fix: memory leak on http tunnel

httptunnel/server.go

@@ -141,7 +141,7 @@ func (s *Server) push(w http.ResponseWriter, r *http.Request) {
 	s.Lock()
 	conn, ok := s.connMap[reqID]
 	if !ok {
-		conn = NewServerConn()
+		conn = NewServerConn(reqID, s.CloseConn)
 		s.connMap[reqID] = conn
 		s.connCh <- conn
 	}
@@ -162,6 +162,13 @@ func (s *Server) push(w http.ResponseWriter, r *http.Request) {
 	writeNoContent(w)
 }
 
+func (s *Server) CloseConn(reqID string) {
+	s.Lock()
+	defer s.Unlock()
+	s.connMap[reqID] = nil
+	delete(s.connMap, reqID)
+}
+
 func writeNotFoundError(w http.ResponseWriter) {
 	http.Error(w, "404 NOT FOUND", http.StatusNotFound)
 }

httptunnel/server_conn.go

@@ -8,15 +8,20 @@ import (
 var _ net.Conn = (*ServerConn)(nil)
 
 type ServerConn struct {
+	reqID     string
+	closeConn func(reqID string)
+
 	local  net.Conn
 	remote net.Conn
 }
 
-func NewServerConn() *ServerConn {
+func NewServerConn(reqID string, closeConn func(reqID string)) *ServerConn {
 	read, write := net.Pipe()
 	return &ServerConn{
-		local:  read,
-		remote: write,
+		reqID:     reqID,
+		closeConn: closeConn,
+		local:     read,
+		remote:    write,
 	}
 }
 
@@ -37,6 +42,11 @@ func (c *ServerConn) Write(b []byte) (n int, err error) {
 }
 
 func (c *ServerConn) Close() error {
+	if c.closeConn != nil {
+		c.closeConn(c.reqID)
+	}
+	c.closeConn = nil
+
 	return c.remote.Close()
 }