From 0ac0dc0a9efe1d35a7ff29ad36945d45dfbb3b1a Mon Sep 17 00:00:00 2001
From: Mees Delzenne <mees.delzenne@gmail.com>
Date: Sun, 11 Feb 2024 17:25:07 +0100
Subject: [PATCH] Mark allocator functions as unsafe

---
 core/src/allocator.rs      |  6 +++---
 core/src/allocator/rust.rs | 12 ++++++------
 2 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/core/src/allocator.rs b/core/src/allocator.rs
index 1344eaf4..de479dca 100644
--- a/core/src/allocator.rs
+++ b/core/src/allocator.rs
@@ -14,13 +14,13 @@ pub type RawMemPtr = *mut u8;
 #[cfg_attr(feature = "doc-cfg", doc(cfg(feature = "allocator")))]
 pub trait Allocator {
     /// Allocate new memory
-    fn alloc(&mut self, size: usize) -> RawMemPtr;
+    unsafe fn alloc(&mut self, size: usize) -> RawMemPtr;
 
     /// De-allocate previously allocated memory
-    fn dealloc(&mut self, ptr: RawMemPtr);
+    unsafe fn dealloc(&mut self, ptr: RawMemPtr);
 
     /// Re-allocate previously allocated memory
-    fn realloc(&mut self, ptr: RawMemPtr, new_size: usize) -> RawMemPtr;
+    unsafe fn realloc(&mut self, ptr: RawMemPtr, new_size: usize) -> RawMemPtr;
 
     /// Get usable size of allocated memory region
     fn usable_size(ptr: RawMemPtr) -> usize
diff --git a/core/src/allocator/rust.rs b/core/src/allocator/rust.rs
index eb2d781a..7ba36dca 100644
--- a/core/src/allocator/rust.rs
+++ b/core/src/allocator/rust.rs
@@ -38,7 +38,7 @@ fn round_size(size: usize) -> usize {
 pub struct RustAllocator;
 
 impl Allocator for RustAllocator {
-    fn alloc(&mut self, size: usize) -> RawMemPtr {
+    unsafe fn alloc(&mut self, size: usize) -> RawMemPtr {
         let size = round_size(size);
         let alloc_size = size + HEADER_SIZE;
         let layout = if let Ok(layout) = Layout::from_size_align(alloc_size, ALLOC_ALIGN) {
@@ -61,7 +61,7 @@ impl Allocator for RustAllocator {
     }
 
     #[allow(clippy::not_unsafe_ptr_arg_deref)]
-    fn dealloc(&mut self, ptr: RawMemPtr) {
+    unsafe fn dealloc(&mut self, ptr: RawMemPtr) {
         let ptr = unsafe { ptr.sub(HEADER_SIZE) };
         let alloc_size = {
             let header = unsafe { &*(ptr as *const Header) };
@@ -73,7 +73,7 @@ impl Allocator for RustAllocator {
     }
 
     #[allow(clippy::not_unsafe_ptr_arg_deref)]
-    fn realloc(&mut self, ptr: RawMemPtr, new_size: usize) -> RawMemPtr {
+    unsafe fn realloc(&mut self, ptr: RawMemPtr, new_size: usize) -> RawMemPtr {
         let new_size = round_size(new_size);
         let ptr = unsafe { ptr.sub(HEADER_SIZE) };
         let alloc_size = {
@@ -116,18 +116,18 @@ mod test {
     struct TestAllocator;
 
     impl Allocator for TestAllocator {
-        fn alloc(&mut self, size: usize) -> crate::allocator::RawMemPtr {
+        unsafe fn alloc(&mut self, size: usize) -> crate::allocator::RawMemPtr {
             let res = RustAllocator.alloc(size);
             ALLOC_SIZE.fetch_add(RustAllocator::usable_size(res), Ordering::AcqRel);
             res
         }
 
-        fn dealloc(&mut self, ptr: crate::allocator::RawMemPtr) {
+        unsafe fn dealloc(&mut self, ptr: crate::allocator::RawMemPtr) {
             ALLOC_SIZE.fetch_sub(RustAllocator::usable_size(ptr), Ordering::AcqRel);
             RustAllocator.dealloc(ptr);
         }
 
-        fn realloc(
+        unsafe fn realloc(
             &mut self,
             ptr: crate::allocator::RawMemPtr,
             new_size: usize,