use crate::ParseError::InvalidSocketAddress;
use crate::parse_port;
use crate::{IPv4Address, IPv6Address, ParseError, SocketAddress, doc_ignored_zone, impl_parse};
impl_parse!(SocketAddress, doc_ignored_zone!());
impl TryFrom<&[u8]> for SocketAddress {
type Error = ParseError;
#[doc = doc_ignored_zone!()]
fn try_from(socket: &[u8]) -> Result<Self, Self::Error> {
let (s, port): (&[u8], u16) = parse_port(socket)?;
if let Some(ip) = IPv6Address::parse_bracketed(s) {
Ok(ip?.to_ip().to_socket(port))
} else {
let ip: IPv4Address = IPv4Address::parse(s).map_err(|_| InvalidSocketAddress)?;
Ok(ip.to_ip().to_socket(port))
}
}
}
#[cfg(test)]
mod tests {
use crate::ParseError::{InvalidIPv6Address, InvalidPort, InvalidSocketAddress};
use crate::{IPv4Address, IPv6Address, ParseError, SocketAddress};
use std::str::FromStr;
#[test]
fn parse() {
let test_cases: &[(&str, Result<SocketAddress, ParseError>)] = &[
("", Err(InvalidPort)),
("[::1]:", Err(InvalidPort)),
("[::1]:xx", Err(InvalidPort)),
(":80", Err(InvalidSocketAddress)),
("xx:80", Err(InvalidSocketAddress)),
("::1:80", Err(InvalidSocketAddress)),
("[]:80", Err(InvalidIPv6Address)),
("[xx]:80", Err(InvalidIPv6Address)),
("[::1%eth0]:80", Err(InvalidIPv6Address)),
("127.0.0.1:80", Ok(IPv4Address::LOCALHOST.to_ip().to_socket(80))),
("0.0.0.0:0", Ok(IPv4Address::UNSPECIFIED.to_ip().to_socket(0))),
(
"255.255.255.255:65535",
Ok(IPv4Address::BROADCAST.to_ip().to_socket(65535)),
),
("[::1]:80", Ok(IPv6Address::LOCALHOST.to_socket(80).to_socket())),
("[::1%1]:80", Ok(IPv6Address::LOCALHOST.to_socket(80).to_socket())),
];
for (input, expected) in test_cases {
let result: Result<SocketAddress, ParseError> = SocketAddress::from_str(input);
assert_eq!(result, *expected, "input={}", input);
let result: Result<SocketAddress, ParseError> = SocketAddress::try_from(*input);
assert_eq!(result, *expected, "input={}", input);
let result: Result<SocketAddress, ParseError> = SocketAddress::try_from(input.as_bytes());
assert_eq!(result, *expected, "input={}", input);
}
}
#[test]
fn round_trip() {
let canonical: &[&str] = &["127.0.0.1:80", "[::1]:443", "[fe80::1]:0", "0.0.0.0:0"];
for input in canonical {
let value: SocketAddress = input.parse().unwrap();
assert_eq!(value.to_string(), *input, "input={}", input);
}
}
}