Simplify verification of parsing in Cdm.set_service_certificate

devine-dl · Dec 6, 2023 · 25e0352 · 25e0352
1 parent a04e751
commit 25e0352
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/pywidevine/cdm.py b/pywidevine/cdm.py
@@ -222,13 +222,10 @@ def set_service_certificate(self, session_id: bytes, certificate: Optional[Union
 
         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: