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 8 #[allow(non_upper_case_globals)] 9 #[allow(non_camel_case_types)] 10 #[allow(non_snake_case)] 11 use core::panic::PanicInfo; 12 13 #[macro_use] 14 mod arch; 15 #[macro_use] 16 mod include; 17 mod ipc; 18 19 #[macro_use] 20 mod libs; 21 mod driver; 22 mod mm; 23 mod process; 24 mod sched; 25 mod smp; 26 mod time; 27 28 extern crate alloc; 29 30 use mm::allocator::KernelAllocator; 31 32 // <3> 33 use crate::{ 34 arch::x86_64::asm::current::current_pcb, 35 include::bindings::bindings::{process_do_exit, BLACK, GREEN}, 36 }; 37 38 // 声明全局的slab分配器 39 #[cfg_attr(not(test), global_allocator)] 40 pub static KERNEL_ALLOCATOR: KernelAllocator = KernelAllocator {}; 41 42 /// 全局的panic处理函数 43 #[panic_handler] 44 #[no_mangle] 45 pub fn panic(info: &PanicInfo) -> ! { 46 kerror!("Kernel Panic Occurred."); 47 48 match info.location() { 49 Some(loc) => { 50 println!( 51 "Location:\n\tFile: {}\n\tLine: {}, Column: {}", 52 loc.file(), 53 loc.line(), 54 loc.column() 55 ); 56 } 57 None => { 58 println!("No location info"); 59 } 60 } 61 62 match info.message() { 63 Some(msg) => { 64 println!("Message:\n\t{}", msg); 65 } 66 None => { 67 println!("No panic message."); 68 } 69 } 70 71 println!("Current PCB:\n\t{:?}", current_pcb()); 72 unsafe { 73 process_do_exit(u64::MAX); 74 }; 75 loop {} 76 } 77 78 /// 该函数用作测试，在process.c的initial_kernel_thread()中调用了此函数 79 #[no_mangle] 80 pub extern "C" fn __rust_demo_func() -> i32 { 81 printk_color!(GREEN, BLACK, "__rust_demo_func()\n"); 82 83 return 0; 84 } 85