Add test before start to check IPC accesibility

Author: bitterpanda63

aikido_zen/__init__.py

@@ -4,6 +4,8 @@
 
 import os
 
+from aikido_zen.background_process.test_uds_file_access import test_uds_file_access
+
 # Re-export functions :
 from aikido_zen.context.users import set_user
 from aikido_zen.middleware import should_block_request
@@ -30,6 +32,9 @@ def protect(mode="daemon", token=""):
     if aikido_disabled_flag_active():
         # Do not run any aikido code when the disabled flag is on
         return
+    if not test_uds_file_access():
+        return  # Unable to start background process
+
     if token:
         os.environ["AIKIDO_TOKEN"] = token
 

aikido_zen/background_process/test_uds_file_access.py

@@ -0,0 +1,27 @@
+import os
+import platform
+import sys
+
+from aikido_zen.background_process import get_uds_filename
+from aikido_zen.helpers.logging import logger
+
+
+def test_uds_file_access():
+    """Returns true if access is possible, false if it's not."""
+    filename = get_uds_filename()
+    directory = "/".join(filename.split("/")[:-1])
+
+    if platform.system() == "Windows":
+        # Python does not support Windows UDS just yet, so we have to rely on INET
+        return True
+
+    if not os.access(directory, os.R_OK) or not os.access(directory, os.W_OK):
+        # We need both read and write permissions in order to start the socket
+        logger.critical(
+            "Cannot start Zen: no read/write permissions for directory %s. "
+            + "You can change the path by setting "
+            + "the AIKIDO_TMP_DIR environment variable.",
+            directory,
+        )
+        return False
+    return True