github/nps
1
0
Fork 0
mirror of https://github.com/ehang-io/nps.git synced 2025-09-02 11:56:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

P2p first version

Browse Source
This commit is contained in:
刘河
2019-02-26 22:40:28 +08:00
parent 204c53ddd3
commit 534d428c6d
23 changed files with 986 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

81
client/local.go
View File

@@ -3,8 +3,10 @@ package client
import (
"github.com/cnlh/nps/lib/common"
"github.com/cnlh/nps/lib/config"
"github.com/cnlh/nps/lib/conn"
"github.com/cnlh/nps/lib/crypt"
"github.com/cnlh/nps/vender/github.com/astaxie/beego/logs"
"github.com/cnlh/nps/vender/github.com/xtaci/kcp"
"net"
"strings"
)
@@ -39,16 +41,81 @@ func StartLocalServer(l *config.LocalServer, config *config.CommonConfig) error
return nil
}
func process(conn net.Conn, config *config.CommonConfig, l *config.LocalServer) {
c, err := NewConn(config.Tp, config.VKey, config.Server, common.WORD_SECRET, config.ProxyUrl)
func process(localTcpConn net.Conn, config *config.CommonConfig, l *config.LocalServer) {
var workType string
if l.Type == "secret" {
workType = common.WORK_SECRET
} else {
workType = common.WORK_P2P
}
remoteConn, err := NewConn(config.Tp, config.VKey, config.Server, workType, config.ProxyUrl)
if err != nil {
logs.Error("Local connection server failed ", err.Error())
}
if _, err := c.Write([]byte(crypt.Md5(l.Password))); err != nil {
if _, err := remoteConn.Write([]byte(crypt.Md5(l.Password))); err != nil {
logs.Error("Local connection server failed ", err.Error())
}
go common.CopyBuffer(c, conn)
common.CopyBuffer(conn, c)
c.Close()
conn.Close()
if l.Type == "secret" {
go common.CopyBuffer(remoteConn, localTcpConn)
common.CopyBuffer(localTcpConn, remoteConn)
remoteConn.Close()
localTcpConn.Close()
} else {
//读取服务端地址、密钥 继续做处理
logs.Warn(111)
if rAddr, err := remoteConn.GetLenContent(); err != nil {
return
} else {
logs.Warn(222)
//与服务端udp建立连接
tmpConn, err := net.Dial("udp", "114.114.114.114:53")
if err != nil {
logs.Warn(err)
}
tmpConn.Close()
//与服务端建立udp连接
localAddr, _ := net.ResolveUDPAddr("udp", tmpConn.LocalAddr().String())
localConn, err := net.ListenUDP("udp", localAddr)
if err != nil {
return
}
logs.Warn(333)
localKcpConn, err := kcp.NewConn(string(rAddr), nil, 150, 3, localConn)
conn.SetUdpSession(localKcpConn)
if err != nil {
logs.Warn(err)
}
localToolConn := conn.NewConn(localKcpConn)
//写入密钥、provider身份
if _, err := localToolConn.Write([]byte(crypt.Md5(l.Password))); err != nil {
return
}
if _, err := localToolConn.Write([]byte(common.WORK_P2P_VISITOR)); err != nil {
return
}
logs.Warn(444)
//接收服务端传的visitor地址
if b, err := localToolConn.GetLenContent(); err != nil {
logs.Warn(err)
return
} else {
logs.Warn("收到服务回传地址", string(b))
//关闭与服务端连接
localConn.Close()
//建立新的连接
localConn, err = net.ListenUDP("udp", localAddr)
udpTunnel, err := kcp.NewConn(string(b), nil, 150, 3, localConn)
if err != nil || udpTunnel == nil {
logs.Warn(err)
return
}
conn.SetUdpSession(udpTunnel)
logs.Warn(udpTunnel.RemoteAddr(), string(b), udpTunnel.LocalAddr())
go common.CopyBuffer(udpTunnel, localTcpConn)
common.CopyBuffer(localTcpConn, udpTunnel)
}
}
}
}