MattJackson
ff5547363b
Audit fixes (14 critical, 22 warnings): - UDF: bounds checks on all ICB/FID parsing from disc data - SCSI Linux: saturating_sub on residual, CDB length guard, buffer size guard - SCSI macOS: SCSITaskStatus u32 (was u8 — stack corruption) - AACS: EC mod_inv returns infinity instead of panic, key reduced mod n - AACS: do_handshake tries all host certs (was returning on first failure) - H.264: bounds check on SPS < 4 bytes - ContentReader: error on missing unit key (was zero-fill) - KEYDB: flat redirect loop (was recursive), 100MB response limit, Windows HOME fallback - ISO writer: AVDP extent order, partition length, allocation cap - Network: removed TCP_NODELAY on bulk stream - MKV: guard on u64::MAX seek - disc.rs: saturating_sub on extent offset, simplified dead region code - cargo fmt (610 violations), cargo clippy --fix (55 auto-fixes) DVD support (new files): - src/ifo.rs — IFO parser (VIDEO_TS.IFO, VTS_XX_0.IFO, PGC chains, cells, streams) — 13 tests - src/mux/ps.rs — MPEG-2 Program Stream demuxer (pack headers, PES, private stream 1) — 12 tests - src/mux/codec/mpeg2.rs — MPEG-2 video parser (sequence headers, I-frame detection) — 15 tests - src/css/crack.rs — split-attack algorithm (LFSR cipher needs verification — test ignored) 226 tests total (was 186), 1 ignored (CSS crack needs cipher verification).
93 lines
2.7 KiB
Rust
93 lines
2.7 KiB
Rust
//! Elementary stream codec parsers.
|
|
//!
|
|
//! Each parser takes PES packets and produces frames suitable for MKV muxing.
|
|
//! Responsibilities:
|
|
//! - Find frame boundaries
|
|
//! - Extract codec initialization data (SPS/PPS, etc.)
|
|
//! - Determine keyframe status
|
|
//! - Convert PTS from 90kHz to nanoseconds
|
|
|
|
pub mod ac3;
|
|
pub mod dts;
|
|
pub mod h264;
|
|
pub mod hevc;
|
|
pub mod mpeg2;
|
|
pub mod pgs;
|
|
pub mod truehd;
|
|
pub mod vc1;
|
|
|
|
use super::ts::PesPacket;
|
|
use crate::disc::Codec;
|
|
|
|
/// A single frame ready for MKV muxing.
|
|
pub struct Frame {
|
|
/// Presentation timestamp in nanoseconds.
|
|
pub pts_ns: i64,
|
|
/// Whether this is a keyframe (used for cue points).
|
|
pub keyframe: bool,
|
|
/// Frame data (elementary stream bytes).
|
|
pub data: Vec<u8>,
|
|
}
|
|
|
|
/// Convert 90kHz PTS to nanoseconds.
|
|
pub fn pts_to_ns(pts: i64) -> i64 {
|
|
pts * 100_000 / 9
|
|
}
|
|
|
|
/// Trait for codec-specific elementary stream parsers.
|
|
pub trait CodecParser: Send {
|
|
/// Parse a PES packet into zero or more frames.
|
|
/// Most codecs: one PES = one frame.
|
|
/// Some (TrueHD): multiple access units per PES.
|
|
fn parse(&mut self, pes: &PesPacket) -> Vec<Frame>;
|
|
|
|
/// Get codec initialization data (e.g., SPS+PPS for H.264).
|
|
/// Returns None until enough data has been seen.
|
|
fn codec_private(&self) -> Option<Vec<u8>>;
|
|
}
|
|
|
|
/// Passthrough parser — treats each PES as one frame, no parsing.
|
|
/// Used for codecs where PES = frame (AC3, DTS, PGS).
|
|
pub struct PassthroughParser {
|
|
keyframe: bool,
|
|
}
|
|
|
|
impl PassthroughParser {
|
|
pub fn new(always_keyframe: bool) -> Self {
|
|
Self {
|
|
keyframe: always_keyframe,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl CodecParser for PassthroughParser {
|
|
fn parse(&mut self, pes: &PesPacket) -> Vec<Frame> {
|
|
let pts_ns = pes.pts.map(pts_to_ns).unwrap_or(0);
|
|
vec![Frame {
|
|
pts_ns,
|
|
keyframe: self.keyframe,
|
|
data: pes.data.clone(),
|
|
}]
|
|
}
|
|
|
|
fn codec_private(&self) -> Option<Vec<u8>> {
|
|
None
|
|
}
|
|
}
|
|
|
|
/// Create the appropriate parser for a codec.
|
|
pub fn parser_for_codec(codec: Codec) -> Box<dyn CodecParser> {
|
|
match codec {
|
|
Codec::H264 => Box::new(h264::H264Parser::new()),
|
|
Codec::Hevc => Box::new(hevc::HevcParser::new()),
|
|
Codec::Mpeg2 => Box::new(mpeg2::Mpeg2Parser::new()),
|
|
Codec::Vc1 => Box::new(vc1::Vc1Parser::new()),
|
|
Codec::Ac3 | Codec::Ac3Plus => Box::new(ac3::Ac3Parser::new()),
|
|
Codec::DtsHdMa | Codec::DtsHdHr | Codec::Dts => Box::new(dts::DtsParser::new()),
|
|
Codec::TrueHd => Box::new(truehd::TrueHdParser::new()),
|
|
Codec::Pgs => Box::new(pgs::PgsParser::new()),
|
|
Codec::Lpcm => Box::new(PassthroughParser::new(true)),
|
|
_ => Box::new(PassthroughParser::new(true)),
|
|
}
|
|
}
|