refactor: unify project handling (cmdline = projects/default)

This commit is contained in:
Dobin
2024-03-29 20:29:33 +00:00
parent 70eb0bf798
commit 4410685d0d
7 changed files with 53 additions and 44 deletions
+6 -2
View File
@@ -14,7 +14,7 @@ import phases.injector
from observer import observer
from pe.pehelper import extract_code_from_exe_file_ep
from sender import scannerDetectsBytes
from model.project import Project
from model.project import Project, prepare_project
from model.settings import Settings
from model.defs import *
from log import setup_logging
@@ -96,7 +96,8 @@ def main():
settings.inject_exe_in = args.inject
settings.inject_exe_out = args.inject.replace(".exe", ".infected.exe")
settings.prep()
settings.prep_web("default")
write_webproject("default", settings)
exit_code = start(settings)
exit(exit_code)
@@ -112,6 +113,9 @@ def start(settings: Settings) -> int:
# And logs
observer.reset()
# Prepare the project: copy all files to projects/<project_name>/
prepare_project("default", settings)
# Do the thing and catch the errors
try:
start_real(settings)