diff --git a/ch2/user/src/bin/00hello_world.rs b/ch2/user/src/bin/00hello_world.rs index 965827b..6995fb6 100644 --- a/ch2/user/src/bin/00hello_world.rs +++ b/ch2/user/src/bin/00hello_world.rs @@ -4,6 +4,7 @@ use user_lib::*; #[no_mangle] -fn main() { +fn main() -> i32 { println!("123"); + 0 } \ No newline at end of file diff --git a/ch2/user/src/lib.rs b/ch2/user/src/lib.rs index e859d12..23b813a 100644 --- a/ch2/user/src/lib.rs +++ b/ch2/user/src/lib.rs @@ -2,12 +2,37 @@ #![feature(linkage)] // 开启弱链接特性 #![feature(panic_info_message)] -pub mod syscall; + pub mod user_lang_items; pub use user_lang_items::*; +pub mod syscall; +use syscall::{sys_exit}; + +extern "C" { + fn start_bss(); + fn end_bss(); +} + + +// 只要使用这个包, 那么这里就会作为bin程序的开始 +#[no_mangle] +#[link_section = ".text.entry"] // 链接到指定的段 +pub extern "C" fn _start() -> ! { + clear_bss(); + sys_exit(main()); // 这里执行的main程序是我们 bin文件夹里面的main, 而不是下面的, 下面的main程序只有在bin程序没有main函数的时候才会链接 + // 正常是不会走到这一步的, 因为上面已经退出了程序 + panic!("unreachable after sys_exit!"); +} + #[linkage = "weak"] // 设置我们默认的main函数, 弱链接 #[no_mangle] fn main() -> i32 { panic!("Cannot find main!"); -} \ No newline at end of file +} + +fn clear_bss() { + unsafe { + (start_bss as usize..end_bss as usize).for_each(|p| (p as *mut u8).write_unaligned(0)) + }; +}