mirror of https://github.com/barzamin/sparkle
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.6 KiB
49 lines
1.6 KiB
//! Wires rust up to the kheap, so that `alloc::` works.
|
|
|
|
use spin::Mutex;
|
|
use linked_list_allocator::Heap;
|
|
use alloc::allocator::{Alloc, Layout, AllocErr};
|
|
|
|
/// Base location of the kheap.
|
|
pub const HEAP_START: usize = 0o_000_001_000_000_0000;
|
|
/// Size of the kheap.
|
|
pub const HEAP_SIZE: usize = 100 * 1024; // 100 KiB
|
|
|
|
/// Locked ownership of an optional kheap. Initialized on boot; is `None` before then.
|
|
static HEAP: Mutex<Option<Heap>> = Mutex::new(None);
|
|
|
|
/// Initialize the kheap. Called at boot.
|
|
pub unsafe fn heap_init(start: usize, size: usize) {
|
|
*HEAP.lock() = Some(Heap::new(start, size));
|
|
}
|
|
|
|
/// Wraps whatever allocator backend we're using, and implements `alloc::allocator::Alloc`.
|
|
pub struct Allocator;
|
|
|
|
unsafe impl<'a> Alloc for &'a Allocator {
|
|
#[inline]
|
|
unsafe fn alloc(&mut self, layout: Layout) -> Result<*mut u8, AllocErr> {
|
|
if let Some(ref mut heap) = *HEAP.lock() {
|
|
// heap.allocate(layout.size(), layout.align())
|
|
// .ok_or(AllocErr::Exhausted { request: layout })
|
|
heap.allocate_first_fit(layout)
|
|
} else {
|
|
panic!("kheap: attempting alloc w/ uninitialized heap");
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
unsafe fn dealloc(&mut self, ptr: *mut u8, layout: Layout) {
|
|
if let Some(ref mut heap) = *HEAP.lock() {
|
|
heap.deallocate(ptr, layout)
|
|
// heap.deallocate(ptr, layout.size(), layout.align());
|
|
} else {
|
|
panic!("kheap: attempting dealloc w/ uninitialized heap");
|
|
}
|
|
}
|
|
|
|
#[inline]
|
|
fn oom(&mut self, err: AllocErr) -> ! {
|
|
panic!("kheap OOM: {:?}", err);
|
|
}
|
|
}
|