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

Cdm: Save Service Certificate in SignedMessage form 

We may need the signature for external verification, and most APIs require it to be in a SignedMessage to be accepted, even though the SignedMessage is pretty much empty (not even actually signed lol).
This commit is contained in:
rlaphoenix 2022-07-31 13:17:49 +01:00
parent a0fa559255
commit 0c85abb2d4
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pywidevine/cdm.py
View File

@ -150,6 +150,10 @@ class Cdm:
signed_drm_certificate.ParseFromString(certificate) signed_drm_certificate.ParseFromString(certificate)
if signed_drm_certificate.SerializeToString() != certificate: if signed_drm_certificate.SerializeToString() != certificate:
raise DecodeError() raise DecodeError()
# Craft a SignedMessage as it's stored as a SignedMessage
signed_message.Clear()
signed_message.msg = signed_drm_certificate.SerializeToString()
# we don't need to sign this message, this is normal
except DecodeError: except DecodeError:
# could be a direct unsigned DrmCertificate, but reject those anyway # could be a direct unsigned DrmCertificate, but reject those anyway
raise DecodeError("Could not parse certificate as a SignedDrmCertificate") raise DecodeError("Could not parse certificate as a SignedDrmCertificate")
@ -164,9 +168,9 @@ 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: else:
session.service_certificate = signed_message
drm_certificate = DrmCertificate() drm_certificate = DrmCertificate()
drm_certificate.ParseFromString(signed_drm_certificate.drm_certificate) drm_certificate.ParseFromString(signed_drm_certificate.drm_certificate)
session.service_certificate = drm_certificate
return drm_certificate.provider_id return drm_certificate.provider_id
def get_license_challenge( def get_license_challenge(

4
pywidevine/session.py
View File

@ -3,12 +3,12 @@ from typing import Optional
from Crypto.Random import get_random_bytes from Crypto.Random import get_random_bytes
from pywidevine.key import Key from pywidevine.key import Key
from pywidevine.license_protocol_pb2 import DrmCertificate from pywidevine.license_protocol_pb2 import SignedMessage
class Session: class Session:
def __init__(self): def __init__(self):
self.id = get_random_bytes(16) self.id = get_random_bytes(16)
self.service_certificate: Optional[DrmCertificate] = None self.service_certificate: Optional[SignedMessage] = None
self.context: dict[bytes, tuple[bytes, bytes]] = {} self.context: dict[bytes, tuple[bytes, bytes]] = {}
self.keys: list[Key] = [] self.keys: list[Key] = []