uefi

Author: phip1611

uefi-test-runner/src/proto/network/pxe.rs

@@ -76,16 +76,11 @@ pub fn test() {
 
         let mut src_ip = server_ip;
         let mut src_port = EXAMPLE_SERVICE_PORT;
-        
-        // TODO add code to put this behind a type
-        let mut dest_ip = if base_code.mode().using_ipv6() {
-            IpAddr::V6(Ipv6Addr::from(base_code.mode().station_ip().octets()))
-        } else {
-            let octets: [u8; 4] = base_code.mode().station_ip().octets()[0..4]
-                .try_into()
-                .unwrap();
-            IpAddr::V4(Ipv4Addr::from(octets))
-        };
+
+        let mut dest_ip = base_code
+            .mode()
+            .station_ip()
+            .to_ip_addr(base_code.mode().using_ipv6());
         let mut dest_port = write_src_port;
         let mut header = [0; 1];
         let mut received = [0; 4];