From 5a0e1f0fb9d54b69abb647e2c4fc367817f86ce8 Mon Sep 17 00:00:00 2001 From: Dobin Date: Thu, 15 Feb 2024 18:08:08 +0000 Subject: [PATCH] refactor: rename verify std to peb --- supermega.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/supermega.py b/supermega.py index 9861ea0..25755cc 100644 --- a/supermega.py +++ b/supermega.py @@ -22,7 +22,6 @@ main_exe_file = os.path.join(build_dir, "main.exe") main_shc_file = os.path.join(build_dir, "main.bin") - # ANSI escape sequences for colors class LogColors: HEADER = '\033[95m' @@ -34,7 +33,6 @@ class LogColors: BOLD = '\033[1m' UNDERLINE = '\033[4m' - # Custom formatter to include colors in log output class CustomFormatter(logging.Formatter): #format = "%(asctime)s - %(name)-12s - [%(levelname)-8s] - %(message)s (%(filename)s:%(lineno)d)" @@ -53,7 +51,6 @@ class CustomFormatter(logging.Formatter): formatter = logging.Formatter(log_fmt, datefmt="%Y-%m-%d %H:%M:%S") return formatter.format(record) - # Configure logging handler = logging.StreamHandler() handler.setFormatter(CustomFormatter()) @@ -83,7 +80,7 @@ def main(): project.try_start_final_shellcode = False project.try_start_final_infected_exe = False - if args.verify == "std": + if args.verify == "peb": project.source_style = SourceStyle.peb_walk project.inject = True project.inject_mode = "1,1" @@ -142,6 +139,7 @@ def start(): if project.exe_capabilities.has_all(): project.source_style = SourceStyle.iat_reuse else: + logger.info("--[ Some imports are missing for the shellcode to use IAT_REUSE") project.source_style = SourceStyle.peb_walk #observer.add_json("capabilities_a", project.exe_capabilities)