uefi: allocator: implement Allocator trait from allocator_api

We have not been requested yet to do this, but this gives downstream
users more flexibility.

Author: phip1611

uefi/CHANGELOG.md

@@ -28,8 +28,10 @@
 - The `Display` impl for `CStr8` now excludes the trailing null character.
 - `VariableKeys` initializes with a larger name buffer to work around firmware
   bugs on some devices.
-- The UEFI `allocator::Allocator` has been optimized for page-aligned 
+- The UEFI `allocator::Allocator` has been optimized for page-aligned
   allocations.
+- The UEFI `allocator::Allocator` now implements `core::alloc::Allocator`
+  (`allocator_api`), when the `--unstable` feature is used.
 
 
 # uefi - 0.34.1 (2025-02-07)

uefi/src/allocator.rs

@@ -20,6 +20,9 @@ use core::ptr::{self, NonNull};
 use core::sync::atomic::{AtomicU32, Ordering};
 use uefi_raw::table::boot::PAGE_SIZE;
 
+#[cfg(feature = "unstable")]
+use core::alloc as alloc_api;
+
 /// Get the memory type to use for allocation.
 ///
 /// The first time this is called, the data type of the loaded image will be
@@ -160,3 +163,17 @@ unsafe impl GlobalAlloc for Allocator {
         }
     }
 }
+
+#[cfg(feature = "unstable")]
+unsafe impl alloc_api::Allocator for Allocator {
+    fn allocate(&self, layout: Layout) -> Result<NonNull<[u8]>, alloc_api::AllocError> {
+        let ptr = unsafe { <Allocator as GlobalAlloc>::alloc(self, layout) };
+        NonNull::new(ptr)
+            .ok_or(alloc_api::AllocError)
+            .map(|ptr| NonNull::slice_from_raw_parts(ptr, layout.size()))
+    }
+
+    unsafe fn deallocate(&self, ptr: NonNull<u8>, layout: Layout) {
+        unsafe { <Allocator as GlobalAlloc>::dealloc(self, ptr.as_ptr(), layout) }
+    }
+}