Reset changes made to SIGTERM/SIGINT stuff

AikidoSec · Aug 22, 2024 · 617e0bf · 617e0bf
1 parent b6ffc5b
commit 617e0bf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/aikido_firewall/background_process/comms.py b/aikido_firewall/background_process/comms.py
@@ -62,7 +62,6 @@ def start_aikido_listener(self):
             # the aikido server, so we shouldn't start the background process
             return
         #  Daemon is set to True so that the process kills itself when the main process dies
-        self.send_data_to_bg_process("KILL", None)
         self.background_process = Process(
             target=AikidoBackgroundProcess, args=(self.address, self.key), daemon=True
         )

diff --git a/aikido_firewall/decorators/gunicorn.py b/aikido_firewall/decorators/gunicorn.py
@@ -2,11 +2,7 @@
 Includes all the wrappers for gunicorn config file
 """
 
-import atexit
 import aikido_firewall
-from aikido_firewall.background_process import get_comms
-from aikido_firewall.helpers.logging import logger
-
 
 def when_ready(prev_func):
     """
@@ -15,6 +11,7 @@ def when_ready(prev_func):
     """
 
     def aik_when_ready(server):
+        aikido_firewall.protect("background-process-only")
         prev_func(server)
 
     return aik_when_ready
@@ -27,7 +24,7 @@ def post_fork(prev_func):
     """
 
     def aik_post_fork(server, worker):
-        aikido_firewall.protect()
+        aikido_firewall.protect(server=False)
         prev_func(server, worker)
 
     return aik_post_fork