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

impl Display trait for PacketInfo

Browse source
This commit is contained in:
zedddie 2026-03-23 19:11:40 +01:00 • committed by tuturuu
parent 1a5b7da6ae
commit d0513bec6a
No known key found for this signature in database
GPG key ID: B352C3C2894405A7
1 changed files with 38 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

57
src/sniffing/headers.rs
View file

@ -12,11 +12,11 @@ pub enum Protocol {
} }
type SourceV4Ip = Ipv4; type SourceV4Ip = Ipv4;
type SourceV6Ip = Ipv6; type SourceV6Ip = Ipv6;
#[derive(PartialEq, Debug)]
pub enum IpVersion { pub enum IpVersion {
V4, V4,
V6 V6
} }
// type IpVersion = String;
type Ipv4 = [u8; 4]; type Ipv4 = [u8; 4];
type Ipv6 = [u8; 16]; type Ipv6 = [u8; 16];
type Port = u16; type Port = u16;
@ -44,12 +44,16 @@ pub enum PacketInfo {
impl fmt::Display for PacketInfo { impl fmt::Display for PacketInfo {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
if self.version() == "Ipv4" { if self.version() == &IpVersion::V4 {
let src_ip = self.src_ipv4_ip(); let src_ip = self.src_ipv4_ip().unwrap();
let dst_ip = self.dst_ipv4_ip(); let dst_ip = self.dst_ipv4_ip().unwrap();
write!(f, "{} {}.{}.{}.{}:{} -> {}.{}.{}.{}:{} PROTO {} DNS? {}", self.version(), src_ip[0], src_ip[1], src_ip[2], src_ip[3], self.src_port, dst_ip[0], dst_ip[1], dst_ip[2], dst_ip[3], self.dst_port(), self.protocol(), self.dns()) write!(f, "{:?} {}.{}.{}.{}:{} -> {}.{}.{}.{}:{} {:?} is dns? {:?}", self.version(), src_ip[0], src_ip[1], src_ip[2], src_ip[3], self.src_port(), dst_ip[0], dst_ip[1], dst_ip[2], dst_ip[3], self.dst_port(), self.protocol(), self.dns())
} else {
let src_ip = self.src_ipv6_ip().unwrap();
let dst_ip = self.dst_ipv6_ip().unwrap();
// y:y:y:y:y:y:y:y = 8 hexademical
write!(f, "{:?} {}:{}:{}:{}:{}:{}:{}:{} port:{} -> {}:{}:{}:{}:{}:{}:{}:{} port:{} {:?} is dns? {:?}", self.version(), src_ip[0], src_ip[1], src_ip[2], src_ip[3], src_ip[4], src_ip[5], src_ip[6], src_ip[7], self.src_port(), dst_ip[0], dst_ip[1], dst_ip[2], dst_ip[3], dst_ip[4], dst_ip[5], dst_ip[6], dst_ip[7], self.dst_port(), self.protocol(), self.dns())
} }
// write!(f, "{} {}:{} -> {}:{} PROTO {} DNS? {}", self.version(), self.)
} }
} }
@ -60,26 +64,41 @@ impl PacketInfo {
PacketInfo::V6 { dns, ..} => dns, PacketInfo::V6 { dns, ..} => dns,
} }
} }
pub fn dst_ipv4_ip(&self) -> &SourceV4Ip { pub fn src_ipv6_ip(&self) -> Option<&SourceV6Ip> {
match self { match self {
PacketInfo::V4 { dst_ip, .. } => dst_ip, PacketInfo::V6 { src_ip, .. } => Some(src_ip),
_ => &[0x0, 0x0, 0x0, 0x0].try_into().expect("this never should fail or even be called in the first place.") _ => None
} }
} }
pub fn src_ipv6_ip(&self) -> &SourceV6Ip { pub fn dst_ipv6_ip(&self) -> Option<&SourceV6Ip> {
match self { match self {
PacketInfo::V6 { src_ip, .. } => src_ip, PacketInfo::V6 { dst_ip, .. } => Some(dst_ip),
_ => &[0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0].try_into().expect("this never should fail or even be called in the first place.") _ => None
} }
} }
pub fn src_ipv4_ip(&self) -> &SourceV4Ip { pub fn src_ipv4_ip(&self) -> Option<&SourceV4Ip> {
match self { match self {
PacketInfo::V4 { src_ip, .. } => src_ip, PacketInfo::V4 { src_ip, .. } => Some(src_ip),
_ => &[0x0, 0x0, 0x0, 0x0].try_into().expect("this never should fail or even be called in the first place.") _ => None,
} }
} }
pub fn src_ipv6_ip(&self) -> &SourceV6Ip { pub fn dst_ipv4_ip(&self) -> Option<&SourceV4Ip> {
PacketInfo::V6.src_ip match self {
PacketInfo::V4 { dst_ip, .. } => Some(dst_ip),
_ => None
}
}
pub fn src_port(&self) -> &Port {
match self {
PacketInfo::V4 { src_port, .. } => src_port,
PacketInfo::V6 { src_port, .. } => src_port
}
}
pub fn dst_port(&self) -> &Port {
match self {
PacketInfo::V4 { dst_port, .. } => dst_port,
PacketInfo::V6 { dst_port, .. } => dst_port
}
} }
pub fn version(&self) -> &IpVersion { pub fn version(&self) -> &IpVersion {
match self { match self {
@ -123,7 +142,7 @@ pub fn sniff_raw_packets(packet: &Packet) -> SniffedPacket {
dns dns
}; };
if !matches!(v4.protocol(), Protocol::Unsupported(_)) { if !matches!(v4.protocol(), Protocol::Unsupported(_)) {
println!("{v4:?}"); println!("{v4}");
} else { } else {
// TODO: make --debug option which will include this diagnostic, for general use this // TODO: make --debug option which will include this diagnostic, for general use this
// should be off // should be off
@ -148,7 +167,7 @@ pub fn sniff_raw_packets(packet: &Packet) -> SniffedPacket {
dns dns
}; };
if !matches!(v6.protocol(), Protocol::Unsupported(_)) { if !matches!(v6.protocol(), Protocol::Unsupported(_)) {
println!("{v6:?}"); println!("{v6}");
} else { } else {
// TODO: make --debug option which will include this diagnostic, for general use this // TODO: make --debug option which will include this diagnostic, for general use this
// should be off // should be off