use super::*;
use broadcast_common::len::need_out;
const XAIT_PID_LEN: usize = 2;
pub const MAX_PID: u16 = 0x1FFF;
impl ExtensionBodyDef<'_> for XaitPid {
const TAG_EXTENSION: u8 = 0x0C;
const NAME: &'static str = "XAIT_PID";
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize))]
pub struct XaitPid {
pub xait_pid: u16,
}
impl XaitPid {}
impl<'a> Parse<'a> for XaitPid {
type Error = crate::error::Error;
fn parse(sel: &'a [u8]) -> Result<Self> {
if sel.len() != XAIT_PID_LEN {
return Err(Error::InvalidDescriptor {
tag: crate::descriptors::extension::TAG,
reason: "xait_pid body length must equal 2",
});
}
Ok(Self {
xait_pid: u16::from_be_bytes([sel[0], sel[1]]),
})
}
}
impl Serialize for XaitPid {
type Error = crate::error::Error;
fn serialized_len(&self) -> usize {
XAIT_PID_LEN
}
fn serialize_into(&self, buf: &mut [u8]) -> Result<usize> {
let len = self.serialized_len();
need_out(buf, len)?;
buf[0..2].copy_from_slice(&self.xait_pid.to_be_bytes());
Ok(len)
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::descriptors::extension::test_support::*;
use crate::descriptors::extension::{ExtensionBody, ExtensionDescriptor, ExtensionTag};
#[test]
fn parse_default_pid() {
let sel = [0x1F, 0xFC];
let bytes = wrap(0x0C, &sel);
let d = ExtensionDescriptor::parse(&bytes).unwrap();
assert_eq!(d.kind(), Some(ExtensionTag::XaitPid));
match &d.body {
ExtensionBody::XaitPid(b) => {
assert_eq!(b.xait_pid, 0x1FFC);
}
other => panic!("expected XaitPid, got {other:?}"),
}
}
#[test]
fn serialize_round_trip() {
let d = ExtensionDescriptor {
tag_extension: 0x0C,
body: ExtensionBody::XaitPid(XaitPid { xait_pid: 0x0100 }),
};
round_trip(&d);
}
#[test]
fn serialize_byte_identical() {
let sel = [0x01, 0x00];
let bytes = wrap(0x0C, &sel);
let d = ExtensionDescriptor::parse(&bytes).unwrap();
let mut buf = vec![0u8; d.serialized_len()];
d.serialize_into(&mut buf).unwrap();
assert_eq!(buf.as_slice(), &bytes[..]);
}
#[test]
fn high_bits_preserved_verbatim() {
let sel = [0xE1, 0x00];
let bytes = wrap(0x0C, &sel);
let d = ExtensionDescriptor::parse(&bytes).unwrap();
match &d.body {
ExtensionBody::XaitPid(b) => {
assert_eq!(b.xait_pid, 0xE100);
}
other => panic!("expected XaitPid, got {other:?}"),
}
let mut buf = vec![0u8; d.serialized_len()];
d.serialize_into(&mut buf).unwrap();
assert_eq!(buf.as_slice(), &bytes[..]);
}
}