Jeremy Soller
40 lines
1014 B
Rust
40 lines
1014 B
Rust
#![deny(warnings)]
|
|
#![feature(alloc)]
|
|
#![feature(allocator_api)]
|
|
#![feature(const_fn)]
|
|
#![no_std]
|
|
|
|
extern crate alloc;
|
|
extern crate spin;
|
|
extern crate linked_list_allocator;
|
|
|
|
use alloc::heap::{Alloc, AllocErr, Layout};
|
|
use spin::Mutex;
|
|
use linked_list_allocator::Heap;
|
|
|
|
static HEAP: Mutex<Option<Heap>> = Mutex::new(None);
|
|
|
|
pub unsafe fn init(offset: usize, size: usize) {
|
|
*HEAP.lock() = Some(Heap::new(offset, size));
|
|
}
|
|
|
|
pub struct Allocator;
|
|
|
|
unsafe impl<'a> Alloc for &'a Allocator {
|
|
unsafe fn alloc(&mut self, layout: Layout) -> Result<*mut u8, AllocErr> {
|
|
if let Some(ref mut heap) = *HEAP.lock() {
|
|
heap.allocate_first_fit(layout)
|
|
} else {
|
|
panic!("__rust_allocate: heap not initialized");
|
|
}
|
|
}
|
|
|
|
unsafe fn dealloc(&mut self, ptr: *mut u8, layout: Layout) {
|
|
if let Some(ref mut heap) = *HEAP.lock() {
|
|
heap.deallocate(ptr, layout)
|
|
} else {
|
|
panic!("__rust_deallocate: heap not initialized");
|
|
}
|
|
}
|
|
}
|