1 files changed, 45 insertions, 0 deletions
diff --git a/functions/exit.sh b/functions/exit.sh
new file mode 100755
index 000000000..5fa1fa86f
--- /dev/null
+++ b/functions/exit.sh
@@ -0,0 +1,45 @@
+#!/bin/sh
+
+## live-build(7) - System Build Scripts
+## Copyright (C) 2006-2010 Daniel Baumann <daniel@debian.org>
+##
+## live-build comes with ABSOLUTELY NO WARRANTY; for details see COPYING.
+## This is free software, and you are welcome to redistribute it
+## under certain conditions; see COPYING for details.
+
+
+Exit ()
+{
+	VALUE="${?}"
+
+	if [ "${_DEBUG}" = "true" ]
+	then
+		# Dump variables
+		set | grep -e ^LH
+	fi
+
+	# Always exit true in case we are not able to unmount
+	# (e.g. due to running processes in chroot from user customizations)
+	Echo_message "Begin unmounting filesystems..."
+
+	if [ -e /proc/mounts ]
+	then
+		for DIRECTORY in $(awk -v dir="${PWD}/chroot/" '$2 ~ dir { print $2 }' /proc/mounts | sort -r)
+		do
+			umount ${DIRECTORY} > /dev/null 2>&1 || true
+		done
+	else
+		for DIRECTORY in /dev/shm /dev/pts /dev /proc /selinux /sys
+		do
+			umount -f chroot/${DIRECTORY} > /dev/null 2>&1 || true
+		done
+	fi
+
+	return ${VALUE}
+}
+
+Setup_cleanup ()
+{
+	Echo_message "Setting up cleanup function"
+	trap 'Exit' EXIT HUP INT QUIT TERM
+}