优化全局数据存储方式，以节点 id 为 key 保存相关数据；修复节点下线监听未关闭问题

gateway/fwd/auth/auth.go

@@ -36,8 +36,8 @@ func Protect(conn net.Conn, proto Protocol, username, password *string) (*core.A
 	}
 
 	// 查找权限配置
-	var permit, ok = app.Permits.Load(uint16(localPort))
-	if !ok {
+	var permit = app.LoadPermit(uint16(localPort))
+	if permit == nil {
 		return nil, errors.New("没有权限")
 	}
 

gateway/fwd/ctrl.go

@@ -29,7 +29,6 @@ const (
 
 func (s *Service) listenCtrl() error {
 	ctrlPort := env.AppCtrlPort
-	slog.Debug("监听控制通道", slog.Uint64("port", uint64(ctrlPort)))
 
 	// 监听端口
 	ls, err := net.Listen("tcp", ":"+strconv.Itoa(int(ctrlPort)))
@@ -80,22 +79,19 @@ func (s *Service) listenCtrl() error {
 	}
 }
 
-func (s *Service) processCtrlConn(ctx context.Context, conn net.Conn) (err error) {
+func (s *Service) processCtrlConn(_ctx context.Context, conn net.Conn) (err error) {
+	// 通道上下文
+	ctx, cancel := context.WithCancel(_ctx)
+
+	// 结束后清理资源
+	var fwdPort uint16
 	defer func() {
-		_, portStr, err := net.SplitHostPort(conn.LocalAddr().String())
-		if err != nil {
-			slog.Error("获取控制通道端口失败", "err", err)
-			return
-		}
-
-		port, err := strconv.ParseUint(portStr, 10, 16)
-		if err != nil {
-			slog.Error("解析控制通道端口失败", "err", err)
-			return
-		}
-
-		app.DelEdge(uint16(port))
+		slog.Debug("关闭控制通道", "port", fwdPort)
+		app.DelEdge(fwdPort)
 	}()
+
+	// 处理控制命令
+	defer cancel()
 	reader := bufio.NewReader(conn)
 	for {
 		// 循环等待直到服务关闭
@@ -130,7 +126,7 @@ func (s *Service) processCtrlConn(ctx context.Context, conn net.Conn) (err error
 				return fmt.Errorf("读取节点 ID 失败: %w", err)
 			}
 			var client = int32(binary.BigEndian.Uint32(recv))
-			err = s.onOpen(conn, client)
+			fwdPort, err = s.onOpen(ctx, conn, client)
 			if err != nil {
 				return fmt.Errorf("处理连接建立命令失败: %w", err)
 			}
@@ -157,17 +153,16 @@ func (s *Service) processCtrlConn(ctx context.Context, conn net.Conn) (err error
 	}
 }
 
-func (s *Service) onOpen(writer io.Writer, edge int32) (err error) {
+func (s *Service) onOpen(ctx context.Context, writer io.Writer, edge int32) (port uint16, err error) {
 	// open 命令全局只执行一次
 	_, ok := app.Edges.Load(edge)
 	if ok {
-		return fmt.Errorf("节点 ID %d 已经连接", edge)
+		return 0, fmt.Errorf("节点 ID %d 已经连接", edge)
 	}
 
 	// 分配端口
 	var minim uint16 = 20000
 	var maxim uint16 = 60000
-	var port uint16
 	for i := minim; i < maxim; i++ {
 		var _, ok = app.Assigns.Load(i)
 		if !ok {
@@ -177,17 +172,17 @@ func (s *Service) onOpen(writer io.Writer, edge int32) (err error) {
 		}
 	}
 	if port == 0 {
-		return errors.New("没有可用的端口")
+		return 0, errors.New("没有可用的端口")
 	}
 
 	// 报告端口分配
 	if err = report.Assigned(edge, port); err != nil {
-		return fmt.Errorf("报告端口分配失败: %w", err)
+		return 0, fmt.Errorf("报告端口分配失败: %w", err)
 	}
 
 	// 响应节点
 	if err = s.sendPong(writer); err != nil {
-		return fmt.Errorf("响应节点失败: %w", err)
+		return 0, fmt.Errorf("响应节点失败: %w", err)
 	}
 
 	// 启动转发服务
@@ -195,13 +190,13 @@ func (s *Service) onOpen(writer io.Writer, edge int32) (err error) {
 	go func() {
 		defer s.fwdLesWg.Done()
 		slog.Info("监听转发端口", "port", port, "edge", edge)
-		err = s.listenUser(port, writer)
+		err = s.listenUser(ctx, port, writer)
 		if err != nil {
 			slog.Error("监听转发端口失败", "port", port, "edge", edge, "err", err)
 		}
 	}()
 
-	return nil
+	return port, nil
 }
 
 func (s *Service) onPing(writer io.Writer) (err error) {

gateway/fwd/data.go

@@ -19,7 +19,6 @@ import (
 
 func (s *Service) listenData() error {
 	dataPort := env.AppDataPort
-	slog.Debug("监听数据通道", slog.Uint64("port", uint64(dataPort)))
 
 	// 监听端口
 	ls, err := net.Listen("tcp", ":"+strconv.Itoa(int(dataPort)))

gateway/fwd/dispatcher/dispatch.go

@@ -42,7 +42,7 @@ func New(port uint16, readTimeout time.Duration) (*Server, error) {
 	}, nil
 }
 
-func (s *Server) Close() {
+func (s *Server) Stop() {
 	s.cancel()
 }
 

gateway/fwd/fwd.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"log/slog"
 	"proxy-server/gateway/core"
+	"proxy-server/gateway/env"
 	"proxy-server/utils"
 	"sync"
 )
@@ -29,7 +30,7 @@ func New() *Service {
 }
 
 func (s *Service) Run() error {
-	slog.Info("启动转发服务")
+	slog.Info("启动转发服务", "控制通道", env.AppCtrlPort, "数据通道", env.AppDataPort)
 
 	errQuit := make(chan struct{}, 2)
 	defer close(errQuit)

gateway/fwd/user.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/hex"
 	"errors"
+	"fmt"
 	"io"
 	"log/slog"
 	"proxy-server/gateway/core"
@@ -14,25 +15,33 @@ import (
 	"time"
 )
 
-func (s *Service) listenUser(port uint16, ctrl io.Writer) error {
+func (s *Service) listenUser(ctx context.Context, port uint16, ctrl io.Writer) error {
 	dspt, err := dispatcher.New(port, time.Duration(env.AppUserTimeout)*time.Second)
 	if err != nil {
 		return err
 	}
-	defer dspt.Close()
+	defer dspt.Stop()
 
+	var errCh = make(chan error)
 	go func() {
 		err := dspt.Run()
 		if err != nil {
-			slog.Error("代理服务运行失败", "err", err)
+			// slog.Error("代理服务运行失败", "err", err)
+			err = fmt.Errorf("协议嗅探服务运行失败: %w", err)
 		}
+		errCh <- err
 	}()
 
 	// 处理连接
 	for {
 		select {
-		case <-s.ctx.Done():
+		case <-ctx.Done():
 			return nil
+		case err := <-errCh:
+			if err != nil {
+				err = fmt.Errorf("监听转发端口失败: %w", err)
+			}
+			return err
 		case user := <-dspt.Conn:
 			metrics.TimerAuth.Store(user.Conn, time.Now())
 			s.userConnWg.Add(1)