瀏覽代碼

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

FreddleSpl0it 1 年之前
父節點
當前提交
2072301d89
共有 1 個文件被更改,包括 6 次插入2 次删除
  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
   global quit_now
   quit_now = True
   quit_now = True
 
 
+def quit_clear(signum, frame):
+  global exit_code
+  clear()
+  sys.exit(exit_code)
+
 
 
 if __name__ == '__main__':
 if __name__ == '__main__':
   # init Logger
   # init Logger
@@ -474,8 +479,7 @@ if __name__ == '__main__':
   whitelistupdate_thread.daemon = True
   whitelistupdate_thread.daemon = True
   whitelistupdate_thread.start()
   whitelistupdate_thread.start()
 
 
-  signal.signal(signal.SIGTERM, quit)
-  atexit.register(clear)
+  signal.signal(signal.SIGTERM, quit_clear)
 
 
   while not quit_now:
   while not quit_now:
     time.sleep(0.5)
     time.sleep(0.5)