From 095c00c6ee598f74f86c83fc2c2d42e4cbc73faf Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Mon, 18 Nov 2019 20:57:00 -0700
Subject: [PATCH] Ensure SSH tunnel is closed atexit

---
 .bin/Scripts/functions/osticket.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/.bin/Scripts/functions/osticket.py b/.bin/Scripts/functions/osticket.py
index f2369534..da5ea521 100644
--- a/.bin/Scripts/functions/osticket.py
+++ b/.bin/Scripts/functions/osticket.py
@@ -1,5 +1,6 @@
 # Wizard Kit: Functions - osTicket
 
+import atexit
 import mysql.connector as mariadb
 
 from functions.data import *
@@ -50,7 +51,11 @@ class osTicket():
 
     # Only open tunnel if one doesn't exist
     if self.tunnel_proc is None or self.tunnel_proc.poll() is not None:
+      if self.tunnel_proc:
+        # Unregister previous terminate
+        atexit.unregister(self.tunnel_proc.terminate)
       self.tunnel_proc = popen_program(cmd)
+      atexit.register(self.tunnel_proc.terminate)
 
     # Connect to database
     for x in range(5):