FantasyPvP
70 lines
1.4 KiB
Rust
70 lines
1.4 KiB
Rust
|
|
#![no_std]
|
|
#![no_main]
|
|
#![feature(custom_test_frameworks)]
|
|
#![test_runner(CrystalOS::test_runner)]
|
|
#![reexport_test_harness_main = "test_main"]
|
|
|
|
extern crate alloc;
|
|
|
|
use bootloader::{entry_point, BootInfo};
|
|
use core::panic::PanicInfo;
|
|
|
|
use alloc::{ boxed::Box, vec::Vec };
|
|
|
|
|
|
entry_point!(main);
|
|
|
|
fn main(boot_info: &'static BootInfo) -> ! {
|
|
use CrystalOS::kernel::allocator;
|
|
use CrystalOS::kernel::memory::{self, BootInfoFrameAllocator};
|
|
use x86_64::VirtAddr;
|
|
|
|
CrystalOS::start();
|
|
|
|
let physical_memory_offset = VirtAddr::new(boot_info.physical_memory_offset);
|
|
let mut mapper = unsafe { memory::init(physical_memory_offset)};
|
|
let mut frame_allocator = unsafe {
|
|
BootInfoFrameAllocator::init(&boot_info.memory_map)
|
|
};
|
|
allocator::init_heap(&mut mapper, &mut frame_allocator).expect("failed to initialise heap");
|
|
|
|
test_main();
|
|
|
|
loop {}
|
|
}
|
|
|
|
#[panic_handler]
|
|
fn panic(info: &PanicInfo) -> ! {
|
|
CrystalOS::test_panic_handler(info)
|
|
}
|
|
|
|
#[test_case]
|
|
fn box_allocation() {
|
|
let heap_1 = Box::new(69);
|
|
let heap_2 = Box::new(420);
|
|
assert_eq!(*heap_1, 69);
|
|
assert_eq!(*heap_2, 420);
|
|
}
|
|
|
|
#[test_case]
|
|
fn vec_allocation() {
|
|
let x = 1000;
|
|
let mut vector = Vec::new();
|
|
for i in 0..x {
|
|
vector.push(i);
|
|
}
|
|
assert_eq!(vector.iter().sum::<u64>(), (x-1) * x/2);
|
|
|
|
}
|
|
|
|
#[test_case]
|
|
fn reallocation() {
|
|
use CrystalOS::kernel::allocator::HEAP_SIZE;
|
|
|
|
for i in 0..HEAP_SIZE {
|
|
let x = Box::new(i);
|
|
assert_eq!(*x, i);
|
|
}
|
|
}
|