12813126eSlogin #![no_main] // <1> 2bb24249fSLoGin #![feature(alloc_error_handler)] 340fe15e0SLoGin #![feature(allocator_api)] 440fe15e0SLoGin #![feature(arbitrary_self_types)] 51496ba7bSLoGin #![feature(asm_const)] 65c4224e5SLoGin #![feature(concat_idents)] 7c566df45SLoGin #![feature(const_for)] 8935f40ecSlogin #![feature(const_mut_refs)] 906d5e247SLoGin #![feature(const_trait_impl)] 1092849878SLoGin #![feature(const_transmute_copy)] 11a03c4f9dSLoGin #![feature(const_refs_to_cell)] 1240fe15e0SLoGin #![feature(core_intrinsics)] 1313776c11Slogin #![feature(c_void_variant)] 141a72a751SLoGin #![feature(extract_if)] 155c4224e5SLoGin #![feature(fn_align)] 1670a4e555SLoGin #![feature(inline_const)] 171496ba7bSLoGin #![feature(naked_functions)] 185c4224e5SLoGin #![feature(new_uninit)] 19bb24249fSLoGin #![feature(panic_info_message)] 20bb24249fSLoGin #![feature(ptr_internals)] 215c4224e5SLoGin #![feature(ptr_to_from_bits)] 22cde5492fSlogin #![feature(trait_upcasting)] 2340fe15e0SLoGin #![feature(slice_ptr_get)] 2440fe15e0SLoGin #![feature(vec_into_raw_parts)] 25fba56231SLoGin #![cfg_attr(target_os = "none", no_std)] 26*840045afSLoGin // clippy的配置 27*840045afSLoGin #![deny(clippy::all)] 28*840045afSLoGin // DragonOS允许在函数中使用return语句(尤其是长函数时,我们推荐这么做) 29*840045afSLoGin #![allow(clippy::let_and_return)] 30*840045afSLoGin #![allow(clippy::needless_pass_by_ref_mut)] 31*840045afSLoGin #![allow(clippy::needless_return)] 32*840045afSLoGin #![allow(clippy::upper_case_acronyms)] 33fba56231SLoGin 34fba56231SLoGin #[cfg(test)] 35fba56231SLoGin #[macro_use] 36fba56231SLoGin extern crate std; 3740fe15e0SLoGin 382813126eSlogin #[allow(non_upper_case_globals)] 392813126eSlogin #[allow(non_camel_case_types)] 402813126eSlogin #[allow(non_snake_case)] 41fb6c29d0Slogin use core::panic::PanicInfo; 42fb6c29d0Slogin 43adc1846bSlogin /// 导出x86_64架构相关的代码,命名为arch模块 4466f67c6aSlogin #[macro_use] 4566f67c6aSlogin mod arch; 4682d2e446Slogin #[macro_use] 47fb6c29d0Slogin mod libs; 48004e86ffSlogin #[macro_use] 49004e86ffSlogin mod include; 507b32f508SLoGin mod debug; 51004e86ffSlogin mod driver; // 如果driver依赖了libs,应该在libs后面导出 52ec53d23eSlogin mod exception; 53004e86ffSlogin mod filesystem; 54fba56231SLoGin mod init; 55004e86ffSlogin mod ipc; 56196b75dcSLoGin mod misc; 5766f67c6aSlogin mod mm; 584454d1a2Slogin mod net; 5966f67c6aSlogin mod process; 6066f67c6aSlogin mod sched; 6166f67c6aSlogin mod smp; 624454d1a2Slogin mod syscall; 6301876902SGou Ngai mod time; 644fda81ceSLoGin 654fda81ceSLoGin #[cfg(target_arch = "x86_64")] 6640314b30SXiaoye Zheng mod virt; 6782d2e446Slogin 68004e86ffSlogin #[macro_use] 69fb6c29d0Slogin extern crate alloc; 70c2481452Shoumkh #[macro_use] 71393f6915SLoGin extern crate atomic_enum; 72393f6915SLoGin #[macro_use] 734454d1a2Slogin extern crate bitflags; 7440fe15e0SLoGin extern crate elf; 754454d1a2Slogin #[macro_use] 76c2481452Shoumkh extern crate lazy_static; 774454d1a2Slogin extern crate num; 78004e86ffSlogin #[macro_use] 794454d1a2Slogin extern crate num_derive; 804454d1a2Slogin extern crate smoltcp; 81fba56231SLoGin #[macro_use] 82fba56231SLoGin extern crate intertrait; 83bacd691cSlogin #[cfg(target_arch = "x86_64")] 84bacd691cSlogin extern crate x86; 85bacd691cSlogin 867b32f508SLoGin extern crate klog_types; 877a29d4fcSLoGin extern crate uefi; 887a29d4fcSLoGin extern crate uefi_raw; 897b32f508SLoGin 9040fe15e0SLoGin use crate::mm::allocator::kernel_allocator::KernelAllocator; 912813126eSlogin 921496ba7bSLoGin use crate::process::ProcessManager; 93fb6c29d0Slogin 944fda81ceSLoGin #[cfg(all(feature = "backtrace", target_arch = "x86_64"))] 95d470019bSLoGin extern crate mini_backtrace; 96d470019bSLoGin 97d470019bSLoGin extern "C" { 98d470019bSLoGin fn lookup_kallsyms(addr: u64, level: i32) -> i32; 99d470019bSLoGin } 100d470019bSLoGin 101abe3a6eaShanjiezhou // 声明全局的分配器 102fb6c29d0Slogin #[cfg_attr(not(test), global_allocator)] 10340fe15e0SLoGin pub static KERNEL_ALLOCATOR: KernelAllocator = KernelAllocator; 104fb6c29d0Slogin 105fb6c29d0Slogin /// 全局的panic处理函数 106fba56231SLoGin #[cfg(target_os = "none")] 1072813126eSlogin #[panic_handler] 1082813126eSlogin #[no_mangle] 10966f67c6aSlogin pub fn panic(info: &PanicInfo) -> ! { 11066f67c6aSlogin kerror!("Kernel Panic Occurred."); 11166f67c6aSlogin 11266f67c6aSlogin match info.location() { 11366f67c6aSlogin Some(loc) => { 11466f67c6aSlogin println!( 11566f67c6aSlogin "Location:\n\tFile: {}\n\tLine: {}, Column: {}", 11666f67c6aSlogin loc.file(), 11766f67c6aSlogin loc.line(), 11866f67c6aSlogin loc.column() 11966f67c6aSlogin ); 12066f67c6aSlogin } 12166f67c6aSlogin None => { 12266f67c6aSlogin println!("No location info"); 12366f67c6aSlogin } 12466f67c6aSlogin } 12566f67c6aSlogin 12666f67c6aSlogin match info.message() { 12766f67c6aSlogin Some(msg) => { 12866f67c6aSlogin println!("Message:\n\t{}", msg); 12966f67c6aSlogin } 13066f67c6aSlogin None => { 13166f67c6aSlogin println!("No panic message."); 13266f67c6aSlogin } 13366f67c6aSlogin } 13466f67c6aSlogin 1354fda81ceSLoGin #[cfg(all(feature = "backtrace", target_arch = "x86_64"))] 136d470019bSLoGin { 137d470019bSLoGin unsafe { 138d470019bSLoGin let bt = mini_backtrace::Backtrace::<16>::capture(); 139d470019bSLoGin println!("Rust Panic Backtrace:"); 140d470019bSLoGin let mut level = 0; 141d470019bSLoGin for frame in bt.frames { 142d470019bSLoGin lookup_kallsyms(frame as u64, level); 143d470019bSLoGin level += 1; 144d470019bSLoGin } 145d470019bSLoGin }; 146d470019bSLoGin } 147d470019bSLoGin 1481496ba7bSLoGin println!("Current PCB:\n\t{:?}", *(ProcessManager::current_pcb())); 1491496ba7bSLoGin ProcessManager::exit(usize::MAX); 1502813126eSlogin } 151