diff --git a/app/storage.py b/app/storage.py index f535230..1344dc6 100644 --- a/app/storage.py +++ b/app/storage.py @@ -22,7 +22,6 @@ class Storage(): project = self.get_project(project_name) if project is None: continue - project.settings.prep_web() projects.append(project) return projects @@ -35,7 +34,6 @@ class Storage(): return None with open(json_path, "rb") as f: project = pickle.load(f) - project.settings.prep_web() return project diff --git a/model/settings.py b/model/settings.py index 9e169e3..a9af17d 100644 --- a/model/settings.py +++ b/model/settings.py @@ -34,8 +34,7 @@ class Settings(): self.fix_missing_iat = False self.payload_location = PayloadLocation.DATA - - def prep_web(self): + # directories and filenames self.main_dir = "{}{}/".format(PATH_WEB_PROJECT, self.project_name) self.main_c_path = self.main_dir + "main.c" self.main_asm_path = self.main_dir + "main.asm" diff --git a/supermega.py b/supermega.py index 3d3f735..2818291 100644 --- a/supermega.py +++ b/supermega.py @@ -88,7 +88,6 @@ def main(): settings.inject_exe_in = args.inject settings.inject_exe_out = args.inject.replace(".exe", ".infected.exe").replace(".dll", ".infected.dll") - settings.prep_web() write_webproject("default", settings) exit_code = start(settings) exit(exit_code) diff --git a/tester.py b/tester.py index 9eeb4f5..db44e82 100644 --- a/tester.py +++ b/tester.py @@ -27,7 +27,6 @@ def test_exe_code(): settings.verify = True settings.try_start_final_infected_exe = False settings.payload_location = PayloadLocation.CODE - settings.prep_web() # 7z, peb-walk, change-entrypoint print("Test EXE 1/4: 7z, peb-walk, change-entrypoint") @@ -73,7 +72,6 @@ def test_exe_data(): settings.verify = True settings.try_start_final_infected_exe = False settings.payload_location = PayloadLocation.DATA - settings.prep_web() # 7z, peb-walk, change-entrypoint print("Test EXE 1/4: 7z, peb-walk, change-entrypoint") @@ -119,7 +117,6 @@ def test_dll_code(): settings.verify = True settings.try_start_final_infected_exe = False settings.payload_location = PayloadLocation.CODE - settings.prep_web() print("Test DLL 1/6: libbz2-1.dll, peb-walk, change-entrypoint dllMain (func=None)") settings.carrier_name = "peb_walk" @@ -163,7 +160,6 @@ def test_dll_data(): settings.verify = True settings.try_start_final_infected_exe = False settings.payload_location = PayloadLocation.DATA - settings.prep_web() print("Test DLL 1/6: libbz2-1.dll, peb-walk, change-entrypoint dllMain (func=None)") settings.carrier_name = "peb_walk"