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  *****************************************************************************/
  25
  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"
  32
  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"
  40
  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"
  90
  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"
 109
 110 #define DESCRIPTION "\
 111 This plugin gets system information on a remote server via snmp.\n"
 112
 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"
 119
 120 #include "common.h"
 121 #include "utils.h"
 122 #include "popen.h"
 123
 124 #define mark(a) ((a)!=0?"*":"")
 125
 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
 147
 148 #define MAX_OIDS 8
 149 #define MAX_DELIM_LENGTH 8
 150 #define DEFAULT_DELIMITER "="
 151 #define DEFAULT_OUTPUT_DELIMITER " "
 152
 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);
 163
 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
 176
 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;
 209
 210
 211 int
 212 main (int argc, char **argv)
 213 {
 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;
 226
 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;
 232
 233         if (process_arguments (argc, argv) == ERROR)
 234                 usage ("Incorrect arguments supplied\n");
 235
 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);
 241
 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         }
 248
 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         }
 253
 254         while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process))
 255                 asprintf (&output, "%s%s", output, input_buffer);
 256
 257         if (verbose)
 258                 printf ("%s\n", output);
 259
 260         ptr = output;
 261
 262         while (ptr) {
 263
 264                 ptr = strstr (ptr, delimiter);
 265                 if (ptr == NULL)
 266                         break;
 267
 268                 ptr += strlen (delimiter);
 269                 ptr += strspn (ptr, " ");
 270
 271                 found++;
 272
 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                 }
 294
 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;
 307
 308                 iresult = STATE_DEPENDENT;
 309
 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                         response_value[i] = strtoul (p2, NULL, 10);
 325                         iresult = check_num (i);
 326                         asprintf (&show, "%lu", response_value[i]);
 327                 }
 328
 329                 /* Process this block for string matching */
 330                 else if (eval_method[i] & CRIT_STRING) {
 331                         if (strcmp (response, string_value))
 332                                 iresult = STATE_CRITICAL;
 333                         else
 334                                 iresult = STATE_OK;
 335                 }
 336
 337                 /* Process this block for regex matching */
 338                 else if (eval_method[i] & CRIT_REGEX) {
 339 #ifdef HAVE_REGEX_H
 340                         excode = regexec (&preg, response, 10, pmatch, eflags);
 341                         if (excode == 0) {
 342                                 iresult = STATE_OK;
 343                         }
 344                         else if (excode != REG_NOMATCH) {
 345                                 regerror (excode, &preg, errbuf, MAX_INPUT_BUFFER);
 346                                 printf ("Execute Error: %s\n", errbuf);
 347                                 exit (STATE_CRITICAL);
 348                         }
 349                         else {
 350                                 iresult = STATE_CRITICAL;
 351                         }
 352 #else
 353                         printf ("%s UNKNOWN: call for regex which was not a compiled option", label);
 354                         exit (STATE_UNKNOWN);
 355 #endif
 356                 }
 357
 358                 /* Process this block for existence-nonexistence checks */
 359                 else {
 360                         if (eval_method[i] & CRIT_PRESENT)
 361                                 iresult = STATE_CRITICAL;
 362                         else if (eval_method[i] & WARN_PRESENT)
 363                                 iresult = STATE_WARNING;
 364                         else if (response && iresult == STATE_DEPENDENT)
 365                                 iresult = STATE_OK;
 366                 }
 367
 368                 /* Result is the worst outcome of all the OIDs tested */
 369                 result = max_state (result, iresult);
 370
 371                 /* Prepend a label for this OID if there is one */
 372                 if (nlabels > 1 && i < nlabels && labels[i] != NULL)
 373                         asprintf (&outbuff, "%s%s%s %s%s%s", outbuff,
 374                                   (i == 0) ? " " : output_delim,
 375                                   labels[i], mark (iresult), show, mark (iresult));
 376                 else
 377                         asprintf (&outbuff, "%s%s%s%s%s", outbuff, (i == 0) ? " " : output_delim,
 378                                   mark (iresult), show, mark (iresult));
 379
 380                 /* Append a unit string for this OID if there is one */
 381                 if (nunits > 0 && i < nunits && unitv[i] != NULL)
 382                         asprintf (&outbuff, "%s %s", outbuff, unitv[i]);
 383
 384                 i++;
 385
 386         }                                                                                                                       /* end while (ptr) */
 387
 388         if (found == 0)
 389                 terminate
 390                         (STATE_UNKNOWN,
 391                          "%s problem - No data recieved from host\nCMD: %s\n",
 392                          label, command_line);
 393
 394         /* WARNING if output found on stderr */
 395         if (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_stderr))
 396                 result = max_state (result, STATE_WARNING);
 397
 398         /* close stderr */
 399         (void) fclose (child_stderr);
 400
 401         /* close the pipe */
 402         if (spclose (child_process))
 403                 result = max_state (result, STATE_WARNING);
 404
 405 /*      if (nunits == 1 || i == 1) */
 406 /*              printf ("%s %s -%s %s\n", label, state_text (result), outbuff, units); */
 407 /*      else */
 408         printf ("%s %s -%s\n", label, state_text (result), outbuff);
 409
 410         return result;
 411 }
 412
 413 /* process command-line arguments */
 414 int
 415 process_arguments (int argc, char **argv)
 416 {
 417         char *ptr;
 418         int c = 1;
 419         int j = 0, jj = 0, ii = 0;
 420
 421 #ifdef HAVE_GETOPT_H
 422         int option_index = 0;
 423         static struct option long_options[] = {
 424                 STD_LONG_OPTS,
 425                 {"community", required_argument, 0, 'C'},
 426                 {"oid", required_argument, 0, 'o'},
 427                 {"object", required_argument, 0, 'o'},
 428                 {"delimiter", required_argument, 0, 'd'},
 429                 {"output-delimiter", required_argument, 0, 'D'},
 430                 {"string", required_argument, 0, 's'},
 431                 {"regex", required_argument, 0, 'r'},
 432                 {"ereg", required_argument, 0, 'r'},
 433                 {"eregi", required_argument, 0, 'R'},
 434                 {"label", required_argument, 0, 'l'},
 435                 {"units", required_argument, 0, 'u'},
 436                 {"port", required_argument, 0, 'p'},
 437                 {"miblist", required_argument, 0, 'm'},
 438                 {"protocol", required_argument, 0, 'P'},
 439                 {"seclevel", required_argument, 0, 'L'},
 440                 {"secname", required_argument, 0, 'U'},
 441                 {"authproto", required_argument, 0, 'a'},
 442                 {"authpasswd", required_argument, 0, 'A'},
 443                 {"privpasswd", required_argument, 0, 'X'},
 444                 {0, 0, 0, 0}
 445         };
 446 #endif
 447
 448         if (argc < 2)
 449                 return ERROR;
 450
 451         /* reverse compatibility for very old non-POSIX usage forms */
 452         for (c = 1; c < argc; c++) {
 453                 if (strcmp ("-to", argv[c]) == 0)
 454                         strcpy (argv[c], "-t");
 455                 if (strcmp ("-wv", argv[c]) == 0)
 456                         strcpy (argv[c], "-w");
 457                 if (strcmp ("-cv", argv[c]) == 0)
 458                         strcpy (argv[c], "-c");
 459         }
 460
 461         while (1) {
 462 #ifdef HAVE_GETOPT_H
 463                 c =
 464                         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:",
 465                                                                          long_options, &option_index);
 466 #else
 467                 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:");
 468 #endif
 469
 470                 if (c == -1 || c == EOF)
 471                         break;
 472
 473                 switch (c) {
 474                 case '?':       /* usage */
 475                         usage3 ("Unknown argument", optopt);
 476                 case 'h':       /* help */
 477                         print_help ();
 478                         exit (STATE_OK);
 479                 case 'V':       /* version */
 480                         print_revision (progname, REVISION);
 481                         exit (STATE_OK);
 482                 case 'v': /* verbose */
 483                         verbose = TRUE;
 484                         break;
 485
 486         /* Connection info */
 487                 case 'C':                                                                       /* group or community */
 488                         community = strscpy (community, optarg);
 489                         break;
 490                 case 'H':                                                                       /* Host or server */
 491                         server_address = strscpy (server_address, optarg);
 492                         break;
 493                 case 'p':       /* TCP port number */
 494                         port = strscpy(port, optarg);
 495                         break;
 496                 case 'm':      /* List of MIBS  */
 497                         miblist = strscpy(miblist, optarg);
 498                         break;
 499                 case 'P':     /* SNMP protocol version */
 500                         proto = strscpy(proto, optarg);
 501                         break;
 502                 case 'L':     /* security level */
 503                         seclevel = strscpy(seclevel,optarg);
 504                         break;
 505                 case 'U':     /* security username */
 506                         secname = strscpy(secname, optarg);
 507                         break;
 508                 case 'a':     /* auth protocol */
 509                         asprintf (&authproto, optarg);
 510                         break;
 511                 case 'A':     /* auth passwd */
 512                         authpasswd = strscpy(authpasswd, optarg);
 513                         break;
 514                 case 'X':     /* priv passwd */
 515                         privpasswd = strscpy(privpasswd, optarg);
 516                         break;
 517                 case 't':       /* timeout period */
 518                         if (!is_integer (optarg))
 519                                 usage2 ("Timeout Interval must be an integer", optarg);
 520                         timeout_interval = atoi (optarg);
 521                         break;
 522
 523         /* Test parameters */
 524                 case 'c':                                                                       /* critical time threshold */
 525                         if (strspn (optarg, "0123456789:,") < strlen (optarg)) {
 526                                 printf ("Invalid critical threshold: %s\n", optarg);
 527                                 print_usage ();
 528                                 exit (STATE_UNKNOWN);
 529                         }
 530                         for (ptr = optarg; ptr && jj < MAX_OIDS; jj++) {
 531                                 if (lu_getll (&lower_crit_lim[jj], ptr) == 1)
 532                                         eval_method[jj] |= CRIT_LT;
 533                                 if (lu_getul (&upper_crit_lim[jj], ptr) == 1)
 534                                         eval_method[jj] |= CRIT_GT;
 535                                 (ptr = index (ptr, ',')) ? ptr++ : ptr;
 536                         }
 537                         break;
 538                 case 'w':                                                                       /* warning time threshold */
 539                         if (strspn (optarg, "0123456789:,") < strlen (optarg)) {
 540                                 printf ("Invalid warning threshold: %s\n", optarg);
 541                                 print_usage ();
 542                                 exit (STATE_UNKNOWN);
 543                         }
 544                         for (ptr = optarg; ptr && ii < MAX_OIDS; ii++) {
 545                                 if (lu_getll (&lower_warn_lim[ii], ptr) == 1)
 546                                         eval_method[ii] |= WARN_LT;
 547                                 if (lu_getul (&upper_warn_lim[ii], ptr) == 1)
 548                                         eval_method[ii] |= WARN_GT;
 549                                 (ptr = index (ptr, ',')) ? ptr++ : ptr;
 550                         }
 551                         break;
 552                 case 'o':                                                                       /* object identifier */
 553                 case 'e': /* PRELIMINARY - may change */
 554                 case 'E': /* PRELIMINARY - may change */
 555                         for (ptr = optarg; (ptr = index (ptr, ',')); ptr++)
 556                                 ptr[0] = ' '; /* relpace comma with space */
 557                         for (ptr = optarg; (ptr = index (ptr, ' ')); ptr++)
 558                                 j++; /* count OIDs */
 559                         asprintf (&oid, "%s %s", (oid?oid:""), optarg);
 560                         if (c == 'E' || c == 'e') {
 561                                 jj++;
 562                                 ii++;
 563                         }
 564                         if (c == 'E')
 565                                 eval_method[j+1] |= WARN_PRESENT;
 566                         else if (c == 'e')
 567                                 eval_method[j+1] |= CRIT_PRESENT;
 568                         break;
 569                 case 's':                                                                       /* string or substring */
 570                         strncpy (string_value, optarg, sizeof (string_value) - 1);
 571                         string_value[sizeof (string_value) - 1] = 0;
 572                         eval_method[jj++] = CRIT_STRING;
 573                         ii++;
 574                         break;
 575                 case 'R':                                                                       /* regex */
 576 #ifdef HAVE_REGEX_H
 577                         cflags = REG_ICASE;
 578 #endif
 579                 case 'r':                                                                       /* regex */
 580 #ifdef HAVE_REGEX_H
 581                         cflags |= REG_EXTENDED | REG_NOSUB | REG_NEWLINE;
 582                         strncpy (regex_expect, optarg, sizeof (regex_expect) - 1);
 583                         regex_expect[sizeof (regex_expect) - 1] = 0;
 584                         errcode = regcomp (&preg, regex_expect, cflags);
 585                         if (errcode != 0) {
 586                                 regerror (errcode, &preg, errbuf, MAX_INPUT_BUFFER);
 587                                 printf ("Could Not Compile Regular Expression");
 588                                 return ERROR;
 589                         }
 590                         eval_method[jj++] = CRIT_REGEX;
 591                         ii++;
 592 #else
 593                         printf ("%s UNKNOWN: call for regex which was not a compiled option", label);
 594                         exit (STATE_UNKNOWN);
 595 #endif
 596                         break;
 597
 598         /* Format */
 599                 case 'd':                                                                       /* delimiter */
 600                         delimiter = strscpy (delimiter, optarg);
 601                         break;
 602                 case 'D':                                                                       /* output-delimiter */
 603                         output_delim = strscpy (output_delim, optarg);
 604                         break;
 605                 case 'l':                                                                       /* label */
 606                         label = optarg;
 607                         nlabels++;
 608                         if (nlabels >= labels_size) {
 609                                 labels_size += 8;
 610                                 labels = realloc (labels, labels_size);
 611                                 if (labels == NULL)
 612                                         terminate (STATE_UNKNOWN,
 613                                                                                  "Could not realloc() labels[%d]", nlabels);
 614                         }
 615                         labels[nlabels - 1] = optarg;
 616                         ptr = thisarg (optarg);
 617                         if (strstr (ptr, "'") == ptr)
 618                                 labels[nlabels - 1] = ptr + 1;
 619                         else
 620                                 labels[nlabels - 1] = ptr;
 621                         while (ptr && (ptr = nextarg (ptr))) {
 622                                 if (nlabels >= labels_size) {
 623                                         labels_size += 8;
 624                                         labels = realloc (labels, labels_size);
 625                                         if (labels == NULL)
 626                                                 terminate (STATE_UNKNOWN, "Could not realloc() labels\n");
 627                                 }
 628                                 labels++;
 629                                 ptr = thisarg (ptr);
 630                                 if (strstr (ptr, "'") == ptr)
 631                                         labels[nlabels - 1] = ptr + 1;
 632                                 else
 633                                         labels[nlabels - 1] = ptr;
 634                         }
 635                         break;
 636                 case 'u':                                                                       /* units */
 637                         units = optarg;
 638                         nunits++;
 639                         if (nunits >= unitv_size) {
 640                                 unitv_size += 8;
 641                                 unitv = realloc (unitv, unitv_size);
 642                                 if (unitv == NULL)
 643                                         terminate (STATE_UNKNOWN,
 644                                                                                  "Could not realloc() units [%d]\n", nunits);
 645                         }
 646                         unitv[nunits - 1] = optarg;
 647                         ptr = thisarg (optarg);
 648                         if (strstr (ptr, "'") == ptr)
 649                                 unitv[nunits - 1] = ptr + 1;
 650                         else
 651                                 unitv[nunits - 1] = ptr;
 652                         while (ptr && (ptr = nextarg (ptr))) {
 653                                 if (nunits >= unitv_size) {
 654                                         unitv_size += 8;
 655                                         unitv = realloc (unitv, unitv_size);
 656                                         if (units == NULL)
 657                                                 terminate (STATE_UNKNOWN, "Could not realloc() units\n");
 658                                 }
 659                                 nunits++;
 660                                 ptr = thisarg (ptr);
 661                                 if (strstr (ptr, "'") == ptr)
 662                                         unitv[nunits - 1] = ptr + 1;
 663                                 else
 664                                         unitv[nunits - 1] = ptr;
 665                         }
 666                         break;
 667
 668                 }
 669         }
 670
 671         if (server_address == NULL)
 672                 asprintf (&server_address, argv[optind]);
 673
 674         return validate_arguments ();
 675 }
 676
 677 /******************************************************************************
 678
 679 @@-
 680 <sect3>
 681 <title>validate_arguments</title>
 682
 683 <para>&PROTO_validate_arguments;</para>
 684
 685 <para>Given a database name, this function returns TRUE if the string
 686 is a valid PostgreSQL database name, and returns false if it is
 687 not.</para>
 688
 689 <para>Valid PostgreSQL database names are less than &NAMEDATALEN;
 690 characters long and consist of letters, numbers, and underscores. The
 691 first character cannot be a number, however.</para>
 692
 693 </sect3>
 694 -@@
 695 ******************************************************************************/
 696
 697 int
 698 validate_arguments ()
 699 {
 700
 701         /* Need better checks to verify seclevel and authproto choices */
 702
 703         if (seclevel == NULL)
 704                 asprintf (&seclevel, "noAuthNoPriv");
 705
 706
 707         if (authproto == NULL )
 708                 asprintf(&authproto, DEFAULT_AUTH_PROTOCOL);
 709
 710
 711
 712         if (proto == NULL || (strcmp(proto,DEFAULT_PROTOCOL) == 0) ) {        /* default protocol version */
 713                 asprintf(&proto, DEFAULT_PROTOCOL);
 714                 asprintf(&authpriv, "%s%s", "-c ", community);
 715         }
 716         else if ( strcmp (proto, "3") == 0 ) {                 /* snmpv3 args */
 717                 asprintf(&proto, "%s", "3");
 718
 719                 if ( (strcmp(seclevel, "noAuthNoPriv") == 0) || seclevel == NULL ) {
 720                         asprintf(&authpriv, "%s", "-l noAuthNoPriv" );
 721                 }
 722                 else if ( strcmp(seclevel, "authNoPriv") == 0 ) {
 723                         if ( secname == NULL || authpasswd == NULL) {
 724                                 printf ("Missing secname (%s) or authpassword (%s) ! \n",secname, authpasswd );
 725                                 print_usage ();
 726                                 exit (STATE_UNKNOWN);
 727                         }
 728                         asprintf(&authpriv, "-l authNoPriv -a %s -u %s -A %s ", authproto, secname, authpasswd);
 729                 }
 730                 else if ( strcmp(seclevel, "authPriv") == 0 ) {
 731                         if ( secname == NULL || authpasswd == NULL || privpasswd == NULL ) {
 732                                 printf ("Missing secname (%s), authpassword (%s), or privpasswd (%s)! \n",secname, authpasswd,privpasswd );
 733                                 print_usage ();
 734                                 exit (STATE_UNKNOWN);
 735                         }
 736                         asprintf(&authpriv, "-l authPriv -a %s -u %s -A %s -x DES -X %s ", authproto, secname, authpasswd, privpasswd);
 737                 }
 738
 739
 740         }
 741         else {
 742                 printf ("Invalid SNMP version: %s\n", proto);
 743                 print_usage ();
 744                 exit (STATE_UNKNOWN);
 745         }
 746
 747
 748
 749
 750         return OK;
 751 }
 752
 753
 754
 755 void
 756 print_help (void)
 757 {
 758         print_revision (progname, REVISION);
 759         printf
 760                 ("Copyright (c) %s %s <%s>\n\n%s\n", COPYRIGHT, AUTHOR, EMAIL, SUMMARY);
 761         print_usage ();
 762         printf
 763                 ("\nOptions:\n" LONGOPTIONS "\n" DESCRIPTION "\n" NOTES "\n",
 764                  DEFAULT_COMMUNITY, DEFAULT_PORT, DEFAULT_DELIMITER, DEFAULT_TIMEOUT);
 765         support ();
 766 }
 767
 768 void
 769 print_usage (void)
 770 {
 771         printf
 772                 ("Usage:\n" " %s %s\n"
 773                  " %s (-h | --help) for detailed help\n"
 774                  " %s (-V | --version) for version information\n",
 775                  progname, OPTIONS, progname, progname);
 776 }
 777 \f
 778
 779
 780 char *
 781 clarify_message (char *msg)
 782 {
 783         int i = 0;
 784         int foo;
 785         char tmpmsg_c[MAX_INPUT_BUFFER];
 786         char *tmpmsg = (char *) &tmpmsg_c;
 787         tmpmsg = strcpy (tmpmsg, msg);
 788         if (!strncmp (tmpmsg, " Hex:", 5)) {
 789                 tmpmsg = strtok (tmpmsg, ":");
 790                 while ((tmpmsg = strtok (NULL, " "))) {
 791                         foo = strtol (tmpmsg, NULL, 16);
 792                         /* Translate chars that are not the same value in the printers
 793                          * character set.
 794                          */
 795                         switch (foo) {
 796                         case 208:
 797                                 {
 798                                         foo = 197;
 799                                         break;
 800                                 }
 801                         case 216:
 802                                 {
 803                                         foo = 196;
 804                                         break;
 805                                 }
 806                         }
 807                         msg[i] = foo;
 808                         i++;
 809                 }
 810                 msg[i] = 0;
 811         }
 812         return (msg);
 813 }
 814
 815
 816 int
 817 check_num (int i)
 818 {
 819         int result;
 820         result = STATE_OK;
 821         if (eval_method[i] & WARN_GT && eval_method[i] & WARN_LT &&
 822                         lower_warn_lim[i] > upper_warn_lim[i]) {
 823                 if (response_value[i] <= lower_warn_lim[i] &&
 824                                 response_value[i] >= upper_warn_lim[i]) {
 825                         result = STATE_WARNING;
 826                 }
 827         }
 828         else if
 829                 ((eval_method[i] & WARN_GT && response_value[i] > upper_warn_lim[i]) ||
 830                  (eval_method[i] & WARN_GE && response_value[i] >= upper_warn_lim[i]) ||
 831                  (eval_method[i] & WARN_LT && response_value[i] < lower_warn_lim[i]) ||
 832                  (eval_method[i] & WARN_LE && response_value[i] <= lower_warn_lim[i]) ||
 833                  (eval_method[i] & WARN_EQ && response_value[i] == upper_warn_lim[i]) ||
 834                  (eval_method[i] & WARN_NE && response_value[i] != upper_warn_lim[i])) {
 835                 result = STATE_WARNING;
 836         }
 837
 838         if (eval_method[i] & CRIT_GT && eval_method[i] & CRIT_LT &&
 839                         lower_warn_lim[i] > upper_warn_lim[i]) {
 840                 if (response_value[i] <= lower_crit_lim[i] &&
 841                                 response_value[i] >= upper_crit_lim[i]) {
 842                         result = STATE_CRITICAL;
 843                 }
 844         }
 845         else if
 846                 ((eval_method[i] & CRIT_GT && response_value[i] > upper_crit_lim[i]) ||
 847                  (eval_method[i] & CRIT_GE && response_value[i] >= upper_crit_lim[i]) ||
 848                  (eval_method[i] & CRIT_LT && response_value[i] < lower_crit_lim[i]) ||
 849                  (eval_method[i] & CRIT_LE && response_value[i] <= lower_crit_lim[i]) ||
 850                  (eval_method[i] & CRIT_EQ && response_value[i] == upper_crit_lim[i]) ||
 851                  (eval_method[i] & CRIT_NE && response_value[i] != upper_crit_lim[i])) {
 852                 result = STATE_CRITICAL;
 853         }
 854
 855         return result;
 856 }
 857
 858
 859 int
 860 lu_getll (unsigned long *ll, char *str)
 861 {
 862         char tmp[100];
 863         if (strchr (str, ':') == NULL)
 864                 return 0;
 865         if (strchr (str, ',') != NULL && (strchr (str, ',') < strchr (str, ':')))
 866                 return 0;
 867         if (sscanf (str, "%lu%[:]", ll, tmp) == 2)
 868                 return 1;
 869         return 0;
 870 }
 871
 872 int
 873 lu_getul (unsigned long *ul, char *str)
 874 {
 875         char tmp[100];
 876         if (sscanf (str, "%lu%[^,]", ul, tmp) == 1)
 877                 return 1;
 878         if (sscanf (str, ":%lu%[^,]", ul, tmp) == 1)
 879                 return 1;
 880         if (sscanf (str, "%*u:%lu%[^,]", ul, tmp) == 1)
 881                 return 1;
 882         return 0;
 883 }
 884
 885
 886
 887
 888
 889
 890 /* trim leading whitespace
 891          if there is a leading quote, make sure it balances */
 892
 893 char *
 894 thisarg (char *str)
 895 {
 896         str += strspn (str, " \t\r\n"); /* trim any leading whitespace */
 897         if (strstr (str, "'") == str) { /* handle SIMPLE quoted strings */
 898                 if (strlen (str) == 1 || !strstr (str + 1, "'"))
 899                         terminate (STATE_UNKNOWN, "Unbalanced quotes\n");
 900         }
 901         return str;
 902 }
 903
 904
 905 /* if there's a leading quote, advance to the trailing quote
 906          set the trailing quote to '\x0'
 907          if the string continues, advance beyond the comma */
 908
 909 char *
 910 nextarg (char *str)
 911 {
 912         if (strstr (str, "'") == str) {
 913                 if (strlen (str) > 1) {
 914                         str = strstr (str + 1, "'");
 915                         str[0] = 0;
 916                         return (++str);
 917                 }
 918                 else {
 919                         str[0] = 0;
 920                         return NULL;
 921                 }
 922         }
 923         if (strstr (str, ",") == str) {
 924                 if (strlen (str) > 1) {
 925                         str[0] = 0;
 926                         return (++str);
 927                 }
 928                 else {
 929                         str[0] = 0;
 930                         return NULL;
 931                 }
 932         }
 933         if ((str = strstr (str, ",")) && strlen (str) > 1) {
 934                 str[0] = 0;
 935                 return (++str);
 936         }
 937         return NULL;
 938 }