1use crate::descriptors::DescriptorLoop;
11use crate::error::{Error, Result};
12use crate::traits::Table;
13use dvb_common::{Parse, Serialize};
14
15pub const TABLE_ID_PF_ACTUAL: u8 = 0x4E;
17pub const TABLE_ID_PF_OTHER: u8 = 0x4F;
19pub const TABLE_ID_SCHEDULE_ACTUAL_FIRST: u8 = 0x50;
21pub const TABLE_ID_SCHEDULE_ACTUAL_LAST: u8 = 0x5F;
23pub const TABLE_ID_SCHEDULE_OTHER_FIRST: u8 = 0x60;
25pub const TABLE_ID_SCHEDULE_OTHER_LAST: u8 = 0x6F;
27pub const PID: u16 = 0x0012;
29
30const MIN_HEADER_LEN: usize = 3;
31const EXTENSION_HEADER_LEN: usize = 5;
32const POST_EXTENSION_LEN: usize = 6;
35const CRC_LEN: usize = 4;
36const EVENT_HEADER_LEN: usize = 12;
37
38#[derive(Debug, Clone, Copy, PartialEq, Eq)]
40#[cfg_attr(feature = "serde", derive(serde::Serialize))]
41pub enum EitKind {
42 PresentFollowingActual,
44 PresentFollowingOther,
46 ScheduleActual,
48 ScheduleOther,
50}
51
52impl EitKind {
53 #[must_use]
55 pub fn from_table_id(table_id: u8) -> Option<Self> {
56 match table_id {
57 TABLE_ID_PF_ACTUAL => Some(Self::PresentFollowingActual),
58 TABLE_ID_PF_OTHER => Some(Self::PresentFollowingOther),
59 TABLE_ID_SCHEDULE_ACTUAL_FIRST..=TABLE_ID_SCHEDULE_ACTUAL_LAST => {
60 Some(Self::ScheduleActual)
61 }
62 TABLE_ID_SCHEDULE_OTHER_FIRST..=TABLE_ID_SCHEDULE_OTHER_LAST => {
63 Some(Self::ScheduleOther)
64 }
65 _ => None,
66 }
67 }
68}
69
70#[derive(Debug, Clone, PartialEq, Eq)]
72#[cfg_attr(feature = "serde", derive(serde::Serialize))]
73#[cfg_attr(feature = "yoke", derive(yoke::Yokeable))]
74pub struct EitEvent<'a> {
75 pub event_id: u16,
77 pub start_time_raw: [u8; 5],
79 pub duration_raw: [u8; 3],
81 pub running_status: u8,
83 pub free_ca_mode: bool,
85 pub descriptors: DescriptorLoop<'a>,
88}
89
90#[derive(Debug, Clone, PartialEq, Eq)]
92#[cfg_attr(feature = "serde", derive(serde::Serialize))]
93#[cfg_attr(feature = "yoke", derive(yoke::Yokeable))]
94pub struct Eit<'a> {
95 pub kind: EitKind,
97 pub table_id: u8,
99 pub service_id: u16,
101 pub version_number: u8,
103 pub current_next_indicator: bool,
105 pub section_number: u8,
107 pub last_section_number: u8,
109 pub transport_stream_id: u16,
111 pub original_network_id: u16,
113 pub segment_last_section_number: u8,
115 pub last_table_id: u8,
117 pub events: Vec<EitEvent<'a>>,
119}
120
121impl<'a> Parse<'a> for Eit<'a> {
122 type Error = crate::error::Error;
123 fn parse(bytes: &'a [u8]) -> Result<Self> {
124 let min_len = MIN_HEADER_LEN + EXTENSION_HEADER_LEN + POST_EXTENSION_LEN + CRC_LEN;
125 if bytes.len() < min_len {
126 return Err(Error::BufferTooShort {
127 need: min_len,
128 have: bytes.len(),
129 what: "Eit",
130 });
131 }
132
133 let table_id = bytes[0];
134 let kind = EitKind::from_table_id(table_id).ok_or(Error::UnexpectedTableId {
135 table_id,
136 what: "Eit",
137 expected: &[
138 TABLE_ID_PF_ACTUAL,
139 TABLE_ID_PF_OTHER,
140 TABLE_ID_SCHEDULE_ACTUAL_FIRST,
141 TABLE_ID_SCHEDULE_OTHER_FIRST,
142 ],
143 })?;
144
145 let section_length = ((bytes[1] & 0x0F) as u16) << 8 | bytes[2] as u16;
146 let total = MIN_HEADER_LEN + section_length as usize;
147 if bytes.len() < total {
148 return Err(Error::SectionLengthOverflow {
149 declared: section_length as usize,
150 available: bytes.len() - MIN_HEADER_LEN,
151 });
152 }
153
154 let service_id = u16::from_be_bytes([bytes[3], bytes[4]]);
155 let version_number = (bytes[5] >> 1) & 0x1F;
156 let current_next_indicator = (bytes[5] & 0x01) != 0;
157 let section_number = bytes[6];
158 let last_section_number = bytes[7];
159
160 let transport_stream_id = u16::from_be_bytes([bytes[8], bytes[9]]);
161 let original_network_id = u16::from_be_bytes([bytes[10], bytes[11]]);
162 let segment_last_section_number = bytes[12];
163 let last_table_id = bytes[13];
164
165 let events_start = MIN_HEADER_LEN + EXTENSION_HEADER_LEN + POST_EXTENSION_LEN;
166 let events_end = total - CRC_LEN;
167 let mut events = Vec::new();
168 let mut pos = events_start;
169 while pos + EVENT_HEADER_LEN <= events_end {
170 let event_id = u16::from_be_bytes([bytes[pos], bytes[pos + 1]]);
171 let start_time_raw = [
172 bytes[pos + 2],
173 bytes[pos + 3],
174 bytes[pos + 4],
175 bytes[pos + 5],
176 bytes[pos + 6],
177 ];
178 let duration_raw = [bytes[pos + 7], bytes[pos + 8], bytes[pos + 9]];
179 let status_and_len_hi = bytes[pos + 10];
180 let running_status = (status_and_len_hi >> 5) & 0x07;
181 let free_ca_mode = (status_and_len_hi & 0x10) != 0;
182 let descriptors_loop_length =
183 (((status_and_len_hi & 0x0F) as usize) << 8) | bytes[pos + 11] as usize;
184 let desc_start = pos + EVENT_HEADER_LEN;
185 let desc_end = desc_start + descriptors_loop_length;
186 if desc_end > events_end {
187 return Err(Error::SectionLengthOverflow {
188 declared: descriptors_loop_length,
189 available: events_end - desc_start,
190 });
191 }
192 events.push(EitEvent {
193 event_id,
194 start_time_raw,
195 duration_raw,
196 running_status,
197 free_ca_mode,
198 descriptors: DescriptorLoop::new(&bytes[desc_start..desc_end]),
199 });
200 pos = desc_end;
201 }
202
203 Ok(Eit {
204 kind,
205 table_id,
206 service_id,
207 version_number,
208 current_next_indicator,
209 section_number,
210 last_section_number,
211 transport_stream_id,
212 original_network_id,
213 segment_last_section_number,
214 last_table_id,
215 events,
216 })
217 }
218}
219
220impl Serialize for Eit<'_> {
221 type Error = crate::error::Error;
222 fn serialized_len(&self) -> usize {
223 let ev_bytes: usize = self
224 .events
225 .iter()
226 .map(|e| EVENT_HEADER_LEN + e.descriptors.len())
227 .sum();
228 MIN_HEADER_LEN + EXTENSION_HEADER_LEN + POST_EXTENSION_LEN + ev_bytes + CRC_LEN
229 }
230
231 fn serialize_into(&self, buf: &mut [u8]) -> Result<usize> {
232 let len = self.serialized_len();
233 if buf.len() < len {
234 return Err(Error::OutputBufferTooSmall {
235 need: len,
236 have: buf.len(),
237 });
238 }
239 let section_length: u16 = (len - MIN_HEADER_LEN) as u16;
240 buf[0] = self.table_id;
241 buf[1] = 0xB0 | ((section_length >> 8) as u8 & 0x0F);
242 buf[2] = (section_length & 0xFF) as u8;
243 buf[3..5].copy_from_slice(&self.service_id.to_be_bytes());
244 buf[5] = 0xC0 | ((self.version_number & 0x1F) << 1) | u8::from(self.current_next_indicator);
245 buf[6] = self.section_number;
246 buf[7] = self.last_section_number;
247 buf[8..10].copy_from_slice(&self.transport_stream_id.to_be_bytes());
248 buf[10..12].copy_from_slice(&self.original_network_id.to_be_bytes());
249 buf[12] = self.segment_last_section_number;
250 buf[13] = self.last_table_id;
251
252 let mut pos = MIN_HEADER_LEN + EXTENSION_HEADER_LEN + POST_EXTENSION_LEN;
253 for ev in &self.events {
254 buf[pos..pos + 2].copy_from_slice(&ev.event_id.to_be_bytes());
255 buf[pos + 2..pos + 7].copy_from_slice(&ev.start_time_raw);
256 buf[pos + 7..pos + 10].copy_from_slice(&ev.duration_raw);
257 let dll = ev.descriptors.len() as u16;
258 buf[pos + 10] = ((ev.running_status & 0x07) << 5)
259 | (u8::from(ev.free_ca_mode) << 4)
260 | ((dll >> 8) as u8 & 0x0F);
261 buf[pos + 11] = (dll & 0xFF) as u8;
262 let desc_start = pos + EVENT_HEADER_LEN;
263 buf[desc_start..desc_start + ev.descriptors.len()]
264 .copy_from_slice(ev.descriptors.raw());
265 pos = desc_start + ev.descriptors.len();
266 }
267
268 let crc_pos = len - CRC_LEN;
269 let crc = dvb_common::crc32_mpeg2::compute(&buf[..crc_pos]);
270 buf[crc_pos..len].copy_from_slice(&crc.to_be_bytes());
271 Ok(len)
272 }
273}
274
275impl<'a> Table<'a> for Eit<'a> {
276 const TABLE_ID: u8 = TABLE_ID_PF_ACTUAL;
277 const PID: u16 = PID;
278}
279
280impl<'a> crate::traits::TableDef<'a> for Eit<'a> {
281 const TABLE_ID_RANGES: &'static [(u8, u8)] =
282 &[(TABLE_ID_PF_ACTUAL, TABLE_ID_SCHEDULE_OTHER_LAST)];
283 const NAME: &'static str = "EVENT_INFORMATION";
284}
285
286#[cfg(feature = "chrono")]
287impl EitEvent<'_> {
288 #[must_use]
293 pub fn start_time(&self) -> Option<chrono::DateTime<chrono::Utc>> {
294 use chrono::{NaiveDate, NaiveDateTime, TimeZone};
295 let mjd = u16::from_be_bytes([self.start_time_raw[0], self.start_time_raw[1]]);
296 let (y, m, d) = mjd_to_ymd(mjd);
297 let h = bcd_byte(self.start_time_raw[2])?;
298 let mi = bcd_byte(self.start_time_raw[3])?;
299 let s = bcd_byte(self.start_time_raw[4])?;
300 let date = NaiveDate::from_ymd_opt(y, m, d)?;
301 let time = chrono::NaiveTime::from_hms_opt(u32::from(h), u32::from(mi), u32::from(s))?;
302 let naive = NaiveDateTime::new(date, time);
303 chrono::Utc.from_local_datetime(&naive).single()
304 }
305}
306
307#[cfg(feature = "chrono")]
308fn bcd_byte(b: u8) -> Option<u8> {
309 let hi = b >> 4;
310 let lo = b & 0x0F;
311 if hi > 9 || lo > 9 {
312 return None;
313 }
314 Some(hi * 10 + lo)
315}
316
317#[cfg(feature = "chrono")]
318fn mjd_to_ymd(mjd: u16) -> (i32, u32, u32) {
319 let mjd = i64::from(mjd);
321 let y_prime = ((mjd as f64 - 15_078.2) / 365.25) as i64;
322 let m_prime = ((mjd as f64 - 14_956.1 - (y_prime as f64 * 365.25).floor()) / 30.6001) as i64;
323 let d = mjd
324 - 14_956
325 - (y_prime as f64 * 365.25).floor() as i64
326 - (m_prime as f64 * 30.6001).floor() as i64;
327 let k = if m_prime == 14 || m_prime == 15 { 1 } else { 0 };
328 let y = y_prime + k + 1900;
329 let m = m_prime - 1 - k * 12;
330 (y as i32, m as u32, d as u32)
331}
332
333#[cfg(test)]
334mod tests {
335 use super::*;
336
337 type TestEvent = (u16, [u8; 5], [u8; 3], u8, bool, Vec<u8>);
338
339 fn build_eit(
340 table_id: u8,
341 service_id: u16,
342 version: u8,
343 tsid: u16,
344 onid: u16,
345 events: &[TestEvent],
346 ) -> Vec<u8> {
347 let ev_bytes: usize = events
348 .iter()
349 .map(|(_, _, _, _, _, d)| EVENT_HEADER_LEN + d.len())
350 .sum();
351 let section_length: u16 =
352 (EXTENSION_HEADER_LEN + POST_EXTENSION_LEN + ev_bytes + CRC_LEN) as u16;
353 let mut v = Vec::new();
354 v.push(table_id);
355 v.push(0xB0 | ((section_length >> 8) as u8 & 0x0F));
356 v.push((section_length & 0xFF) as u8);
357 v.extend_from_slice(&service_id.to_be_bytes());
358 v.push(0xC0 | ((version & 0x1F) << 1) | 0x01);
359 v.push(0);
360 v.push(0);
361 v.extend_from_slice(&tsid.to_be_bytes());
362 v.extend_from_slice(&onid.to_be_bytes());
363 v.push(0);
364 v.push(table_id);
365 for (eid, start, dur, rs, fca, desc) in events {
366 v.extend_from_slice(&eid.to_be_bytes());
367 v.extend_from_slice(start);
368 v.extend_from_slice(dur);
369 let dll = desc.len() as u16;
370 v.push(((*rs & 0x07) << 5) | (u8::from(*fca) << 4) | ((dll >> 8) as u8 & 0x0F));
371 v.push((dll & 0xFF) as u8);
372 v.extend_from_slice(desc);
373 }
374 v.extend_from_slice(&[0, 0, 0, 0]);
375 v
376 }
377
378 #[test]
379 fn parse_pf_actual_and_other_map_to_correct_kind() {
380 for (tid, expected) in [
381 (TABLE_ID_PF_ACTUAL, EitKind::PresentFollowingActual),
382 (TABLE_ID_PF_OTHER, EitKind::PresentFollowingOther),
383 ] {
384 let bytes = build_eit(tid, 1, 0, 0x20, 0x30, &[]);
385 assert_eq!(Eit::parse(&bytes).unwrap().kind, expected);
386 }
387 }
388
389 #[test]
390 fn schedule_tables_0x50_through_0x5f_all_decode_as_schedule_actual() {
391 for tid in TABLE_ID_SCHEDULE_ACTUAL_FIRST..=TABLE_ID_SCHEDULE_ACTUAL_LAST {
392 let bytes = build_eit(tid, 1, 0, 0x20, 0x30, &[]);
393 assert_eq!(Eit::parse(&bytes).unwrap().kind, EitKind::ScheduleActual);
394 }
395 }
396
397 #[test]
398 fn schedule_tables_0x60_through_0x6f_all_decode_as_schedule_other() {
399 for tid in TABLE_ID_SCHEDULE_OTHER_FIRST..=TABLE_ID_SCHEDULE_OTHER_LAST {
400 let bytes = build_eit(tid, 1, 0, 0x20, 0x30, &[]);
401 assert_eq!(Eit::parse(&bytes).unwrap().kind, EitKind::ScheduleOther);
402 }
403 }
404
405 #[test]
406 fn event_loop_with_descriptor_bytes_preserved() {
407 let desc = vec![0x4D, 0x05, 0x01, 0x02, 0x03, 0x04, 0x05];
408 let bytes = build_eit(
409 TABLE_ID_PF_ACTUAL,
410 1,
411 0,
412 0x20,
413 0x30,
414 &[(
415 42,
416 [0xDF, 0xA1, 0x12, 0x34, 0x56],
417 [0x00, 0x30, 0x00],
418 4,
419 false,
420 desc.clone(),
421 )],
422 );
423 let eit = Eit::parse(&bytes).unwrap();
424 assert_eq!(eit.events.len(), 1);
425 assert_eq!(eit.events[0].event_id, 42);
426 assert_eq!(eit.events[0].descriptors.raw(), &desc[..]);
427 }
428
429 #[test]
430 fn running_status_extracted() {
431 let bytes = build_eit(
432 TABLE_ID_PF_ACTUAL,
433 1,
434 0,
435 0x20,
436 0x30,
437 &[(1, [0; 5], [0; 3], 2, false, vec![])],
438 );
439 assert_eq!(Eit::parse(&bytes).unwrap().events[0].running_status, 2);
440 }
441
442 #[test]
443 fn free_ca_mode_flag_extracted() {
444 let bytes = build_eit(
445 TABLE_ID_PF_ACTUAL,
446 1,
447 0,
448 0x20,
449 0x30,
450 &[(1, [0; 5], [0; 3], 0, true, vec![])],
451 );
452 assert!(Eit::parse(&bytes).unwrap().events[0].free_ca_mode);
453 }
454
455 #[test]
456 fn serialize_round_trip_preserves_all_events() {
457 let desc1: [u8; 2] = [0x54, 0x00];
458 let eit = Eit {
459 kind: EitKind::PresentFollowingActual,
460 table_id: TABLE_ID_PF_ACTUAL,
461 service_id: 0x0100,
462 version_number: 3,
463 current_next_indicator: true,
464 section_number: 0,
465 last_section_number: 0,
466 transport_stream_id: 0x1234,
467 original_network_id: 0x0020,
468 segment_last_section_number: 0,
469 last_table_id: TABLE_ID_PF_ACTUAL,
470 events: vec![
471 EitEvent {
472 event_id: 1,
473 start_time_raw: [0xDF, 0xA1, 0x12, 0x34, 0x56],
474 duration_raw: [0x00, 0x30, 0x00],
475 running_status: 4,
476 free_ca_mode: false,
477 descriptors: DescriptorLoop::new(&desc1),
478 },
479 EitEvent {
480 event_id: 2,
481 start_time_raw: [0xDF, 0xA1, 0x13, 0x00, 0x00],
482 duration_raw: [0x01, 0x00, 0x00],
483 running_status: 1,
484 free_ca_mode: true,
485 descriptors: DescriptorLoop::new(&[]),
486 },
487 ],
488 };
489 let mut buf = vec![0u8; eit.serialized_len()];
490 eit.serialize_into(&mut buf).unwrap();
491 let re = Eit::parse(&buf).unwrap();
492 assert_eq!(eit, re);
493 }
494
495 #[test]
496 fn zero_events_is_valid() {
497 let bytes = build_eit(TABLE_ID_PF_ACTUAL, 1, 0, 0x20, 0x30, &[]);
498 let eit = Eit::parse(&bytes).unwrap();
499 assert_eq!(eit.events.len(), 0);
500 }
501
502 #[test]
503 #[cfg(feature = "chrono")]
504 fn event_start_time_decodes_to_utc_datetime() {
505 let mjd: u16 = 59945;
507 let ev = EitEvent {
508 event_id: 1,
509 start_time_raw: [(mjd >> 8) as u8, (mjd & 0xFF) as u8, 0x12, 0x34, 0x56],
510 duration_raw: [0, 0, 0],
511 running_status: 0,
512 free_ca_mode: false,
513 descriptors: DescriptorLoop::new(&[]),
514 };
515 let dt = ev.start_time().unwrap();
516 use chrono::Datelike;
517 assert_eq!(dt.year(), 2023);
518 assert_eq!(dt.month(), 1);
519 assert_eq!(dt.day(), 1);
520 use chrono::Timelike;
521 assert_eq!(dt.hour(), 12);
522 assert_eq!(dt.minute(), 34);
523 assert_eq!(dt.second(), 56);
524 }
525}