#!/bin/bash
# Disk usage monitorization plugin for Pandora FMS agent
# Author: Manuel A. Gulín Bejarano
# Version: 1.2
# Date: 19/03/2013
#
# This program is free software: you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation, either version 3 of the License, or
#  (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
#  along with this program.  If not, see <http://www.gnu.org/licenses/>.

function help {
        echo -e "Disk usage monitorization agent plugin for Pandora FMS agent. http://pandorafms.com" 
        echo -e "Syntax:"
        echo -e "\t\t$0"
        exit
}

if [ $# -ne 0 ]
then
        help
        exit -1
fi


# Info about used swap - all retrieved with one query, later it's going to be appropiately parsed
SWP_INFO=`free -m | grep Swap:`

# Info about swap
SWP_SPACE=`echo $SWP_INFO | awk '{print $2}'`
SWP_USAGE=`echo $SWP_INFO | awk '{print $3}'`
SWP_USAGE100=`echo "$SWP_USAGE $SWP_SPACE"  | awk '{print ($1*100/$2)}'`

# Info about used disks - all retrieved with one query, later it's going to be appropiately parsed
DISK_INFO=`df -m | tail --lines=+2`

# Loop to iterate through disks; size info
df -Pm | tail --lines=+2 | while read line; do
	DISK=`echo $line | awk '{print $1}'`

	if [[ "$DISK" == /dev/sd* || "$DISK" == /dev/hd* || "$DISK" == /dev/mapper/* ]]; then
		SPACE=`echo $line | awk '{print $2}'`
		USAGE=`echo $line | awk '{print $3}'`
		USAGE100=`echo $line | awk '{print $5}' | grep -o [0-9]*`
		MOUNT=`echo $line | awk '{print $6}'`

		# Module to get the total space on disk
		echo -e "\t<module>"
		echo -e "\t<name><![CDATA[Disk: Total space on $MOUNT (MB)]]></name>"
		echo -e "\t<description><![CDATA[Total disk space on $MOUNT - $DISK in megabytes]]></description>"
		echo -e "\t<type>async_data</type>"
		echo -e "\t<unit>MB</unit>"
		echo -e "\t<min>0</min>"
		echo -e "\t<disabled>0</disabled>"
		echo -e "\t<data><![CDATA[$SPACE]]></data>"
		#echo -e "\t<crontab>0 * * * *</crontab>""
		echo -e "\t</module>"

		# Module to get the free space on disk
		echo -e "\t<module>"
		echo -e "\t<name><![CDATA[Disk: Used space on $MOUNT (MB)]]></name>"
		echo -e "\t<description><![CDATA[Used disk space on $MOUNT - $DISK in megabytes]]></description>"
		echo -e "\t<type>generic_data</type>"
		echo -e "\t<unit>MB</unit>"
		echo -e "\t<min>0</min>"
		echo -e "\t<disabled>0</disabled>"
		echo -e "\t<data><![CDATA[$USAGE]]></data>"
		echo -e "\t</module>"

		# Module to get the % of used space on disk
		echo -e "\t<module>"
		echo -e "\t<name><![CDATA[Disk: Used space on $MOUNT (%)]]></name>"
		echo -e "\t<description><![CDATA[Percentage of used disk space on $MOUNT - $DISK]]></description>"
		echo -e "\t<type>generic_data</type>"
		echo -e "\t<unit>%</unit>"
		echo -e "\t<min>0</min>"
		echo -e "\t<max>100</max>"
		echo -e "\t<min_warning>85</min_warning>"
		echo -e "\t<max_warning>89.9</max_warning>"
		echo -e "\t<min_critical>90</min_critical>"
		echo -e "\t<max_critical>100</max_critical>"
		echo -e "\t<disabled>0</disabled>"
		echo -e "\t<data><![CDATA[$USAGE100]]></data>"
		echo -e "\t</module>"
	fi
done


# Module to get the total swap space
echo -e "\t<module>"
echo -e "\t<name><![CDATA[Disk: Total swap space (MB)]]></name>"
echo -e "\t<description><![CDATA[Total swap space in megabytes]]></description>"
echo -e "\t<type>async_data</type>"
echo -e "\t<unit>MB</unit>"
echo -e "\t<min>0</min>"
echo -e "\t<disabled>0</disabled>"
echo -e "\t<data><![CDATA[$SWP_SPACE]]></data>"
#echo -e "\t<crontab>0 * * * *</crontab>""
echo -e "\t</module>"

# Module to get the used swap space
echo -e "\t<module>"
echo -e "\t<name><![CDATA[Disk: Used swap space (MB)]]></name>"
echo -e "\t<description><![CDATA[Used swap space in megabytes]]></description>"
echo -e "\t<type>generic_data</type>"
echo -e "\t<unit>MB</unit>"
echo -e "\t<min>0</min>"
echo -e "\t<max>$SWP_SPACE</max>"
echo -e "\t<min_warning>100</min_warning>"
echo -e "\t<max_warning>499</max_warning>"
echo -e "\t<min_critical>500</min_critical>"
echo -e "\t<max_critical>$SWP_SPACE</max_critical>"
echo -e "\t<disabled>0</disabled>"
echo -e "\t<data><![CDATA[$SWP_USAGE]]></data>"
echo -e "\t</module>"

# Module to get the % of used swap space
echo -e "\t<module>"
echo -e "\t<name><![CDATA[Disk: Used swap space (%)]]></name>"
echo -e "\t<description><![CDATA[Percentage of used swap space]]></description>"
echo -e "\t<type>generic_data</type>"
echo -e "\t<unit>%</unit>"
echo -e "\t<min>0</min>"
echo -e "\t<max>100</max>"
echo -e "\t<min_warning>40</min_warning>"
echo -e "\t<max_warning>89.9</max_warning>"
echo -e "\t<min_critical>90</min_critical>"
echo -e "\t<max_critical>100</max_critical>"
echo -e "\t<disabled>0</disabled>"
echo -e "\t<data><![CDATA[$SWP_USAGE100]]></data>"
echo -e "\t</module>"


# Loop to iterate through disks; I/O info
for line in `ls /dev/{hd,sd}? 2> /dev/null`; do
	DISK_D=`echo $line | cut -d "/" -f3`
	IO_INFO=`iostat -m $DISK | grep $DISK_D`

        if [ ! -z "$IO_INFO" ]; then
		IO_TPS=`echo $IO_INFO | awk '{print $2}'`
		IO_RMBS=`echo $IO_INFO | awk '{print $3}'`
		IO_WMBS=`echo $IO_INFO | awk '{print $4}'`

		# Module to get the transfers per second on disk
		echo -e "\t<module>"
		echo -e "\t<name><![CDATA[Disk IO: Tps on $DISK_D]]></name>"
		echo -e "\t<description><![CDATA[Transfers per second on $DISK_D (tps)]]></description>"
		echo -e "\t<type>generic_data</type>"
		echo -e "\t<unit>tps</unit>"
		echo -e "\t<min>0</min>"
		echo -e "\t<disabled>0</disabled>"
		echo -e "\t<data><![CDATA[$IO_TPS]]></data>"
		echo -e "\t</module>"

		# Module to get the read speed on disk
		echo -e "\t<module>"
		echo -e "\t<name><![CDATA[Disk IO: Read speed on $DISK_D (MB/s)]]></name>"
		echo -e "\t<description><![CDATA[Read speed on $DISK_D in megabytes per second]]></description>"
		echo -e "\t<type>generic_data</type>"
		echo -e "\t<unit>MB/sec</unit>"
		echo -e "\t<min>0</min>"
		echo -e "\t<disabled>0</disabled>"
		echo -e "\t<data><![CDATA[$IO_RMBS]]></data>"
		echo -e "\t</module>"

		# Module to get the write speed on disk
		echo -e "\t<module>"
		echo -e "\t<name><![CDATA[Disk IO: Write speed on $DISK_D (MB/s)]]></name>"
		echo -e "\t<description><![CDATA[Write speed on $DISK_D in megabytes per second]]></description>"
		echo -e "\t<type>generic_data</type>"
		echo -e "\t<unit>MB/sec</unit>"
		echo -e "\t<min>0</min>"
		echo -e "\t<disabled>0</disabled>"
		echo -e "\t<data><![CDATA[$IO_WMBS]]></data>"
		echo -e "\t</module>"
	fi
done