use std::net::{SocketAddr, SocketAddrV4, SocketAddrV6};
use crate::{IPAddress, SocketAddress, SocketAddressV4, SocketAddressV6};
impl SocketAddressV4 {
#[must_use]
pub const fn to_std(self) -> SocketAddrV4 {
SocketAddrV4::new(self.ip().to_std(), self.port())
}
}
impl From<SocketAddrV4> for SocketAddressV4 {
fn from(std: SocketAddrV4) -> Self {
Self::new((*std.ip()).into(), std.port())
}
}
impl From<SocketAddressV4> for SocketAddrV4 {
fn from(socket: SocketAddressV4) -> Self {
socket.to_std()
}
}
impl SocketAddressV6 {
#[must_use]
pub const fn to_std(self) -> SocketAddrV6 {
self.to_std_with(0, 0)
}
#[must_use]
pub const fn to_std_with(self, flow_info: u32, scope_id: u32) -> SocketAddrV6 {
SocketAddrV6::new(self.ip().to_std(), self.port(), flow_info, scope_id)
}
}
impl From<SocketAddrV6> for SocketAddressV6 {
fn from(std: SocketAddrV6) -> Self {
Self::new((*std.ip()).into(), std.port())
}
}
impl From<SocketAddressV6> for SocketAddrV6 {
fn from(socket: SocketAddressV6) -> Self {
socket.to_std()
}
}
impl SocketAddress {
#[must_use]
pub const fn to_std(self) -> SocketAddr {
self.to_std_with(0, 0)
}
#[must_use]
pub const fn to_std_with(self, flow_info: u32, scope_id: u32) -> SocketAddr {
match self.ip() {
IPAddress::V4(ip) => SocketAddr::V4(SocketAddrV4::new(ip.to_std(), self.port())),
IPAddress::V6(ip) => SocketAddr::V6(SocketAddrV6::new(
ip.to_std(),
self.port(),
flow_info,
scope_id,
)),
}
}
}
impl From<SocketAddr> for SocketAddress {
fn from(std: SocketAddr) -> Self {
Self::new(std.ip().into(), std.port())
}
}
impl From<SocketAddress> for SocketAddr {
fn from(socket: SocketAddress) -> Self {
socket.to_std()
}
}
#[cfg(test)]
mod tests {
use std::net::{Ipv4Addr, Ipv6Addr, SocketAddr, SocketAddrV4, SocketAddrV6};
use crate::{IPv4Address, IPv6Address, SocketAddress, SocketAddressV4, SocketAddressV6};
#[test]
fn v4() {
let socket: SocketAddressV4 = SocketAddressV4::new(IPv4Address::LOCALHOST, 80);
let result: SocketAddrV4 = socket.to_std();
let expected: SocketAddrV4 = SocketAddrV4::new(Ipv4Addr::LOCALHOST, 80);
assert_eq!(result, expected);
let socket: SocketAddressV4 = SocketAddressV4::new(IPv4Address::LOCALHOST, 80);
let result: SocketAddrV4 = socket.into();
let expected: SocketAddrV4 = SocketAddrV4::new(Ipv4Addr::LOCALHOST, 80);
assert_eq!(result, expected);
let socket: SocketAddrV4 = SocketAddrV4::new(Ipv4Addr::LOCALHOST, 80);
let result: SocketAddressV4 = socket.into();
let expected: SocketAddressV4 = SocketAddressV4::new(IPv4Address::LOCALHOST, 80);
assert_eq!(result, expected);
}
#[test]
fn v6() {
let socket: SocketAddressV6 = SocketAddressV6::new(IPv6Address::LOCALHOST, 80);
let result: SocketAddrV6 = socket.to_std_with(123, 456);
let expected: SocketAddrV6 = SocketAddrV6::new(Ipv6Addr::LOCALHOST, 80, 123, 456);
assert_eq!(result, expected);
let socket: SocketAddressV6 = SocketAddressV6::new(IPv6Address::LOCALHOST, 80);
let result: SocketAddrV6 = socket.to_std();
let expected: SocketAddrV6 = SocketAddrV6::new(Ipv6Addr::LOCALHOST, 80, 0, 0);
assert_eq!(result, expected);
let socket: SocketAddressV6 = SocketAddressV6::new(IPv6Address::LOCALHOST, 80);
let result: SocketAddrV6 = socket.into();
let expected: SocketAddrV6 = SocketAddrV6::new(Ipv6Addr::LOCALHOST, 80, 0, 0);
assert_eq!(result, expected);
let socket: SocketAddrV6 = SocketAddrV6::new(Ipv6Addr::LOCALHOST, 80, 123, 456);
let result: SocketAddressV6 = socket.into();
let expected: SocketAddressV6 = SocketAddressV6::new(IPv6Address::LOCALHOST, 80);
assert_eq!(result, expected);
}
#[test]
fn socket() {
let socket: SocketAddress = SocketAddress::new(IPv4Address::LOCALHOST.to_ip(), 80);
let result: SocketAddr = socket.to_std_with(123, 456);
let expected: SocketAddr = SocketAddr::V4(SocketAddrV4::new(Ipv4Addr::LOCALHOST, 80));
assert_eq!(result, expected);
let socket: SocketAddress = SocketAddress::new(IPv6Address::LOCALHOST.to_ip(), 80);
let result: SocketAddr = socket.to_std_with(123, 456);
let expected: SocketAddr =
SocketAddr::V6(SocketAddrV6::new(Ipv6Addr::LOCALHOST, 80, 123, 456));
assert_eq!(result, expected);
let socket: SocketAddress = SocketAddress::new(IPv6Address::LOCALHOST.to_ip(), 80);
let result: SocketAddr = socket.to_std();
let expected: SocketAddr = SocketAddr::V6(SocketAddrV6::new(Ipv6Addr::LOCALHOST, 80, 0, 0));
assert_eq!(result, expected);
let socket: SocketAddress = SocketAddress::new(IPv4Address::LOCALHOST.to_ip(), 80);
let result: SocketAddr = socket.into();
let expected: SocketAddr = SocketAddr::V4(SocketAddrV4::new(Ipv4Addr::LOCALHOST, 80));
assert_eq!(result, expected);
let socket: SocketAddress = SocketAddress::new(IPv6Address::LOCALHOST.to_ip(), 80);
let result: SocketAddr = socket.into();
let expected: SocketAddr = SocketAddr::V6(SocketAddrV6::new(Ipv6Addr::LOCALHOST, 80, 0, 0));
assert_eq!(result, expected);
let std: SocketAddr = SocketAddr::V4(SocketAddrV4::new(Ipv4Addr::LOCALHOST, 80));
let result: SocketAddress = std.into();
let expected: SocketAddress = SocketAddress::new(IPv4Address::LOCALHOST.to_ip(), 80);
assert_eq!(result, expected);
let std: SocketAddr = SocketAddr::V6(SocketAddrV6::new(Ipv6Addr::LOCALHOST, 80, 123, 456));
let result: SocketAddress = std.into();
let expected: SocketAddress = SocketAddress::new(IPv6Address::LOCALHOST.to_ip(), 80);
assert_eq!(result, expected);
}
}