upload

pywidevine/decrypt/wvdecryptcustom.py

@@ -0,0 +1,79 @@
+import logging
+import subprocess
+import re
+import base64
+
+from pywidevine.cdm import cdm, deviceconfig
+
+
+class WvDecrypt(object):
+    WV_SYSTEM_ID = [
+        237,
+        239,
+        139,
+        169,
+        121,
+        214,
+        74,
+        206,
+        163,
+        200,
+        39,
+        220,
+        213,
+        29,
+        33,
+        237,
+    ]
+
+    def __init__(self, init_data_b64, cert_data_b64, device):
+        self.init_data_b64 = init_data_b64
+        self.cert_data_b64 = cert_data_b64
+        self.device = device
+
+        self.cdm = cdm.Cdm()
+
+        def check_pssh(pssh_b64):
+            pssh = base64.b64decode(pssh_b64)
+            if not pssh[12:28] == bytes(self.WV_SYSTEM_ID):
+                new_pssh = bytearray([0, 0, 0])
+                new_pssh.append(32 + len(pssh))
+                new_pssh[4:] = bytearray(b"pssh")
+                new_pssh[8:] = [0, 0, 0, 0]
+                new_pssh[13:] = self.WV_SYSTEM_ID
+                new_pssh[29:] = [0, 0, 0, 0]
+                new_pssh[31] = len(pssh)
+                new_pssh[32:] = pssh
+                return base64.b64encode(new_pssh)
+            else:
+                return pssh_b64
+
+        self.session = self.cdm.open_session(
+            check_pssh(self.init_data_b64), deviceconfig.DeviceConfig(self.device)
+        )
+
+        if self.cert_data_b64:
+            self.cdm.set_service_certificate(self.session, self.cert_data_b64)
+
+    def log_message(self, msg):
+        return "{}".format(msg)
+
+    def start_process(self):
+        keyswvdecrypt = []
+        try:
+            for key in self.cdm.get_keys(self.session):
+                if key.type == "CONTENT":
+                    keyswvdecrypt.append(
+                        self.log_message("{}:{}".format(key.kid.hex(), key.key.hex()))
+                    )
+        except Exception:
+            return False, keyswvdecrypt
+
+        return True, keyswvdecrypt
+    def get_challenge(self):
+        return self.cdm.get_license_request(self.session)
+
+    def update_license(self, license_b64):
+        self.cdm.provide_license(self.session, license_b64)
+        return True
+