admin/SuperMega
1
0
Fork 0
mirror of https://github.com/dobin/SuperMega synced 2026-06-02 17:27:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: better file cleanup

Browse Source
This commit is contained in:
Dobin
2024-02-04 08:40:17 +00:00
parent 9f7466d775
commit efe2039984
3 changed files with 23 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
supermega.py
+8 -1
View File
@@ -25,6 +25,9 @@ options_default = {
"verify": False,
"cleanup_files_on_start": True,
"cleanup_files_on_exit": True, # all is just in out/
"generate_asm_from_c": True,
"generate_shc_from_asm": True,
"test_loader_shellcode": False,
@@ -49,6 +52,8 @@ options_verify = {
"verify": True,
"cleanup_files_on_start": True,
"cleanup_files_on_exit": True, # all is just in out/
"generate_asm_from_c": True,
"generate_shc_from_asm": True,
"test_loader_shellcode": False,
@@ -124,12 +129,14 @@ def main():
# 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"]:
print("--[ Verify final exe ]")
verify_injected_exe(options["inject_exe_out"])
if options["cleanup_files_on_exit"]:
clean_files()
if __name__ == "__main__":
main()