89 lines
2.3 KiB
Rust
89 lines
2.3 KiB
Rust
#![no_std]
|
|
#![no_main]
|
|
#![feature(custom_test_frameworks)]
|
|
#![test_runner(blog_os::test_runner)]
|
|
#![reexport_test_harness_main = "test_main"]
|
|
|
|
extern crate alloc;
|
|
|
|
use alloc::{boxed::Box, rc::Rc, vec, vec::Vec};
|
|
use core::panic::PanicInfo;
|
|
|
|
use blog_os::println;
|
|
use blog_os::task::{keyboard, Task, executor::Executor};
|
|
use bootloader::{entry_point, BootInfo};
|
|
|
|
/// This function is called on panic.
|
|
#[cfg(not(test))]
|
|
#[panic_handler]
|
|
fn panic(info: &PanicInfo) -> ! {
|
|
println!("{}", info);
|
|
blog_os::hlt_loop();
|
|
}
|
|
|
|
#[cfg(test)]
|
|
#[panic_handler]
|
|
fn panic(info: &PanicInfo) -> ! {
|
|
blog_os::test_panic_handler(info)
|
|
}
|
|
|
|
entry_point!(kernel_main);
|
|
|
|
fn kernel_main(boot_info: &'static BootInfo) -> ! {
|
|
use blog_os::allocator;
|
|
use blog_os::memory::{self, BootInfoFrameAllocator};
|
|
use x86_64::VirtAddr;
|
|
|
|
println!("Hello World{}", "!");
|
|
blog_os::init();
|
|
|
|
let mut mapper = unsafe { memory::init(VirtAddr::new(boot_info.physical_memory_offset)) };
|
|
let mut frame_allocator = unsafe { BootInfoFrameAllocator::init(&boot_info.memory_map) };
|
|
|
|
allocator::init_heap(&mut mapper, &mut frame_allocator).expect("heap initialization failed");
|
|
|
|
// allocate a number on the heap
|
|
let heap_value = Box::new(41);
|
|
println!("heap_value at {:p}", heap_value);
|
|
|
|
// create a dynamically sized vector
|
|
let mut vec = Vec::new();
|
|
for i in 0..500 {
|
|
vec.push(i);
|
|
}
|
|
println!("vec at {:p}", vec.as_slice());
|
|
|
|
// create a reference counted vector -> will be freed when count reaches 0
|
|
let reference_counted = Rc::new(vec![1, 2, 3]);
|
|
let cloned_reference = reference_counted.clone();
|
|
println!(
|
|
"current reference count is {}",
|
|
Rc::strong_count(&cloned_reference)
|
|
);
|
|
core::mem::drop(reference_counted);
|
|
println!(
|
|
"reference count is {} now",
|
|
Rc::strong_count(&cloned_reference)
|
|
);
|
|
|
|
let mut executor = Executor::new();
|
|
executor.spawn(Task::new(example_task()));
|
|
executor.spawn(Task::new(keyboard::print_keypresses()));
|
|
executor.run();
|
|
|
|
#[cfg(test)]
|
|
test_main();
|
|
|
|
println!("It did not crash!");
|
|
blog_os::hlt_loop();
|
|
}
|
|
|
|
async fn async_number() -> u32 {
|
|
42
|
|
}
|
|
|
|
async fn example_task() {
|
|
let number = async_number().await;
|
|
println!("async number: {}", number);
|
|
}
|