diff --git a/keydive/__main__.py b/keydive/__main__.py index 43a5af9..52d6f9b 100644 --- a/keydive/__main__.py +++ b/keydive/__main__.py @@ -2,12 +2,14 @@ import argparse import logging import subprocess import time + from datetime import datetime from pathlib import Path import coloredlogs import keydive + from keydive.cdm import Cdm from keydive.constants import CDM_VENDOR_API from keydive.core import Core diff --git a/keydive/cdm.py b/keydive/cdm.py index 511b491..f0cdbe2 100644 --- a/keydive/cdm.py +++ b/keydive/cdm.py @@ -2,16 +2,17 @@ import base64 import json import logging import re + from pathlib import Path from typing import Union from zlib import crc32 +from unidecode import unidecode from Cryptodome.PublicKey import RSA from Cryptodome.PublicKey.RSA import RsaKey from pywidevine.device import Device, DeviceTypes from pywidevine.license_protocol_pb2 import (SignedMessage, LicenseRequest, ClientIdentification, SignedDrmCertificate, DrmCertificate, EncryptedClientIdentification) -from unidecode import unidecode def sanitize(path: Path) -> Path: diff --git a/keydive/core.py b/keydive/core.py index a4f0ac7..6cfaf50 100644 --- a/keydive/core.py +++ b/keydive/core.py @@ -2,10 +2,12 @@ import json import logging import re import subprocess + from pathlib import Path import frida import xmltodict + from frida.core import Device, Session, Script from keydive.cdm import Cdm