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

island: Add type hinting in file_utils.py

This commit is contained in:
Shreya 2021-06-15 19:11:57 +05:30
parent 22c3c5a11b
commit 64ac1fe706
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/monkey_island/cc/server_utils/file_utils.py
View File

@ -54,7 +54,7 @@ def _create_secure_directory_windows(path: str):
raise ex raise ex
def get_file_descriptor_for_new_secure_file(path: str): def get_file_descriptor_for_new_secure_file(path: str) -> int:
if not os.path.isfile(path): if not os.path.isfile(path):
if is_windows_os(): if is_windows_os():
return _get_file_descriptor_for_new_secure_file_windows(path) return _get_file_descriptor_for_new_secure_file_windows(path)
@ -62,7 +62,7 @@ def get_file_descriptor_for_new_secure_file(path: str):
return _get_file_descriptor_for_new_secure_file_linux(path) return _get_file_descriptor_for_new_secure_file_linux(path)
def _get_file_descriptor_for_new_secure_file_linux(path: str): def _get_file_descriptor_for_new_secure_file_linux(path: str) -> int:
try: try:
mode = stat.S_IRUSR | stat.S_IWUSR mode = stat.S_IRUSR | stat.S_IWUSR
flags = ( flags = (
@ -77,7 +77,7 @@ def _get_file_descriptor_for_new_secure_file_linux(path: str):
raise ex raise ex
def _get_file_descriptor_for_new_secure_file_windows(path: str): def _get_file_descriptor_for_new_secure_file_windows(path: str) -> int:
try: try:
file_access = win32file.GENERIC_READ | win32file.GENERIC_WRITE file_access = win32file.GENERIC_READ | win32file.GENERIC_WRITE
# subsequent open operations on the object will succeed only if read access is requested # subsequent open operations on the object will succeed only if read access is requested