From 0ffd0aa1aac84d2c389dba3c0bce0b5646e4cea5 Mon Sep 17 00:00:00 2001
From: klks <klks84@gmail.com>
Date: Thu, 9 Apr 2020 20:42:59 -0700
Subject: [PATCH] Adding simpler performance test/profiling

---
 .gitignore         |  1 +
 tests/test_perf.py | 53 ++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 54 insertions(+)
 create mode 100644 tests/test_perf.py

diff --git a/.gitignore b/.gitignore
index a913db15..2bb66d41 100644
--- a/.gitignore
+++ b/.gitignore
@@ -30,6 +30,7 @@ test.file
 *.d
 *.o
 core
+*.perf
 
 # dont keep raw malware
 wannacry.bin
diff --git a/tests/test_perf.py b/tests/test_perf.py
new file mode 100644
index 00000000..d705e420
--- /dev/null
+++ b/tests/test_perf.py
@@ -0,0 +1,53 @@
+#!/usr/bin/env python3
+# 
+# Cross Platform and Multi Architecture Advanced Binary Emulation Framework
+# Built on top of Unicorn emulator (www.unicorn-engine.org) 
+import cProfile
+import pstats
+import sys
+import inspect
+import os
+sys.path.append("..")
+from qiling import *
+from test_elf import *
+from test_macho import *
+from test_shellcode import *
+perf_res_dir = "./perf_results/"
+
+test_mapping = [
+
+]
+
+def populate_tests():
+    global test_mapping
+
+    unit_tests = [ELFTest(), MACHOTest(), TestShellcode()]
+
+    for ut in unit_tests:
+        ut_functions = inspect.getmembers(ut, predicate = inspect.ismethod)
+        for test_name, test_fn in ut_functions:
+            if not test_name.startswith("test_"): continue
+            outfile = perf_res_dir + test_name + ".perf"
+            test_mapping.append( (test_fn, outfile) )
+    
+def ql_profile(run_fn, outfile):
+    pr = cProfile.Profile()
+    pr.enable()
+    run_fn()
+    pr.disable()
+    pr.dump_stats(outfile)
+    pr.print_stats()
+
+def profile_all_functions():
+    if not os.path.isdir("perf_results"): os.mkdir("perf_results")
+    
+    populate_tests()
+
+    for tm_func, rm_outfile in test_mapping:
+        try:
+            ql_profile(tm_func, rm_outfile)
+        except:
+            pass
+
+if __name__ == "__main__":
+    profile_all_functions()
\ No newline at end of file