12813126eSlogin #![no_std] // <1> 22813126eSlogin #![no_main] // <1> 32813126eSlogin #![feature(core_intrinsics)] // <2> 4fb6c29d0Slogin #![feature(alloc_error_handler)] 566f67c6aSlogin #![feature(panic_info_message)] 61a2eaa40Slogin #![feature(drain_filter)] // 允许Vec的drain_filter特性 7fb6c29d0Slogin 82813126eSlogin #[allow(non_upper_case_globals)] 92813126eSlogin #[allow(non_camel_case_types)] 102813126eSlogin #[allow(non_snake_case)] 11fb6c29d0Slogin use core::panic::PanicInfo; 12fb6c29d0Slogin 13*adc1846bSlogin /// 导出x86_64架构相关的代码,命名为arch模块 14*adc1846bSlogin #[cfg(target_arch = "x86_64")] 15*adc1846bSlogin #[path = "arch/x86_64/mod.rs"] 1666f67c6aSlogin #[macro_use] 1766f67c6aSlogin mod arch; 18*adc1846bSlogin 1966f67c6aSlogin #[macro_use] 2066f67c6aSlogin mod include; 2166f67c6aSlogin mod ipc; 22fb6c29d0Slogin 2382d2e446Slogin #[macro_use] 24fb6c29d0Slogin mod libs; 2501876902SGou Ngai mod driver; 2666f67c6aSlogin mod mm; 2766f67c6aSlogin mod process; 2866f67c6aSlogin mod sched; 2966f67c6aSlogin mod smp; 3001876902SGou Ngai mod time; 3182d2e446Slogin 32*adc1846bSlogin 33*adc1846bSlogin 34fb6c29d0Slogin extern crate alloc; 3582d2e446Slogin 36fb6c29d0Slogin use mm::allocator::KernelAllocator; 372813126eSlogin 38fb6c29d0Slogin // <3> 3901876902SGou Ngai use crate::{ 40*adc1846bSlogin arch::asm::current::current_pcb, 4101876902SGou Ngai include::bindings::bindings::{process_do_exit, BLACK, GREEN}, 4201876902SGou Ngai }; 43fb6c29d0Slogin 44fb6c29d0Slogin // 声明全局的slab分配器 45fb6c29d0Slogin #[cfg_attr(not(test), global_allocator)] 46fb6c29d0Slogin pub static KERNEL_ALLOCATOR: KernelAllocator = KernelAllocator {}; 47fb6c29d0Slogin 48fb6c29d0Slogin /// 全局的panic处理函数 492813126eSlogin #[panic_handler] 502813126eSlogin #[no_mangle] 5166f67c6aSlogin pub fn panic(info: &PanicInfo) -> ! { 5266f67c6aSlogin kerror!("Kernel Panic Occurred."); 5366f67c6aSlogin 5466f67c6aSlogin match info.location() { 5566f67c6aSlogin Some(loc) => { 5666f67c6aSlogin println!( 5766f67c6aSlogin "Location:\n\tFile: {}\n\tLine: {}, Column: {}", 5866f67c6aSlogin loc.file(), 5966f67c6aSlogin loc.line(), 6066f67c6aSlogin loc.column() 6166f67c6aSlogin ); 6266f67c6aSlogin } 6366f67c6aSlogin None => { 6466f67c6aSlogin println!("No location info"); 6566f67c6aSlogin } 6666f67c6aSlogin } 6766f67c6aSlogin 6866f67c6aSlogin match info.message() { 6966f67c6aSlogin Some(msg) => { 7066f67c6aSlogin println!("Message:\n\t{}", msg); 7166f67c6aSlogin } 7266f67c6aSlogin None => { 7366f67c6aSlogin println!("No panic message."); 7466f67c6aSlogin } 7566f67c6aSlogin } 7666f67c6aSlogin 7766f67c6aSlogin println!("Current PCB:\n\t{:?}", current_pcb()); 7866f67c6aSlogin unsafe { 7966f67c6aSlogin process_do_exit(u64::MAX); 8066f67c6aSlogin }; 8101876902SGou Ngai loop {} 822813126eSlogin } 832813126eSlogin 84fb6c29d0Slogin /// 该函数用作测试,在process.c的initial_kernel_thread()中调用了此函数 852813126eSlogin #[no_mangle] 862813126eSlogin pub extern "C" fn __rust_demo_func() -> i32 { 87fb6c29d0Slogin printk_color!(GREEN, BLACK, "__rust_demo_func()\n"); 882813126eSlogin return 0; 892813126eSlogin } 90