use crate::descriptors::entry_list::{Entry, list_len, parse_list, write_list};
use crate::error::Result;
use alloc::vec::Vec;
use broadcast_common::{Parse, Serialize};
use core::net::Ipv6Addr;
pub const TAG: u8 = 0x11;
const ENTRY_LEN: usize = 17;
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize))]
pub struct Ipv6Slash {
pub addr: Ipv6Addr,
pub slash_mask: u8,
}
impl Entry<ENTRY_LEN> for Ipv6Slash {
fn read(b: &[u8; ENTRY_LEN]) -> Self {
let mut a = [0u8; 16];
a.copy_from_slice(&b[..16]);
Self {
addr: Ipv6Addr::from(a),
slash_mask: b[16],
}
}
fn write(&self, b: &mut [u8; ENTRY_LEN]) {
b[..16].copy_from_slice(&self.addr.octets());
b[16] = self.slash_mask;
}
}
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize))]
#[cfg_attr(feature = "yoke", derive(yoke::Yokeable))]
pub struct TargetIpv6SlashDescriptor<'a> {
pub entries: Vec<Ipv6Slash>,
pub trailing: &'a [u8],
}
impl<'a> Parse<'a> for TargetIpv6SlashDescriptor<'a> {
type Error = crate::error::Error;
fn parse(bytes: &'a [u8]) -> Result<Self> {
let (entries, trailing) =
parse_list::<ENTRY_LEN, Ipv6Slash>(bytes, TAG, "TargetIpv6SlashDescriptor")?;
Ok(Self { entries, trailing })
}
}
impl Serialize for TargetIpv6SlashDescriptor<'_> {
type Error = crate::error::Error;
fn serialized_len(&self) -> usize {
list_len(self.entries.len(), ENTRY_LEN, self.trailing.len())
}
fn serialize_into(&self, buf: &mut [u8]) -> Result<usize> {
write_list::<ENTRY_LEN, _>(buf, TAG, &self.entries, self.trailing)
}
}
impl<'a> crate::traits::DescriptorDef<'a> for TargetIpv6SlashDescriptor<'a> {
const TAG: u8 = TAG;
const NAME: &'static str = "TARGET_IPV6_SLASH";
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn serialize_rejects_entry_sized_trailing() {
let mut d = TargetIpv6SlashDescriptor::parse(&[TAG, 0]).unwrap();
let long = [0u8; ENTRY_LEN];
d.trailing = &long;
assert!(d.serialize_into(&mut [0u8; 64]).is_err());
let mut buf = [0u8; 64];
d.trailing = &long[..ENTRY_LEN - 1];
let n = d.serialize_into(&mut buf).unwrap();
assert_eq!(TargetIpv6SlashDescriptor::parse(&buf[..n]).unwrap(), d);
}
#[test]
fn parse_and_round_trip() {
let a = Ipv6Addr::new(1, 2, 3, 4, 5, 6, 7, 8);
let mut bytes = vec![TAG, 17];
bytes.extend_from_slice(&a.octets());
bytes.push(72);
let d = TargetIpv6SlashDescriptor::parse(&bytes).unwrap();
assert_eq!(
d.entries,
[Ipv6Slash {
addr: a,
slash_mask: 72
}]
);
let mut out = vec![0u8; d.serialized_len()];
d.serialize_into(&mut out).unwrap();
assert_eq!(out, bytes);
}
#[test]
fn empty_and_partial_tail_round_trip() {
for bytes in [&[TAG, 0][..], &[TAG, 2, 7, 7][..]] {
let d = TargetIpv6SlashDescriptor::parse(bytes).unwrap();
let mut out = vec![0u8; d.serialized_len()];
d.serialize_into(&mut out).unwrap();
assert_eq!(out, bytes);
}
assert!(TargetIpv6SlashDescriptor::parse(&[0x12, 0]).is_err());
}
}