def main(argv): files_dir = '' emi_dir = '' decompiler = '' EMI = False try: opts, args = getopt.getopt(argv, "", ["help", "EMI", "decompiler=", "files_dir=", "emi_dir="]) except getopt.GetoptError: print('reproduce.py \n' '--decompiler <decompiler name> \n' '--files_dir <directory of C programs> \n' '--emi_dir <where to store EMI variants> \n' '--EMI generate EMI variants') sys.exit(2) for opt, arg in opts: # print(opt, arg) if opt == '--help': print('reproduce.py \n' '--decompiler <decompiler name> \n' '--files_dir <directory of C programs> \n' '--emi_dir <where to store EMI variants> \n' '--EMI generate EMI variants') sys.exit() elif opt == "--EMI": EMI = True elif opt == "--files_dir": files_dir = arg elif opt == "--emi_dir": emi_dir = arg elif opt == "--decompiler": decompiler = arg Config.set_decompiler(decompiler) if EMI: pass fuzzer.seed_test_AE(files_dir, emi_dir, os.path.join(files_dir, 'config.txt')) else: pass fuzzer.emi_test_AE(files_dir, os.path.join(files_dir, 'config.txt'))
#!/usr/bin/python3 # -*- coding: UTF-8 -*- # import sys, getopt from src import fuzzer from src import Config Config.set_decompiler("retdec") # fuzzer.seed_test_AE('./seed_for_retdec/', './seed_for_retdec/emi/', './seed_for_retdec/config.txt') # fuzzer.emi_test_AE('./seed_for_retdec/emi/', './seed_for_retdec/emi/config.txt') Config.set_decompiler("r2") fuzzer.seed_test_AE('./seed_for_r2/', './seed_for_r2/emi/', './seed_for_r2/config.txt') fuzzer.emi_test_AE('./seed_for_r2/emi/', './seed_for_r2/emi/config.txt')