Android-Tools/KeyDive
1
0
Fork 0
mirror of https://github.com/hyugogirubato/KeyDive.git synced 2024-12-22 11:04:26 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Improve library vendor detection

Browse Source
This commit is contained in:
hyugogirubato 2024-07-07 16:27:47 +02:00
parent 405c184558
commit 4bcabf1101
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
keydive/__main__.py
View File

@ -107,6 +107,17 @@ def main() -> None:
# Initialize Core instance for interacting with the device
core = Core(cdm=cdm, device=args.device, functions=args.functions)
# Map process keys to their compatible CDM vendors
cdm_vendor = {}
for key, vendors in CDM_VENDOR_API.items():
for vendor in vendors:
# Check if vendor's SDK matches the core's SDK or the previous version
if vendor.sdk in (core.sdk_api, core.sdk_api - 1):
cdm_vendor.setdefault(key, []).append(vendor)
if not cdm_vendor:
raise NotImplementedError('SDK version is not supported.')
# Process watcher loop
logger.info('Watcher delay: %ss' % args.delay)
current = None
@ -115,7 +126,7 @@ def main() -> None:
processes = {
key: (name, pid)
for name, pid in core.enumerate_processes().items()
for key in CDM_VENDOR_API.keys() if key in name
for key in cdm_vendor.keys() if key in name
}
if not processes:
@ -136,7 +147,7 @@ def main() -> None:
for key, (name, pid) in processes.items():
if current:
break
for vendor in CDM_VENDOR_API[key]:
for vendor in cdm_vendor[key]:
if core.hook_process(pid=pid, vendor=vendor):
logger.info('Process: %s (%s)', pid, name)
current = pid