ninjajiraiya
/
devine
forked from DRMTalks/devine
1
0
Fork 0
Code Pull Requests Activity

Implement CEKNotFound and EmptyLicense exceptions to Widevine

This commit is contained in:
rlaphoenix 2023-03-04 11:18:28 +00:00
parent c3a22431f0
commit 4b330c0478
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
devine/core/drm/widevine.py
View File

@ -210,10 +210,10 @@ class Widevine:
for key in cdm.get_keys(session_id, "CONTENT") for key in cdm.get_keys(session_id, "CONTENT")
} }
if not self.content_keys: if not self.content_keys:
raise ValueError("No Content Keys were returned by the License") raise Widevine.Exceptions.EmptyLicense("No Content Keys were within the License")
if kid not in self.content_keys: if kid not in self.content_keys:
raise ValueError(f"No Content Key with the KID ({kid.hex}) was returned") raise Widevine.Exceptions.CEKNotFound(f"No Content Key for KID {kid.hex} within the License")
finally: finally:
cdm.close(session_id) cdm.close(session_id)
@ -303,5 +303,11 @@ class Widevine:
class KIDNotFound(Exception): class KIDNotFound(Exception):
"""KID (Encryption Key ID) was not found.""" """KID (Encryption Key ID) was not found."""
class CEKNotFound(Exception):
"""CEK (Content Encryption Key) for KID was not found in License."""
class EmptyLicense(Exception):
"""License returned no Content Encryption Keys."""
__ALL__ = (Widevine,) __ALL__ = (Widevine,)