mirror of
https://github.com/frida/frida-python
synced 2026-06-08 14:16:17 +00:00
Ole André Vadla Ravnås
66bdd0ed30
This way it is automatically cancellable, and we get the performance benefit of not having to wait for all backends to start.
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import sys
|
|
import threading
|
|
import time
|
|
try:
|
|
import unittest2 as unittest
|
|
except:
|
|
import unittest
|
|
|
|
import frida
|
|
|
|
|
|
class TestCore(unittest.TestCase):
|
|
def test_enumerate_devices(self):
|
|
devices = frida.get_device_manager().enumerate_devices()
|
|
self.assertTrue(len(devices) > 0)
|
|
|
|
def test_get_existing_device(self):
|
|
device = frida.get_device_matching(lambda d: d.id == 'local')
|
|
self.assertEqual(device.name, "Local System")
|
|
|
|
device = frida.get_device_manager().get_device_matching(lambda d: d.id == 'local')
|
|
self.assertEqual(device.name, "Local System")
|
|
|
|
def test_get_nonexistent_device(self):
|
|
def get_nonexistent():
|
|
frida.get_device_manager().get_device_matching(lambda device: device.type == 'lol')
|
|
self.assertRaisesMatching(frida.InvalidArgumentError, "device not found", get_nonexistent)
|
|
|
|
def test_wait_for_nonexistent_device(self):
|
|
def wait_for_nonexistent():
|
|
frida.get_device_manager().get_device_matching(lambda device: device.type == 'lol', timeout=0.1)
|
|
self.assertRaisesMatching(frida.InvalidArgumentError, "device not found", wait_for_nonexistent)
|
|
|
|
def test_cancel_wait_for_nonexistent_device(self):
|
|
cancellable = frida.Cancellable()
|
|
|
|
def wait_for_nonexistent():
|
|
frida.get_device_manager().get_device_matching(lambda device: device.type == 'lol', timeout=-1, cancellable=cancellable)
|
|
|
|
def cancel_after_100ms():
|
|
time.sleep(0.1)
|
|
cancellable.cancel()
|
|
|
|
threading.Thread(target=cancel_after_100ms).start()
|
|
self.assertRaisesMatching(frida.OperationCancelledError, "operation was cancelled", wait_for_nonexistent)
|
|
|
|
def assertRaisesMatching(self, exception, regex, operation):
|
|
m = self.assertRaisesRegex if sys.version_info[0] >= 3 else self.assertRaisesRegexp
|
|
m(exception, regex, operation)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|