40 lines
1.1 KiB
Rust
40 lines
1.1 KiB
Rust
use std::net::{IpAddr, SocketAddr};
|
|
|
|
#[derive(Clone, Copy)]
|
|
pub enum SocketAddrSpace {
|
|
Unspecified,
|
|
Global,
|
|
}
|
|
|
|
impl SocketAddrSpace {
|
|
pub fn new(allow_private_addr: bool) -> Self {
|
|
if allow_private_addr {
|
|
SocketAddrSpace::Unspecified
|
|
} else {
|
|
SocketAddrSpace::Global
|
|
}
|
|
}
|
|
|
|
/// Returns true if the IP address is valid.
|
|
#[must_use]
|
|
pub fn check(&self, addr: &SocketAddr) -> bool {
|
|
if matches!(self, SocketAddrSpace::Unspecified) {
|
|
return true;
|
|
}
|
|
// TODO: remove these once IpAddr::is_global is stable.
|
|
match addr.ip() {
|
|
IpAddr::V4(addr) => {
|
|
// TODO: Consider excluding:
|
|
// addr.is_link_local() || addr.is_broadcast()
|
|
// || addr.is_documentation() || addr.is_unspecified()
|
|
!(addr.is_private() || addr.is_loopback())
|
|
}
|
|
IpAddr::V6(addr) => {
|
|
// TODO: Consider excluding:
|
|
// addr.is_unspecified(),
|
|
!addr.is_loopback()
|
|
}
|
|
}
|
|
}
|
|
}
|