check_cpu_proc/check_cpu_proc.sh at master · thomasweaver/check_cpu_proc · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
#!/bin/bash

CRITICAL=0
WARNING=0
PROC=0
EXITCODE=0
MEMWARN=0
MEMCRIT=0

function Usage {
        echo "
        This script looks at a command and its processes and calculates its CPU and memory usage

        OPTIONS:
        -p - The process name to look for
        -w - The warning to use for the CPU percentage used
        -c - The critical to use for the CPU percentage used
        -m - The warning to use for the Memory percentage used
        -n - The critical to use for the Memory percentage used

        EXAMPLES:
                Check the usage for apache processes and alert warning if over 80% CPU utilised and critical if 90%
                ./check_cpu_proc.sh -p apache2 -w 80 -c 90

                Check the usage for nagios processes and alert warning if over 20% memory Utilised and critical if 30%
                ./check_cpu_proc.sh -p nagios -m 20 -n 30
"
        exit 3
}

while getopts "p:w:c:m:n:" OPTION
do
        case $OPTION in
                p)
                        PROC=$OPTARG
                  ;;

                w)
                        WARNING=$OPTARG
                  ;;

                c)
                        CRITICAL=$OPTARG
                  ;;

                m)
                        MEMWARN=$OPTARG
                  ;;

                n)
                        MEMCRIT=$OPTARG
                  ;;
        esac
done;

if [[ $PROC == 0 ]]; then
        echo "Must specify a process name"
        Usage
fi

PSOUTPUT=`ps aux | grep $PROC`
OIFS="${IFS}"
NIFS=$'\n'

IFS="${NIFS}"

OVERALCPU=0.0
OVERALMEM=0.0
OVERALRSS=0.0
OVERALVSZ=0.0
COUNT=0

for LINE in ${PSOUTPUT}; do
        CPU=$(echo $LINE | awk '{ print $3 }')
        COMMAND=$(echo $LINE | awk '{ print $11 }')
        MEM=$(echo $LINE | awk '{ print $4 }')
        RSS=$(echo $LINE | awk '{ print $6 }')
        VSZ=$(echo $LINE | awk '{ print $5 }')

        if [[ $COMMAND == *$PROC* ]]; then
                OVERALCPU=`echo "${OVERALCPU} + ${CPU}" | bc -l`
                OVERALMEM=`echo "${OVERALMEM} + ${MEM}" | bc -l`
                OVERALRSS=`echo "${OVERALRSS} + ${RSS}" | bc -l`
                OVERALVSZ=`echo "${OVERALVSZ} + ${VSZ}" | bc -l`
                COUNT=`echo "${COUNT} + 1" | bc -l`
                ACTCOMMAND=$COMMAND
        fi

done

if [ $WARNING != 0 ] || [ $CRITICAL != 0 ]; then
        if [ $WARNING == 0 ] || [ $CRITICAL == 0 ]; then
                echo "Must Specify both warning and critical"
                Usage
        fi

        #Work out CPU
        if [ `echo $OVERALCPU'>'$WARNING | bc -l` == 1 ]; then
                #echo $OVERALCPU'>'$WARNING
                #echo $OVERALCPU'>'$WARNING | bc -l
                EXITCODE=1

                if [ `echo $OVERALCPU'>'$CRITICAL | bc -l` == 1 ]; then
                        #echo $OVERALCPU'>'$CRITICAL
                        #echo $OVERALCPU'>'$CRITICAL | bc -l
                        EXITCODE=2
                fi
        fi
fi

if [ $MEMWARN != 0 ] || [ $MEMCRIT != 0 ]; then
        if [ $MEMWARN == 0 ] || [ $MEMCRIT == 0 ]; then
                echo "Must Specify both warning and critical"
                Usage
        fi

        #Work out Memory
        if [ `echo $OVERALMEM'>'$MEMWARN | bc -l` == 1 ]; then
                #echo $OVERALCPU'>'$WARNING
                #echo $OVERALCPU'>'$WARNING | bc -l
                EXITCODE=1

                if [ `echo $OVERALMEM'>'$MEMCRIT | bc -l` == 1 ]; then
                        #echo $OVERALCPU'>'$CRITICAL
                        #echo $OVERALCPU'>'$CRITICAL | bc -l
                        EXITCODE=2
                fi
        fi
fi

EXITTEXT="OK"

case "$EXITCODE" in
        1)
                EXITTEXT="WARNING"
        ;;

        2)
                EXITTEXT="CRITICAL"
        ;;

        3)
                EXITTEXT="UNKNOWN"
        ;;
esac


IFS="${OIFS}"

echo "${EXITTEXT} ${ACTCOMMAND} CPU: ${OVERALCPU}% MEM: ${OVERALMEM}% over ${COUNT} processes | proc=${COUNT} mem=${OVERALMEM}% cpu=${OVERALCPU}% rss=${OVERALRSS}KB vsz=${OVERALVSZ}KB"

exit $EXITCODE