rswidevine 0.1.1

Widevine CDM (Content Decryption Module) implementation in Rust.
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

//! Session module for managing Widevine CDM sessions.
//!
//! Each session stores request context and decrypted keys loaded from a license.

use std::collections::HashMap;

use rsa::rand_core::{OsRng, RngCore};

use crate::key::Key;
use crate::license_protocol::SignedDrmCertificate;

/// A Widevine CDM session.
///
/// Session ids are randomly generated 16-byte values. The `context` map
/// associates request_id to the derived encryption and MAC contexts.
#[derive(Debug, Clone)]
pub struct Session {
    /// Session number (1-indexed).
    pub number: u32,
    /// Unique session identifier (16 random bytes).
    pub id: Vec<u8>,
    /// Service certificate for privacy mode.
    pub service_certificate: Option<SignedDrmCertificate>,
    /// Context data: maps request_id to (enc_context, mac_context).
    pub context: HashMap<Vec<u8>, (Vec<u8>, Vec<u8>)>,
    /// Decrypted keys from the license.
    pub keys: Vec<Key>,
}

impl Session {
    /// Create a new session with the given session number.
    #[must_use]
    pub fn new(number: u32) -> Self {
        let mut id = vec![0u8; 16];
        let mut rng = OsRng;
        rng.fill_bytes(&mut id);

        Self {
            number,
            id,
            service_certificate: None,
            context: HashMap::new(),
            keys: Vec::new(),
        }
    }
}