| use crate::cmp::Ordering;use crate::fmt::{self, Write as FmtWrite}; |
| use crate::hash; |
| use crate::io::Write as IoWrite; |
| use crate::mem::transmute; |
| use crate::sys::net::netc as c; |
| use crate::sys_common::{AsInner, FromInner, IntoInner}; |
| #[derive(Copy, Clone, Eq, PartialEq, Hash, PartialOrd, Ord)] |
| pub enum IpAddr { V4(Ipv4Addr), V6(Ipv6Addr), } |
| #[derive(Copy)] |
| pub struct Ipv4Addr { |
| inner: c::in_addr, |
| } |
| #[derive(Copy)] |
| pub struct Ipv6Addr { |
| inner: c::in6_addr, |
| } |
| #[derive(Copy, PartialEq, Eq, Clone, Hash, Debug)] |
| #[non_exhaustive] |
| pub enum Ipv6MulticastScope { |
| InterfaceLocal, |
| LinkLocal, |
| RealmLocal, |
| AdminLocal, |
| SiteLocal, |
| OrganizationLocal, |
| Global, |
| } |
| impl IpAddr { |
| pub const fn is_unspecified(&self) -> bool { |
| match self { |
| IpAddr::V4(ip) => ip.is_unspecified(), |
| IpAddr::V6(ip) => ip.is_unspecified(), |
| } |
| } |
| pub const fn is_loopback(&self) -> bool { |
| match self { |
| IpAddr::V4(ip) => ip.is_loopback(), |
| IpAddr::V6(ip) => ip.is_loopback(), |
| } |
| } |
| pub const fn is_global(&self) -> bool { |
| match self { |
| IpAddr::V4(ip) => ip.is_global(), |
| IpAddr::V6(ip) => ip.is_global(), |
| } |
| } |
| pub const fn is_multicast(&self) -> bool { |
| match self { |
| IpAddr::V4(ip) => ip.is_multicast(), |
| IpAddr::V6(ip) => ip.is_multicast(), |
| } |
| } |
| pub const fn is_documentation(&self) -> bool { |
| match self { |
| IpAddr::V4(ip) => ip.is_documentation(), |
| IpAddr::V6(ip) => ip.is_documentation(), |
| } |
| } |
| pub const fn is_benchmarking(&self) -> bool { |
| match self { |
| IpAddr::V4(ip) => ip.is_benchmarking(), |
| IpAddr::V6(ip) => ip.is_benchmarking(), |
| } |
| } |
| pub const fn is_ipv4(&self) -> bool { matches!(self, IpAddr :: V4(_)) } |
| pub const fn is_ipv6(&self) -> bool { matches!(self, IpAddr :: V6(_)) } |
| pub const fn to_canonical(&self) -> IpAddr { |
| match self { |
| &v4 @ IpAddr::V4(_) => v4, |
| IpAddr::V6(v6) => v6.to_canonical(), |
| } |
| } |
| } |
| impl Ipv4Addr { |
| pub const fn new(a: u8, b: u8, c: u8, d: u8) -> Ipv4Addr { |
| Ipv4Addr { |
| inner: c::in_addr { s_addr: u32::from_ne_bytes([a, b, c, d]) }, |
| } |
| } |
| pub const LOCALHOST: Self = Ipv4Addr::new(127, 0, 0, 1); |
| #[doc(alias = "INADDR_ANY")] |
| pub const UNSPECIFIED: Self = Ipv4Addr::new(0, 0, 0, 0); |
| pub const BROADCAST: Self = Ipv4Addr::new(255, 255, 255, 255); |
| pub const fn octets(&self) -> [u8; 4] { self.inner.s_addr.to_ne_bytes() } |
| pub const fn is_unspecified(&self) -> bool { self.inner.s_addr == 0 } |
| pub const fn is_loopback(&self) -> bool { self.octets()[0] == 127 } |
| pub const fn is_private(&self) -> bool { |
| match self.octets() { |
| [10, ..] => true, |
| [172, b, ..] if b >= 16 && b <= 31 => true, |
| [192, 168, ..] => true, |
| _ => false, |
| } |
| } |
| pub const fn is_link_local(&self) -> bool { |
| matches!(self.octets(), [169, 254, ..]) |
| } |
| pub const fn is_global(&self) -> bool { |
| if u32::from_be_bytes(self.octets()) == 0xc0000009 || |
| u32::from_be_bytes(self.octets()) == 0xc000000a { |
| return true; |
| } |
| !self.is_private() && !self.is_loopback() && !self.is_link_local() && |
| !self.is_broadcast() && !self.is_documentation() && |
| !self.is_shared() && |
| !(self.octets()[0] == 192 && self.octets()[1] == 0 && |
| self.octets()[2] == 0) && !self.is_reserved() && |
| !self.is_benchmarking() && self.octets()[0] != 0 |
| } |
| pub const fn is_shared(&self) -> bool { |
| self.octets()[0] == 100 && |
| (self.octets()[1] & 0b1100_0000 == 0b0100_0000) |
| } |
| pub const fn is_benchmarking(&self) -> bool { |
| self.octets()[0] == 198 && (self.octets()[1] & 0xfe) == 18 |
| } |
| pub const fn is_reserved(&self) -> bool { |
| self.octets()[0] & 240 == 240 && !self.is_broadcast() |
| } |
| pub const fn is_multicast(&self) -> bool { |
| self.octets()[0] >= 224 && self.octets()[0] <= 239 |
| } |
| pub const fn is_broadcast(&self) -> bool { |
| u32::from_be_bytes(self.octets()) == |
| u32::from_be_bytes(Self::BROADCAST.octets()) |
| } |
| pub const fn is_documentation(&self) -> bool { |
| matches!(self.octets(), [192, 0, 2, _] | [198, 51, 100, _] | |
| [203, 0, 113, _]) |
| } |
| pub const fn to_ipv6_compatible(&self) -> Ipv6Addr { |
| let [a, b, c, d] = self.octets(); |
| Ipv6Addr { |
| inner: c::in6_addr { |
| s6_addr: [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, a, b, c, d], |
| }, |
| } |
| } |
| pub const fn to_ipv6_mapped(&self) -> Ipv6Addr { |
| let [a, b, c, d] = self.octets(); |
| Ipv6Addr { |
| inner: c::in6_addr { |
| s6_addr: [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0xFF, 0xFF, a, b, c, |
| d], |
| }, |
| } |
| } |
| } |
| impl fmt::Display for IpAddr { |
| fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result { |
| match self { |
| IpAddr::V4(ip) => ip.fmt(fmt), |
| IpAddr::V6(ip) => ip.fmt(fmt), |
| } |
| } |
| } |
| impl fmt::Debug for IpAddr { |
| fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result { |
| fmt::Display::fmt(self, fmt) |
| } |
| } |
| impl From<Ipv4Addr> for IpAddr { |
| fn from(ipv4: Ipv4Addr) -> IpAddr { IpAddr::V4(ipv4) } |
| } |
| impl From<Ipv6Addr> for IpAddr { |
| fn from(ipv6: Ipv6Addr) -> IpAddr { IpAddr::V6(ipv6) } |
| } |
| impl fmt::Display for Ipv4Addr { |
| fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result { |
| let octets = self.octets(); |
| if fmt.precision().is_none() && fmt.width().is_none() { |
| write!(fmt, "{}.{}.{}.{}", octets [0], octets [1], octets [2], |
| octets [3]) |
| } else { |
| const IPV4_BUF_LEN: usize = 15; |
| let mut buf = [0u8; IPV4_BUF_LEN]; |
| let mut buf_slice = &mut buf[..]; |
| write!(buf_slice, "{}.{}.{}.{}", octets [0], octets [1], octets |
| [2], octets [3]).unwrap(); |
| let len = IPV4_BUF_LEN - buf_slice.len(); |
| let buf = |
| unsafe { crate::str::from_utf8_unchecked(&buf[..len]) }; |
| fmt.pad(buf) |
| } |
| } |
| } |
| impl fmt::Debug for Ipv4Addr { |
| fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result { |
| fmt::Display::fmt(self, fmt) |
| } |
| } |
| impl Clone for Ipv4Addr { |
| fn clone(&self) -> Ipv4Addr { *self } |
| } |
| impl PartialEq for Ipv4Addr { |
| fn eq(&self, other: &Ipv4Addr) -> bool { |
| self.inner.s_addr == other.inner.s_addr |
| } |
| } |
| impl PartialEq<Ipv4Addr> for IpAddr { |
| fn eq(&self, other: &Ipv4Addr) -> bool { |
| match self { IpAddr::V4(v4) => v4 == other, IpAddr::V6(_) => false, } |
| } |
| } |
| impl PartialEq<IpAddr> for Ipv4Addr { |
| fn eq(&self, other: &IpAddr) -> bool { |
| match other { IpAddr::V4(v4) => self == v4, IpAddr::V6(_) => false, } |
| } |
| } |
| impl Eq for Ipv4Addr {} |
| impl hash::Hash for Ipv4Addr { |
| fn hash<H: hash::Hasher>(&self, s: &mut H) { |
| { self.inner.s_addr }.hash(s) |
| } |
| } |
| impl PartialOrd for Ipv4Addr { |
| fn partial_cmp(&self, other: &Ipv4Addr) -> Option<Ordering> { |
| Some(self.cmp(other)) |
| } |
| } |
| impl PartialOrd<Ipv4Addr> for IpAddr { |
| fn partial_cmp(&self, other: &Ipv4Addr) -> Option<Ordering> { |
| match self { |
| IpAddr::V4(v4) => v4.partial_cmp(other), |
| IpAddr::V6(_) => Some(Ordering::Greater), |
| } |
| } |
| } |
| impl PartialOrd<IpAddr> for Ipv4Addr { |
| fn partial_cmp(&self, other: &IpAddr) -> Option<Ordering> { |
| match other { |
| IpAddr::V4(v4) => self.partial_cmp(v4), |
| IpAddr::V6(_) => Some(Ordering::Less), |
| } |
| } |
| } |
| impl Ord for Ipv4Addr { |
| fn cmp(&self, other: &Ipv4Addr) -> Ordering { |
| u32::from_be(self.inner.s_addr).cmp(&u32::from_be(other.inner.s_addr)) |
| } |
| } |
| impl IntoInner<c::in_addr> for Ipv4Addr { |
| fn into_inner(self) -> c::in_addr { self.inner } |
| } |
| impl From<Ipv4Addr> for u32 { |
| fn from(ip: Ipv4Addr) -> u32 { |
| let ip = ip.octets(); |
| u32::from_be_bytes(ip) |
| } |
| } |
| impl From<u32> for Ipv4Addr { |
| fn from(ip: u32) -> Ipv4Addr { Ipv4Addr::from(ip.to_be_bytes()) } |
| } |
| impl From<[u8; 4]> for Ipv4Addr { |
| fn from(octets: [u8; 4]) -> Ipv4Addr { |
| Ipv4Addr::new(octets[0], octets[1], octets[2], octets[3]) |
| } |
| } |
| impl From<[u8; 4]> for IpAddr { |
| fn from(octets: [u8; 4]) -> IpAddr { IpAddr::V4(Ipv4Addr::from(octets)) } |
| } |
| impl Ipv6Addr { |
| pub const fn new(a: u16, b: u16, c: u16, d: u16, e: u16, f: u16, g: u16, |
| h: u16) -> Ipv6Addr { |
| let addr16 = |
| [a.to_be(), b.to_be(), c.to_be(), d.to_be(), e.to_be(), f.to_be(), |
| g.to_be(), h.to_be()]; |
| Ipv6Addr { |
| inner: c::in6_addr { |
| s6_addr: unsafe { transmute::<_, [u8; 16]>(addr16) }, |
| }, |
| } |
| } |
| pub const LOCALHOST: Self = Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 1); |
| pub const UNSPECIFIED: Self = Ipv6Addr::new(0, 0, 0, 0, 0, 0, 0, 0); |
| pub const fn segments(&self) -> [u16; 8] { |
| let [a, b, c, d, e, f, g, h] = |
| unsafe { transmute::<_, [u16; 8]>(self.inner.s6_addr) }; |
| [u16::from_be(a), u16::from_be(b), u16::from_be(c), u16::from_be(d), |
| u16::from_be(e), u16::from_be(f), u16::from_be(g), |
| u16::from_be(h)] |
| } |
| pub const fn is_unspecified(&self) -> bool { |
| u128::from_be_bytes(self.octets()) == |
| u128::from_be_bytes(Ipv6Addr::UNSPECIFIED.octets()) |
| } |
| pub const fn is_loopback(&self) -> bool { |
| u128::from_be_bytes(self.octets()) == |
| u128::from_be_bytes(Ipv6Addr::LOCALHOST.octets()) |
| } |
| pub const fn is_global(&self) -> bool { |
| match self.multicast_scope() { |
| Some(Ipv6MulticastScope::Global) => true, |
| None => self.is_unicast_global(), |
| _ => false, |
| } |
| } |
| pub const fn is_unique_local(&self) -> bool { |
| (self.segments()[0] & 0xfe00) == 0xfc00 |
| } |
| pub const fn is_unicast(&self) -> bool { !self.is_multicast() } |
| pub const fn is_unicast_link_local(&self) -> bool { |
| (self.segments()[0] & 0xffc0) == 0xfe80 |
| } |
| pub const fn is_documentation(&self) -> bool { |
| (self.segments()[0] == 0x2001) && (self.segments()[1] == 0xdb8) |
| } |
| pub const fn is_benchmarking(&self) -> bool { |
| (self.segments()[0] == 0x2001) && (self.segments()[1] == 0x2) && |
| (self.segments()[2] == 0) |
| } |
| pub const fn is_unicast_global(&self) -> bool { |
| self.is_unicast() && !self.is_loopback() && |
| !self.is_unicast_link_local() && !self.is_unique_local() && |
| !self.is_unspecified() && !self.is_documentation() |
| } |
| pub const fn multicast_scope(&self) -> Option<Ipv6MulticastScope> { |
| if self.is_multicast() { |
| match self.segments()[0] & 0x000f { |
| 1 => Some(Ipv6MulticastScope::InterfaceLocal), |
| 2 => Some(Ipv6MulticastScope::LinkLocal), |
| 3 => Some(Ipv6MulticastScope::RealmLocal), |
| 4 => Some(Ipv6MulticastScope::AdminLocal), |
| 5 => Some(Ipv6MulticastScope::SiteLocal), |
| 8 => Some(Ipv6MulticastScope::OrganizationLocal), |
| 14 => Some(Ipv6MulticastScope::Global), |
| _ => None, |
| } |
| } else { None } |
| } |
| pub const fn is_multicast(&self) -> bool { |
| (self.segments()[0] & 0xff00) == 0xff00 |
| } |
| pub const fn to_ipv4_mapped(&self) -> Option<Ipv4Addr> { |
| match self.octets() { |
| [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0xff, 0xff, a, b, c, d] => { |
| Some(Ipv4Addr::new(a, b, c, d)) |
| } |
| _ => None, |
| } |
| } |
| pub const fn to_ipv4(&self) -> Option<Ipv4Addr> { |
| if let [0, 0, 0, 0, 0, 0 | 0xffff, ab, cd] = self.segments() { |
| let [a, b] = ab.to_be_bytes(); |
| let [c, d] = cd.to_be_bytes(); |
| Some(Ipv4Addr::new(a, b, c, d)) |
| } else { None } |
| } |
| pub const fn to_canonical(&self) -> IpAddr { |
| if let Some(mapped) = self.to_ipv4_mapped() { |
| return IpAddr::V4(mapped); |
| } |
| IpAddr::V6(*self) |
| } |
| pub const fn octets(&self) -> [u8; 16] { self.inner.s6_addr } |
| } |
| impl fmt::Display for Ipv6Addr { |
| fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
| if f.precision().is_none() && f.width().is_none() { |
| let segments = self.segments(); |
| if self.is_unspecified() { |
| f.write_str("::") |
| } else if self.is_loopback() { |
| f.write_str("::1") |
| } else if let Some(ipv4) = self.to_ipv4() { |
| match segments[5] { |
| 0 => write!(f, "::{}", ipv4), |
| 0xffff => write!(f, "::ffff:{}", ipv4), |
| _ => unreachable!(), |
| } |
| } else { |
| #[derive(Copy, Clone, Default)] |
| struct Span { |
| start: usize, |
| len: usize, |
| } |
| let zeroes = |
| { |
| let mut longest = Span::default(); |
| let mut current = Span::default(); |
| for (i, &segment) in segments.iter().enumerate() { |
| if segment == 0 { |
| if current.len == 0 { current.start = i; } |
| current.len += 1; |
| if current.len > longest.len { longest = current; } |
| } else { current = Span::default(); } |
| } |
| longest |
| }; |
| #[doc = " Write a colon-separated part of the address"] |
| #[inline] |
| fn fmt_subslice(f: &mut fmt::Formatter<'_>, chunk: &[u16]) |
| -> fmt::Result { |
| if let Some((first, tail)) = chunk.split_first() { |
| write!(f, "{:x}", first)?; |
| for segment in tail { |
| f.write_char(':')?; |
| write!(f, "{:x}", segment)?; |
| } |
| } |
| Ok(()) |
| } |
| if zeroes.len > 1 { |
| fmt_subslice(f, &segments[..zeroes.start])?; |
| f.write_str("::")?; |
| fmt_subslice(f, &segments[zeroes.start + zeroes.len..]) |
| } else { fmt_subslice(f, &segments) } |
| } |
| } else { |
| const IPV6_BUF_LEN: usize = (4 * 8) + 7; |
| let mut buf = [0u8; IPV6_BUF_LEN]; |
| let mut buf_slice = &mut buf[..]; |
| write!(buf_slice, "{}", self).unwrap(); |
| let len = IPV6_BUF_LEN - buf_slice.len(); |
| let buf = |
| unsafe { crate::str::from_utf8_unchecked(&buf[..len]) }; |
| f.pad(buf) |
| } |
| } |
| } |
| impl fmt::Debug for Ipv6Addr { |
| fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result { |
| fmt::Display::fmt(self, fmt) |
| } |
| } |
| impl Clone for Ipv6Addr { |
| fn clone(&self) -> Ipv6Addr { *self } |
| } |
| impl PartialEq for Ipv6Addr { |
| fn eq(&self, other: &Ipv6Addr) -> bool { |
| self.inner.s6_addr == other.inner.s6_addr |
| } |
| } |
| impl PartialEq<IpAddr> for Ipv6Addr { |
| fn eq(&self, other: &IpAddr) -> bool { |
| match other { IpAddr::V4(_) => false, IpAddr::V6(v6) => self == v6, } |
| } |
| } |
| impl PartialEq<Ipv6Addr> for IpAddr { |
| fn eq(&self, other: &Ipv6Addr) -> bool { |
| match self { IpAddr::V4(_) => false, IpAddr::V6(v6) => v6 == other, } |
| } |
| } |
| impl Eq for Ipv6Addr {} |
| impl hash::Hash for Ipv6Addr { |
| fn hash<H: hash::Hasher>(&self, s: &mut H) { self.inner.s6_addr.hash(s) } |
| } |
| impl PartialOrd for Ipv6Addr { |
| fn partial_cmp(&self, other: &Ipv6Addr) -> Option<Ordering> { |
| Some(self.cmp(other)) |
| } |
| } |
| impl PartialOrd<Ipv6Addr> for IpAddr { |
| fn partial_cmp(&self, other: &Ipv6Addr) -> Option<Ordering> { |
| match self { |
| IpAddr::V4(_) => Some(Ordering::Less), |
| IpAddr::V6(v6) => v6.partial_cmp(other), |
| } |
| } |
| } |
| impl PartialOrd<IpAddr> for Ipv6Addr { |
| fn partial_cmp(&self, other: &IpAddr) -> Option<Ordering> { |
| match other { |
| IpAddr::V4(_) => Some(Ordering::Greater), |
| IpAddr::V6(v6) => self.partial_cmp(v6), |
| } |
| } |
| } |
| impl Ord for Ipv6Addr { |
| fn cmp(&self, other: &Ipv6Addr) -> Ordering { |
| self.segments().cmp(&other.segments()) |
| } |
| } |
| impl AsInner<c::in6_addr> for Ipv6Addr { |
| fn as_inner(&self) -> &c::in6_addr { &self.inner } |
| } |
| impl FromInner<c::in6_addr> for Ipv6Addr { |
| fn from_inner(addr: c::in6_addr) -> Ipv6Addr { Ipv6Addr { inner: addr } } |
| } |
| impl From<Ipv6Addr> for u128 { |
| fn from(ip: Ipv6Addr) -> u128 { |
| let ip = ip.octets(); |
| u128::from_be_bytes(ip) |
| } |
| } |
| impl From<u128> for Ipv6Addr { |
| fn from(ip: u128) -> Ipv6Addr { Ipv6Addr::from(ip.to_be_bytes()) } |
| } |
| impl From<[u8; 16]> for Ipv6Addr { |
| fn from(octets: [u8; 16]) -> Ipv6Addr { |
| let inner = c::in6_addr { s6_addr: octets }; |
| Ipv6Addr::from_inner(inner) |
| } |
| } |
| impl From<[u16; 8]> for Ipv6Addr { |
| fn from(segments: [u16; 8]) -> Ipv6Addr { |
| let [a, b, c, d, e, f, g, h] = segments; |
| Ipv6Addr::new(a, b, c, d, e, f, g, h) |
| } |
| } |
| impl From<[u8; 16]> for IpAddr { |
| fn from(octets: [u8; 16]) -> IpAddr { IpAddr::V6(Ipv6Addr::from(octets)) } |
| } |
| impl From<[u16; 8]> for IpAddr { |
| fn from(segments: [u16; 8]) -> IpAddr { |
| IpAddr::V6(Ipv6Addr::from(segments)) |
| } |
| } |
