SagerNet/sing
1
0
Fork 0
mirror of https://github.com/SagerNet/sing.git synced 2025-04-05 12:57:38 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add vectorised interface

Browse source
This commit is contained in:
世界 2022-08-12 13:46:55 +08:00
parent 13f394e202
commit 05f9836bff
No known key found for this signature in database
GPG key ID: CD109927C34A63C4
12 changed files with 371 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

57
common/bufio/vectorised_unix.go Normal file
View file

@ -0,0 +1,57 @@
//go:build !windows
package bufio
import (
"unsafe"
"github.com/sagernet/sing/common/buf"
M "github.com/sagernet/sing/common/metadata"
"golang.org/x/sys/unix"
)
func (w *SyscallVectorisedWriter) WriteVectorised(buffers []*buf.Buffer) error {
defer buf.ReleaseMulti(buffers)
iovecList := make([]unix.Iovec, 0, len(buffers))
for _, buffer := range buffers {
var iovec unix.Iovec
iovec.Base = &buffer.Bytes()[0]
iovec.SetLen(buffer.Len())
iovecList = append(iovecList, iovec)
}
var innerErr unix.Errno
err := w.rawConn.Write(func(fd uintptr) (done bool) {
_, _, innerErr = unix.Syscall(unix.SYS_WRITEV, fd, uintptr(unsafe.Pointer(&iovecList[0])), uintptr(len(iovecList)))
return innerErr != unix.EAGAIN && innerErr != unix.EWOULDBLOCK
})
if innerErr != 0 {
err = innerErr
}
return err
}
func (w *SyscallVectorisedPacketWriter) WriteVectorisedPacket(buffers []*buf.Buffer, destination M.Socksaddr) error {
iovecList := make([]unix.Iovec, 0, len(buffers))
for _, buffer := range buffers {
var iovec unix.Iovec
iovec.Base = &buffer.Bytes()[0]
iovec.SetLen(buffer.Len())
iovecList = append(iovecList, iovec)
}
name, nameLen := destination.Sockaddr()
var msgHdr unix.Msghdr
msgHdr.Name = (*byte)(name)
msgHdr.Namelen = nameLen
msgHdr.Iov = &iovecList[0]
msgHdr.SetIovlen(len(iovecList))
var innerErr unix.Errno
err := w.rawConn.Write(func(fd uintptr) (done bool) {
_, _, innerErr = unix.Syscall(unix.SYS_SENDMSG, fd, uintptr(unsafe.Pointer(&msgHdr)), 0)
return innerErr != unix.EAGAIN && innerErr != unix.EWOULDBLOCK
})
if innerErr != 0 {
err = innerErr
}
return err
}