重命名包 client 为 edge；重命名包 server 为 gateway

edge/core/consts.go

@@ -0,0 +1,9 @@
+package core
+
+const Version byte = 1
+
+const FwdCtrlPort uint = 18080
+const FwdDataPort uint = 18081
+
+const RetryInterval uint = 5
+const HeartbeatInterval uint = 30

edge/edge.go

@@ -0,0 +1,296 @@
+package edge
+
+import (
+	"bufio"
+	"context"
+	"encoding/binary"
+	"errors"
+	"fmt"
+	"io"
+	"log/slog"
+	"net"
+	_ "net/http/pprof"
+	"os"
+	"os/signal"
+	"proxy-server/edge/core"
+	"proxy-server/edge/env"
+	"proxy-server/edge/geo"
+	"proxy-server/edge/report"
+	"proxy-server/pkg/utils"
+	"time"
+)
+
+func Start() error {
+
+	// 初始化环境变量
+	slog.Debug("初始化环境变量...")
+	err := env.Init()
+	if err != nil {
+		return fmt.Errorf("初始化环境变量失败: %w", err)
+	}
+
+	// 获取归属地
+	slog.Debug("获取节点归属地...")
+	err = geo.Query()
+	if err != nil {
+		return fmt.Errorf("获取节点归属地失败: %w", err)
+	}
+
+	// 注册节点
+	slog.Debug("注册节点...")
+	id, host, err := report.Online(geo.Prov, geo.City, geo.Isp)
+	if err != nil {
+		return fmt.Errorf("注册节点失败: %w", err)
+	}
+
+	// 连接到网关
+	var ctx, cancel = signal.NotifyContext(context.Background(), os.Interrupt, os.Kill)
+	defer cancel()
+
+	var errCh = make(chan error)
+	go func() {
+		for {
+			err = ctrl(ctx, id, host)
+			if err == nil {
+				errCh <- nil
+				return
+			}
+
+			select {
+			case <-ctx.Done():
+				return
+			default:
+				slog.Error("建立控制通道失败", "err", err)
+				slog.Info(fmt.Sprintf("%d 秒后重试", core.RetryInterval))
+			}
+
+			select {
+			case <-ctx.Done():
+				return
+			case <-time.After(time.Duration(core.RetryInterval) * time.Second):
+			}
+		}
+	}()
+
+	// 等待退出
+	select {
+	case err := <-errCh:
+		if err != nil {
+			slog.Error("控制通道发生错误", "err", err)
+		}
+	}
+
+	// 下线节点
+	slog.Debug("下线节点...")
+	err = report.Offline()
+	if err != nil {
+		slog.Error("下线节点失败", "err", err)
+	}
+
+	return ctx.Err()
+}
+
+func ctrl(ctx context.Context, id int32, host string) error {
+	ctrlAddr := net.JoinHostPort(host, fmt.Sprintf("%d", core.FwdCtrlPort))
+	dataAddr := net.JoinHostPort(host, fmt.Sprintf("%d", core.FwdDataPort))
+
+	slog.Info("建立控制通道", "addr", ctrlAddr)
+	conn, err := net.Dial("tcp", ctrlAddr)
+	if err != nil {
+		return errors.New("连接失败")
+	}
+	defer utils.Close(conn)
+	var reader = bufio.NewReader(conn)
+
+	// 发送节点连接命令
+	err = sendOpen(reader, conn, id)
+	if err != nil {
+		return fmt.Errorf("发送节点信息失败: %w", err)
+	}
+
+	// 异步定时发送心跳
+	go func() {
+		ticker := time.NewTicker(time.Duration(core.HeartbeatInterval) * time.Second)
+		defer ticker.Stop()
+		for {
+			select {
+			case <-ctx.Done():
+				return
+			case tick := <-ticker.C:
+				err := sendPing(reader, conn)
+				if err != nil {
+					slog.Error("发送心跳失败", "time", tick, "err", err)
+				}
+			}
+		}
+	}()
+
+	// 等待用户连接
+	// 读写失败后退出重连，防止后续数据读写顺序错位导致卡死控制通道
+	slog.Info("等待用户连接")
+	for loop := true; loop; {
+		select {
+		case <-ctx.Done():
+			loop = false
+		default:
+			// 接收 dst
+			tag, addr, err := onConn(reader)
+			if err != nil {
+				return fmt.Errorf("接收连接命令失败: %w", err)
+			}
+
+			// 建立数据通道
+			go func() {
+				err := data(dataAddr, addr, tag)
+				if err != nil {
+					slog.Error("建立数据通道失败", "err", err)
+				}
+			}()
+		}
+	}
+
+	// 发送关闭连接（不 return err，否则会重新连接）
+	err = sendClose(reader, conn)
+	if err != nil {
+		slog.Error("发送关闭连接失败", "err", err)
+	}
+
+	return nil
+}
+
+func data(proxy string, dest string, tag [16]byte) error {
+
+	// 向目标地址建立连接
+	var result = 1
+	var dstErr error
+	dst, err := net.Dial("tcp", dest)
+	if err != nil {
+		dstErr = fmt.Errorf("连接目标地址失败: %w", dstErr)
+		result = 0
+	}
+	defer utils.Close(dst)
+
+	// 向服务端建立连接
+	src, err := net.Dial("tcp", proxy)
+	if err != nil {
+		return errors.New("连接服务端失败")
+	}
+	defer utils.Close(src)
+
+	// 发送连接状态
+	var buf = make([]byte, 17)
+	copy(buf[0:16], tag[:])
+	buf[16] = byte(result)
+	_, err = src.Write(buf)
+	if err != nil {
+		return errors.New("发送连接状态失败")
+	}
+
+	if result == 0 {
+		return dstErr
+	}
+
+	go func() {
+		defer utils.Close(dst)
+		_, err := io.Copy(dst, src)
+		if err != nil && !errors.Is(err, net.ErrClosed) {
+			slog.Error("上行流量代理失败", "err", err)
+		}
+	}()
+	go func() {
+		defer utils.Close(src)
+		_, err := io.Copy(src, dst)
+		if err != nil && !errors.Is(err, net.ErrClosed) {
+			slog.Error("下行流量代理失败", "err", err)
+		}
+	}()
+	return nil
+}
+
+func sendOpen(reader io.Reader, writer io.Writer, id int32) error {
+
+	// 发送打开连接
+	var buf = make([]byte, 5)
+	buf[0] = 3
+	binary.BigEndian.PutUint32(buf[1:], uint32(id))
+
+	_, err := writer.Write(buf)
+	if err != nil {
+		return fmt.Errorf("发送打开连接失败: %w", err)
+	}
+
+	// 等待服务端响应
+	respBuf := make([]byte, 1)
+	_, err = io.ReadFull(reader, respBuf)
+	if err != nil {
+		return fmt.Errorf("接收服务端响应失败: %w", err)
+	}
+	if respBuf[0] != 1 {
+		return errors.New("服务端响应失败")
+	}
+
+	return nil
+}
+
+func sendClose(reader io.Reader, writer io.Writer) error {
+	// 发送关闭连接
+	_, err := writer.Write([]byte{4})
+	if err != nil {
+		return err
+	}
+
+	// 等待服务端响应
+	respBuf := make([]byte, 1)
+	_, err = io.ReadFull(reader, respBuf)
+	if err != nil {
+		return fmt.Errorf("接收服务端响应失败: %w", err)
+	}
+	if respBuf[0] != 1 {
+		return errors.New("服务端响应失败")
+	}
+
+	return nil
+}
+
+func sendPing(reader io.Reader, writer io.Writer) error {
+	_, err := writer.Write([]byte{2})
+	if err != nil {
+		return err
+	}
+
+	// 等待服务端响应
+	respBuf := make([]byte, 1)
+	_, err = io.ReadFull(reader, respBuf)
+	if err != nil {
+		return fmt.Errorf("接收服务端响应失败: %w", err)
+	}
+	if respBuf[0] != 1 {
+		return errors.New("服务端响应失败")
+	}
+
+	return nil
+}
+
+func onConn(reader io.Reader) (tag [16]byte, addr string, err error) {
+	var buf = make([]byte, 1+16+2)
+	_, err = io.ReadFull(reader, buf)
+	if err != nil {
+		return [16]byte{}, "", err
+	}
+
+	if buf[0] != 5 {
+		return [16]byte{}, "", errors.New("命令错误")
+	}
+
+	tag = [16]byte(buf[1:17])
+
+	var addrLen = binary.BigEndian.Uint16(buf[17:19])
+	var addrBuf = make([]byte, addrLen)
+	_, err = io.ReadFull(reader, addrBuf)
+	if err != nil {
+		return [16]byte{}, "", err
+	}
+
+	addr = string(addrBuf)
+	return tag, addr, nil
+}

