mirror of
https://github.com/dobin/SuperMega
synced 2026-06-03 01:27:11 +00:00
refactor: rename verify std to peb
This commit is contained in:
+2
-4
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user