1 #![no_main] // <1>
2 #![feature(alloc_error_handler)]
3 #![feature(allocator_api)]
4 #![feature(arbitrary_self_types)]
5 #![feature(asm_const)]
6 #![feature(concat_idents)]
7 #![feature(const_for)]
8 #![feature(const_mut_refs)]
9 #![feature(const_trait_impl)]
10 #![feature(const_transmute_copy)]
11 #![feature(const_refs_to_cell)]
12 #![feature(core_intrinsics)]
13 #![feature(c_void_variant)]
14 #![feature(extract_if)]
15 #![feature(fn_align)]
16 #![feature(inline_const)]
17 #![feature(naked_functions)]
18 #![feature(new_uninit)]
19 #![feature(panic_info_message)]
20 #![feature(ptr_internals)]
21 #![feature(ptr_to_from_bits)]
22 #![feature(trait_upcasting)]
23 #![feature(slice_ptr_get)]
24 #![feature(vec_into_raw_parts)]
25 #![cfg_attr(target_os = "none", no_std)]
26 // clippy的配置
27 #![deny(clippy::all)]
28 // DragonOS允许在函数中使用return语句（尤其是长函数时，我们推荐这么做）
29 #![allow(clippy::let_and_return)]
30 #![allow(clippy::needless_pass_by_ref_mut)]
31 #![allow(clippy::needless_return)]
32 #![allow(clippy::upper_case_acronyms)]
33
34 #[cfg(test)]
35 #[macro_use]
36 extern crate std;
37
38 #[allow(non_upper_case_globals)]
39 #[allow(non_camel_case_types)]
40 #[allow(non_snake_case)]
41 use core::panic::PanicInfo;
42
43 /// 导出x86_64架构相关的代码，命名为arch模块
44 #[macro_use]
45 mod arch;
46 #[macro_use]
47 mod libs;
48 #[macro_use]
49 mod include;
50 mod debug;
51 mod driver; // 如果driver依赖了libs，应该在libs后面导出
52 mod exception;
53 mod filesystem;
54 mod init;
55 mod ipc;
56 mod misc;
57 mod mm;
58 mod net;
59 mod process;
60 mod sched;
61 mod smp;
62 mod syscall;
63 mod time;
64
65 #[cfg(target_arch = "x86_64")]
66 mod virt;
67
68 #[macro_use]
69 extern crate alloc;
70 #[macro_use]
71 extern crate atomic_enum;
72 #[macro_use]
73 extern crate bitflags;
74 extern crate elf;
75 #[macro_use]
76 extern crate lazy_static;
77 extern crate num;
78 #[macro_use]
79 extern crate num_derive;
80 extern crate smoltcp;
81 #[macro_use]
82 extern crate intertrait;
83 #[cfg(target_arch = "x86_64")]
84 extern crate x86;
85
86 extern crate klog_types;
87 extern crate uefi;
88 extern crate uefi_raw;
89
90 use crate::mm::allocator::kernel_allocator::KernelAllocator;
91
92 use crate::process::ProcessManager;
93
94 #[cfg(all(feature = "backtrace", target_arch = "x86_64"))]
95 extern crate mini_backtrace;
96
97 extern "C" {
lookup_kallsyms(addr: u64, level: i32) -> i3298 fn lookup_kallsyms(addr: u64, level: i32) -> i32;
99 }
100
101 // 声明全局的分配器
102 #[cfg_attr(not(test), global_allocator)]
103 pub static KERNEL_ALLOCATOR: KernelAllocator = KernelAllocator;
104
105 /// 全局的panic处理函数
106 #[cfg(target_os = "none")]
107 #[panic_handler]
108 #[no_mangle]
panic(info: &PanicInfo) -> !109 pub fn panic(info: &PanicInfo) -> ! {
110 kerror!("Kernel Panic Occurred.");
111
112 match info.location() {
113 Some(loc) => {
114 println!(
115 "Location:\n\tFile: {}\n\tLine: {}, Column: {}",
116 loc.file(),
117 loc.line(),
118 loc.column()
119 );
120 }
121 None => {
122 println!("No location info");
123 }
124 }
125
126 match info.message() {
127 Some(msg) => {
128 println!("Message:\n\t{}", msg);
129 }
130 None => {
131 println!("No panic message.");
132 }
133 }
134
135 #[cfg(all(feature = "backtrace", target_arch = "x86_64"))]
136 {
137 unsafe {
138 let bt = mini_backtrace::Backtrace::<16>::capture();
139 println!("Rust Panic Backtrace:");
140 let mut level = 0;
141 for frame in bt.frames {
142 lookup_kallsyms(frame as u64, level);
143 level += 1;
144 }
145 };
146 }
147
148 println!("Current PCB:\n\t{:?}", *(ProcessManager::current_pcb()));
149 ProcessManager::exit(usize::MAX);
150 }
151