aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorIsmaël Bouya <ismael.bouya@normalesup.org>2019-10-24 00:01:47 +0200
committerIsmaël Bouya <ismael.bouya@normalesup.org>2019-10-24 00:01:47 +0200
commit64d41da73cce0005757ca0f7dd88151f9ba5cf4a (patch)
treee4aa39affac1b965b9f99f7c2928aaddb4439c5d
parent441031085f3144247ae97489e4b08cdb41a17e4b (diff)
downloadPuppet-64d41da73cce0005757ca0f7dd88151f9ba5cf4a.tar.gz
Puppet-64d41da73cce0005757ca0f7dd88151f9ba5cf4a.tar.zst
Puppet-64d41da73cce0005757ca0f7dd88151f9ba5cf4a.zip
Use json for send_nrdp
Diffstat
-rwxr-xr-xmodules/profile/templates/monitoring/send_nrdp.sh.erb288
1 files changed, 38 insertions, 250 deletions
diff --git a/modules/profile/templates/monitoring/send_nrdp.sh.erb b/modules/profile/templates/monitoring/send_nrdp.sh.erb
index 41f58e5..0e38908 100755
--- a/modules/profile/templates/monitoring/send_nrdp.sh.erb
+++ b/modules/profile/templates/monitoring/send_nrdp.sh.erb
@@ -1,271 +1,59 @@
1#!/bin/bash 1#!/bin/bash
2#
3# check_nrdp.sh
4#
5# Copyright (c) 2010-2017 - Nagios Enterprises, LLC.
6# Written by: Scott Wilkerson (nagios@nagios.org)
7#
8# 2017-09-25 Troy Lea aka BOX293
9# - Fixed script not working with arguments when run as a cron job
10# or if being used as a nagios command like obsessive compulsive.
11# ... "if [ ! -t 0 ]" was the reason why.
12# 2017-12-08 Jørgen van der Meulen (Conclusion Xforce)
13# - Fixed typo in NRDP abbreviation
14 2
3TEMPLATE='{
4 "cmd": "submitcheck",
5 "token": $token,
6 "checkresult": [{
7 "hostname": $hostname,
8 "state": $state,
9 "output": $output,
10 "type": $type,
11 "servicename": $servicename
12 }]
13}'
15 14
16PROGNAME=$(basename $0)
17RELEASE="Revision 0.6.1"
18
19print_release() {
20 echo "$RELEASE"
21}
22
23print_usage() {
24 echo ""
25 echo "$PROGNAME $RELEASE - Send NRDP script for Nagios"
26 echo ""
27 echo "Usage: send_nrdp.sh -u URL -t token [options]"
28 echo ""
29 echo "Usage: $PROGNAME -h display help"
30 echo ""
31}
32
33print_help() {
34 print_usage
35 echo ""
36 echo "This script is used to send NRDP data to a Nagios server"
37 echo ""
38 echo "Required:"
39 echo " -u"," URL of NRDP server. Usually http://<IP_ADDRESS>/nrdp/"
40 echo " -t"," Shared token. Must be the same token set in NRDP Server"
41 echo ""
42 echo "Options:"
43 echo " Single Check:"
44 echo " -H host name"
45 echo " -s service name"
46 echo " -S State"
47 echo " -o output"
48 echo ""
49 echo " STDIN:"
50 echo " [-d delimiter] (default -d \"\\t\")"
51 echo " With only the required parameters $PROGNAME is capable of"
52 echo " processing data piped to it either from a file or other"
53 echo " process. By default, we use \t as the delimiter however this"
54 echo " may be specified with the -d option data should be in the"
55 echo " following formats one entry per line."
56 echo " For Host checks:"
57 echo " hostname State output"
58 echo " For Service checks"
59 echo " hostname servicename State output"
60 echo ""
61 echo " File:"
62 echo " -f /full/path/to/file"
63 echo " This file will be sent to the NRDP server specified in -u"
64 echo " The file should be an XML file in the following format"
65 echo " ##################################################"
66 echo ""
67 echo " <?xml version='1.0'?>"
68 echo " <checkresults>"
69 echo " <checkresult type=\"host\" checktype=\"1\">"
70 echo " <hostname>YOUR_HOSTNAME</hostname>"
71 echo " <state>0</state>"
72 echo " <output>OK|perfdata=1.00;5;10;0</output>"
73 echo " </checkresult>"
74 echo " <checkresult type=\"service\" checktype=\"1\">"
75 echo " <hostname>YOUR_HOSTNAME</hostname>"
76 echo " <servicename>YOUR_SERVICENAME</servicename>"
77 echo " <state>0</state>"
78 echo " <output>OK|perfdata=1.00;5;10;0</output>"
79 echo " </checkresult>"
80 echo " </checkresults>"
81 echo " ##################################################"
82 echo ""
83 echo " Directory:"
84 echo " -D /path/to/temp/dir"
85 echo " This is a directory that contains XML files in the format"
86 echo " above. Additionally, if the -d flag is specified, $PROGNAME"
87 echo " will create temp files here if the server could not be reached."
88 echo " On additional calls with the same -D path, if a connection to"
89 echo " the server is successful, all temp files will be sent."
90 exit 0
91}
92
93send_data() {
94 pdata="token=$token&cmd=submitcheck"
95 if [ $file ]; then
96 fdata="--data-urlencode XMLDATA@$file"
97 rslt=`curl -f --silent --insecure -d "$pdata" $fdata "$url/"`
98 else
99 pdata="$pdata&XMLDATA=$1"
100 rslt=`curl -f --silent --insecure -d "$pdata" "$url/"`
101 fi
102
103 ret=$?
104
105 status=`echo $rslt | sed -n 's|.*<status>\(.*\)</status>.*|\1|p'`
106 message=`echo $rslt | sed -n 's|.*<message>\(.*\)</message>.*|\1|p'`
107 if [ $ret != 0 ];then
108 echo "ERROR: could not connect to NRDP server at $url"
109 # verify we are not processing the directory already and then write to the directory
110 if [ ! "$2" ] && [ $directory ];then
111 if [ ! -d "$directory" ];then
112 mkdir -p "$directory"
113 fi
114 # This is where we write to the tmp directory
115 echo $xml > `mktemp $directory/nrdp.XXXXXX`
116 fi
117 exit 1
118 fi
119
120 if [ "$status" != "0" ];then
121 # This means we couldn't connect to NRPD server
122 echo "ERROR: The NRDP Server said $message"
123 # verify we are not processing the directory already and then write to the directory
124 if [ ! "$2" ] && [ $directory ];then
125 if [ ! -d "$directory" ];then
126 mkdir -p "$directory"
127 fi
128 # This is where we write to the tmp directory
129 echo $xml > `mktemp $directory/nrdp.XXXXXX`
130 fi
131
132 exit 2
133 fi
134
135 # If this was a directory call and was successful, remove the file
136 if [ $2 ] && [ "$status" == "0" ];then
137 rm -f "$2"
138 fi
139
140 # If we weren't successful error
141 if [ $ret != 0 ];then
142 echo "exited with error "$ret
143 exit $ret
144 fi
145}
146
147# Parse parameters
148url="<%= @naemon_url %>" 15url="<%= @naemon_url %>"
149token="<%= @naemon_token %>" 16token="<%= @naemon_token %>"
150 17while getopts "u:t:H:s:S:o:" option
151while getopts "u:t:H:s:S:o:f:d:c:D:hv" option
152do 18do
153 case $option in 19 case $option in
154 u) url=$OPTARG ;; 20 u) url=$OPTARG ;;
155 t) token=$OPTARG ;; 21 t) token=$OPTARG ;;
156 H) host=$OPTARG ;; 22 H) hostname=$OPTARG ;;
157 s) service=$OPTARG ;; 23 s) servicename=$OPTARG ;;
158 S) State=$OPTARG ;; 24 S) state=$OPTARG ;;
159 o) output=$OPTARG ;; 25 o) output=$OPTARG ;;
160 f) file=$OPTARG ;;
161 d) delim=$OPTARG ;;
162 c) checktype=$OPTARG ;;
163 D) directory=$OPTARG ;;
164 h) print_help 0;;
165 v) print_release
166 exit 0 ;;
167 esac 26 esac
168done 27done
169 28
170if [ ! $checktype ]; then 29if [ -n "$servicename" ]; then
171 checktype=1 30 checktype="service"
172fi 31else
173if [ ! $delim ]; then 32 checktype="host"
174 delim=`echo -e "\t"`
175fi 33fi
176 34
177if [ "x$url" == "x" -o "x$token" == "x" ] 35payload=$(jq -n \
178then 36 --arg type "$checktype" \
179 echo "Usage: send_nrdp -u url -t token" 37 --arg hostname "$hostname" \
180 exit 1 38 --arg servicename "$servicename" \
181fi 39 --arg output "$output" \
182# detecting curl 40 --arg token "$token" \
183if [[ `which curl` =~ "/curl" ]] 41 --arg state "$state" \
184 then curl=1; 42 "$TEMPLATE")
185fi
186
187if [[ ! $curl ]];
188then
189 echo "Either curl or wget are required to run $PROGNAME"
190 exit 1
191fi
192
193checkcount=0
194 43
195if [ $host ]; then 44rslt=$(curl -f --silent --insecure -d "$payload" -H "Content-Type: application/json" "$url")
196 xml="" 45ret=$?
197 # we are not getting piped results
198 if [ "$host" == "" ] || [ "$State" == "" ]; then
199 echo "You must provide a host -H and State -S"
200 exit 2
201 fi
202 if [ "$service" != "" ]; then
203 xml="$xml<checkresult type='service' checktype='$checktype'><servicename>$service</servicename>"
204 else
205 xml="$xml<checkresult type='host' checktype='$checktype'>"
206 fi
207
208 # urlencode XML special chars
209 output=${output//&/%26}
210 output=${output//</%3C}
211 output=${output//>/%3E}
212
213 xml="$xml<hostname>$host</hostname><state>$State</state><output><![CDATA["$output"]]></output></checkresult>"
214 checkcount=1
215fi
216
217 # If only url and token have been provided then it is assumed that data is being piped
218########################
219if [[ ! $host && ! $State && ! $file && ! $directory ]]; then
220 xml=""
221 # we know we are being piped results
222 IFS=$delim
223
224 while read -r line ; do
225 arr=($line)
226 if [ ${#arr[@]} != 0 ];then
227 if [[ ${#arr[@]} < 3 ]] || [[ ${#arr[@]} > 4 ]];then
228 echo "ERROR: STDIN must be either 3 or 4 fields long, I found "${#arr[@]}
229 else
230 if [ ${#arr[@]} == 4 ]; then
231 xml="$xml<checkresult type='service' checktype='$checktype'>
232 <servicename>${arr[1]}</servicename>
233 <hostname>${arr[0]}</hostname>
234 <state>${arr[2]}</state>
235 <output>${arr[3]}</output>"
236 else
237 xml="$xml<checkresult type='host' checktype='$checktype'>
238 <hostname>${arr[0]}</hostname>
239 <state>${arr[1]}</state>
240 <output>${arr[2]}</output>"
241 fi
242
243 xml="$xml</checkresult>"
244 checkcount=$[checkcount+1]
245 fi
246 fi
247 done
248 IFS=" "
249fi
250 46
251if [ $file ]; then 47if [ $ret != 0 ];then
252 xml=`cat $file` 48 echo "ERROR: could not connect to NRDP server at $url"
253 send_data "$xml" 49 exit 1
254fi 50fi
255 51
256if [ $directory ]; then 52status=$(echo "$rslt" | jq -r .status)
257 #echo "Processing directory..." 53message=$(echo "$rslt" | jq -r .message)
258 for f in `ls $directory`
259 do
260 #echo "Processing $f file..."
261 # take action on each file. $f store current file name
262 xml=`cat $directory/$f`
263 send_data "$xml" "$directory/$f"
264 done
265fi
266 54
267if [ "x$file" == "x" ] && [ "x$directory" == "x" ]; then 55if [ "$status" != "ok" ];then
268 xml="<?xml version='1.0'?><checkresults>$xml</checkresults>" 56 echo "ERROR: The NRDP Server said $message"
269 send_data "$xml" 57 exit 2
270 echo "Sent $checkcount checks to $url"
271fi 58fi
59echo "Sent 1 checks to $url"
generated by cgit v1.2.3 (git 2.25.1) at 2024-04-18 05:41:55 +0000