blob: 05aacc319f908baa2b8a208e611d1afb89d67227 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
|
#
# rdiff-backup handler script for backupninja
# requires rdiff-backup
#
### FUNCTIONS ###
function test_connection() {
# given a user and host,
# tests the connection.
# if user or host is missing, returns 0
# (ie, assume it's a local connection).
if [ $# -lt 2 ]; then
debug "(local is assumed to be a good connection)"
return 0
fi
local user=$1
local host=$2
debug "ssh -o PasswordAuthentication=no $host -l $user 'echo -n 1'"
local ret=`ssh -o PasswordAuthentication=no $host -l $user 'echo -n host is alive'`
if echo $ret | grep "host is alive"; then
debug "Connected to $host as $user successfully"
else
fatal "Can't connect to $host as $user."
fi
}
function get_version() {
# given no arguments, returns the local version.
# given a user and host, returns the remote version.
# if user or host is missing, returns the local version.
if [ "$#" -lt 2 ]; then
debug "$RDIFFBACKUP -V"
echo `$RDIFFBACKUP -V`
else
local user=$1
local host=$2
debug "ssh $host -l $user '$RDIFFBACKUP -V'"
echo `ssh $host -l $user "$RDIFFBACKUP -V | grep rdiff-backup"`
fi
}
function check_consistency() {
local section=$1
local type=$2
local user=$3
local host=$4
if [ "$type" == "local" ]; then
if [ "$user" != "" ]; then
warning "User should not be specified for local $section."
fi
if [ "$host" != "" ]; then
warning "Host should not be specified for local $section."
fi
fi
if [ "$type" == "remote" ]; then
if [ "$user" == "" ]; then
fatal "User must be specified for remote $section."
fi
if [ "host" == "" ]; then
fatal "Host must be specifed for remote $section."
fi
fi
}
### GET CONFIG ###
getconf options
getconf testconnect yes
getconf nicelevel 0
setsection source
getconf type; sourcetype=$type
getconf user; sourceuser=$user
getconf host; sourcehost=$host
check_consistency "source" "$type" "$user" "$host"
getconf label
getconf keep 60
getconf include
getconf vsinclude
getconf exclude
setsection dest
getconf directory; destdir=$directory
# strip trailing /
destdir=${destdir%/}
getconf type; desttype=$type
getconf user; destuser=$user
getconf host; desthost=$host
check_consistency "destination" "$type" "$user" "$host"
### CHECK CONFIG ###
# See if vservers are configured
if [ "$vservers" = "yes" ]
then
if [ ! -d $VROOTDIR ]
then
fatal "vservers enabled, but $VROOTDIR does not exist!"
else
info "vserver method enabled"
usevserver=1
fi
fi
# check the connection at the source and destination
if [ "$testconnect" = "yes" ] || [ "${test}" -eq 1 ]; then
test_connection $sourceuser $sourcehost
test_connection $destuser $desthost
fi
# see that rdiff-backup has the same version at the source and destination
sourceversion=`get_version $sourceuser $sourcehost`
destversion=`get_version $destuser $desthost`
if [ "$sourceversion" != "$destversion" ]; then
fatal "rdiff-backup does not have the same version at the source and at the destination."
fi
# source specific checks
[ "$include" != "" -o "$vsinclude" != "" ] || fatal "No source includes specified"
#TODO should I test for vsinclude if usevservers=1?
case $sourcetype in
remote ) execstr_sourcepart="$sourceuser@$sourcehost::/" ;;
local ) execstr_sourcepart="/" ;;
* ) fatal "sourcetype '$sourcetype' is neither local nor remote" ;;
esac
# destination specific checks
[ "$destdir" != "" ] || fatal "Destination directory not set"
case $desttype in
remote ) execstr_destpart="$destuser@$desthost::$destdir/$label" ;;
local ) execstr_destpart="$destdir/$label" ;;
* ) fatal "desttype '$desttype' is neither local nor remote" ;;
esac
### REMOVE OLD BACKUPS ###
if [ "`echo $keep | tr -d 0-9`" == "" ]; then
# add D if no other date unit is specified
keep="${keep}D"
fi
removestr="$RDIFFBACKUP --force --remove-older-than $keep "
if [ "$desttype" == "remote" ]; then
removestr="${removestr}${destuser}@${desthost}::"
fi
removestr="${removestr}${destdir}/${label}";
debug "$removestr"
if [ ! $test ]; then
output=`$removestr 2>&1`
if [ $? = 0 ]; then
debug $output
info "Removing backups older than $keep days succeeded."
else
warning $output
warning "Failed removing backups older than $keep."
fi
fi
### EXECUTE ###
execstr="$RDIFFBACKUP $options --print-statistics "
# TODO: order the includes and excludes
# excludes
for i in $exclude; do
str="${i//__star__/*}"
execstr="${execstr}--exclude '$str' "
done
# includes
for i in $include; do
[ "$i" != "/" ] || fatal "Sorry, you cannot use 'include = /'"
str="${i//__star__/*}"
execstr="${execstr}--include '$str' "
done
# vsinclude
if [ $usevserver ]; then
for vserver in `ls $VROOTDIR|grep -v lost+found`; do
for vi in $vsinclude; do
str="${vi//__star__/*}"
execstr="${execstr}--include '$VROOTDIR/$vserver$str' "
done
done
fi
# exclude everything else
execstr="${execstr}--exclude '/*' "
# include client-part and server-part
execstr="${execstr}$execstr_sourcepart $execstr_destpart"
debug "$execstr"
if [ ! $test ]; then
output=`nice -n $nicelevel su -c "$execstr" 2>&1`
if [ $? = 0 ]; then
debug $output
info "Successfully finished backing up source $label"
else
warning $output
warning "Failed backup up source $label"
fi
fi
return 0
|
