From 3e6c1e06cf6dc880fbe7e3b43998f477b4a7fd21 Mon Sep 17 00:00:00 2001 From: Dobin Date: Mon, 29 Apr 2024 19:24:29 +0100 Subject: [PATCH] feature: keep exe and dll files in project folder --- model/project.py | 9 +++++++++ supermega.py | 2 +- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/model/project.py b/model/project.py index bd9c377..89e1ea4 100644 --- a/model/project.py +++ b/model/project.py @@ -46,6 +46,15 @@ def prepare_project(project_name, settings): continue if file.startswith("."): continue + if file.endswith(".exe"): + # keep all exes except: + if file != "main.exe" and not file.endswith(".infected.exe"): + continue + if file.endswith(".dll"): + # keep all dlls except: + if not file.endswith(".infected.dll"): + continue + os.remove(dst + file) # copy *.c *.h files from src directory to dst directory diff --git a/supermega.py b/supermega.py index 4bc0273..aa7246e 100644 --- a/supermega.py +++ b/supermega.py @@ -86,7 +86,7 @@ def main(): logger.info("Could not find: {}".format(args.inject)) return settings.inject_exe_in = args.inject - settings.inject_exe_out = args.inject.replace(".exe", ".infected.exe") + settings.inject_exe_out = args.inject.replace(".exe", ".infected.exe").replace(".dll", ".infected.dll") settings.prep_web("default") write_webproject("default", settings)