From 00bf13005274c46f124b9c2a6ad64976ca96b694 Mon Sep 17 00:00:00 2001
From: Jason Mehring <nrgaway@gmail.com>
Date: Tue, 10 Feb 2015 10:09:04 -0500
Subject: [PATCH] Use cleanup function to trap errors

---
 qubeize_image | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/qubeize_image b/qubeize_image
index 0333754..35e38eb 100755
--- a/qubeize_image
+++ b/qubeize_image
@@ -46,7 +46,11 @@ fi
 # Cleanup function
 # ------------------------------------------------------------------------------
 function cleanup() {
-    umount_kill "$PWD/mnt" || :
+    errval=$?
+    trap - ERR
+    trap
+    umount_kill "$PWD/mnt" || true
+    exit $errval
 }
 trap cleanup ERR
 
@@ -66,7 +70,7 @@ export INSTALLDIR=mnt
 # ------------------------------------------------------------------------------
 # Run qubeize script
 # ------------------------------------------------------------------------------
-"$SCRIPTSDIR/04_install_qubes.sh" || { umount "$INSTALLDIR"; exit 1; }
+"$SCRIPTSDIR/04_install_qubes.sh"
 
 # ------------------------------------------------------------------------------
 # Create App Menus
@@ -110,7 +114,7 @@ fi
 # Finsh - unmount image
 # ------------------------------------------------------------------------------
 echo "--> Unmounting $IMG"
-cleanup
+umount_kill "$PWD/mnt" || true
 
 echo "Qubeized image stored at: $IMG"