Add alignment

rusterlium · Mar 15, 2024 · f63205a · f63205a
1 parent b13d16c
commit f63205a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/rustler/src/alloc.rs b/rustler/src/alloc.rs
@@ -10,7 +10,10 @@ pub struct EnifAllocator;
 
 unsafe impl GlobalAlloc for EnifAllocator {
     unsafe fn alloc(&self, layout: Layout) -> *mut u8 {
-        // TODO: Check the requested alignment.
+        // TODO: Check enif_alloc's real alignment
+        if layout.align() > (((((layout.size()) - 1) / 8) + 1) * 8) {
+            return std::ptr::null_mut();
+        }
         rustler_sys::enif_alloc(layout.size()) as *mut u8
     }