2010-12-02 01:48:02 +03:00
# Hey Emacs, this is a -*- shell-script -*- !!!
2007-06-01 14:54:26 +04:00
# utility functions for ctdb event scripts
2009-11-19 07:00:17 +03:00
PATH=/bin:/usr/bin:/usr/sbin:/sbin:$PATH
2009-11-13 10:28:25 +03:00
2010-12-15 02:08:16 +03:00
[ -z "$CTDB_VARDIR" ] && {
export CTDB_VARDIR="/var/ctdb"
}
2011-06-07 09:57:29 +04:00
[ -z "$CTDB_ETCDIR" ] && {
export CTDB_ETCDIR="/etc"
}
2010-12-15 02:08:16 +03:00
2007-06-03 16:07:07 +04:00
#######################################
# pull in a system config file, if any
2010-08-31 11:40:40 +04:00
_loadconfig() {
2009-11-13 10:28:25 +03:00
2009-11-20 08:45:36 +03:00
if [ -z "$1" ] ; then
2009-11-19 07:00:17 +03:00
foo="${service_config:-${service_name}}"
if [ -n "$foo" ] ; then
loadconfig "$foo"
fi
2010-01-22 09:19:12 +03:00
elif [ "$1" != "ctdb" ] ; then
loadconfig "ctdb"
2009-11-13 10:28:25 +03:00
fi
2011-06-07 09:57:29 +04:00
if [ -f $CTDB_ETCDIR/sysconfig/$1 ]; then
. $CTDB_ETCDIR/sysconfig/$1
elif [ -f $CTDB_ETCDIR/default/$1 ]; then
. $CTDB_ETCDIR/default/$1
2009-11-20 08:45:36 +03:00
elif [ -f $CTDB_BASE/sysconfig/$1 ]; then
. $CTDB_BASE/sysconfig/$1
2007-06-03 16:07:07 +04:00
fi
}
2010-08-31 11:40:40 +04:00
loadconfig () {
_loadconfig "$@"
}
2009-01-16 15:26:57 +03:00
##############################################################
# determine on what type of system (init style) we are running
detect_init_style() {
# only do detection if not already set:
test "x$CTDB_INIT_STYLE" != "x" && return
if [ -x /sbin/startproc ]; then
CTDB_INIT_STYLE="suse"
elif [ -x /sbin/start-stop-daemon ]; then
2009-09-15 13:33:35 +04:00
CTDB_INIT_STYLE="debian"
2009-01-16 15:26:57 +03:00
else
CTDB_INIT_STYLE="redhat"
fi
}
2007-06-01 14:54:26 +04:00
2007-06-02 12:51:05 +04:00
######################################################
# simulate /sbin/service on platforms that don't have it
2011-06-07 09:57:29 +04:00
# _service() makes it easier to hook the service() function for
# testing.
_service ()
{
2009-11-13 10:28:25 +03:00
_service_name="$1"
_op="$2"
2009-01-16 15:31:02 +03:00
# do nothing, when no service was specified
2009-11-20 08:45:36 +03:00
[ -z "$_service_name" ] && return
2009-01-16 15:31:02 +03:00
2007-06-02 12:51:05 +04:00
if [ -x /sbin/service ]; then
2011-06-07 09:57:29 +04:00
$_nice /sbin/service "$_service_name" "$_op"
elif [ -x $CTDB_ETCDIR/init.d/$_service_name ]; then
$_nice $CTDB_ETCDIR/init.d/$_service_name "$_op"
elif [ -x $CTDB_ETCDIR/rc.d/init.d/$_service_name ]; then
$_nice $CTDB_ETCDIR/rc.d/init.d/$_service_name "$_op"
2007-06-02 12:51:05 +04:00
fi
}
2011-06-07 09:57:29 +04:00
service()
{
_nice=""
_service "$@"
}
2008-02-13 00:20:20 +03:00
######################################################
# simulate /sbin/service (niced) on platforms that don't have it
2011-06-07 09:57:29 +04:00
nice_service()
{
_nice="nice"
_service "$@"
2008-02-13 00:20:20 +03:00
}
2007-06-17 05:57:42 +04:00
2011-06-28 08:54:33 +04:00
######################################################
# wrapper around /proc/ settings to allow them to be hooked
# for testing
# 1st arg is relative path under /proc/, 2nd arg is value to set
set_proc ()
{
echo "$2" >"/proc/$1"
}
######################################################
# wrapper around getting file contents from /proc/ to allow
# this to be hooked for testing
# 1st arg is relative path under /proc/
get_proc ()
{
cat "/proc/$1"
}
2007-06-06 06:08:42 +04:00
######################################################
# check that a rpc server is registered with portmap
# and responding to requests
# usage: ctdb_check_rpc SERVICE_NAME PROGNUM VERSION
######################################################
ctdb_check_rpc() {
2009-11-20 08:45:36 +03:00
progname="$1"
2007-06-06 06:08:42 +04:00
prognum="$2"
version="$3"
2010-11-16 11:31:18 +03:00
ctdb_check_rpc_out=$(rpcinfo -u localhost $prognum $version 2>&1)
if [ $? -ne 0 ] ; then
ctdb_check_rpc_out="ERROR: $progname failed RPC check:
$ctdb_check_rpc_out"
echo "$ctdb_check_rpc_out"
return 1
fi
2007-06-06 06:08:42 +04:00
}
######################################################
# check a set of directories is available
2009-10-12 09:32:49 +04:00
# return 1 on a missing directory
2008-07-23 09:35:46 +04:00
# usage: ctdb_check_directories_probe SERVICE_NAME <directories...>
2007-06-06 06:08:42 +04:00
######################################################
2008-07-23 09:35:46 +04:00
ctdb_check_directories_probe() {
2009-11-20 08:45:36 +03:00
while IFS="" read d ; do
case "$d" in
*%*)
continue
;;
*)
2010-03-26 18:40:00 +03:00
[ -d "${d}/." ] || return 1
2009-11-20 08:45:36 +03:00
esac
done
2008-07-23 09:35:46 +04:00
}
######################################################
# check a set of directories is available
# usage: ctdb_check_directories SERVICE_NAME <directories...>
######################################################
ctdb_check_directories() {
2009-11-20 08:45:36 +03:00
n="${1:-${service_name}}"
ctdb_check_directories_probe || {
echo "ERROR: $n directory \"$d\" not available"
exit 1
}
2007-06-06 06:08:42 +04:00
}
######################################################
# check a set of tcp ports
2009-11-20 08:45:36 +03:00
# usage: ctdb_check_tcp_ports <ports...>
2007-06-06 06:08:42 +04:00
######################################################
ctdb_check_tcp_ports() {
2009-01-30 18:10:05 +03:00
2009-11-20 08:45:36 +03:00
for p ; do
if ! netstat -a -t -n | grep -q "0\.0\.0\.0:$p .*LISTEN" ; then
if ! netstat -a -t -n | grep -q ":::$p .*LISTEN" ; then
echo "ERROR: $service_name tcp port $p is not responding"
return 1
fi
fi
done
2007-06-06 06:08:42 +04:00
}
2007-06-17 06:05:29 +04:00
2009-05-19 02:47:19 +04:00
######################################################
# check a unix socket
# usage: ctdb_check_unix_socket SERVICE_NAME <socket_path>
######################################################
ctdb_check_unix_socket() {
2009-11-20 08:45:36 +03:00
socket_path="$1"
[ -z "$socket_path" ] && return
2009-05-19 02:47:19 +04:00
2009-11-20 08:45:36 +03:00
if ! netstat --unix -a -n | grep -q "^unix.*LISTEN.*${socket_path}$"; then
echo "ERROR: $service_name socket $socket_path not found"
return 1
2009-05-19 02:47:19 +04:00
fi
}
2007-06-17 06:05:29 +04:00
######################################################
# check a command returns zero status
# usage: ctdb_check_command SERVICE_NAME <command>
######################################################
ctdb_check_command() {
service_name="$1"
wait_cmd="$2"
[ -z "$wait_cmd" ] && return;
$wait_cmd > /dev/null 2>&1 || {
2008-01-16 14:06:44 +03:00
echo "ERROR: $service_name - $wait_cmd returned error"
2007-06-17 06:05:29 +04:00
exit 1
}
}
2007-10-11 01:27:38 +04:00
################################################
# kill off any TCP connections with the given IP
################################################
kill_tcp_connections() {
_IP="$1"
_failed=0
_killcount=0
2010-09-03 06:35:25 +04:00
connfile="$CTDB_VARDIR/state/connections.$_IP"
2008-07-09 04:03:21 +04:00
netstat -tn |egrep "^tcp.*[[:space:]]+$_IP:.*ESTABLISHED" | awk '{print $4" "$5}' > $connfile
2008-08-20 06:50:50 +04:00
netstat -tn |egrep "^tcp.*[[:space:]]+::ffff:$_IP:.*ESTABLISHED" | awk '{print $4" "$5}' >> $connfile
2008-08-20 03:47:00 +04:00
2007-10-11 01:27:38 +04:00
while read dest src; do
2008-08-20 03:47:00 +04:00
srcip=`echo $src | sed -e "s/:[^:]*$//"`
srcport=`echo $src | sed -e "s/^.*://"`
destip=`echo $dest | sed -e "s/:[^:]*$//"`
destport=`echo $dest | sed -e "s/^.*://"`
2008-01-16 14:06:44 +03:00
echo "Killing TCP connection $srcip:$srcport $destip:$destport"
2008-08-01 08:17:50 +04:00
ctdb killtcp $srcip:$srcport $destip:$destport >/dev/null 2>&1 || _failed=1
case $destport in
2009-11-06 01:54:03 +03:00
# we only do one-way killtcp for CIFS
139|445) : ;;
2008-08-01 08:17:50 +04:00
# for all others we do 2-way
*)
ctdb killtcp $destip:$destport $srcip:$srcport >/dev/null 2>&1 || _failed=1
;;
esac
2007-10-11 01:27:38 +04:00
_killcount=`expr $_killcount + 1`
2008-08-20 03:47:00 +04:00
done < $connfile
2011-04-13 06:08:09 +04:00
rm -f $connfile
2008-08-20 03:47:00 +04:00
2007-10-11 01:27:38 +04:00
[ $_failed = 0 ] || {
2008-01-16 14:06:44 +03:00
echo "Failed to send killtcp control"
2007-10-11 01:27:38 +04:00
return;
}
[ $_killcount -gt 0 ] || {
return;
}
2009-03-24 06:05:31 +03:00
_count=0
while netstat -tn |egrep "^tcp.*[[:space:]]+$_IP:.*ESTABLISHED" > /dev/null; do
sleep 1
_count=`expr $_count + 1`
[ $_count -gt 3 ] && {
echo "Timed out killing tcp connections for IP $_IP"
return;
}
done
echo "killed $_killcount TCP connections to released IP $_IP"
}
##################################################################
# kill off the local end for any TCP connections with the given IP
##################################################################
kill_tcp_connections_local_only() {
_IP="$1"
_failed=0
_killcount=0
2010-09-03 06:35:25 +04:00
connfile="$CTDB_VARDIR/state/connections.$_IP"
2009-03-24 06:05:31 +03:00
netstat -tn |egrep "^tcp.*[[:space:]]+$_IP:.*ESTABLISHED" | awk '{print $4" "$5}' > $connfile
netstat -tn |egrep "^tcp.*[[:space:]]+::ffff:$_IP:.*ESTABLISHED" | awk '{print $4" "$5}' >> $connfile
while read dest src; do
srcip=`echo $src | sed -e "s/:[^:]*$//"`
srcport=`echo $src | sed -e "s/^.*://"`
destip=`echo $dest | sed -e "s/:[^:]*$//"`
destport=`echo $dest | sed -e "s/^.*://"`
echo "Killing TCP connection $srcip:$srcport $destip:$destport"
ctdb killtcp $srcip:$srcport $destip:$destport >/dev/null 2>&1 || _failed=1
_killcount=`expr $_killcount + 1`
done < $connfile
2011-04-13 06:08:09 +04:00
rm -f $connfile
2009-03-24 06:05:31 +03:00
[ $_failed = 0 ] || {
echo "Failed to send killtcp control"
return;
}
[ $_killcount -gt 0 ] || {
return;
}
2007-10-11 01:27:38 +04:00
_count=0
2008-07-09 04:03:21 +04:00
while netstat -tn |egrep "^tcp.*[[:space:]]+$_IP:.*ESTABLISHED" > /dev/null; do
2007-10-11 01:27:38 +04:00
sleep 1
_count=`expr $_count + 1`
[ $_count -gt 3 ] && {
2008-01-16 14:06:44 +03:00
echo "Timed out killing tcp connections for IP $_IP"
2007-10-11 01:27:38 +04:00
return;
}
done
2008-01-16 14:06:44 +03:00
echo "killed $_killcount TCP connections to released IP $_IP"
2007-10-11 01:27:38 +04:00
}
2009-12-18 11:43:20 +03:00
##################################################################
# tickle any TCP connections with the given IP
##################################################################
tickle_tcp_connections() {
_IP="$1"
_failed=0
_killcount=0
2010-09-03 06:35:25 +04:00
connfile="$CTDB_VARDIR/state/connections.$_IP"
2009-12-18 11:43:20 +03:00
netstat -tn |egrep "^tcp.*[[:space:]]+$_IP:.*ESTABLISHED" | awk '{print $4" "$5}' > $connfile
netstat -tn |egrep "^tcp.*[[:space:]]+::ffff:$_IP:.*ESTABLISHED" | awk '{print $4" "$5}' >> $connfile
while read dest src; do
srcip=`echo $src | sed -e "s/:[^:]*$//"`
srcport=`echo $src | sed -e "s/^.*://"`
destip=`echo $dest | sed -e "s/:[^:]*$//"`
destport=`echo $dest | sed -e "s/^.*://"`
echo "Tickle TCP connection $srcip:$srcport $destip:$destport"
ctdb tickle $srcip:$srcport $destip:$destport >/dev/null 2>&1 || _failed=1
echo "Tickle TCP connection $destip:$destport $srcip:$srcport"
ctdb tickle $destip:$destport $srcip:$srcport >/dev/null 2>&1 || _failed=1
done < $connfile
2011-04-13 06:08:09 +04:00
rm -f $connfile
2009-12-18 11:43:20 +03:00
[ $_failed = 0 ] || {
echo "Failed to send tickle control"
return;
}
}
2008-02-11 01:35:37 +03:00
########################################################
# start/stop the nfs service on different platforms
########################################################
startstop_nfs() {
PLATFORM="unknown"
2011-06-07 09:57:29 +04:00
[ -x $CTDB_ETCDIR/init.d/nfsserver ] && {
2008-02-11 01:35:37 +03:00
PLATFORM="sles"
}
2011-06-07 09:57:29 +04:00
[ -x $CTDB_ETCDIR/init.d/nfslock ] && {
2008-02-11 01:35:37 +03:00
PLATFORM="rhel"
}
case $PLATFORM in
sles)
case $1 in
start)
service nfsserver start
;;
stop)
service nfsserver stop > /dev/null 2>&1
;;
2010-08-19 01:18:22 +04:00
restart)
2011-06-28 08:58:13 +04:00
set_proc "fs/nfsd/threads" 0
2011-01-11 09:06:48 +03:00
service nfsserver stop > /dev/null 2>&1
pkill -9 nfsd
service nfsserver start
2010-08-19 01:18:22 +04:00
;;
2008-02-11 01:35:37 +03:00
esac
;;
rhel)
case $1 in
start)
service nfslock start
service nfs start
;;
stop)
service nfs stop > /dev/null 2>&1
service nfslock stop > /dev/null 2>&1
;;
2010-08-19 01:18:22 +04:00
restart)
2011-06-28 08:58:13 +04:00
set_proc "fs/nfsd/threads" 0
2011-01-11 09:06:48 +03:00
service nfs stop > /dev/null 2>&1
service nfslock stop > /dev/null 2>&1
pkill -9 nfsd
service nfslock start
service nfs start
2010-08-19 01:18:22 +04:00
;;
2008-02-11 01:35:37 +03:00
esac
;;
*)
echo "Unknown platform. NFS is not supported with ctdb"
exit 1
;;
esac
}
2008-02-11 01:52:09 +03:00
########################################################
# start/stop the nfs lockmanager service on different platforms
########################################################
startstop_nfslock() {
PLATFORM="unknown"
2011-06-07 09:57:29 +04:00
[ -x $CTDB_ETCDIR/init.d/nfsserver ] && {
2008-02-11 01:52:09 +03:00
PLATFORM="sles"
}
2011-06-07 09:57:29 +04:00
[ -x $CTDB_ETCDIR/init.d/nfslock ] && {
2008-02-11 01:52:09 +03:00
PLATFORM="rhel"
}
case $PLATFORM in
sles)
# for sles there is no service for lockmanager
# so we instead just shutdown/restart nfs
case $1 in
start)
service nfsserver start
;;
stop)
service nfsserver stop > /dev/null 2>&1
;;
2010-10-14 01:12:41 +04:00
restart)
service nfsserver stop
service nfsserver start
;;
2008-02-11 01:52:09 +03:00
esac
;;
rhel)
case $1 in
start)
service nfslock start
;;
stop)
service nfslock stop > /dev/null 2>&1
;;
2010-10-14 01:12:41 +04:00
restart)
service nfslock stop
service nfslock start
;;
2008-02-11 01:52:09 +03:00
esac
;;
*)
echo "Unknown platform. NFS locking is not supported with ctdb"
exit 1
;;
esac
}
2008-04-10 00:50:12 +04:00
2010-01-20 13:10:48 +03:00
# better use delete_ip_from_iface() together with add_ip_to_iface
2010-02-12 12:24:44 +03:00
# remove_ip should be removed in future
2009-04-08 06:49:28 +04:00
remove_ip() {
2010-02-12 12:24:44 +03:00
local _ip_maskbits=$1
local _iface=$2
local _ip=`echo "$_ip_maskbits" | cut -d '/' -f1`
local _maskbits=`echo "$_ip_maskbits" | cut -d '/' -f2`
delete_ip_from_iface "$_iface" "$_ip" "$_maskbits"
return $?
2009-04-08 06:49:28 +04:00
}
2010-01-20 13:10:48 +03:00
add_ip_to_iface()
{
local _iface=$1
local _ip=$2
local _maskbits=$3
2010-09-03 06:35:25 +04:00
local _state_dir="$CTDB_VARDIR/state/interface_modify"
2010-02-12 12:23:17 +03:00
local _lockfile="$_state_dir/$_iface.flock"
local _readd_base="$_state_dir/$_iface.readd.d"
mkdir -p $_state_dir || {
ret=$?
echo "Failed to mkdir -p $_state_dir - $ret"
return $ret
}
2010-01-20 13:10:48 +03:00
test -f $_lockfile || {
touch $_lockfile
}
2010-02-12 11:48:01 +03:00
flock --timeout 30 $_lockfile $CTDB_BASE/interface_modify.sh add "$_iface" "$_ip" "$_maskbits" "$_readd_base"
2010-01-20 13:10:48 +03:00
return $?
}
delete_ip_from_iface()
{
local _iface=$1
local _ip=$2
local _maskbits=$3
2010-09-03 06:35:25 +04:00
local _state_dir="$CTDB_VARDIR/state/interface_modify"
2010-02-12 12:23:17 +03:00
local _lockfile="$_state_dir/$_iface.flock"
local _readd_base="$_state_dir/$_iface.readd.d"
mkdir -p $_state_dir || {
ret=$?
echo "Failed to mkdir -p $_state_dir - $ret"
return $ret
}
2010-01-20 13:10:48 +03:00
test -f $_lockfile || {
touch $_lockfile
}
2010-02-12 11:48:01 +03:00
flock --timeout 30 $_lockfile $CTDB_BASE/interface_modify.sh delete "$_iface" "$_ip" "$_maskbits" "$_readd_base"
return $?
}
setup_iface_ip_readd_script()
{
local _iface=$1
local _ip=$2
local _maskbits=$3
local _readd_script=$4
2010-09-03 06:35:25 +04:00
local _state_dir="$CTDB_VARDIR/state/interface_modify"
2010-02-12 12:23:17 +03:00
local _lockfile="$_state_dir/$_iface.flock"
local _readd_base="$_state_dir/$_iface.readd.d"
mkdir -p $_state_dir || {
ret=$?
echo "Failed to mkdir -p $_state_dir - $ret"
return $ret
}
2010-02-12 11:48:01 +03:00
test -f $_lockfile || {
touch $_lockfile
}
flock --timeout 30 $_lockfile $CTDB_BASE/interface_modify.sh readd_script "$_iface" "$_ip" "$_maskbits" "$_readd_base" "$_readd_script"
2010-01-20 13:10:48 +03:00
return $?
}
2009-09-30 15:05:16 +04:00
########################################################
# some simple logic for counting events - per eventscript
2009-11-19 07:00:17 +03:00
# usage: ctdb_counter_init
# ctdb_counter_incr
# ctdb_check_counter_limit <limit>
# ctdb_check_counter_limit succeeds when count >= <limit>
2009-09-30 15:05:16 +04:00
########################################################
_ctdb_counter_common () {
2009-11-19 07:00:17 +03:00
_counter_file="$ctdb_fail_dir/$service_name"
2009-09-30 15:05:16 +04:00
mkdir -p "${_counter_file%/*}" # dirname
}
ctdb_counter_init () {
2009-11-13 10:28:25 +03:00
_ctdb_counter_common
2009-09-30 15:05:16 +04:00
2009-11-19 08:48:19 +03:00
>"$_counter_file"
2009-09-30 15:05:16 +04:00
}
ctdb_counter_incr () {
2009-11-13 10:28:25 +03:00
_ctdb_counter_common
2009-09-30 15:05:16 +04:00
# unary counting!
echo -n 1 >> "$_counter_file"
}
2009-11-13 10:28:25 +03:00
ctdb_check_counter_limit () {
_ctdb_counter_common
2009-09-30 15:05:16 +04:00
2009-11-19 08:48:19 +03:00
_limit="${1:-${service_fail_limit}}"
_quiet="$2"
2009-09-30 15:05:16 +04:00
# unary counting!
_size=$(stat -c "%s" "$_counter_file" 2>/dev/null || echo 0)
2009-11-19 08:48:19 +03:00
if [ $_size -ge $_limit ] ; then
echo "ERROR: more than $_limit consecutive failures for $service_name, marking cluster unhealthy"
2009-11-13 10:28:25 +03:00
exit 1
2009-11-19 08:48:19 +03:00
elif [ $_size -gt 0 -a -z "$_quiet" ] ; then
echo "WARNING: less than $_limit consecutive failures ($_size) for $service_name, not unhealthy yet"
2009-11-13 10:28:25 +03:00
fi
}
2010-11-17 05:50:56 +03:00
ctdb_check_counter_equal () {
_ctdb_counter_common
_limit=$1
# unary counting!
_size=$(stat -c "%s" "$_counter_file" 2>/dev/null || echo 0)
if [ $_size -eq $_limit ] ; then
return 1
fi
return 0
}
2009-11-13 10:28:25 +03:00
########################################################
2010-12-15 02:21:39 +03:00
ctdb_status_dir="$CTDB_VARDIR/status"
ctdb_fail_dir="$CTDB_VARDIR/failcount"
2010-12-15 02:45:17 +03:00
########################################################
# Managed status history, for auto-start/stop
ctdb_managed_dir="$CTDB_VARDIR/managed_history"
_ctdb_managed_common ()
{
_service_name="${1:-${service_name}}"
_ctdb_managed_file="$ctdb_managed_dir/$_service_name"
}
ctdb_service_managed ()
{
_ctdb_managed_common "$@"
mkdir -p "$ctdb_managed_dir"
touch "$_ctdb_managed_file"
}
ctdb_service_unmanaged ()
{
_ctdb_managed_common "$@"
rm -f "$_ctdb_managed_file"
}
is_ctdb_previously_managed_service ()
{
_ctdb_managed_common "$@"
[ -f "$_ctdb_managed_file" ]
}
########################################################
# Check and set status
2009-11-13 10:28:25 +03:00
2009-11-25 08:34:49 +03:00
log_status_cat ()
{
2010-03-10 12:39:31 +03:00
echo "node is \"$1\", \"${script_name}\" reports problem: $(cat $2)"
2009-11-25 08:34:49 +03:00
}
2009-11-13 10:28:25 +03:00
ctdb_checkstatus ()
{
2009-11-25 08:42:14 +03:00
if [ -r "$ctdb_status_dir/$script_name/unhealthy" ] ; then
log_status_cat "unhealthy" "$ctdb_status_dir/$script_name/unhealthy"
2009-11-13 10:28:25 +03:00
return 1
2009-11-25 08:42:14 +03:00
elif [ -r "$ctdb_status_dir/$script_name/banned" ] ; then
log_status_cat "banned" "$ctdb_status_dir/$script_name/banned"
2009-11-13 10:28:25 +03:00
return 2
else
return 0
fi
}
ctdb_setstatus ()
{
2009-11-25 08:42:14 +03:00
d="$ctdb_status_dir/$script_name"
2009-11-13 10:28:25 +03:00
case "$1" in
unhealthy|banned)
mkdir -p "$d"
cat "$2" >"$d/$1"
;;
*)
for i in "banned" "unhealthy" ; do
rm -f "$d/$i"
done
;;
esac
}
ctdb_service_needs_reconfigure ()
{
[ -e "$ctdb_status_dir/$service_name/reconfigure" ]
2009-09-30 15:05:16 +04:00
}
2009-11-13 10:28:25 +03:00
ctdb_service_set_reconfigure ()
{
d="$ctdb_status_dir/$service_name"
mkdir -p "$d"
>"$d/reconfigure"
}
ctdb_service_unset_reconfigure ()
{
rm -f "$ctdb_status_dir/$service_name/reconfigure"
}
ctdb_service_reconfigure ()
{
2011-01-11 09:10:55 +03:00
echo "Reconfiguring service \"$service_name\"..."
2009-11-13 10:28:25 +03:00
if [ -n "$service_reconfigure" ] ; then
2009-11-19 08:48:19 +03:00
eval $service_reconfigure
2009-11-13 10:28:25 +03:00
else
service "$service_name" restart
fi
ctdb_service_unset_reconfigure
ctdb_counter_init
}
ctdb_compat_managed_service ()
{
if [ "$1" = "yes" ] ; then
t="$t $2 "
fi
}
is_ctdb_managed_service ()
{
2010-11-18 03:04:52 +03:00
_service_name="${1:-${service_name}}"
2009-11-13 10:28:25 +03:00
t=" $CTDB_MANAGED_SERVICES "
ctdb_compat_managed_service "$CTDB_MANAGES_VSFTPD" "vsftpd"
ctdb_compat_managed_service "$CTDB_MANAGES_SAMBA" "samba"
ctdb_compat_managed_service "$CTDB_MANAGES_SCP" "scp"
2010-11-18 07:40:19 +03:00
ctdb_compat_managed_service "$CTDB_MANAGES_WINBIND" "winbind"
2009-11-13 10:28:25 +03:00
ctdb_compat_managed_service "$CTDB_MANAGES_HTTPD" "httpd"
ctdb_compat_managed_service "$CTDB_MANAGES_ISCSI" "iscsi"
ctdb_compat_managed_service "$CTDB_MANAGES_CLAMD" "clamd"
2009-11-25 08:34:49 +03:00
ctdb_compat_managed_service "$CTDB_MANAGES_NFS" "nfs"
2010-12-06 03:26:43 +03:00
ctdb_compat_managed_service "$CTDB_MANAGES_NFS" "nfs-ganesha-gpfs"
2009-11-13 10:28:25 +03:00
2010-11-18 03:04:52 +03:00
# Returns 0 if "<space>$_service_name<space>" appears in $t
[ "${t#* ${_service_name} }" != "${t}" ]
2009-11-13 10:28:25 +03:00
}
ctdb_start_stop_service ()
{
2010-11-18 03:04:52 +03:00
_service_name="${1:-${service_name}}"
2011-01-11 09:12:03 +03:00
[ "$event_name" = "monitor" ] || return 0
2010-12-15 02:45:17 +03:00
if is_ctdb_managed_service "$_service_name" ; then
if ! is_ctdb_previously_managed_service "$_service_name" ; then
2010-11-18 03:04:52 +03:00
echo "Starting service $_service_name"
2009-11-13 10:28:25 +03:00
ctdb_service_start || exit $?
2010-12-15 02:45:17 +03:00
ctdb_service_managed "$_service_name"
2009-11-13 10:28:25 +03:00
exit 0
fi
2010-11-18 07:40:19 +03:00
else
2010-12-15 02:45:17 +03:00
if is_ctdb_previously_managed_service "$_service_name" ; then
2010-11-18 03:04:52 +03:00
echo "Stopping service $_service_name"
2009-11-13 10:28:25 +03:00
ctdb_service_stop || exit $?
2010-12-15 02:45:17 +03:00
ctdb_service_unmanaged "$_service_name"
2010-11-18 07:40:19 +03:00
exit 0
2009-11-13 10:28:25 +03:00
fi
fi
}
ctdb_service_start ()
{
if [ -n "$service_start" ] ; then
2010-11-18 03:27:10 +03:00
eval $service_start || return $?
2009-11-13 10:28:25 +03:00
else
2010-11-18 03:27:10 +03:00
service "$service_name" start || return $?
2009-11-13 10:28:25 +03:00
fi
ctdb_counter_init
}
ctdb_service_stop ()
{
if [ -n "$service_stop" ] ; then
2009-11-19 08:48:19 +03:00
eval $service_stop
2009-11-13 10:28:25 +03:00
else
service "$service_name" stop
fi
}
2009-12-01 09:43:47 +03:00
ctdb_standard_event_handler ()
{
case "$1" in
status)
ctdb_checkstatus
exit
;;
setstatus)
2010-03-10 12:39:31 +03:00
shift
2009-12-01 09:43:47 +03:00
ctdb_setstatus "$@"
exit
;;
esac
}
2009-12-11 21:56:36 +03:00
ipv4_host_addr_to_net_addr()
{
local HOST=$1
local MASKBITS=$2
local HOST0=$(echo $HOST | awk -F . '{print $4}')
local HOST1=$(echo $HOST | awk -F . '{print $3}')
local HOST2=$(echo $HOST | awk -F . '{print $2}')
local HOST3=$(echo $HOST | awk -F . '{print $1}')
local HOST_NUM=$(( $HOST0 + $HOST1 * 256 + $HOST2 * (256 ** 2) + $HOST3 * (256 ** 3) ))
local MASK_NUM=$(( ( (2**32 - 1) * (2**(32 - $MASKBITS)) ) & (2**32 - 1) ))
local NET_NUM=$(( $HOST_NUM & $MASK_NUM))
local NET0=$(( $NET_NUM & 255 ))
local NET1=$(( ($NET_NUM & (255 * 256)) / 256 ))
local NET2=$(( ($NET_NUM & (255 * 256**2)) / 256**2 ))
local NET3=$(( ($NET_NUM & (255 * 256**3)) / 256**3 ))
echo "$NET3.$NET2.$NET1.$NET0"
}
ipv4_maskbits_to_net_mask()
{
local MASKBITS=$1
local MASK_NUM=$(( ( (2**32 - 1) * (2**(32 - $MASKBITS)) ) & (2**32 - 1) ))
local MASK0=$(( $MASK_NUM & 255 ))
local MASK1=$(( ($MASK_NUM & (255 * 256)) / 256 ))
local MASK2=$(( ($MASK_NUM & (255 * 256**2)) / 256**2 ))
local MASK3=$(( ($MASK_NUM & (255 * 256**3)) / 256**3 ))
echo "$MASK3.$MASK2.$MASK1.$MASK0"
}
ipv4_is_valid_addr()
{
local ADDR=$1
local fail=0
local N=`echo $ADDR | sed -e 's/[0-9][0-9]*\.[0-9][0-9]*\.[0-9][0-9]*\.[0-9][0-9]*//'`
test -n "$N" && fail=1
local ADDR0=$(echo $ADDR | awk -F . '{print $4}')
local ADDR1=$(echo $ADDR | awk -F . '{print $3}')
local ADDR2=$(echo $ADDR | awk -F . '{print $2}')
local ADDR3=$(echo $ADDR | awk -F . '{print $1}')
test "$ADDR0" -gt 255 && fail=1
test "$ADDR1" -gt 255 && fail=1
test "$ADDR2" -gt 255 && fail=1
test "$ADDR3" -gt 255 && fail=1
test x"$fail" != x"0" && {
#echo "IPv4: '$ADDR' is not a valid address"
return 1;
}
return 0;
}
2010-07-12 09:41:42 +04:00
# iptables doesn't like being re-entered, so flock-wrap it.
iptables()
{
2011-06-28 09:04:58 +04:00
flock -w 30 $CTDB_VARDIR/iptables-ctdb.flock /sbin/iptables "$@"
2010-07-12 09:41:42 +04:00
}
2010-08-26 08:59:59 +04:00
########################################################
# tickle handling
########################################################
# Temporary directory for tickles.
2010-09-03 06:35:25 +04:00
tickledir="$CTDB_VARDIR/state/tickles"
2010-08-26 08:59:59 +04:00
mkdir -p "$tickledir"
update_tickles ()
{
_port="$1"
mkdir -p "$tickledir" # Just in case
# Who am I?
_pnn=$(ctdb pnn) ; _pnn=${_pnn#PNN:}
# What public IPs do I hold?
_ips=$(ctdb -Y ip | awk -F: -v pnn=$_pnn '$3 == pnn {print $2}')
# IPs as a regexp choice
_ipschoice="($(echo $_ips | sed -e 's/ /|/g' -e 's/\./\\\\./g'))"
# Record connections to our public IPs in a temporary file
_my_connections="${tickledir}/${_port}.connections"
rm -f "$_my_connections"
netstat -tn |
awk -v destpat="^${_ipschoice}:${_port}\$" \
'$1 == "tcp" && $6 == "ESTABLISHED" && $4 ~ destpat {print $5, $4}' |
sort >"$_my_connections"
# Record our current tickles in a temporary file
_my_tickles="${tickledir}/${_port}.tickles"
rm -f "$_my_tickles"
for _i in $_ips ; do
ctdb -Y gettickles $_i $_port |
awk -F: 'NR > 1 { printf "%s:%s %s:%s\n", $2, $3, $4, $5 }'
done |
sort >"$_my_tickles"
# Add tickles for connections that we haven't already got tickles for
comm -23 "$_my_connections" "$_my_tickles" |
while read _src _dst ; do
ctdb addtickle $_src $_dst
done
# Remove tickles for connections that are no longer there
comm -13 "$_my_connections" "$_my_tickles" |
while read _src _dst ; do
ctdb deltickle $_src $_dst
done
rm -f "$_my_connections" "$_my_tickles"
}
2008-04-10 00:50:12 +04:00
########################################################
# load a site local config file
########################################################
2011-06-28 09:06:10 +04:00
[ -n "$CTDB_RC_LOCAL" -a -x "$CTDB_RC_LOCAL" ] && {
. "$CTDB_RC_LOCAL"
}
2008-04-10 01:01:22 +04:00
[ -x $CTDB_BASE/rc.local ] && {
2008-04-10 00:50:12 +04:00
. $CTDB_BASE/rc.local
}
2008-08-20 03:47:00 +04:00
2009-10-19 09:22:15 +04:00
[ -d $CTDB_BASE/rc.local.d ] && {
for i in $CTDB_BASE/rc.local.d/* ; do
[ -x "$i" ] && . "$i"
done
}
2009-12-01 09:43:47 +03:00
script_name="${0##*/}" # basename
service_name="$script_name" # default is just the script name
2009-11-19 08:48:19 +03:00
service_fail_limit=1
2011-01-11 09:12:03 +03:00
event_name="$1"