From 2072301d89683cdbe9a398150b199630d1109d81 Mon Sep 17 00:00:00 2001
From: FreddleSpl0it <patschul@posteo.de>
Date: Fri, 2 Feb 2024 11:08:44 +0100
Subject: [PATCH] [Netfilter] only perform cleanup at exit if SIGTERM was
 recieved

---
 data/Dockerfiles/netfilter/main.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/data/Dockerfiles/netfilter/main.py b/data/Dockerfiles/netfilter/main.py
index 8480c9ef..7f8dd9fb 100644
--- a/data/Dockerfiles/netfilter/main.py
+++ b/data/Dockerfiles/netfilter/main.py
@@ -376,6 +376,11 @@ def quit(signum, frame):
   global quit_now
   quit_now = True
 
+def quit_clear(signum, frame):
+  global exit_code
+  clear()
+  sys.exit(exit_code)
+
 
 if __name__ == '__main__':
   # init Logger
@@ -474,8 +479,7 @@ if __name__ == '__main__':
   whitelistupdate_thread.daemon = True
   whitelistupdate_thread.start()
 
-  signal.signal(signal.SIGTERM, quit)
-  atexit.register(clear)
+  signal.signal(signal.SIGTERM, quit_clear)
 
   while not quit_now:
     time.sleep(0.5)