mirror of
https://github.com/frida/frida-python
synced 2026-06-08 14:16:17 +00:00
Use ConsoleApplication for discoverer and tracer
This commit is contained in:
Ole André Vadla Ravnås
+15
-66
@@ -149,72 +149,29 @@ class UI(object):
|
||||
|
||||
|
||||
def main():
|
||||
from optparse import OptionParser
|
||||
import sys
|
||||
from frida.application import ConsoleApplication
|
||||
|
||||
import colorama
|
||||
from colorama import Fore, Back, Style
|
||||
class DiscovererApplication(ConsoleApplication, UI):
|
||||
def _usage(self):
|
||||
return "usage: %prog [options] process-name-or-id"
|
||||
|
||||
import frida
|
||||
from frida.core import Reactor
|
||||
|
||||
colorama.init(autoreset=True)
|
||||
|
||||
usage = "usage: %prog [options] process-name-or-id"
|
||||
parser = OptionParser(usage=usage)
|
||||
(options, args) = parser.parse_args()
|
||||
if len(args) != 1:
|
||||
parser.error("process name or id must be specified")
|
||||
try:
|
||||
target = int(args[0])
|
||||
except:
|
||||
target = args[0]
|
||||
|
||||
class Application(UI):
|
||||
def __init__(self, target):
|
||||
self._target = target
|
||||
self._process = None
|
||||
def _initialize(self, parser, options, args):
|
||||
self._discoverer = None
|
||||
self._status_updated = False
|
||||
self._exit_status = 0
|
||||
self._reactor = Reactor(await_enter)
|
||||
self._reactor.schedule(self._start)
|
||||
|
||||
def run(self):
|
||||
self._reactor.run()
|
||||
self._stop()
|
||||
return self._exit_status
|
||||
def _target_specifier(self, parser, options, args):
|
||||
if len(args) != 1:
|
||||
parser.error("process name or id must be specified")
|
||||
return args[0]
|
||||
|
||||
def _start(self):
|
||||
try:
|
||||
self._update_status("Attaching...")
|
||||
self._process = frida.attach(self._target)
|
||||
except Exception as e:
|
||||
self._update_status("Failed to attach: %s" % e)
|
||||
self._exit_status = 1
|
||||
self._reactor.schedule(self._stop)
|
||||
return
|
||||
self._update_status("Injecting script...")
|
||||
self._discoverer = Discoverer(self._reactor)
|
||||
self._discoverer.start(self._process, self)
|
||||
|
||||
def _stop(self):
|
||||
if self._discoverer is not None:
|
||||
print("Stopping...")
|
||||
self._discoverer.stop()
|
||||
self._discoverer = None
|
||||
if self._process is not None:
|
||||
self._process.detach()
|
||||
self._process = None
|
||||
self._reactor.stop()
|
||||
|
||||
def _update_status(self, message):
|
||||
if self._status_updated:
|
||||
cursor_position = "\033[A"
|
||||
else:
|
||||
cursor_position = ""
|
||||
print("%-80s" % (cursor_position + Style.BRIGHT + message,))
|
||||
self._status_updated = True
|
||||
print("Stopping...")
|
||||
self._discoverer.stop()
|
||||
self._discoverer = None
|
||||
|
||||
def on_sample_progress(self, begin, end, total):
|
||||
self._update_status("Sampling %d threads: %d through %d..." % (total, begin, end))
|
||||
@@ -233,18 +190,10 @@ def main():
|
||||
for function, rate in sorted(dynamic_functions, key=lambda item: item[1], reverse=True):
|
||||
print("\t%-10d\t%s" % (rate, function))
|
||||
|
||||
self._reactor.schedule(self._stop)
|
||||
self._exit(0)
|
||||
|
||||
def await_enter():
|
||||
if sys.version_info[0] >= 3:
|
||||
input()
|
||||
else:
|
||||
raw_input()
|
||||
|
||||
app = Application(target)
|
||||
status = app.run()
|
||||
frida.shutdown()
|
||||
sys.exit(status)
|
||||
app = DiscovererApplication()
|
||||
app.run()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
+28
-79
@@ -2,7 +2,6 @@
|
||||
|
||||
import os
|
||||
import fnmatch
|
||||
import sys
|
||||
import time
|
||||
|
||||
from frida.core import ModuleFunction
|
||||
@@ -413,65 +412,36 @@ class UI(object):
|
||||
|
||||
|
||||
def main():
|
||||
from optparse import OptionParser
|
||||
import sys
|
||||
from frida.application import ConsoleApplication
|
||||
|
||||
import colorama
|
||||
from colorama import Fore, Back, Style
|
||||
class TracerApplication(ConsoleApplication, UI):
|
||||
def _add_options(self, parser):
|
||||
pb = TracerProfileBuilder()
|
||||
def process_builder_arg(option, opt_str, value, parser, method, **kwargs):
|
||||
method(value)
|
||||
parser.add_option("-I", "--include-module=MODULE", help="include MODULE", metavar="MODULE",
|
||||
type='string', action='callback', callback=process_builder_arg, callback_args=(pb.include_modules,))
|
||||
parser.add_option("-X", "--exclude-module=MODULE", help="exclude MODULE", metavar="MODULE",
|
||||
type='string', action='callback', callback=process_builder_arg, callback_args=(pb.exclude_modules,))
|
||||
parser.add_option("-i", "--include=FUNCTION", help="include FUNCTION", metavar="FUNCTION",
|
||||
type='string', action='callback', callback=process_builder_arg, callback_args=(pb.include,))
|
||||
parser.add_option("-x", "--exclude=FUNCTION", help="exclude FUNCTION", metavar="FUNCTION",
|
||||
type='string', action='callback', callback=process_builder_arg, callback_args=(pb.exclude,))
|
||||
self._profile_builder = pb
|
||||
|
||||
import frida
|
||||
from frida.core import Reactor
|
||||
def _usage(self):
|
||||
return "usage: %prog [options] process-name-or-id"
|
||||
|
||||
colorama.init(autoreset=True)
|
||||
|
||||
tp = TracerProfileBuilder()
|
||||
def process_builder_arg(option, opt_str, value, parser, method, **kwargs):
|
||||
method(value)
|
||||
|
||||
usage = "usage: %prog [options] process-name-or-id"
|
||||
parser = OptionParser(usage=usage)
|
||||
parser.add_option("-I", "--include-module=MODULE", help="include MODULE", metavar="MODULE",
|
||||
type='string', action='callback', callback=process_builder_arg, callback_args=(tp.include_modules,))
|
||||
parser.add_option("-X", "--exclude-module=MODULE", help="exclude MODULE", metavar="MODULE",
|
||||
type='string', action='callback', callback=process_builder_arg, callback_args=(tp.exclude_modules,))
|
||||
parser.add_option("-i", "--include=FUNCTION", help="include FUNCTION", metavar="FUNCTION",
|
||||
type='string', action='callback', callback=process_builder_arg, callback_args=(tp.include,))
|
||||
parser.add_option("-x", "--exclude=FUNCTION", help="exclude FUNCTION", metavar="FUNCTION",
|
||||
type='string', action='callback', callback=process_builder_arg, callback_args=(tp.exclude,))
|
||||
(options, args) = parser.parse_args()
|
||||
if len(args) != 1:
|
||||
parser.error("process name or id must be specified")
|
||||
try:
|
||||
target = int(args[0])
|
||||
except:
|
||||
target = args[0]
|
||||
profile = tp.build()
|
||||
|
||||
class Application(UI):
|
||||
def __init__(self, target, profile):
|
||||
self._target = target
|
||||
self._process = None
|
||||
def _initialize(self, parser, options, args):
|
||||
self._tracer = None
|
||||
self._profile = profile
|
||||
self._status_updated = False
|
||||
self._exit_status = 0
|
||||
self._reactor = Reactor(await_enter)
|
||||
self._reactor.schedule(self._start)
|
||||
self._profile = self._profile_builder.build()
|
||||
|
||||
def run(self):
|
||||
self._reactor.run()
|
||||
self._stop()
|
||||
return self._exit_status
|
||||
def _target_specifier(self, parser, options, args):
|
||||
if len(args) != 1:
|
||||
parser.error("process name or id must be specified")
|
||||
return args[0]
|
||||
|
||||
def _start(self):
|
||||
try:
|
||||
self._update_status("Attaching...")
|
||||
self._process = frida.attach(self._target)
|
||||
except Exception as e:
|
||||
self._update_status("Failed to attach: %s" % e)
|
||||
self._exit_status = 1
|
||||
self._reactor.schedule(self._stop)
|
||||
return
|
||||
self._tracer = Tracer(self._reactor, FileRepository(), self._profile)
|
||||
targets = self._tracer.start_trace(self._process, self)
|
||||
if len(targets) == 1:
|
||||
@@ -481,14 +451,9 @@ def main():
|
||||
self._update_status("Started tracing %d function%s. Press ENTER to stop." % (len(targets), plural))
|
||||
|
||||
def _stop(self):
|
||||
if self._tracer is not None:
|
||||
print("Stopping...")
|
||||
self._tracer.stop()
|
||||
self._tracer = None
|
||||
if self._process is not None:
|
||||
self._process.detach()
|
||||
self._process = None
|
||||
self._reactor.stop()
|
||||
print("Stopping...")
|
||||
self._tracer.stop()
|
||||
self._tracer = None
|
||||
|
||||
def on_trace_progress(self, operation):
|
||||
if operation == 'resolve':
|
||||
@@ -509,24 +474,8 @@ def main():
|
||||
def on_trace_handler_load(self, function, handler, source):
|
||||
print("%s: Loaded handler at \"%s\"" % (function, source))
|
||||
|
||||
def _update_status(self, message):
|
||||
if self._status_updated:
|
||||
cursor_position = "\033[A"
|
||||
else:
|
||||
cursor_position = ""
|
||||
print("%-80s" % (cursor_position + Style.BRIGHT + message,))
|
||||
self._status_updated = True
|
||||
|
||||
def await_enter():
|
||||
if sys.version_info[0] >= 3:
|
||||
input()
|
||||
else:
|
||||
raw_input()
|
||||
|
||||
app = Application(target, profile)
|
||||
status = app.run()
|
||||
frida.shutdown()
|
||||
sys.exit(status)
|
||||
app = TracerApplication()
|
||||
app.run()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
Reference in New Issue
Block a user