mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 20:15:10 +00:00
imSp4rky
6294566823
cdm lookups used the raw service key while decryption upper-cases its keys, so a case-mismatched cdm entry silently used the default. Add a shared ci_get helper, route cdm/decryption through resolvers, drop a redundant decryption mutation and dead downloader writes, and add a regression bench.
74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
"""Pins per-service decryption-tool and CDM resolution, including key casing.
|
|
|
|
A per-service `cdm:`/`decryption:` entry must resolve regardless of the key's case vs the
|
|
service tag, falling back to the configured default otherwise. Guards the silent
|
|
"falls back to default on a case mismatch" foot-gun.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from unshackle.core.config import Config, resolve_cdm_name, resolve_decryption
|
|
from unshackle.core.utils.collections import ci_get
|
|
|
|
pytestmark = pytest.mark.unit
|
|
|
|
|
|
def test_ci_get_exact_and_case_insensitive():
|
|
m = {"NF": "a", "Default": "b"}
|
|
assert ci_get(m, "NF") == "a"
|
|
assert ci_get(m, "nf") == "a"
|
|
assert ci_get(m, "default") == "b"
|
|
assert ci_get(m, "missing") is None
|
|
assert ci_get(m, "missing", "fallback") == "fallback"
|
|
assert ci_get({}, "NF", "fallback") == "fallback"
|
|
|
|
|
|
def test_decryption_map_keys_upper_cased_on_build():
|
|
c = Config(decryption={"cr": "mp4decrypt", "default": "shaka"})
|
|
assert c.decryption_map == {"CR": "mp4decrypt", "DEFAULT": "shaka"}
|
|
assert c.decryption == "shaka"
|
|
|
|
|
|
def test_resolve_decryption_per_service_and_default():
|
|
c = Config(decryption={"CR": "mp4decrypt", "default": "shaka"})
|
|
assert resolve_decryption(c.decryption_map, c.decryption, "CR") == "mp4decrypt"
|
|
# service not mapped -> default
|
|
assert resolve_decryption(c.decryption_map, c.decryption, "NF") == "shaka"
|
|
# case-insensitive both ways (lowercase yaml key, mixed-case lookup)
|
|
c2 = Config(decryption={"cr": "mp4decrypt", "default": "shaka"})
|
|
assert resolve_decryption(c2.decryption_map, c2.decryption, "Cr") == "mp4decrypt"
|
|
|
|
|
|
def test_decryption_scalar_form():
|
|
c = Config(decryption="mp4decrypt")
|
|
assert c.decryption_map == {}
|
|
assert resolve_decryption(c.decryption_map, c.decryption, "ANY") == "mp4decrypt"
|
|
|
|
|
|
def test_cdm_keys_preserved_on_build():
|
|
c = Config(cdm={"nf": "dev_a", "default": "dev_b"})
|
|
assert c.cdm == {"nf": "dev_a", "default": "dev_b"}
|
|
|
|
|
|
def test_resolve_cdm_name_case_insensitive():
|
|
# lowercase yaml key must resolve for an uppercase service tag (the bug being fixed)
|
|
c = Config(cdm={"nf": "dev_a", "default": "dev_b"})
|
|
assert resolve_cdm_name(c.cdm, "NF") == "dev_a"
|
|
# unmapped service -> default
|
|
assert resolve_cdm_name(c.cdm, "ATV") == "dev_b"
|
|
# uppercase yaml key, lowercase lookup
|
|
c2 = Config(cdm={"NF": "dev_a", "default": "dev_b"})
|
|
assert resolve_cdm_name(c2.cdm, "nf") == "dev_a"
|
|
|
|
|
|
def test_resolve_cdm_name_override_wins():
|
|
c = Config(cdm={"NF": "dev_a", "default": "dev_b"})
|
|
assert resolve_cdm_name(c.cdm, "NF", override="dev_override") == "dev_override"
|
|
|
|
|
|
def test_resolve_cdm_name_no_match_no_default():
|
|
c = Config(cdm={"NF": "dev_a"})
|
|
assert resolve_cdm_name(c.cdm, "ATV") is None
|