diff --git a/supermega.py b/supermega.py index 50f5d76..b1102be 100644 --- a/supermega.py +++ b/supermega.py @@ -52,7 +52,10 @@ def main(): project.inject_exe_out = "out/iatttest-full-a.exe" else: print("Unknown verify option {}, use std/iat".format(args.verify)) + else: + project.try_start_final_infected_exe = True + if args.shellcode: if not os.path.isfile(args.shellcode): print("Could not find: {}".format(args.shellcode)) @@ -65,6 +68,7 @@ def main(): project.inject = True project.inject_exe_in = args.inject project.inject_exe_out = args.inject.replace(".exe", ".infected.exe") + start()