check_upis.sh

#!/bin/bash
# UPiS plugin for Nagios
# Written by Udo Seidel
#
# Description:
#
# This plugin will check the status of a UPiS connected to the RPi
#
# Location of the sudo and i2cget command (if not in path)
SUDO="/usr/bin/sudo"
I2CGET="/usr/sbin/i2cget"
BC="/usr/bin/bc"
MYTEST=""
CUSTOMWARNCRIT=0 # no external defined warning and critical levels

# sudo is needed if i2cget cannot be executed by the nagios 
# user context w/o sudo granted priviledges


# Nagios return codes
STATE_OK=0
STATE_WARNING=1
STATE_CRITICAL=2
STATE_UNKNOWN=3
STATE_DEPENDENT=4

EXITSTATUS=$STATE_UNKNOWN #default


PROGNAME=`basename $0`

print_usage() {
	echo 
	echo " This plugin will check the power and temperature status of an locally attached UPiS."
	echo 
	echo 
        echo " Usage: $PROGNAME -<p|t|h> -w <warning level> -c <critical level>"
        echo
        echo "   -p: Power status"
        echo "   -t: Temperature in Grad Celsius"
        echo "   -w: WARNING level for power/temperature"
        echo "   -c: CRITICAL level for power/temperature" 
	echo 
}

if [ "$#" -lt 1 ]; then
	print_usage
        EXITSTATUS=$STATE_UNKNOWN
        exit $EXITSTATUS
fi

check_i2cget() {
if [ ! -x "$I2CGET" ]
then
        echo "UNKNOWN: $I2CGET not found or is not executable by the nagios user"
        EXITSTATUS=$STATE_UNKNOWN
        exit $EXITSTATUS
fi
}


check_power() {
I2CGET_ARG="-y 1 0x6A 0x00"
# Run basic i2cget and find our status
I2CGET_OUTPUT=`$SUDO $I2CGET $I2CGET_ARG 2>&1`

if [ $? -ne 0 ]
then
EXITSTATUS=$STATE_CRITICAL
else
EXITSTATUS=$STATE_OK
fi

CLEANED_I2CGET_OUTPUT=`sudo $I2CGET $I2CGET_ARG |sed -e 's/0x//g'|awk '{print $1}' 2>&1`

if [ $CUSTOMWARNCRIT -ne 0 ]; then
	case $CLEANED_I2CGET_OUTPUT in
		$WARNLEVEL)
		EXITSTATUS=$STATE_WARNING
		POWERMSG="UPiS WARNING - "
		;;
		$CRITLEVEL)
		EXITSTATUS=$STATE_CRITICAL
		POWERMSG="UPiS CRITICAL - "
		;;
		*)
		EXITSTATUS=$STATE_OK
		POWERMSG="UPiS OK - "
		;;
	esac
else
	case $CLEANED_I2CGET_OUTPUT in
		1)
		EXITSTATUS=$STATE_OK
		POWERMSG="UPiS OK - "
		;;
		2)
		EXITSTATUS=$STATE_OK
		POWERMSG="UPiS OK - "
		;;
		3)
		EXITSTATUS=$STATE_OK
		POWERMSG="UPiS OK - "
		;;
		4)
		EXITSTATUS=$STATE_WARNING
		POWERMSG="UPiS WARNING - "
		;;
		5)
		EXITSTATUS=$STATE_CRITICAL
		POWERMSG="UPiS WARNING - "
		;;
		*)
		EXITSTATUS=$STATE_UNKNOWN
		POWERMSG="UPiS UNKNOWN "
		;;
	esac
fi

case $CLEANED_I2CGET_OUTPUT in
                1)
                POWERMSG="$POWERMSG EPR | EPR (External Power)"
                ;;
                2)
                POWERMSG="$POWERMSG USB | USB (USB Power)"
                ;;
                3)
                POWERMSG="$POWERMSG RPI | RPI (RPI USB Power)"
                ;;
                4)
                POWERMSG="$POWERMSG BAT | BAT (Battery Power)"
                ;;
                5)
                POWERMSG="$POWERMSG LPR | LPR (Low Power)"
                ;;
                *)
                POWERMSG="$POWERMSG | UPiS Status Unknown"
                ;;
