mirror of
https://github.com/frida/frida-python
synced 2026-06-08 14:16:17 +00:00
Ole André Vadla Ravnås
87 lines
2.9 KiB
Python
87 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import threading
|
|
|
|
|
|
def spawn(command_line, device_id = None):
|
|
return get_device_manager().get_device(device_id).spawn(command_line)
|
|
|
|
def resume(target, device_id = None):
|
|
get_device_manager().get_device(device_id).resume(target)
|
|
|
|
def kill(target, device_id = None):
|
|
get_device_manager().get_device(device_id).kill(target)
|
|
|
|
def attach(target, device_id = None):
|
|
return get_device_manager().get_device(device_id).attach(target)
|
|
|
|
def get_usb_device(timeout = 0):
|
|
return _get_device('tether', timeout)
|
|
|
|
def get_remote_device(timeout = 0):
|
|
return _get_device('remote', timeout)
|
|
|
|
def _get_device(type, timeout):
|
|
mgr = get_device_manager()
|
|
def find_usb_device():
|
|
usb_devices = [device for device in mgr.enumerate_devices() if device.type == type]
|
|
if len(usb_devices) > 0:
|
|
return usb_devices[0]
|
|
else:
|
|
return None
|
|
device = find_usb_device()
|
|
if device is None:
|
|
result = [None]
|
|
event = threading.Event()
|
|
def on_devices_changed():
|
|
result[0] = find_usb_device()
|
|
if result[0] is not None:
|
|
event.set()
|
|
mgr.on('changed', on_devices_changed)
|
|
device = find_usb_device()
|
|
if device is None:
|
|
event.wait(timeout)
|
|
device = result[0]
|
|
mgr.off('changed', on_devices_changed)
|
|
if device is None:
|
|
raise TimeoutError("timed out while waiting for device to appear")
|
|
return device
|
|
|
|
def shutdown():
|
|
get_device_manager()._manager.close()
|
|
|
|
class TimeoutError(Exception):
|
|
pass
|
|
|
|
|
|
global _device_manager
|
|
_device_manager = None
|
|
def get_device_manager():
|
|
global _device_manager
|
|
if _device_manager is None:
|
|
from . import core
|
|
try:
|
|
import _frida
|
|
except Exception as ex:
|
|
import colorama
|
|
from colorama import Back, Fore, Style
|
|
import sys
|
|
colorama.init(autoreset=True)
|
|
print("")
|
|
print("***")
|
|
if str(ex).startswith("No module named "):
|
|
print(Back.RED + Fore.WHITE + Style.BRIGHT + "Frida native extension not found")
|
|
print(Fore.WHITE + Style.BRIGHT + "Please check your PYTHONPATH.")
|
|
else:
|
|
print(Back.RED + Fore.WHITE + Style.BRIGHT + "Failed to load the Frida native extension: %s" % ex)
|
|
if sys.version_info[0] == 2:
|
|
current_python_version = "%d.%d" % sys.version_info[:2]
|
|
else:
|
|
current_python_version = "%d.x" % sys.version_info[0]
|
|
print(Fore.WHITE + Style.BRIGHT + "Please ensure that the extension was compiled for Python " + current_python_version + ".")
|
|
print("***")
|
|
print("")
|
|
raise ex
|
|
_device_manager = core.DeviceManager(_frida.DeviceManager())
|
|
return _device_manager
|