#!/bin/bash
#
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice, this
#   list of conditions and the following disclaimer.
#
# * Redistributions in binary form must reproduce the above copyright notice,
#   this list of conditions and the following disclaimer in the documentation
#   and/or other materials provided with the distribution.
#
# * Neither the name of the Novell nor the names of its contributors may be
#   used to endorse or promote products derived from this software without
#   specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#

export PATH="/usr/local/bin:/usr/bin:/bin"

REVISION=0.0.2
PROGNAME=$(basename "$0")
PROGPATH=$(dirname "$PROGNAME")
STATEFILE='/var/run/keepalived.state'
PIDFILE='/var/run/keepalived.pid'

PROGPATH='/usr/lib/nagios/plugins/'

if [ -r "$PROGPATH/utils.sh" ]; then
    . "$PROGPATH/utils.sh"
fi

print_usage() {
    echo "Usage: $PROGNAME [-F keepalived_statefile] [-p keepalived_pidfile] [-S|-M] [-h] [-v]"
    echo "       -F keepalived_statefile : URI to the status file (default: $STATEFILE)"
    echo "       -p keepalived_pidfile   : URI to the pidfile of the keepalived process (default: $PIDFILE)"
    echo "       -S                      : expect the machine to run in SLAVE state"
    echo "       -M                      : expect the machine to run in MASTER state"
    echo "       -h                      : print this usage"
    echo "       -v                      : print version information"
    echo
    echo "To successfully run this script, the following pre-requires need to be fulfilled:"
    echo "* in your keepalived.conf, add 'notify /usr/bin/keepalived_notify_monitoring.sh'" 
    echo "  in at least one vrrp_instance section (see 'keepalived_notify_monitoring.sh -h'"
    echo "  output for more details"
    echo "* keepalive should be restarted after the changes"
    echo    
}

print_help() {
    print_revision $PROGNAME $REVISION
    echo
    print_usage
    echo
    support
    exit $STATE_OK
}

while getopts 'F:p:SMhvV' OPTION ; do
    case $OPTION in
        v|V)
            print_revision $PROGNAME $REVISION
            exit $STATE_OK
        ;;
        F)
            STATEFILE="$OPTARG"
        ;;
        M)
            MASTER='true'
        ;;
        p)
            PIDFILE="$OPTARG"
        ;;
        S)
            SLAVE='true'
        ;;
        h)
            print_help
            exit $STATE_OK
        ;;
        *)
            echo "Unkown argument: $OPTION" >&2
            print_usage
            exit $STATE_UNKNOWN
        ;;
    esac
done

if [ -r "$PIDFILE" ]; then
    PID=$(cat "$PIDFILE")
    KEEPALIVERUN=$(cat /proc/$PID/status | grep ^State: | awk '" " { print $2 }')
    case $KEEPALIVERUN in
        S|R)
            KEEPALIVESTATE=$STATE_OK
        ;;
        Z)
            echo "WARNING: main keepalived process is a Zombie"
            exit $STATE_WARNING
        ;;
        *)
            echo "UNKOWN: found $PIDFILE but got no state for keepalived"
            exit $STATE_UNKNOWN
        ;;
    esac
else
    KEEPALIVENUM=$(ps ax | grep [k]eepalived | wc -l | tr -d "\n")
    if [ $KEEPALIVENUM -gt 0 ]; then
        KEEPALIVESTATE=$STATE_OK
    fi
fi

if [ $KEEPALIVESTATE -eq $STATE_OK ]; then
    if [ -r "$STATEFILE" ]; then
        STATE=$(cat "$STATEFILE")
    else
        echo "UNKOWN: could not read $STATEFILE"
        exit $STATE_UNKNOWN
    fi
    if [[ $STATE == *"FAULT"* ]];then
        echo $STATE
        exit $STATE_CRITICAL
    fi

    if [ "$MASTER" == "true" ]; then
        if [[ $STATE == *"MASTER"* ]]; then
            echo $STATE
            exit $STATE_OK
        fi
        if [[ $STATE == *"BACKUP"* ]]; then
            echo $STATE
            exit $STATE_CRITICAL
        fi
    elif [ "$SLAVE" == "true" ]; then
        if [[ $STATE == *"BACKUP"* ]]; then
            echo $STATE
            exit $STATE_OK
        fi
        if [[ $STATE == *"MASTER"* ]]; then
            echo $STATE
            exit $STATE_CRITICAL
        fi
    else
        if [[ $STATE == *"BACKUP"* ]] || [[ $STATE == *"MASTER"* ]]; then
            echo $STATE
            exit $STATE_OK
        fi
    fi
else
    echo "Keepalived is in UNKNOWN state"
    exit $STATE_UNKNOWN
fi
