Simplify verification of parsing in Cdm.set_service_certificate

This commit is contained in:
rlaphoenix 2023-12-06 16:00:52 +00:00
parent a04e751aa1
commit 25e03529f6
1 changed files with 3 additions and 6 deletions

View File

@ -222,13 +222,10 @@ class Cdm:
try:
signed_message.ParseFromString(certificate)
if (
signed_message.SerializeToString() == certificate or
if all(
# See https://github.com/devine-dl/pywidevine/issues/41
all(
bytes(chunk) == signed_message.SerializeToString()
for chunk in zip(*[iter(certificate)] * len(signed_message.SerializeToString()))
)
bytes(chunk) == signed_message.SerializeToString()
for chunk in zip(*[iter(certificate)] * len(signed_message.SerializeToString()))
):
signed_drm_certificate.ParseFromString(signed_message.msg)
else: