p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

island: Consolidate duplicate delete post breach file functionality

This commit is contained in:
Mike Salvatore 2021-05-11 08:56:04 -04:00
parent 71029cb7f9
commit 5742e85ff5
2 changed files with 4 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
monkey/monkey_island/cc/resources/pba_file_upload.py
View File

@ -1,6 +1,5 @@
import copy
import logging
import os
from pathlib import Path
import flask_restful
@ -84,16 +83,7 @@ class FileUpload(flask_restful.Resource):
)
filename = ConfigService.get_config_value(filename_path)
if filename:
file_path = Path(PostBreachFilesService.get_custom_pba_directory()).joinpath(filename)
FileUpload._delete_file(file_path)
PostBreachFilesService.remove_file(filename)
ConfigService.set_config_value(filename_path, "")
return {}
@staticmethod
def _delete_file(file_path):
try:
if os.path.exists(file_path):
os.remove(file_path)
except OSError as e:
LOG.error("Couldn't remove previously uploaded post breach files: %s" % e)

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

@ -36,12 +36,12 @@ class PostBreachFilesService:
PBA_LINUX_FILENAME_PATH
)
if linux_filename:
PostBreachFilesService._remove_file(linux_filename)
PostBreachFilesService.remove_file(linux_filename)
if windows_filename:
PostBreachFilesService._remove_file(windows_filename)
PostBreachFilesService.remove_file(windows_filename)
@staticmethod
def _remove_file(file_name):
def remove_file(file_name):
file_path = os.path.join(PostBreachFilesService.get_custom_pba_directory(), file_name)
try:
if os.path.exists(file_path):