From 955ad5ba8b10e94fa40e2b3101b44ea3ef77fffe Mon Sep 17 00:00:00 2001
From: Dave Reisner <d@falconindy.com>
Date: Tue, 12 Jan 2010 19:19:49 -0500
Subject: Add checks for root user in rollback, backup, and unmount

---
 squashfu | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/squashfu b/squashfu
index 555a181..d693a37 100755
--- a/squashfu
+++ b/squashfu
@@ -169,6 +169,10 @@ unmount_all () {
 # Args: none
 # Returns: none
 
+    if [[ $UID -ne 0 ]]; then
+        die "Must be root to unmount."
+    fi
+
     # Union MUST be unmounted first
     unmount_union
     unmount_squash
@@ -221,6 +225,10 @@ action_backup () {
 # Args: options array squashfu was invoked with, shifted 1
 # Returns: none
 
+    if [[ $UID -ne 0 ]]; then
+        die "Must be root to perform a backup"
+    fi
+
     # Does the binventory exist? If not, prompt to make sure this is an initialization
     # FIRST_RUN=0
     if [[ ! -f "$BINVENTORY" || ! -f "$SEED" ]]; then
@@ -260,6 +268,13 @@ action_backup () {
 }
 
 action_rollback () {
+# Args: number of backups to roll back
+# Returns: none
+
+    if [[ $UID -ne 0 ]]; then
+        die "Must be root to perform a rollback"
+    fi
+
     # Validate input with test cases
     if [[ -z $1 ]]; then
         die "The rollback action requires 1 additional argument."
@@ -310,7 +325,7 @@ action_report () {
 }
 
 usage () {
-    info "SquashFu: a backup solution hewn out of boredom"
+    info "SquashFu: Super Awesome Backup Express (Professional Edition)"
     cat <<HELP
 
 USAGE
-- 
cgit v1.2.3