ram: lock for concurrent use
this may be reverted if ram allocation shall be more separate.
This commit is contained in:
parent
4f8a76e29b
commit
6ffcf7d4a4
17
src/ram.rs
17
src/ram.rs
|
@ -1,36 +1,35 @@
|
||||||
use core::cell::RefCell;
|
|
||||||
use core::alloc::GlobalAlloc;
|
use core::alloc::GlobalAlloc;
|
||||||
use core::ptr::NonNull;
|
use core::ptr::NonNull;
|
||||||
use alloc::alloc::Layout;
|
use alloc::alloc::Layout;
|
||||||
use linked_list_allocator::Heap;
|
use linked_list_allocator::Heap;
|
||||||
|
use crate::cortex_a9::mutex::Mutex;
|
||||||
use crate::zynq::ddr::DdrRam;
|
use crate::zynq::ddr::DdrRam;
|
||||||
|
|
||||||
#[global_allocator]
|
#[global_allocator]
|
||||||
static ALLOCATOR: HeapAlloc = HeapAlloc(RefCell::new(Heap::empty()));
|
static ALLOCATOR: CortexA9Alloc = CortexA9Alloc(Mutex::new(Heap::empty()));
|
||||||
|
|
||||||
/// LockedHeap doesn't locking properly
|
/// LockedHeap doesn't locking properly
|
||||||
struct HeapAlloc(RefCell<Heap>);
|
struct CortexA9Alloc(Mutex<Heap>);
|
||||||
|
|
||||||
/// FIXME: unsound; lock properly
|
unsafe impl Sync for CortexA9Alloc {}
|
||||||
unsafe impl Sync for HeapAlloc {}
|
|
||||||
|
|
||||||
unsafe impl GlobalAlloc for HeapAlloc {
|
unsafe impl GlobalAlloc for CortexA9Alloc {
|
||||||
unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
|
unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
|
||||||
self.0.borrow_mut()
|
self.0.lock()
|
||||||
.allocate_first_fit(layout)
|
.allocate_first_fit(layout)
|
||||||
.ok()
|
.ok()
|
||||||
.map_or(0 as *mut u8, |allocation| allocation.as_ptr())
|
.map_or(0 as *mut u8, |allocation| allocation.as_ptr())
|
||||||
}
|
}
|
||||||
|
|
||||||
unsafe fn dealloc(&self, ptr: *mut u8, layout: Layout) {
|
unsafe fn dealloc(&self, ptr: *mut u8, layout: Layout) {
|
||||||
self.0.borrow_mut()
|
self.0.lock()
|
||||||
.deallocate(NonNull::new_unchecked(ptr), layout)
|
.deallocate(NonNull::new_unchecked(ptr), layout)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn init_alloc(ddr: &mut DdrRam) {
|
pub fn init_alloc(ddr: &mut DdrRam) {
|
||||||
unsafe {
|
unsafe {
|
||||||
ALLOCATOR.0.borrow_mut()
|
ALLOCATOR.0.lock()
|
||||||
.init(ddr.ptr::<u8>() as usize, ddr.size());
|
.init(ddr.ptr::<u8>() as usize, ddr.size());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue