Added thread join for divertg

2020-03-13 08:45:54 -04:00 · 2020-03-13 08:45:54 -04:00 · 06bf5612e1
parent 81b2e935c9
commit 06bf5612e1
1 changed files with 7 additions and 0 deletions
--- a/engine.py
+++ b/engine.py
@ -166,10 +166,17 @@ class WindowsEngine(GenericEngine):
        """
        Closes the divert connection
        """
+        self.logger.debug("Shutting down Divert engine...")
        if self.divert:
+            self.logger.debug("Closing Divert engine")
            self.divert.close()
            self.divert = None

+        if self.divert_thread:
+            self.logger.debug("Joining divert thread")
+            self.divert_thread.join()
+        self.logger.debug("Shutdown complete.")
+
    def run_divert(self):
        """
        Runs actions on packets