Allow passing additional parameters to rsync
[public/mirror.git] / csc-sync-standard
1 #! /bin/bash
2
3 if [ $# -lt 3 ]; then
4   echo 'Usage: sync local_dir rsync_host rsync_dir'
5   exit 1
6 fi
7
8 set -e
9
10 RSYNC="rsync"
11 TOP_DIR=/mirror
12 TO=$TOP_DIR/root/$1
13 RSYNC_HOST=$2
14 RSYNC_DIR=$3
15 LOGDIR=/var/log/mirror/$1_$2
16 ADDRESS=129.97.134.71
17 EXTRA=$4
18
19 mkdir -p $LOGDIR
20
21 # LOCK_TIMEOUT is a timeout in minutes.  Defaults to 360 (6 hours).
22 # This program creates a lock to ensure that only one copy
23 # of it is mirroring any one archive at any one time.
24 # Locks held for longer than the timeout are broken, unless
25 # a running rsync process appears to be connected to $RSYNC_HOST.
26
27 LOCK_TIMEOUT=360
28
29 # There should be no need to edit anything below this point, unless there
30 # are problems.
31
32 #-----------------------------------------------------------------------------#
33
34 # If you are accessing a rsync server/module which is password-protected,
35 # uncomment the following lines (and edit the other file).
36
37 # . ftpsync.conf
38 export RSYNC_PASSWORD
39 if [[ "$RSYNC_USER" != "" ]]; then
40     RSYNC_HOST=$RSYNC_USER@$RSYNC_HOST
41 fi
42
43 #-----------------------------------------------------------------------------#
44
45 # Check for some environment variables
46 if [ -z $TO ] || [ -z $RSYNC_HOST ] || [ -z $RSYNC_DIR ] || [ -z $LOGDIR ]; then
47         echo "One of the following variables seems to be empty:"
48         echo "TO, RSYNC_HOST, RSYNC_DIR or LOGDIR"
49         exit 2
50 fi
51
52 # Note: on some non-Debian systems, hostname doesn't accept -f option.
53 # If that's the case on your system, make sure hostname prints the full
54 # hostname, and remove the -f option. If there's no hostname command,
55 # explicitly replace `hostname -f` with the hostname.
56
57 HOSTNAME=`hostname -f`
58
59 # The hostname must match the "Site" field written in the list of mirrors.
60 # If hostname doesn't returns the correct value, fill and uncomment below 
61 HOSTNAME=mirror.csclub.uwaterloo.ca
62  
63 LOCK="${TO}/Archive-Update-in-Progress-${HOSTNAME}"
64
65 # The temp directory used by rsync --delay-updates is not
66 # world-readable remotely. It must be excluded to avoid errors. 
67 TMP_EXCLUDE="--exclude .~tmp~/"
68
69 # Logfile
70 LOGFILE=$LOGDIR/mirror.log
71
72 # Get in the right directory and set the umask to be group writable
73
74 cd $HOME
75 umask 002
76
77 # Check to see if another sync is in progress
78 if [ -f "$LOCK" ]; then
79 # Note: this requires the findutils find; for other finds, adjust as necessary
80   if [ "`find $LOCK -maxdepth 1 -amin -$LOCK_TIMEOUT`" = "" ]; then
81 # Note: this requires the procps ps; for other ps', adjust as necessary
82     if ps ax | grep '[r]'sync | grep -q $RSYNC_HOST; then
83       echo "stale lock found, but a rsync is still running, aiee!" > /dev/stderr
84       exit 1
85     else
86       echo "stale lock found (not accessed in the last $LOCK_TIMEOUT minutes), forcing update!"
87       rm -f $LOCK
88     fi
89   else
90     echo "current lock file exists, unable to start rsync!"
91     exit 1
92   fi
93 fi
94
95 touch $LOCK
96 # Note: on some non-Debian systems, trap doesn't accept "exit" as signal
97 # specification. If that's the case on your system, try using "0".
98 trap "rm -f $LOCK" exit
99
100 set +e
101
102 # Now sync the remaining stuff
103 $RSYNC -rlHtv --delete \
104      --exclude "Archive-Update-in-Progress-${HOSTNAME}" \
105      --timeout=3600 --address=$ADDRESS $TMP_EXCLUDE $EXTRA \
106      $RSYNC_HOST::$RSYNC_DIR $TO >> $LOGFILE 2>&1
107 if [[ "$?" != "0" ]]; then
108     echo "ERROR: Help, something weird happened" | tee -a $LOGFILE
109     echo "mirroring /pool exited with exitcode" $result | tee -a $LOGFILE
110 fi
111
112 savelog $LOGFILE >/dev/null
113
114 rm $LOCK