zishuo/netstack
1
0
Fork 0
mirror of https://github.com/impact-eintr/netstack.git synced 2025-10-20 19:36:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

fdbased endpoint

Browse Source
This commit is contained in:
impact-eintr
2022-11-23 11:31:21 +08:00
parent 6011e0f8f3
commit 990a0c2901
7 changed files with 314 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
tcpip/link/rawfile/blockingpoll_unsafe.go
View File

@@ -41,10 +41,46 @@ func NonBlockingWrite(fd int, buf []byte) *tcpip.Error {
if len(buf) > 0 {
ptr = unsafe.Pointer(&buf[0])
}
_, _, e := syscall.RawSyscall(syscall.SYS_WRITE, uintptr(fd),
uintptr(ptr), uintptr(len(buf)))
if e != 0 {
return TranslateErrno(e)
}
return nil
}
func NonBlockingWrite2(fd int, b1, b2 []byte) *tcpip.Error {
if len(b2) == 0 {
return NonBlockingWrite(fd, b1)
}
/*
#include <sys/uio.h>
struct iovec {
void *iov_base;
size_t iov_len;
};
**/
iovec := [...]syscall.Iovec{
{
Base: &b1[0],
Len: uint64(len(b1)),
},
{
Base: &b2[0],
Len: uint64(len(b2)),
},
}
// ssize_t writev(int fildes, const struct iovec *iov, int iovcnt);
_, _, e := syscall.RawSyscall(syscall.SYS_WRITEV, uintptr(fd),
uintptr(unsafe.Pointer(&iovec[0])), uintptr(len(iovec)))
if e != 0 {
return TranslateErrno(e)
}
return nil
}
func BlockingRead(fd int, b []byte) (int, *tcpip.Error) {
@@ -69,7 +105,8 @@ func BlockingRead(fd int, b []byte) (int, *tcpip.Error) {
func BlockingReadv(fd int, iovecs []syscall.Iovec) (int, *tcpip.Error) {
for {
n, _, e := syscall.RawSyscall(syscall.SYS_READV, uintptr(fd), uintptr(unsafe. Pointer(&iovecs[0])), uintptr(len(iovecs)))
n, _, e := syscall.RawSyscall(syscall.SYS_READV, uintptr(fd),
uintptr(unsafe.Pointer(&iovecs[0])), uintptr(len(iovecs)))
if e == 0 {
return int(n), nil
}