12813126eSlogin #![no_std] // <1> 22813126eSlogin #![no_main] // <1> 32813126eSlogin #![feature(core_intrinsics)] // <2> 4fb6c29d0Slogin #![feature(alloc_error_handler)] 5*66f67c6aSlogin #![feature(panic_info_message)] 6fb6c29d0Slogin 72813126eSlogin #[allow(non_upper_case_globals)] 82813126eSlogin #[allow(non_camel_case_types)] 92813126eSlogin #[allow(non_snake_case)] 10fb6c29d0Slogin use core::panic::PanicInfo; 11fb6c29d0Slogin 12*66f67c6aSlogin #[macro_use] 13*66f67c6aSlogin mod arch; 14*66f67c6aSlogin #[macro_use] 15*66f67c6aSlogin mod include; 16*66f67c6aSlogin mod ipc; 17fb6c29d0Slogin 1882d2e446Slogin #[macro_use] 19fb6c29d0Slogin mod libs; 20*66f67c6aSlogin mod mm; 21*66f67c6aSlogin mod process; 22*66f67c6aSlogin mod sched; 23*66f67c6aSlogin mod smp; 2482d2e446Slogin 25fb6c29d0Slogin extern crate alloc; 2682d2e446Slogin 27fb6c29d0Slogin use mm::allocator::KernelAllocator; 282813126eSlogin 29fb6c29d0Slogin // <3> 30*66f67c6aSlogin use crate::{include::bindings::bindings::{process_do_exit, BLACK, GREEN}, arch::x86_64::asm::current::current_pcb}; 31fb6c29d0Slogin 32fb6c29d0Slogin // 声明全局的slab分配器 33fb6c29d0Slogin #[cfg_attr(not(test), global_allocator)] 34fb6c29d0Slogin pub static KERNEL_ALLOCATOR: KernelAllocator = KernelAllocator {}; 35fb6c29d0Slogin 36fb6c29d0Slogin /// 全局的panic处理函数 372813126eSlogin #[panic_handler] 382813126eSlogin #[no_mangle] 39*66f67c6aSlogin pub fn panic(info: &PanicInfo) -> ! { 40*66f67c6aSlogin kerror!("Kernel Panic Occurred."); 41*66f67c6aSlogin 42*66f67c6aSlogin match info.location() { 43*66f67c6aSlogin Some(loc) => { 44*66f67c6aSlogin println!( 45*66f67c6aSlogin "Location:\n\tFile: {}\n\tLine: {}, Column: {}", 46*66f67c6aSlogin loc.file(), 47*66f67c6aSlogin loc.line(), 48*66f67c6aSlogin loc.column() 49*66f67c6aSlogin ); 50*66f67c6aSlogin } 51*66f67c6aSlogin None => { 52*66f67c6aSlogin println!("No location info"); 53*66f67c6aSlogin } 54*66f67c6aSlogin } 55*66f67c6aSlogin 56*66f67c6aSlogin match info.message() { 57*66f67c6aSlogin Some(msg) => { 58*66f67c6aSlogin println!("Message:\n\t{}", msg); 59*66f67c6aSlogin } 60*66f67c6aSlogin None => { 61*66f67c6aSlogin println!("No panic message."); 62*66f67c6aSlogin } 63*66f67c6aSlogin } 64*66f67c6aSlogin 65*66f67c6aSlogin println!("Current PCB:\n\t{:?}", current_pcb()); 66*66f67c6aSlogin unsafe { 67*66f67c6aSlogin process_do_exit(u64::MAX); 68*66f67c6aSlogin }; 69*66f67c6aSlogin loop { 70*66f67c6aSlogin 71*66f67c6aSlogin } 722813126eSlogin } 732813126eSlogin 74fb6c29d0Slogin /// 该函数用作测试,在process.c的initial_kernel_thread()中调用了此函数 752813126eSlogin #[no_mangle] 762813126eSlogin pub extern "C" fn __rust_demo_func() -> i32 { 77fb6c29d0Slogin printk_color!(GREEN, BLACK, "__rust_demo_func()\n"); 78fb6c29d0Slogin 792813126eSlogin return 0; 802813126eSlogin } 81