posix-socket 0.2.0

use crate::SpecificSocketAddress;

/// IPv6 socket address.
///
/// This includes an IPv6 address and a 16-bit port number.
#[derive(Clone)]
#[repr(C)]
pub struct Inet6SocketAddress {
	/// The inner C-compatible socket address.
	inner: libc::sockaddr_in6,
}

impl Inet6SocketAddress {
	/// Create an IPv6 socket address.
	pub fn new(ip: std::net::Ipv6Addr, port: u16, flowinfo: u32, scope_id: u32) -> Self {
		let inner = libc::sockaddr_in6 {
			sin6_family: Self::static_family(),
			sin6_addr: libc::in6_addr { s6_addr: ip.octets() },
			sin6_port: port.to_be(),
			sin6_flowinfo: flowinfo,
			sin6_scope_id: scope_id,
		};
		Self::from_raw(inner)
	}

	/// Create an IPv6 socket address from a [`libc::sockaddr_in6`].
	pub fn from_raw(inner: libc::sockaddr_in6) -> Self {
		Self { inner }
	}

	/// Convert the [`SocketAddress`] into raw [`libc`] parts.
	pub fn into_raw(self) -> libc::sockaddr_in6 {
		self.inner
	}

	/// Get the IP address associated with the socket address.
	pub fn ip(&self) -> std::net::Ipv6Addr {
		self.inner.sin6_addr.s6_addr.into()
	}

	/// Set the IP address associated with the socket address.
	pub fn set_ip(&mut self, ip: std::net::Ipv6Addr) {
		self.inner.sin6_addr.s6_addr = ip.octets();
	}

	/// Get the port number associated with the socket address.
	pub fn port(&self) -> u16 {
		u16::from_be(self.inner.sin6_port)
	}

	/// Set the port number associated with the socket address.
	pub fn set_port(&mut self, port: u16) {
		self.inner.sin6_port = port.to_be();
	}

	/// Get the flow information associated with the socket address.
	fn flowinfo(&self) -> u32 {
		self.inner.sin6_flowinfo
	}

	/// Set the flow information associated with the socket address.
	pub fn set_flowinfo(&mut self, flowinfo: u32) {
		self.inner.sin6_flowinfo = flowinfo;
	}

	/// Get the scope ID associated with the socket address.
	fn scope_id(&self) -> u32 {
		self.inner.sin6_scope_id
	}

	/// Set the scope ID associated with the socket address.
	pub fn set_scope_id(&mut self, scope_id: u32) {
		self.inner.sin6_scope_id = scope_id;
	}
}

impl SpecificSocketAddress for Inet6SocketAddress {
	fn static_family() -> libc::sa_family_t {
		libc::AF_INET6 as libc::sa_family_t
	}
}

unsafe impl crate::AsSocketAddress for Inet6SocketAddress {
	fn as_sockaddr(&self) -> *const libc::sockaddr {
		&self.inner as *const _ as *const _
	}

	fn as_sockaddr_mut(address: &mut std::mem::MaybeUninit<Self>) -> *mut libc::sockaddr {
		unsafe { &mut address.as_mut_ptr().as_mut().unwrap().inner as *mut _ as *mut _ }
	}

	fn len(&self) -> libc::socklen_t {
		Self::max_len()
	}

	fn finalize(address: std::mem::MaybeUninit<Self>, len: libc::socklen_t) -> std::io::Result<Self> {
		unsafe {
			let address = address.assume_init();
			if address.family() != Self::static_family() {
				return Err(std::io::Error::new(std::io::ErrorKind::InvalidData, "wrong address family, expeced AF_INET6"));
			}
			if len != Self::max_len() {
				return Err(std::io::Error::new(std::io::ErrorKind::InvalidData, "wrong address size"));
			}
			Ok(address)
		}
	}

	fn max_len() -> libc::socklen_t {
		std::mem::size_of::<libc::sockaddr_in6>() as libc::socklen_t
	}
}

impl From<Inet6SocketAddress> for crate::SocketAddress {
	fn from(other: Inet6SocketAddress) -> Self {
		Self::from(&other)
	}
}

impl From<&Inet6SocketAddress> for crate::SocketAddress {
	fn from(other: &Inet6SocketAddress) -> Self {
		Self::from_other(other)
	}
}

impl From<std::net::SocketAddrV6> for Inet6SocketAddress {
	fn from(other: std::net::SocketAddrV6) -> Self {
		Self::from(&other)
	}
}

impl From<&std::net::SocketAddrV6> for Inet6SocketAddress {
	fn from(other: &std::net::SocketAddrV6) -> Self {
		Self::new(*other.ip(), other.port(), other.flowinfo(), other.scope_id())
	}
}

impl From<Inet6SocketAddress> for std::net::SocketAddrV6 {
	fn from(other: Inet6SocketAddress) -> Self {
		Self::from(&other)
	}
}

impl From<&Inet6SocketAddress> for std::net::SocketAddrV6 {
	fn from(other: &Inet6SocketAddress) -> Self {
		Self::new(other.ip(), other.port(), other.flowinfo(), other.scope_id())
	}
}