aboutsummaryrefslogtreecommitdiff
path: root/handlers/rdiff
blob: 82160f4fb5b03b86a36cdf45a4944611672dc383 (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
# -*- mode: sh; sh-basic-offset: 3; indent-tabs-mode: nil; -*-
#
# 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
local usevserver=no
if [ $vservers_are_available = yes -a -n "$vsinclude" ]; then
   info "vserver method enabled"
   usevserver=yes
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"
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 = yes ]; 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