use crate::ParseError::InvalidIPAddress;
use crate::{IPAddress, ParseError, impl_parse};
use std::net::IpAddr;
use std::str::FromStr;
impl IPAddress {
pub fn parse(text: &[u8]) -> Result<Self, ParseError> {
const MAX_STR_LEN: usize = "ffff:ffff:ffff:ffff:ffff:ffff:255.255.255.255".len();
if text.len() > MAX_STR_LEN {
return Err(InvalidIPAddress);
}
let text: &str = std::str::from_utf8(text).map_err(|_| InvalidIPAddress)?;
Ok(IpAddr::from_str(text).map_err(|_| InvalidIPAddress)?.into())
}
}
impl_parse!(IPAddress);
#[cfg(test)]
mod tests {
use crate::ParseError::InvalidIPAddress;
use crate::{IPAddress, IPv4Address, IPv6Address, ParseError};
use std::str::FromStr;
type TestCase<'a> = (&'a [u8], Result<IPAddress, ParseError>);
const MAX_STR_LEN: usize = "ffff:ffff:ffff:ffff:ffff:ffff:255.255.255.255".len();
#[test]
fn parse() {
let over_max: Vec<u8> = vec![b'0'; MAX_STR_LEN + 1];
let test_cases: &[TestCase] = &[
(b"", Err(InvalidIPAddress)),
(b"127.0.0.1", Ok(IPv4Address::LOCALHOST.to_ip())),
(b"::1", Ok(IPv6Address::LOCALHOST.to_ip())),
(
b"::ffff:1.2.3.4",
Ok(IPv6Address::from([0, 0, 0, 0, 0, 0xFFFF, 0x0102, 0x0304]).to_ip()),
),
(b"[::1]", Err(InvalidIPAddress)),
(b"fe80::1%1", Err(InvalidIPAddress)),
(b"localhost", Err(InvalidIPAddress)),
(b"\xFF", Err(InvalidIPAddress)),
(over_max.as_slice(), Err(InvalidIPAddress)),
];
for (input, expected) in test_cases {
let result: Result<IPAddress, ParseError> = IPAddress::parse(input);
assert_eq!(result, *expected, "parse input={:?}", input);
let Ok(text) = std::str::from_utf8(input) else {
continue;
};
let result: Result<IPAddress, ParseError> = IPAddress::from_str(text);
assert_eq!(result, *expected, "from_str input={}", text);
let result: Result<IPAddress, ParseError> = IPAddress::try_from(text);
assert_eq!(result, *expected, "try_from(&str) input={}", text);
}
}
#[test]
fn round_trip() {
let canonical: &[&str] = &[
"127.0.0.1",
"255.255.255.255",
"::1",
"fe80::1",
"::ffff:1.2.3.4",
];
for input in canonical {
let value: IPAddress = input.parse().unwrap();
assert_eq!(value.to_string(), *input, "input={}", input);
}
}
}