helpers: Add AlignedBuffer

AlignedBuffer is a helper class that manages the livetime of a memory region, allocated using
a certain alignment. Like Box, it handles deallocation when the object isn't used anymore.

Author: seijikun

uefi/src/helpers/aligned_buffer.rs

@@ -0,0 +1,45 @@
+// SPDX-License-Identifier: MIT OR Apache-2.0
+
+use alloc::alloc::{alloc, dealloc, Layout, LayoutError};
+
+/// Helper class to maintain the lifetime of a memory region allocated with a non-standard alignment.
+/// Facilitates RAII to properly deallocate when lifetime of the object ends.
+#[derive(Debug)]
+pub struct AlignedBuffer {
+    ptr: *mut u8,
+    layout: Layout,
+}
+impl AlignedBuffer {
+    /// Allocate a new memory region with the requested len and alignment.
+    pub fn alloc(len: usize, alignment: usize) -> Result<Self, LayoutError> {
+        let layout = Layout::from_size_align(len, alignment)?;
+        let ptr = unsafe { alloc(layout) };
+        Ok(Self { ptr, layout })
+    }
+    /// Get a pointer to the aligned memory region managed by this instance.
+    pub fn ptr(&self) -> *const u8 {
+        self.ptr as *const u8
+    }
+    /// Get a mutable pointer to the aligned memory region managed by this instance.
+    pub fn ptr_mut(&mut self) -> *mut u8 {
+        self.ptr as *mut u8
+    }
+    /// Get the size of the aligned memory region managed by this instance.
+    pub fn len(&self) -> usize {
+        self.layout.size()
+    }
+    /// Fill the aligned memory region with data from the given buffer.
+    pub fn copy_from(&mut self, data: &[u8]) {
+        let len = data.len().min(self.len());
+        unsafe {
+            self.ptr.copy_from(data.as_ptr(), len);
+        }
+    }
+}
+impl Drop for AlignedBuffer {
+    fn drop(&mut self) {
+        unsafe {
+            dealloc(self.ptr, self.layout);
+        }
+    }
+}

uefi/src/helpers/mod.rs

@@ -24,6 +24,12 @@ use crate::Result;
 #[doc(hidden)]
 pub use println::_print;
 
+#[cfg(feature = "alloc")]
+mod aligned_buffer;
+#[doc(hidden)]
+#[cfg(feature = "alloc")]
+pub use aligned_buffer::AlignedBuffer;
+
 #[cfg(feature = "global_allocator")]
 mod global_allocator;
 #[cfg(feature = "logger")]