1use crate::error::{Error, Result};
34use crate::traits::Table;
35use dvb_common::{Parse, Serialize};
36
37pub const TABLE_ID: u8 = 0x75;
39
40pub const PID: u16 = 0x0000;
43
44const HEADER_LEN: usize = 3;
46
47const EXTENSION_HEADER_LEN: usize = 5;
50
51const CRC_LEN: usize = 4;
53
54const MIN_LEN: usize = HEADER_LEN + EXTENSION_HEADER_LEN + CRC_LEN;
56
57#[derive(Debug, Clone, PartialEq, Eq)]
62#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
63pub struct Container<'a> {
64 pub private_indicator: bool,
66 pub container_id: u16,
68 pub version_number: u8,
70 pub current_next_indicator: bool,
72 pub section_number: u8,
74 pub last_section_number: u8,
76 #[cfg_attr(feature = "serde", serde(borrow))]
79 pub container_data: &'a [u8],
80}
81
82impl<'a> Parse<'a> for Container<'a> {
83 type Error = crate::error::Error;
84
85 fn parse(bytes: &'a [u8]) -> Result<Self> {
86 if bytes.len() < MIN_LEN {
87 return Err(Error::BufferTooShort {
88 need: MIN_LEN,
89 have: bytes.len(),
90 what: "Container",
91 });
92 }
93
94 if bytes[0] != TABLE_ID {
95 return Err(Error::UnexpectedTableId {
96 table_id: bytes[0],
97 what: "Container",
98 expected: &[TABLE_ID],
99 });
100 }
101
102 let section_length = (((bytes[1] & 0x0F) as usize) << 8) | bytes[2] as usize;
106 let total = HEADER_LEN + section_length;
107 if bytes.len() < total {
108 return Err(Error::SectionLengthOverflow {
109 declared: section_length,
110 available: bytes.len() - HEADER_LEN,
111 });
112 }
113
114 let private_indicator = (bytes[1] & 0x40) != 0;
115
116 let container_id = u16::from_be_bytes([bytes[3], bytes[4]]);
118 let version_number = (bytes[5] >> 1) & 0x1F;
120 let current_next_indicator = (bytes[5] & 0x01) != 0;
121 let section_number = bytes[6];
122 let last_section_number = bytes[7];
123
124 let data_start = HEADER_LEN + EXTENSION_HEADER_LEN;
126 let data_end = total - CRC_LEN;
127 let container_data = &bytes[data_start..data_end];
128
129 Ok(Container {
130 private_indicator,
131 container_id,
132 version_number,
133 current_next_indicator,
134 section_number,
135 last_section_number,
136 container_data,
137 })
138 }
139}
140
141impl Serialize for Container<'_> {
142 type Error = crate::error::Error;
143
144 fn serialized_len(&self) -> usize {
145 HEADER_LEN + EXTENSION_HEADER_LEN + self.container_data.len() + CRC_LEN
146 }
147
148 fn serialize_into(&self, buf: &mut [u8]) -> Result<usize> {
149 let len = self.serialized_len();
150 if buf.len() < len {
151 return Err(Error::OutputBufferTooSmall {
152 need: len,
153 have: buf.len(),
154 });
155 }
156
157 let section_length = (len - HEADER_LEN) as u16;
158
159 buf[0] = TABLE_ID;
161 buf[1] = 0x80
164 | (u8::from(self.private_indicator) << 6)
165 | 0x30
166 | ((section_length >> 8) as u8 & 0x0F);
167 buf[2] = (section_length & 0xFF) as u8;
169
170 buf[3..5].copy_from_slice(&self.container_id.to_be_bytes());
172 buf[5] = 0xC0 | ((self.version_number & 0x1F) << 1)
174 | u8::from(self.current_next_indicator);
175 buf[6] = self.section_number;
176 buf[7] = self.last_section_number;
177
178 let data_start = HEADER_LEN + EXTENSION_HEADER_LEN;
180 let data_end = data_start + self.container_data.len();
181 buf[data_start..data_end].copy_from_slice(self.container_data);
182
183 let crc = dvb_common::crc32_mpeg2::compute(&buf[..data_end]);
185 buf[data_end..len].copy_from_slice(&crc.to_be_bytes());
186
187 Ok(len)
188 }
189}
190
191impl<'a> Table<'a> for Container<'a> {
192 const TABLE_ID: u8 = TABLE_ID;
193 const PID: u16 = PID;
194}
195
196#[cfg(test)]
197mod tests {
198 use super::*;
199
200 fn build_container(
203 container_id: u16,
204 version: u8,
205 current_next: bool,
206 section_number: u8,
207 last_section_number: u8,
208 container_data: &[u8],
209 ) -> Vec<u8> {
210 let c = Container {
211 private_indicator: true,
212 container_id,
213 version_number: version,
214 current_next_indicator: current_next,
215 section_number,
216 last_section_number,
217 container_data,
218 };
219 let mut buf = vec![0u8; c.serialized_len()];
220 c.serialize_into(&mut buf).unwrap();
221 buf
222 }
223
224 #[test]
225 fn parse_happy_path() {
226 let data = [0x00u8, 0xDE, 0xAD, 0xBE, 0xEF];
227 let bytes = build_container(0x1234, 7, true, 1, 3, &data);
228 let c = Container::parse(&bytes).unwrap();
229 assert!(c.private_indicator);
230 assert_eq!(c.container_id, 0x1234);
231 assert_eq!(c.version_number, 7);
232 assert!(c.current_next_indicator);
233 assert_eq!(c.section_number, 1);
234 assert_eq!(c.last_section_number, 3);
235 assert_eq!(c.container_data, &data[..]);
236 }
237
238 #[test]
239 fn parse_empty_container_data() {
240 let bytes = build_container(0x0000, 0, false, 0, 0, &[]);
241 let c = Container::parse(&bytes).unwrap();
242 assert_eq!(c.container_id, 0x0000);
243 assert_eq!(c.version_number, 0);
244 assert!(!c.current_next_indicator);
245 assert!(c.container_data.is_empty());
246 }
247
248 #[test]
249 fn parse_rejects_wrong_tag() {
250 let mut bytes = build_container(0x0001, 0, true, 0, 0, &[]);
251 bytes[0] = 0x70; assert!(matches!(
253 Container::parse(&bytes).unwrap_err(),
254 Error::UnexpectedTableId { table_id: 0x70, .. }
255 ));
256 }
257
258 #[test]
259 fn parse_rejects_short_buffer() {
260 assert!(matches!(
261 Container::parse(&[0x75, 0x80]).unwrap_err(),
262 Error::BufferTooShort { .. }
263 ));
264 }
265
266 #[test]
267 fn parse_rejects_section_length_overflow() {
268 let mut bytes = build_container(0x0001, 0, true, 0, 0, &[]);
269 let fake_sl: u16 = (bytes.len() as u16) + 100 - HEADER_LEN as u16;
270 bytes[1] = (bytes[1] & 0xF0) | ((fake_sl >> 8) as u8 & 0x0F);
271 bytes[2] = (fake_sl & 0xFF) as u8;
272 assert!(matches!(
273 Container::parse(&bytes).unwrap_err(),
274 Error::SectionLengthOverflow { .. }
275 ));
276 }
277
278 #[test]
279 fn serialize_round_trip() {
280 let data = [0x01u8, 0x02, 0x03];
281 let original = Container {
282 private_indicator: false,
283 container_id: 0xABCD,
284 version_number: 15,
285 current_next_indicator: false,
286 section_number: 2,
287 last_section_number: 4,
288 container_data: &data,
289 };
290 let mut buf = vec![0u8; original.serialized_len()];
291 original.serialize_into(&mut buf).unwrap();
292 assert_eq!(Container::parse(&buf).unwrap(), original);
293 }
294
295 #[test]
296 fn serialize_rejects_output_buffer_too_small() {
297 let c = Container {
298 private_indicator: false,
299 container_id: 0x0001,
300 version_number: 0,
301 current_next_indicator: true,
302 section_number: 0,
303 last_section_number: 0,
304 container_data: &[],
305 };
306 let mut buf = vec![0u8; 2];
307 assert!(matches!(
308 c.serialize_into(&mut buf).unwrap_err(),
309 Error::OutputBufferTooSmall { .. }
310 ));
311 }
312
313 #[test]
314 fn table_trait_constants() {
315 assert_eq!(<Container as Table>::TABLE_ID, 0x75);
316 assert_eq!(<Container as Table>::PID, 0x0000);
317 }
318
319 #[cfg(feature = "serde")]
320 #[test]
321 fn serde_json_round_trip() {
322 let data = [0xCAu8, 0xFE];
327 let bytes = build_container(0xBEEF, 9, true, 0, 0, &data);
328 let c = Container::parse(&bytes).unwrap();
329 let v: serde_json::Value = serde_json::to_value(&c).unwrap();
330 assert_eq!(v["container_id"], 0xBEEF);
331 assert_eq!(v["version_number"], 9);
332 assert_eq!(v["current_next_indicator"], true);
333 assert_eq!(v["container_data"], serde_json::json!([0xCA, 0xFE]));
334 }
335}