52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
import argparse
|
|
import logging
|
|
import time
|
|
|
|
import coloredlogs
|
|
from _frida import Process
|
|
from pathlib import Path
|
|
|
|
from extractor.cdm import Cdm
|
|
|
|
coloredlogs.install(
|
|
fmt='%(asctime)s [%(levelname).1s] %(name)s: %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S',
|
|
level=logging.DEBUG)
|
|
|
|
if __name__ == '__main__':
|
|
logger = logging.getLogger('KeyDive')
|
|
|
|
# Parse command line arguments for device ID
|
|
parser = argparse.ArgumentParser(description='Extract Widevine L3 keys from an Android device.')
|
|
parser.add_argument('--device', required=False, type=str, help='Target Android device ID.')
|
|
parser.add_argument('--symbols', required=False, type=Path, help='Ghidra XML symbols file.')
|
|
args = parser.parse_args()
|
|
|
|
try:
|
|
symbols = args.symbols
|
|
if symbols and not symbols.is_file():
|
|
raise FileNotFoundError('Symbols file not found')
|
|
|
|
# Initialize CDM handler with given device
|
|
cdm = Cdm(device=args.device, symbols=symbols)
|
|
|
|
# Find Widevine process on the device
|
|
process: Process = next((p for p in cdm.device.enumerate_processes() if cdm.vendor.process == p.name), None)
|
|
if not process:
|
|
raise Exception('Failed to find Widevine process')
|
|
logger.info('Process: %s (%s)', process.pid, process.name)
|
|
|
|
# Hook into the process to extract DRM keys
|
|
if not cdm.hook_process(process):
|
|
raise Exception('Failed to hook into the process')
|
|
logger.info('Successfully hooked. To test, play a DRM-protected video: https://bitmovin.com/demos/drm')
|
|
|
|
# Keep script running while extracting keys
|
|
while cdm.running:
|
|
time.sleep(1)
|
|
except KeyboardInterrupt:
|
|
pass
|
|
except Exception as e:
|
|
logger.critical(e)
|
|
logger.info('Exiting')
|