Devine
/
pywidevine
mirror of https://github.com/devine-dl/pywidevine.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Cdm: Add parsing error handlers to Service Cert DrmCertificates

This commit is contained in:
rlaphoenix 2022-09-28 07:37:17 +01:00
parent fa00bbd8e4
commit 42b825dcd5
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pywidevine/cdm.py
View File

@ -206,6 +206,7 @@ class Cdm:
signed_message = SignedMessage() signed_message = SignedMessage()
signed_drm_certificate = SignedDrmCertificate() signed_drm_certificate = SignedDrmCertificate()
drm_certificate = DrmCertificate()
try: try:
signed_message.ParseFromString(certificate) signed_message.ParseFromString(certificate)
@ -232,10 +233,15 @@ class Cdm:
) )
except (ValueError, TypeError): except (ValueError, TypeError):
raise SignatureMismatch("Signature Mismatch on SignedDrmCertificate, rejecting certificate") raise SignatureMismatch("Signature Mismatch on SignedDrmCertificate, rejecting certificate")
else:
session.service_certificate = signed_message try:
drm_certificate = DrmCertificate()
drm_certificate.ParseFromString(signed_drm_certificate.drm_certificate) drm_certificate.ParseFromString(signed_drm_certificate.drm_certificate)
if drm_certificate.SerializeToString() != signed_drm_certificate.drm_certificate:
raise DecodeError("partial parse")
except DecodeError as e:
raise DecodeError(f"Could not parse signed certificate's message as a DrmCertificate, {e}")
session.service_certificate = signed_message
return drm_certificate.provider_id return drm_certificate.provider_id
def get_service_certificate(self, session_id: bytes) -> Optional[SignedMessage]: def get_service_certificate(self, session_id: bytes) -> Optional[SignedMessage]: