mirror of
https://github.com/e1732a364fed/v2ray_simple.git
synced 2025-11-01 12:32:49 +08:00
354 lines
7.4 KiB
Go
354 lines
7.4 KiB
Go
package vless
|
||
|
||
import (
|
||
"bufio"
|
||
"bytes"
|
||
"io"
|
||
"log"
|
||
"net"
|
||
"net/url"
|
||
"strconv"
|
||
"sync"
|
||
|
||
"github.com/hahahrfool/v2ray_simple/netLayer"
|
||
"github.com/hahahrfool/v2ray_simple/proxy"
|
||
"github.com/hahahrfool/v2ray_simple/utils"
|
||
)
|
||
|
||
func init() {
|
||
proxy.RegisterClient(Name, ClientCreator{})
|
||
}
|
||
|
||
//实现 proxy.UserClient，以及 netLayer.UDP_Putter
|
||
type Client struct {
|
||
proxy.ProxyCommonStruct
|
||
|
||
udpResponseChan chan netLayer.UDPAddrData
|
||
|
||
version int
|
||
|
||
user *proxy.V2rayUser
|
||
|
||
is_CRUMFURS_established bool
|
||
|
||
mutex sync.RWMutex
|
||
knownUDPDestinations map[string]io.ReadWriter
|
||
|
||
crumfursBuf *bufio.Reader //在不使用ws或者grpc时，需要一个缓存 来读取 CRUMFURS
|
||
}
|
||
|
||
type ClientCreator struct{}
|
||
|
||
func (_ ClientCreator) NewClientFromURL(u *url.URL) (proxy.Client, error) {
|
||
return NewClientByURL(u)
|
||
}
|
||
|
||
func (_ ClientCreator) NewClient(dc *proxy.DialConf) (proxy.Client, error) {
|
||
|
||
uuidStr := dc.Uuid
|
||
id, err := proxy.NewV2rayUser(uuidStr)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
c := Client{
|
||
user: id,
|
||
}
|
||
|
||
c.knownUDPDestinations = make(map[string]io.ReadWriter)
|
||
c.udpResponseChan = make(chan netLayer.UDPAddrData, 20)
|
||
|
||
v := dc.Version
|
||
if v >= 0 {
|
||
|
||
if v == 1 {
|
||
c.version = 1
|
||
}
|
||
|
||
}
|
||
|
||
return &c, nil
|
||
}
|
||
|
||
func NewClientByURL(url *url.URL) (proxy.Client, error) {
|
||
uuidStr := url.User.Username()
|
||
id, err := proxy.NewV2rayUser(uuidStr)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
c := Client{
|
||
user: id,
|
||
}
|
||
c.knownUDPDestinations = make(map[string]io.ReadWriter)
|
||
c.udpResponseChan = make(chan netLayer.UDPAddrData, 20)
|
||
|
||
vStr := url.Query().Get("version")
|
||
if vStr != "" {
|
||
v, err := strconv.Atoi(vStr)
|
||
if err == nil {
|
||
if v == 1 {
|
||
c.version = 1
|
||
}
|
||
}
|
||
}
|
||
|
||
return &c, nil
|
||
}
|
||
|
||
func (c *Client) Name() string { return Name }
|
||
func (c *Client) Version() int { return c.version }
|
||
func (c *Client) GetUser() proxy.User {
|
||
return c.user
|
||
}
|
||
|
||
func (c *Client) Handshake(underlay net.Conn, target *netLayer.Addr) (io.ReadWriteCloser, error) {
|
||
var err error
|
||
|
||
if underlay == nil {
|
||
underlay, err = target.Dial() //不建议传入underlay为nil，因为这里dial处于裸奔状态
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
port := target.Port
|
||
addr, atyp := target.AddressBytes()
|
||
|
||
cmd := CmdTCP
|
||
if target.Network == "udp" {
|
||
if c.version == 1 && !c.is_CRUMFURS_established {
|
||
|
||
//log.Println("尝试拨号 Cmd_CRUMFURS 信道")
|
||
|
||
UMFURS_conn, err := target.Dial()
|
||
if err != nil {
|
||
if utils.CanLogErr() {
|
||
|
||
log.Println("尝试拨号 Cmd_CRUMFURS 信道时发生错误")
|
||
}
|
||
return nil, err
|
||
}
|
||
buf := c.getBufWithCmd(Cmd_CRUMFURS)
|
||
|
||
UMFURS_conn.Write(buf.Bytes())
|
||
|
||
utils.PutBuf(buf)
|
||
|
||
bs := []byte{0}
|
||
n, err := UMFURS_conn.Read(bs)
|
||
if err != nil || n == 0 || bs[0] != CRUMFURS_ESTABLISHED {
|
||
if utils.CanLogErr() {
|
||
|
||
log.Println("尝试读取 Cmd_CRUMFURS 信道返回值 时发生错误")
|
||
}
|
||
return nil, err
|
||
}
|
||
|
||
c.is_CRUMFURS_established = true
|
||
|
||
// 循环监听 UMFURS 信息
|
||
go c.handle_CRUMFURS(UMFURS_conn)
|
||
|
||
}
|
||
|
||
cmd = CmdUDP
|
||
|
||
}
|
||
|
||
buf := c.getBufWithCmd(cmd)
|
||
|
||
buf.WriteByte(byte(uint16(port) >> 8))
|
||
buf.WriteByte(byte(uint16(port) << 8 >> 8))
|
||
|
||
buf.WriteByte(atyp)
|
||
buf.Write(addr)
|
||
|
||
_, err = underlay.Write(buf.Bytes())
|
||
|
||
utils.PutBuf(buf)
|
||
|
||
return &UserConn{
|
||
Conn: underlay,
|
||
uuid: *c.user,
|
||
version: c.version,
|
||
isUDP: target.IsUDP(),
|
||
underlayIsBasic: netLayer.IsBasicConn(underlay),
|
||
}, err
|
||
}
|
||
|
||
func (c *Client) GetNewUDPResponse() (net.UDPAddr, []byte, error) {
|
||
x := <-c.udpResponseChan //v1的话，由 handle_CRUMFURS 以及 WriteUDPRequest 中的 goroutine 填充；v0的话，由 WriteUDPRequest 填充
|
||
return x.Addr, x.Data, nil
|
||
}
|
||
|
||
//一般由socks5或者透明代理等地方 获取到 udp请求后，被传入这里
|
||
func (c *Client) WriteUDPRequest(a net.UDPAddr, b []byte, dialFunc func(targetAddr netLayer.Addr) (io.ReadWriter, error)) (err error) {
|
||
|
||
astr := a.String()
|
||
|
||
c.mutex.RLock()
|
||
knownConn := c.knownUDPDestinations[astr]
|
||
c.mutex.RUnlock()
|
||
|
||
if knownConn == nil {
|
||
|
||
knownConn, err = dialFunc(netLayer.NewAddrFromUDPAddr(&a))
|
||
if err != nil || knownConn == nil {
|
||
return utils.ErrInErr{ErrDesc: "vless WriteUDPRequest, err when creating an underlay", ErrDetail: err}
|
||
}
|
||
//这里原来的代码是调用 c.Handshake，会自动帮我们拨号代理节点CmdUDP
|
||
// 但是似乎有问题，因为不应该由client自己拨号vless，因为我们还有上层的tls;
|
||
// 自己拨号的话，那就是裸奔状态
|
||
// 最新代码采用dialFunc的方式解决
|
||
//knownConn, err = c.Handshake(newUnderlay, netLayer.NewAddrFromUDPAddr(a))
|
||
|
||
c.mutex.Lock()
|
||
c.knownUDPDestinations[astr] = knownConn
|
||
c.mutex.Unlock()
|
||
|
||
go func() {
|
||
bs := make([]byte, netLayer.MaxUDP_packetLen)
|
||
for {
|
||
n, err := knownConn.Read(bs)
|
||
if err != nil {
|
||
break
|
||
}
|
||
if n <= 0 {
|
||
continue
|
||
}
|
||
msg := make([]byte, n)
|
||
copy(msg, bs[:n])
|
||
|
||
c.udpResponseChan <- netLayer.UDPAddrData{
|
||
Addr: a,
|
||
Data: msg,
|
||
}
|
||
}
|
||
}()
|
||
}
|
||
|
||
_, err = knownConn.Write(b)
|
||
return
|
||
}
|
||
|
||
func (c *Client) getBufWithCmd(cmd byte) *bytes.Buffer {
|
||
v := c.version
|
||
buf := utils.GetBuf()
|
||
buf.WriteByte(byte(v)) //version
|
||
buf.Write(c.user[:])
|
||
if v == 0 {
|
||
buf.WriteByte(0) //addon length
|
||
}
|
||
buf.WriteByte(cmd) // cmd
|
||
return buf
|
||
}
|
||
|
||
// 把在 CRUMFURS 信道中 获取到的 未知流量 转发到 UDPResponseWriter （本作中就是 转发到 inServer 中, 而且 只有 socks5 这一种 inServer 实现了该方法， 见 main.go)
|
||
func (c *Client) handle_CRUMFURS(UMFURS_conn net.Conn) {
|
||
|
||
if c.udpResponseChan == nil {
|
||
c.is_CRUMFURS_established = false
|
||
return
|
||
}
|
||
|
||
for {
|
||
//之前讨论了，udp信息通通要传长度头，CRUMFURS 也不例外，在 没有AdvancedLayer时，统一都要加udp长度头
|
||
|
||
if c.AdvancedL != "" {
|
||
|
||
buf_for_umfurs := utils.GetPacket()
|
||
n, err := UMFURS_conn.Read(buf_for_umfurs)
|
||
if err != nil {
|
||
break
|
||
}
|
||
if n < 7 {
|
||
|
||
break
|
||
}
|
||
msg := buf_for_umfurs[:n]
|
||
atyp := msg[0]
|
||
portIndex := net.IPv4len
|
||
switch atyp {
|
||
case netLayer.AtypIP6:
|
||
portIndex = net.IPv6len
|
||
default:
|
||
//不合法，必须是ipv4或者ipv6
|
||
break
|
||
|
||
}
|
||
theIP := make(net.IP, portIndex)
|
||
copy(theIP, msg[1:portIndex])
|
||
|
||
port := int16(msg[portIndex])<<8 + int16(msg[portIndex+1])
|
||
|
||
c.udpResponseChan <- netLayer.UDPAddrData{
|
||
Addr: net.UDPAddr{
|
||
IP: theIP,
|
||
Port: int(port),
|
||
},
|
||
Data: msg[portIndex+2:],
|
||
}
|
||
} else {
|
||
if c.crumfursBuf == nil {
|
||
c.crumfursBuf = bufio.NewReader(UMFURS_conn)
|
||
}
|
||
|
||
atyp, err := c.crumfursBuf.ReadByte()
|
||
if err != nil {
|
||
break
|
||
}
|
||
|
||
ipLen := net.IPv4len
|
||
switch atyp {
|
||
case netLayer.AtypIP6:
|
||
ipLen = net.IPv6len
|
||
default:
|
||
//不合法，必须是ipv4或者ipv6
|
||
break
|
||
}
|
||
|
||
theIP := make(net.IP, ipLen)
|
||
_, err = c.crumfursBuf.Read(theIP)
|
||
if err != nil {
|
||
break
|
||
}
|
||
|
||
twoBytes, err := c.crumfursBuf.Peek(2)
|
||
if err != nil {
|
||
break
|
||
}
|
||
|
||
port := int(int16(twoBytes[0])<<8 + int16(twoBytes[1]))
|
||
|
||
c.crumfursBuf.Discard(2)
|
||
|
||
twoBytes, err = c.crumfursBuf.Peek(2)
|
||
if err != nil {
|
||
break
|
||
}
|
||
|
||
packetLen := int16(twoBytes[0])<<8 + int16(twoBytes[1])
|
||
c.crumfursBuf.Discard(2)
|
||
|
||
msg := make([]byte, packetLen)
|
||
|
||
_, err = io.ReadFull(c.crumfursBuf, msg)
|
||
if err != nil {
|
||
break
|
||
}
|
||
|
||
c.udpResponseChan <- netLayer.UDPAddrData{
|
||
Addr: net.UDPAddr{
|
||
IP: theIP,
|
||
Port: port,
|
||
},
|
||
Data: msg,
|
||
}
|
||
|
||
}
|
||
}
|
||
|
||
c.is_CRUMFURS_established = false
|
||
}
|