diff options
| author | zedddie <rust@zedddie.rs> | 2026-03-17 22:27:24 +0100 |
|---|---|---|
| committer | tuturuu <zedddiezxc@gmail.com> | 2026-03-17 22:27:24 +0100 |
| commit | 50524cb5aee1f1f4464f60467122a44befc11203 (patch) | |
| tree | a01bc74d5d0c2dbb79002ef697f783d0d6bd39a8 /src/sniffing/headers.rs | |
| parent | da8e70f2e3c841796c122ca90617d74cb044b763 (diff) | |
use ihl to derive port bytes
Diffstat (limited to 'src/sniffing/headers.rs')
| -rw-r--r-- | src/sniffing/headers.rs | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/sniffing/headers.rs b/src/sniffing/headers.rs index a5b0480..2472981 100644 --- a/src/sniffing/headers.rs +++ b/src/sniffing/headers.rs @@ -52,13 +52,15 @@ pub fn sniff_raw_packets(packet: &Packet) -> SniffedPacket { let ver = packet[0] >> 4; match ver { 4 => { - let dst_port = Port::from_be_bytes([packet[22], packet[23]]); + // Internet Header Length (IHL). + let ihl = (packet[0] & 0x0F) as usize * 4; + let dst_port = Port::from_be_bytes([packet[ihl+2], packet[ihl+3]]); let dns; if dst_port == 53 { dns = true; } else { dns = false; }; // FIXME: hardcoded IPv4 port offset let v4 = PacketInfo::V4{ src_ip: <Ipv4>::try_from(&packet[12..16])?, - src_port: Port::from_be_bytes([packet[20], packet[21]]), + src_port: Port::from_be_bytes([packet[ihl], packet[ihl+1]]), dst_ip: <Ipv4>::try_from(&packet[16..20])?, dst_port, protocol: match packet[9] { |