authorgravatar for oezgurmakkurt@gmail.comÖzgür Akkurt <oezgurmakkurt@gmail.com> 2025-08-16 13:18:49+06:00
committergravatar for noreply@github.comGitHub <noreply@github.com> 2025-08-16 09:18:49+02:00
log99c489055951c09c1a22973078e5d7dbda14bac1
treeab4ba4e2d7da8247ba4236ecd2d17c21443af56f
parent98547713a354efef406ca79731db8984eb26c051
signaturebadge-check Signed by PGP key B5690EEEBB952194

implement registering NAPI on IoUring (#24850)

2 files changed, 44 insertions(+), 2 deletions(-)

lib/std/os/linux.zig+32-2
......@@ -6587,7 +6587,7 @@ pub const IORING_REGISTER = enum(u32) {
65876587
65886588 // register/unregister io_uring fd with the ring
65896589 REGISTER_RING_FDS,
6590 NREGISTER_RING_FDS,
6590 UNREGISTER_RING_FDS,
65916591
65926592 // register ring based provide buffer group
65936593 REGISTER_PBUF_RING,
......@@ -6599,8 +6599,31 @@ pub const IORING_REGISTER = enum(u32) {
65996599 // register a range of fixed file slots for automatic slot allocation
66006600 REGISTER_FILE_ALLOC_RANGE,
66016601
6602 // return status information for a buffer group
6603 REGISTER_PBUF_STATUS,
6604
6605 // set/clear busy poll settings
6606 REGISTER_NAPI,
6607 UNREGISTER_NAPI,
6608
6609 REGISTER_CLOCK,
6610
6611 // clone registered buffers from source ring to current ring
6612 REGISTER_CLONE_BUFFERS,
6613
6614 // send MSG_RING without having a ring
6615 REGISTER_SEND_MSG_RING,
6616
6617 // register a netdev hw rx queue for zerocopy
6618 REGISTER_ZCRX_IFQ,
6619
6620 // resize CQ ring
6621 REGISTER_RESIZE_RINGS,
6622
6623 REGISTER_MEM_REGION,
6624
66026625 // flag added to the opcode to use a registered ring fd
6603 IORING_REGISTER_USE_REGISTERED_RING = 1 << 31,
6626 REGISTER_USE_REGISTERED_RING = 1 << 31,
66046627
66056628 _,
66066629};
......@@ -6657,6 +6680,13 @@ pub const io_uring_notification_register = extern struct {
66576680 resv3: u64,
66586681};
66596682
6683pub const io_uring_napi = extern struct {
6684 busy_poll_to: u32,
6685 prefer_busy_poll: u8,
6686 _pad: [3]u8,
6687 resv: u64,
6688};
6689
66606690/// Skip updating fd indexes set to this value in the fd table */
66616691pub const IORING_REGISTER_FILES_SKIP = -2;
66626692
lib/std/os/linux/IoUring.zig+12
......@@ -1270,6 +1270,18 @@ pub fn unregister_eventfd(self: *IoUring) !void {
12701270 try handle_registration_result(res);
12711271}
12721272
1273pub fn register_napi(self: *IoUring, napi: *linux.io_uring_napi) !void {
1274 assert(self.fd >= 0);
1275 const res = linux.io_uring_register(self.fd, .REGISTER_NAPI, napi, 1);
1276 try handle_registration_result(res);
1277}
1278
1279pub fn unregister_napi(self: *IoUring, napi: *linux.io_uring_napi) !void {
1280 assert(self.fd >= 0);
1281 const res = linux.io_uring_register(self.fd, .UNREGISTER_NAPI, napi, 1);
1282 try handle_registration_result(res);
1283}
1284
12731285/// Registers an array of buffers for use with `read_fixed` and `write_fixed`.
12741286pub fn register_buffers(self: *IoUring, buffers: []const posix.iovec) !void {
12751287 assert(self.fd >= 0);