mirror of
https://github.com/frida/frida-python
synced 2026-06-08 14:16:17 +00:00
Ole André Vadla Ravnås
98733a3388
Because `/bin/cat` is now considered a system program we are no longer allowed to attach to it using `task_for_pid()`. Bundle our own test program for Mac as a workaround.
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from .data import target_program
|
|
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()
|
|
cls.target = subprocess.Popen([target_program])
|
|
cls.session = frida.attach(cls.target.pid)
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
cls.session.detach()
|
|
cls.target.terminate()
|
|
|
|
def test_basics(self):
|
|
test_ui = TestUI()
|
|
reactor = Reactor(lambda reactor: test_ui.on_result.wait())
|
|
def start():
|
|
d = Discoverer(reactor)
|
|
d.start(self.session, test_ui)
|
|
reactor.schedule(d.stop, 0.1)
|
|
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()
|