Răsfoiți Sursa

[Netfilter] only perform cleanup at exit if SIGTERM was recieved

FreddleSpl0it 1 an în urmă
părinte
comite
2072301d89
1 a modificat fișierele cu 6 adăugiri și 2 ștergeri
  1. 6 2
      data/Dockerfiles/netfilter/main.py

+ 6 - 2
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)