Paul/tiny_os
1
0
Fork 0
mirror of https://github.com/panpaul/tiny_os synced 2024-09-20 09:45:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: entry: store N_CPUS

Browse Source
This commit is contained in:
Paul Pan 2024-05-20 16:16:03 +08:00
parent 5d5c18592f
commit 32e21b3468
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
kernel/src/entry.rs
View File

@ -4,16 +4,21 @@ use crate::plat::timer::{Timer, TimerOps};
use crate::plat::trap::{Trap, TrapOps};
use crate::scheduler::SCHEDULER;
use core::cell::Cell;
use core::sync::atomic::{AtomicUsize, Ordering};
use fdt::Fdt;
use log::{debug, error, info, warn};
#[thread_local]
pub static HART_ID: Cell<usize> = Cell::new(0);
pub static N_CPUS: AtomicUsize = AtomicUsize::new(1);
pub fn rust_main() -> ! {
let fdt_addr = crate::arch::FDT.load(core::sync::atomic::Ordering::Acquire);
let fdt_addr = crate::arch::FDT.load(Ordering::Acquire);
let fdt = unsafe { Fdt::from_ptr(fdt_addr).unwrap() };
N_CPUS.store(fdt.cpus().count(), Ordering::Release);
setup_console(&fdt);
info!("Kernel Started");