refactor: new data/ structure

This commit is contained in:
Dobin
2024-03-27 20:13:14 +00:00
parent f08334dc1a
commit 4064cf94ba
33 changed files with 151 additions and 52 deletions
+11 -11
View File
@@ -2,6 +2,7 @@ from typing import Dict
from helper import *
from config import config
from model.defs import *
from model.settings import Settings
from log import setup_logging
@@ -9,20 +10,19 @@ from supermega import start
def main():
"""Argument parsing for when called from command line"""
logger.info("Super Mega")
logger.info("Super Mega Tester")
config.load()
settings = Settings()
settings.payload_path = "data/shellcodes/createfile.bin"
settings.payload_path = PATH_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"
settings.inject_exe_in = PATH_EXES + "7z.exe"
settings.inject_exe_out = PATH_EXES + "7z-verify.exe"
if start(settings) != 0:
print("Error")
return 1
@@ -30,8 +30,8 @@ def main():
# 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"
settings.inject_exe_in = PATH_EXES + "7z.exe"
settings.inject_exe_out = PATH_EXES + "7z-verify.exe"
if start(settings) != 0:
print("Error")
return 1
@@ -39,8 +39,8 @@ def main():
# 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"
settings.inject_exe_in = PATH_EXES + "procexp64.exe"
settings.inject_exe_out = PATH_EXES + "procexp64-verify.exe"
if start(settings) != 0:
print("Error")
return 1
@@ -48,8 +48,8 @@ def main():
# 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"
settings.inject_exe_in = PATH_EXES + "procexp64.exe"
settings.inject_exe_out = PATH_EXES + "procexp64-verify.exe"
if start(settings) != 0:
print("Error")
return 1