d00ce9c345fc94449e0f3173f795034f34edb24f
1 /******************************************************************************
2 *
3 * CHECK_UDP.C
4 *
5 * Program: UDP port plugin for Nagios
6 * License: GPL
7 * Copyright (c) 1999 Ethan Galstad (nagios@nagios.org)
8 *
9 * Last Modified: $Date$
10 *
11 * Command line: CHECK_UDP <host_address> [-p port] [-s send] [-e expect] \
12 * [-wt warn_time] [-ct crit_time] [-to to_sec]
13 *
14 * Description:
15 *
16 * This plugin will attempt to connect to the specified port
17 * on the host. Successul connects return STATE_OK, refusals
18 * and timeouts return STATE_CRITICAL, other errors return
19 * STATE_UNKNOWN.
20 *
21 * License Information:
22 *
23 * This program is free software; you can redistribute it and/or modify
24 * it under the terms of the GNU General Public License as published by
25 * the Free Software Foundation; either version 2 of the License, or
26 * (at your option) any later version.
27 *
28 * This program is distributed in the hope that it will be useful,
29 * but WITHOUT ANY WARRANTY; without even the implied warranty of
30 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
31 * GNU General Public License for more details.
32 *
33 * You should have received a copy of the GNU General Public License
34 * along with this program; if not, write to the Free Software
35 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
36 *
37 *****************************************************************************/
39 #include "config.h"
40 #include "common.h"
41 #include "netutils.h"
42 #include "utils.h"
44 #define PROGNAME "check_udp"
46 int warning_time = 0;
47 int check_warning_time = FALSE;
48 int critical_time = 0;
49 int check_critical_time = FALSE;
51 int process_arguments (int, char **);
52 int call_getopt (int, char **);
53 void print_usage (void);
54 void print_help (void);
56 int verbose = FALSE;
57 int server_port = 0;
58 char *server_address = NULL;
59 char *server_expect = NULL;
60 char *server_send = NULL;
62 int
63 main (int argc, char **argv)
64 {
65 int result;
66 char recv_buffer[MAX_INPUT_BUFFER];
68 if (process_arguments (argc, argv) == ERROR)
69 usage ("\n");
71 /* initialize alarm signal handling */
72 signal (SIGALRM, socket_timeout_alarm_handler);
74 /* set socket timeout */
75 alarm (socket_timeout);
77 time (&start_time);
78 result =
79 process_udp_request (server_address, server_port, server_send,
80 recv_buffer, MAX_INPUT_BUFFER - 1);
81 time (&end_time);
83 if (result != STATE_OK) {
84 printf ("No response from host on port %d\n", server_port);
85 result = STATE_CRITICAL;
86 }
88 else {
90 /* check to see if we got the response we wanted */
91 if (server_expect) {
92 if (!strstr (recv_buffer, server_expect)) {
93 printf ("Invalid response received from host on port %d\n",
94 server_port);
95 result = STATE_CRITICAL;
96 }
97 }
98 }
100 /* we connected, so close connection before exiting */
101 if (result == STATE_OK) {
103 if (check_critical_time == TRUE
104 && (end_time - start_time) > critical_time) result = STATE_CRITICAL;
105 else if (check_warning_time == TRUE
106 && (end_time - start_time) > warning_time) result =
107 STATE_WARNING;
109 printf ("Connection %s on port %d - %d second response time\n",
110 (result == STATE_OK) ? "accepted" : "problem", server_port,
111 (int) (end_time - start_time));
112 }
114 /* reset the alarm */
115 alarm (0);
117 return result;
118 }
123 /* process command-line arguments */
124 int
125 process_arguments (int argc, char **argv)
126 {
127 int c;
129 if (argc < 2)
130 usage ("\n");
132 for (c = 1; c < argc; c++) {
133 if (strcmp ("-to", argv[c]) == 0)
134 strcpy (argv[c], "-t");
135 else if (strcmp ("-wt", argv[c]) == 0)
136 strcpy (argv[c], "-w");
137 else if (strcmp ("-ct", argv[c]) == 0)
138 strcpy (argv[c], "-c");
139 }
141 c = 0;
142 while ((c += call_getopt (argc - c, &argv[c])) < argc) {
144 if (is_option (argv[c]))
145 continue;
147 if (server_address == NULL) {
148 if (argc > c) {
149 if (is_host (argv[c]) == FALSE)
150 usage ("Invalid host name/address\n");
151 server_address = argv[c];
152 }
153 else {
154 usage ("Host name was not supplied\n");
155 }
156 }
157 }
159 if (server_send == NULL)
160 server_send = strscpy (server_send, "");
162 return OK;
163 }
169 int
170 call_getopt (int argc, char **argv)
171 {
172 int c, i = 0;
174 #ifdef HAVE_GETOPT_H
175 int option_index = 0;
176 static struct option long_options[] = {
177 {"hostname", required_argument, 0, 'H'},
178 {"critical", required_argument, 0, 'c'},
179 {"warning", required_argument, 0, 'w'},
180 {"timeout", required_argument, 0, 't'},
181 {"port", required_argument, 0, 'p'},
182 {"expect", required_argument, 0, 'e'},
183 {"send", required_argument, 0, 's'},
184 {"verbose", no_argument, 0, 'v'},
185 {"version", no_argument, 0, 'V'},
186 {"help", no_argument, 0, 'h'},
187 {0, 0, 0, 0}
188 };
189 #endif
191 while (1) {
192 #ifdef HAVE_GETOPT_H
193 c =
194 getopt_long (argc, argv, "+hVvH:e:s:c:w:t:p:", long_options,
195 &option_index);
196 #else
197 c = getopt (argc, argv, "+hVvH:e:s:c:w:t:p:");
198 #endif
200 i++;
202 if (c == -1 || c == EOF || c == 1)
203 break;
205 switch (c) {
206 case 'H':
207 case 'c':
208 case 'w':
209 case 't':
210 case 'p':
211 case 'e':
212 case 's':
213 i++;
214 }
216 switch (c) {
217 case '?': /* print short usage statement if args not parsable */
218 printf ("%s: Unknown argument: %s\n\n", my_basename (argv[0]), optarg);
219 print_usage ();
220 exit (STATE_UNKNOWN);
221 case 'h': /* help */
222 print_help ();
223 exit (STATE_OK);
224 case 'V': /* version */
225 print_revision (my_basename (argv[0]), "$Revision$");
226 exit (STATE_OK);
227 case 'v': /* verbose mode */
228 verbose = TRUE;
229 break;
230 case 'H': /* hostname */
231 if (is_host (optarg) == FALSE)
232 usage ("Invalid host name/address\n");
233 server_address = optarg;
234 break;
235 case 'c': /* critical */
236 if (!is_intnonneg (optarg))
237 usage ("Critical threshold must be a nonnegative integer\n");
238 critical_time = atoi (optarg);
239 check_critical_time = TRUE;
240 break;
241 case 'w': /* warning */
242 if (!is_intnonneg (optarg))
243 usage ("Warning threshold must be a nonnegative integer\n");
244 warning_time = atoi (optarg);
245 check_warning_time = TRUE;
246 break;
247 case 't': /* timeout */
248 if (!is_intnonneg (optarg))
249 usage ("Timeout interval must be a nonnegative integer\n");
250 socket_timeout = atoi (optarg);
251 break;
252 case 'p': /* port */
253 if (!is_intnonneg (optarg))
254 usage ("Serevr port must be a nonnegative integer\n");
255 server_port = atoi (optarg);
256 break;
257 case 'e': /* expect */
258 server_expect = optarg;
259 break;
260 case 's': /* send */
261 server_send = optarg;
262 break;
263 }
264 }
265 return i;
266 }
272 void
273 print_usage (void)
274 {
275 printf
276 ("Usage: %s -H <host_address> [-p port] [-w warn_time] [-c crit_time]\n"
277 " [-e expect] [-s send] [-t to_sec] [-v]\n", PROGNAME);
278 }
284 void
285 print_help (void)
286 {
287 print_revision (PROGNAME, "$Revision$");
288 printf
289 ("Copyright (c) 1999 Ethan Galstad (nagios@nagios.org)\n\n"
290 "This plugin tests an UDP connection with the specified host.\n\n");
291 print_usage ();
292 printf
293 ("Options:\n"
294 " -H, --hostname=ADDRESS\n"
295 " Host name argument for servers using host headers (use numeric\n"
296 " address if possible to bypass DNS lookup).\n"
297 " -p, --port=INTEGER\n"
298 " Port number\n"
299 " -e, --expect=STRING <optional>\n"
300 " String to expect in first line of server response\n"
301 " -s, --send=STRING <optional>\n"
302 " String to send to the server when initiating the connection\n"
303 " -w, --warning=INTEGER <optional>\n"
304 " Response time to result in warning status (seconds)\n"
305 " -c, --critical=INTEGER <optional>\n"
306 " Response time to result in critical status (seconds)\n"
307 " -t, --timeout=INTEGER <optional>\n"
308 " Seconds before connection times out (default: %d)\n"
309 " -v, --verbose <optional>\n"
310 " Show details for command-line debugging (do not use with nagios server)\n"
311 " -h, --help\n"
312 " Print detailed help screen and exit\n"
313 " -V, --version\n"
314 " Print version information and exit\n", DEFAULT_SOCKET_TIMEOUT);
315 }