uefi-raw

Author: phip1611

uefi-raw/src/lib.rs

@@ -153,14 +153,14 @@ impl IpAddress {
 
     /// Returns a raw pointer to the IP address.
     #[must_use]
-    pub const fn as_ptr(&self) -> *const u8 {
-        core::ptr::addr_of!(*self).cast()
+    pub const fn as_ptr(&self) -> *const Self {
+        core::ptr::addr_of!(*self)
     }
 
     /// Returns a raw mutable pointer to the IP address.
     #[must_use]
-    pub fn as_ptr_mut(&mut self) -> *mut u8 {
-        core::ptr::addr_of_mut!(*self).cast()
+    pub fn as_ptr_mut(&mut self) -> *mut Self {
+        core::ptr::addr_of_mut!(*self)
     }
 }
 
@@ -262,11 +262,12 @@ mod tests {
     #[test]
     fn ip_ptr() {
         let mut ip = IpAddress::new_v4([0; 4]);
+        let ptr = ip.as_ptr_mut().cast::<u8>();
         unsafe {
-            core::ptr::write(ip.as_ptr_mut(), 192);
-            core::ptr::write(ip.as_ptr_mut().add(1), 168);
-            core::ptr::write(ip.as_ptr_mut().add(2), 42);
-            core::ptr::write(ip.as_ptr_mut().add(3), 73);
+            core::ptr::write(ptr, 192);
+            core::ptr::write(ptr.add(1), 168);
+            core::ptr::write(ptr.add(2), 42);
+            core::ptr::write(ptr.add(3), 73);
         }
         unsafe { assert_eq!(ip.v4.octets(), [192, 168, 42, 73]) }
     }