samba-mirror/ctdb/config/events/legacy/95.database.script

#!/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

	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').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
}

maybe_backup_persistent_tdbs()
{
	_dir="${CTDB_PERSISTENT_DB_BACKUP_DIR:-}"
	if [ -z "$_dir" ]; then
		return 0
	fi

	if [ ! -d "$_dir" ]; then
		echo "Creating CTDB_PERSISTENT_DB_BACKUP_DIR=${_dir}"
		if ! mkdir -p "$_dir"; then
			die "ERROR: unable to create ${_dir}"
		fi
	fi

	# Don't backup if there are backup files from within the past day
	_out=$(find "$_dir" -type f -mtime -1)
	if [ -n "$_out" ]; then
		return 0
	fi

	# Script will ignore if this isn't leader node, so don't
	# double-check that here...
	"${CTDB_BASE}/ctdb-backup-persistent-tdbs.sh" -l "$_dir"

	# Remove backups beyond the limit (default 14)
	_limit="${CTDB_PERSISTENT_DB_BACKUP_LIMIT:-14}"
	_offset=$((_limit + 1))
	# Can't sort by time using find instead of ls
	# shellcheck disable=SC2012
	ls -t "$_dir"/* 2>/dev/null | tail -n "+${_offset}" | xargs rm -f
}

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

ctdb_check_args "$@"

case "$1" in
init)
	# Load/cache database options from configuration file
	ctdb_get_db_options

	if select_tdb_checker; then
		check_persistent_databases || exit $?
		check_non_persistent_databases
	fi
	;;
monitor)
	maybe_backup_persistent_tdbs
	;;
esac

# all OK
exit 0