1 files changed, 36 insertions, 14 deletions
diff --git a/src/sniffing/headers.rs b/src/sniffing/headers.rs
index 0a1a742..c90a849 100644
--- a/src/sniffing/headers.rs
+++ b/src/sniffing/headers.rs
@@ -4,9 +4,10 @@ use tun::Error;
 use crate::config::Config;
 
 #[derive(Debug)]
-enum Protocol {
+pub enum Protocol {
     TCP,
-    UDP
+    UDP,
+    Unsupported(u8)
 }
 type Ipv4 = [u8; 4];
 type Ipv6 = [u8; 16];
@@ -30,36 +31,57 @@ pub enum PacketInfo {
         protocol: Protocol
     }
 }
+
+impl PacketInfo {
+    pub fn protocol(&self) -> &Protocol {
+        match self {
+            PacketInfo::V4 { protocol, .. } => protocol,
+            PacketInfo::V6 { protocol, .. } => protocol,
+        }
+    }
+}
 pub fn sniff_raw_packets(packet: &[u8]) -> Result<PacketInfo, Box<dyn std::error::Error + Send + Sync + 'static>> {
-    println!("something");
     let ver = packet[0] >> 4;
     dbg!(ver);
     match ver {
         4 => {
-            Ok(PacketInfo::V4{
-                src_ip: packet[12..16].try_into()?,
+            let v4 = PacketInfo::V4{
+                src_ip: <[u8; 4]>::try_from(&packet[12..16])?,
                 src_port: u16::from_be_bytes([packet[20], packet[21]]),
-                dst_ip: packet[16..20].try_into()?,
+                dst_ip: <[u8; 4]>::try_from(&packet[16..20])?,
                 dst_port: u16::from_be_bytes([packet[22], packet[23]]),
                 protocol: match packet[9] {
                     6 => Protocol::TCP,
-                    4 => Protocol::UDP,
-                    p => return Err(format!("unsuppiorted protocol: {p}").into())
+                    17 => Protocol::UDP,
+                    p => Protocol::Unsupported(p)
                 }
-            })
+            };
+        if !matches!(v4.protocol(), Protocol::Unsupported(_)) {
+            println!("{v4:?}");
+        } else {
+            println!("oppsie unsupported");
+            }
+            Ok(v4)
         },
         6 => {
-            Ok(PacketInfo::V6{
-                src_ip: packet[8..24].try_into()?,
+    println!("im in 6!");
+            let v6 = PacketInfo::V6{
+                src_ip: <[u8; 16]>::try_from(&packet[8..24])?,
                 src_port: u16::from_be_bytes([packet[40], packet[41]]),
-                dst_ip: packet[24..40].try_into()?,
+                dst_ip: <[u8; 16]>::try_from(&packet[24..40])?,
                 dst_port: u16::from_be_bytes([packet[42], packet[43]]),
                 protocol: match packet[6] {
                     6 => Protocol::TCP,
                     4 => Protocol::UDP,
-                    p => return Err(format!("unsuppiorted protocol: {p}").into())
+                    p => Protocol::Unsupported(p)
                 }
-            })
+            };
+        if !matches!(v6.protocol(), Protocol::Unsupported(_)) {
+            println!("{v6:?}");
+        } else {
+            println!("oppsie unsupported");
+            }
+            Ok(v6)
         },
         ver => {
             Err(format!("unsuppiorted ver: {ver}").into())