客户端服务端分离

2025-09-01 02:46:52 +00:00 · 2019-01-09 20:33:00 +08:00
parent dcd21f211d
commit 1f61b99387
46 changed files with 1062 additions and 1431 deletions
--- a/client/client.go
+++ b/client/client.go
@@ -0,0 +1,128 @@
+package client
+
+import (
+	"github.com/cnlh/easyProxy/utils"
+	"log"
+	"net"
+	"sync"
+	"time"
+)
+
+type TRPClient struct {
+	svrAddr string
+	tcpNum  int
+	sync.Mutex
+	vKey string
+}
+
+//new client
+func NewRPClient(svraddr string, tcpNum int, vKey string) *TRPClient {
+	c := new(TRPClient)
+	c.svrAddr = svraddr
+	c.tcpNum = tcpNum
+	c.vKey = vKey
+	return c
+}
+
+//start
+func (s *TRPClient) Start() error {
+	for i := 0; i < s.tcpNum; i++ {
+		go s.NewConn()
+	}
+	return nil
+}
+
+//新建
+func (s *TRPClient) NewConn() error {
+	s.Lock()
+	conn, err := net.Dial("tcp", s.svrAddr)
+	if err != nil {
+		log.Println("连接服务端失败,五秒后将重连")
+		time.Sleep(time.Second * 5)
+		s.Unlock()
+		go s.NewConn()
+		return err
+	}
+	s.Unlock()
+	return s.process(utils.NewConn(conn))
+}
+
+//处理
+func (s *TRPClient) process(c *utils.Conn) error {
+	c.SetAlive()
+	if _, err := c.Write([]byte(utils.Getverifyval(s.vKey))); err != nil {
+		return err
+	}
+	c.WriteMain()
+	for {
+		flags, err := c.ReadFlag()
+		if err != nil {
+			log.Println("服务端断开,五秒后将重连", err)
+			time.Sleep(5 * time.Second)
+			go s.NewConn()
+			break
+		}
+		switch flags {
+		case utils.VERIFY_EER:
+			log.Fatalln("vkey:", s.vKey, "不正确,服务端拒绝连接,请检查")
+		case utils.WORK_CHAN: //隧道模式，每次开启10个，加快连接速度
+			for i := 0; i < 5; i++ {
+				go s.dealChan()
+			}
+		case utils.RES_MSG:
+			log.Println("服务端返回错误。")
+		default:
+			log.Println("无法解析该错误。", flags)
+		}
+	}
+	return nil
+}
+
+//隧道模式处理
+func (s *TRPClient) dealChan() {
+	var err error
+	//创建一个tcp连接
+	conn, err := net.Dial("tcp", s.svrAddr)
+	if err != nil {
+		log.Println("connect to ", s.svrAddr, "error:", err)
+		return
+	}
+	//验证
+	if _, err := conn.Write([]byte(utils.Getverifyval(s.vKey))); err != nil {
+		log.Println("connect to ", s.svrAddr, "error:", err)
+		return
+	}
+	//默认长连接保持
+	c := utils.NewConn(conn)
+	c.SetAlive()
+	//写标志
+	c.WriteChan()
+re:
+	//获取连接的host type(tcp or udp)
+	typeStr, host, en, de, crypt, mux, err := c.GetHostFromConn()
+	if err != nil {
+		log.Println("get host info error:", err)
+		c.Close()
+		return
+	}
+	Process(c, typeStr, host, en, de, crypt, mux)
+	if mux {
+		utils.FlushConn(conn)
+		goto re
+	} else {
+		c.Close()
+	}
+}
+
+func Process(c *utils.Conn, typeStr, host string, en, de int, crypt, mux bool) {
+	//与目标建立连接,超时时间为3
+	server, err := net.DialTimeout(typeStr, host, time.Second*3)
+	if err != nil {
+		log.Println("connect to ", host, "error:", err, mux)
+		c.WriteFail()
+		return
+	}
+	c.WriteSuccess()
+	go utils.Relay(server, c.Conn, de, crypt, mux)
+	utils.Relay(c.Conn, server, en, crypt, mux)
+}