diff --git a/app/views_project.py b/app/views_project.py index 3ddd3a0..2e98155 100644 --- a/app/views_project.py +++ b/app/views_project.py @@ -54,7 +54,9 @@ def project(name): exes = [] for file in os.listdir(PATH_EXES): - exes.append(file) + exes.append(PATH_EXES + file) + for file in os.listdir(PATH_EXES_MORE): + exes.append(PATH_EXES_MORE + file) shellcodes = [] for file in os.listdir(PATH_SHELLCODES): @@ -99,8 +101,8 @@ def add_project(): else: settings.cleanup_files_on_exit = False - settings.inject_exe_in = PATH_EXES + request.form['exe'] - settings.inject_exe_out = PATH_EXES + request.form['exe'].replace(".exe", ".infected.exe") + settings.inject_exe_in = request.form['exe'] + settings.inject_exe_out = request.form['exe'].replace(".exe", ".infected.exe") source_style = request.form['source_style'] settings.source_style = SourceStyle[source_style] @@ -128,7 +130,10 @@ def add_project(): else: # GET exes = [] for file in os.listdir(PATH_EXES): - exes.append(file) + exes.append(PATH_EXES + file) + + for file in os.listdir(PATH_EXES_MORE): + exes.append(PATH_EXES_MORE + file) shellcodes = [] for file in os.listdir(PATH_SHELLCODES): diff --git a/data/binary/exes_more/.gitkeep b/data/binary/exes_more/.gitkeep new file mode 100644 index 0000000..49cc8ef Binary files /dev/null and b/data/binary/exes_more/.gitkeep differ diff --git a/model/defs.py b/model/defs.py index 50957bd..3786279 100644 --- a/model/defs.py +++ b/model/defs.py @@ -9,6 +9,7 @@ VerifyFilename: FilePath = r'C:\Temp\a' # Directory structure PATH_EXES = "data/binary/exes/" +PATH_EXES_MORE = "data/binary/exes_more/" PATH_SHELLCODES = "data/binary/shellcodes/" PATH_CARRIER = "data/source/carrier/" PATH_PEB_WALK = "data/source/carrier/peb_walk/"