Fix vmess mux leak

mux.go

@@ -20,6 +20,8 @@ import (
 )
 
 func HandleMuxConnection(ctx context.Context, conn net.Conn, source M.Socksaddr, handler Handler) error {
+	ctx, cancel := context.WithCancel(ctx)
+	defer cancel()
 	session := &serverSession{
 		ctx:          ctx,
 		source:       source,
@@ -29,12 +31,10 @@ func HandleMuxConnection(ctx context.Context, conn net.Conn, source M.Socksaddr,
 		streams:      make(map[uint16]*serverStream),
 		writer:       std_bufio.NewWriter(conn),
 	}
-	if ctx.Done() != nil {
-		go func() {
-			<-ctx.Done()
-			session.cleanup(ctx.Err())
-		}()
-	}
+	go func() {
+		<-ctx.Done()
+		session.cleanup(ctx.Err())
+	}()
 	return session.recvLoop()
 }
 
@@ -72,6 +72,7 @@ func (c *serverSession) cleanup(err error) {
 	for _, stream := range c.streams {
 		_ = stream.pipe.CloseWithError(err)
 	}
+	c.streams = nil
 	c.streamAccess.Unlock()
 }