1 #![no_std] // <1> 2 #![no_main] // <1> 3 #![feature(core_intrinsics)] // <2> 4 #![feature(alloc_error_handler)] 5 #![feature(panic_info_message)] 6 #![feature(drain_filter)] // 允许Vec的drain_filter特性 7 #![feature(c_void_variant)] //not stable, used in /home/su/Documents/VSCode/DragonOS/kernel/src/exception/softirq.rs 8 #[allow(non_upper_case_globals)] 9 #[allow(non_camel_case_types)] 10 #[allow(non_snake_case)] 11 use core::panic::PanicInfo; 12 13 /// 导出x86_64架构相关的代码，命名为arch模块 14 #[cfg(target_arch = "x86_64")] 15 #[path = "arch/x86_64/mod.rs"] 16 #[macro_use] 17 mod arch; 18 19 mod driver; 20 mod filesystem; 21 #[macro_use] 22 mod include; 23 mod ipc; 24 #[macro_use] 25 mod libs; 26 mod exception; 27 mod mm; 28 mod process; 29 mod sched; 30 mod smp; 31 mod time; 32 33 extern crate alloc; 34 35 use mm::allocator::KernelAllocator; 36 37 // <3> 38 use crate::{ 39 arch::asm::current::current_pcb, 40 include::bindings::bindings::{process_do_exit, BLACK, GREEN}, 41 }; 42 43 // 声明全局的slab分配器 44 #[cfg_attr(not(test), global_allocator)] 45 pub static KERNEL_ALLOCATOR: KernelAllocator = KernelAllocator {}; 46 47 /// 全局的panic处理函数 48 #[panic_handler] 49 #[no_mangle] 50 pub fn panic(info: &PanicInfo) -> ! { 51 kerror!("Kernel Panic Occurred."); 52 53 match info.location() { 54 Some(loc) => { 55 println!( 56 "Location:\n\tFile: {}\n\tLine: {}, Column: {}", 57 loc.file(), 58 loc.line(), 59 loc.column() 60 ); 61 } 62 None => { 63 println!("No location info"); 64 } 65 } 66 67 match info.message() { 68 Some(msg) => { 69 println!("Message:\n\t{}", msg); 70 } 71 None => { 72 println!("No panic message."); 73 } 74 } 75 76 println!("Current PCB:\n\t{:?}", current_pcb()); 77 unsafe { 78 process_do_exit(u64::MAX); 79 }; 80 loop {} 81 } 82 83 /// 该函数用作测试，在process.c的initial_kernel_thread()中调用了此函数 84 #[no_mangle] 85 pub extern "C" fn __rust_demo_func() -> i32 { 86 printk_color!(GREEN, BLACK, "__rust_demo_func()\n"); 87 88 return 0; 89 } 90