dvb-tools 6.2.0

//! `dvb-tools t2mi <file> [--pid 0xNNN|raw] [--inner] [--plp N]` — T2-MI
//! pump + optional inner-TS extraction.
//!
//! Without `--inner` this is byte-identical to the former
//! `dvb-t2mi/examples/t2mi_dump.rs`. With `--inner` it chains
//! [`dvb_t2mi::inner_ts::InnerTsRecovery`] to recover the inner MPEG-TS
//! carried inside the BBFrames of the T2-MI stream and writes the recovered
//! 188-byte packets to stdout, so a caller can `> inner.ts` and pipe them
//! back into `dvb-tools dump`.
use std::io::{BufWriter, Write};
use std::process::ExitCode;

use dvb_t2mi::inner_ts::InnerTsRecovery;
use dvb_t2mi::pump::T2miPump;

use crate::util::{for_each_packet, read_file};

/// Default T2-MI PID when none is specified (EN 302 755 V1.4.1 §5.1 — the usual
/// data-piping PID for T2-MI in TS).
const DEFAULT_T2MI_PID: u16 = 0x0006;

/// `--pid` value: a TS-encapsulated target PID, or raw bytes.
enum PidArg {
    Ts(u16),
    Raw,
}

/// Parse `--pid` (decimal or `0x`-prefixed hex) or a literal `raw`.
fn parse_pid(s: &str) -> Option<PidArg> {
    if s.eq_ignore_ascii_case("raw") {
        return Some(PidArg::Raw);
    }
    let value = if let Some(hex) = s.strip_prefix("0x").or_else(|| s.strip_prefix("0X")) {
        u16::from_str_radix(hex, 16).ok()?
    } else {
        s.parse::<u16>().ok()?
    };
    Some(PidArg::Ts(value))
}

pub fn run(args: &[String]) -> ExitCode {
    let mut path: Option<String> = None;
    let mut pid: PidArg = PidArg::Ts(DEFAULT_T2MI_PID);
    let mut plp: Option<u8> = None;
    let mut inner = false;

    let mut i = 0;
    while i < args.len() {
        match args[i].as_str() {
            "--pid" => {
                i += 1;
                match args.get(i).and_then(|s| parse_pid(s)) {
                    Some(p) => pid = p,
                    None => {
                        eprintln!("dvb-tools t2mi: --pid expects a number or 'raw'");
                        return ExitCode::FAILURE;
                    }
                }
            }
            "--plp" => {
                i += 1;
                match args.get(i).and_then(|s| s.parse::<u8>().ok()) {
                    Some(n) => plp = Some(n),
                    None => {
                        eprintln!("dvb-tools t2mi: --plp expects a u8");
                        return ExitCode::FAILURE;
                    }
                }
            }
            "--inner" => inner = true,
            "-h" | "--help" => {
                eprintln!(
                    "usage: dvb-tools t2mi <file.t2mi|file.ts> \
                     [--pid 0xNNN|raw] [--inner] [--plp N]"
                );
                return ExitCode::SUCCESS;
            }
            other if other.starts_with('-') => {
                eprintln!("dvb-tools t2mi: unknown option {other}");
                return ExitCode::FAILURE;
            }
            other => path = Some(other.to_string()),
        }
        i += 1;
    }

    if plp.is_some() && !inner {
        eprintln!("dvb-tools t2mi: --plp requires --inner");
        return ExitCode::FAILURE;
    }

    if inner && matches!(pid, PidArg::Raw) {
        eprintln!("dvb-tools t2mi: --inner is not valid with --pid raw");
        return ExitCode::FAILURE;
    }

    let Some(path) = path else {
        eprintln!(
            "usage: dvb-tools t2mi <file.t2mi|file.ts> \
             [--pid 0xNNN|raw] [--inner] [--plp N]"
        );
        return ExitCode::FAILURE;
    };

    let data = match read_file(&path, "dvb-tools t2mi") {
        Ok(d) => d,
        Err(code) => return code,
    };

    if inner {
        let PidArg::Ts(t2mi_pid) = pid else {
            // Unreachable — guarded above.
            return ExitCode::FAILURE;
        };
        let mut rec = match plp {
            Some(p) => InnerTsRecovery::new_for_plp(t2mi_pid, p),
            None => InnerTsRecovery::new(t2mi_pid),
        };
        let stdout = std::io::stdout();
        let mut out = BufWriter::new(stdout.lock());
        for packet in for_each_packet(&data) {
            for inner_pkt in rec.feed(packet) {
                if let Err(e) = out.write_all(inner_pkt) {
                    if e.kind() == std::io::ErrorKind::BrokenPipe {
                        break;
                    }
                    eprintln!("dvb-tools t2mi: write stdout: {e}");
                    return ExitCode::FAILURE;
                }
            }
        }
        if let Err(e) = out.flush() {
            eprintln!("dvb-tools t2mi: flush stdout: {e}");
            return ExitCode::FAILURE;
        }
        let s = rec.stats();
        eprintln!(
            "-- inner packets written: ts_packets={} t2mi_packets={} \
             crc_failures={} malformed={} filtered_bbframes={}",
            s.ts_packets,
            s.t2mi_packets,
            s.crc_failures,
            s.malformed_packets,
            rec.filtered_bbframes(),
        );
        return ExitCode::SUCCESS;
    }

    // Default behaviour (no --inner): byte-identical to the old t2mi_dump.
    let mut pump = match pid {
        PidArg::Ts(p) => T2miPump::new(p),
        PidArg::Raw => T2miPump::raw(),
    };

    let print_event = |event: dvb_t2mi::pump::T2miEvent| match event.header() {
        Ok(hdr) => println!(
            "type={} (0x{:02X}) count={} superframe_idx={} stream_id={} payload_bits={}",
            event.payload().map(|p| p.name()).unwrap_or("PARSE_ERROR"),
            event.packet_type(),
            hdr.packet_count,
            hdr.superframe_idx,
            hdr.t2mi_stream_id,
            hdr.payload_len_bits,
        ),
        Err(e) => eprintln!(
            "packet_type=0x{:02X} header error: {e}",
            event.packet_type()
        ),
    };

    match pid {
        PidArg::Ts(_) => {
            for packet in for_each_packet(&data) {
                for event in pump.feed_ts(packet) {
                    print_event(event);
                }
            }
        }
        PidArg::Raw => {
            for event in pump.feed_raw(&data) {
                print_event(event);
            }
        }
    }

    let s = pump.stats();
    eprintln!(
        "-- ts_packets={} t2mi_packets={} crc_failures={} malformed={}",
        s.ts_packets, s.t2mi_packets, s.crc_failures, s.malformed_packets
    );
    ExitCode::SUCCESS
}