pssh-box 0.1.11

Parsing and serialization support for PSSH boxes used in DRM systems
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141

/// generate.rs -- tests for generating PSSH boxes


use base64::prelude::{Engine as _, BASE64_STANDARD};
use test_log::test;
use pssh_box::{pprint, from_bytes, from_base64, from_hex, ToBytes, PsshBox, PsshData, DRMKeyId};
use pssh_box::{WIDEVINE_SYSTEM_ID, PLAYREADY_SYSTEM_ID};


#[test]
fn test_roundtrip_widevine_bytes() {
    let mut pssh = PsshBox::new_widevine();
    assert_eq!(pssh.system_id, WIDEVINE_SYSTEM_ID);
    let kid = DRMKeyId::try_from("4444aaaa2222bbbb8888eeee7777cccc").unwrap();
    pssh.add_key_id(kid);
    if let PsshData::Widevine(ref mut pd) = pssh.pssh_data {
        pd.provider = Some(String::from("pssh-box-rs"));
        pd.content_id = Some(hex::encode("DEADBEAF").into());
        assert_eq!(pd.policy, None);
    }
    pprint(&pssh);
    let serialized = pssh.clone().to_bytes();
    let boxes = from_bytes(&serialized).unwrap();
    assert_eq!(boxes.len(), 1);
    let parsed = &boxes[0];
    assert_eq!(parsed.system_id, WIDEVINE_SYSTEM_ID);
    assert_eq!(parsed.key_ids[0], kid);
    if let PsshData::Widevine(ref pd) = parsed.pssh_data {
        assert_eq!(pd.provider, Some(String::from("pssh-box-rs")));
        assert_eq!(pd.content_id, Some(hex::encode("DEADBEAF").into()));
        assert_eq!(pd.policy, None);
    }
}


#[test]
fn test_roundtrip_widevine_base64() {
    let mut pssh = PsshBox::new_widevine();
    assert_eq!(pssh.system_id, WIDEVINE_SYSTEM_ID);
    let kid = DRMKeyId::try_from("4444aaaa2222bbbb8888eeee7777cccc").unwrap();
    pssh.add_key_id(kid);
    if let PsshData::Widevine(ref mut pd) = pssh.pssh_data {
        pd.provider = Some(String::from("pssh-box-rs"));
        pd.content_id = Some(hex::encode("DEADBEAF").into());
    }
    pprint(&pssh);
    let b64 = pssh.clone().to_base64();
    let boxes = from_base64(&b64).unwrap();
    assert_eq!(boxes.len(), 1);
    let parsed = &boxes[0];
    assert_eq!(parsed.system_id, WIDEVINE_SYSTEM_ID);
    assert_eq!(parsed.key_ids[0], kid);
    if let PsshData::Widevine(ref pd) = parsed.pssh_data {
        assert_eq!(pd.provider, Some(String::from("pssh-box-rs")));
        assert_eq!(pd.content_id, Some(hex::encode("DEADBEAF").into()));
    }
}


#[test]
fn test_roundtrip_widevine_hex() {
    let mut pssh = PsshBox::new_widevine();
    assert_eq!(pssh.system_id, WIDEVINE_SYSTEM_ID);
    let kid1 = DRMKeyId::try_from("4444aaaa2222bbbb8888eeee7777cccc").unwrap();
    pssh.add_key_id(kid1);
    let kid2 = DRMKeyId::try_from("AAAAaaaa222200008888dddd3333eeee").unwrap();
    pssh.add_key_id(kid2);
    if let PsshData::Widevine(ref mut pd) = pssh.pssh_data {
        pd.provider = Some(String::from("pssh-box-rs"));
        pd.content_id = Some(hex::encode("DEADBEAF").into());
    }
    pprint(&pssh);
    let hx = pssh.clone().to_hex();
    let boxes = from_hex(&hx).unwrap();
    assert_eq!(boxes.len(), 1);
    let parsed = &boxes[0];
    assert_eq!(parsed.system_id, WIDEVINE_SYSTEM_ID);
    assert_eq!(parsed.key_ids[0], kid1);
    assert_eq!(parsed.key_ids[1], kid2);
    if let PsshData::Widevine(ref pd) = parsed.pssh_data {
        assert_eq!(pd.provider, Some(String::from("pssh-box-rs")));
        assert_eq!(pd.content_id, Some(hex::encode("DEADBEAF").into()));
    }
}


#[test]
fn test_roundtrip_playready_bytes() {
    let mut pssh = PsshBox::new_playready();
    assert_eq!(pssh.system_id, PLAYREADY_SYSTEM_ID);
    let kid1 = DRMKeyId::try_from("4444aaaa2222bbbb8888eeee7777cccc").unwrap();
    pssh.add_key_id(kid1);
    let kid2 = DRMKeyId::try_from("AAAAaaaa222200008888dddd3333eeee").unwrap();
    pssh.add_key_id(kid2);
    if let PsshData::PlayReady(ref mut pd) = pssh.pssh_data {
        let wrmh = &mut pd.record[0].record_value;
        wrmh.data.checksum = Some(BASE64_STANDARD.decode("7zDsYfDVHUY=").unwrap());
        wrmh.data.lui_url = Some(String::from("http://www.example.com/"));
    }
    pprint(&pssh);
    let serialized = pssh.to_bytes();
    let boxes = from_bytes(&serialized).unwrap();
    let parsed = &boxes[0];
    assert_eq!(parsed.system_id, PLAYREADY_SYSTEM_ID);
    assert_eq!(parsed.key_ids[0], kid1);
    assert_eq!(parsed.key_ids[1], kid2);
    println!("PlayReady> {pssh:?}");
    if let PsshData::PlayReady(ref pd) = parsed.pssh_data {
        let wrmh = &pd.record[0].record_value;
        assert_eq!(wrmh.data.checksum, Some(BASE64_STANDARD.decode("7zDsYfDVHUY=").unwrap()));
    }
}


#[test]
fn test_roundtrip_playready_base64() {
    let mut pssh = PsshBox::new_playready();
    assert_eq!(pssh.system_id, PLAYREADY_SYSTEM_ID);
    let kid1 = DRMKeyId::try_from("4444aaaa2222bbbb8888eeee7777cccc").unwrap();
    pssh.add_key_id(kid1);
    let kid2 = DRMKeyId::try_from("AAAAaaaa222200008888dddd3333eeee").unwrap();
    pssh.add_key_id(kid2);
    if let PsshData::PlayReady(ref mut pd) = pssh.pssh_data {
        let wrmh = &mut pd.record[0].record_value;
        wrmh.data.checksum = Some(BASE64_STANDARD.decode("7zDsYfDVHUY=").unwrap());
        wrmh.data.lui_url = Some(String::from("http://www.example.com/"));
    }
    pprint(&pssh);
    let serialized = pssh.clone().to_base64();
    let boxes = from_base64(&serialized).unwrap();
    let parsed = &boxes[0];
    assert_eq!(parsed.system_id, PLAYREADY_SYSTEM_ID);
    assert_eq!(parsed.key_ids[0], kid1);
    assert_eq!(parsed.key_ids[1], kid2);
    println!("PlayReady> {pssh:?}");
    if let PsshData::PlayReady(ref pd) = parsed.pssh_data {
        let wrmh = &pd.record[0].record_value;
        assert_eq!(wrmh.data.checksum, Some(BASE64_STANDARD.decode("7zDsYfDVHUY=").unwrap()));
        assert_eq!(wrmh.data.lui_url, Some(String::from("http://www.example.com/")));
    }
}