qiling/tests/test_perf.py

#!/usr/bin/env python3
# 
# Cross Platform and Multi Architecture Advanced Binary Emulation Framework
#

import cProfile
import pstats
import sys
import inspect
import os as pyos

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 pyos.path.isdir("perf_results"):
        pyos.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()