pub mod icmp;
pub mod tcp;
pub mod udp;
use std::net::{IpAddr, SocketAddr};
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum SocketFamily {
IPV4,
IPV6,
}
impl SocketFamily {
pub fn from_ip(ip: &IpAddr) -> Self {
match ip {
IpAddr::V4(_) => SocketFamily::IPV4,
IpAddr::V6(_) => SocketFamily::IPV6,
}
}
pub fn from_socket_addr(addr: &SocketAddr) -> Self {
match addr {
SocketAddr::V4(_) => SocketFamily::IPV4,
SocketAddr::V6(_) => SocketFamily::IPV6,
}
}
pub fn is_v4(&self) -> bool {
matches!(self, SocketFamily::IPV4)
}
pub fn is_v6(&self) -> bool {
matches!(self, SocketFamily::IPV6)
}
pub(crate) fn to_domain(&self) -> socket2::Domain {
match self {
SocketFamily::IPV4 => socket2::Domain::IPV4,
SocketFamily::IPV6 => socket2::Domain::IPV6,
}
}
}