p17860254
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

island: Add PostBreachFilesService.get_custom_pba_directory()

This commit is contained in:
Mike Salvatore 2021-05-11 08:30:59 -04:00
parent ee19eed596
commit 4190797ca2
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
monkey/monkey_island/cc/services/post_breach_files.py
View File

@ -14,6 +14,7 @@ PBA_LINUX_FILENAME_PATH = ["monkey", "post_breach", "PBA_linux_filename"]
class PostBreachFilesService: class PostBreachFilesService:
DATA_DIR = None DATA_DIR = None
CUSTOM_PBA_DIRNAME = "custom_pbas"
@staticmethod @staticmethod
def remove_PBA_files(): def remove_PBA_files():
@ -31,13 +32,19 @@ class PostBreachFilesService:
@staticmethod @staticmethod
def _remove_file(file_name): def _remove_file(file_name):
file_path = os.path.join(PostBreachFilesService.DATA_DIR, file_name) file_path = os.path.join(PostBreachFilesService.get_custom_pba_directory(), file_name)
try: try:
if os.path.exists(file_path): if os.path.exists(file_path):
os.remove(file_path) os.remove(file_path)
except OSError as e: except OSError as e:
logger.error("Can't remove previously uploaded post breach files: %s" % e) logger.error("Can't remove previously uploaded post breach files: %s" % e)
@staticmethod
def get_custom_pba_directory():
return os.path.join(
PostBreachFilesService.DATA_DIR, PostBreachFilesService.CUSTOM_PBA_DIRNAME
)
@staticmethod @staticmethod
def set_config_PBA_files(config_json): def set_config_PBA_files(config_json):
""" """