mirror of
https://github.com/frida/frida-python
synced 2026-06-08 14:16:17 +00:00
Ole André Vadla Ravnås
6284b97c37
This introduces the -U/--usb switch for using a USB-tethered device instead of the local system. Only iOS devices are supported for now (PR for completing Android support in frida-core is most appreciated!).
57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import platform
|
|
import subprocess
|
|
import threading
|
|
try:
|
|
import unittest2 as unittest
|
|
except:
|
|
import unittest
|
|
|
|
import frida
|
|
from frida.application import Reactor
|
|
from frida.discoverer import Discoverer, UI
|
|
|
|
|
|
class TestDiscoverer(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
system = platform.system()
|
|
if system == 'Windows':
|
|
cls.target = subprocess.Popen([r"C:\Windows\notepad.exe"])
|
|
else:
|
|
cls.target = subprocess.Popen(["/bin/cat"])
|
|
cls.process = frida.attach(cls.target.pid)
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
cls.process.detach()
|
|
cls.target.terminate()
|
|
|
|
def test_basics(self):
|
|
test_ui = TestUI()
|
|
reactor = Reactor(test_ui.on_result.wait)
|
|
def start():
|
|
d = Discoverer(reactor)
|
|
d.start(self.process, test_ui)
|
|
reactor.schedule(start)
|
|
reactor.run()
|
|
self.assertIsInstance(test_ui.module_functions, dict)
|
|
self.assertIsInstance(test_ui.dynamic_functions, list)
|
|
|
|
class TestUI(UI):
|
|
def __init__(self):
|
|
super(UI, self).__init__()
|
|
self.module_functions = None
|
|
self.dynamic_functions = None
|
|
self.on_result = threading.Event()
|
|
|
|
def on_sample_result(self, module_functions, dynamic_functions):
|
|
self.module_functions = module_functions
|
|
self.dynamic_functions = dynamic_functions
|
|
self.on_result.set()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|