49 lines
1.0 KiB
Rust
49 lines
1.0 KiB
Rust
#![no_std]
|
|
#![no_main]
|
|
#![feature(custom_test_frameworks)]
|
|
#![test_runner(xe_os::test_runner)]
|
|
#![reexport_test_harness_main = "test_main"]
|
|
|
|
extern crate alloc;
|
|
|
|
use alloc::boxed::Box;
|
|
use bootloader::{entry_point, BootInfo};
|
|
use core::panic::PanicInfo;
|
|
use xe_os::println;
|
|
|
|
entry_point!(kernel_main);
|
|
|
|
fn kernel_main(boot_info: &'static BootInfo) -> ! {
|
|
use x86_64::VirtAddr;
|
|
use xe_os::allocator;
|
|
use xe_os::memory::{self, BootInfoFrameAllocator};
|
|
|
|
xe_os::init(&boot_info);
|
|
println!("Hello World{}", "!");
|
|
|
|
// allocate a number on the heap
|
|
let heap_value = Box::new(41);
|
|
println!("heap_value at {:p}", heap_value);
|
|
|
|
#[cfg(test)]
|
|
test_main();
|
|
|
|
println!("It did not crash!");
|
|
xe_os::hlt_loop();
|
|
}
|
|
|
|
// our existing panic handler
|
|
#[cfg(not(test))] // new attribute
|
|
#[panic_handler]
|
|
fn panic(info: &PanicInfo) -> ! {
|
|
println!("{}", info);
|
|
xe_os::hlt_loop();
|
|
}
|
|
|
|
// our panic handler in test mode
|
|
#[cfg(test)]
|
|
#[panic_handler]
|
|
fn panic(info: &PanicInfo) -> ! {
|
|
xe_os::test_panic_handler(info);
|
|
}
|