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

Move `has_sufficient_permissions` to a separate file in utils/

This commit is contained in:
Shreya 2021-06-03 18:38:13 +05:30
parent 6f1154f911
commit a2bd59c377
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
monkey/monkey_island/cc/services/utils/file_permissions.py Normal file
View File

@ -0,0 +1,8 @@
import os
def has_sufficient_permissions(path: str, required_permissions: str) -> bool:
file_mode = os.stat(path).st_mode
file_permissions = oct(file_mode & 0o777)
return file_permissions == required_permissions

12
monkey/monkey_island/cc/setup/certificate/certificate_setup.py
View File

@ -2,6 +2,7 @@ import logging
import os
from common.utils.exceptions import InsecurePermissionsError
from monkey_island.cc.services.utils.file_permissions import has_sufficient_permissions
from monkey_island.setup.island_config_options import IslandConfigOptions
logger = logging.getLogger(__name__)
@ -16,7 +17,7 @@ def setup_certificate(config_options: IslandConfigOptions) -> (str, str):
if not os.path.exists(file):
raise FileNotFoundError(f"File not found at {file}. Exiting.")
if not has_sufficient_permissions(file):
if not has_sufficient_permissions(path=file, required_permissions="0o400"):
raise InsecurePermissionsError(
f"{file} has insecure permissions. Required permissions: r--------. Exiting."
)
@ -24,12 +25,3 @@ def setup_certificate(config_options: IslandConfigOptions) -> (str, str):
logger.INFO(f"Using certificate path: {crt_path}, and key path: {key_path}.")
return crt_path, key_path
def has_sufficient_permissions(path: str) -> bool:
required_permissions = "0o400"
file_mode = os.stat(path).st_mode
file_permissions = oct(file_mode & 0o777)
return file_permissions == required_permissions