12813126eSlogin #![no_std] // <1> 22813126eSlogin #![no_main] // <1> 3*935f40ecSlogin #![feature(const_mut_refs)] 42813126eSlogin #![feature(core_intrinsics)] // <2> 5fb6c29d0Slogin #![feature(alloc_error_handler)] 666f67c6aSlogin #![feature(panic_info_message)] 71a2eaa40Slogin #![feature(drain_filter)] // 允许Vec的drain_filter特性 862e46139SGou Ngai #![feature(c_void_variant)] //not stable, used in /home/su/Documents/VSCode/DragonOS/kernel/src/exception/softirq.rs 92813126eSlogin #[allow(non_upper_case_globals)] 102813126eSlogin #[allow(non_camel_case_types)] 112813126eSlogin #[allow(non_snake_case)] 12fb6c29d0Slogin use core::panic::PanicInfo; 13fb6c29d0Slogin 14adc1846bSlogin /// 导出x86_64架构相关的代码,命名为arch模块 15adc1846bSlogin #[cfg(target_arch = "x86_64")] 16adc1846bSlogin #[path = "arch/x86_64/mod.rs"] 1766f67c6aSlogin #[macro_use] 1866f67c6aSlogin mod arch; 19adc1846bSlogin 20587086d3Slogin mod driver; 21587086d3Slogin mod filesystem; 2266f67c6aSlogin #[macro_use] 2366f67c6aSlogin mod include; 2466f67c6aSlogin mod ipc; 2582d2e446Slogin #[macro_use] 26fb6c29d0Slogin mod libs; 27ec53d23eSlogin mod exception; 2866f67c6aSlogin mod mm; 2966f67c6aSlogin mod process; 3066f67c6aSlogin mod sched; 3166f67c6aSlogin mod smp; 3201876902SGou Ngai mod time; 3382d2e446Slogin 34fb6c29d0Slogin extern crate alloc; 3582d2e446Slogin 36fb6c29d0Slogin use mm::allocator::KernelAllocator; 372813126eSlogin 38fb6c29d0Slogin // <3> 3901876902SGou Ngai use crate::{ 40adc1846bSlogin 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"); 8861de2cdcSlogin 892813126eSlogin return 0; 902813126eSlogin } 91