src/collectd-nagios.c

   1 /**
   2  * collectd-nagios - src/collectd-nagios.c
   3  * Copyright (C) 2008-2010  Florian octo Forster
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License as published by the
   7  * Free Software Foundation; only version 2 of the License is applicable.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  17  *
  18  * Authors:
  19  *   Florian octo Forster <octo at verplant.org>
  20  **/
  21
  22 #if HAVE_CONFIG_H
  23 # include "config.h"
  24 #endif
  25
  26 #if !defined(__GNUC__) || !__GNUC__
  27 # define __attribute__(x) /**/
  28 #endif
  29
  30 #include <stdlib.h>
  31 #include <unistd.h>
  32 #include <stdio.h>
  33 #include <errno.h>
  34 #include <string.h>
  35 #include <strings.h>
  36 #include <assert.h>
  37
  38 #if NAN_STATIC_DEFAULT
  39 # include <math.h>
  40 /* #endif NAN_STATIC_DEFAULT*/
  41 #elif NAN_STATIC_ISOC
  42 # ifndef __USE_ISOC99
  43 #  define DISABLE_ISOC99 1
  44 #  define __USE_ISOC99 1
  45 # endif /* !defined(__USE_ISOC99) */
  46 # include <math.h>
  47 # if DISABLE_ISOC99
  48 #  undef DISABLE_ISOC99
  49 #  undef __USE_ISOC99
  50 # endif /* DISABLE_ISOC99 */
  51 /* #endif NAN_STATIC_ISOC */
  52 #elif NAN_ZERO_ZERO
  53 # include <math.h>
  54 # ifdef NAN
  55 #  undef NAN
  56 # endif
  57 # define NAN (0.0 / 0.0)
  58 # ifndef isnan
  59 #  define isnan(f) ((f) != (f))
  60 # endif /* !defined(isnan) */
  61 # ifndef isfinite
  62 #  define isfinite(f) (((f) - (f)) == 0.0)
  63 # endif
  64 # ifndef isinf
  65 #  define isinf(f) (!isfinite(f) && !isnan(f))
  66 # endif
  67 #endif /* NAN_ZERO_ZERO */
  68
  69 #include "libcollectdclient/client.h"
  70
  71 #define RET_OKAY     0
  72 #define RET_WARNING  1
  73 #define RET_CRITICAL 2
  74 #define RET_UNKNOWN  3
  75
  76 #define CON_NONE     0
  77 #define CON_AVERAGE  1
  78 #define CON_SUM      2
  79 #define CON_PERCENTAGE  3
  80
  81 struct range_s
  82 {
  83         double min;
  84         double max;
  85         int    invert;
  86 };
  87 typedef struct range_s range_t;
  88
  89 extern char *optarg;
  90 extern int optind, opterr, optopt;
  91
  92 static char *socket_file_g = NULL;
  93 static char *value_string_g = NULL;
  94 static char *hostname_g = NULL;
  95
  96 static range_t range_critical_g;
  97 static range_t range_warning_g;
  98 static int consolitation_g = CON_NONE;
  99 static _Bool nan_is_error_g = 0;
 100
 101 static char **match_ds_g = NULL;
 102 static int    match_ds_num_g = 0;
 103
 104 /* `strdup' is an XSI extension. I don't want to pull in all of XSI just for
 105  * that, so here's an own implementation.. It's easy enough. The GCC attributes
 106  * are supposed to get good performance..  -octo */
 107 __attribute__((malloc, nonnull (1)))
 108 static char *cn_strdup (const char *str) /* {{{ */
 109 {
 110   size_t strsize;
 111   char *ret;
 112
 113   strsize = strlen (str) + 1;
 114   ret = (char *) malloc (strsize);
 115   if (ret != NULL)
 116     memcpy (ret, str, strsize);
 117   return (ret);
 118 } /* }}} char *cn_strdup */
 119
 120 static int filter_ds (size_t *values_num,
 121                 double **values, char ***values_names)
 122 {
 123         gauge_t *new_values;
 124         char   **new_names;
 125
 126         size_t i;
 127
 128         if (match_ds_g == NULL)
 129                 return (RET_OKAY);
 130
 131         new_values = (gauge_t *)calloc (match_ds_num_g, sizeof (*new_values));
 132         if (new_values == NULL)
 133         {
 134                 fprintf (stderr, "malloc failed: %s\n", strerror (errno));
 135                 return (RET_UNKNOWN);
 136         }
 137
 138         new_names = (char **)calloc (match_ds_num_g, sizeof (*new_names));
 139         if (new_names == NULL)
 140         {
 141                 fprintf (stderr, "malloc failed: %s\n", strerror (errno));
 142                 free (new_values);
 143                 return (RET_UNKNOWN);
 144         }
 145
 146         for (i = 0; i < (size_t) match_ds_num_g; i++)
 147         {
 148                 size_t j;
 149
 150                 /* match_ds_g keeps pointers into argv but the names will be freed */
 151                 new_names[i] = cn_strdup (match_ds_g[i]);
 152                 if (new_names[i] == NULL)
 153                 {
 154                         fprintf (stderr, "cn_strdup failed: %s\n", strerror (errno));
 155                         free (new_values);
 156                         for (j = 0; j < i; j++)
 157                                 free (new_names[j]);
 158                         free (new_names);
 159                         return (RET_UNKNOWN);
 160                 }
 161
 162                 for (j = 0; j < *values_num; j++)
 163                         if (strcasecmp (new_names[i], (*values_names)[j]) == 0)
 164                                 break;
 165
 166                 if (j == *values_num)
 167                 {
 168                         printf ("ERROR: DS `%s' is not available.\n", new_names[i]);
 169                         free (new_values);
 170                         for (j = 0; j <= i; j++)
 171                                 free (new_names[j]);
 172                         free (new_names);
 173                         return (RET_CRITICAL);
 174                 }
 175
 176                 new_values[i] = (*values)[j];
 177         }
 178
 179         free (*values);
 180         for (i = 0; i < *values_num; i++)
 181                 free ((*values_names)[i]);
 182         free (*values_names);
 183
 184         *values       = new_values;
 185         *values_names = new_names;
 186         *values_num   = match_ds_num_g;
 187         return (RET_OKAY);
 188 } /* int filter_ds */
 189
 190 static void parse_range (char *string, range_t *range)
 191 {
 192         char *min_ptr;
 193         char *max_ptr;
 194
 195         if (*string == '@')
 196         {
 197                 range->invert = 1;
 198                 string++;
 199         }
 200
 201         max_ptr = strchr (string, ':');
 202         if (max_ptr == NULL)
 203         {
 204                 min_ptr = NULL;
 205                 max_ptr = string;
 206         }
 207         else
 208         {
 209                 min_ptr = string;
 210                 *max_ptr = '\0';
 211                 max_ptr++;
 212         }
 213
 214         assert (max_ptr != NULL);
 215
 216         /* `10' == `0:10' */
 217         if (min_ptr == NULL)
 218                 range->min = 0.0;
 219         /* :10 == ~:10 == -inf:10 */
 220         else if ((*min_ptr == '\0') || (*min_ptr == '~'))
 221                 range->min = NAN;
 222         else
 223                 range->min = atof (min_ptr);
 224
 225         if ((*max_ptr == '\0') || (*max_ptr == '~'))
 226                 range->max = NAN;
 227         else
 228                 range->max = atof (max_ptr);
 229 } /* void parse_range */
 230
 231 static int match_range (range_t *range, double value)
 232 {
 233         int ret = 0;
 234
 235         if (!isnan (range->min) && (range->min > value))
 236                 ret = 1;
 237         if (!isnan (range->max) && (range->max < value))
 238                 ret = 1;
 239
 240         return (((ret - range->invert) == 0) ? 0 : 1);
 241 } /* int match_range */
 242
 243 static void usage (const char *name)
 244 {
 245         fprintf (stderr, "Usage: %s <-s socket> <-n value_spec> <-H hostname> [options]\n"
 246                         "\n"
 247                         "Valid options are:\n"
 248                         "  -s <socket>    Path to collectd's UNIX-socket.\n"
 249                         "  -n <v_spec>    Value specification to get from collectd.\n"
 250                         "                 Format: `plugin-instance/type-instance'\n"
 251                         "  -d <ds>        Select the DS to examine. May be repeated to examine multiple\n"
 252                         "                 DSes. By default all DSes are used.\n"
 253                         "  -g <consol>    Method to use to consolidate several DSes.\n"
 254                         "                 See below for a list of valid arguments.\n"
 255                         "  -H <host>      Hostname to query the values for.\n"
 256                         "  -c <range>     Critical range\n"
 257                         "  -w <range>     Warning range\n"
 258                         "  -m             Treat \"Not a Number\" (NaN) as critical (default: warning)\n"
 259                         "\n"
 260                         "Consolidation functions:\n"
 261                         "  none:          Apply the warning- and critical-ranges to each data-source\n"
 262                         "                 individually.\n"
 263                         "  average:       Calculate the average of all matching DSes and apply the\n"
 264                         "                 warning- and critical-ranges to the calculated average.\n"
 265                         "  sum:           Apply the ranges to the sum of all DSes.\n"
 266                         "  percentage:    Apply the ranges to the ratio (in percent) of the first value\n"
 267                         "                 and the sum of all values."
 268                         "\n", name);
 269         exit (1);
 270 } /* void usage */
 271
 272 static int do_listval (lcc_connection_t *connection)
 273 {
 274         lcc_identifier_t *ret_ident = NULL;
 275         size_t ret_ident_num = 0;
 276
 277         int status;
 278         size_t i;
 279
 280         status = lcc_listval (connection, &ret_ident, &ret_ident_num);
 281         if (status != 0) {
 282                 printf ("UNKNOWN: %s\n", lcc_strerror (connection));
 283                 if (ret_ident != NULL)
 284                         free (ret_ident);
 285                 return (RET_UNKNOWN);
 286         }
 287
 288         for (i = 0; i < ret_ident_num; ++i) {
 289                 char id[1024];
 290
 291                 if ((hostname_g != NULL) && (strcasecmp (hostname_g, ret_ident[i].host)))
 292                         continue;
 293
 294                 status = lcc_identifier_to_string (connection,
 295                                 id, sizeof (id), ret_ident + i);
 296                 if (status != 0) {
 297                         printf ("ERROR: listval: Failed to convert returned "
 298                                         "identifier to a string: %s\n",
 299                                         lcc_strerror (connection));
 300                         continue;
 301                 }
 302
 303                 printf ("%s\n", id);
 304         }
 305
 306         if (ret_ident != NULL)
 307                 free (ret_ident);
 308         return (RET_OKAY);
 309 } /* int do_listval */
 310
 311 static int do_check_con_none (size_t values_num,
 312                 double *values, char **values_names)
 313 {
 314         int num_critical = 0;
 315         int num_warning  = 0;
 316         int num_okay = 0;
 317         const char *status_str = "UNKNOWN";
 318         int status_code = RET_UNKNOWN;
 319         size_t i;
 320
 321         for (i = 0; i < values_num; i++)
 322         {
 323                 if (isnan (values[i]))
 324                 {
 325                         if (nan_is_error_g)
 326                                 num_critical++;
 327                         else
 328                                 num_warning++;
 329                 }
 330                 else if (match_range (&range_critical_g, values[i]) != 0)
 331                         num_critical++;
 332                 else if (match_range (&range_warning_g, values[i]) != 0)
 333                         num_warning++;
 334                 else
 335                         num_okay++;
 336         }
 337
 338         if ((num_critical == 0) && (num_warning == 0) && (num_okay == 0))
 339         {
 340                 printf ("WARNING: No defined values found\n");
 341                 return (RET_WARNING);
 342         }
 343         else if ((num_critical == 0) && (num_warning == 0))
 344         {
 345                 status_str = "OKAY";
 346                 status_code = RET_OKAY;
 347         }
 348         else if (num_critical == 0)
 349         {
 350                 status_str = "WARNING";
 351                 status_code = RET_WARNING;
 352         }
 353         else
 354         {
 355                 status_str = "CRITICAL";
 356                 status_code = RET_CRITICAL;
 357         }
 358
 359         printf ("%s: %i critical, %i warning, %i okay", status_str,
 360                         num_critical, num_warning, num_okay);
 361         if (values_num > 0)
 362         {
 363                 printf (" |");
 364                 for (i = 0; i < values_num; i++)
 365                         printf (" %s=%f;;;;", values_names[i], values[i]);
 366         }
 367         printf ("\n");
 368
 369         return (status_code);
 370 } /* int do_check_con_none */
 371
 372 static int do_check_con_average (size_t values_num,
 373                 double *values, char **values_names)
 374 {
 375         size_t i;
 376         double total;
 377         int total_num;
 378         double average;
 379         const char *status_str = "UNKNOWN";
 380         int status_code = RET_UNKNOWN;
 381
 382         total = 0.0;
 383         total_num = 0;
 384         for (i = 0; i < values_num; i++)
 385         {
 386                 if (isnan (values[i]))
 387                 {
 388                         if (!nan_is_error_g)
 389                                 continue;
 390
 391                         printf ("CRITICAL: Data source \"%s\" is NaN\n",
 392                                         values_names[i]);
 393                         return (RET_CRITICAL);
 394                 }
 395
 396                 total += values[i];
 397                 total_num++;
 398         }
 399
 400         if (total_num == 0)
 401         {
 402                 printf ("WARNING: No defined values found\n");
 403                 return (RET_WARNING);
 404         }
 405
 406         average = total / total_num;
 407
 408         if (match_range (&range_critical_g, average) != 0)
 409         {
 410                 status_str = "CRITICAL";
 411                 status_code = RET_CRITICAL;
 412         }
 413         else if (match_range (&range_warning_g, average) != 0)
 414         {
 415                 status_str = "WARNING";
 416                 status_code = RET_WARNING;
 417         }
 418         else
 419         {
 420                 status_str = "OKAY";
 421                 status_code = RET_OKAY;
 422         }
 423
 424         printf ("%s: %g average |", status_str, average);
 425         for (i = 0; i < values_num; i++)
 426                 printf (" %s=%f;;;;", values_names[i], values[i]);
 427         printf ("\n");
 428
 429         return (status_code);
 430 } /* int do_check_con_average */
 431
 432 static int do_check_con_sum (size_t values_num,
 433                 double *values, char **values_names)
 434 {
 435         size_t i;
 436         double total;
 437         int total_num;
 438         const char *status_str = "UNKNOWN";
 439         int status_code = RET_UNKNOWN;
 440
 441         total = 0.0;
 442         total_num = 0;
 443         for (i = 0; i < values_num; i++)
 444         {
 445                 if (isnan (values[i]))
 446                 {
 447                         if (!nan_is_error_g)
 448                                 continue;
 449
 450                         printf ("CRITICAL: Data source \"%s\" is NaN\n",
 451                                         values_names[i]);
 452                         return (RET_CRITICAL);
 453                 }
 454
 455                 total += values[i];
 456                 total_num++;
 457         }
 458
 459         if (total_num == 0)
 460         {
 461                 printf ("WARNING: No defined values found\n");
 462                 return (RET_WARNING);
 463         }
 464
 465         if (match_range (&range_critical_g, total) != 0)
 466         {
 467                 status_str = "CRITICAL";
 468                 status_code = RET_CRITICAL;
 469         }
 470         else if (match_range (&range_warning_g, total) != 0)
 471         {
 472                 status_str = "WARNING";
 473                 status_code = RET_WARNING;
 474         }
 475         else
 476         {
 477                 status_str = "OKAY";
 478                 status_code = RET_OKAY;
 479         }
 480
 481         printf ("%s: %g sum |", status_str, total);
 482         for (i = 0; i < values_num; i++)
 483                 printf (" %s=%f;;;;", values_names[i], values[i]);
 484         printf ("\n");
 485
 486         return (status_code);
 487 } /* int do_check_con_sum */
 488
 489 static int do_check_con_percentage (size_t values_num,
 490                 double *values, char **values_names)
 491 {
 492         size_t i;
 493         double sum = 0.0;
 494         double percentage;
 495
 496         const char *status_str  = "UNKNOWN";
 497         int         status_code = RET_UNKNOWN;
 498
 499         if ((values_num < 1) || (isnan (values[0])))
 500         {
 501                 printf ("WARNING: The first value is not defined\n");
 502                 return (RET_WARNING);
 503         }
 504
 505         for (i = 0; i < values_num; i++)
 506         {
 507                 if (isnan (values[i]))
 508                 {
 509                         if (!nan_is_error_g)
 510                                 continue;
 511
 512                         printf ("CRITICAL: Data source \"%s\" is NaN\n",
 513                                         values_names[i]);
 514                         return (RET_CRITICAL);
 515                 }
 516
 517                 sum += values[i];
 518         }
 519
 520         if (sum == 0.0)
 521         {
 522                 printf ("WARNING: Values sum up to zero\n");
 523                 return (RET_WARNING);
 524         }
 525
 526         percentage = 100.0 * values[0] / sum;
 527
 528         if (match_range (&range_critical_g, percentage) != 0)
 529         {
 530                 status_str  = "CRITICAL";
 531                 status_code = RET_CRITICAL;
 532         }
 533         else if (match_range (&range_warning_g, percentage) != 0)
 534         {
 535                 status_str  = "WARNING";
 536                 status_code = RET_WARNING;
 537         }
 538         else
 539         {
 540                 status_str  = "OKAY";
 541                 status_code = RET_OKAY;
 542         }
 543
 544         printf ("%s: %lf percent |", status_str, percentage);
 545         for (i = 0; i < values_num; i++)
 546                 printf (" %s=%lf;;;;", values_names[i], values[i]);
 547         return (status_code);
 548 } /* int do_check_con_percentage */
 549
 550 static int do_check (lcc_connection_t *connection)
 551 {
 552         gauge_t *values;
 553         char   **values_names;
 554         size_t   values_num;
 555         char ident_str[1024];
 556         lcc_identifier_t ident;
 557         size_t i;
 558         int status;
 559
 560         snprintf (ident_str, sizeof (ident_str), "%s/%s",
 561                         hostname_g, value_string_g);
 562         ident_str[sizeof (ident_str) - 1] = 0;
 563
 564         memset (&ident, 0, sizeof (ident));
 565         status = lcc_string_to_identifier (connection, &ident, ident_str);
 566         if (status != 0)
 567         {
 568                 printf ("ERROR: Creating an identifier failed: %s.\n",
 569                                 lcc_strerror (connection));
 570                 LCC_DESTROY (connection);
 571                 return (RET_CRITICAL);
 572         }
 573
 574         status = lcc_getval (connection, &ident,
 575                         &values_num, &values, &values_names);
 576         if (status != 0)
 577         {
 578                 printf ("ERROR: Retrieving values from the daemon failed: %s.\n",
 579                                 lcc_strerror (connection));
 580                 LCC_DESTROY (connection);
 581                 return (RET_CRITICAL);
 582         }
 583
 584         LCC_DESTROY (connection);
 585
 586         status = filter_ds (&values_num, &values, &values_names);
 587         if (status != RET_OKAY)
 588                 return (status);
 589
 590         status = RET_UNKNOWN;
 591         if (consolitation_g == CON_NONE)
 592                 status =  do_check_con_none (values_num, values, values_names);
 593         else if (consolitation_g == CON_AVERAGE)
 594                 status =  do_check_con_average (values_num, values, values_names);
 595         else if (consolitation_g == CON_SUM)
 596                 status = do_check_con_sum (values_num, values, values_names);
 597         else if (consolitation_g == CON_PERCENTAGE)
 598                 status = do_check_con_percentage (values_num, values, values_names);
 599
 600         free (values);
 601         if (values_names != NULL)
 602                 for (i = 0; i < values_num; i++)
 603                         free (values_names[i]);
 604         free (values_names);
 605
 606         return (status);
 607 } /* int do_check */
 608
 609 int main (int argc, char **argv)
 610 {
 611         char address[1024];
 612         lcc_connection_t *connection;
 613
 614         int status;
 615
 616         range_critical_g.min = NAN;
 617         range_critical_g.max = NAN;
 618         range_critical_g.invert = 0;
 619
 620         range_warning_g.min = NAN;
 621         range_warning_g.max = NAN;
 622         range_warning_g.invert = 0;
 623
 624         while (42)
 625         {
 626                 int c;
 627
 628                 c = getopt (argc, argv, "w:c:s:n:H:g:d:hm");
 629                 if (c < 0)
 630                         break;
 631
 632                 switch (c)
 633                 {
 634                         case 'c':
 635                                 parse_range (optarg, &range_critical_g);
 636                                 break;
 637                         case 'w':
 638                                 parse_range (optarg, &range_warning_g);
 639                                 break;
 640                         case 's':
 641                                 socket_file_g = optarg;
 642                                 break;
 643                         case 'n':
 644                                 value_string_g = optarg;
 645                                 break;
 646                         case 'H':
 647                                 hostname_g = optarg;
 648                                 break;
 649                         case 'g':
 650                                 if (strcasecmp (optarg, "none") == 0)
 651                                         consolitation_g = CON_NONE;
 652                                 else if (strcasecmp (optarg, "average") == 0)
 653                                         consolitation_g = CON_AVERAGE;
 654                                 else if (strcasecmp (optarg, "sum") == 0)
 655                                         consolitation_g = CON_SUM;
 656                                 else if (strcasecmp (optarg, "percentage") == 0)
 657                                         consolitation_g = CON_PERCENTAGE;
 658                                 else
 659                                 {
 660                                         fprintf (stderr, "Unknown consolidation function `%s'.\n",
 661                                                         optarg);
 662                                         usage (argv[0]);
 663                                 }
 664                                 break;
 665                         case 'd':
 666                         {
 667                                 char **tmp;
 668                                 tmp = (char **) realloc (match_ds_g,
 669                                                 (match_ds_num_g + 1)
 670                                                 * sizeof (char *));
 671                                 if (tmp == NULL)
 672                                 {
 673                                         fprintf (stderr, "realloc failed: %s\n",
 674                                                         strerror (errno));
 675                                         return (RET_UNKNOWN);
 676                                 }
 677                                 match_ds_g = tmp;
 678                                 match_ds_g[match_ds_num_g] = cn_strdup (optarg);
 679                                 if (match_ds_g[match_ds_num_g] == NULL)
 680                                 {
 681                                         fprintf (stderr, "cn_strdup failed: %s\n",
 682                                                         strerror (errno));
 683                                         return (RET_UNKNOWN);
 684                                 }
 685                                 match_ds_num_g++;
 686                                 break;
 687                         }
 688                         case 'm':
 689                                 nan_is_error_g = 1;
 690                                 break;
 691                         default:
 692                                 usage (argv[0]);
 693                 } /* switch (c) */
 694         }
 695
 696         if ((socket_file_g == NULL) || (value_string_g == NULL)
 697                         || ((hostname_g == NULL) && (strcasecmp (value_string_g, "LIST"))))
 698         {
 699                 fprintf (stderr, "Missing required arguments.\n");
 700                 usage (argv[0]);
 701         }
 702
 703         snprintf (address, sizeof (address), "unix:%s", socket_file_g);
 704         address[sizeof (address) - 1] = 0;
 705
 706         connection = NULL;
 707         status = lcc_connect (address, &connection);
 708         if (status != 0)
 709         {
 710                 printf ("ERROR: Connecting to daemon at %s failed.\n",
 711                                 socket_file_g);
 712                 return (RET_CRITICAL);
 713         }
 714
 715         if (0 == strcasecmp (value_string_g, "LIST"))
 716                 return (do_listval (connection));
 717
 718         return (do_check (connection));
 719 } /* int main */