refactor: dedicated tester

tester.py

@@ -0,0 +1,60 @@
+from typing import Dict
+
+from helper import *
+from config import config
+
+from model.settings import Settings
+from log import setup_logging
+from supermega import start
+
+
+def main():
+    """Argument parsing for when called from command line"""
+    logger.info("Super Mega")
+    config.load()
+
+    settings = Settings()
+    settings.payload_path = "data/shellcodes/createfile.bin"
+    settings.verify = True
+    settings.try_start_final_infected_exe = False
+    
+    # 7z, peb-walk, change-entrypoint
+    settings.source_style = SourceStyle.peb_walk
+    settings.inject_mode = InjectStyle.ChangeEntryPoint
+    settings.inject_exe_in = "data/exes/7z.exe"
+    settings.inject_exe_out = "data/exes/7z-verify.exe"
+    if start(settings) != 0:
+        print("Error")
+        return 1
+
+    # 7z, peb-walk, hijack
+    settings.source_style = SourceStyle.peb_walk
+    settings.inject_mode = InjectStyle.BackdoorCallInstr
+    settings.inject_exe_in = "data/exes/7z.exe"
+    settings.inject_exe_out = "data/exes/7z-verify.exe"
+    if start(settings) != 0:
+        print("Error")
+        return 1
+
+    # procexp, iat-reuse, change-entrypoint
+    settings.source_style = SourceStyle.iat_reuse
+    settings.inject_mode = InjectStyle.ChangeEntryPoint
+    settings.inject_exe_in = "data/exes/procexp64.exe"
+    settings.inject_exe_out = "data/exes/procexp64-verify.exe"
+    if start(settings) != 0:
+        print("Error")
+        return 1
+
+    # procexp, iat-reuse, change-entrypoint
+    settings.source_style = SourceStyle.iat_reuse
+    settings.inject_mode = InjectStyle.ChangeEntryPoint
+    settings.inject_exe_in = "data/exes/procexp64.exe"
+    settings.inject_exe_out = "data/exes/procexp64-verify.exe"
+    if start(settings) != 0:
+        print("Error")
+        return 1
+
+
+if __name__ == "__main__":
+    setup_logging(level=logging.WARN)
+    main()