Adding simpler performance test/profiling

.gitignore

@@ -30,6 +30,7 @@ test.file
 *.d
 *.o
 core
+*.perf
 
 # dont keep raw malware
 wannacry.bin

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()