vsd-mp4 0.2.0

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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

use cipher::{BlockModeDecrypt, KeyIvInit};

type Aes128Cbc = cbc::Decryptor<aes::Aes128>;

/// A decrypter for HTTP Live Streaming (HLS) AES-128 encrypted segments.
///
/// In HLS AES-128, the entire segment file is encrypted using AES-128 in Cipher Block
/// Chaining (CBC) mode with PKCS7 padding.
#[derive(Clone)]
pub struct HlsAes128Decrypter {
    key: [u8; 16],
    iv: [u8; 16],
}

impl HlsAes128Decrypter {
    /// Creates a new `HlsAes128Decrypter` with the given key and IV.
    pub fn new(key: &[u8; 16], iv: &[u8; 16]) -> Self {
        Self { key: *key, iv: *iv }
    }

    /// Increments the initialization vector (IV) by 1.
    pub fn increment_iv(&mut self) {
        self.iv = (u128::from_be_bytes(self.iv) + 1).to_be_bytes();
    }

    /// Decrypts the given encrypted segment.
    pub fn decrypt(&self, mut input: Vec<u8>) -> Vec<u8> {
        let slice_len = {
            let slice = Aes128Cbc::new((&self.key).into(), (&self.iv).into())
                .decrypt_padded::<cipher::block_padding::Pkcs7>(&mut input)
                .unwrap();
            slice.len()
        };

        input.truncate(slice_len);
        input
    }
}

/// A decrypter for HTTP Live Streaming (HLS) SAMPLE-AES encrypted segments.
///
/// In HLS SAMPLE-AES, only media samples (e.g., individual audio or video frames) are
/// encrypted, while container metadata and structural headers remain unencrypted.
#[derive(Clone)]
pub struct HlsSampleAesDecrypter {
    key: [u8; 16],
    iv: [u8; 16],
}

impl HlsSampleAesDecrypter {
    /// Creates a new `HlsSampleAesDecrypter` with the given key and IV.
    pub fn new(key: &[u8; 16], iv: &[u8; 16]) -> Self {
        Self { key: *key, iv: *iv }
    }

    /// Increments the initialization vector (IV) by 1.
    pub fn increment_iv(&mut self) {
        self.iv = (u128::from_be_bytes(self.iv) + 1).to_be_bytes();
    }

    /// Decrypts the given encrypted segment.
    pub fn decrypt(&self, input: Vec<u8>) -> Vec<u8> {
        let mut input = std::io::Cursor::new(input);
        let mut output = Vec::new();
        iori_ssa::decrypt(&mut input, &mut output, self.key, self.iv).unwrap();
        output
    }
}