mirror of
https://github.com/e1732a364fed/v2ray_simple.git
synced 2025-12-24 13:27:56 +08:00
104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
package netLayer
|
||
|
||
import (
|
||
"io"
|
||
"net"
|
||
"syscall"
|
||
|
||
"github.com/e1732a364fed/v2ray_simple/utils"
|
||
)
|
||
|
||
//经过测试，网速越快、延迟越小，越不需要readv, 此时首包buf越大越好, 因为一次系统底层读取就会读到一大块数据, 此时再用readv分散写入 实际上就是反效果; readv的数量则不需要太多
|
||
|
||
//在内网单机自己连自己测速时,readv会导致降速.
|
||
|
||
const (
|
||
DefaultReadvOption = true
|
||
)
|
||
|
||
var (
|
||
|
||
// 是否会在转发过程中使用readv
|
||
UseReadv bool
|
||
)
|
||
|
||
/*
|
||
ReadvFrom 用于读端 为rawRead的情况，如 从socks5或direct读取 数据, 等裸协议的情况。
|
||
|
||
rm可为nil，但不建议，因为提供非nil的readvMem 可以节省内存分配开销。
|
||
|
||
返回错误时，会返回 原buffer 或 在函数内部新分配的buffer.
|
||
|
||
本函数不负责 释放分配的内存. 因为有时需要重复利用缓存。
|
||
|
||
TryCopy函数使用到了本函数 来进行readv相关操作。
|
||
*/
|
||
func ReadvFrom(rawReadConn syscall.RawConn, rm *utils.ReadvMem) ([][]byte, error) {
|
||
|
||
if rm == nil {
|
||
rm = utils.Get_readvMem()
|
||
}
|
||
|
||
allocedBuffers := rm.Buffers
|
||
|
||
var nBytes uint32
|
||
err := rawReadConn.Read(func(fd uintptr) bool {
|
||
n, e := rm.Mr.Read(fd)
|
||
if e != nil {
|
||
return false
|
||
}
|
||
|
||
nBytes = n
|
||
return true
|
||
})
|
||
if err != nil {
|
||
|
||
return allocedBuffers, err
|
||
}
|
||
if nBytes == 0 {
|
||
return allocedBuffers, io.EOF
|
||
}
|
||
|
||
nBuf := utils.ShrinkBuffers(allocedBuffers, int(nBytes), utils.ReadvSingleBufLen)
|
||
|
||
return allocedBuffers[:nBuf], nil
|
||
}
|
||
|
||
// 依次试图使用 readv、ReadBuffers 以及 原始 Read 读取数据
|
||
func ReadBuffersFrom(c io.Reader, rawReadConn syscall.RawConn, mr utils.MultiReader) (buffers [][]byte, err error) {
|
||
|
||
if rawReadConn != nil {
|
||
readv_mem := utils.Get_readvMem()
|
||
//defer put_readvMem(readv_mem) //因为返回的buffers还需使用, 所以不能立刻放回 readv_mem
|
||
|
||
buffers, err = ReadvFrom(rawReadConn, readv_mem)
|
||
|
||
} else if mr != nil {
|
||
return mr.ReadBuffers()
|
||
} else {
|
||
packet := utils.GetPacket()
|
||
var n int
|
||
n, err = c.Read(packet)
|
||
if err != nil {
|
||
return
|
||
}
|
||
buffers = append(buffers, packet[:n])
|
||
}
|
||
return
|
||
}
|
||
|
||
// if r!=0, then it means c can be used in readv. 1 means syscall.RawConn, 2 means utils.MultiReader
|
||
func IsConnGoodForReadv(c net.Conn) (r int, rawReadConn syscall.RawConn, mr utils.MultiReader) {
|
||
rawReadConn = GetRawConn(c)
|
||
var ok bool
|
||
if rawReadConn != nil {
|
||
r = 1
|
||
return
|
||
|
||
} else if mr, ok = c.(utils.MultiReader); ok {
|
||
r = 2
|
||
return
|
||
}
|
||
return
|
||
}
|