mirror of
https://github.com/e1732a364fed/v2ray_simple.git
synced 2025-10-30 11:36:35 +08:00
f2adcdcd45
修订代码,文档,示例 添加 NumErr 结构; 从 handshakeInserver_and_passToOutClient 函数 分离出一个 dialClient 函数。 在socks5包中添加 client.go 文件，以及三个udp相关的客户端请求udp函数 之前的udp associate代码被证明是有很多bug的，现在被我一一修复，并通过了 udp_test.go的测试。
365 lines
8.4 KiB
Go
365 lines
8.4 KiB
Go
// Package vless provies vless proxy support for proxy.Client and proxy.Server
|
||
package vless
|
||
|
||
import (
|
||
"bufio"
|
||
"bytes"
|
||
"errors"
|
||
"io"
|
||
"net"
|
||
|
||
"github.com/hahahrfool/v2ray_simple/netLayer"
|
||
"github.com/hahahrfool/v2ray_simple/utils"
|
||
)
|
||
|
||
const Name = "vless"
|
||
|
||
const (
|
||
CRUMFURS_ESTABLISHED byte = 20
|
||
|
||
CRUMFURS_Established_Str = "CRUMFURS_Established"
|
||
)
|
||
|
||
// CMD types, for vless and vmess
|
||
const (
|
||
_ byte = iota
|
||
CmdTCP
|
||
CmdUDP
|
||
CmdMux
|
||
Cmd_CRUMFURS //byte = 4 // start from vless v1
|
||
|
||
)
|
||
|
||
//实现 net.Conn, io.ReaderFrom, utils.MultiWriter, netLayer.Splicer
|
||
type UserConn struct {
|
||
net.Conn
|
||
optionalReader io.Reader //在使用了缓存读取握手包头后，就产生了buffer中有剩余数据的可能性，此时就要使用MultiReader
|
||
|
||
remainFirstBufLen int //记录读取握手包头时读到的buf的长度. 如果我们读超过了这个部分的话,实际上我们就可以不再使用 optionalReader 读取, 而是直接从Conn读取
|
||
|
||
underlayIsBasic bool
|
||
|
||
uuid [16]byte
|
||
convertedUUIDStr string
|
||
version int
|
||
isUDP bool
|
||
isServerEnd bool //for v0
|
||
|
||
// udpUnreadPart 不为空，则表示上一次读取没读完整个包（给Read传入的buf太小），须接着读
|
||
udpUnreadPart []byte //for udp
|
||
|
||
bufr *bufio.Reader //for udp
|
||
isntFirstPacket bool //for v0
|
||
|
||
hasAdvancedLayer bool //for v1, 在用ws或grpc时，这个开关保持打开
|
||
}
|
||
|
||
func (uc *UserConn) GetProtocolVersion() int {
|
||
return uc.version
|
||
}
|
||
func (uc *UserConn) GetIdentityStr() string {
|
||
if uc.convertedUUIDStr == "" {
|
||
uc.convertedUUIDStr = utils.UUIDToStr(uc.uuid)
|
||
}
|
||
|
||
return uc.convertedUUIDStr
|
||
}
|
||
|
||
//当前连接状态是否可以直接写入底层Conn而不经任何改动/包装
|
||
func (c *UserConn) canDirectWrite() bool {
|
||
return c.version == 1 && !c.isUDP || c.version == 0 && !(c.isServerEnd && !c.isntFirstPacket)
|
||
}
|
||
|
||
func (c *UserConn) EverPossibleToSplice() bool {
|
||
if c.isUDP {
|
||
return false //udp 在底层就是无法splice的，没办法
|
||
}
|
||
if netLayer.IsBasicConn(c.Conn) {
|
||
return true
|
||
}
|
||
if s, ok := c.Conn.(netLayer.Splicer); ok {
|
||
return s.EverPossibleToSplice()
|
||
}
|
||
return false
|
||
}
|
||
|
||
func (c *UserConn) CanSplice() (r bool, conn net.Conn) {
|
||
|
||
if c.isUDP {
|
||
return
|
||
}
|
||
|
||
if !c.canDirectWrite() {
|
||
return
|
||
}
|
||
|
||
if netLayer.IsBasicConn(c.Conn) {
|
||
r = true
|
||
conn = c.Conn
|
||
|
||
} else if s, ok := c.Conn.(netLayer.Splicer); ok {
|
||
r, conn = s.CanSplice()
|
||
}
|
||
|
||
return
|
||
}
|
||
|
||
func (c *UserConn) WriteBuffers(buffers [][]byte) (int64, error) {
|
||
|
||
if c.canDirectWrite() {
|
||
|
||
//底层连接可以是 ws，或者 tls，或者 基本连接; tls 我们暂不支持 utils.MultiWriter
|
||
// 理论上tls是可以支持的，但是要我们魔改tls库
|
||
|
||
//本作的 ws.Conn 实现了 utils.MultiWriter
|
||
|
||
if c.underlayIsBasic {
|
||
//如果是基本Conn，则不用担心 WriteTo篡改buffers的问题, 因为它会直接调用底层 writev
|
||
//nb := net.Buffers(buffers)
|
||
//return nb.WriteTo(c.Conn) //发现它还是会篡改？？什么鬼
|
||
return utils.BuffersWriteTo(buffers, c.Conn)
|
||
|
||
} else if mr, ok := c.Conn.(utils.MultiWriter); ok {
|
||
return mr.WriteBuffers(buffers)
|
||
}
|
||
}
|
||
|
||
bigbs, dup := utils.MergeBuffers(buffers)
|
||
n, e := c.Write(bigbs)
|
||
if dup {
|
||
utils.PutPacket(bigbs)
|
||
}
|
||
return int64(n), e
|
||
|
||
}
|
||
|
||
//专门适用于 裸奔splice的情况
|
||
func (uc *UserConn) ReadFrom(r io.Reader) (written int64, err error) {
|
||
if uc.isUDP {
|
||
return netLayer.ClassicReadFrom(uc, r)
|
||
}
|
||
return netLayer.TryReadFrom_withSplice(uc, uc.Conn, r, func() bool {
|
||
return uc.canDirectWrite()
|
||
})
|
||
}
|
||
|
||
//如果是udp，则是多线程不安全的，如果是tcp，则安不安全看底层的链接。
|
||
// 这里规定，如果是UDP，则 每Write一遍，都要Write一个 完整的UDP 数据包
|
||
func (uc *UserConn) Write(p []byte) (int, error) {
|
||
|
||
if uc.version == 0 {
|
||
|
||
originalSupposedWrittenLenth := len(p)
|
||
|
||
var writeBuf *bytes.Buffer
|
||
|
||
if uc.isServerEnd && !uc.isntFirstPacket {
|
||
uc.isntFirstPacket = true
|
||
|
||
writeBuf = utils.GetBuf()
|
||
|
||
//v0 中，服务端的回复的第一个包也是要有数据头的(和客户端的handshake类似，只是第一个包有)，第一字节版本，第二字节addon长度（都是0）
|
||
|
||
writeBuf.WriteByte(0)
|
||
writeBuf.WriteByte(0)
|
||
|
||
}
|
||
|
||
if !uc.isUDP {
|
||
|
||
if writeBuf != nil {
|
||
writeBuf.Write(p)
|
||
|
||
_, err := uc.Conn.Write(writeBuf.Bytes()) //“直接return这个的长度” 是错的，因为写入长度只能小于等于len(p)
|
||
|
||
utils.PutBuf(writeBuf)
|
||
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
return originalSupposedWrittenLenth, nil
|
||
|
||
} else {
|
||
_, err := uc.Conn.Write(p) //“直接return这个的长度” 是错的，因为写入长度只能小于等于len(p)
|
||
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
return originalSupposedWrittenLenth, nil
|
||
}
|
||
|
||
} else {
|
||
l := int16(len(p))
|
||
if writeBuf == nil {
|
||
writeBuf = utils.GetBuf()
|
||
}
|
||
|
||
writeBuf.WriteByte(byte(l >> 8))
|
||
writeBuf.WriteByte(byte(l << 8 >> 8))
|
||
writeBuf.Write(p)
|
||
|
||
_, err := uc.Conn.Write(writeBuf.Bytes()) //“直接return这个的长度” 是错的，因为写入长度只能小于等于len(p)
|
||
|
||
utils.PutBuf(writeBuf)
|
||
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
return originalSupposedWrittenLenth, nil
|
||
}
|
||
|
||
} else {
|
||
if uc.isUDP && !uc.hasAdvancedLayer {
|
||
|
||
// 这里暂时认为包裹它的连接是 tcp或者tls，而不是udp，如果udp的话，就不需要考虑粘包问题了，比如socks5的实现
|
||
// 我们目前认为只有tls是最防墙的，而且 魔改tls是有毒的，所以反推过来，这里udp就必须加长度头。
|
||
|
||
// 目前是这个样子。之后verysimple实现了websocket和grpc后，会添加判断，如果连接是websocket或者grpc连接，则不再加长度头
|
||
|
||
// tls和tcp都是基于流的，可以分开写两次，不需要buf存在；如果连接是websocket或者grpc的话，直接传输。
|
||
|
||
l := int16(len(p))
|
||
var lenBytes []byte
|
||
|
||
if l <= 255 {
|
||
lenBytes = []byte{0, byte(l)}
|
||
}
|
||
|
||
lenBytes = []byte{byte(l >> 8), byte(l << 8 >> 8)}
|
||
|
||
_, err := uc.Conn.Write(lenBytes)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
|
||
return uc.Conn.Write(p)
|
||
|
||
}
|
||
return uc.Conn.Write(p)
|
||
|
||
}
|
||
}
|
||
|
||
//如果是udp，则是多线程不安全的，如果是tcp，则安不安全看底层的链接。
|
||
// 这里规定，如果是UDP，则 每次 Read 得到的都是一个 完整的UDP 数据包，除非p给的太小……
|
||
func (uc *UserConn) Read(p []byte) (int, error) {
|
||
|
||
var from io.Reader = uc.Conn
|
||
if uc.optionalReader != nil {
|
||
from = uc.optionalReader
|
||
}
|
||
|
||
if uc.version == 0 {
|
||
|
||
if !uc.isUDP {
|
||
|
||
if !uc.isServerEnd && !uc.isntFirstPacket {
|
||
//先读取响应头
|
||
|
||
uc.isntFirstPacket = true
|
||
|
||
bs := utils.GetPacket()
|
||
n, e := from.Read(bs)
|
||
|
||
if e != nil {
|
||
return 0, e
|
||
}
|
||
|
||
if n < 2 {
|
||
return 0, errors.New("vless response head too short")
|
||
}
|
||
n = copy(p, bs[2:n])
|
||
utils.PutPacket(bs)
|
||
return n, nil
|
||
|
||
}
|
||
|
||
return from.Read(p)
|
||
} else {
|
||
|
||
if uc.bufr == nil {
|
||
uc.bufr = bufio.NewReader(from)
|
||
}
|
||
|
||
if len(uc.udpUnreadPart) > 0 {
|
||
copiedN := copy(p, uc.udpUnreadPart)
|
||
if copiedN < len(uc.udpUnreadPart) {
|
||
uc.udpUnreadPart = uc.udpUnreadPart[copiedN:]
|
||
} else {
|
||
uc.udpUnreadPart = nil
|
||
}
|
||
return copiedN, nil
|
||
}
|
||
|
||
//v0 先读取vless响应头，再读取udp长度头
|
||
|
||
if !uc.isServerEnd && !uc.isntFirstPacket {
|
||
uc.isntFirstPacket = true
|
||
|
||
_, err := uc.bufr.ReadByte() //version byte
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
_, err = uc.bufr.ReadByte() //addon len byte
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
|
||
}
|
||
|
||
return uc.readudp_withLenthHead(p)
|
||
}
|
||
|
||
} else {
|
||
if uc.isUDP && !uc.hasAdvancedLayer {
|
||
|
||
if len(uc.udpUnreadPart) > 0 {
|
||
copiedN := copy(p, uc.udpUnreadPart)
|
||
if copiedN < len(uc.udpUnreadPart) {
|
||
uc.udpUnreadPart = uc.udpUnreadPart[copiedN:]
|
||
} else {
|
||
uc.udpUnreadPart = nil
|
||
}
|
||
return copiedN, nil
|
||
}
|
||
|
||
return uc.readudp_withLenthHead(p)
|
||
}
|
||
return from.Read(p)
|
||
|
||
}
|
||
}
|
||
|
||
func (uc *UserConn) readudp_withLenthHead(p []byte) (int, error) {
|
||
if uc.bufr == nil {
|
||
uc.bufr = bufio.NewReader(uc.Conn)
|
||
}
|
||
b1, err := uc.bufr.ReadByte()
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
b2, err := uc.bufr.ReadByte()
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
|
||
l := int(int16(b1)<<8 + int16(b2))
|
||
bs := utils.GetBytes(l)
|
||
n, err := io.ReadFull(uc.bufr, bs)
|
||
if err != nil {
|
||
return 0, err
|
||
}
|
||
/*// 测试代码
|
||
if uc.version == 1 {
|
||
log.Println("read", bs[:n], string(bs[:n]))
|
||
|
||
}*/
|
||
|
||
copiedN := copy(p, bs)
|
||
if copiedN < n { //p is short
|
||
uc.udpUnreadPart = bs[copiedN:]
|
||
}
|
||
|
||
return copiedN, nil
|
||
}
|