feat(cmaf): add qbz-cmaf crate with crypto module

HKDF-SHA256 session key derivation, AES-128-CBC content key unwrap,
AES-128-CTR frame decryption, and MD5 request signing for Qobuz
CMAF segmented streaming.

cc
This commit is contained in:
vicrodh committed 2026-04-06 13:10:00 -06:00
1 parent 9a95a54e0f
commit 794dd9f24c
5 files changed
+192

No files matched your search

+1
View File
@@ -11,6 +11,7 @@ members = [
"qbz-cache",
"qbz-library",
"qbz-cast",
"qbz-cmaf",
"qconnect-core",
"qconnect-protocol",
"qconnect-transport-ws",
+17
View File
@@ -0,0 +1,17 @@
[package]
name = "qbz-cmaf"
version.workspace = true
edition.workspace = true
license.workspace = true
description = "CMAF segment parsing and crypto for Qobuz streaming"
[dependencies]
aes = "0.8"
cbc = "0.1"
ctr = "0.9"
hkdf = "0.12"
sha2 = "0.10"
base64 = { workspace = true }
md-5 = { workspace = true }
thiserror = { workspace = true }
log = { workspace = true }
+154
View File
@@ -0,0 +1,154 @@
use aes::cipher::{BlockDecryptMut, KeyIvInit, StreamCipher};
use base64::{Engine, engine::general_purpose::URL_SAFE_NO_PAD};
use hkdf::Hkdf;
use sha2::Sha256;
use crate::error::CmafError;
const RNG_INIT: &str = "abb21364945c0583309667d13ca3d93a";
type Aes128CbcDec = cbc::Decryptor<aes::Aes128>;
type Aes128Ctr = ctr::Ctr128BE<aes::Aes128>;
fn hex_decode(hex: &str) -> Vec<u8> {
(0..hex.len())
.step_by(2)
.map(|i| u8::from_str_radix(&hex[i..i + 2], 16).expect("RNG_INIT is valid hex"))
.collect()
}
/// Derive the 16-byte session key from the session/start `infos` field.
///
/// `infos` format: `"salt_b64url.info_b64url"`
/// IKM = hex-decoded RNG_INIT (16 bytes)
pub fn derive_session_key(infos: &str) -> Result<[u8; 16], CmafError> {
let parts: Vec<&str> = infos.split('.').collect();
if parts.len() < 2 {
return Err(CmafError::InvalidInfos(
"session infos must have at least 2 dot-separated parts".into(),
));
}
let salt = URL_SAFE_NO_PAD.decode(parts[0])?;
let info = URL_SAFE_NO_PAD.decode(parts[1])?;
let ikm = hex_decode(RNG_INIT);
let hk = Hkdf::<Sha256>::new(Some(&salt), &ikm);
let mut okm = [0u8; 16];
hk.expand(&info, &mut okm).map_err(|_| CmafError::HkdfExpand)?;
Ok(okm)
}
/// Unwrap the per-track content key using the session key.
///
/// `key_str` format: `"qbz-1.wrapped_key_b64url.iv_b64url"`
pub fn unwrap_content_key(session_key: &[u8; 16], key_str: &str) -> Result<[u8; 16], CmafError> {
let parts: Vec<&str> = key_str.split('.').collect();
if parts.len() < 3 {
return Err(CmafError::InvalidKey(
"key string must have at least 3 dot-separated parts".into(),
));
}
let wrapped = URL_SAFE_NO_PAD.decode(parts[1])?;
let iv = URL_SAFE_NO_PAD.decode(parts[2])?;
if iv.len() != 16 {
return Err(CmafError::InvalidKey(format!(
"unwrap IV must be 16 bytes, got {}",
iv.len()
)));
}
let mut buf = wrapped.clone();
let decrypted =
Aes128CbcDec::new(session_key.into(), iv.as_slice().into())
.decrypt_padded_mut::<aes::cipher::block_padding::Pkcs7>(&mut buf)
.map_err(|e| CmafError::AesDecrypt(format!("AES-CBC unwrap failed: {e}")))?;
if decrypted.len() != 16 {
return Err(CmafError::InvalidKey(format!(
"unwrapped key must be 16 bytes, got {}",
decrypted.len()
)));
}
let mut key = [0u8; 16];
key.copy_from_slice(decrypted);
Ok(key)
}
/// Decrypt a FLAC frame in-place using AES-128-CTR.
///
/// `iv_8` = 8-byte IV from the segment UUID box entry, zero-padded to 16 bytes.
pub fn decrypt_frame(content_key: &[u8; 16], iv_8: &[u8; 8], data: &mut [u8]) {
let mut nonce = [0u8; 16];
nonce[..8].copy_from_slice(iv_8);
Aes128Ctr::new(content_key.into(), &nonce.into()).apply_keystream(data);
}
/// Compute the MD5 request signature for Qobuz CMAF API calls.
///
/// Concatenates method + sorted key-value pairs + timestamp + RNG_INIT,
/// then returns the lowercase hex MD5 digest.
pub fn compute_request_sig(
method: &str,
args: &std::collections::BTreeMap<&str, String>,
timestamp: &str,
) -> String {
use md5::{Digest, Md5};
let mut hasher = Md5::new();
hasher.update(method.as_bytes());
for (k, v) in args {
hasher.update(k.as_bytes());
hasher.update(v.as_bytes());
}
hasher.update(timestamp.as_bytes());
hasher.update(RNG_INIT.as_bytes());
format!("{:x}", hasher.finalize())
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_compute_request_sig() {
let mut args = std::collections::BTreeMap::new();
args.insert("profile", "qbz-1".to_string());
let sig = compute_request_sig("sessionstart", &args, "1775500000");
assert_eq!(sig.len(), 32); // MD5 hex = 32 chars
// Verify deterministic
let sig2 = compute_request_sig("sessionstart", &args, "1775500000");
assert_eq!(sig, sig2);
}
#[test]
fn test_decrypt_frame_roundtrip() {
let key = [0x42u8; 16];
let iv = [0x01u8; 8];
let original = b"Hello FLAC frame data here!".to_vec();
let mut data = original.clone();
decrypt_frame(&key, &iv, &mut data);
assert_ne!(data, original);
decrypt_frame(&key, &iv, &mut data);
assert_eq!(data, original);
}
#[test]
fn test_derive_session_key_invalid_infos() {
let result = derive_session_key("no_dot_here");
assert!(result.is_err());
}
#[test]
fn test_unwrap_content_key_invalid_format() {
let key = [0u8; 16];
let result = unwrap_content_key(&key, "only.two");
assert!(result.is_err());
}
}
+15
View File
@@ -0,0 +1,15 @@
#[derive(Debug, thiserror::Error)]
pub enum CmafError {
#[error("Invalid infos format: {0}")]
InvalidInfos(String),
#[error("Invalid key format: {0}")]
InvalidKey(String),
#[error("Base64 decode error: {0}")]
Base64(#[from] base64::DecodeError),
#[error("HKDF expand error")]
HkdfExpand,
#[error("AES decrypt error: {0}")]
AesDecrypt(String),
#[error("CMAF parse error: {0}")]
ParseError(String),
}
+5
View File
@@ -0,0 +1,5 @@
pub mod crypto;
pub mod error;
pub use crypto::{compute_request_sig, decrypt_frame, derive_session_key, unwrap_content_key};
pub use error::CmafError;