admin/frida-frida-python
1
0
Fork 0
mirror of https://github.com/frida/frida-python synced 2026-06-08 14:16:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add %time command

Browse Source
This commit is contained in:
Asger Hautop Drewsen
2015-06-17 16:15:09 +02:00
parent 806fff0df3
commit 2d13844563
1 changed files with 27 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/frida/repl.py
+27 -15
View File
@@ -147,21 +147,24 @@ def main():
elif expression == "help":
print("Help: #TODO :)")
else:
try:
(t, value) = self._evaluate(expression)
if t in ('function', 'undefined', 'null'):
output = t
elif t == 'binary':
output = hexdump(value).rstrip("\n")
else:
output = json.dumps(value, sort_keys=True, indent=4, separators=(",", ": "))
except frida.InvalidOperationError:
return
except Exception as ex:
error = ex.message
output = Fore.RED + Style.BRIGHT + error['name'] + Style.RESET_ALL + ": " + error['message']
sys.stdout.write(output + "\n")
sys.stdout.flush()
self._eval_and_print(expression)
def _eval_and_print(self, expression):
try:
(t, value) = self._evaluate(expression)
if t in ('function', 'undefined', 'null'):
output = t
elif t == 'binary':
output = hexdump(value).rstrip("\n")
else:
output = json.dumps(value, sort_keys=True, indent=4, separators=(",", ": "))
except frida.InvalidOperationError:
return
except Exception as ex:
error = ex.message
output = Fore.RED + Style.BRIGHT + error['name'] + Style.RESET_ALL + ": " + error['message']
sys.stdout.write(output + "\n")
sys.stdout.flush()
def _print_startup_message(self):
print(""" _____
@@ -230,6 +233,15 @@ def main():
elif command == 'unload' and len(args) == 0:
self._user_script = None
self._reload()
elif command == 'time' and len(args) > 0:
self._eval_and_print('''
(function() {{
var _startTime = new Date().getTime();
var _result = {expression};
var _endTime = new Date().getTime();
console.log('Time: ' + (_endTime - _startTime).toLocaleString() + ' ms.');
return _result;
}})();'''.format(expression=' '.join(args)))
else:
print("Unknown command: {command}".format(command=command))