check if cpus are initialized

anubis-rs · Jan 23, 2024 · be00af2 · be00af2
1 parent d771bc8
commit be00af2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/xernel/kernel/src/cpu.rs b/xernel/kernel/src/cpu.rs
@@ -155,6 +155,10 @@ pub fn register_cpu() {
 }
 
 pub fn current_cpu() -> Pin<&'static Cpu> {
+    if !CPU_COUNT.is_completed() || *CPU_COUNT != CPU_ID_COUNTER.load(Ordering::SeqCst) {
+        panic!("current_cpu called before all cpus registered");
+    }
+
     unsafe { Pin::new_unchecked(&*core::ptr::from_exposed_addr(rdmsr(KERNEL_GS_BASE) as usize)) }
 }