hare

commit 19b3839657aea9781638e4f4b6280928f77746e5
parent c17aaed2bddaf6ac9b15fbf0e31b79e176ac607b
Author: Drew DeVault <sir@cmpwn.com>
Date:   Wed, 19 May 2021 12:58:31 -0400

linux::io_uring: send/recv/sendmsg/recvmsg

Signed-off-by: Drew DeVault <sir@cmpwn.com>

Diffstat:
M
linux/io_uring/sqe.ha
 | 
56
++++++++++++++++++++++++++++++++++++++++++++++++++++++++
M
linux/io_uring/uring.ha
 | 
2
+-

2 files changed, 57 insertions(+), 1 deletion(-)
diff --git a/linux/io_uring/sqe.ha b/linux/io_uring/sqe.ha
@@ -153,3 +153,59 @@ export fn poll_remove(sqe: *sqe, user_data: *void, flags: sqe_flags...) void = {
 	preprw(sqe, op::POLL_REMOVE, -1, null, 0, 0, flags...);
 	set_user(sqe, user_data);
 };
+
+// Prepares a sendmsg operation for an [[sqe]], equivalent to the sendmsg(2)
+// system call.
+export fn sendmsg(
+	sqe: *sqe,
+	fd: int,
+	msghdr: *rt::msghdr,
+	sendmsg_flags: int,
+	flags: sqe_flags...
+) void = {
+	preprw(sqe, op::SENDMSG, fd, msghdr, 0, 0, flags...);
+	sqe.msg_flags = sendmsg_flags;
+};
+
+// Prepares a recvmsg operation for an [[sqe]], equivalent to the sendmsg(2)
+// system call.
+export fn recvmsg(
+	sqe: *sqe,
+	fd: int,
+	msghdr: *rt::msghdr,
+	recvmsg_flags: int,
+	flags: sqe_flags...
+) void = {
+	preprw(sqe, op::RECVMSG, fd, msghdr, 0, 0, flags...);
+	sqe.msg_flags = recvmsg_flags;
+};
+
+// Prepares a send operation for an [[sqe]], equivalent to the send(2) system
+// call.
+export fn send(
+	sqe: *sqe,
+	fd: int,
+	buf: *void,
+	count: size,
+	send_flags: int,
+	flags: sqe_flags...
+) void = {
+	assert(count <= types::U32_MAX);
+	preprw(sqe, op::SEND, fd, buf, count: u32, 0, flags...);
+	sqe.msg_flags = send_flags;
+};
+
+// Prepares a recv operation for an [[sqe]], equivalent to the send(2) system
+// call.
+export fn recv(
+	sqe: *sqe,
+	fd: int,
+	buf: *void,
+	count: size,
+	recv_flags: int,
+	flags: sqe_flags...
+) void = {
+	assert(count <= types::U32_MAX);
+	preprw(sqe, op::RECV, fd, buf, count: u32, 0, flags...);
+	sqe.msg_flags = recv_flags;
+};
diff --git a/linux/io_uring/uring.ha b/linux/io_uring/uring.ha
@@ -111,7 +111,7 @@ export type sqe = struct {
 		poll_events: u16,
 		poll32_events: u32,
 		sync_range_flags: u32,
-		msg_flags: u32,
+		msg_flags: int,
 		timeout_flags: timeout_flags,
 		accept_flags: u32,
 		cancel_flags: u32,