Clean import

This commit is contained in:
hyugogirubato 2024-04-01 13:17:28 +02:00
parent 539a895541
commit b59551c708
1 changed files with 4 additions and 2 deletions

View File

@ -7,7 +7,7 @@ from pathlib import Path
import frida
import xmltodict
from _frida import Process
from frida.core import Device, Session, Script, RPCException
from frida.core import Device, Session, Script
from Cryptodome.PublicKey import RSA
from extractor.license_protocol_pb2 import SignedMessage, LicenseRequest, ClientIdentification, DrmCertificate, SignedDrmCertificate
@ -89,7 +89,9 @@ class Cdm:
name = func['@NAME']
# Add symbol if it matches specific criteria
if any(keyword in name for keyword in ['UsePrivacyMode', 'PrepareKeyRequest']) or name == target or (not target and re.match(r'^[a-z]+$', name)):
if any(keyword in name for keyword in
['UsePrivacyMode', 'PrepareKeyRequest']) or name == target or (
not target and re.match(r'^[a-z]+$', name)):
addr = hex(int(func['@ADDRESS'], 16) - base_addr)
symbols[addr] = {'name': name, 'address': addr}
except Exception: