From 6f418c11e1ad1150fbdb002cfe1be92bda4e93cb Mon Sep 17 00:00:00 2001
From: David CARLIER <devnexen@gmail.com>
Date: Fri, 12 May 2023 22:13:25 +0100
Subject: [PATCH] linux adding some NUMA support

---
 lib/std/c/linux.zig  | 16 ++++++++++++++++
 lib/std/os/linux.zig | 16 ++++++++++++++++
 2 files changed, 32 insertions(+)

diff --git a/lib/std/c/linux.zig b/lib/std/c/linux.zig
index 3eeb3bd68b..856d64b21c 100644
--- a/lib/std/c/linux.zig
+++ b/lib/std/c/linux.zig
@@ -371,3 +371,19 @@ pub const dirent64 = struct {
     d_type: u8,
     d_name: [256]u8,
 };
+
+pub const MPOL = struct {
+    pub const F_NODE = 1 << 0;
+    pub const F_ADDR = 1 << 1;
+    pub const F_MEMS_ALLOWED = 1 << 2;
+    /// flags for SYS_mbind
+    pub const MF_STRICT = 1 << 0;
+    pub const MF_MOVE = 1 << 1;
+    pub const MF_MOVE_ALL = 1 << 2;
+    pub const MF_LAZY = 1 << 3;
+    pub const MF_INTERNAL = 1 << 4;
+    pub const MF_VALID = MPOL.MF_STRICT | MPOL.MF_MOVE | MPOL.MOVE_ALL;
+};
+
+pub extern "c" fn getcpu(cpu: *c_uint, node: *c_uint) c_int;
+pub extern "c" fn sched_getcpu() c_int;
diff --git a/lib/std/os/linux.zig b/lib/std/os/linux.zig
index 3dde287612..896169ed57 100644
--- a/lib/std/os/linux.zig
+++ b/lib/std/os/linux.zig
@@ -1519,6 +1519,22 @@ pub fn sched_getaffinity(pid: pid_t, size: usize, set: *cpu_set_t) usize {
     return 0;
 }
 
+pub fn getcpu(cpu: *u32, node: *u32) usize {
+    return syscall3(.getcpu, cpu, node, null);
+}
+
+pub fn sched_getcpu() usize {
+    var cpu: u32 = undefined;
+    const rc = syscall3(.getcpu, &cpu, null, null);
+    if (@bitCast(isize, rc) < 0) return rc;
+    return @intCast(usize, cpu);
+}
+
+/// libc has no wrapper for this syscall
+pub fn mbind(addr: ?*anyopaque, len: u32, mode: i32, nodemask: *const u32, maxnode: u32, flags: u32) usize {
+    return syscall6(.mbind, addr, len, mode, nodemask, maxnode, flags);
+}
+
 pub fn epoll_create() usize {
     return epoll_create1(0);
 }