Added safegaurds to prevent script from running on top of itself

[zfs-ubuntu/.git] / zfs-replicate-wrapper

diff --git a/zfs-replicate-wrapper b/zfs-replicate-wrapper
index faeaeb0eecce314f4961afca0606f254683c4df2..b52e64e05e3ffbb7c8fb77a88af8756013ed61e3 100755 (executable)
--- a/zfs-replicate-wrapper
+++ b/zfs-replicate-wrapper
@@ -9,41 +9,51 @@ replicate="/usr/local/etc/bin/zfs-replicate"
 logfile_parser="/usr/local/etc/bin/zfs-log-parser"
 logfile="/var/log/zfs/zfs-replicate.log"
 lockdir="/tmp/zfs-admin-lock"
+mailto=root
 destpool="backups"
 maxsleeptime=60
+maxattempts=600
 released_lock_date=0
+date=`date`
 
 # Setup our cleanup and exit trap
 cleanup() { 
   rm -rf "$lockdir"
   if [ $released_lock_date == 0 ]; then 
+    zpool export $destpool
+    /usr/local/etc/bin/usb-drive-power off
     echo `date` ZFS admin lock released >> $logfile
   fi
 }
 trap cleanup EXIT
 
 # Auto snapshot every zfs filesystem on the system specified below
-date=`date`;
 echo "$date Polling for ZFS admin lock" >> $logfile
 
 # Poll for a lock on the zfs subsystem, and make the lock once we can do so
+attempts=0
 while true; do
   if ! mkdir "$lockdir" >/dev/null 2>&1; then
     # Another zfs admin tool is running.
     # Wait a random amount of time and try again
     ransleep=$(($RANDOM % $maxsleeptime))
     sleep $ransleep
+    ((attempts=attempts+1))
   else 
     # No other zfs admin tool is running, we can now.
     break
   fi
+  if [[ $attempts -gt $maxattempts ]]; then
+    # We've exceeded our maximum while loop count
+    ls -ld $lockdir | /usr/bin/mailx -s "zfs-replicate-all unable to obtain zfs admin lock" $mailto
+    exit 1
+  fi
 done
 date=`date`;
 echo "$date ZFS admin lock obtained" >> $logfile
 
 # Poweron the destpool and import it
-/usr/local/bin/br -x /dev/ttyd0 A3 ON
-sleep 10
+/usr/local/etc/bin/usb-drive-power on >> $logfile
 zpool import $destpool
 
 # List the filesystems to replicate
@@ -62,7 +72,7 @@ $replicate tank/backup $destpool
 
 # Export the destpool and power it down
 zpool export $destpool
-/usr/local/bin/br -x /dev/ttyd0 A3 OFF
+/usr/local/etc/bin/usb-drive-power off >> $logfile
 
 # Release our lock
 released_lock_date=1
@@ -70,3 +80,4 @@ echo `date` ZFS admin lock released >> $logfile
 
 # Parse the log file and extract our backup stats
 $logfile_parser "$logfile" "$date" >> $logfile
+