a bunch of really (not)good tests

1 files changed, 235 insertions, 0 deletions

diff --git a/tests/headers.rs b/tests/headers.rs
new file mode 100644
index 0000000..d7daef5
--- /dev/null
+++ b/tests/headers.rs
@@ -0,0 +1,235 @@
+use nsc::sniffing::*;
+
+use crate::headers::Protocol;
+use nsc::sniffing::headers::sniff_raw_packets;
+use nsc::sniffing::headers::PacketInfo;
+
+#[test]
+fn generic_typeck() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
+    let test_suite: [&[u8]; 10] = [
+        // 1. IPv4 TCP — 192.168.1.100:4832 → 93.184.216.34:443
+        &[
+            0x45, 0x00, 0x00, 0x28, 0x1A, 0x2B, 0x40, 0x00, 0x40, 0x06, 0x00, 0x00, 192, 168, 1,
+            100, // src ip
+            93, 184, 216, 34, // dst ip
+            0x12, 0xE0, // src port 4832
+            0x01, 0xBB, // dst port 443
+            0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02, 0xFF, 0xFF, 0x00, 0x00,
+            0x00, 0x00,
+        ],
+        // 2. IPv4 UDP — 10.0.0.9:5353 → 224.0.0.251:5353 (mDNS)
+        &[
+            0x45, 0x00, 0x00, 0x30, 0x8A, 0xF0, 0x40, 0x00, 0x01, 0x11, 0x00, 0x00, 10, 0, 0, 9,
+            224, 0, 0, 251, 0x14, 0xE9, // src port 5353
+            0x14, 0xE9, // dst port 5353
+            0x00, 0x1C, 0x00, 0x00,
+        ],
+        // 3. IPv4 UDP — 10.0.0.9:1024 → 8.8.8.8:53 (DNS)
+        &[
+            0x45, 0x00, 0x00, 0x3C, 0x00, 0x01, 0x00, 0x00, 0x40, 0x11, 0x00, 0x00, 10, 0, 0, 9, 8,
+            8, 8, 8, 0x04, 0x00, // src port 1024
+            0x00, 0x35, // dst port 53
+            0x00, 0x28, 0x00, 0x00,
+        ],
+        // 4. IPv4 TCP — 10.0.0.5:54321 → 10.0.0.1:80 (HTTP)
+        &[
+            0x45, 0x00, 0x00, 0x28, 0x00, 0x02, 0x40, 0x00, 0x40, 0x06, 0x00, 0x00, 10, 0, 0, 5,
+            10, 0, 0, 1, 0xD4, 0x31, // src port 54321
+            0x00, 0x50, // dst port 80
+            0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02, 0xFF, 0xFF, 0x00, 0x00,
+            0x00, 0x00,
+        ],
+        // 5. IPv4 TCP — 172.16.0.1:65535 → 172.16.0.2:8080
+        &[
+            0x45, 0x00, 0x00, 0x28, 0x00, 0x03, 0x40, 0x00, 0x40, 0x06, 0x00, 0x00, 172, 16, 0, 1,
+            172, 16, 0, 2, 0xFF, 0xFF, // src port 65535
+            0x1F, 0x90, // dst port 8080
+            0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02, 0xFF, 0xFF, 0x00, 0x00,
+            0x00, 0x00,
+        ],
+        // 6. IPv4 with IHL=6 (24-byte header, has IP options)
+        //    10.0.0.1:9090 → 10.0.0.2:22 TCP
+        &[
+            0x46, 0x00, 0x00, 0x2C, 0x00, 0x04, 0x40, 0x00, 0x40, 0x06, 0x00, 0x00, 10, 0, 0, 1,
+            10, 0, 0, 2, 0x01, 0x02, 0x03, 0x04, // 4 bytes of IP options (padding)
+            0x23, 0x82, // src port 9090 (at byte 24, not 20!)
+            0x00, 0x16, // dst port 22
+            0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02, 0xFF, 0xFF, 0x00, 0x00,
+            0x00, 0x00,
+        ],
+        // 7. IPv4 ICMP — unsupported protocol (proto=1)
+        //    10.0.0.9 → 10.0.0.1
+        &[
+            0x45, 0x00, 0x00, 0x1C, 0x00, 0x05, 0x00, 0x00, 0x40, 0x01, 0x00,
+            0x00, // protocol = 1 (ICMP)
+            10, 0, 0, 9, 10, 0, 0, 1, 0x08, 0x00, 0x00,
+            0x00, // ICMP echo request (will be read as "ports")
+            0x00, 0x01, 0x00, 0x01,
+        ],
+        // 8. IPv6 TCP — [::1]:4000 → [2606:4700::1]:443
+        &[
+            0x60, 0x00, 0x00, 0x00, // version=6, traffic class, flow label
+            0x00, 0x14, // payload length = 20
+            0x06, // next header = 6 (TCP)
+            0x40, // hop limit = 64
+            // src ip: ::1 (bytes 8-23)
+            0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+            0x00, 0x01, // dst ip: 2606:4700::1 (bytes 24-39)
+            0x26, 0x06, 0x47, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+            0x00, 0x01, // TCP header (bytes 40+)
+            0x0F, 0xA0, // src port 4000
+            0x01, 0xBB, // dst port 443
+            0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x50, 0x02, 0xFF, 0xFF, 0x00, 0x00,
+            0x00, 0x00,
+        ],
+        // 9. IPv6 UDP — [fd00::9]:1234 → [fd00::1]:53 (DNS)
+        &[
+            0x60, 0x00, 0x00, 0x00, 0x00, 0x08, // payload length = 8
+            0x11, // next header = 17 (UDP)
+            0x40, // src ip: fd00::9
+            0xFD, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+            0x00, 0x09, // dst ip: fd00::1
+            0xFD, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+            0x00, 0x01, // UDP header
+            0x04, 0xD2, // src port 1234
+            0x00, 0x35, // dst port 53
+            0x00, 0x08, 0x00, 0x00,
+        ],
+        // 10. IPv6 ICMPv6 — unsupported (next header = 58)
+        //     [fe80::1] → [ff02::1]
+        &[
+            0x60, 0x00, 0x00, 0x00, 0x00, 0x08, 0x3A, // next header = 58 (ICMPv6)
+            0xFF, // src ip: fe80::1
+            0xFE, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+            0x00, 0x01, // dst ip: ff02::1
+            0xFF, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+            0x00, 0x01, // ICMPv6 body (will be read as "ports" if you parse blindly)
+            0x80, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x01,
+        ],
+    ];
+    // for test in test_suite {
+    //     let abc = sniff_raw_packets(test);
+    //     eprintln!("{:?}", abc);
+    // }
+
+    // [0] IPv4 TCP 192.168.1.100:4832 → 93.184.216.34:443
+    assert_eq!(
+        sniff_raw_packets(test_suite[0])?,
+        PacketInfo::V4 {
+            src_ip: [192, 168, 1, 100],
+            src_port: 4832,
+            dst_ip: [93, 184, 216, 34],
+            dst_port: 443,
+            protocol: Protocol::TCP
+        }
+    );
+
+    // [1] IPv4 UDP 10.0.0.9:5353 → 224.0.0.251:5353 (mDNS)
+    assert_eq!(
+        sniff_raw_packets(test_suite[1])?,
+        PacketInfo::V4 {
+            src_ip: [10, 0, 0, 9],
+            src_port: 5353,
+            dst_ip: [224, 0, 0, 251],
+            dst_port: 5353,
+            protocol: Protocol::UDP
+        }
+    );
+
+    // [2] IPv4 UDP 10.0.0.9:1024 → 8.8.8.8:53 (DNS)
+    assert_eq!(
+        sniff_raw_packets(test_suite[2])?,
+        PacketInfo::V4 {
+            src_ip: [10, 0, 0, 9],
+            src_port: 1024,
+            dst_ip: [8, 8, 8, 8],
+            dst_port: 53,
+            protocol: Protocol::UDP
+        }
+    );
+
+    // [3] IPv4 TCP 10.0.0.5:54321 → 10.0.0.1:80 (HTTP)
+    assert_eq!(
+        sniff_raw_packets(test_suite[3])?,
+        PacketInfo::V4 {
+            src_ip: [10, 0, 0, 5],
+            src_port: 54321,
+            dst_ip: [10, 0, 0, 1],
+            dst_port: 80,
+            protocol: Protocol::TCP
+        }
+    );
+
+    // [4] IPv4 TCP 172.16.0.1:65535 → 172.16.0.2:8080
+    assert_eq!(
+        sniff_raw_packets(test_suite[4])?,
+        PacketInfo::V4 {
+            src_ip: [172, 16, 0, 1],
+            src_port: 65535,
+            dst_ip: [172, 16, 0, 2],
+            dst_port: 8080,
+            protocol: Protocol::TCP
+        }
+    );
+
+    // [5] IPv4 TCP IHL=6 10.0.0.1:9090 → 10.0.0.2:22 (requires IHL-based offset)
+    assert_eq!(
+        sniff_raw_packets(test_suite[5])?,
+        PacketInfo::V4 {
+            src_ip: [10, 0, 0, 1],
+            src_port: 9090,
+            dst_ip: [10, 0, 0, 2],
+            dst_port: 22,
+            protocol: Protocol::TCP
+        }
+    );
+
+    // [6] IPv4 ICMP (unsupported, "ports" are just ICMP body bytes)
+    assert_eq!(
+        sniff_raw_packets(test_suite[6])?,
+        PacketInfo::V4 {
+            src_ip: [10, 0, 0, 9],
+            src_port: 2048,
+            dst_ip: [10, 0, 0, 1],
+            dst_port: 0,
+            protocol: Protocol::Unsupported(1)
+        }
+    );
+
+    // [7] IPv6 TCP [::1]:4000 → [2606:4700::1]:443
+    assert_eq!(
+        sniff_raw_packets(test_suite[7])?,
+        PacketInfo::V6 {
+            src_ip: [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
+            src_port: 4000,
+            dst_ip: [0x26, 0x06, 0x47, 0x00, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
+            dst_port: 443,
+            protocol: Protocol::TCP
+        }
+    );
+
+    // [8] IPv6 UDP [fd00::9]:1234 → [fd00::1]:53 (requires fixing 17 => UDP)
+    assert_eq!(
+        sniff_raw_packets(test_suite[8])?,
+        PacketInfo::V6 {
+            src_ip: [0xFD, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 9],
+            src_port: 1234,
+            dst_ip: [0xFD, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
+            dst_port: 53,
+            protocol: Protocol::UDP
+        }
+    );
+
+    // [9] IPv6 ICMPv6 (unsupported, "ports" are ICMPv6 body bytes)
+    assert_eq!(
+        sniff_raw_packets(test_suite[9])?,
+        PacketInfo::V6 {
+            src_ip: [0xFE, 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
+            src_port: 32768,
+            dst_ip: [0xFF, 0x02, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1],
+            dst_port: 0,
+            protocol: Protocol::Unsupported(58)
+        }
+    );
+    Ok(())
+}