summaryrefslogtreecommitdiff
path: root/ctdb/config/events.d/00.ctdb
blob: 272dcb1ba355392422b78b718c5a4bd6e189b9c1 (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
#!/bin/sh

# Event script for ctdb-specific setup and other things that don't fit
# elsewhere.

[ -n "$CTDB_BASE" ] || \
    CTDB_BASE=$(d=$(dirname "$0") ; cd -P "$d" ; dirname "$PWD")

. "${CTDB_BASE}/functions"

load_script_options

############################################################

# type is commonly supported and more portable than which(1)
# shellcheck disable=SC2039
select_tdb_checker ()
{
    # Find the best TDB consistency check available.
    use_tdb_tool_check=false
    type tdbtool >/dev/null 2>&1 && found_tdbtool=true
    type tdbdump >/dev/null 2>&1 && found_tdbdump=true

    if $found_tdbtool && echo "help" | tdbtool | grep -q check ; then
	    use_tdb_tool_check=true
    elif $found_tdbtool && $found_tdbdump ; then
	    cat <<EOF
WARNING: The installed 'tdbtool' does not offer the 'check' subcommand.
 Using 'tdbdump' for database checks.
 Consider updating 'tdbtool' for better checks!
EOF
    elif $found_tdbdump ; then
	cat <<EOF
WARNING: 'tdbtool' is not available.
 Using 'tdbdump' to check the databases.
 Consider installing a recent 'tdbtool' for better checks!
EOF
    else
	cat <<EOF
WARNING: Cannot check databases since neither
 'tdbdump' nor 'tdbtool check' is available.
 Consider installing tdbtool or at least tdbdump!
EOF
        return 1
    fi
}

check_tdb ()
{
    _db="$1"

    if $use_tdb_tool_check ; then
	# tdbtool always exits with 0  :-(
	if timeout 10 tdbtool "$_db" check 2>/dev/null |
	    grep -q "Database integrity is OK" ; then
	    return 0
	else
	    return 1
	fi
    else
	timeout 10 tdbdump "$_db" >/dev/null 2>/dev/null
	return $?
    fi
}

check_persistent_databases ()
{
    _dir="${CTDB_DBDIR_PERSISTENT:-${CTDB_VARDIR}/persistent}"
    [ -d "$_dir" ] || return 0

    [ "${CTDB_MAX_PERSISTENT_CHECK_ERRORS:-0}" = "0" ] || return 0

    for _db in "$_dir/"*.tdb.*[0-9] ; do
	[ -r "$_db" ] || continue
	check_tdb "$_db" || \
	    die "Persistent database $_db is corrupted! CTDB will not start."
    done
}

check_non_persistent_databases ()
{
    _dir="${CTDB_DBDIR:-${CTDB_VARDIR}}"
    [ -d "$_dir" ] || return 0

    for _db in "${_dir}/"*.tdb.*[0-9] ; do
	[ -r "$_db" ] || continue
	check_tdb "$_db" || {
	    _backup="${_db}.$(date +'%Y%m%d.%H%M%S.%N').corrupt"
	    cat <<EOF
WARNING: database ${_db} is corrupted.
 Moving to backup ${_backup} for later analysis.
EOF
	    mv "$_db" "$_backup"

	    # Now remove excess backups
	    _max="${CTDB_MAX_CORRUPT_DB_BACKUPS:-10}"
	    _bdb="${_db##*/}" # basename
	    find "$_dir" -name "${_bdb}.*.corrupt" |
		    sort -r |
		    tail -n +$((_max + 1)) |
		    xargs rm -f
	}
    done
}

set_ctdb_variables ()
{
	_f="${CTDB_BASE}/ctdb.tunables"
	if [ ! -r "$_f" ] ; then
		return
	fi

	while IFS="=" read _var _val ; do
		case "$_var" in
		\#*|"") continue ;;
		esac

		if $CTDB setvar "$_var" "$_val" ; then
			echo "Set $_var to $_val"
		else
			echo "Invalid tunable: ${_var}=${_val}"
			return 1
		fi
	done <"$_f"
}

############################################################

ctdb_check_args "$@"

case "$1" in
init)
        # make sure we have a blank state directory for the scripts to work with
	rm -rf "$CTDB_SCRIPT_VARDIR"
	mkdir -p "$CTDB_SCRIPT_VARDIR" || \
	    die "mkdir -p ${CTDB_SCRIPT_VARDIR} - failed - $?" $?

	if select_tdb_checker ; then
	    check_persistent_databases || exit $?
	    check_non_persistent_databases
	fi
	;;

setup)
	# Set any tunables from the config file
	set_ctdb_variables || \
	    die "Aborting setup due to invalid configuration - fix typos, remove unknown tunables"
	;;

startup)
	$CTDB attach ctdb.tdb persistent
	;;
esac

# all OK
exit 0