feature: keep exe and dll files in project folder

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

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)