aboutsummaryrefslogtreecommitdiff
path: root/handlers/mysql.in
blob: 6e894fee967d19a864e034e6bc5ba0fb48a25fda (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
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
# -*- mode: sh; sh-basic-offset: 3; indent-tabs-mode: nil; -*-
#
# mysql handler script for backupninja
#

getconf backupdir /var/backups/mysql
getconf databases all
getconf ignores
getconf nodata
getconf dbhost localhost
getconf hotcopy no
getconf sqldump no
getconf compress yes
getconf vsname

# authentication:
getconf user
getconf dbusername
getconf dbpassword
getconf configfile /etc/mysql/debian.cnf


# Decide if the handler should operate on a vserver or on the host.
# In the former case, check that $vsname exists and is running.
local usevserver=no
local vroot
if [ $vservers_are_available = yes ]; then
   if [ -n "$vsname" ]; then
      # does it exist ?
      if ! vservers_exist "$vsname" ; then
         fatal "The vserver given in vsname ($vsname) does not exist."
      fi
      # is it running ?
      vservers_running $vsname || fatal "The vserver $vsname is not running."
      # everything ok
      info "Using vserver '$vsname'."
      usevserver=yes
      vroot="$VROOTDIR/$vsname"
   else
      info "No vserver name specified, actions will be performed on the host."
   fi
fi

## Prepare ignore part of the command
## This only works for mysqldump at the moment

ignore=''
for i in $ignores $nodata; do
       ignore="$ignore --ignore-table=$i"
done
	
# create backup dirs, $vroot will be empty if no vsname was specified
# and we will instead proceed to operate on the host
[ -d $vroot$backupdir ] || mkdir -p $vroot$backupdir
[ -d $vroot$backupdir ] || fatal "Backup directory '$vroot$backupdir'"
hotdir="$backupdir/hotcopy"
dumpdir="$backupdir/sqldump"

if [ $usevserver = yes ]
then
	[ "$sqldump" == "no" -o -d $vroot$dumpdir ] || $VSERVER $vsname exec mkdir -p $dumpdir
	[ "$hotcopy" == "no" -o -d $vroot$hotdir ] || $VSERVER $vsname exec mkdir -p $hotdir
else
	[ "$sqldump" == "no" -o -d $dumpdir ] || mkdir -p $dumpdir
	[ "$hotcopy" == "no" -o -d $hotdir ] || mkdir -p $hotdir
fi

#######################################################################
## AUTHENTICATION

#
# one of three authentication methods:
# 1. setting the user, so that /home/user/.my.cnf is used.
# 2. specifying the user and password in the handler config,
#    which generates a temporary .my.cnf in /root/.my.cnf
# 3. specify the config file with --defaults-extra-file
#    (this option DOESN'T WORK WITH MYSQLHOTCOPY)
#

# create .my.cnf
# only if dbusername and dbpassword specified.
# we create a tmp file because we don't want to
# specify the password on the command line.

defaultsfile=""

if [ "$dbusername" != "" -a "$dbpassword" != "" ]
then
    if [ $usevserver = yes ]
    then
    	home=`$VSERVER $vsname exec getent passwd "root" | @AWK@ -F: '{print $6}'`
    else
    	home=`getent passwd "root" | @AWK@ -F: '{print $6}'`
    fi

    [ -d $home ] || fatal "Can't find root's home directory ($home)."
    
    mycnf="$home/.my.cnf"

    if [ $usevserver = yes ]
    then
      workcnf="$vroot$mycnf"
    else
      workcnf="$mycnf"
    fi

    if [ -f $workcnf ]
    then
      # rename temporarily
      tmpcnf="$workcnf.disable"
      debug "mv $workcnf $tmpcnf"
      mv $workcnf $tmpcnf
    fi
    
    oldmask=`umask`
    umask 077
    cat > $workcnf <<EOF
# auto generated backupninja mysql conf
[mysql]
host=$dbhost
user=$dbusername
password="$dbpassword"

[mysqldump]
host=$dbhost
user=$dbusername
password="$dbpassword"

[mysqlhotcopy]
host=$dbhost
user=$dbusername
password="$dbpassword"

[mysqladmin]
host=$dbhost
user=$dbusername
password="$dbpassword"
EOF
	umask $oldmask
	defaultsfile="--defaults-extra-file=$mycnf"
fi

# if a user is not set, use $configfile, otherwise use $mycnf
if [ "$user" == "" ]; then
	user=root;
	defaultsfile="--defaults-extra-file=$configfile"
else
	userset=true;
	if [ $usevserver = yes ]
	then
	    userhome=`$VSERVER $vsname exec getent passwd "$user" | @AWK@ -F: '{print $6}'`
	    if [ $? -eq 2 ]
	    then
	    	fatal "User $user not found in /etc/passwd"
	    fi
	    debug "User home set to: $vroot$userhome"
	    [ -f $vroot$userhome/.my.cnf ] || fatal "Can't find config file in $userhome/.my.cnf"
	else
	    userhome=`getent passwd "$user" | @AWK@ -F: '{print $6}'`
	    if [ $? -eq 2 ]
	    then
	    	fatal "User $user not found in /etc/passwd"
	    fi
	    debug "User home set to: $userhome"
	    [ -f $userhome/.my.cnf ] || fatal "Can't find config file in $userhome/.my.cnf"
	fi
	
	defaultsfile="--defaults-extra-file=$userhome/.my.cnf"
	debug "using $defaultsfile"
fi

#######################################################################
## HOT COPY

if [ "$hotcopy" == "yes" ]
then
    info "Initializing hotcopy method"
    if [ "$databases" == "all" ]
    then
	if [ $usevserver = yes ]
	then
	    	info "dbhost: $dbhost"
		execstr="$VSERVER $vsname exec $MYSQLHOTCOPY -h $dbhost --quiet --allowold --regexp /.\*/./.\*/ $hotdir"
	else
		execstr="$MYSQLHOTCOPY --quiet --allowold --regexp /.\*/./.\*/ $hotdir"
	fi
	debug "su $user -c \"$execstr\""
	if [ ! $test ]
	then
		output=`su $user -c "$execstr" 2>&1`
		code=$?
		if [ "$code" == "0" ]
		then
			debug $output
			info "Successfully finished hotcopy of all mysql databases"
		else
			warning $output
			warning "Failed to hotcopy all mysql databases"
		fi
	fi
    else	
	for db in $databases
	do
		if [ $usevserver = yes ]
		then
			execstr="$VSERVER $vsname exec $MYSQLHOTCOPY --allowold $db $hotdir"
		else
			execstr="$MYSQLHOTCOPY --allowold $db $hotdir"
		fi
		debug 'su $user -c \"$execstr\"'
		if [ ! $test ]
		then
			output=`su $user -c "$execstr" 2>&1`
			code=$?
			if [ "$code" == "0" ]
			then
				debug $output
				info "Successfully finished hotcopy of mysql database $db"
			else
				warning $output
				warning "Failed to hotcopy mysql database $db"
			fi
		fi
	done
     fi
fi

##########################################################################
## SQL DUMP

if [ "$sqldump" == "yes" ]
then
    info "Initializing SQL dump method"
    if [ "$databases" == "all" ]
    then
	if [ $usevserver = yes ]
	then
	    debug 'echo show databases | $VSERVER $vsname exec su $user -c \"$MYSQL $defaultsfile\" | grep -v Database'
	    databases=`echo 'show databases' | $VSERVER $vsname exec su $user -c "$MYSQL $defaultsfile" | grep -v Database`
	    if [ $? -ne 0 ]
	    then
	        fatal "Authentication problem, maybe user/password is wrong or mysqld is not running?"
	    fi
	else
		databases=$(su $user -c "$MYSQL $defaultsfile -N -B -e 'show databases'" | sed 's/|//g;/\+----/d')
		if [ $? -ne 0 ]
		then
		    fatal "Authentication problem, maybe user/password is wrong or mysqld is not running?"
		fi
	fi
    fi

	for db in $databases
	do
           DUMP_BASE="$MYSQLDUMP $defaultsfile --lock-tables --complete-insert --add-drop-table --quick --quote-names"

           # Dumping structure and data
           DUMP="$DUMP_BASE $ignore $db"

           # If requested, dump only the table structure for this database
           if echo "$nodata" | grep -E '(^|[[:space:]])'"$db\." >/dev/null
           then
              # Get the structure of the tables, without data
              DUMP_STRUCT="$DUMP_BASE --no-data $db"
              for qualified_table in $nodata
              do
                 table=$( expr match "$qualified_table" "$db\.\([^\w]*\)" )
                 DUMP_STRUCT="$DUMP_STRUCT $table"
              done
              DUMP="( $DUMP; $DUMP_STRUCT )"
           fi
		if [ $usevserver = yes ]
		then
                   # Test to make sure mysqld is running, if it is not sqldump will not work
                   $VSERVER $vsname exec su $user -c "$MYSQLADMIN $defaultsfile ping"
                   if [ $? -ne 0 ]; then
                      fatal "Either you have an authentication problem, or mysqld doesn't appear to be running!"
                   fi
                   if [ "$compress" == "yes" ]; then
                      execstr="$VSERVER $vsname exec $DUMP | $GZIP > $vroot$dumpdir/${db}.sql.gz"
                   else
                      execstr="$VSERVER $vsname exec $DUMP -r $vroot$dumpdir/${db}.sql"
                   fi
		else
                   # Test to make sure mysqld is running, if it is not sqldump will not work
                   su $user -c "$MYSQLADMIN $defaultsfile ping"
                   if [ $? -ne 0 ]; then
                      fatal "Either you have an authentication problem, or mysqld doesn't appear to be running!"
                   fi
                   if [ "$compress" == "yes" ]; then
                      execstr="$DUMP | $GZIP > $dumpdir/${db}.sql.gz"
                   else
                      execstr="$DUMP -r $dumpdir/${db}.sql"
                   fi
		fi
		debug "su $user -c \"$execstr\""
		if [ ! $test ]
		then
			output=`su $user -c "$execstr" 2>&1`
			code=$?
			if [ "$code" == "0" ]
			then
				debug $output
				info "Successfully finished dump of mysql database $db"
			else
				warning $output
				warning "Failed to dump mysql databases $db"
			fi
		fi
	done
fi

# clean up tmp config file
if [ "$dbusername" != "" -a "$dbpassword" != "" ]
then
	## clean up tmp config file
	debug "rm $workcnf"
	rm $workcnf
	if [ -f "$tmpcnf" ]
	then
		debug "mv $tmpcnf $workcnf"
		mv $tmpcnf $workcnf
	fi
fi

return 0