Add MKV muxer, IsoWriter, disc pipeline, and network tests

- MKV muxer: EBML header, segment, cluster, cues, multi-track, keyframe flags — 6 tests - MkvStream: BD-TS roundtrip, metadata preservation — 2 tests - IsoWriter: valid UDF, file size update, custom names, empty content — 4 tests - Disc pipeline: format detection (UHD/BD/DVD), content format, capacity, duration — 5 tests - Network: listen/connect roundtrip, metadata flow — 2 tests (ignored for CI) - Encryption: no AACS dir, no keydb — 2 tests - 297 tests total, all passing
2026-04-11 17:27:36 +00:00
parent 63accb6718
commit 9cb0c369e4
6 changed files with 1001 additions and 0 deletions
@@ -140,3 +140,239 @@ fn scan_options_with_keydb_pathbuf() {
    let opts = ScanOptions::with_keydb(path.clone());
    assert_eq!(opts.keydb_path.unwrap(), path);
 }
+
+// ── detect_format integration tests ───────────────────────────────────────
+
+use libfreemkv::{Codec, ColorSpace, ContentFormat, HdrFormat, Stream, VideoStream};
+
+fn title_with_video(codec: Codec, resolution: &str, content_format: ContentFormat) -> DiscTitle {
+    DiscTitle {
+        playlist: "00800.mpls".into(),
+        playlist_id: 800,
+        duration_secs: 7200.0,
+        size_bytes: 0,
+        clips: Vec::new(),
+        streams: vec![Stream::Video(VideoStream {
+            pid: 0x1011,
+            codec,
+            resolution: resolution.into(),
+            frame_rate: "23.976".into(),
+            hdr: HdrFormat::Sdr,
+            color_space: ColorSpace::Bt709,
+            secondary: false,
+            label: String::new(),
+        })],
+        extents: Vec::new(),
+        content_format,
+    }
+}
+
+#[test]
+fn disc_title_duration_display_edge_cases() {
+    let mut t = DiscTitle::empty();
+
+    // 0 seconds
+    t.duration_secs = 0.0;
+    assert_eq!(t.duration_display(), "0h 00m");
+
+    // 1 second
+    t.duration_secs = 1.0;
+    assert_eq!(t.duration_display(), "0h 00m");
+
+    // 59 minutes
+    t.duration_secs = 59.0 * 60.0;
+    assert_eq!(t.duration_display(), "0h 59m");
+
+    // 24 hours exactly
+    t.duration_secs = 24.0 * 3600.0;
+    assert_eq!(t.duration_display(), "24h 00m");
+}
+
+#[test]
+fn content_format_default_bdts() {
+    let t = title_with_video(Codec::H264, "1080p", ContentFormat::BdTs);
+    assert_eq!(t.content_format, ContentFormat::BdTs);
+}
+
+#[test]
+fn content_format_dvd_mpegps() {
+    let t = title_with_video(Codec::Mpeg2, "480i", ContentFormat::MpegPs);
+    assert_eq!(t.content_format, ContentFormat::MpegPs);
+}
+
+// ── UDF helpers for encryption resolution tests ───────────────────────────
+
+/// Build an AVDP sector (tag_id=2) pointing to VDS at the given LBA.
+fn make_avdp_sector(vds_lba: u32) -> Vec<u8> {
+    let mut s = vec![0u8; SECTOR_SIZE];
+    s[0..2].copy_from_slice(&2u16.to_le_bytes());
+    s[16..20].copy_from_slice(&vds_lba.to_le_bytes());
+    s[20..24].copy_from_slice(&(6u32 * SECTOR_SIZE as u32).to_le_bytes());
+    s
+}
+
+fn make_pvd_sector(volume_id: &str) -> Vec<u8> {
+    let mut s = vec![0u8; SECTOR_SIZE];
+    s[0..2].copy_from_slice(&1u16.to_le_bytes());
+    if !volume_id.is_empty() {
+        let id_bytes = volume_id.as_bytes();
+        s[24] = 8;
+        let copy_len = id_bytes.len().min(30);
+        s[25..25 + copy_len].copy_from_slice(&id_bytes[..copy_len]);
+        s[55] = (1 + copy_len) as u8;
+    }
+    s
+}
+
+fn make_partition_desc(partition_start: u32) -> Vec<u8> {
+    let mut s = vec![0u8; SECTOR_SIZE];
+    s[0..2].copy_from_slice(&5u16.to_le_bytes());
+    s[188..192].copy_from_slice(&partition_start.to_le_bytes());
+    s
+}
+
+fn make_lvd_sector_simple() -> Vec<u8> {
+    let mut s = vec![0u8; SECTOR_SIZE];
+    s[0..2].copy_from_slice(&6u16.to_le_bytes());
+    s[268..272].copy_from_slice(&1u32.to_le_bytes());
+    s
+}
+
+fn make_terminator() -> Vec<u8> {
+    let mut s = vec![0u8; SECTOR_SIZE];
+    s[0..2].copy_from_slice(&8u16.to_le_bytes());
+    s
+}
+
+fn make_fsd_sector(root_meta_lba: u32) -> Vec<u8> {
+    let mut s = vec![0u8; SECTOR_SIZE];
+    s[0..2].copy_from_slice(&256u16.to_le_bytes());
+    s[400..404].copy_from_slice(&(SECTOR_SIZE as u32).to_le_bytes());
+    s[404..408].copy_from_slice(&root_meta_lba.to_le_bytes());
+    s
+}
+
+fn make_dir_icb(data_meta_lba: u32, data_len: u32) -> Vec<u8> {
+    let mut s = vec![0u8; SECTOR_SIZE];
+    s[0..2].copy_from_slice(&266u16.to_le_bytes());
+    s[56..64].copy_from_slice(&(data_len as u64).to_le_bytes());
+    s[208..212].copy_from_slice(&0u32.to_le_bytes());
+    s[212..216].copy_from_slice(&8u32.to_le_bytes());
+    s[216..220].copy_from_slice(&data_len.to_le_bytes());
+    s[220..224].copy_from_slice(&data_meta_lba.to_le_bytes());
+    s
+}
+
+fn make_parent_fid() -> Vec<u8> {
+    let fid_len = ((38 + 0 + 0 + 3) & !3) as usize;
+    let mut fid = vec![0u8; fid_len];
+    fid[0..2].copy_from_slice(&257u16.to_le_bytes());
+    fid[18] = 0x08;
+    fid[19] = 0;
+    fid
+}
+
+fn make_fid(name: &str, icb_meta_lba: u32, is_dir: bool) -> Vec<u8> {
+    let mut name_bytes = vec![8u8];
+    name_bytes.extend_from_slice(name.as_bytes());
+    let l_fi = name_bytes.len() as u8;
+    let file_chars: u8 = if is_dir { 0x02 } else { 0x00 };
+    let fid_len = ((38 + 0 + l_fi as usize + 3) & !3) as usize;
+    let mut fid = vec![0u8; fid_len];
+    fid[0..2].copy_from_slice(&257u16.to_le_bytes());
+    fid[18] = file_chars;
+    fid[19] = l_fi;
+    fid[20..24].copy_from_slice(&(SECTOR_SIZE as u32).to_le_bytes());
+    fid[24..28].copy_from_slice(&icb_meta_lba.to_le_bytes());
+    fid[36..38].copy_from_slice(&0u16.to_le_bytes());
+    fid[38..38 + name_bytes.len()].copy_from_slice(&name_bytes);
+    fid
+}
+
+/// Build a minimal UDF image with an empty root directory (no /AACS).
+fn build_minimal_udf(reader: &mut MockSectorReader) {
+    let partition_start: u32 = 512;
+    reader.sectors.insert(256, make_avdp_sector(32));
+    reader.sectors.insert(32, make_pvd_sector("TEST_DISC"));
+    reader.sectors.insert(33, make_partition_desc(partition_start));
+    reader.sectors.insert(34, make_lvd_sector_simple());
+    reader.sectors.insert(35, make_terminator());
+
+    reader.sectors.insert(partition_start, make_fsd_sector(1));
+
+    let parent_fid = make_parent_fid();
+    let dir_data_len = parent_fid.len() as u32;
+    reader
+        .sectors
+        .insert(partition_start + 1, make_dir_icb(2, dir_data_len));
+    let mut sector = vec![0u8; SECTOR_SIZE];
+    sector[..parent_fid.len()].copy_from_slice(&parent_fid);
+    reader.sectors.insert(partition_start + 2, sector);
+}
+
+/// Build a UDF image with an /AACS directory (empty).
+fn build_udf_with_aacs_dir(reader: &mut MockSectorReader) {
+    let partition_start: u32 = 512;
+    reader.sectors.insert(256, make_avdp_sector(32));
+    reader.sectors.insert(32, make_pvd_sector("ENCRYPTED_DISC"));
+    reader.sectors.insert(33, make_partition_desc(partition_start));
+    reader.sectors.insert(34, make_lvd_sector_simple());
+    reader.sectors.insert(35, make_terminator());
+
+    reader.sectors.insert(partition_start, make_fsd_sector(1));
+
+    // Root -> AACS (dir)
+    let parent_fid = make_parent_fid();
+    let aacs_fid = make_fid("AACS", 3, true);
+    let mut root_data = Vec::new();
+    root_data.extend_from_slice(&parent_fid);
+    root_data.extend_from_slice(&aacs_fid);
+    let root_data_len = root_data.len() as u32;
+
+    reader
+        .sectors
+        .insert(partition_start + 1, make_dir_icb(2, root_data_len));
+    let mut sector = vec![0u8; SECTOR_SIZE];
+    sector[..root_data.len()].copy_from_slice(&root_data);
+    reader.sectors.insert(partition_start + 2, sector);
+
+    // AACS dir (empty)
+    let aacs_parent = make_parent_fid();
+    let aacs_data_len = aacs_parent.len() as u32;
+    reader
+        .sectors
+        .insert(partition_start + 3, make_dir_icb(4, aacs_data_len));
+    let mut sector2 = vec![0u8; SECTOR_SIZE];
+    sector2[..aacs_parent.len()].copy_from_slice(&aacs_parent);
+    reader.sectors.insert(partition_start + 4, sector2);
+}
+
+#[test]
+fn resolve_encryption_no_aacs_dir() {
+    // A UDF image with no /AACS directory should result in no encryption
+    let mut reader = MockSectorReader::new();
+    build_minimal_udf(&mut reader);
+
+    let opts = ScanOptions::default();
+    let disc = Disc::scan_image(&mut reader, 1000, &opts).unwrap();
+
+    assert!(!disc.encrypted, "disc without /AACS should not be encrypted");
+    assert!(disc.aacs.is_none(), "aacs should be None without /AACS dir");
+}
+
+#[test]
+fn resolve_encryption_no_keydb() {
+    // A UDF image with /AACS directory but no keydb path -> aacs is None
+    let mut reader = MockSectorReader::new();
+    build_udf_with_aacs_dir(&mut reader);
+
+    // No keydb configured and no standard keydb on the system
+    let opts = ScanOptions::with_keydb("/nonexistent/path/KEYDB.cfg");
+    let disc = Disc::scan_image(&mut reader, 1000, &opts).unwrap();
+
+    // The disc detects encryption but can't resolve keys without a keydb
+    assert!(
+        disc.aacs.is_none(),
+        "aacs should be None when keydb is unavailable"
+    );
+}