完善客户端与服务端注册，端口分配和协议交互逻辑

client/client.go

@@ -2,42 +2,43 @@ package client
 
 import (
 	"bufio"
+	"encoding/binary"
 	"fmt"
 	"io"
 	"log/slog"
 	"net"
-	"os"
 	"proxy-server/client/core"
+	"proxy-server/client/env"
 	"proxy-server/client/geo"
 	"proxy-server/client/report"
 	"proxy-server/pkg/utils"
 	"time"
 
 	"errors"
-	"github.com/joho/godotenv"
-
 	_ "net/http/pprof"
 )
 
-var Geo geo.Func = geo.Ipapi
-
 func Start() error {
 
 	// 初始化环境变量
-	slog.SetLogLoggerLevel(slog.LevelDebug)
-
-	err := godotenv.Load()
+	slog.Debug("初始化环境变量...")
+	err := env.Init()
 	if err != nil {
-		slog.Debug("没有本地环境变量文件")
-	} else {
-		online := os.Getenv("ENDPOINT_ONLINE")
-		if online != "" {
-			core.EndpointOnline = online
-		}
-		offline := os.Getenv("ENDPOINT_OFFLINE")
-		if offline != "" {
-			core.EndpointOffline = offline
-		}
+		return fmt.Errorf("初始化环境变量失败: %w", err)
+	}
+
+	// 获取归属地
+	slog.Debug("获取节点归属地...")
+	err = geo.Query()
+	if err != nil {
+		slog.Error("获取归属地失败", "err", err)
+	}
+
+	// 注册节点
+	slog.Debug("注册节点...")
+	id, host, err := report.Online(geo.Prov, geo.City, geo.Isp)
+	if err != nil {
+		return fmt.Errorf("注册节点失败: %w", err)
 	}
 
 	// 性能监控
@@ -49,24 +50,9 @@ func Start() error {
 	// 	}
 	// }()
 
-	// 获取归属地
-	slog.Debug("获取节点归属地...")
-	prov, city, isp, err := Geo()
-	if err != nil {
-		slog.Error("获取归属地失败", "err", err)
-	}
-
-	// 注册节点
-	slog.Debug("注册节点...")
-	host, err := report.Online(prov, city, isp)
-	if err != nil {
-		slog.Error("节点注册失败", "err", err)
-		return err
-	}
-
 	// 建立控制通道
 	for {
-		err := ctrl(host)
+		err := ctrl(id, host)
 		if err != nil {
 			slog.Error("建立控制通道失败", "err", err)
 			slog.Info(fmt.Sprintf("%d 秒后重试", core.RetryInterval))
@@ -75,7 +61,7 @@ func Start() error {
 	}
 }
 
-func ctrl(host string) error {
+func ctrl(id int32, host string) error {
 	ctrlAddr := net.JoinHostPort(host, fmt.Sprintf("%d", core.FwdCtrlPort))
 	dataAddr := net.JoinHostPort(host, fmt.Sprintf("%d", core.FwdDataPort))
 
@@ -86,25 +72,19 @@ func ctrl(host string) error {
 	}
 	defer utils.Close(conn)
 
-	reader := bufio.NewReader(conn)
-
-	// 请求转发端口
-	_, err = conn.Write([]byte{core.Version})
+	// 发送客户端信息
+	var buf = make([]byte, 4)
+	_, err = binary.Encode(buf, binary.BigEndian, id)
 	if err != nil {
-		return errors.New("发送版本号失败")
+		return fmt.Errorf("编码客户端 ID 失败: %w", err)
 	}
-
-	// 发送客户端名称
-	nameLen := byte(len(core.Name))
-	nameBuf := make([]byte, 1+nameLen)
-	nameBuf[0] = nameLen
-	copy(nameBuf[1:], core.Name)
-	_, err = conn.Write(nameBuf)
+	_, err = conn.Write(buf)
 	if err != nil {
-		return errors.New("发送 name 失败")
+		return fmt.Errorf("发送客户端 ID 失败: %w", err)
 	}
 
 	// 等待服务端响应
+	reader := bufio.NewReader(conn)
 	respBuf, err := reader.ReadByte()
 	if err != nil {
 		return errors.New("接收响应失败")