Add Cleanup function

Signed-off-by: Jonatan Schlag <jonatan.schlag@ipfire.org>

diff --git a/generate_image.sh b/generate_image.sh
index 9309f7afc39c910fef791546a5101d5fce4f2f9d..3fed861c419c38584a691773184c2f18de69f7c4 100755 (executable)
--- a/generate_image.sh
+++ b/generate_image.sh
@@ -56,6 +56,7 @@ cmd() {
        esac
 }
 
+
 cleanup_stage_1() {
        # Unmount image
        umount ${IMAGE_MOUNT_DIR}
@@ -74,6 +75,11 @@ cleanup_stage_2() {
        fi
 }
 
+cleanup() {
+       cleanup_stage_1
+       cleanup_stage_2
+}
+
 generate_image_filename() {
        local distro=${1}
        local version=${2}