some unwrap clarifications

mrak · Aug 29, 2024 · 9c25eb8 · 9c25eb8
1 parent 2c09d7e
commit 9c25eb8
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -140,7 +140,7 @@ fn capture_packets(settings: &Settings, sender: Sender<(u32, Vec<u8>)>) {
                     match rx.next() {
                         Ok(packet) => child_snd
                             .send((interface.index, packet.to_owned()))
-                            .unwrap(),
+                            .expect("packet should be able to be sent on channel"),
                         Err(e) => panic!("nx: Unable to receive packet: {}", e),
                     };
                 }
@@ -156,16 +156,19 @@ fn print_packets(settings: &Settings, receiver: Receiver<(u32, Vec<u8>)>) {
     loop {
         match receiver.recv() {
             Ok((index, packet)) => {
-                let ethernet_packet = EthernetPacket::new(packet.as_slice()).unwrap();
-                match ethernet_packet.get_ethertype() {
-                    EtherTypes::Ipv4 => {
-                        process_ipv4(settings, &inames[&index][..], &ethernet_packet)
-                    }
-                    EtherTypes::Ipv6 => {
-                        process_ipv6(settings, &inames[&index][..], &ethernet_packet)
+                if let Some(ethernet_packet) = EthernetPacket::new(packet.as_slice()) {
+                    match ethernet_packet.get_ethertype() {
+                        EtherTypes::Ipv4 => {
+                            process_ipv4(settings, &inames[&index][..], &ethernet_packet)
+                        }
+                        EtherTypes::Ipv6 => {
+                            process_ipv6(settings, &inames[&index][..], &ethernet_packet)
+                        }
+                        EtherTypes::Arp => {
+                            process_arp(settings, &inames[&index][..], &ethernet_packet)
+                        }
+                        _ => {}
                     }
-                    EtherTypes::Arp => process_arp(settings, &inames[&index][..], &ethernet_packet),
-                    _ => {}
                 }
             }
             Err(_) => panic!("All interfaces closed"),