esac

echo $POWERMSG

}

bcdbyte2dec() {
# convert the hex output to a proper dec number
if [ ! -x "$BC" ]
then
        echo "UNKNOWN: $BC not found or is not executable by the nagios user"
        EXITSTATUS=$STATE_UNKNOWN
        exit $EXITSTATUS
fi

DEC=`echo "ibase=16; $1" |bc -l`
A=$((DEC/16))
A=$((A&15))
B=$((DEC&15))
RESULT=`echo "10*$A+$B"|bc -l`
}

check_temperature() {
I2CGET_ARG="-y 1 0x6A 0x0B"
# Run basic i2cget and find our status
I2CGET_OUTPUT=`$SUDO $I2CGET $I2CGET_ARG 2>&1`

if [ $? -ne 0 ]
then
EXITSTATUS=$STATE_CRITICAL
else
EXITSTATUS=$STATE_OK
fi

CLEANED_I2CGET_OUTPUT=`sudo $I2CGET $I2CGET_ARG |sed -e 's/0x//g'|awk '{print $1}' 2>&1`

bcdbyte2dec $CLEANED_I2CGET_OUTPUT

if [ $CUSTOMWARNCRIT -ne 0 ]; then
	# check if the levels are integers
	echo $WARNLEVEL | awk '{ exit ! /^[0-9]+$/ }'
	if [ $? -ne 0 ]; then
		echo " warning level ($WARNLEVEL) is not an integer"
		exit $STATE_UNKNOWN
	fi
	echo $CRITLEVEL | awk '{ exit ! /^[0-9]+$/ }'
	if [ $? -ne 0 ]; then
		echo " critical level ($CRITLEVEL) is not an integer"
		exit $STATE_UNKNOWN
	fi
	if [ $WARNLEVEL -gt $CRITLEVEL ]; then
		echo
		echo " The value for critical level has to be equal or higher than the one for warning level"
		echo " Your values are: critcal ($CRITLEVEL) and warning ($WARNLEVEL)"
		echo
		exit $STATE_UNKNOWN
	fi
	if [ $CLEANED_I2CGET_OUTPUT -lt $WARNLEVEL ]; then
		EXITSTATUS=$STATE_OK
		echo "Temperature OK - $CLEANED_I2CGET_OUTPUT 'C | $CLEANED_I2CGET_OUTPUT"
	else
		EXITSTATUS=$STATE_WARNING
		if [ $CLEANED_I2CGET_OUTPUT -lt $CRITLEVEL ]; then
			echo "Temperature WARNING - $CLEANED_I2CGET_OUTPUT 'C | $CLEANED_I2CGET_OUTPUT"
		else
			EXITSTATUS=$STATE_CRITICAL
				echo "Temperature CRITICAL - $CLEANED_I2CGET_OUTPUT 'C | $CLEANED_I2CGET_OUTPUT"
		fi
	fi


else
	echo "Temperature OK - $CLEANED_I2CGET_OUTPUT 'C | $CLEANED_I2CGET_OUTPUT"
fi
}


while getopts "hptw:c:" OPT
do		
	case "$OPT" in
	h)
		print_usage
		exit $STATE_UNKNOWN
		;;
	p)
		MYCHECK=power
		;;
	t)
		MYCHECK=temperature
		;;
        w)
                WARNLEVEL=$3
		CUSTOMWARNCRIT=1
                ;;
        c)
                CRITLEVEL=$5
		CUSTOMWARNCRIT=1
                ;;
	*)
		print_usage
		exit $STATE_UNKNOWN
	esac
done

check_i2cget
check_$MYCHECK

exit $EXITSTATUS