vsd-mp4 0.2.1

MP4 parser ported from shaka-player with decryption and subtitle extraction support.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use std::{
    error::Error,
    fs::{self, File},
    io::Write,
    path::PathBuf,
    sync::LazyLock,
};
use vsd_mp4::decrypt::CencDecrypter;

const VIDEO_KEY: &str = "100b6c20940f779a4589152b57d2dacb";
const AUDIO_KEY: &str = "3bda3329158a4789880816a70e7e436d";

static SAMPLES_DIR: LazyLock<PathBuf> =
    LazyLock::new(|| PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("tests/samples"));

static OUTPUT_DIR: LazyLock<PathBuf> = LazyLock::new(|| {
    let dir = PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("../target/vsd-mp4-samples");
    fs::create_dir_all(&dir).ok();
    dir
});

macro_rules! sample {
    ($test_name: ident, $scheme: literal, $track: literal) => {
        #[test]
        fn $test_name() -> Result<(), Box<dyn Error>> {
            let init_data = fs::read(SAMPLES_DIR.join(concat!($scheme, "/", $track, "_init.mp4")))?;
            let segment_data = fs::read(SAMPLES_DIR.join(concat!($scheme, "/", $track, "_1.m4s")))?;

            let decrypted = CencDecrypter::new(if $track == "video" {
                VIDEO_KEY
            } else {
                AUDIO_KEY
            })?
            .decrypt_fragment(segment_data, Some(init_data.as_slice()))?;

            let mut f = File::create(OUTPUT_DIR.join(concat!($scheme, "-", $track, ".mp4")))?;
            f.write_all(&init_data)?;
            f.write_all(&decrypted)?;
            Ok(())
        }
    };
}

sample!(test_cenc_video, "cenc", "video");
sample!(test_cenc_audio, "cenc", "audio");
sample!(test_cens_video, "cens", "video");
sample!(test_cens_audio, "cens", "audio");
sample!(test_cbc1_video, "cbc1", "video");
sample!(test_cbc1_audio, "cbc1", "audio");
sample!(test_cbcs_video, "cbcs", "video");
sample!(test_cbcs_audio, "cbcs", "audio");