Fix incorrect types in sys_getcpu

kernel/src/syscall/getcpu.rs

@@ -1,27 +1,32 @@
 // SPDX-License-Identifier: MPL-2.0
 
-use ostd::{cpu::PinCurrentCpu, task::disable_preempt};
+use ostd::cpu::CpuId;
 
 use super::SyscallReturn;
 use crate::prelude::*;
 
 pub fn sys_getcpu(cpu: Vaddr, node: Vaddr, _tcache: Vaddr, ctx: &Context) -> Result<SyscallReturn> {
-    // The third argument tcache is unused after Linux 2.6.24 so we ignore it
+    // The third argument `tcache` is unused since Linux 2.6.24, so we ignore it.
-    let preempt_guard = disable_preempt();
+
-    let cpuid = preempt_guard.current_cpu();
+    // The system call itself is inherently racy, so using `current_racy` here should be fine.
-    drop(preempt_guard);
+    let current_cpu = CpuId::current_racy().as_usize() as u32;
+    // TODO: Support NUMA.
+    let current_node = 0u32;
+
     debug!(
-        "getcpu: cpuid = {}, total_cpus = {}",
+        "[sys_getcpu]: cpu = {} (total {}), node = {}",
-        cpuid.as_usize(),
+        current_cpu,
-        ostd::cpu::num_cpus()
+        ostd::cpu::num_cpus(),
+        current_node,
     );
-    // Since cpu and node can be NULL, we need to check them before writing
+
+    // `cpu` and `node` can be NULL, so we need to check before writing.
     if cpu != 0 {
-        ctx.user_space()
+        ctx.user_space().write_val(cpu, &current_cpu)?;
-            .write_val::<usize>(cpu, &cpuid.as_usize())?;
     }
     if node != 0 {
-        ctx.user_space().write_val::<usize>(node, &0)?; // TODO: NUMA is not supported
+        ctx.user_space().write_val(node, &current_node)?;
     }
+
     Ok(SyscallReturn::Return(0))
 }

test/apps/getcpu/getcpu.c

@@ -1,24 +1,29 @@
 // SPDX-License-Identifier: MPL-2.0
 
-#define _GNU_SOURCE
+#include "../network/test.h"
-#include <stdio.h>
+
-#include <stdlib.h>
-#include <unistd.h>
 #include <sys/syscall.h>
+#include <unistd.h>
 
-int main()
+FN_TEST(getcpu)
 {
-	unsigned int cpu, node;
+	struct int_pair {
+		unsigned int first;
+		unsigned int second;
+	};
 
-	// Directly test the getcpu syscall because glibc's getcpu() may not
+	struct int_pair cpu = { .first = 0xdeadbeef, .second = 0xbeefdead };
-	// use the getcpu syscall to retrieve CPU info
+	struct int_pair node = { .first = 0xbeefdead, .second = 0xdeadbeef };
-	long ret = syscall(SYS_getcpu, &cpu, &node, NULL);
-	if (ret != 0) {
-		perror("syscall getcpu");
-		exit(EXIT_FAILURE);
-	}
 
-	printf("getcpu syscall: cpu = %u, node = %u\n", cpu, node);
+	// Use `syscall()` here because `getcpu()` from glibc may not use
+	// the system call to retrieve CPU information.
+	TEST_SUCC(syscall(SYS_getcpu, NULL, NULL));
 
-	return 0;
+	// Our CI has a maximum of 4 CPUs, and NUMA support is not yet
-}
+	// available. Update these conditions if we have more than 4 CPUs
+	// or if NUMA support is added.
+	TEST_RES(syscall(SYS_getcpu, &cpu, &node),
+		 cpu.first < 4 && cpu.second == 0xbeefdead && node.first == 0 &&
+			 node.second == 0xdeadbeef);
+}
+END_TEST()

test/apps/scripts/process.sh

@@ -21,6 +21,7 @@ exit/exit_procfs
 eventfd2/eventfd2
 fork/fork
 fork_c/fork
+getcpu/getcpu
 getpid/getpid
 hello_pie/hello
 hello_world/hello_world