mirror of
https://github.com/rcore-os/rCore.git
synced 2024-11-27 02:03:29 +04:00
153 lines
3.6 KiB
Rust
153 lines
3.6 KiB
Rust
#![feature(ptr_internals)]
|
||
#![feature(lang_items)]
|
||
#![feature(const_fn)]
|
||
#![feature(alloc)]
|
||
#![feature(const_unique_new, const_atomic_usize_new)]
|
||
#![feature(unique)]
|
||
#![feature(allocator_api)]
|
||
#![feature(global_allocator)]
|
||
#![feature(abi_x86_interrupt)]
|
||
#![feature(iterator_step_by)]
|
||
#![feature(unboxed_closures)]
|
||
#![feature(naked_functions)]
|
||
#![feature(asm)]
|
||
#![no_std]
|
||
|
||
|
||
#[macro_use]
|
||
extern crate alloc;
|
||
|
||
extern crate rlibc;
|
||
extern crate volatile;
|
||
extern crate spin;
|
||
extern crate multiboot2;
|
||
#[macro_use]
|
||
extern crate bitflags;
|
||
#[macro_use]
|
||
extern crate x86_64;
|
||
#[macro_use]
|
||
extern crate once;
|
||
extern crate linked_list_allocator;
|
||
#[macro_use]
|
||
extern crate lazy_static;
|
||
extern crate bit_field;
|
||
extern crate syscall;
|
||
extern crate xmas_elf;
|
||
|
||
#[macro_use] // print!
|
||
mod io;
|
||
mod memory;
|
||
mod lang;
|
||
mod util;
|
||
#[macro_use]
|
||
mod macros;
|
||
mod consts;
|
||
mod process;
|
||
|
||
#[allow(dead_code)]
|
||
#[cfg(target_arch = "x86_64")]
|
||
#[path = "arch/x86_64/mod.rs"]
|
||
mod arch;
|
||
|
||
// The entry point of Rust kernel
|
||
#[no_mangle]
|
||
pub extern "C" fn rust_main(multiboot_information_address: usize) -> ! {
|
||
arch::cpu::init();
|
||
|
||
// ATTENTION: we have a very small stack and no guard page
|
||
println!("Hello World{}", "!");
|
||
|
||
let boot_info = unsafe { multiboot2::load(multiboot_information_address) };
|
||
|
||
// set up guard page and map the heap pages
|
||
let mut memory_controller = memory::init(boot_info);
|
||
unsafe {
|
||
use consts::{KERNEL_HEAP_OFFSET, KERNEL_HEAP_SIZE};
|
||
HEAP_ALLOCATOR.lock().init(KERNEL_HEAP_OFFSET, KERNEL_HEAP_SIZE);
|
||
}
|
||
|
||
arch::gdt::init();
|
||
arch::idt::init();
|
||
|
||
test!(global_allocator);
|
||
test!(guard_page);
|
||
test!(find_mp);
|
||
|
||
let acpi = arch::driver::init(
|
||
|addr: usize| memory_controller.map_page_identity(addr));
|
||
// memory_controller.print_page_table();
|
||
|
||
// FIXME: 开启SMP后，导致switch_to_user中设置rsp无效
|
||
// arch::smp::start_other_cores(&acpi, &mut memory_controller);
|
||
process::init(&mut memory_controller);
|
||
|
||
unsafe{ arch::interrupt::enable(); }
|
||
|
||
unsafe{
|
||
use arch::syscall;
|
||
// 在用户模式下触发时钟中断，会导致GPF
|
||
// （可能是由于没有合理分离栈）
|
||
no_interrupt!({
|
||
syscall::switch_to_user();
|
||
println!("Now in user mode");
|
||
syscall::switch_to_kernel();
|
||
println!("Now in kernel mode");
|
||
});
|
||
}
|
||
|
||
loop{
|
||
println!("init ...");
|
||
let mut i = 0;
|
||
while i < 1 << 22 {
|
||
i += 1;
|
||
}
|
||
}
|
||
|
||
test_end!();
|
||
unreachable!();
|
||
}
|
||
|
||
#[no_mangle]
|
||
pub extern "C" fn other_main() -> ! {
|
||
arch::cpu::init();
|
||
arch::gdt::init();
|
||
arch::idt::init();
|
||
arch::driver::apic::other_init();
|
||
let cpu_id = arch::driver::apic::lapic_id();
|
||
println!("Hello world! from CPU {}!", arch::driver::apic::lapic_id());
|
||
unsafe{ arch::smp::notify_started(cpu_id); }
|
||
// unsafe{ let a = *(0xdeadbeaf as *const u8); } // Page fault
|
||
loop {}
|
||
}
|
||
|
||
use linked_list_allocator::LockedHeap;
|
||
|
||
#[global_allocator]
|
||
static HEAP_ALLOCATOR: LockedHeap = LockedHeap::empty();
|
||
|
||
mod test {
|
||
pub fn global_allocator() {
|
||
for i in 0..10000 {
|
||
format!("Some String");
|
||
}
|
||
}
|
||
pub fn find_mp() {
|
||
use arch;
|
||
let mp = arch::driver::mp::find_mp();
|
||
assert!(mp.is_some());
|
||
}
|
||
pub fn guard_page() {
|
||
use x86_64;
|
||
// invoke a breakpoint exception
|
||
x86_64::instructions::interrupts::int3();
|
||
|
||
fn stack_overflow() {
|
||
stack_overflow(); // for each recursion, the return address is pushed
|
||
}
|
||
|
||
// trigger a stack overflow
|
||
stack_overflow();
|
||
|
||
println!("It did not crash!");
|
||
}
|
||
} |