Dump-Tools
/
KeyDive
mirror of https://github.com/hyugogirubato/KeyDive.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add ADB env check + auto start

This commit is contained in:
hyugogirubato 2024-04-01 12:24:00 +02:00
parent 3910b2b531
commit 262efe8c0c
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
extractor/cdm.py
View File

@ -71,6 +71,10 @@ class Cdm:
""" """
symbols = {} symbols = {}
if path: if path:
# Verify symbols file path
if not path.is_file():
raise FileNotFoundError('Symbols file not found')
try: try:
# Parse the XML file # Parse the XML file
program = xmltodict.parse(path.read_bytes())['PROGRAM'] program = xmltodict.parse(path.read_bytes())['PROGRAM']

10
keydive.py
View File

@ -1,5 +1,6 @@
import argparse import argparse
import logging import logging
import subprocess
import time import time
import coloredlogs import coloredlogs
@ -23,12 +24,13 @@ if __name__ == '__main__':
args = parser.parse_args() args = parser.parse_args()
try: try:
symbols = args.symbols # Start ADB server
if symbols and not symbols.is_file(): exitcode, _ = subprocess.getstatusoutput('adb start-server')
raise FileNotFoundError('Symbols file not found') if exitcode != 0:
raise EnvironmentError('ADB is not recognized as an environment variable')
# Initialize CDM handler with given device # Initialize CDM handler with given device
cdm = Cdm(device=args.device, symbols=symbols) cdm = Cdm(device=args.device, symbols=args.symbols)
# Find Widevine process on the device # Find Widevine process on the device
process: Process = next((p for p in cdm.device.enumerate_processes() if cdm.vendor.process == p.name), None) process: Process = next((p for p in cdm.device.enumerate_processes() if cdm.vendor.process == p.name), None)