12813126eSlogin #![no_main] // <1> 2bb24249fSLoGin #![feature(alloc_error_handler)] 340fe15e0SLoGin #![feature(allocator_api)] 440fe15e0SLoGin #![feature(arbitrary_self_types)] 51496ba7bSLoGin #![feature(asm_const)] 6935f40ecSlogin #![feature(const_mut_refs)] 706d5e247SLoGin #![feature(const_trait_impl)] 8a03c4f9dSLoGin #![feature(const_refs_to_cell)] 940fe15e0SLoGin #![feature(core_intrinsics)] 1013776c11Slogin #![feature(c_void_variant)] 1140fe15e0SLoGin #![feature(drain_filter)] 1270a4e555SLoGin #![feature(inline_const)] 13729a96efSXshine #![feature(is_some_and)] 141496ba7bSLoGin #![feature(naked_functions)] 15bb24249fSLoGin #![feature(panic_info_message)] 16bb24249fSLoGin #![feature(ptr_internals)] 17cde5492fSlogin #![feature(trait_upcasting)] 1840fe15e0SLoGin #![feature(slice_ptr_get)] 1940fe15e0SLoGin #![feature(vec_into_raw_parts)] 2040314b30SXiaoye Zheng #![feature(new_uninit)] 2140314b30SXiaoye Zheng #![feature(ptr_to_from_bits)] 2240314b30SXiaoye Zheng #![feature(concat_idents)] 23fba56231SLoGin #![cfg_attr(target_os = "none", no_std)] 24971462beSGnoCiYeH #![feature(atomic_mut_ptr)] 25fba56231SLoGin 26fba56231SLoGin #[cfg(test)] 27fba56231SLoGin #[macro_use] 28fba56231SLoGin extern crate std; 2940fe15e0SLoGin 302813126eSlogin #[allow(non_upper_case_globals)] 312813126eSlogin #[allow(non_camel_case_types)] 322813126eSlogin #[allow(non_snake_case)] 33fb6c29d0Slogin use core::panic::PanicInfo; 34fb6c29d0Slogin 35adc1846bSlogin /// 导出x86_64架构相关的代码,命名为arch模块 3666f67c6aSlogin #[macro_use] 3766f67c6aSlogin mod arch; 3882d2e446Slogin #[macro_use] 39fb6c29d0Slogin mod libs; 40004e86ffSlogin #[macro_use] 41004e86ffSlogin mod include; 42*7b32f508SLoGin mod debug; 43004e86ffSlogin mod driver; // 如果driver依赖了libs,应该在libs后面导出 44ec53d23eSlogin mod exception; 45004e86ffSlogin mod filesystem; 46fba56231SLoGin mod init; 47004e86ffSlogin mod ipc; 4866f67c6aSlogin mod mm; 494454d1a2Slogin mod net; 5066f67c6aSlogin mod process; 5166f67c6aSlogin mod sched; 5266f67c6aSlogin mod smp; 534454d1a2Slogin mod syscall; 5401876902SGou Ngai mod time; 5540314b30SXiaoye Zheng mod virt; 5682d2e446Slogin 57004e86ffSlogin #[macro_use] 58fb6c29d0Slogin extern crate alloc; 59c2481452Shoumkh #[macro_use] 604454d1a2Slogin extern crate bitflags; 6140fe15e0SLoGin extern crate elf; 624454d1a2Slogin #[macro_use] 63c2481452Shoumkh extern crate lazy_static; 644454d1a2Slogin extern crate num; 65004e86ffSlogin #[macro_use] 664454d1a2Slogin extern crate num_derive; 674454d1a2Slogin extern crate smoltcp; 68fba56231SLoGin #[macro_use] 69fba56231SLoGin extern crate intertrait; 70bacd691cSlogin #[cfg(target_arch = "x86_64")] 71bacd691cSlogin extern crate x86; 72bacd691cSlogin 73*7b32f508SLoGin extern crate klog_types; 74*7b32f508SLoGin 7540fe15e0SLoGin use crate::mm::allocator::kernel_allocator::KernelAllocator; 762813126eSlogin 771496ba7bSLoGin use crate::process::ProcessManager; 78fb6c29d0Slogin 79d470019bSLoGin #[cfg(feature = "backtrace")] 80d470019bSLoGin extern crate mini_backtrace; 81d470019bSLoGin 82d470019bSLoGin extern "C" { 83d470019bSLoGin fn lookup_kallsyms(addr: u64, level: i32) -> i32; 84d470019bSLoGin } 85d470019bSLoGin 86abe3a6eaShanjiezhou // 声明全局的分配器 87fb6c29d0Slogin #[cfg_attr(not(test), global_allocator)] 8840fe15e0SLoGin pub static KERNEL_ALLOCATOR: KernelAllocator = KernelAllocator; 89fb6c29d0Slogin 90fb6c29d0Slogin /// 全局的panic处理函数 91fba56231SLoGin #[cfg(target_os = "none")] 922813126eSlogin #[panic_handler] 932813126eSlogin #[no_mangle] 9466f67c6aSlogin pub fn panic(info: &PanicInfo) -> ! { 9566f67c6aSlogin kerror!("Kernel Panic Occurred."); 9666f67c6aSlogin 9766f67c6aSlogin match info.location() { 9866f67c6aSlogin Some(loc) => { 9966f67c6aSlogin println!( 10066f67c6aSlogin "Location:\n\tFile: {}\n\tLine: {}, Column: {}", 10166f67c6aSlogin loc.file(), 10266f67c6aSlogin loc.line(), 10366f67c6aSlogin loc.column() 10466f67c6aSlogin ); 10566f67c6aSlogin } 10666f67c6aSlogin None => { 10766f67c6aSlogin println!("No location info"); 10866f67c6aSlogin } 10966f67c6aSlogin } 11066f67c6aSlogin 11166f67c6aSlogin match info.message() { 11266f67c6aSlogin Some(msg) => { 11366f67c6aSlogin println!("Message:\n\t{}", msg); 11466f67c6aSlogin } 11566f67c6aSlogin None => { 11666f67c6aSlogin println!("No panic message."); 11766f67c6aSlogin } 11866f67c6aSlogin } 11966f67c6aSlogin 120d470019bSLoGin #[cfg(feature = "backtrace")] 121d470019bSLoGin { 122d470019bSLoGin unsafe { 123d470019bSLoGin let bt = mini_backtrace::Backtrace::<16>::capture(); 124d470019bSLoGin println!("Rust Panic Backtrace:"); 125d470019bSLoGin let mut level = 0; 126d470019bSLoGin for frame in bt.frames { 127d470019bSLoGin lookup_kallsyms(frame as u64, level); 128d470019bSLoGin level += 1; 129d470019bSLoGin } 130d470019bSLoGin }; 131d470019bSLoGin } 132d470019bSLoGin 1331496ba7bSLoGin println!("Current PCB:\n\t{:?}", *(ProcessManager::current_pcb())); 1341496ba7bSLoGin ProcessManager::exit(usize::MAX); 1352813126eSlogin } 136