use crate::descriptors::entry_list::{HEADER_LEN, parse_masked, write_masked};
use crate::error::Result;
use alloc::vec::Vec;
use broadcast_common::{Parse, Serialize};
use core::net::Ipv6Addr;
pub const TAG: u8 = 0x0A;
const ADDR_LEN: usize = 16;
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize))]
#[cfg_attr(feature = "yoke", derive(yoke::Yokeable))]
pub struct TargetIpv6AddressDescriptor<'a> {
pub addr_mask: Ipv6Addr,
pub addrs: Vec<Ipv6Addr>,
pub trailing: &'a [u8],
}
impl<'a> Parse<'a> for TargetIpv6AddressDescriptor<'a> {
type Error = crate::error::Error;
fn parse(bytes: &'a [u8]) -> Result<Self> {
let (addr_mask, addrs, trailing) =
parse_masked::<ADDR_LEN, Ipv6Addr>(bytes, TAG, "TargetIpv6AddressDescriptor")?;
Ok(Self {
addr_mask,
addrs,
trailing,
})
}
}
impl Serialize for TargetIpv6AddressDescriptor<'_> {
type Error = crate::error::Error;
fn serialized_len(&self) -> usize {
HEADER_LEN + ADDR_LEN + self.addrs.len() * ADDR_LEN + self.trailing.len()
}
fn serialize_into(&self, buf: &mut [u8]) -> Result<usize> {
write_masked::<ADDR_LEN, Ipv6Addr>(buf, TAG, self.addr_mask, &self.addrs, self.trailing)
}
}
impl<'a> crate::traits::DescriptorDef<'a> for TargetIpv6AddressDescriptor<'a> {
const TAG: u8 = TAG;
const NAME: &'static str = "TARGET_IPV6_ADDRESS";
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn parse_and_round_trip() {
let mask = Ipv6Addr::new(0xffff, 0xf0, 0, 0, 0, 0, 0, 0);
let a1 = Ipv6Addr::new(0xfe80, 0, 0, 0, 0x93a3, 0xdea0, 0x2108, 0xb81e);
let mut bytes = vec![TAG, 48];
for a in [mask, a1, Ipv6Addr::LOCALHOST] {
bytes.extend_from_slice(&a.octets());
}
let d = TargetIpv6AddressDescriptor::parse(&bytes).unwrap();
assert_eq!(d.addr_mask, mask);
assert_eq!(d.addrs, [a1, Ipv6Addr::LOCALHOST]);
let mut out = vec![0u8; d.serialized_len()];
d.serialize_into(&mut out).unwrap();
assert_eq!(out, bytes);
}
#[test]
fn partial_tail_round_trip_and_rejects_short_mask() {
let mut bytes = vec![TAG, 18];
bytes.extend_from_slice(&[0xAA; 18]);
let d = TargetIpv6AddressDescriptor::parse(&bytes).unwrap();
assert_eq!(d.trailing.len(), 2);
let mut out = vec![0u8; d.serialized_len()];
d.serialize_into(&mut out).unwrap();
assert_eq!(out, bytes);
assert!(TargetIpv6AddressDescriptor::parse(&[TAG, 3, 1, 2, 3]).is_err());
}
}