backmeupscotty

   1 #!/bin/bash
   2
   3 REMOTE_HOST=localhost
   4 REMOTE_DIR=/tmp/backmeupscotty/test
   5 ARCHIVE_KEEPNBACKUPS=30
   6 ARCHIVE_KEEPNDAYS=30
   7 BACKUP_RUNEVERYNTHDAY=1
   8
   9 _UPPERME=$(echo $(basename $0) | tr '[:lower:]' '[:upper:]')
  10
  11 function timestamp {
  12     date +"[%y-%m-%d %H:%M:%S]"
  13 }
  14
  15 function scottyinfo {
  16     echo $(timestamp) $_UPPERME: $@
  17 }
  18
  19 function scottyerror {
  20     echo $(timestamp) $_UPPERME: $@ >&2
  21 }
  22
  23 function ssh255 {
  24     ssh $@
  25     sshret=$?
  26
  27     if [ $sshret -eq 255 ]; then
  28         scottyerror "SSH connection failed!"
  29         exit 1
  30     else
  31         return $sshret
  32     fi
  33 }
  34
  35 function grepbackups {
  36     ssh255 $REMOTE_HOST "ls $REMOTE_DIR" | grep -E '[0-9]+-[0-9]+'
  37 }
  38
  39 function isIncomplete {
  40     if ( ssh255 $REMOTE_HOST '[ -d '$REMOTE_DIR/incomplete' ]' ); then
  41         return 0
  42     else
  43         return 1
  44     fi
  45 }
  46
  47 function isNthDay {
  48     if [ $(( ( $(date +%s) / (60*60*24) ) % $BACKUP_RUNEVERYNTHDAY )) -eq 0 ];
  49     then
  50         return 0
  51     else
  52         return 1
  53     fi
  54 }
  55
  56 function latestTooOld {
  57     for oldbackup in $(grepbackups); do
  58         tstamp=$(echo $oldbackup | cut -d'-' -f1)
  59
  60         if [ $(( $(date +%s) - $tstamp )) -lt \
  61             $(( ($BACKUP_RUNEVERYNTHDAY*24+12)*60*60 )) ]
  62         then
  63             return 1
  64         fi
  65     done
  66
  67     return 0
  68 }
  69
  70 function scottysync {
  71     timestamp=$(date +%s)
  72
  73     scottyinfo "Syncing $SYNC_SRC to $REMOTE_HOST:$REMOTE_DIR @$timestamp."
  74
  75     if [ ! -d "$SYNC_SRC" ]; then
  76         scottyerror "Source dir $SYNC_SRC does not exist. Not syncing!"
  77         return 1
  78     fi
  79
  80     if [ $(ls -A "$SYNC_SRC" | wc -l) -eq 0 ]; then
  81         scottyerror "Source dir $SYNC_SRC is empty. Not syncing!"
  82         return 1
  83     fi
  84
  85     dir_current=$REMOTE_DIR/current
  86     dir_incomplete=$REMOTE_DIR/incomplete
  87     dir_timestamped=$REMOTE_DIR/$timestamp-$(date -d @$timestamp +%Y%m%d%H%M%S)
  88
  89     if [ -z $SYNC_EXC ]; then
  90         rsync_exclude=""
  91     else
  92         rsync_exclude=$(eval echo --exclude={$SYNC_EXC} | tr -d {})
  93     fi
  94
  95     if (ssh255 $REMOTE_HOST '[ ! -d '$REMOTE_DIR' ]'); then
  96         scottyinfo "Creating destination directory $REMOTE_HOST:$REMOTE_DIR."
  97         ssh255 $REMOTE_HOST "mkdir $REMOTE_DIR"
  98     fi
  99
 100     if isIncomplete; then
 101         scottyerror "Continuing old incomplete backup."
 102     fi
 103
 104     scottyinfo "Starting rsync."
 105     rsync -e ssh \
 106         -v -aHAX --numeric-ids --delete --delete-excluded \
 107         --link-dest=$dir_current \
 108         $rsync_exclude \
 109         $SYNC_SRC/ $REMOTE_HOST:$dir_incomplete/
 110
 111     if [ $? -eq 0 ]; then
 112         scottyinfo "Timestamping completed backup and linking to current backup."
 113         ssh255 $REMOTE_HOST \
 114             "mv $dir_incomplete $dir_timestamped && rm -f $dir_current && ln -s $(basename $dir_timestamped) $dir_current"
 115     fi
 116
 117     while [ $(grepbackups | wc -l) -gt $ARCHIVE_KEEPNBACKUPS ]; do
 118         oldestbackup=$(grepbackups | head -1)
 119         oldestbackuptstamp=$(echo $oldestbackup | cut -d'-' -f1)
 120
 121         if [ $oldestbackuptstamp -lt $(( $(date +%s) - $ARCHIVE_KEEPNDAYS*60*60*24 )) ]; then
 122             scottyinfo "Removing old backup $oldestbackup."
 123             ssh255 $REMOTE_HOST rm -r "$REMOTE_DIR/$oldestbackup"
 124         else
 125             break
 126         fi
 127     done
 128 }
 129
 130 function deleteLock {
 131     if ! rmdir /var/lock/$(basename $0); then
 132         scottyerror "Could not delete lockfile /tmp/$(basename $0).lock!"
 133     fi
 134 }
 135
 136 function cleanup_abort {
 137     scottyerror "Caught exit signal! Cleaning up."
 138
 139     cleanup ABORT
 140
 141     if [ $(jobs -p) ]; then
 142         scottyerror "TERMinating remaining child processes."
 143         kill $(jobs -p)
 144     fi
 145
 146     deleteLock
 147
 148     exit
 149 }
 150
 151 function cleanup {
 152     scottyinfo "No cleanup function was defined."
 153 }
 154
 155 function prepare {
 156     scottyinfo "No prepare function was defined."
 157 }
 158
 159 function cleanup_normal {
 160     cleanup
 161     deleteLock
 162 }
 163
 164 function printhelp {
 165     cat <<EOF
 166 Usage: $(basename $0) [OPTION]...
 167
 168 Recognized options:
 169   -q   Only output errors
 170   -n   Run only on nth day
 171   -h   Print out this help
 172 EOF
 173 }
 174
 175 function backmeupscotty {
 176     while getopts "qn:h" opt; do
 177         case $opt in
 178             q)
 179                 exec > /dev/null
 180                 ;;
 181             n)
 182                 BACKUP_RUNEVERYNTHDAY=$OPTARG
 183                 ;;
 184             h)
 185                 printhelp
 186                 exit 0
 187                 ;;
 188         esac
 189     done
 190
 191     ssh255 $REMOTE_HOST exit
 192
 193     if latestTooOld; then
 194         scottyerror "The latest backup is too old."
 195     elif isNthDay; then
 196         scottyinfo "This is the nth day."
 197     else
 198         scottyinfo "No backup has to be done. Exiting."
 199         exit 0
 200     fi
 201
 202     scottyinfo "Performing backup."
 203
 204     trap cleanup_abort EXIT
 205
 206     prepare
 207     scottysync
 208
 209     trap cleanup_normal EXIT
 210
 211     exit 0
 212 }
 213
 214 if ! mkdir /var/lock/$(basename $0); then
 215     scottyerror "Another instance of $(basename $0) is still running!"
 216     exit 1
 217 else
 218     trap deleteLock EXIT
 219 fi