pssh_box/
widevine.rs

1//! Definitions for PSSH data in the Widevine DRM system.
2//
3//
4// Widevine PSSH data and licence messages use a protobuf encoding, which we decode using the Prost
5// crate.
6
7use std::fmt;
8use prost::Message;
9use serde_json::{json, Value};
10use crate::ToBytes;
11
12// This file is generated by Prost in our build script
13include!(concat!(env!("OUT_DIR"), "/widevine.rs"));
14
15// include!("widevine-generated.rs");
16
17impl ToBytes for WidevinePsshData {
18    fn to_bytes(&self) -> Vec<u8> {
19        self.encode_to_vec()
20    }
21}
22
23impl WidevinePsshData {
24    pub fn to_json(&self) -> Value {
25        let mut out = json!({});
26        if let Some(a) = self.algorithm {
27            out["algorithm"] = if a == 0 {
28                Value::String(String::from("unencrypted"))
29            } else {
30                Value::String(String::from("Aesctr"))
31            };
32        }
33        if let Some(p) = &self.provider {
34            out["provider"] = Value::String(p.to_string());
35        }
36        if let Some(p) = &self.policy {
37            if !p.is_empty() {
38                out["policy"] = Value::String(p.to_string());
39            }
40        }
41        if let Some(cpi) = &self.crypto_period_index {
42            out["crypto_period_index"] = Value::String(cpi.to_string());
43        }
44        if let Some(gl) = &self.grouped_license {
45            out["grouped_licence"] = Value::String(hex::encode(gl));
46        }
47        if let Some(ps) = &self.protection_scheme {
48            let scheme = match widevine_pssh_data::ProtectionScheme::try_from(*ps) {
49                Ok(s) => String::from(s.as_str_name()),
50                Err(_) => format!("unknown ({ps})"),
51            };
52            out["protection_scheme"] = Value::String(scheme);
53        }
54        let mut keys = Vec::new();
55        for kid in &self.key_id {
56            keys.push(Value::String(hex::encode(kid)));
57        }
58        if !keys.is_empty() {
59            out["key_id"] = Value::Array(keys);
60        }
61        if let Some(cid) = &self.content_id {
62            out["content_id"] = Value::String(hex::encode(cid));
63        }
64        out
65    }
66}
67
68impl fmt::Debug for WidevinePsshData {
69    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
70        let mut items = Vec::new();
71        if let Some(a) = &self.algorithm {
72            items.push(if *a == 0 { String::from("unencrypted") } else { String::from("Aesctr") });
73        }
74        if let Some(p) = &self.provider {
75            items.push(format!("provider: {p}"));
76        }
77        if let Some(p) = &self.policy {
78            if !p.is_empty() {
79                items.push(format!("policy: {p}"));
80            }
81        }
82        if let Some(cpi) = &self.crypto_period_index {
83            items.push(format!("crypto_period_index: {cpi}"));
84        }
85        if let Some(gl) = &self.grouped_license {
86            items.push(format!("grouped_licence: {}", hex::encode(gl)));
87        }
88        // In the 2016 version of the protobuf for WidevinePsshData, the protection_scheme field is
89        // specified as a uint32. In 2018 versions there is a ProtectionScheme enum which specifies
90        // values for the uint32.
91        if let Some(ps) = &self.protection_scheme {
92            let scheme = match widevine_pssh_data::ProtectionScheme::try_from(*ps) {
93                Ok(s) => String::from(s.as_str_name()),
94                Err(_) => format!("unknown ({ps})"),
95            };
96            items.push(format!("protection_scheme: {scheme}"));
97        }
98        for kid in &self.key_id {
99            items.push(format!("keyid: {}", hex::encode(kid)));
100        }
101        if let Some(cid) = &self.content_id {
102            items.push(format!("content_id: {}", hex::encode(cid)));
103        }
104        write!(f, "WidevinePsshData<{}>", items.join(", "))
105    }
106}
pssh_box/widevine.rs

pssh_box/
widevine.rs