edge/env/env.go

@@ -0,0 +1,53 @@
+package env
+
+import (
+	"errors"
+	"flag"
+	"log/slog"
+)
+
+var Mode = "dev"
+var Name = "dev-edge"
+
+var EndpointOnline = "https://api.lanhuip.com/api/edge/online"
+var EndpointOffline = "https://api.lanhuip.com/api/edge/offline"
+
+func Init() error {
+
+	var env = flag.String("e", "dev", "环境变量，可选值 dev 或 prod")
+	var name = flag.String("n", "", "客户端唯一标识")
+	var online = flag.String("online", "", "服务注册地址")
+	var offline = flag.String("offline", "", "服务注销地址")
+
+	flag.Parse()
+
+	if env != nil && *env != "" {
+		if *env == "dev" || *env == "prod" {
+			Mode = *env
+		} else {
+			return errors.New("环境变量只能为 dev 或 prod")
+		}
+	}
+
+	if name != nil && *name != "" {
+		Name = *name
+	} else {
+		return errors.New("客户端唯一标识不能为空")
+	}
+
+	if online != nil && *online != "" {
+		EndpointOnline = *online
+	}
+
+	if offline != nil && *offline != "" {
+		EndpointOffline = *offline
+	}
+
+	if Mode == "dev" {
+		slog.SetLogLoggerLevel(slog.LevelDebug)
+	} else {
+		slog.SetLogLoggerLevel(slog.LevelWarn)
+	}
+
+	return nil
+}

