diff --git a/helper.py b/helper.py index 7ba6748..30ed44a 100644 --- a/helper.py +++ b/helper.py @@ -162,7 +162,8 @@ def obfuscate_shc_loader(file_shc_in, file_shc_out): print("Error") return else: - print(" > Generated main-clean-sgn.bin") + print(" > Success obfuscation") + pass def test_shellcode(shc_name): @@ -191,8 +192,10 @@ def verify_shellcode(shc_name): # better to remove it immediately. If cleanup on start is not performed, # there may be false positives os.remove(verify_filename) + return True else: print("---> Verify FAIL. Payload did not create file.") + return False def inject_exe(shc_file, exe_in, exe_out): diff --git a/supermega.py b/supermega.py index 75614c3..c350008 100644 --- a/supermega.py +++ b/supermega.py @@ -58,7 +58,7 @@ options_verify = { "inject_exe": True, "inject_exe_in": "exes/procexp64.exe", - "inject_exe_out": "exes/procexp64-a.exe", + "inject_exe_out": "out/procexp64-a.exe", "alloc_style": AllocStyle.RWX, "exec_style": ExecStyle.CALL, @@ -121,6 +121,10 @@ def main(): print("--[ Test Append shellcode ]") test_shellcode("main-clean-append.bin") + # copy it to out + shutil.copyfile("main-clean-append.bin", os.path.join("out/", "main-clean-append.bin")) + + if options["inject_exe"]: inject_exe("main-clean-append.bin", options["inject_exe_in"], options["inject_exe_out"]) if options["verify"]: