From 5d8a1ec64fc6955daa42f3e75dcac45fbfb560d9 Mon Sep 17 00:00:00 2001 From: Dobin Date: Mon, 1 Apr 2024 19:49:41 +0100 Subject: [PATCH] feature: more exes --- app/views_project.py | 13 +++++++++---- data/binary/exes_more/.gitkeep | Bin 0 -> 6 bytes model/defs.py | 1 + 3 files changed, 10 insertions(+), 4 deletions(-) create mode 100644 data/binary/exes_more/.gitkeep 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 0000000000000000000000000000000000000000..49cc8ef0e116cef009fe0bd72473a964bbd07f9b GIT binary patch literal 6 NcmezWkC%aq0RRg=0u=xN literal 0 HcmV?d00001 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/"