edge/geo/geo.go

@@ -0,0 +1,159 @@
+package geo
+
+import (
+	"bufio"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"net/textproto"
+	"proxy-server/edge/env"
+	"strings"
+)
+
+var (
+	Ip   string
+	Prov string
+	City string
+	Isp  string
+)
+
+func Query() (err error) {
+
+	switch env.Mode {
+	case "dev":
+		err = dev()
+	default:
+		err = ipapi()
+	}
+	return err
+}
+
+func dev() (err error) {
+	Prov = "河南省"
+	City = "郑州市"
+	Isp = "电信"
+	return nil
+}
+
+func cip() (err error) {
+	//goland:noinspection HttpUrlsUsage
+	const endpoint = "http://cip.cc"
+
+	req, err := http.NewRequest("GET", endpoint, nil)
+	if err != nil {
+		return fmt.Errorf("创建请求失败: %w", err)
+	}
+	req.Header.Set("User-Agent", "curl/8.9.1")
+	req.Header.Set("Accept", "*/*")
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return fmt.Errorf("执行请求失败: %w", err)
+	}
+	defer func(Body io.ReadCloser) {
+		_ = Body.Close()
+	}(resp.Body)
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("状态码: %s", resp.Status)
+	}
+
+	reader := textproto.NewReader(bufio.NewReader(resp.Body))
+	ipLine, err := reader.ReadLine()
+	if err != nil {
+		return fmt.Errorf("读取响应失败: %w", err)
+	}
+	Ip = strings.TrimSpace(strings.Split(ipLine, ":")[1])
+
+	addrLine, err := reader.ReadLine()
+	if err != nil {
+		return fmt.Errorf("读取响应失败: %w", err)
+	}
+	addr := strings.Split(strings.Split(addrLine, ":")[1], " ")
+	Prov = strings.TrimSpace(addr[1])
+	City = strings.TrimSpace(addr[2])
+
+	ispLine, err := reader.ReadLine()
+	if err != nil {
+		return fmt.Errorf("读取响应失败: %w", err)
+	}
+	Isp = strings.TrimSpace(strings.Split(ispLine, ":")[1])
+
+	return nil
+}
+
+func ipapi() (err error) {
+	//goland:noinspection HttpUrlsUsage
+	const endpoint = "http://ip-api.com/json/?fields=regionName,city,as,query&lang=zh-CN"
+
+	resp, err := http.Get(endpoint)
+	if err != nil {
+		return fmt.Errorf("执行请求失败: %w", err)
+	}
+	defer func(Body io.ReadCloser) {
+		_ = Body.Close()
+	}(resp.Body)
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("状态码: %s", resp.Status)
+	}
+
+	var data struct {
+		RegionName string `json:"regionName"`
+		City       string `json:"city"`
+		As         string `json:"as"`
+		Query      string `json:"query"`
+	}
+	err = json.NewDecoder(resp.Body).Decode(&data)
+	if err != nil {
+		return fmt.Errorf("解析响应失败: %w", err)
+	}
+
+	Ip = data.Query
+	Prov = data.RegionName
+	City = data.City
+
+	var telecom = []string{"AS4134", "AS4812", "AS134419", "AS140292"}
+	var unicom = []string{"AS4837", "AS17621", "AS17816"}
+	var mobile = []string{
+		"AS9808", "AS24444", "AS24445", "AS24547", "AS38019",
+		"AS56040", "AS56041", "AS56042", "AS56044", "AS56046", "AS56047",
+		"AS132525", "AS134810",
+	}
+	var foreign = []string{
+		"AS9299",
+	}
+
+	for _, telecomAsn := range telecom {
+		if strings.HasPrefix(data.As, telecomAsn) {
+			Isp = "电信"
+			break
+		}
+	}
+	if Isp == "" {
+		for _, unicomAsn := range unicom {
+			if strings.HasPrefix(data.As, unicomAsn) {
+				Isp = "联通"
+				break
+			}
+		}
+	}
+	if Isp == "" {
+		for _, mobileAsn := range mobile {
+			if strings.HasPrefix(data.As, mobileAsn) {
+				Isp = "移动"
+				break
+			}
+		}
+	}
+	if Isp == "" {
+		for _, foreignAsn := range foreign {
+			if strings.HasPrefix(data.As, foreignAsn) {
+				Isp = "国外"
+				break
+			}
+		}
+	}
+
+	return nil
+}

