#!/bin/sh
# event strict to manage vsftpd in a cluster environment

. $CTDB_BASE/functions
loadconfig ctdb
loadconfig vsftpd

[ "$CTDB_MANAGES_VSFTPD" = "yes" ] || exit 0

cmd="$1"
shift

# Count the number of monitor failures.  The cluster only becomes
# unhealthy after 2 failures.
VSFTPD_FAILS="fail-count"
VSFTPD_LIMIT=2

case $cmd in 
     startup)
	/bin/mkdir -p $CTDB_BASE/state/vsftpd

	# make sure the service is stopped first
	service vsftpd stop > /dev/null 2>&1
	service vsftpd start

	ctdb_counter_init "$VSFTPD_FAILS"
	;;

     shutdown)
	service vsftpd stop
	;;

     takeip)
	echo "restart" > $CTDB_BASE/state/vsftpd/restart
	;;

     releaseip)
	echo "restart" > $CTDB_BASE/state/vsftpd/restart
	;;

     recovered)
	# if we have taken or released any ips we must 
	# restart vsftpd to ensure that all tcp connections are reset
	[ -f $CTDB_BASE/state/vsftpd/restart ] && {
		service vsftpd stop > /dev/null 2>&1
		service vsftpd start
		/bin/rm -f $CTDB_BASE/state/vsftpd/restart 2>/dev/null
		ctdb_counter_init "$VSFTPD_FAILS"
	} >/dev/null 2>&1
	;;

     monitor)
	# Subshell catches the "exit 1"
	if (ctdb_check_tcp_ports "ftp" 21) ; then
	    ctdb_counter_init "$VSFTPD_FAILS"
	else
	    ctdb_counter_incr "$VSFTPD_FAILS"
	    if ctdb_counter_limit "$VSFTPD_FAILS" $VSFTPD_LIMIT ; then
		echo "ERROR: more than $VSFTPD_LIMIT consecutive failures, marking cluster unhealthy"
		exit 1
	    else
		echo "WARNING: less than $VSFTPD_LIMIT consecutive failures, not unhealthy yet"
	    fi
		
	fi
	;;
esac

exit 0