优化项目机构和服务端协程控制逻辑

server/pkg/socks5/auth.go

@@ -0,0 +1,82 @@
+package socks5
+
+import (
+	"context"
+	"io"
+	"log/slog"
+	"proxy-server/pkg/utils"
+	"slices"
+
+	"github.com/pkg/errors"
+)
+
+type AuthMethod byte
+
+const (
+	AuthVersion  = byte(1)
+	AuthSuccess  = byte(0)
+	AuthFailure  = byte(1)
+	NoAuth       = AuthMethod(0)
+	UserPassAuth = AuthMethod(2)
+	NoAcceptable = byte(0xFF)
+)
+
+type Authenticator interface {
+	Method() AuthMethod
+	Authenticate(ctx context.Context, reader io.Reader, writer io.Writer) (*AuthContext, error)
+}
+
+// authenticate 执行认证流程
+func (server *Server) authenticate(reader io.Reader, writer io.Writer) (*AuthContext, error) {
+
+	// 版本检查
+	err := checkVersion(reader)
+	if err != nil {
+		return nil, err
+	}
+
+	// 获取客户端认证方式
+	nAuth, err := utils.ReadByte(reader)
+	if err != nil {
+		return nil, err
+	}
+	methods, err := utils.ReadBuffer(reader, int(nAuth))
+	if err != nil {
+		return nil, err
+	}
+
+	// 认证客户端连接
+	for _, authenticator := range server.config.AuthMethods {
+		method := authenticator.Method()
+		if slices.Contains(methods, byte(method)) {
+			slog.Debug("使用的认证方式", method)
+
+			_, err := writer.Write([]byte{SocksVersion, byte(method)})
+			if err != nil {
+				slog.Error("响应认证方式失败", err)
+				return nil, err
+			}
+
+			ctx := context.WithValue(context.Background(), "service", server)
+			authContext, err := authenticator.Authenticate(ctx, reader, writer)
+			if err != nil {
+				return nil, err
+			}
+			return authContext, nil
+		}
+	}
+
+	// 无适用的认证方式
+	_, err = writer.Write([]byte{SocksVersion, NoAcceptable})
+	if err != nil {
+		return nil, err
+	}
+
+	return nil, errors.New("没有适用的认证方式")
+}
+
+type AuthContext struct {
+	Method  AuthMethod
+	Timeout uint
+	Payload map[string]any
+}