edge/report/report.go

@@ -0,0 +1,95 @@
+package report
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"net/http"
+	"proxy-server/edge/core"
+	"proxy-server/edge/env"
+	"strings"
+)
+
+func Online(prov, city, isp string) (id int32, host string, err error) {
+
+	var ispInt = 0
+	switch isp {
+	case "电信":
+		ispInt = 1
+	case "联通":
+		ispInt = 2
+	case "移动":
+		ispInt = 3
+	}
+
+	body, err := json.Marshal(map[string]any{
+		"prov":    prov,
+		"city":    city,
+		"isp":     ispInt,
+		"name":    env.Name,
+		"version": core.Version,
+	})
+	if err != nil {
+		return 0, "", err
+	}
+
+	req, err := http.NewRequest("POST", env.EndpointOnline, strings.NewReader(string(body)))
+	if err != nil {
+		return 0, "", fmt.Errorf("创建请求失败: %w", err)
+	}
+
+	req.Header.Set("Content-Type", "application/json")
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return 0, "", fmt.Errorf("执行请求失败: %w", err)
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != http.StatusOK {
+		return 0, "", errors.New("状态码: " + resp.Status)
+	}
+
+	bytes, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return 0, "", fmt.Errorf("读取响应失败: %w", err)
+	}
+	var respBody struct {
+		Id   int32  `json:"id"`
+		Host string `json:"host"`
+	}
+	err = json.Unmarshal(bytes, &respBody)
+	if err != nil {
+		return 0, "", fmt.Errorf("解析响应失败: %w", err)
+	}
+	if respBody.Host == "" {
+		return 0, "", errors.New("响应体为空")
+	}
+
+	return respBody.Id, respBody.Host, nil
+}
+
+func Offline() error {
+	var bytes, err = json.Marshal(map[string]any{
+		"name": env.Name,
+	})
+	if err != nil {
+		return err
+	}
+	var body = strings.NewReader(string(bytes))
+
+	req, err := http.NewRequest("POST", env.EndpointOffline, body)
+	if err != nil {
+		return fmt.Errorf("创建请求失败: %w", err)
+	}
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return fmt.Errorf("执行请求失败: %w", err)
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != http.StatusOK {
+		return errors.New("状态码: " + resp.Status)
+	}
+
+	return nil
+}