From b33f97f9a45de8fa5d32738af7936c37e6390e3f Mon Sep 17 00:00:00 2001
From: Mike Reeves <mike.reeves@securityonionsolutions.com>
Date: Thu, 12 Apr 2018 11:16:28 -0400
Subject: [PATCH] Setup Script - Add cleanup of install dir

---
 so-setup-network.sh | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/so-setup-network.sh b/so-setup-network.sh
index 457dd49ba..a0b125efb 100644
--- a/so-setup-network.sh
+++ b/so-setup-network.sh
@@ -237,12 +237,20 @@ got_root() {
 
 }
 
+install_cleanup() {
+
+  # Clean up after ourselves
+  rm -rf ./installtmp
+
+}
 install_prep() {
 
   # Create a tmp space that isn't in /tmp
   mkdir ./installtmp
   TMP=./installtmp
+
 }
+
 install_master() {
 
   # Install the salt master package
@@ -480,6 +488,7 @@ whiptail_bond_nics_mtu() {
 whiptail_cancel() {
 
   whiptail --title "Security Onion Setup" --msgbox "Cancelling Setup. No changes have been made." 8 78
+  install_cleanup
   exit
 
 }
@@ -698,7 +707,7 @@ if (whiptail_you_sure) then
 
   # Create a dir to get started
   install_prep
-  
+
 	# Let folks know they need their management interface already set up.
 	whiptail_network_notice