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 1366f67c6aSlogin #[macro_use] 1466f67c6aSlogin mod arch; 1566f67c6aSlogin #[macro_use] 1666f67c6aSlogin mod include; 1766f67c6aSlogin mod ipc; 18fb6c29d0Slogin 1982d2e446Slogin #[macro_use] 20fb6c29d0Slogin mod libs; 21*01876902SGou Ngai mod driver; 2266f67c6aSlogin mod mm; 2366f67c6aSlogin mod process; 2466f67c6aSlogin mod sched; 2566f67c6aSlogin mod smp; 26*01876902SGou Ngai mod time; 2782d2e446Slogin 28fb6c29d0Slogin extern crate alloc; 2982d2e446Slogin 30fb6c29d0Slogin use mm::allocator::KernelAllocator; 312813126eSlogin 32fb6c29d0Slogin // <3> 33*01876902SGou Ngai use crate::{ 34*01876902SGou Ngai arch::x86_64::asm::current::current_pcb, 35*01876902SGou Ngai include::bindings::bindings::{process_do_exit, BLACK, GREEN}, 36*01876902SGou Ngai }; 37fb6c29d0Slogin 38fb6c29d0Slogin // 声明全局的slab分配器 39fb6c29d0Slogin #[cfg_attr(not(test), global_allocator)] 40fb6c29d0Slogin pub static KERNEL_ALLOCATOR: KernelAllocator = KernelAllocator {}; 41fb6c29d0Slogin 42fb6c29d0Slogin /// 全局的panic处理函数 432813126eSlogin #[panic_handler] 442813126eSlogin #[no_mangle] 4566f67c6aSlogin pub fn panic(info: &PanicInfo) -> ! { 4666f67c6aSlogin kerror!("Kernel Panic Occurred."); 4766f67c6aSlogin 4866f67c6aSlogin match info.location() { 4966f67c6aSlogin Some(loc) => { 5066f67c6aSlogin println!( 5166f67c6aSlogin "Location:\n\tFile: {}\n\tLine: {}, Column: {}", 5266f67c6aSlogin loc.file(), 5366f67c6aSlogin loc.line(), 5466f67c6aSlogin loc.column() 5566f67c6aSlogin ); 5666f67c6aSlogin } 5766f67c6aSlogin None => { 5866f67c6aSlogin println!("No location info"); 5966f67c6aSlogin } 6066f67c6aSlogin } 6166f67c6aSlogin 6266f67c6aSlogin match info.message() { 6366f67c6aSlogin Some(msg) => { 6466f67c6aSlogin println!("Message:\n\t{}", msg); 6566f67c6aSlogin } 6666f67c6aSlogin None => { 6766f67c6aSlogin println!("No panic message."); 6866f67c6aSlogin } 6966f67c6aSlogin } 7066f67c6aSlogin 7166f67c6aSlogin println!("Current PCB:\n\t{:?}", current_pcb()); 7266f67c6aSlogin unsafe { 7366f67c6aSlogin process_do_exit(u64::MAX); 7466f67c6aSlogin }; 75*01876902SGou Ngai loop {} 762813126eSlogin } 772813126eSlogin 78fb6c29d0Slogin /// 该函数用作测试,在process.c的initial_kernel_thread()中调用了此函数 792813126eSlogin #[no_mangle] 802813126eSlogin pub extern "C" fn __rust_demo_func() -> i32 { 81fb6c29d0Slogin printk_color!(GREEN, BLACK, "__rust_demo_func()\n"); 82fb6c29d0Slogin 832813126eSlogin return 0; 842813126eSlogin } 85