Add 20ms timeout for firewall lists

Author: bitterpanda63

aikido_zen/background_process/comms.py

@@ -44,15 +44,15 @@ def __init__(self, address, key):
         global comms
         comms = self
 
-    def send_data_to_bg_process(self, action, obj, receive=False):
+    def send_data_to_bg_process(self, action, obj, receive=False, timeout_in_sec=0.1):
         """Try-catched send_data_to_bg_process"""
         try:
-            return self._send_data_to_bg_process(action, obj, receive)
+            return self._send_data_to_bg_process(action, obj, receive, timeout_in_sec)
         except Exception as e:
             logger.debug("Exception happened in send_data_to_bg_process : %s", e)
             return {"success": False, "error": "unknown"}
 
-    def _send_data_to_bg_process(self, action, obj, receive=False):
+    def _send_data_to_bg_process(self, action, obj, receive=False, timeout_in_sec=0.1):
         """
         This creates a new client for comms to the background process
         """
@@ -88,7 +88,7 @@ def target(address, key, receive, data, result_obj):
 
         # Start and join the thread for 100ms, afterwards the thread is forced to close (daemon=True)
         t.start()
-        t.join(timeout=0.1)
+        t.join(timeout=timeout_in_sec)
         if not result_obj[0]:
             logger.debug(
                 " Failure in communication to background process, %s(%s)", action, obj

aikido_zen/sources/functions/request_handler.py

@@ -58,6 +58,7 @@ def pre_response():
             "user-agent": context.get_user_agent(),
         },
         receive=True,
+        timeout_in_sec=(20 / 1000),
     )
     if not check_fw_lists_res["success"] or not check_fw_lists_res["data"]["blocked"]:
         return