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

island: Use stat.S_IRWXU in place of 0o700 in cc/environment/utils.py

This commit is contained in:
shreyamalviya 2021-06-15 12:20:48 +05:30
parent 5ea046eda5
commit d7565fc515
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
monkey/monkey_island/cc/environment/utils.py
View File

@ -1,6 +1,7 @@
import logging import logging
import os import os
import platform import platform
import stat
def is_windows_os() -> bool: def is_windows_os() -> bool:
@ -29,7 +30,7 @@ def _create_secure_directory_linux(path: str):
try: try:
# Don't split directory creation and permission setting # Don't split directory creation and permission setting
# because it will temporarily create an accessible directory which anyone can use. # because it will temporarily create an accessible directory which anyone can use.
os.mkdir(path, mode=0o700) os.mkdir(path, mode=stat.S_IRWXU)
except Exception as ex: except Exception as ex:
LOG.error(f'Could not create a directory at "{path}": {str(ex)}') LOG.error(f'Could not create a directory at "{path}": {str(ex)}')
@ -62,7 +63,7 @@ def _create_secure_file_linux(path: str):
flags = ( flags = (
os.O_RDWR | os.O_CREAT | os.O_EXCL os.O_RDWR | os.O_CREAT | os.O_EXCL
) # read/write, create new, throw error if file exists ) # read/write, create new, throw error if file exists
mode = 0o700 # read/write/execute permissions to owner mode = stat.S_IRWXU # read/write/execute permissions to owner
os.close(os.open(path, flags, mode)) os.close(os.open(path, flags, mode))
except Exception as ex: except Exception as ex: