bjorn3
51 lines
1.4 KiB
Rust
51 lines
1.4 KiB
Rust
use crate::memory::KernelMapper;
|
|
use core::{
|
|
alloc::{GlobalAlloc, Layout},
|
|
ptr::NonNull,
|
|
};
|
|
use linked_list_allocator::Heap;
|
|
use spin::Mutex;
|
|
|
|
static HEAP: Mutex<Option<Heap>> = Mutex::new(None);
|
|
|
|
pub struct Allocator;
|
|
|
|
impl Allocator {
|
|
pub unsafe fn init(offset: usize, size: usize) {
|
|
unsafe {
|
|
*HEAP.lock() = Some(Heap::new(offset, size));
|
|
}
|
|
}
|
|
}
|
|
|
|
unsafe impl GlobalAlloc for Allocator {
|
|
unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
|
|
unsafe {
|
|
while let Some(ref mut heap) = *HEAP.lock() {
|
|
match heap.allocate_first_fit(layout) {
|
|
Ok(ptr) => return ptr.as_ptr(),
|
|
Err(()) => {
|
|
let size = heap.size();
|
|
super::map_heap(
|
|
&mut KernelMapper::lock_rw(),
|
|
crate::KERNEL_HEAP_OFFSET + size,
|
|
super::KERNEL_HEAP_SIZE,
|
|
);
|
|
heap.extend(super::KERNEL_HEAP_SIZE);
|
|
}
|
|
}
|
|
}
|
|
panic!("__rust_allocate: heap not initialized");
|
|
}
|
|
}
|
|
|
|
unsafe fn dealloc(&self, ptr: *mut u8, layout: Layout) {
|
|
unsafe {
|
|
HEAP.lock()
|
|
.as_mut()
|
|
.expect("heap not initialized")
|
|
.deallocate(NonNull::new_unchecked(ptr), layout)
|
|
}
|
|
}
|
|
}
|