Skip to main content

dvb_si/descriptors/
any.rs

1//! Unified descriptor dispatch: [`AnyDescriptor`] + [`parse_loop`].
2//!
3//! [`AnyDescriptor`] is generated from a single declarative list
4//! (`declare_descriptors!`) — one line per crate-implemented descriptor tag.
5//! The list is the single source of truth: it produces the enum, the
6//! `From<T>` conversions, and the tag → type dispatcher, and a drift test
7//! pins each tag literal to the type's [`crate::traits::DescriptorDef::TAG`].
8//!
9//! [`parse_loop`] lazily walks a raw descriptor loop (the variable-length
10//! `descriptor()` sequence inside tables), yielding one [`AnyDescriptor`] per
11//! entry. It never panics: a malformed entry whose length is known yields an
12//! `Err` and iteration continues; a truncated final header/body yields one
13//! final `Err` and then fuses.
14//!
15//! ```
16//! use dvb_si::descriptors::{parse_loop, AnyDescriptor};
17//!
18//! // A two-descriptor loop: short_event (tag 0x4D, "eng" / "News") then an
19//! // unrecognised private tag 0xA7 — the walker yields a typed value for the
20//! // first and `Unknown` for the second, never panicking.
21//! let loop_bytes = [
22//!     0x4D, 0x07, b'e', b'n', b'g', 0x02, b'H', b'i', 0x00, // short_event
23//!     0xA7, 0x02, 0xCA, 0xFE,                               // unknown 0xA7
24//! ];
25//! let items: Vec<_> = parse_loop(&loop_bytes).collect();
26//! assert_eq!(items.len(), 2);
27//! match items[0].as_ref().unwrap() {
28//!     AnyDescriptor::ShortEvent(se) => {
29//!         assert_eq!(se.language_code.as_str(), "eng");
30//!         assert_eq!(se.event_name.decode(), "Hi");
31//!     }
32//!     other => panic!("expected ShortEvent, got {other:?}"),
33//! }
34//! assert!(matches!(items[1].as_ref().unwrap(), AnyDescriptor::Unknown { tag: 0xA7, .. }));
35//! ```
36//!
37//! # Adding a descriptor
38//!
39//! 1. Create the module with the wire layout, a `pub const TAG: u8`, and the
40//!    symmetric [`dvb_common::Parse`]/[`dvb_common::Serialize`] impls +
41//!    round-trip tests (copy an existing module).
42//! 2. `impl DescriptorDef` for the type (`TAG` from the module const, `NAME`
43//!    in SCREAMING_SNAKE without the `_descriptor` suffix).
44//! 3. Add one line to the `declare_descriptors!` invocation below — the enum
45//!    variant, dispatcher arm, and drift test are generated from it.
46//! 4. The integration completeness test walks the generated
47//!    [`AnyDescriptor::DISPATCHED_TAGS`] automatically — no test edits needed.
48
49use alloc::boxed::Box;
50#[cfg(feature = "serde")]
51use alloc::string::ToString;
52
53/// Declares [`AnyDescriptor`] + its dispatcher from one tag list.
54///
55/// Each line is `Variant = 0xTAG => module::Type[<'a>]`. The optional trailing
56/// `@no_dispatch …` section adds variants that are NOT reachable from the
57/// generated dispatcher (private / context-dependent tags such as 0x83
58/// logical_channel) — the variant exists for callers that opt in via the
59/// registry, but `dispatch` never produces it.
60macro_rules! declare_descriptors {
61    (
62        $lt:lifetime;
63        $( $variant:ident = $tag:literal => $($path:ident)::+ $(<$plt:lifetime>)? ),+ $(,)?
64        $( ; @no_dispatch $( $nd_variant:ident => $($nd_path:ident)::+ $(<$nd_plt:lifetime>)? ),+ $(,)? )?
65    ) => {
66        /// Every crate-implemented descriptor, plus an `Unknown` fallthrough.
67        ///
68        /// serde uses external tagging with camelCase variant keys —
69        /// a parsed short_event_descriptor serializes as `{"shortEvent": {…}}`.
70        /// Variant names map 1:1 to the descriptor modules; see each module
71        /// for the wire layout.
72        #[derive(Debug)]
73        #[cfg_attr(feature = "serde", derive(serde::Serialize))]
74        #[cfg_attr(feature = "serde", serde(rename_all = "camelCase"))]
75        // Every variant is covariant in `$lt`: typed variants hold only
76        // lifetime-parametrised views, `Unknown` holds `&$lt [u8]`, and the
77        // `Other` value is a `'static` `Box<dyn DescriptorObject>`. The derive
78        // accepts the `'static` field unchanged, so the impl is sound.
79        #[cfg_attr(feature = "yoke", derive(yoke::Yokeable))]
80        #[non_exhaustive]
81        pub enum AnyDescriptor<$lt> {
82            $(
83                #[allow(missing_docs)]
84                $variant($($path)::+ $(<$plt>)?),
85            )+
86            $($(
87                #[allow(missing_docs)]
88                $nd_variant($($nd_path)::+ $(<$nd_plt>)?),
89            )+)?
90            /// Runtime-registered custom descriptor (see [`DescriptorRegistry`]).
91            ///
92            /// [`DescriptorRegistry`]: crate::descriptors::registry::DescriptorRegistry
93            Other {
94                /// The raw descriptor_tag byte.
95                tag: u8,
96                /// The parsed, type-erased descriptor value. Call `downcast_ref`
97                /// on it (see [`DescriptorObject`](crate::descriptors::registry::DescriptorObject))
98                /// to recover the concrete type.
99                #[cfg_attr(
100                    feature = "serde",
101                    serde(serialize_with = "crate::descriptors::registry::serialize_erased")
102                )]
103                value: Box<dyn crate::descriptors::registry::DescriptorObject>,
104            },
105            /// Tag with no typed implementation; `body` is the payload sans
106            /// the 2-byte (tag, length) header.
107            Unknown {
108                /// The raw descriptor_tag byte.
109                tag: u8,
110                /// The raw payload bytes (descriptor_length bytes).
111                body: &$lt [u8],
112            },
113        }
114
115        $(
116            impl<$lt> From<$($path)::+ $(<$plt>)?> for AnyDescriptor<$lt> {
117                fn from(d: $($path)::+ $(<$plt>)?) -> Self {
118                    Self::$variant(d)
119                }
120            }
121        )+
122        $($(
123            impl<$lt> From<$($nd_path)::+ $(<$nd_plt>)?> for AnyDescriptor<$lt> {
124                fn from(d: $($nd_path)::+ $(<$nd_plt>)?) -> Self {
125                    Self::$nd_variant(d)
126                }
127            }
128        )+)?
129
130        impl<$lt> AnyDescriptor<$lt> {
131            /// Every tag the generated dispatcher routes (excludes `@no_dispatch`
132            /// variants and [`AnyDescriptor::Unknown`]).
133            pub const DISPATCHED_TAGS: &'static [u8] = &[$($tag),+];
134
135            /// Diagnostic name of the contained descriptor — the type's
136            /// [`DescriptorDef::NAME`](crate::traits::DescriptorDef::NAME)
137            /// (`"SHORT_EVENT"`, `"NETWORK_NAME"`, …); `"CUSTOM"` for
138            /// [`AnyDescriptor::Other`] (runtime-registered) and `"UNKNOWN"`
139            /// for [`AnyDescriptor::Unknown`].
140            #[must_use]
141            pub fn name(&self) -> &'static str {
142                match self {
143                    $(
144                        Self::$variant(_) =>
145                            <$($path)::+ as crate::traits::DescriptorDef>::NAME,
146                    )+
147                    $($(
148                        Self::$nd_variant(_) =>
149                            <$($nd_path)::+ as crate::traits::DescriptorDef>::NAME,
150                    )+)?
151                    Self::Other { .. } => "CUSTOM",
152                    Self::Unknown { .. } => "UNKNOWN",
153                }
154            }
155
156            /// Parse one full descriptor (2-byte header included) by its tag.
157            ///
158            /// `None` means no typed implementation exists for `tag` (the
159            /// caller turns that into [`AnyDescriptor::Unknown`]). `Some(Err)`
160            /// is a typed parse failure for a recognised tag.
161            pub(crate) fn dispatch(tag: u8, full: &$lt [u8]) -> Option<crate::Result<Self>> {
162                use dvb_common::Parse;
163                match tag {
164                    $(
165                        $tag => Some(<$($path)::+>::parse(full).map(Self::$variant)),
166                    )+
167                    _ => None,
168                }
169            }
170        }
171
172        #[cfg(test)]
173        mod macro_drift {
174            #[test]
175            fn tag_literals_match_descriptor_def() {
176                use crate::traits::DescriptorDef;
177                $(
178                    assert_eq!(
179                        $tag,
180                        <$($path)::+ as DescriptorDef>::TAG,
181                        concat!("tag literal drift for ", stringify!($variant)),
182                    );
183                    assert!(
184                        !<$($path)::+ as DescriptorDef>::NAME.is_empty(),
185                        concat!("empty NAME for ", stringify!($variant)),
186                    );
187                )+
188                $($(
189                    assert!(
190                        !<$($nd_path)::+ as DescriptorDef>::NAME.is_empty(),
191                        concat!("empty NAME for ", stringify!($nd_variant)),
192                    );
193                )+)?
194            }
195        }
196    };
197}
198
199declare_descriptors! {'a;
200    // MPEG-2 systems descriptors (ISO/IEC 13818-1) used outside table context.
201    Registration = 0x05 => crate::descriptors::registration::RegistrationDescriptor<'a>,
202    DataStreamAlignment = 0x06 => crate::descriptors::data_stream_alignment::DataStreamAlignmentDescriptor,
203    Ca = 0x09 => crate::descriptors::ca::CaDescriptor<'a>,
204    Iso639Language = 0x0A => crate::descriptors::iso_639_language::Iso639LanguageDescriptor,
205    PrivateDataIndicator = 0x0F => crate::descriptors::private_data_indicator::PrivateDataIndicatorDescriptor,
206    CarouselIdentifier = 0x13 => crate::descriptors::carousel_identifier::CarouselIdentifierDescriptor<'a>,
207    // DVB descriptors (ETSI EN 300 468) — contiguous 0x40..=0x7F.
208    NetworkName = 0x40 => crate::descriptors::network_name::NetworkNameDescriptor<'a>,
209    ServiceList = 0x41 => crate::descriptors::service_list::ServiceListDescriptor,
210    Stuffing = 0x42 => crate::descriptors::stuffing::StuffingDescriptor<'a>,
211    SatelliteDeliverySystem = 0x43 => crate::descriptors::satellite_delivery_system::SatelliteDeliverySystemDescriptor,
212    CableDeliverySystem = 0x44 => crate::descriptors::cable_delivery_system::CableDeliverySystemDescriptor,
213    VbiData = 0x45 => crate::descriptors::vbi_data::VbiDataDescriptor<'a>,
214    VbiTeletext = 0x46 => crate::descriptors::vbi_teletext::VbiTeletextDescriptor,
215    BouquetName = 0x47 => crate::descriptors::bouquet_name::BouquetNameDescriptor<'a>,
216    Service = 0x48 => crate::descriptors::service::ServiceDescriptor<'a>,
217    CountryAvailability = 0x49 => crate::descriptors::country_availability::CountryAvailabilityDescriptor,
218    Linkage = 0x4A => crate::descriptors::linkage::LinkageDescriptor<'a>,
219    NvodReference = 0x4B => crate::descriptors::nvod_reference::NvodReferenceDescriptor,
220    TimeShiftedService = 0x4C => crate::descriptors::time_shifted_service::TimeShiftedServiceDescriptor,
221    ShortEvent = 0x4D => crate::descriptors::short_event::ShortEventDescriptor<'a>,
222    ExtendedEvent = 0x4E => crate::descriptors::extended_event::ExtendedEventDescriptor<'a>,
223    TimeShiftedEvent = 0x4F => crate::descriptors::time_shifted_event::TimeShiftedEventDescriptor,
224    Component = 0x50 => crate::descriptors::component::ComponentDescriptor<'a>,
225    Mosaic = 0x51 => crate::descriptors::mosaic::MosaicDescriptor,
226    StreamIdentifier = 0x52 => crate::descriptors::stream_identifier::StreamIdentifierDescriptor,
227    CaIdentifier = 0x53 => crate::descriptors::ca_identifier::CaIdentifierDescriptor,
228    Content = 0x54 => crate::descriptors::content::ContentDescriptor,
229    ParentalRating = 0x55 => crate::descriptors::parental_rating::ParentalRatingDescriptor,
230    Teletext = 0x56 => crate::descriptors::teletext::TeletextDescriptor,
231    Telephone = 0x57 => crate::descriptors::telephone::TelephoneDescriptor<'a>,
232    LocalTimeOffset = 0x58 => crate::descriptors::local_time_offset::LocalTimeOffsetDescriptor,
233    Subtitling = 0x59 => crate::descriptors::subtitling::SubtitlingDescriptor,
234    TerrestrialDeliverySystem = 0x5A => crate::descriptors::terrestrial_delivery_system::TerrestrialDeliverySystemDescriptor,
235    MultilingualNetworkName = 0x5B => crate::descriptors::multilingual_network_name::MultilingualNetworkNameDescriptor<'a>,
236    MultilingualBouquetName = 0x5C => crate::descriptors::multilingual_bouquet_name::MultilingualBouquetNameDescriptor<'a>,
237    MultilingualServiceName = 0x5D => crate::descriptors::multilingual_service_name::MultilingualServiceNameDescriptor<'a>,
238    MultilingualComponent = 0x5E => crate::descriptors::multilingual_component::MultilingualComponentDescriptor<'a>,
239    PrivateDataSpecifier = 0x5F => crate::descriptors::private_data_specifier::PrivateDataSpecifierDescriptor,
240    ServiceMove = 0x60 => crate::descriptors::service_move::ServiceMoveDescriptor,
241    ShortSmoothingBuffer = 0x61 => crate::descriptors::short_smoothing_buffer::ShortSmoothingBufferDescriptor<'a>,
242    FrequencyList = 0x62 => crate::descriptors::frequency_list::FrequencyListDescriptor,
243    PartialTransportStream = 0x63 => crate::descriptors::partial_transport_stream::PartialTransportStreamDescriptor,
244    DataBroadcast = 0x64 => crate::descriptors::data_broadcast::DataBroadcastDescriptor<'a>,
245    Scrambling = 0x65 => crate::descriptors::scrambling::ScramblingDescriptor,
246    DataBroadcastId = 0x66 => crate::descriptors::data_broadcast_id::DataBroadcastIdDescriptor<'a>,
247    TransportStream = 0x67 => crate::descriptors::transport_stream::TransportStreamDescriptor<'a>,
248    Dsng = 0x68 => crate::descriptors::dsng::DsngDescriptor<'a>,
249    Pdc = 0x69 => crate::descriptors::pdc::PdcDescriptor,
250    Ac3 = 0x6A => crate::descriptors::ac3::Ac3Descriptor<'a>,
251    AncillaryData = 0x6B => crate::descriptors::ancillary_data::AncillaryDataDescriptor,
252    CellList = 0x6C => crate::descriptors::cell_list::CellListDescriptor,
253    CellFrequencyLink = 0x6D => crate::descriptors::cell_frequency_link::CellFrequencyLinkDescriptor,
254    AnnouncementSupport = 0x6E => crate::descriptors::announcement_support::AnnouncementSupportDescriptor,
255    ApplicationSignalling = 0x6F => crate::descriptors::application_signalling::ApplicationSignallingDescriptor,
256    AdaptationFieldData = 0x70 => crate::descriptors::adaptation_field_data::AdaptationFieldDataDescriptor,
257    ServiceIdentifier = 0x71 => crate::descriptors::service_identifier::ServiceIdentifierDescriptor<'a>,
258    ServiceAvailability = 0x72 => crate::descriptors::service_availability::ServiceAvailabilityDescriptor,
259    DefaultAuthority = 0x73 => crate::descriptors::default_authority::DefaultAuthorityDescriptor<'a>,
260    RelatedContent = 0x74 => crate::descriptors::related_content::RelatedContentDescriptor,
261    TvaId = 0x75 => crate::descriptors::tva_id::TvaIdDescriptor,
262    ContentIdentifier = 0x76 => crate::descriptors::content_identifier::ContentIdentifierDescriptor<'a>,
263    TimeSliceFecIdentifier = 0x77 => crate::descriptors::time_slice_fec_identifier::TimeSliceFecIdentifierDescriptor<'a>,
264    EcmRepetitionRate = 0x78 => crate::descriptors::ecm_repetition_rate::EcmRepetitionRateDescriptor<'a>,
265    S2SatelliteDeliverySystem = 0x79 => crate::descriptors::s2_satellite_delivery_system::S2SatelliteDeliverySystemDescriptor,
266    EnhancedAc3 = 0x7A => crate::descriptors::enhanced_ac3::EnhancedAc3Descriptor<'a>,
267    Dts = 0x7B => crate::descriptors::dts::DtsDescriptor<'a>,
268    Aac = 0x7C => crate::descriptors::aac::AacDescriptor<'a>,
269    XaitLocation = 0x7D => crate::descriptors::xait_location::XaitLocationDescriptor,
270    FtaContentManagement = 0x7E => crate::descriptors::fta_content_management::FtaContentManagementDescriptor,
271    Extension = 0x7F => crate::descriptors::extension::ExtensionDescriptor<'a>;
272    // Private / context-dependent: variant exists but is NOT auto-dispatched.
273    // 0x83 logical_channel requires private_data_specifier context; enabled
274    // via the descriptor registry (Task 4).
275    @no_dispatch
276    LogicalChannel => crate::descriptors::logical_channel::LogicalChannelDescriptor,
277}
278
279/// Lazily walk a raw descriptor loop. Never panics.
280///
281/// Per-descriptor parse errors yield `Err` and iteration continues (the
282/// descriptor_length field bounds each entry, so the walker can always
283/// advance past a malformed body). A truncated final header or body yields
284/// one `Err` and then the iterator fuses (returns `None` forever after).
285#[must_use]
286pub fn parse_loop(bytes: &[u8]) -> DescriptorIter<'_> {
287    DescriptorIter {
288        bytes,
289        pos: 0,
290        fused: false,
291    }
292}
293
294/// Extract the next descriptor-loop entry (tag + full-entry slice) from the
295/// raw byte stream. Shared by [`DescriptorIter::next`] and
296/// [`RegistryIter::next`](crate::descriptors::registry::RegistryIter::next).
297///
298/// On success, advances `pos` past the entry and returns `(tag, full)`.
299/// On truncation, sets `fused = true` and returns `Some(Err(..))`.
300/// When the loop is exhausted, returns `None`.
301pub(crate) fn next_loop_entry<'a>(
302    bytes: &'a [u8],
303    pos: &mut usize,
304    fused: &mut bool,
305) -> Option<crate::Result<(u8, &'a [u8])>> {
306    if *fused || *pos >= bytes.len() {
307        return None;
308    }
309    let rem = &bytes[*pos..];
310    if rem.len() < 2 {
311        *fused = true;
312        return Some(Err(crate::Error::BufferTooShort {
313            need: 2,
314            have: rem.len(),
315            what: "descriptor header in loop",
316        }));
317    }
318    let tag = rem[0];
319    let len = rem[1] as usize;
320    let total = 2 + len;
321    if rem.len() < total {
322        *fused = true;
323        return Some(Err(crate::Error::BufferTooShort {
324            need: total,
325            have: rem.len(),
326            what: "descriptor body in loop",
327        }));
328    }
329    let full = &rem[..total];
330    *pos += total;
331    Some(Ok((tag, full)))
332}
333
334/// Iterator over a raw descriptor loop; see [`parse_loop`].
335#[derive(Debug, Clone)]
336pub struct DescriptorIter<'a> {
337    bytes: &'a [u8],
338    pos: usize,
339    fused: bool,
340}
341
342impl<'a> Iterator for DescriptorIter<'a> {
343    type Item = crate::Result<AnyDescriptor<'a>>;
344
345    fn next(&mut self) -> Option<Self::Item> {
346        let (tag, full) = match next_loop_entry(self.bytes, &mut self.pos, &mut self.fused)? {
347            Ok(v) => v,
348            Err(e) => return Some(Err(e)),
349        };
350        Some(match AnyDescriptor::dispatch(tag, full) {
351            Some(res) => res,
352            None => Ok(AnyDescriptor::Unknown {
353                tag,
354                body: &full[2..],
355            }),
356        })
357    }
358}
359
360impl core::iter::FusedIterator for DescriptorIter<'_> {}
361
362/// A raw descriptor loop, borrowed from the section. Zero-copy: walk it
363/// typed via [`DescriptorLoop::iter`]; serde serializes the typed walk.
364///
365/// This is the table-loop analogue of [`crate::text::DvbText`]: it wraps the
366/// raw `descriptor()` sequence (the variable-length region inside a table) and
367/// decodes — i.e. dispatches each entry to a typed [`AnyDescriptor`] — only on
368/// demand. Parsing stays zero-copy; the typed walk happens when you call
369/// [`DescriptorLoop::iter`] or serialize.
370///
371/// ```
372/// use dvb_si::descriptors::{AnyDescriptor, DescriptorLoop};
373///
374/// // short_event (tag 0x4D, "eng" / "Hi") then an unknown private tag 0xA7.
375/// let raw = [
376///     0x4D, 0x07, b'e', b'n', b'g', 0x02, b'H', b'i', 0x00,
377///     0xA7, 0x02, 0xCA, 0xFE,
378/// ];
379/// let loop_ = DescriptorLoop::new(&raw);
380/// let items: Vec<_> = loop_.iter().collect();
381/// assert_eq!(items.len(), 2);
382/// assert!(matches!(items[0].as_ref().unwrap(), AnyDescriptor::ShortEvent(_)));
383/// assert_eq!(loop_.raw(), &raw[..]); // bytes preserved verbatim
384/// ```
385#[derive(Clone, Copy, PartialEq, Eq, Hash, Default)]
386#[cfg_attr(feature = "yoke", derive(yoke::Yokeable))]
387pub struct DescriptorLoop<'a>(&'a [u8]);
388
389/// Bytes of a descriptor header: `descriptor_tag` (1) + `descriptor_length` (1).
390const DESC_HEADER_LEN: usize = 2;
391
392impl<'a> DescriptorLoop<'a> {
393    /// Wrap a raw descriptor-loop slice (the `descriptor()` bytes only — no
394    /// enclosing length field).
395    #[must_use]
396    pub const fn new(raw: &'a [u8]) -> Self {
397        Self(raw)
398    }
399
400    /// The raw wire bytes of the loop, verbatim. These are what a serializer
401    /// writes back; use them for the byte length of the loop.
402    #[must_use]
403    pub const fn raw(&self) -> &'a [u8] {
404        self.0
405    }
406
407    /// Lazily walk the loop, yielding one typed [`AnyDescriptor`] per entry
408    /// (or [`AnyDescriptor::Unknown`] for tags with no implementation).
409    /// Delegates to [`parse_loop`]; never panics.
410    #[must_use]
411    pub fn iter(&self) -> DescriptorIter<'a> {
412        parse_loop(self.0)
413    }
414
415    /// Walk the loop's `(tag, body)` pairs by the TLV structure only — never
416    /// typed-parses a body. Each entry is `descriptor_tag` (1 byte) +
417    /// `descriptor_length` (1 byte) + that many body bytes; a truncated final
418    /// entry (declared length running past the buffer) simply ends iteration.
419    ///
420    /// Use this for body-agnostic structural scans (e.g. "is tag X present?")
421    /// where a malformed or empty body should not hide the tag — unlike
422    /// [`iter`](Self::iter), which yields `Err`/`Unknown` for bodies that fail
423    /// to typed-parse.
424    pub fn raw_tags(&self) -> impl Iterator<Item = (u8, &'a [u8])> {
425        let b = self.0;
426        let mut pos = 0usize;
427        core::iter::from_fn(move || {
428            if pos + DESC_HEADER_LEN > b.len() {
429                return None;
430            }
431            let tag = b[pos];
432            let len = b[pos + 1] as usize;
433            let end = pos + DESC_HEADER_LEN + len;
434            if end > b.len() {
435                return None; // truncated final entry
436            }
437            let body = &b[pos + DESC_HEADER_LEN..end];
438            pos = end;
439            Some((tag, body))
440        })
441    }
442
443    /// True if the loop contains a descriptor with `tag`, regardless of whether
444    /// its body parses (or even fits). Pure structural walk: checks the
445    /// `descriptor_tag` byte at each entry header, so an empty (`[tag, 0x00]`)
446    /// or truncated (`[tag, 0x01]`) descriptor still counts as present.
447    #[must_use]
448    pub fn contains_tag(&self, tag: u8) -> bool {
449        let b = self.0;
450        let mut pos = 0usize;
451        while pos < b.len() {
452            if b[pos] == tag {
453                return true;
454            }
455            // Need the length byte to advance to the next header; without it
456            // the tag byte just checked was the last thing in the loop.
457            if pos + 1 >= b.len() {
458                break;
459            }
460            pos += DESC_HEADER_LEN + b[pos + 1] as usize;
461        }
462        false
463    }
464
465    /// Walk this loop through a [`DescriptorRegistry`](crate::descriptors::registry::DescriptorRegistry)
466    /// so registered private/custom descriptors (and PDS-scoped tags) are
467    /// produced as [`AnyDescriptor::Other`]; mirrors [`iter`](Self::iter)
468    /// otherwise.
469    ///
470    /// See [`DescriptorRegistry::parse_loop`](crate::descriptors::registry::DescriptorRegistry::parse_loop)
471    /// for precedence rules and PDS-scoped dispatch.
472    #[must_use]
473    pub fn iter_with<'r>(
474        &self,
475        registry: &'r crate::descriptors::registry::DescriptorRegistry,
476    ) -> crate::descriptors::registry::RegistryIter<'r, 'a> {
477        registry.parse_loop(self.0)
478    }
479
480    /// Walk this loop through both a [`DescriptorRegistry`](crate::descriptors::registry::DescriptorRegistry) and an
481    /// [`ExtensionRegistry`](crate::descriptors::extension::registry::ExtensionRegistry),
482    /// so that custom-registered extension bodies (tag `0x7F` with a known
483    /// `descriptor_tag_extension`) are surfaced as
484    /// [`ExtIterItem::CustomExtension`](crate::descriptors::registry::ExtIterItem::CustomExtension) with the type-erased value available
485    /// for downcast. All other descriptors follow the normal
486    /// [`iter_with`](Self::iter_with) precedence.
487    ///
488    /// This is the only way to reach private extension bodies during a
489    /// descriptor-loop walk; [`iter_with`](Self::iter_with) alone produces
490    /// the built-in [`ExtensionBody::Raw`](crate::descriptors::extension::ExtensionBody::Raw)
491    /// for unrecognised `descriptor_tag_extension` values.
492    #[must_use]
493    pub fn iter_with_extensions<'r>(
494        &self,
495        desc_reg: &'r crate::descriptors::registry::DescriptorRegistry,
496        ext_reg: &'r crate::descriptors::extension::registry::ExtensionRegistry,
497    ) -> crate::descriptors::registry::ExtRegistryIter<'r, 'a> {
498        crate::descriptors::registry::ExtRegistryIter::new(desc_reg, ext_reg, self.0)
499    }
500}
501
502impl<'a> core::ops::Deref for DescriptorLoop<'a> {
503    /// Derefs to the raw wire bytes — `len()`/indexing are **byte counts for
504    /// serialization, not entry counts**. To count entries, use
505    /// [`DescriptorLoop::iter`].
506    type Target = [u8];
507    fn deref(&self) -> &[u8] {
508        self.0
509    }
510}
511
512impl<'a> From<&'a [u8]> for DescriptorLoop<'a> {
513    fn from(raw: &'a [u8]) -> Self {
514        Self(raw)
515    }
516}
517
518impl core::fmt::Debug for DescriptorLoop<'_> {
519    /// Cheap: prints the byte length, not the decoded entries.
520    fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result {
521        write!(f, "DescriptorLoop(<{} bytes>)", self.0.len())
522    }
523}
524
525impl<'a> IntoIterator for &DescriptorLoop<'a> {
526    type Item = crate::Result<AnyDescriptor<'a>>;
527    type IntoIter = DescriptorIter<'a>;
528    fn into_iter(self) -> Self::IntoIter {
529        self.iter()
530    }
531}
532
533#[cfg(feature = "serde")]
534impl serde::Serialize for DescriptorLoop<'_> {
535    /// Serializes as a sequence of the typed walk: each `Ok(d)` becomes the
536    /// [`AnyDescriptor`] (camelCase external tagging), and each `Err(e)`
537    /// becomes a `{"parseError": "<Display>"}` map — parse errors are surfaced,
538    /// never silently dropped.
539    fn serialize<S: serde::Serializer>(&self, s: S) -> Result<S::Ok, S::Error> {
540        struct Entry<'a>(crate::Result<AnyDescriptor<'a>>);
541        impl serde::Serialize for Entry<'_> {
542            fn serialize<S: serde::Serializer>(&self, s: S) -> Result<S::Ok, S::Error> {
543                match &self.0 {
544                    Ok(d) => d.serialize(s),
545                    Err(e) => {
546                        use serde::ser::SerializeMap;
547                        let mut m = s.serialize_map(Some(1))?;
548                        m.serialize_entry("parseError", &e.to_string())?;
549                        m.end()
550                    }
551                }
552            }
553        }
554        s.collect_seq(self.iter().map(Entry))
555    }
556}
557// Serialize-only: the typed walk decodes DVB text and dispatches per-tag —
558// there is no lossless way to reconstruct the raw loop bytes from the
559// serialized form. Structs holding a DescriptorLoop derive Serialize only.
560// To reconstruct, keep the wire bytes and re-`parse` the table.
561
562#[cfg(test)]
563mod tests {
564    use super::*;
565
566    #[test]
567    fn contains_tag_detects_present_tag_regardless_of_body() {
568        // Well-formed AC-3 (0x6A) with a flags byte.
569        assert!(DescriptorLoop::new(&[0x6A, 0x01, 0x80]).contains_tag(0x6A));
570        // Empty body — malformed for AC-3, but the tag is present.
571        assert!(DescriptorLoop::new(&[0x6A, 0x00]).contains_tag(0x6A));
572        // Truncated — declares 1 body byte that isn't there; tag still present.
573        assert!(DescriptorLoop::new(&[0x6A, 0x01]).contains_tag(0x6A));
574        // Present after an earlier well-formed entry.
575        assert!(DescriptorLoop::new(&[0x09, 0x02, 0x00, 0x00, 0x7A, 0x00]).contains_tag(0x7A));
576        // Absent.
577        assert!(!DescriptorLoop::new(&[0x09, 0x02, 0x00, 0x00]).contains_tag(0x6A));
578        assert!(!DescriptorLoop::new(&[]).contains_tag(0x6A));
579    }
580
581    #[test]
582    fn raw_tags_walks_tlv_without_typed_parsing() {
583        // Two complete entries: tag 0x6A (empty body) + tag 0x40 (2-byte body).
584        let loop_ = DescriptorLoop::new(&[0x6A, 0x00, 0x40, 0x02, 0xAA, 0xBB]);
585        let pairs: Vec<_> = loop_.raw_tags().collect();
586        assert_eq!(pairs, vec![(0x6A, &[][..]), (0x40, &[0xAA, 0xBB][..])]);
587        // First entry [0x40,len=1,0xAA] complete, then a truncated [0x6A, len=5]
588        // entry whose declared body runs past the buffer.
589        let trunc = DescriptorLoop::new(&[0x40, 0x01, 0xAA, 0x6A, 0x05]);
590        let tags: Vec<u8> = trunc.raw_tags().map(|(t, _)| t).collect();
591        assert_eq!(tags, vec![0x40]); // the truncated 0x6A entry is dropped by raw_tags
592                                      // ...but contains_tag still sees the truncated tag.
593        assert!(trunc.contains_tag(0x6A));
594    }
595
596    #[test]
597    fn unknown_tag_yields_unknown_with_body_sans_header() {
598        // tag 0xA7 (no typed impl), length 2, body [0xDE, 0xAD].
599        let bytes = [0xA7, 0x02, 0xDE, 0xAD];
600        let items: Vec<_> = parse_loop(&bytes).collect();
601        assert_eq!(items.len(), 1);
602        match items[0].as_ref().unwrap() {
603            AnyDescriptor::Unknown { tag, body } => {
604                assert_eq!(*tag, 0xA7);
605                assert_eq!(*body, &[0xDE, 0xAD]);
606            }
607            other => panic!("expected Unknown, got {other:?}"),
608        }
609    }
610
611    #[test]
612    fn empty_loop_yields_nothing() {
613        assert_eq!(parse_loop(&[]).count(), 0);
614    }
615
616    #[test]
617    fn logical_channel_0x83_is_not_dispatched() {
618        // 0x83 has a variant but no dispatcher entry → Unknown, never panics.
619        let bytes = [0x83, 0x04, 0x00, 0x01, 0xFC, 0x01];
620        let items: Vec<_> = parse_loop(&bytes).collect();
621        assert_eq!(items.len(), 1);
622        assert!(matches!(
623            items[0].as_ref().unwrap(),
624            AnyDescriptor::Unknown { tag: 0x83, .. }
625        ));
626    }
627
628    #[test]
629    fn descriptor_loop_iter_matches_parse_loop() {
630        let raw = [
631            0x4D, 0x07, b'e', b'n', b'g', 0x02, b'H', b'i', 0x00, // short_event
632            0xA7, 0x02, 0xCA, 0xFE, // unknown 0xA7
633        ];
634        let via_loop: Vec<_> = DescriptorLoop::new(&raw)
635            .iter()
636            .map(|r| format!("{r:?}"))
637            .collect();
638        let via_fn: Vec<_> = parse_loop(&raw).map(|r| format!("{r:?}")).collect();
639        assert_eq!(via_loop, via_fn);
640        // raw()/Deref expose the wire bytes (byte length, not entry count).
641        assert_eq!(DescriptorLoop::new(&raw).raw(), &raw[..]);
642        assert_eq!(DescriptorLoop::new(&raw).len(), raw.len());
643        // IntoIterator for &DescriptorLoop.
644        let count = (&DescriptorLoop::new(&raw)).into_iter().count();
645        assert_eq!(count, 2);
646    }
647
648    #[test]
649    fn descriptor_loop_debug_is_cheap() {
650        let raw = [0x4D, 0x02, 0x01, 0x02];
651        assert_eq!(
652            format!("{:?}", DescriptorLoop::new(&raw)),
653            "DescriptorLoop(<4 bytes>)"
654        );
655    }
656
657    #[test]
658    fn iter_with_custom_tag_yields_other() {
659        use crate::descriptors::registry::DescriptorRegistry;
660        use crate::traits::DescriptorDef;
661
662        #[derive(Debug, PartialEq, Eq)]
663        #[cfg_attr(feature = "serde", derive(serde::Serialize))]
664        struct MyTag0xA7 {
665            x: u8,
666        }
667
668        impl<'a> dvb_common::Parse<'a> for MyTag0xA7 {
669            type Error = crate::error::Error;
670            fn parse(bytes: &'a [u8]) -> crate::Result<Self> {
671                if bytes.len() < 3 {
672                    return Err(crate::error::Error::BufferTooShort {
673                        need: 3,
674                        have: bytes.len(),
675                        what: "MyTag0xA7",
676                    });
677                }
678                Ok(Self { x: bytes[2] })
679            }
680        }
681
682        impl<'a> DescriptorDef<'a> for MyTag0xA7 {
683            const TAG: u8 = 0xA7;
684            const NAME: &'static str = "MY_TAG_0xA7";
685        }
686
687        let mut reg = DescriptorRegistry::new();
688        reg.register::<MyTag0xA7>();
689
690        let raw = [
691            0x4D, 0x07, b'e', b'n', b'g', 0x02, b'H', b'i', 0x00, 0xA7, 0x02, 0xCA, 0xFE,
692        ];
693        let loop_ = DescriptorLoop::new(&raw);
694        let items: Vec<_> = loop_.iter_with(&reg).collect::<Result<_, _>>().unwrap();
695        assert_eq!(items.len(), 2);
696        assert!(matches!(items[0], AnyDescriptor::ShortEvent(_)));
697        match &items[1] {
698            AnyDescriptor::Other { tag, value } => {
699                assert_eq!(*tag, 0xA7);
700                assert_eq!(value.downcast_ref::<MyTag0xA7>().unwrap().x, 0xCA);
701            }
702            other => panic!("expected Other, got {other:?}"),
703        }
704    }
705
706    #[test]
707    fn iter_with_empty_registry_matches_iter_for_builtin() {
708        let raw = [
709            0x4D, 0x07, b'e', b'n', b'g', 0x02, b'H', b'i', 0x00, 0xA7, 0x02, 0xCA, 0xFE,
710        ];
711        let loop_ = DescriptorLoop::new(&raw);
712        let reg = crate::descriptors::registry::DescriptorRegistry::new();
713        let via_iter: Vec<_> = loop_.iter().collect();
714        let via_iter_with: Vec<_> = loop_.iter_with(&reg).collect();
715
716        assert_eq!(via_iter.len(), via_iter_with.len());
717        for (a, b) in via_iter.iter().zip(via_iter_with.iter()) {
718            match (a, b) {
719                (Ok(AnyDescriptor::ShortEvent(_)), Ok(AnyDescriptor::ShortEvent(_))) => {}
720                (
721                    Ok(AnyDescriptor::Unknown { tag: t1, body: b1 }),
722                    Ok(AnyDescriptor::Unknown { tag: t2, body: b2 }),
723                ) => {
724                    assert_eq!(t1, t2);
725                    assert_eq!(b1, b2);
726                }
727                (Err(_), Err(_)) => {}
728                (l, r) => panic!("mismatch: {l:?} vs {r:?}"),
729            }
730        }
731    }
732
733    #[cfg(feature = "serde")]
734    #[test]
735    fn descriptor_loop_serializes_typed_unknown_and_parse_error() {
736        // [valid short_event, unknown tag 0xA7, truncated final entry].
737        // A truncated final entry (declared len 5, only 1 body byte present)
738        // makes the walker yield a final Err → {"parseError": …}.
739        let raw = [
740            0x4D, 0x07, b'e', b'n', b'g', 0x02, b'H', b'i', 0x00, // short_event
741            0xA7, 0x02, 0xCA, 0xFE, // unknown 0xA7
742            0x55, 0x05, 0x00, // parental_rating header claims 5 bytes; only 1 present
743        ];
744        let v = serde_json::to_value(DescriptorLoop::new(&raw)).unwrap();
745        let arr = v.as_array().expect("sequence");
746        assert_eq!(arr.len(), 3);
747        // 1. typed short_event under the camelCase variant key.
748        assert!(arr[0].get("shortEvent").is_some(), "got {}", arr[0]);
749        assert_eq!(arr[0]["shortEvent"]["event_name"], "Hi");
750        // 2. unknown tag carries its raw body bytes.
751        let unknown = arr[1].get("unknown").expect("unknown variant");
752        assert_eq!(unknown["tag"], 0xA7);
753        assert_eq!(unknown["body"], serde_json::json!([0xCA, 0xFE]));
754        // 3. truncated entry → parseError, never silently dropped.
755        assert!(
756            arr[2].get("parseError").is_some(),
757            "expected parseError, got {}",
758            arr[2]
759        );
760    }
761}