Xeffy/gouring
1
0
Fork 0
mirror of https://github.com/ii64/gouring.git synced 2025-04-26 05:42:48 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat: added some prepare helper

Browse source 
* PrepAcceptMultishot (renamed)
* PrepAcceptMultishotDirect (renamed)
* PrepRecvmsgMultishot
* PrepSendSetAddr
* PrepSendto
* PrepRecvMultishot
* PrepFgetxattr
This commit is contained in:
Xeffy Chen 2023-04-16 17:50:51 +07:00
parent d3eea9e076
commit 56bbdd81d6
Signed by: Xeffy
GPG key ID: E41C08AD390E7C49
2 changed files with 44 additions and 23 deletions

1
hdr.go
View file

@ -20,6 +20,7 @@ type IoUringSqe_Union1 uint64
func (u *IoUringSqe_Union1) SetOffset(v uint64) { *u = IoUringSqe_Union1(v) }
func (u *IoUringSqe_Union1) SetOffset_RawPtr(v unsafe.Pointer) { *u = IoUringSqe_Union1((uintptr)(v)) }
func (u *IoUringSqe_Union1) SetAddr2(v uint64) { *u = IoUringSqe_Union1(v) }
func (u *IoUringSqe_Union1) SetAddr2_RawPtr(v unsafe.Pointer) { *u = IoUringSqe_Union1((uintptr)(v)) }
func (u *IoUringSqe_Union1) SetCmdOp(v uint32) {
(*struct {
CmdOp uint32

66
prep.go
View file

@ -113,6 +113,14 @@ func PrepAcceptDirect(sqe *IoUringSqe, fd int, rsa *syscall.RawSockaddrAny, rsaS
PrepAccept(sqe, fd, rsa, rsaSz, flags)
__io_uring_set_target_fixed_file(sqe, uint32(fileIndex))
}
func PrepAcceptMultishot(sqe *IoUringSqe, fd int, rsa *syscall.RawSockaddrAny, rsaSz *uintptr, flags uint32) {
PrepAccept(sqe, fd, rsa, rsaSz, flags)
sqe.IoPrio |= IORING_ACCEPT_MULTISHOT
}
func PrepAcceptMultishotDirect(sqe *IoUringSqe, fd int, rsa *syscall.RawSockaddrAny, rsaSz *uintptr, flags uint32) {
PrepAcceptMultishot(sqe, fd, rsa, rsaSz, flags)
__io_uring_set_target_fixed_file(sqe, IORING_FILE_INDEX_ALLOC-1)
}
func PrepConnect(sqe *IoUringSqe, fd int, rsa *syscall.RawSockaddrAny, rsaSz uintptr) {
PrepRW(IORING_OP_CONNECT, sqe, fd, unsafe.Pointer(rsa), 0, uint64(rsaSz))
@ -122,6 +130,10 @@ func PrepRecvmsg(sqe *IoUringSqe, fd int, msg *syscall.Msghdr, flags uint) {
PrepRW(IORING_OP_RECVMSG, sqe, fd, unsafe.Pointer(msg), 1, 0)
sqe.SetMsgFlags(uint32(flags))
}
func PrepRecvmsgMultishot(sqe *IoUringSqe, fd int, msg *syscall.Msghdr, flags uint) {
PrepRecvmsg(sqe, fd, msg, flags)
sqe.IoPrio |= IORING_RECV_MULTISHOT
}
func PrepSendmsg(sqe *IoUringSqe, fd int, msg *syscall.Msghdr, flags uint32) {
PrepRW(IORING_OP_SENDMSG, sqe, fd, unsafe.Pointer(msg), 1, 0)
@ -132,6 +144,11 @@ func PrepSendmsgZc(sqe *IoUringSqe, fd int, msg *syscall.Msghdr, flags uint32) {
sqe.Opcode |= IORING_OP_SENDMSG_ZC
}
func PrepSendSetAddr(sqe *IoUringSqe, destAddr *syscall.RawSockaddrAny, addrLen uint16) {
sqe.SetAddr2_RawPtr(unsafe.Pointer(destAddr))
sqe.SetAddrLen(addrLen)
}
func PrepClose(sqe *IoUringSqe, fd int) {
PrepRW(IORING_OP_CLOSE, sqe, fd, nil, 0, 0)
}
@ -154,6 +171,11 @@ func PrepOpenat(sqe *IoUringSqe, dfd int, path *byte, flags uint32, mode int) {
sqe.SetOpenFlags(flags)
}
func PrepOpenatDirect(sqe *IoUringSqe, dfd int, path *byte, flags uint32, mode int, fileIndex uint32) {
PrepOpenat(sqe, dfd, path, flags, mode)
__io_uring_set_target_fixed_file(sqe, fileIndex)
}
func PrepOpenat2(sqe *IoUringSqe, dfd int, path *byte, how *unix.OpenHow) {
PrepRW(IORING_OP_OPENAT2, sqe, dfd, unsafe.Pointer(path), int(unsafe.Sizeof(*how)), 0)
sqe.SetOffset_RawPtr(unsafe.Pointer(how))
@ -164,11 +186,6 @@ func PrepOpenat2Direct(sqe *IoUringSqe, dfd int, path *byte, how *unix.OpenHow,
__io_uring_set_target_fixed_file(sqe, fileIndex)
}
func PrepOpenatDirect(sqe *IoUringSqe, dfd int, path *byte, flags uint32, mode int, fileIndex uint32) {
PrepOpenat(sqe, dfd, path, flags, mode)
__io_uring_set_target_fixed_file(sqe, fileIndex)
}
func PrepStatx(sqe *IoUringSqe, dfd int, path *byte, flags uint32, mask uint32, statxbuf *unix.Statx_t) {
PrepRW(IORING_OP_STATX, sqe, dfd, unsafe.Pointer(path), int(mask), 0)
sqe.SetOffset_RawPtr(unsafe.Pointer(statxbuf))
@ -184,6 +201,12 @@ func PrepMadvise(sqe *IoUringSqe, addr unsafe.Pointer, length int, advice uint32
sqe.SetFadviseAdvice(advice)
}
func PrepSendto(sqe *IoUringSqe, sockfd int, buf *byte, length int, flags uint32,
addr *syscall.RawSockaddrAny, addrLen uint16) {
PrepSend(sqe, sockfd, buf, length, flags)
PrepSendSetAddr(sqe, addr, addrLen)
}
func PrepSend(sqe *IoUringSqe, sockfd int, buf *byte, length int, flags uint32) {
PrepRW(IORING_OP_SEND, sqe, sockfd, unsafe.Pointer(buf), length, 0)
sqe.SetMsgFlags(flags)
@ -199,9 +222,14 @@ func PrepSendZcFixed(sqe *IoUringSqe, sockfd int, buf *byte, length int, flags u
sqe.SetBufIndex(bufIndex)
}
func PrepRecv(sqe *IoUringSqe, sockfd int, buf *byte, length int, flags uint32) {
func PrepRecv(sqe *IoUringSqe, sockfd int, buf *byte, length int, flags int) {
PrepRW(IORING_OP_RECV, sqe, sockfd, unsafe.Pointer(buf), length, 0)
sqe.SetMsgFlags(flags)
sqe.SetMsgFlags(uint32(flags))
}
func PrepRecvMultishot(sqe *IoUringSqe, sockfd int, buf *byte, length int, flags int) {
PrepRecv(sqe, sockfd, buf, length, flags)
sqe.IoPrio |= IORING_RECV_MULTISHOT
}
func PrepSocket(sqe *IoUringSqe, domain int, _type int, protocol int, flags uint32) {
@ -251,20 +279,6 @@ func PrepFsync(sqe *IoUringSqe, fd int, fsyncFlags uint32) {
sqe.SetFsyncFlags(fsyncFlags)
}
/*
Multishot
*/
func PrepMultishotAccept(sqe *IoUringSqe, fd int, rsa *syscall.RawSockaddrAny, rsaSz *uintptr, flags uint32) {
PrepAccept(sqe, fd, rsa, rsaSz, flags)
sqe.IoPrio |= IORING_ACCEPT_MULTISHOT
}
func PrepMultishotAcceptDirect(sqe *IoUringSqe, fd int, rsa *syscall.RawSockaddrAny, rsaSz *uintptr, flags uint32) {
PrepMultishotAccept(sqe, fd, rsa, rsaSz, flags)
__io_uring_set_target_fixed_file(sqe, IORING_FILE_INDEX_ALLOC-1)
}
/*
Extra
*/
@ -368,8 +382,14 @@ func PrepSetxattr(sqe *IoUringSqe, name *byte, value *byte, path *byte, flags ui
sqe.SetXattrFlags(flags)
}
func PrepFsetxattr(sqe *IoUringSqe, fd int, name *byte, value *byte, flags uint32, length int) {
PrepRW(IORING_OP_FSETXATTR, sqe, fd, unsafe.Pointer(name), length, 0)
func PrepFgetxattr(sqe *IoUringSqe, fd int, name *byte, value *byte, length uint) {
PrepRW(IORING_OP_FGETXATTR, sqe, fd, unsafe.Pointer(name), int(length), 0)
sqe.SetOffset_RawPtr(unsafe.Pointer(value))
sqe.SetXattrFlags(0)
}
func PrepFsetxattr(sqe *IoUringSqe, fd int, name *byte, value *byte, flags uint32, length uint) {
PrepRW(IORING_OP_FSETXATTR, sqe, fd, unsafe.Pointer(name), int(length), 0)
sqe.SetOffset_RawPtr(unsafe.Pointer(value))
sqe.SetXattrFlags(flags)
}