ctdb-scripts: Quote some variable expansions
[nivanova/samba-autobuild/.git] / ctdb / config / events.d / 00.ctdb
1 #!/bin/sh
2
3 # Event script for ctdb-specific setup and other things that don't fit
4 # elsewhere.
5
6 [ -n "$CTDB_BASE" ] || \
7     CTDB_BASE=$(d=$(dirname "$0") ; cd -P "$d" ; dirname "$PWD")
8
9 . "${CTDB_BASE}/functions"
10
11 loadconfig
12
13 ctdb_setup_service_state_dir "ctdb"
14
15 ############################################################
16
17 select_tdb_checker ()
18 {
19     # Find the best TDB consistency check available.
20     use_tdb_tool_check=false
21     type tdbtool >/dev/null 2>&1 && found_tdbtool=true
22     type tdbdump >/dev/null 2>&1 && found_tdbdump=true
23
24     if $found_tdbtool && echo "help" | tdbtool | grep -q check ; then
25             use_tdb_tool_check=true
26     elif $found_tdbtool && $found_tdbdump ; then
27             cat <<EOF
28 WARNING: The installed 'tdbtool' does not offer the 'check' subcommand.
29  Using 'tdbdump' for database checks.
30  Consider updating 'tdbtool' for better checks!
31 EOF
32     elif $found_tdbdump ; then
33         cat <<EOF
34 WARNING: 'tdbtool' is not available.
35  Using 'tdbdump' to check the databases.
36  Consider installing a recent 'tdbtool' for better checks!
37 EOF
38     else
39         cat <<EOF
40 WARNING: Cannot check databases since neither
41  'tdbdump' nor 'tdbtool check' is available.
42  Consider installing tdbtool or at least tdbdump!
43 EOF
44         return 1
45     fi
46 }
47
48 check_tdb ()
49 {
50     _db="$1"
51
52     if $use_tdb_tool_check ; then
53         # tdbtool always exits with 0  :-(
54         if timeout 10 tdbtool "$_db" check 2>/dev/null |
55             grep -q "Database integrity is OK" ; then
56             return 0
57         else
58             return 1
59         fi
60     else
61         timeout 10 tdbdump "$_db" >/dev/null 2>/dev/null
62         return $?
63     fi
64 }
65
66 check_persistent_databases ()
67 {
68     _dir="${CTDB_DBDIR_PERSISTENT:-${CTDB_DBDIR:-${CTDB_VARDIR}}/persistent}"
69     [ -d "$_dir" ] || return 0
70
71     [ "${CTDB_MAX_PERSISTENT_CHECK_ERRORS:-0}" = "0" ] || return 0
72
73     for _db in "$_dir/"*.tdb.*[0-9] ; do
74         [ -r "$_db" ] || continue
75         check_tdb "$_db" || \
76             die "Persistent database $_db is corrupted! CTDB will not start."
77     done
78 }
79
80 check_non_persistent_databases ()
81 {
82     _dir="${CTDB_DBDIR:-${CTDB_VARDIR}}"
83     [ -d "$_dir" ] || return 0
84
85     for _db in "${_dir}/"*.tdb.*[0-9] ; do
86         [ -r "$_db" ] || continue
87         check_tdb "$_db" || {
88             _backup="${_db}.$(date +'%Y%m%d.%H%M%S.%N').corrupt"
89             cat <<EOF
90 WARNING: database ${_db} is corrupted.
91  Moving to backup ${_backup} for later analysis.
92 EOF
93             mv "$_db" "$_backup"
94
95             # Now remove excess backups
96             ls -td "${_db}."*".corrupt" |
97             tail -n +$((${CTDB_MAX_CORRUPT_DB_BACKUPS:-10} + 1)) |
98             xargs rm -f
99         }
100     done
101 }
102
103 set_ctdb_variables ()
104 {
105     # set any tunables from the config file
106     set | sed -n '/^CTDB_SET_/s/=.*//p' |
107     while read v; do
108         varname="${v#CTDB_SET_}"
109         value=$(eval echo "\$$v")
110         if $CTDB setvar "$varname" "$value" ; then
111             echo "Set $varname to $value"
112         else
113             echo "Invalid configuration: CTDB_SET_${varname}=${value}"
114             return 1
115         fi
116     done
117 }
118
119 ############################################################
120
121 ctdb_check_args "$@"
122
123 case "$1" in
124      init)
125         # make sure we have a blank state directory for the scripts to work with
126         rm -rf "$CTDB_SCRIPT_VARDIR"
127         mkdir -p "$CTDB_SCRIPT_VARDIR" || \
128             die "mkdir -p ${CTDB_SCRIPT_VARDIR} - failed - $?" $?
129
130         if select_tdb_checker ; then
131             check_persistent_databases || exit $?
132             check_non_persistent_databases
133         fi
134         ;;
135
136     setup)
137         # Set any tunables from the config file
138         set_ctdb_variables || \
139             die "Aborting setup due to invalid configuration - fix typos, remove unknown tunables"
140         ;;
141
142     startup)
143         $CTDB attach ctdb.tdb persistent
144         ;;
145
146     *)
147         ctdb_standard_event_handler "$@"
148         ;;
149 esac
150
151 # all OK
152 exit 0