src/libstd/sys/sgx/alloc.rs

   1 use crate::alloc::{GlobalAlloc, Layout, System};
   2
   3 use super::waitqueue::SpinMutex;
   4
   5 // Using a SpinMutex because we never want to exit the enclave waiting for the
   6 // allocator.
   7 static DLMALLOC: SpinMutex<dlmalloc::Dlmalloc> = SpinMutex::new(dlmalloc::DLMALLOC_INIT);
   8
   9 #[stable(feature = "alloc_system_type", since = "1.28.0")]
  10 unsafe impl GlobalAlloc for System {
  11     #[inline]
  12     unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
  13         DLMALLOC.lock().malloc(layout.size(), layout.align())
  14     }
  15
  16     #[inline]
  17     unsafe fn alloc_zeroed(&self, layout: Layout) -> *mut u8 {
  18         DLMALLOC.lock().calloc(layout.size(), layout.align())
  19     }
  20
  21     #[inline]
  22     unsafe fn dealloc(&self, ptr: *mut u8, layout: Layout) {
  23         DLMALLOC.lock().free(ptr, layout.size(), layout.align())
  24     }
  25
  26     #[inline]
  27     unsafe fn realloc(&self, ptr: *mut u8, layout: Layout, new_size: usize) -> *mut u8 {
  28         DLMALLOC.lock().realloc(ptr, layout.size(), layout.align(), new_size)
  29     }
  30 }