Code

Avoid core dump with null or invalid data (679400 - Mathieu Masseboeuf)
[nagiosplug.git] / plugins / check_snmp.c
1 /******************************************************************************
2  *
3  * Program: SNMP plugin for Nagios
4  * License: GPL
5  *
6  * License Information:
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21  *
22  *./plugins/check_snmp 127.0.0.1 -c public -o .1.3.6.1.4.1.2021.9.1.2.1
23  *
24  *****************************************************************************/
26 const char *progname = "check_snmp";
27 #define REVISION "$Revision$"
28 #define COPYRIGHT "1999-2002"
29 #define AUTHOR "Ethan Galstad"
30 #define EMAIL "nagios@nagios.org"
31 #define SUMMARY "Check status of remote machines using SNMP.\n"
33 #define OPTIONS "\
34 -H <ip_address> -o <OID> [-w warn_range] [-c crit_range] \n\
35           [-C community] [-s string] [-r regex] [-R regexi] [-t timeout]\n\
36           [-l label] [-u units] [-p port-number] [-d delimiter]\n\
37           [-D output-delimiter] [-m miblist] [-P snmp version]\n\
38           [-L seclevel] [-U secname] [-a authproto] [-A authpasswd]\n\
39           [-X privpasswd]\n"
41 #define LONGOPTIONS "\
42  -H, --hostname=HOST\n\
43     Name or IP address of the device you wish to query\n\
44  -o, --oid=OID(s)\n\
45     Object identifier(s) whose value you wish to query\n\
46  -w, --warning=INTEGER_RANGE(s)\n\
47     Range(s) which will not result in a WARNING status\n\
48  -c, --critical=INTEGER_RANGE(s)\n\
49     Range(s) which will not result in a CRITICAL status\n\
50  -C, --community=STRING\n\
51     Optional community string for SNMP communication\n\
52     (default is \"%s\")\n\
53  -u, --units=STRING\n\
54     Units label(s) for output data (e.g., 'sec.').\n\
55  -p, --port=STRING\n\
56     UDP port number target is listening on. Default is \"%s\"\n\
57  -P, --protocol=[1|3]\n\
58     SNMP protocol version\n\
59  -L, --seclevel=[noAuthNoPriv|authNoPriv|authPriv]\n\
60     SNMPv3 securityLevel\n\
61  -U, --secname=USERNAME\n\
62     SNMPv3 username\n\
63  -a, --authproto=[MD5|SHA]\n\
64     SNMPv3 auth proto\n\
65  -A, --authpassword=PASSWORD\n\
66     SNMPv3 authentication password\n\
67  -X, --privpasswd=PASSWORD\n\
68     SNMPv3 crypt passwd (DES)\n\
69  -d, --delimiter=STRING\n\
70     Delimiter to use when parsing returned data. Default is \"%s\"\n\
71     Any data on the right hand side of the delimiter is considered\n\
72     to be the data that should be used in the evaluation.\n\
73  -t, --timeout=INTEGER\n\
74     Seconds to wait before plugin times out (see also nagios server timeout).\n\
75     Default is %d seconds\n\
76  -D, --output-delimiter=STRING\n\
77     Separates output on multiple OID requests\n\
78  -s, --string=STRING\n\
79     Return OK state (for that OID) if STRING is an exact match\n\
80  -r, --ereg=REGEX\n\
81     Return OK state (for that OID) if extended regular expression REGEX matches\n\
82  -R, --eregi=REGEX\n\
83     Return OK state (for that OID) if case-insensitive extended REGEX matches\n\
84  -l, --label=STRING\n\
85     Prefix label for output from plugin (default -s 'SNMP')\n\
86  -v, --verbose\n\
87     Debugging the output\n\
88  -m, --miblist=STRING\n\
89     List of MIBS to be loaded (default = ALL)\n"
91 #define NOTES "\
92 - This plugin uses the 'snmpget' command included with the NET-SNMP package.\n\
93   If you don't have the package installed, you will need to download it from\n\
94   http://net-snmp.sourceforge.net before you can use this plugin.\n\
95 - Multiple OIDs may be indicated by a comma- or space-delimited list (lists with\n\
96   internal spaces must be quoted) [max 8 OIDs]\n\
97 - Ranges are inclusive and are indicated with colons. When specified as\n\
98   'min:max' a STATE_OK will be returned if the result is within the indicated\n\
99   range or is equal to the upper or lower bound. A non-OK state will be\n\
100   returned if the result is outside the specified range.\n\
101 - If specified in the order 'max:min' a non-OK state will be returned if the\n\
102   result is within the (inclusive) range.\n\
103 - Upper or lower bounds may be omitted to skip checking the respective limit.\n\
104 - Bare integers are interpreted as upper limits.\n\
105 - When checking multiple OIDs, separate ranges by commas like '-w 1:10,1:,:20'\n\
106 - Note that only one string and one regex may be checked at present\n\
107 - All evaluation methods other than PR, STR, and SUBSTR expect that the value\n\
108   returned from the SNMP query is an unsigned integer.\n"
110 #define DESCRIPTION "\
111 This plugin gets system information on a remote server via snmp.\n"
113 #define DEFAULT_COMMUNITY "public"
114 #define DEFAULT_PORT "161"
115 #define DEFAULT_TIMEOUT 10
116 #define DEFAULT_MIBLIST "ALL"
117 #define DEFAULT_PROTOCOL "1"
118 #define DEFAULT_AUTH_PROTOCOL "MD5"
120 #include "common.h"
121 #include "utils.h"
122 #include "popen.h"
124 #define mark(a) ((a)!=0?"*":"")
126 #define CHECK_UNDEF 0
127 #define CRIT_PRESENT 1
128 #define CRIT_STRING 2
129 #define CRIT_REGEX 4
130 #define CRIT_GT 8
131 #define CRIT_LT 16
132 #define CRIT_GE 32
133 #define CRIT_LE 64
134 #define CRIT_EQ 128
135 #define CRIT_NE 256
136 #define CRIT_RANGE 512
137 #define WARN_PRESENT 1024
138 #define WARN_STRING 2048
139 #define WARN_REGEX 4096
140 #define WARN_GT 8192
141 #define WARN_LT 16384
142 #define WARN_GE 32768
143 #define WARN_LE 65536
144 #define WARN_EQ 131072
145 #define WARN_NE 262144
146 #define WARN_RANGE 524288
148 #define MAX_OIDS 8
149 #define MAX_DELIM_LENGTH 8
150 #define DEFAULT_DELIMITER "="
151 #define DEFAULT_OUTPUT_DELIMITER " "
153 void print_usage (void);
154 void print_help (void);
155 int process_arguments (int, char **);
156 int validate_arguments (void);
157 int check_num (int);
158 char *clarify_message (char *);
159 int lu_getll (unsigned long *, char *);
160 int lu_getul (unsigned long *, char *);
161 char *thisarg (char *str);
162 char *nextarg (char *str);
164 #ifdef HAVE_REGEX_H
165 #include <regex.h>
166 char regex_expect[MAX_INPUT_BUFFER] = "";
167 regex_t preg;
168 regmatch_t pmatch[10];
169 char timestamp[10] = "";
170 char regex[MAX_INPUT_BUFFER];
171 char errbuf[MAX_INPUT_BUFFER];
172 int cflags = REG_EXTENDED | REG_NOSUB | REG_NEWLINE;
173 int eflags = 0;
174 int errcode, excode;
175 #endif
177 char *server_address = NULL;
178 char *community = DEFAULT_COMMUNITY;
179 char *authpriv = NULL;
180 char *proto = NULL;
181 char *seclevel = NULL;
182 char *secname = NULL;
183 char *authproto = NULL;
184 char *authpasswd = NULL;
185 char *privpasswd = NULL;
186 char *oid = "";
187 char *label = "SNMP";
188 char *units = "";
189 char *port = DEFAULT_PORT;
190 char string_value[MAX_INPUT_BUFFER] = "";
191 char **labels = NULL;
192 char **unitv = NULL;
193 int nlabels = 0;
194 int labels_size = 8;
195 int nunits = 0;
196 int unitv_size = 8;
197 int verbose = FALSE;
198 unsigned long lower_warn_lim[MAX_OIDS];
199 unsigned long upper_warn_lim[MAX_OIDS];
200 unsigned long lower_crit_lim[MAX_OIDS];
201 unsigned long upper_crit_lim[MAX_OIDS];
202 unsigned long response_value[MAX_OIDS];
203 int check_warning_value = FALSE;
204 int check_critical_value = FALSE;
205 int eval_method[MAX_OIDS];
206 char *delimiter = DEFAULT_DELIMITER;
207 char *output_delim = DEFAULT_OUTPUT_DELIMITER;
208 char *miblist = DEFAULT_MIBLIST;
211 int
212 main (int argc, char **argv)
214         int i = 0;
215         int iresult = STATE_UNKNOWN;
216         int found = 0;
217         int result = STATE_DEPENDENT;
218         char input_buffer[MAX_INPUT_BUFFER];
219         char *command_line = NULL;
220         char *response = NULL;
221         char *outbuff = "";
222         char *output = "";
223         char *ptr = NULL;
224         char *p2 = NULL;
225         char *show = NULL;
227         labels = malloc (labels_size);
228         unitv = malloc (unitv_size);
229         for (i = 0; i < MAX_OIDS; i++)
230                 eval_method[i] = CHECK_UNDEF;
231         i = 0;
233         if (process_arguments (argc, argv) == ERROR)
234                 usage ("Incorrect arguments supplied\n");
236         /* create the command line to execute */
237         asprintf (&command_line, "%s -m %s -v %s %s %s:%s %s",
238                   PATH_TO_SNMPGET, miblist, proto, authpriv, server_address, port, oid);
239         if (verbose)
240                 printf ("%s\n", command_line);
242         /* run the command */
243         child_process = spopen (command_line);
244         if (child_process == NULL) {
245                 printf ("Could not open pipe: %s\n", command_line);
246                 exit (STATE_UNKNOWN);
247         }
249         child_stderr = fdopen (child_stderr_array[fileno (child_process)], "r");
250         if (child_stderr == NULL) {
251                 printf ("Could not open stderr for %s\n", command_line);
252         }
254         while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process))
255                 asprintf (&output, "%s%s", output, input_buffer);
257         if (verbose)
258                 printf ("%s\n", output);
260         ptr = output;
262         while (ptr) {
264                 ptr = strstr (ptr, delimiter);
265                 if (ptr == NULL)
266                         break;
268                 ptr += strlen (delimiter);
269                 ptr += strspn (ptr, " ");
271                 found++;
273                 if (ptr[0] == '"') {
274                         ptr++;
275                         response = strpcpy (response, ptr, "\"");
276                         ptr = strpbrk (ptr, "\"");
277                         ptr += strspn (ptr, "\"\n");
278                 }
279                 else {
280                         response = strpcpy (response, ptr, "\n");
281                         ptr = strpbrk (ptr, "\n");
282                         ptr += strspn (ptr, "\n");
283                         while
284                                 (strstr (ptr, delimiter) &&
285                                  strstr (ptr, "\n") && strstr (ptr, "\n") < strstr (ptr, delimiter)) {
286                                 response = strpcat (response, ptr, "\n");
287                                 ptr = strpbrk (ptr, "\n");
288                         }
289                         if (ptr && strstr (ptr, delimiter) == NULL) {
290                                 response = strscat (response, ptr);
291                                 ptr = NULL;
292                         }
293                 }
295                 /* We strip out the datatype indicator for PHBs */
296                 if (strstr (response, "Gauge: "))
297                         show = strstr (response, "Gauge: ") + 7;
298                 else if (strstr (response, "Gauge32: "))
299                         show = strstr (response, "Gauge32: ") + 9;
300                 else if (strstr (response, "Counter32: "))
301                         show = strstr (response, "Counter32: ") + 11;
302                 else if (strstr (response, "INTEGER: "))
303                         show = strstr (response, "INTEGER: ") + 9;
304                 else
305                         show = response;
306                 p2 = show;
308                 iresult = STATE_DEPENDENT;
310                 /* Process this block for integer comparisons */
311                 if (eval_method[i] & CRIT_GT ||
312                     eval_method[i] & CRIT_LT ||
313                     eval_method[i] & CRIT_GE ||
314                     eval_method[i] & CRIT_LE ||
315                     eval_method[i] & CRIT_EQ ||
316                     eval_method[i] & CRIT_NE ||
317                     eval_method[i] & WARN_GT ||
318                     eval_method[i] & WARN_LT ||
319                     eval_method[i] & WARN_GE ||
320                     eval_method[i] & WARN_LE ||
321                     eval_method[i] & WARN_EQ ||
322                     eval_method[i] & WARN_NE) {
323                         p2 = strpbrk (p2, "0123456789");
324                         if (p2 == NULL) 
325                                 terminate (STATE_UNKNOWN,"No valid data returned");
326                         response_value[i] = strtoul (p2, NULL, 10);
327                         iresult = check_num (i);
328                         asprintf (&show, "%lu", response_value[i]);
329                 }
331                 /* Process this block for string matching */
332                 else if (eval_method[i] & CRIT_STRING) {
333                         if (strcmp (response, string_value))
334                                 iresult = STATE_CRITICAL;
335                         else
336                                 iresult = STATE_OK;
337                 }
339                 /* Process this block for regex matching */
340                 else if (eval_method[i] & CRIT_REGEX) {
341 #ifdef HAVE_REGEX_H
342                         excode = regexec (&preg, response, 10, pmatch, eflags);
343                         if (excode == 0) {
344                                 iresult = STATE_OK;
345                         }
346                         else if (excode != REG_NOMATCH) {
347                                 regerror (excode, &preg, errbuf, MAX_INPUT_BUFFER);
348                                 printf ("Execute Error: %s\n", errbuf);
349                                 exit (STATE_CRITICAL);
350                         }
351                         else {
352                                 iresult = STATE_CRITICAL;
353                         }
354 #else
355                         printf ("%s UNKNOWN: call for regex which was not a compiled option", label);
356                         exit (STATE_UNKNOWN);
357 #endif
358                 }
360                 /* Process this block for existence-nonexistence checks */
361                 else {
362                         if (eval_method[i] & CRIT_PRESENT)
363                                 iresult = STATE_CRITICAL;
364                         else if (eval_method[i] & WARN_PRESENT)
365                                 iresult = STATE_WARNING;
366                         else if (response && iresult == STATE_DEPENDENT) 
367                                 iresult = STATE_OK;
368                 }
370                 /* Result is the worst outcome of all the OIDs tested */
371                 result = max_state (result, iresult);
373                 /* Prepend a label for this OID if there is one */
374                 if (nlabels > 1 && i < nlabels && labels[i] != NULL)
375                         asprintf (&outbuff, "%s%s%s %s%s%s", outbuff,
376                                   (i == 0) ? " " : output_delim,
377                                   labels[i], mark (iresult), show, mark (iresult));
378                 else
379                         asprintf (&outbuff, "%s%s%s%s%s", outbuff, (i == 0) ? " " : output_delim,
380                                   mark (iresult), show, mark (iresult));
382                 /* Append a unit string for this OID if there is one */
383                 if (nunits > 0 && i < nunits && unitv[i] != NULL)
384                         asprintf (&outbuff, "%s %s", outbuff, unitv[i]);
386                 i++;
388         }                                                                                                                       /* end while (ptr) */
390         if (found == 0)
391                 terminate
392                         (STATE_UNKNOWN,
393                          "%s problem - No data recieved from host\nCMD: %s\n",
394                          label, command_line);
396         /* WARNING if output found on stderr */
397         if (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_stderr))
398                 result = max_state (result, STATE_WARNING);
400         /* close stderr */
401         (void) fclose (child_stderr);
403         /* close the pipe */
404         if (spclose (child_process))
405                 result = max_state (result, STATE_WARNING);
407 /*      if (nunits == 1 || i == 1) */
408 /*              printf ("%s %s -%s %s\n", label, state_text (result), outbuff, units); */
409 /*      else */
410         printf ("%s %s -%s\n", label, state_text (result), outbuff);
412         return result;
415 /* process command-line arguments */
416 int
417 process_arguments (int argc, char **argv)
419         char *ptr;
420         int c = 1;
421         int j = 0, jj = 0, ii = 0;
423 #ifdef HAVE_GETOPT_H
424         int option_index = 0;
425         static struct option long_options[] = {
426                 STD_LONG_OPTS,
427                 {"community", required_argument, 0, 'C'},
428                 {"oid", required_argument, 0, 'o'},
429                 {"object", required_argument, 0, 'o'},
430                 {"delimiter", required_argument, 0, 'd'},
431                 {"output-delimiter", required_argument, 0, 'D'},
432                 {"string", required_argument, 0, 's'},
433                 {"regex", required_argument, 0, 'r'},
434                 {"ereg", required_argument, 0, 'r'},
435                 {"eregi", required_argument, 0, 'R'},
436                 {"label", required_argument, 0, 'l'},
437                 {"units", required_argument, 0, 'u'},
438                 {"port", required_argument, 0, 'p'},
439                 {"miblist", required_argument, 0, 'm'},
440                 {"protocol", required_argument, 0, 'P'},
441                 {"seclevel", required_argument, 0, 'L'},
442                 {"secname", required_argument, 0, 'U'},
443                 {"authproto", required_argument, 0, 'a'},
444                 {"authpasswd", required_argument, 0, 'A'},
445                 {"privpasswd", required_argument, 0, 'X'},
446                 {0, 0, 0, 0}
447         };
448 #endif
450         if (argc < 2)
451                 return ERROR;
453         /* reverse compatibility for very old non-POSIX usage forms */
454         for (c = 1; c < argc; c++) {
455                 if (strcmp ("-to", argv[c]) == 0)
456                         strcpy (argv[c], "-t");
457                 if (strcmp ("-wv", argv[c]) == 0)
458                         strcpy (argv[c], "-w");
459                 if (strcmp ("-cv", argv[c]) == 0)
460                         strcpy (argv[c], "-c");
461         }
463         while (1) {
464 #ifdef HAVE_GETOPT_H
465                 c =
466                         getopt_long (argc, argv, "hvVt:c:w:H:C:o:e:E:d:D:s:R:r:l:u:p:m:P:L:U:a:A:X:",
467                                                                          long_options, &option_index);
468 #else
469                 c = getopt (argc, argv, "hvVt:c:w:H:C:o:e:E:d:D:s:R:r:l:u:p:m:P:L:U:a:A:X:");
470 #endif
472                 if (c == -1 || c == EOF)
473                         break;
475                 switch (c) {
476                 case '?':       /* usage */
477                         usage3 ("Unknown argument", optopt);
478                 case 'h':       /* help */
479                         print_help ();
480                         exit (STATE_OK); 
481                 case 'V':       /* version */
482                         print_revision (progname, REVISION);
483                         exit (STATE_OK);
484                 case 'v': /* verbose */
485                         verbose = TRUE;
486                         break;
488         /* Connection info */
489                 case 'C':                                                                       /* group or community */
490                         community = strscpy (community, optarg);
491                         break;
492                 case 'H':                                                                       /* Host or server */
493                         server_address = strscpy (server_address, optarg);
494                         break;
495                 case 'p':       /* TCP port number */
496                         port = strscpy(port, optarg);
497                         break;
498                 case 'm':      /* List of MIBS  */
499                         miblist = strscpy(miblist, optarg);
500                         break;
501                 case 'P':     /* SNMP protocol version */
502                         proto = strscpy(proto, optarg);
503                         break;
504                 case 'L':     /* security level */
505                         seclevel = strscpy(seclevel,optarg);
506                         break;
507                 case 'U':     /* security username */
508                         secname = strscpy(secname, optarg);
509                         break;
510                 case 'a':     /* auth protocol */
511                         asprintf (&authproto, optarg);
512                         break;
513                 case 'A':     /* auth passwd */
514                         authpasswd = strscpy(authpasswd, optarg);
515                         break;
516                 case 'X':     /* priv passwd */
517                         privpasswd = strscpy(privpasswd, optarg);
518                         break;
519                 case 't':       /* timeout period */
520                         if (!is_integer (optarg))
521                                 usage2 ("Timeout Interval must be an integer", optarg);
522                         timeout_interval = atoi (optarg);
523                         break;
525         /* Test parameters */
526                 case 'c':                                                                       /* critical time threshold */
527                         if (strspn (optarg, "0123456789:,") < strlen (optarg)) {
528                                 printf ("Invalid critical threshold: %s\n", optarg);
529                                 print_usage ();
530                                 exit (STATE_UNKNOWN);
531                         }
532                         for (ptr = optarg; ptr && jj < MAX_OIDS; jj++) {
533                                 if (lu_getll (&lower_crit_lim[jj], ptr) == 1)
534                                         eval_method[jj] |= CRIT_LT;
535                                 if (lu_getul (&upper_crit_lim[jj], ptr) == 1)
536                                         eval_method[jj] |= CRIT_GT;
537                                 (ptr = index (ptr, ',')) ? ptr++ : ptr;
538                         }
539                         break;
540                 case 'w':                                                                       /* warning time threshold */
541                         if (strspn (optarg, "0123456789:,") < strlen (optarg)) {
542                                 printf ("Invalid warning threshold: %s\n", optarg);
543                                 print_usage ();
544                                 exit (STATE_UNKNOWN);
545                         }
546                         for (ptr = optarg; ptr && ii < MAX_OIDS; ii++) {
547                                 if (lu_getll (&lower_warn_lim[ii], ptr) == 1)
548                                         eval_method[ii] |= WARN_LT;
549                                 if (lu_getul (&upper_warn_lim[ii], ptr) == 1)
550                                         eval_method[ii] |= WARN_GT;
551                                 (ptr = index (ptr, ',')) ? ptr++ : ptr;
552                         }
553                         break;
554                 case 'o':                                                                       /* object identifier */
555                 case 'e': /* PRELIMINARY - may change */
556                 case 'E': /* PRELIMINARY - may change */
557                         for (ptr = optarg; (ptr = index (ptr, ',')); ptr++)
558                                 ptr[0] = ' '; /* relpace comma with space */
559                         for (ptr = optarg; (ptr = index (ptr, ' ')); ptr++)
560                                 j++; /* count OIDs */
561                         asprintf (&oid, "%s %s", (oid?oid:""), optarg);
562                         if (c == 'E' || c == 'e') {
563                                 jj++;
564                                 ii++;
565                         }
566                         if (c == 'E') 
567                                 eval_method[j+1] |= WARN_PRESENT;
568                         else if (c == 'e')
569                                 eval_method[j+1] |= CRIT_PRESENT;
570                         break;
571                 case 's':                                                                       /* string or substring */
572                         strncpy (string_value, optarg, sizeof (string_value) - 1);
573                         string_value[sizeof (string_value) - 1] = 0;
574                         eval_method[jj++] = CRIT_STRING;
575                         ii++;
576                         break;
577                 case 'R':                                                                       /* regex */
578 #ifdef HAVE_REGEX_H
579                         cflags = REG_ICASE;
580 #endif
581                 case 'r':                                                                       /* regex */
582 #ifdef HAVE_REGEX_H
583                         cflags |= REG_EXTENDED | REG_NOSUB | REG_NEWLINE;
584                         strncpy (regex_expect, optarg, sizeof (regex_expect) - 1);
585                         regex_expect[sizeof (regex_expect) - 1] = 0;
586                         errcode = regcomp (&preg, regex_expect, cflags);
587                         if (errcode != 0) {
588                                 regerror (errcode, &preg, errbuf, MAX_INPUT_BUFFER);
589                                 printf ("Could Not Compile Regular Expression");
590                                 return ERROR;
591                         }
592                         eval_method[jj++] = CRIT_REGEX;
593                         ii++;
594 #else
595                         printf ("%s UNKNOWN: call for regex which was not a compiled option", label);
596                         exit (STATE_UNKNOWN);
597 #endif
598                         break;
600         /* Format */
601                 case 'd':                                                                       /* delimiter */
602                         delimiter = strscpy (delimiter, optarg);
603                         break;
604                 case 'D':                                                                       /* output-delimiter */
605                         output_delim = strscpy (output_delim, optarg);
606                         break;
607                 case 'l':                                                                       /* label */
608                         label = optarg;
609                         nlabels++;
610                         if (nlabels >= labels_size) {
611                                 labels_size += 8;
612                                 labels = realloc (labels, labels_size);
613                                 if (labels == NULL)
614                                         terminate (STATE_UNKNOWN,
615                                                                                  "Could not realloc() labels[%d]", nlabels);
616                         }
617                         labels[nlabels - 1] = optarg;
618                         ptr = thisarg (optarg);
619                         if (strstr (ptr, "'") == ptr)
620                                 labels[nlabels - 1] = ptr + 1;
621                         else
622                                 labels[nlabels - 1] = ptr;
623                         while (ptr && (ptr = nextarg (ptr))) {
624                                 if (nlabels >= labels_size) {
625                                         labels_size += 8;
626                                         labels = realloc (labels, labels_size);
627                                         if (labels == NULL)
628                                                 terminate (STATE_UNKNOWN, "Could not realloc() labels\n");
629                                 }
630                                 labels++;
631                                 ptr = thisarg (ptr);
632                                 if (strstr (ptr, "'") == ptr)
633                                         labels[nlabels - 1] = ptr + 1;
634                                 else
635                                         labels[nlabels - 1] = ptr;
636                         }
637                         break;
638                 case 'u':                                                                       /* units */
639                         units = optarg;
640                         nunits++;
641                         if (nunits >= unitv_size) {
642                                 unitv_size += 8;
643                                 unitv = realloc (unitv, unitv_size);
644                                 if (unitv == NULL)
645                                         terminate (STATE_UNKNOWN,
646                                                                                  "Could not realloc() units [%d]\n", nunits);
647                         }
648                         unitv[nunits - 1] = optarg;
649                         ptr = thisarg (optarg);
650                         if (strstr (ptr, "'") == ptr)
651                                 unitv[nunits - 1] = ptr + 1;
652                         else
653                                 unitv[nunits - 1] = ptr;
654                         while (ptr && (ptr = nextarg (ptr))) {
655                                 if (nunits >= unitv_size) {
656                                         unitv_size += 8;
657                                         unitv = realloc (unitv, unitv_size);
658                                         if (units == NULL)
659                                                 terminate (STATE_UNKNOWN, "Could not realloc() units\n");
660                                 }
661                                 nunits++;
662                                 ptr = thisarg (ptr);
663                                 if (strstr (ptr, "'") == ptr)
664                                         unitv[nunits - 1] = ptr + 1;
665                                 else
666                                         unitv[nunits - 1] = ptr;
667                         }
668                         break;
670                 }
671         }
673         if (server_address == NULL)
674                 asprintf (&server_address, argv[optind]);
676         return validate_arguments ();
679 /******************************************************************************
681 @@-
682 <sect3>
683 <title>validate_arguments</title>
685 <para>&PROTO_validate_arguments;</para>
687 <para>Given a database name, this function returns TRUE if the string
688 is a valid PostgreSQL database name, and returns false if it is
689 not.</para>
691 <para>Valid PostgreSQL database names are less than &NAMEDATALEN;
692 characters long and consist of letters, numbers, and underscores. The
693 first character cannot be a number, however.</para>
695 </sect3>
696 -@@
697 ******************************************************************************/
699 int
700 validate_arguments ()
703         /* Need better checks to verify seclevel and authproto choices */
704         
705         if (seclevel == NULL) 
706                 asprintf (&seclevel, "noAuthNoPriv");
709         if (authproto == NULL ) 
710                 asprintf(&authproto, DEFAULT_AUTH_PROTOCOL);
711         
712          
713         
714         if (proto == NULL || (strcmp(proto,DEFAULT_PROTOCOL) == 0) ) {        /* default protocol version */
715                 asprintf(&proto, DEFAULT_PROTOCOL);
716                 asprintf(&authpriv, "%s%s", "-c ", community);
717         }
718         else if ( strcmp (proto, "3") == 0 ) {                 /* snmpv3 args */
719                 asprintf(&proto, "%s", "3");
720                 
721                 if ( (strcmp(seclevel, "noAuthNoPriv") == 0) || seclevel == NULL ) {
722                         asprintf(&authpriv, "%s", "-l noAuthNoPriv" );
723                 }
724                 else if ( strcmp(seclevel, "authNoPriv") == 0 ) {
725                         if ( secname == NULL || authpasswd == NULL) {
726                                 printf ("Missing secname (%s) or authpassword (%s) ! \n",secname, authpasswd );
727                                 print_usage ();
728                                 exit (STATE_UNKNOWN);
729                         }
730                         asprintf(&authpriv, "-l authNoPriv -a %s -u %s -A %s ", authproto, secname, authpasswd);
731                 }
732                 else if ( strcmp(seclevel, "authPriv") == 0 ) {
733                         if ( secname == NULL || authpasswd == NULL || privpasswd == NULL ) {
734                                 printf ("Missing secname (%s), authpassword (%s), or privpasswd (%s)! \n",secname, authpasswd,privpasswd );
735                                 print_usage ();
736                                 exit (STATE_UNKNOWN);
737                         }
738                         asprintf(&authpriv, "-l authPriv -a %s -u %s -A %s -x DES -X %s ", authproto, secname, authpasswd, privpasswd);
739                 }
740                 
741                                                                         
742         }
743         else {
744                 printf ("Invalid SNMP version: %s\n", proto);
745                 print_usage ();
746                 exit (STATE_UNKNOWN);                           
747         }
748                         
749         
750         
752         return OK;
757 void
758 print_help (void)
760         print_revision (progname, REVISION);
761         printf
762                 ("Copyright (c) %s %s <%s>\n\n%s\n", COPYRIGHT, AUTHOR, EMAIL, SUMMARY);
763         print_usage ();
764         printf
765                 ("\nOptions:\n" LONGOPTIONS "\n" DESCRIPTION "\n" NOTES "\n", 
766                  DEFAULT_COMMUNITY, DEFAULT_PORT, DEFAULT_DELIMITER, DEFAULT_TIMEOUT);
767         support ();
770 void
771 print_usage (void)
773         printf
774                 ("Usage:\n" " %s %s\n"
775                  " %s (-h | --help) for detailed help\n"
776                  " %s (-V | --version) for version information\n",
777                  progname, OPTIONS, progname, progname);
779 \f
782 char *
783 clarify_message (char *msg)
785         int i = 0;
786         int foo;
787         char tmpmsg_c[MAX_INPUT_BUFFER];
788         char *tmpmsg = (char *) &tmpmsg_c;
789         tmpmsg = strcpy (tmpmsg, msg);
790         if (!strncmp (tmpmsg, " Hex:", 5)) {
791                 tmpmsg = strtok (tmpmsg, ":");
792                 while ((tmpmsg = strtok (NULL, " "))) {
793                         foo = strtol (tmpmsg, NULL, 16);
794                         /* Translate chars that are not the same value in the printers
795                          * character set.
796                          */
797                         switch (foo) {
798                         case 208:
799                                 {
800                                         foo = 197;
801                                         break;
802                                 }
803                         case 216:
804                                 {
805                                         foo = 196;
806                                         break;
807                                 }
808                         }
809                         msg[i] = foo;
810                         i++;
811                 }
812                 msg[i] = 0;
813         }
814         return (msg);
818 int
819 check_num (int i)
821         int result;
822         result = STATE_OK;
823         if (eval_method[i] & WARN_GT && eval_method[i] & WARN_LT &&
824                         lower_warn_lim[i] > upper_warn_lim[i]) {
825                 if (response_value[i] <= lower_warn_lim[i] &&
826                                 response_value[i] >= upper_warn_lim[i]) {
827                         result = STATE_WARNING;
828                 }
829         }
830         else if
831                 ((eval_method[i] & WARN_GT && response_value[i] > upper_warn_lim[i]) ||
832                  (eval_method[i] & WARN_GE && response_value[i] >= upper_warn_lim[i]) ||
833                  (eval_method[i] & WARN_LT && response_value[i] < lower_warn_lim[i]) ||
834                  (eval_method[i] & WARN_LE && response_value[i] <= lower_warn_lim[i]) ||
835                  (eval_method[i] & WARN_EQ && response_value[i] == upper_warn_lim[i]) ||
836                  (eval_method[i] & WARN_NE && response_value[i] != upper_warn_lim[i])) {
837                 result = STATE_WARNING;
838         }
840         if (eval_method[i] & CRIT_GT && eval_method[i] & CRIT_LT &&
841                         lower_warn_lim[i] > upper_warn_lim[i]) {
842                 if (response_value[i] <= lower_crit_lim[i] &&
843                                 response_value[i] >= upper_crit_lim[i]) {
844                         result = STATE_CRITICAL;
845                 }
846         }
847         else if
848                 ((eval_method[i] & CRIT_GT && response_value[i] > upper_crit_lim[i]) ||
849                  (eval_method[i] & CRIT_GE && response_value[i] >= upper_crit_lim[i]) ||
850                  (eval_method[i] & CRIT_LT && response_value[i] < lower_crit_lim[i]) ||
851                  (eval_method[i] & CRIT_LE && response_value[i] <= lower_crit_lim[i]) ||
852                  (eval_method[i] & CRIT_EQ && response_value[i] == upper_crit_lim[i]) ||
853                  (eval_method[i] & CRIT_NE && response_value[i] != upper_crit_lim[i])) {
854                 result = STATE_CRITICAL;
855         }
857         return result;
861 int
862 lu_getll (unsigned long *ll, char *str)
864         char tmp[100];
865         if (strchr (str, ':') == NULL)
866                 return 0;
867         if (strchr (str, ',') != NULL && (strchr (str, ',') < strchr (str, ':')))
868                 return 0;
869         if (sscanf (str, "%lu%[:]", ll, tmp) == 2)
870                 return 1;
871         return 0;
874 int
875 lu_getul (unsigned long *ul, char *str)
877         char tmp[100];
878         if (sscanf (str, "%lu%[^,]", ul, tmp) == 1)
879                 return 1;
880         if (sscanf (str, ":%lu%[^,]", ul, tmp) == 1)
881                 return 1;
882         if (sscanf (str, "%*u:%lu%[^,]", ul, tmp) == 1)
883                 return 1;
884         return 0;
892 /* trim leading whitespace
893          if there is a leading quote, make sure it balances */
895 char *
896 thisarg (char *str)
898         str += strspn (str, " \t\r\n"); /* trim any leading whitespace */
899         if (strstr (str, "'") == str) { /* handle SIMPLE quoted strings */
900                 if (strlen (str) == 1 || !strstr (str + 1, "'"))
901                         terminate (STATE_UNKNOWN, "Unbalanced quotes\n");
902         }
903         return str;
907 /* if there's a leading quote, advance to the trailing quote
908          set the trailing quote to '\x0'
909          if the string continues, advance beyond the comma */
911 char *
912 nextarg (char *str)
914         if (strstr (str, "'") == str) {
915                 if (strlen (str) > 1) {
916                         str = strstr (str + 1, "'");
917                         str[0] = 0;
918                         return (++str);
919                 }
920                 else {
921                         str[0] = 0;
922                         return NULL;
923                 }
924         }
925         if (strstr (str, ",") == str) {
926                 if (strlen (str) > 1) {
927                         str[0] = 0;
928                         return (++str);
929                 }
930                 else {
931                         str[0] = 0;
932                         return NULL;
933                 }
934         }
935         if ((str = strstr (str, ",")) && strlen (str) > 1) {
936                 str[0] = 0;
937                 return (++str);
938         }
939         return NULL;