mirror of https://github.com/ehang-io/nps
148 lines
3.0 KiB
Go
Executable File
148 lines
3.0 KiB
Go
Executable File
package lib
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"log"
|
||
"net"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
type TRPClient struct {
|
||
svrAddr string
|
||
tcpNum int
|
||
sync.Mutex
|
||
vKey string
|
||
}
|
||
|
||
func NewRPClient(svraddr string, tcpNum int, vKey string) *TRPClient {
|
||
c := new(TRPClient)
|
||
c.svrAddr = svraddr
|
||
c.tcpNum = tcpNum
|
||
c.vKey = vKey
|
||
return c
|
||
}
|
||
|
||
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(NewConn(conn))
|
||
}
|
||
|
||
func (s *TRPClient) process(c *Conn) error {
|
||
c.SetAlive()
|
||
if _, err := c.Write([]byte(getverifyval(s.vKey))); err != nil {
|
||
return err
|
||
}
|
||
c.wMain()
|
||
for {
|
||
flags, err := c.ReadFlag()
|
||
if err != nil {
|
||
log.Println("服务端断开,五秒后将重连", err)
|
||
time.Sleep(5 * time.Second)
|
||
go s.newConn()
|
||
break
|
||
}
|
||
switch flags {
|
||
case VERIFY_EER:
|
||
log.Fatalln("vkey:", s.vKey, "不正确,服务端拒绝连接,请检查")
|
||
case RES_SIGN: //代理请求模式
|
||
if err := s.dealHttp(c); err != nil {
|
||
log.Println(err)
|
||
return err
|
||
}
|
||
case WORK_CHAN: //隧道模式，每次开启10个，加快连接速度
|
||
for i := 0; i < 10; i++ {
|
||
go s.dealChan()
|
||
}
|
||
case RES_MSG:
|
||
log.Println("服务端返回错误。")
|
||
default:
|
||
log.Println("无法解析该错误。", flags)
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
//隧道模式处理
|
||
func (s *TRPClient) dealChan() error {
|
||
//创建一个tcp连接
|
||
conn, err := net.Dial("tcp", s.svrAddr)
|
||
if err != nil {
|
||
log.Println("connect to ", s.svrAddr, "error:", err)
|
||
return err
|
||
}
|
||
//验证
|
||
if _, err := conn.Write([]byte(getverifyval(s.vKey))); err != nil {
|
||
log.Println("connect to ", s.svrAddr, "error:", err)
|
||
return err
|
||
}
|
||
//默认长连接保持
|
||
c := NewConn(conn)
|
||
c.SetAlive()
|
||
//写标志
|
||
c.wChan()
|
||
//获取连接的host type(tcp or udp)
|
||
typeStr, host, en, de, crypt, err := c.GetHostFromConn()
|
||
if err != nil {
|
||
log.Println("get host info error:", err)
|
||
return err
|
||
}
|
||
//与目标建立连接
|
||
server, err := net.Dial(typeStr, host)
|
||
if err != nil {
|
||
log.Println("connect to ", host, "error:", err)
|
||
return err
|
||
}
|
||
go relay(NewConn(server), c, de, crypt)
|
||
relay(c, NewConn(server), en, crypt)
|
||
return nil
|
||
}
|
||
|
||
//http模式处理
|
||
func (s *TRPClient) dealHttp(c *Conn) error {
|
||
buf := make([]byte, 1024*32)
|
||
en, de, _ := c.GetConnInfoFromConn()
|
||
n, err := c.ReadFromCompress(buf, de)
|
||
if err != nil {
|
||
c.wError()
|
||
return err
|
||
}
|
||
req, err := DecodeRequest(buf[:n])
|
||
if err != nil {
|
||
c.wError()
|
||
return err
|
||
}
|
||
respBytes, err := GetEncodeResponse(req)
|
||
if err != nil {
|
||
c.wError()
|
||
return err
|
||
}
|
||
c.wSign()
|
||
n, err = c.WriteCompress(respBytes, en)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if n != len(respBytes) {
|
||
return errors.New(fmt.Sprintf("发送数据长度错误，已经发送：%dbyte，总字节长：%dbyte\n", n, len(respBytes)))
|
||
}
|
||
return nil
|
||
}
|