Implement the Allocator trait from the unstable allocator_api

Author: rmja

Cargo.toml

@@ -23,6 +23,9 @@ license = "MIT OR Apache-2.0"
 name = "embedded-alloc"
 version = "0.5.0"
 
+[features]
+allocator_api = []
+
 [dependencies]
 critical-section = "1.0"
 

rust-toolchain

@@ -0,0 +1 @@
+nightly

src/lib.rs

@@ -1,5 +1,9 @@
 #![doc = include_str!("../README.md")]
 #![no_std]
+#![cfg_attr(
+    feature = "allocator_api",
+    feature(allocator_api, nonnull_slice_from_raw_parts, alloc_layout_extra)
+)]
 
 use core::alloc::{GlobalAlloc, Layout};
 use core::cell::RefCell;
@@ -88,3 +92,40 @@ unsafe impl GlobalAlloc for Heap {
         });
     }
 }
+
+#[cfg(feature = "allocator_api")]
+mod allocator_api {
+    use core::{
+        alloc::{AllocError, Allocator, Layout},
+        ptr::NonNull,
+    };
+
+    use crate::Heap;
+
+    unsafe impl Allocator for Heap {
+        fn allocate(&self, layout: Layout) -> Result<NonNull<[u8]>, AllocError> {
+            match layout.size() {
+                0 => Ok(NonNull::slice_from_raw_parts(layout.dangling(), 0)),
+                size => critical_section::with(|cs| {
+                    self.heap
+                        .borrow(cs)
+                        .borrow_mut()
+                        .allocate_first_fit(layout)
+                        .map(|allocation| NonNull::slice_from_raw_parts(allocation, size))
+                        .map_err(|_| AllocError)
+                }),
+            }
+        }
+
+        unsafe fn deallocate(&self, ptr: NonNull<u8>, layout: Layout) {
+            if layout.size() != 0 {
+                critical_section::with(|cs| {
+                    self.heap
+                        .borrow(cs)
+                        .borrow_mut()
+                        .deallocate(NonNull::new_unchecked(ptr.as_ptr()), layout)
+                });
+            }
+        }
+    }
+}