zedddie/nsc
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

somewhat working network header parser

Browse source
This commit is contained in:
zedddie 2026-03-15 19:30:19 +01:00 • committed by tuturuu
parent 891a76eaf5
commit d48a7bebfb
No known key found for this signature in database
GPG key ID: B352C3C2894405A7
3 changed files with 44 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

50
src/sniffing/headers.rs
View file

@ -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())