#!/bin/sh

# Copyright 2015 Claudiu Vasadi. All rights reserved.
#
# Redistribution and use in source and binary forms, with or without modification, are
# permitted provided that the following conditions are met:
#
#   1. Redistributions of source code must retain the above copyright notice, this list of
#      conditions and the following disclaimer.
#
#   2. 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.
#
# THIS SOFTWARE IS PROVIDED BY Claudiu Vasadi ''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 Claudiu Vasadi 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.
#
# The views and conclusions contained in the software and documentation are those of the
# authors and should not be interpreted as representing official policies, either expressed
# or implied, of Claudiu Vasadi.
#
#
# Script to check the pool usage percentage
#
# Usage:
# ./check_zpool_size -w <warning_percentage> -c <critic_percentage> -p <pool_name>
#
#
# Date: 25.11.2013
# Author: Claudiu Vasadi
#

. /usr/local/libexec/nagios/utils.sh


while getopts c:w:p: opt
do
  case $opt in
    w) warn="$OPTARG"
      ;;
    c) crit="$OPTARG"
      ;;
    p) pool="$OPTARG"
      ;;
  esac
done

# $warn and $crit need to be specified 
if [ -z $warn ];then
  echo "-w needs to be specified"
  exit $STATE_CRITICAL
fi
echo $warn | grep -Eq '^(\+|-)?[0-9]+$'
if [ $? != "0" ];then
  echo "-w must be an integer"
  exit $STATE_CRITICAL
fi


if [ -z $crit ];then
  echo "-c needs to be specified"
  exit $STATE_CRITICAL
fi
echo $crit | grep -Eq '^(\+|-)?[0-9]+$'
if [ $? != "0" ];then
  echo "-c must be an integer"
  exit $STATE_CRITICAL
fi


# $warn cannot be higher than $crit
if [ $warn -ge $crit ];then
  echo "-w cannot be equal or higher than -c"
  exit $STATE_CRITICAL
fi

if [ -z $pool ];then
  echo "-p not specified"
  exit $STATE_CRITICAL
fi


poolsize="`zpool get capacity $pool |  awk 'FNR == 2 {print $3}' | sed -e 's/\%//g'`"

if [ $poolsize -ge $warn ] && [ $poolsize -lt $crit ];then
  echo "WARNING: \"$pool\" zfs pool size is $poolsize%"
  exit $STATE_WARNING
elif [ $poolsize -gt $crit ];then
  echo "CRITICAL: \"$pool\" zfs pool size is $poolsize%"
  exit $STATE_CRITICAL
elif [ $poolsize -lt $warn ];then
  echo "OK: \"$pool\" zfs pool size is $poolsize%"
  exit $STATE_OK
fi