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

Agent: Use a random, secure /tmp directory for "monkey_dir"

This commit is contained in:
Mike Salvatore 2022-03-02 14:12:52 -05:00
parent 1ca9a21d43
commit 8a6a820d14
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
monkey/infection_monkey/utils/monkey_dir.py
View File

@ -1,16 +1,20 @@
import os
import shutil
import tempfile
MONKEY_DIR_NAME = "monkey_dir"
MONKEY_DIR_PREFIX = "monkey_dir_"
_monkey_dir = None
# TODO: Check if we even need this. Individual plugins can just use tempfile.mkdtemp() or
# tempfile.mkftemp() if they need to.
def create_monkey_dir():
"""
Creates directory for monkey and related files
"""
if not os.path.exists(get_monkey_dir_path()):
os.mkdir(get_monkey_dir_path())
global _monkey_dir
_monkey_dir = tempfile.mkdtemp(prefix=MONKEY_DIR_PREFIX, dir=tempfile.gettempdir())
return _monkey_dir
def remove_monkey_dir():
@ -26,4 +30,4 @@ def remove_monkey_dir():
def get_monkey_dir_path():
return os.path.join(tempfile.gettempdir(), MONKEY_DIR_NAME)
return _monkey_dir