1use std::fmt;
7
8use num_enum::TryFromPrimitive;
9
10use dvb_common::{Parse, Serialize};
11
12#[derive(Debug, Clone, Copy, PartialEq, Eq, TryFromPrimitive)]
14#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
15#[repr(u8)]
16pub enum AddressingFunctionTag {
17 TimeOffset = 0x00,
19 FrequencyOffset = 0x01,
21 Power = 0x02,
23 PrivateData = 0x03,
25 CellId = 0x04,
27 Enable = 0x05,
29 Bandwidth = 0x06,
31 AcePapr = 0x10,
33 MisoGroup = 0x11,
35 TrPapr = 0x12,
37 L1AcePapr = 0x13,
39 TxSigFefSeqNum = 0x15,
41 TxSigAuxStreamTxId = 0x16,
43 Frequency = 0x17,
45}
46
47impl From<AddressingFunctionTag> for u8 {
48 fn from(tag: AddressingFunctionTag) -> Self {
49 tag as u8
50 }
51}
52
53impl fmt::Display for AddressingFunctionTag {
54 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
55 match self {
56 AddressingFunctionTag::TimeOffset => write!(f, "TimeOffset"),
57 AddressingFunctionTag::FrequencyOffset => write!(f, "FrequencyOffset"),
58 AddressingFunctionTag::Power => write!(f, "Power"),
59 AddressingFunctionTag::PrivateData => write!(f, "PrivateData"),
60 AddressingFunctionTag::CellId => write!(f, "CellId"),
61 AddressingFunctionTag::Enable => write!(f, "Enable"),
62 AddressingFunctionTag::Bandwidth => write!(f, "Bandwidth"),
63 AddressingFunctionTag::AcePapr => write!(f, "AcePapr"),
64 AddressingFunctionTag::MisoGroup => write!(f, "MisoGroup"),
65 AddressingFunctionTag::TrPapr => write!(f, "TrPapr"),
66 AddressingFunctionTag::L1AcePapr => write!(f, "L1AcePapr"),
67 AddressingFunctionTag::TxSigFefSeqNum => write!(f, "TxSigFefSeqNum"),
68 AddressingFunctionTag::TxSigAuxStreamTxId => write!(f, "TxSigAuxStreamTxId"),
69 AddressingFunctionTag::Frequency => write!(f, "Frequency"),
70 }
71 }
72}
73
74#[derive(Debug, Clone, PartialEq, Eq)]
76#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
77pub struct FunctionEntry<'a> {
78 pub tag: AddressingFunctionTag,
80 #[cfg_attr(feature = "serde", serde(borrow))]
82 pub raw: &'a [u8],
83}
84
85#[derive(Debug, Clone, PartialEq, Eq)]
94#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
95pub struct IndividualAddressingPayload<'a> {
96 pub rfu: u8,
98 #[cfg_attr(feature = "serde", serde(borrow))]
101 pub individual_addressing_data: &'a [u8],
102}
103
104const HEADER_LEN: usize = 2;
105
106impl<'a> Parse<'a> for IndividualAddressingPayload<'a> {
107 type Error = crate::error::Error;
108
109 fn parse(bytes: &'a [u8]) -> Result<Self, crate::error::Error> {
110 if bytes.len() < HEADER_LEN {
111 return Err(crate::Error::BufferTooShort {
112 need: HEADER_LEN,
113 have: bytes.len(),
114 what: "IndividualAddressingPayload header",
115 });
116 }
117
118 let rfu = bytes[0];
119 let individual_addressing_length = bytes[1] as usize;
120 let need = HEADER_LEN + individual_addressing_length;
122 if bytes.len() < need {
123 return Err(crate::Error::BufferTooShort {
124 need,
125 have: bytes.len(),
126 what: "IndividualAddressingPayload data",
127 });
128 }
129
130 Ok(IndividualAddressingPayload {
131 rfu,
132 individual_addressing_data: &bytes[HEADER_LEN..need],
133 })
134 }
135}
136
137impl Serialize for IndividualAddressingPayload<'_> {
138 type Error = crate::error::Error;
139
140 fn serialized_len(&self) -> usize {
141 HEADER_LEN + self.individual_addressing_data.len()
142 }
143
144 fn serialize_into(&self, buf: &mut [u8]) -> Result<usize, crate::error::Error> {
145 if buf.len() < self.serialized_len() {
146 return Err(crate::Error::OutputBufferTooSmall {
147 need: self.serialized_len(),
148 have: buf.len(),
149 });
150 }
151
152 if self.individual_addressing_data.len() > u8::MAX as usize {
155 return Err(crate::Error::ReservedBitsViolation {
156 field: "individual_addressing_length",
157 reason: "individual_addressing_data exceeds 255 bytes (8-bit length field)",
158 });
159 }
160
161 buf[0] = self.rfu;
162 buf[1] = self.individual_addressing_data.len() as u8;
163
164 if !self.individual_addressing_data.is_empty() {
165 buf[HEADER_LEN..HEADER_LEN + self.individual_addressing_data.len()]
166 .copy_from_slice(self.individual_addressing_data);
167 }
168
169 Ok(self.serialized_len())
170 }
171}
172
173impl fmt::Display for IndividualAddressingPayload<'_> {
174 fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
175 write!(
176 f,
177 "IndividualAddressing {{ rfu: 0x{:02X}, addr_data_len: {} }}",
178 self.rfu,
179 self.individual_addressing_data.len()
180 )
181 }
182}
183
184#[cfg(test)]
185mod tests {
186 use super::*;
187
188 #[test]
189 fn addressing_function_tag_try_from_valid() {
190 assert_eq!(
191 AddressingFunctionTag::try_from(0x10),
192 Ok(AddressingFunctionTag::AcePapr)
193 );
194 assert_eq!(
195 AddressingFunctionTag::try_from(0x17),
196 Ok(AddressingFunctionTag::Frequency)
197 );
198 }
199
200 #[test]
201 fn addressing_function_tag_try_from_rejects_unknown() {
202 assert!(AddressingFunctionTag::try_from(0x14).is_err());
203 assert!(AddressingFunctionTag::try_from(0xFF).is_err());
204 }
205
206 #[test]
207 fn exhaustive_byte_sweep() {
208 let mut matched = 0u16;
209 for byte in 0u8..=0xFF {
210 if let Ok(v) = AddressingFunctionTag::try_from(byte) {
211 assert_eq!(v as u8, byte, "round-trip failed for {byte:#04x}");
212 matched += 1;
213 }
214 }
215 assert_eq!(matched, 14, "expected 14 matched variants");
216 }
217
218 #[test]
219 fn address_function_tag_display() {
220 assert_eq!(AddressingFunctionTag::AcePapr.to_string(), "AcePapr");
221 }
222
223 #[test]
224 fn parse_extracts_rfu_and_addressing_data() {
225 let buf = [0x00u8, 0x04, 0x00, 0x05, 0x04, 0x10];
229 let result = IndividualAddressingPayload::parse(&buf).unwrap();
230 assert_eq!(result.rfu, 0x00);
231 assert_eq!(result.individual_addressing_data, &[0x00, 0x05, 0x04, 0x10]);
232 }
233
234 #[test]
235 fn parse_preserves_rfu_byte() {
236 let buf = [0xFFu8, 0x00];
237 let result = IndividualAddressingPayload::parse(&buf).unwrap();
238 assert_eq!(result.rfu, 0xFF);
239 assert!(result.individual_addressing_data.is_empty());
240 }
241
242 #[test]
243 fn parse_rejects_short_buffer() {
244 assert!(IndividualAddressingPayload::parse(&[0x00]).is_err());
245 }
246
247 #[test]
248 fn parse_rejects_truncated_data() {
249 assert!(IndividualAddressingPayload::parse(&[0x00, 0x04, 0xAA, 0xBB]).is_err());
251 }
252
253 #[test]
254 fn serialize_round_trip() {
255 let orig = IndividualAddressingPayload {
256 rfu: 0x00,
257 individual_addressing_data: &[0x00, 0x03, 0x04, 0xDE, 0xAD],
258 };
259 let mut buf = vec![0u8; orig.serialized_len()];
260 orig.serialize_into(&mut buf).unwrap();
261 assert_eq!(buf[1], 5);
263 let parsed = IndividualAddressingPayload::parse(&buf).unwrap();
264 assert_eq!(orig, parsed);
265 }
266
267 #[test]
268 fn serialize_empty_data() {
269 let orig = IndividualAddressingPayload {
270 rfu: 0x00,
271 individual_addressing_data: &[],
272 };
273 let mut buf = vec![0u8; orig.serialized_len()];
274 orig.serialize_into(&mut buf).unwrap();
275 assert_eq!(buf, [0x00, 0x00]);
276 }
277}