2021-07-23 08:25:03 -07:00
|
|
|
use std::net::{IpAddr, SocketAddr};
|
2021-07-16 10:41:01 -07:00
|
|
|
|
2022-05-22 18:00:42 -07:00
|
|
|
#[derive(Clone, Copy, PartialEq, Eq)]
|
2021-07-23 08:25:03 -07:00
|
|
|
pub enum SocketAddrSpace {
|
|
|
|
Unspecified,
|
|
|
|
Global,
|
2021-07-16 10:41:01 -07:00
|
|
|
}
|
|
|
|
|
2021-07-23 08:25:03 -07:00
|
|
|
impl SocketAddrSpace {
|
|
|
|
pub fn new(allow_private_addr: bool) -> Self {
|
|
|
|
if allow_private_addr {
|
|
|
|
SocketAddrSpace::Unspecified
|
|
|
|
} else {
|
|
|
|
SocketAddrSpace::Global
|
|
|
|
}
|
|
|
|
}
|
2021-07-16 10:41:01 -07:00
|
|
|
|
2021-07-23 08:25:03 -07:00
|
|
|
/// Returns true if the IP address is valid.
|
|
|
|
pub fn check(&self, addr: &SocketAddr) -> bool {
|
|
|
|
if self == &SocketAddrSpace::Unspecified {
|
|
|
|
return true;
|
2021-07-16 10:41:01 -07:00
|
|
|
}
|
2021-07-23 08:25:03 -07:00
|
|
|
// TODO: remove these once IpAddr::is_global is stable.
|
|
|
|
match addr.ip() {
|
|
|
|
IpAddr::V4(addr) => {
|
|
|
|
// TODO: Consider excluding:
|
|
|
|
// addr.is_loopback() || addr.is_link_local()
|
|
|
|
// || addr.is_broadcast() || addr.is_documentation()
|
|
|
|
// || addr.is_unspecified()
|
|
|
|
!addr.is_private()
|
|
|
|
}
|
|
|
|
IpAddr::V6(_) => {
|
|
|
|
// TODO: Consider excluding:
|
|
|
|
// addr.is_loopback() || addr.is_unspecified(),
|
|
|
|
true
|
|
|
|
}
|
2021-07-16 10:41:01 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|