From 59a17c547563dab8a70830185ba763d8343ac109 Mon Sep 17 00:00:00 2001 From: zedddie Date: Sun, 15 Mar 2026 16:57:56 +0100 Subject: networking abstractions & early parsing logic --- src/sniffing/headers.rs | 62 +++++++++++++++++++++++++++++++++++++++++++++++-- 1 file changed, 60 insertions(+), 2 deletions(-) diff --git a/src/sniffing/headers.rs b/src/sniffing/headers.rs index 7ed1a88..d1d0a84 100644 --- a/src/sniffing/headers.rs +++ b/src/sniffing/headers.rs @@ -1,9 +1,67 @@ +use tun::Error; + // Here we will recieve bytes and try to get their destanation & apply Rules for them. use crate::config::Config; -struct PacketInfo; +enum Protocol { + TCP, + UDP +} +type Ipv4 = [u8; 4]; +type Ipv6 = [u8; 16]; +type Port = u16; +enum PacketInfo { + // + V4 { + src_ip: Ipv4, + src_port: Port, + dst_ip: Ipv4, + dst_port: Port, + protocol: Protocol + }, + // + V6 { + src_ip: Ipv6, + src_port: Port, + dst_ip: Ipv6, + dst_port: Port, + protocol: Protocol + } +} pub fn sniff_raw_packets(packet: &[u8]) -> Result> { - todo!() + let ver = packet[0] >> 4; + match ver { + 4 => { + PacketInfo::V4{ + src_ip: packet[12..16], + src_port: u16::from_be_bytes([packet[20], packet[21]]), + dst_ip: packet[16..20], + dst_port: u16::from_be_bytes([packet[22], packet[23]]), + protocol: match packet[9] { + 6 => Protocol::TCP, + 4 => Protocol::UDP, + _ => return Err(format!("unsuppiorted protocol: {p}").into()) + } + } + }, + 6 => { + PacketInfo::V6{ + src_ip: packet[8..24], + src_port: u16::from_be_bytes([packet[40], packet[41]]), + dst_ip: packet[24..40], + dst_port: u16::from_be_bytes([packet[42], packet[43]]), + protocol: match packet[6] { + 6 => Protocol::TCP, + 4 => Protocol::UDP, + _ => return Err(format!("unsuppiorted protocol: {p}").into()) + } + } + }, + ver => { + panic!("unexpected packet ver: {ver}"); + Error + } + } } pub fn apply_rules(config: Config, pinfo: PacketInfo) { -- cgit v1.2.3