mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-06-22 17:07:23 +00:00
4422c975c3
FairPlay HLS (SAMPLE-AES/cbcs) ships a content KID in its skd:// key but no PlayReady/Widevine PSSH. Synthesize a PlayReady header from that KID so a PlayReady CDM can license and decrypt it whenever the backend is multi-DRM. - FairPlay DRM (PlayReady subclass) + FairPlay.from_kid; fairplay_kid_from_skd extracts the KID from GUID / ?KID= / 32-hex skd forms (core/drm/fairplay.py) - build_pr_header_from_kid synthesizes a WRMHEADER/PRO (cbcs->AESCBC v4.3, cenc->AESCTR v4.0); PlayReady.from_track falls back to a tenc KID when no PSSH - Service.get_fairplay_license hook (defaults to get_playready_license) - dl.py routes FairPlay tracks to get_fairplay_license through the PlayReady CDM - EXAMPLE service + config.yaml document the bridge end to end - tests for the synthesized header and skd KID extraction
82 lines
2.8 KiB
Python
82 lines
2.8 KiB
Python
"""Tests for the FairPlay -> PlayReady bridge DRM type and skd KID extraction.
|
|
|
|
FairPlay HLS (SAMPLE-AES / cbcs) carries a content KID but no PlayReady PSSH. The
|
|
``FairPlay`` DRM synthesizes a PlayReady header from that KID so a PlayReady CDM can
|
|
license it; it is a PlayReady subclass so the pipeline reuses CDM/decrypt logic and
|
|
only differs in routing the request to ``get_fairplay_license``.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Optional
|
|
from uuid import UUID
|
|
|
|
import pytest
|
|
|
|
from unshackle.core.drm import FairPlay, PlayReady
|
|
from unshackle.core.drm.fairplay import fairplay_kid_from_skd
|
|
|
|
|
|
def test_fairplay_is_playready_subclass() -> None:
|
|
# Pipeline relies on isinstance(drm, PlayReady) (get_drm_for_cdm, prepare_drm branch).
|
|
fp = FairPlay.from_kid(UUID("279926a3-d9b9-4b6f-8b2c-1a2b3c4d5e6f"))
|
|
assert isinstance(fp, PlayReady)
|
|
assert isinstance(fp, FairPlay)
|
|
|
|
|
|
def test_from_kid_round_trips_kid_and_sets_pssh() -> None:
|
|
kid = UUID("279926a3-d9b9-4b6f-8b2c-1a2b3c4d5e6f")
|
|
fp = FairPlay.from_kid(kid)
|
|
assert fp.pssh_b64
|
|
assert [k.hex for k in fp.kids] == [kid.hex]
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("skd", "expected"),
|
|
[
|
|
# hyphenated GUID (with optional :IV)
|
|
("skd://72ea9ec1-abb4-40d8-b3a9-e0bd1833bd58:0123456789ABCDEF", "72ea9ec1abb440d8b3a9e0bd1833bd58"),
|
|
# GUID as a query param
|
|
("skd://lic.example/fairplay?KID=4376a4b3-d8ef-4f21-9a6b-faa81a2e59e3", "4376a4b3d8ef4f219a6bfaa81a2e59e3"),
|
|
# bare 32-hex path segment
|
|
("skd://0000000069176cc2af6f93cce3515ced/clip/x", "0000000069176cc2af6f93cce3515ced"),
|
|
],
|
|
)
|
|
def test_kid_from_skd_common_forms(skd: str, expected: str) -> None:
|
|
kid = fairplay_kid_from_skd(skd)
|
|
assert kid is not None
|
|
assert kid.hex == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"skd",
|
|
[
|
|
"skd://example.com/p123456/c1", # service-specific encoding, not a GUID/32-hex
|
|
"skd://opaque-asset-token-abc",
|
|
"",
|
|
],
|
|
)
|
|
def test_kid_from_skd_returns_none_for_non_kid_forms(skd: str) -> None:
|
|
# Services with non-standard skd encodings derive the KID themselves; the generic
|
|
# helper must not mis-parse them.
|
|
assert fairplay_kid_from_skd(skd) is None
|
|
|
|
|
|
def test_base_get_fairplay_license_delegates_to_playready() -> None:
|
|
captured: dict = {}
|
|
|
|
class FakeService:
|
|
# Bind the unbound base methods to a minimal stand-in.
|
|
from unshackle.core.service import Service
|
|
|
|
get_fairplay_license = Service.get_fairplay_license
|
|
|
|
def get_playready_license(self, *, challenge: bytes, title, track) -> Optional[bytes]:
|
|
captured["called"] = (challenge, title, track)
|
|
return b"LICENSE"
|
|
|
|
svc = FakeService()
|
|
out = svc.get_fairplay_license(challenge=b"chal", title="T", track="K")
|
|
assert out == b"LICENSE"
|
|
assert captured["called"] == (b"chal", "T", "K")
|