feat: adding more utility methods to Ptr64 and Array

Author: WolverinDEV

raw_struct/src/builtins/array.rs

@@ -24,10 +24,9 @@ use crate::{
 
 pub trait Array<T: ?Sized> {
     fn start_address(&self) -> u64;
+    fn len(&self) -> Option<usize>;
 }
 
-pub trait SizedArray<T: ?Sized, const N: usize>: Array<T> {}
-
 impl<T: FromMemoryView> dyn Array<T> {
     pub fn element_at(&self, memory: &dyn MemoryView, index: usize) -> Result<T, AccessError> {
         let offset = (index * mem::size_of::<T>()) as u64;
@@ -146,3 +145,11 @@ where
         Ok(result.into_iter().map(Copy::<T>::new).collect::<Vec<_>>())
     }
 }
+
+pub trait SizedArray<T: ?Sized, const N: usize>: Array<T> {}
+
+impl<T: ?Sized, const N: usize> dyn SizedArray<T, N> {
+    pub fn len(&self) -> usize {
+        N
+    }
+}

raw_struct/src/builtins/ptr.rs

@@ -38,6 +38,42 @@ impl<T: ?Sized + 'static> Clone for Ptr64<T> {
 }
 impl<T: ?Sized + 'static> marker::Copy for Ptr64<T> {}
 
+impl<T: ?Sized> Ptr64<T> {
+    pub fn is_null(&self) -> bool {
+        self.address == 0
+    }
+
+    pub fn cast<V: ?Sized>(&self) -> Ptr64<V> {
+        Ptr64::<V> {
+            address: self.address,
+            _dummy: Default::default(),
+        }
+    }
+}
+
+impl<T: marker::Copy> Ptr64<T> {
+    /// Create a copy of the value the pointer points to
+    #[must_use = "copied result must be used"]
+    pub fn read_value(&self, memory: &dyn MemoryView) -> Result<Option<T>, AccessError> {
+        if self.address > 0 {
+            let memory = T::read_object(memory, self.address).map_err(|err| AccessError {
+                source: err,
+
+                member: None,
+                object: "T".into(),
+                mode: AccessMode::Read,
+
+                offset: self.address,
+                size: mem::size_of::<T>(),
+            })?;
+
+            Ok(Some(memory))
+        } else {
+            Ok(None)
+        }
+    }
+}
+
 impl<T: ?Sized + Viewable<T>> Ptr64<T> {
     #[must_use]
     pub fn value_reference(&self, memory: Arc<dyn MemoryView>) -> Option<Reference<T>> {
@@ -75,6 +111,10 @@ impl<T> Array<T> for Ptr64<[T]> {
     fn start_address(&self) -> u64 {
         self.address
     }
+
+    fn len(&self) -> Option<usize> {
+        None
+    }
 }
 
 impl<T> Deref for Ptr64<[T]> {
@@ -89,6 +129,10 @@ impl<T, const N: usize> Array<T> for Ptr64<[T; N]> {
     fn start_address(&self) -> u64 {
         self.address
     }
+
+    fn len(&self) -> Option<usize> {
+        Some(N)
+    }
 }
 
 impl<T, const N: usize> Deref for Ptr64<[T; N]> {
@@ -103,6 +147,10 @@ impl<T: ?Sized> Array<T> for Ptr64<dyn Array<T>> {
     fn start_address(&self) -> u64 {
         self.address
     }
+
+    fn len(&self) -> Option<usize> {
+        None
+    }
 }
 
 impl<T: ?Sized> Deref for Ptr64<dyn Array<T>> {
@@ -117,6 +165,10 @@ impl<T: ?Sized, const N: usize> Array<T> for Ptr64<dyn SizedArray<T, N>> {
     fn start_address(&self) -> u64 {
         self.address
     }
+
+    fn len(&self) -> Option<usize> {
+        Some(N)
+    }
 }
 
 impl<T: ?Sized, const N: usize> Deref for Ptr64<dyn SizedArray<T, N>> {