uefi-raw

Author: phip1611

uefi-raw/src/lib.rs

@@ -150,6 +150,18 @@ impl IpAddress {
     pub const fn octets(&self) -> [u8; 16] {
         unsafe { self.v6.octets() }
     }
+
+    /// Returns a raw pointer to the IP address.
+    #[must_use]
+    pub const fn as_ptr(&self) -> *const u8 {
+        core::ptr::addr_of!(*self).cast()
+    }
+
+    /// 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()
+    }
 }
 
 impl Debug for IpAddress {
@@ -246,16 +258,29 @@ mod tests {
         assert_eq!(size_of::<Ipv6Addr>(), 16);
         assert_eq!(align_of::<Ipv6Addr>(), 1);
     }
+
+    #[test]
+    fn ip_ptr() {
+        let mut ip = IpAddress::new_v4([0; 4]);
+        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);
+        }
+        unsafe { assert_eq!(ip.v4.octets(), [192, 168, 42, 73]) }
+    }
+
     /// Test conversion from `core::net::IpAddr` to `IpvAddress`.
     ///
     /// Note that conversion in the other direction is not possible.
     #[test]
     fn test_ip_addr_conversion() {
-        let core_addr = IpAddr::V4(core::net::Ipv4Addr::from(TEST_IPV4));
+        let core_addr = IpAddr::V4(Ipv4Addr::from(TEST_IPV4));
         let uefi_addr = IpAddress::from(core_addr);
         assert_eq!(unsafe { uefi_addr.v4.octets() }, TEST_IPV4);
 
-        let core_addr = IpAddr::V6(core::net::Ipv6Addr::from(TEST_IPV6));
+        let core_addr = IpAddr::V6(Ipv6Addr::from(TEST_IPV6));
         let uefi_addr = IpAddress::from(core_addr);
         assert_eq!(unsafe { uefi_addr.v6.octets() }, TEST_IPV6);
     }