feat: add get_field function

1 files changed, 31 insertions, 0 deletions

diff --git a/src/util.rs b/src/util.rs
new file mode 100644
index 0000000..6b7180a
--- /dev/null
+++ b/src/util.rs
@@ -0,0 +1,31 @@
+use std::mem::MaybeUninit;
+
+pub trait MaybeUninitByteSlice
+{
+    unsafe fn cast<Item: 'static>(&self) -> &[Item];
+}
+
+impl MaybeUninitByteSlice for &[MaybeUninit<u8>]
+{
+    unsafe fn cast<Item: 'static>(&self) -> &[Item]
+    {
+        assert_eq!(self.len() % size_of::<Item>(), 0, "Invalid item size");
+
+        assert_eq!(
+            self.as_ptr().addr() % align_of::<Item>(),
+            0,
+            "Invalid item alignment"
+        );
+
+        if size_of::<Item>() == 0 {
+            return &[];
+        }
+
+        unsafe {
+            std::slice::from_raw_parts(
+                self.as_ptr().cast::<Item>(),
+                self.len() / size_of::<Item>(),
+            )
+        }
+    }
+}