37 lines
1.2 KiB
Rust
37 lines
1.2 KiB
Rust
|
use core::alloc::{GlobalAlloc, Layout};
|
||
|
|
|
||
|
|
pub struct Allocator;
|
||
|
|
|
||
|
|
#[cfg(unix)]
|
||
|
|
unsafe impl GlobalAlloc for Allocator {
|
||
|
|
unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
|
||
|
|
unsafe { libc::memalign(layout.align(), layout.size()) as *mut u8 }
|
||
|
|
}
|
||
|
|
|
||
|
|
unsafe fn dealloc(&self, ptr: *mut u8, _layout: Layout) {
|
||
|
|
unsafe { libc::free(ptr as *mut libc::c_void) }
|
||
|
|
}
|
||
|
|
|
||
|
|
unsafe fn realloc(&self, ptr: *mut u8, _layout: Layout, new_size: usize) -> *mut u8 {
|
||
|
|
unsafe { libc::realloc(ptr as *mut libc::c_void, new_size) as *mut u8 }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(windows)]
|
||
|
|
unsafe impl GlobalAlloc for Allocator {
|
||
|
|
unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
|
||
|
|
use windows_sys::Win32::System::Memory::*;
|
||
|
|
unsafe { HeapAlloc(GetProcessHeap(), 0, layout.size()) as *mut u8 }
|
||
|
|
}
|
||
|
|
|
||
|
|
unsafe fn dealloc(&self, ptr: *mut u8, _layout: Layout) {
|
||
|
|
use windows_sys::Win32::System::Memory::*;
|
||
|
|
unsafe { HeapFree(GetProcessHeap(), 0, ptr as *mut core::ffi::c_void) };
|
||
|
|
}
|
||
|
|
|
||
|
|
unsafe fn realloc(&self, ptr: *mut u8, _layout: Layout, new_size: usize) -> *mut u8 {
|
||
|
|
use windows_sys::Win32::System::Memory::*;
|
||
|
|
unsafe { HeapReAlloc(GetProcessHeap(), 0, ptr as *mut core::ffi::c_void, new_size) as *mut u8 }
|
||
|
|
}
|
||
|
|
}
|