2 * collectd-nagios - src/collectd-nagios.c
3 * Copyright (C) 2008 Florian octo Forster
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.
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.
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
19 * Florian octo Forster <octo at verplant.org>
22 /* Set to C99 and POSIX code */
23 #ifndef _ISOC99_SOURCE
24 # define _ISOC99_SOURCE
27 # define _POSIX_SOURCE
29 #ifndef _POSIX_C_SOURCE
30 # define _POSIX_C_SOURCE 200112L
36 /* Disable non-standard extensions */
47 #if !defined(__GNUC__) || !__GNUC__
48 # define __attribute__(x) /**/
61 #include <sys/socket.h>
64 #include "libcollectdclient/client.h"
67 * This is copied directly from collectd.h. Make changes there!
69 #if NAN_STATIC_DEFAULT
71 /* #endif NAN_STATIC_DEFAULT*/
74 # define DISABLE_ISOC99 1
75 # define __USE_ISOC99 1
76 # endif /* !defined(__USE_ISOC99) */
79 # undef DISABLE_ISOC99
81 # endif /* DISABLE_ISOC99 */
82 /* #endif NAN_STATIC_ISOC */
88 # define NAN (0.0 / 0.0)
90 # define isnan(f) ((f) != (f))
91 # endif /* !defined(isnan) */
92 #endif /* NAN_ZERO_ZERO */
96 #define RET_CRITICAL 2
100 #define CON_AVERAGE 1
109 typedef struct range_s range_t;
112 extern int optind, opterr, optopt;
114 static char *socket_file_g = NULL;
115 static char *value_string_g = NULL;
116 static char *hostname_g = NULL;
118 static range_t range_critical_g;
119 static range_t range_warning_g;
120 static int consolitation_g = CON_NONE;
122 static char **match_ds_g = NULL;
123 static int match_ds_num_g = 0;
125 /* `strdup' is an XSI extension. I don't want to pull in all of XSI just for
126 * that, so here's an own implementation.. It's easy enough. The GCC attributes
127 * are supposed to get good performance.. -octo */
128 __attribute__((malloc, nonnull (1)))
129 static char *cn_strdup (const char *str) /* {{{ */
134 strsize = strlen (str) + 1;
135 ret = (char *) malloc (strsize);
137 memcpy (ret, str, strsize);
139 } /* }}} char *cn_strdup */
141 static int filter_ds (size_t *values_num,
142 double **values, char ***values_names)
149 if (match_ds_g == NULL)
152 new_values = (gauge_t *)calloc (match_ds_num_g, sizeof (*new_values));
153 if (new_values == NULL)
155 fprintf (stderr, "malloc failed: %s\n", strerror (errno));
156 return (RET_UNKNOWN);
159 new_names = (char **)calloc (match_ds_num_g, sizeof (*new_names));
160 if (new_names == NULL)
162 fprintf (stderr, "malloc failed: %s\n", strerror (errno));
164 return (RET_UNKNOWN);
167 for (i = 0; i < match_ds_num_g; i++)
171 /* match_ds_g keeps pointers into argv but the names will be freed */
172 new_names[i] = cn_strdup (match_ds_g[i]);
173 if (new_names[i] == NULL)
175 fprintf (stderr, "cn_strdup failed: %s\n", strerror (errno));
177 for (j = 0; j < i; j++)
180 return (RET_UNKNOWN);
183 for (j = 0; j < *values_num; j++)
184 if (strcasecmp (new_names[i], (*values_names)[j]) == 0)
187 if (j == *values_num)
189 printf ("ERROR: DS `%s' is not available.\n", new_names[i]);
191 for (j = 0; j <= i; j++)
194 return (RET_CRITICAL);
197 new_values[i] = (*values)[j];
201 for (i = 0; i < *values_num; i++)
202 free ((*values_names)[i]);
203 free (*values_names);
205 *values = new_values;
206 *values_names = new_names;
207 *values_num = match_ds_num_g;
209 } /* int filter_ds */
211 static void parse_range (char *string, range_t *range)
222 max_ptr = strchr (string, ':');
235 assert (max_ptr != NULL);
240 /* :10 == ~:10 == -inf:10 */
241 else if ((*min_ptr == '\0') || (*min_ptr == '~'))
244 range->min = atof (min_ptr);
246 if ((*max_ptr == '\0') || (*max_ptr == '~'))
249 range->max = atof (max_ptr);
250 } /* void parse_range */
252 static int match_range (range_t *range, double value)
256 if (!isnan (range->min) && (range->min > value))
258 if (!isnan (range->max) && (range->max < value))
261 return (((ret - range->invert) == 0) ? 0 : 1);
262 } /* int match_range */
264 static void usage (const char *name)
266 fprintf (stderr, "Usage: %s <-s socket> <-n value_spec> <-H hostname> [options]\n"
268 "Valid options are:\n"
269 " -s <socket> Path to collectd's UNIX-socket.\n"
270 " -n <v_spec> Value specification to get from collectd.\n"
271 " Format: `plugin-instance/type-instance'\n"
272 " -d <ds> Select the DS to examine. May be repeated to examine multiple\n"
273 " DSes. By default all DSes are used.\n"
274 " -g <consol> Method to use to consolidate several DSes.\n"
275 " Valid arguments are `none', `average' and `sum'\n"
276 " -H <host> Hostname to query the values for.\n"
277 " -c <range> Critical range\n"
278 " -w <range> Warning range\n"
280 "Consolidation functions:\n"
281 " none: Apply the warning- and critical-ranges to each data-source\n"
283 " average: Calculate the average of all matching DSes and apply the\n"
284 " warning- and critical-ranges to the calculated average.\n"
285 " sum: Apply the ranges to the sum of all DSes.\n"
290 static int do_check_con_none (size_t values_num,
291 double *values, char **values_names)
293 int num_critical = 0;
296 const char *status_str = "UNKNOWN";
297 int status_code = RET_UNKNOWN;
300 for (i = 0; i < values_num; i++)
302 if (isnan (values[i]))
304 else if (match_range (&range_critical_g, values[i]) != 0)
306 else if (match_range (&range_warning_g, values[i]) != 0)
312 if ((num_critical == 0) && (num_warning == 0) && (num_okay == 0))
314 printf ("WARNING: No defined values found\n");
315 return (RET_WARNING);
317 else if ((num_critical == 0) && (num_warning == 0))
320 status_code = RET_OKAY;
322 else if (num_critical == 0)
324 status_str = "WARNING";
325 status_code = RET_WARNING;
329 status_str = "CRITICAL";
330 status_code = RET_CRITICAL;
333 printf ("%s: %i critical, %i warning, %i okay", status_str,
334 num_critical, num_warning, num_okay);
338 for (i = 0; i < values_num; i++)
339 printf (" %s=%g;;;;", values_names[i], values[i]);
343 return (status_code);
344 } /* int do_check_con_none */
346 static int do_check_con_average (size_t values_num,
347 double *values, char **values_names)
353 const char *status_str = "UNKNOWN";
354 int status_code = RET_UNKNOWN;
358 for (i = 0; i < values_num; i++)
360 if (!isnan (values[i]))
369 printf ("WARNING: No defined values found\n");
370 return (RET_WARNING);
373 average = total / total_num;
375 if (match_range (&range_critical_g, average) != 0)
377 status_str = "CRITICAL";
378 status_code = RET_CRITICAL;
380 else if (match_range (&range_warning_g, average) != 0)
382 status_str = "WARNING";
383 status_code = RET_WARNING;
388 status_code = RET_OKAY;
391 printf ("%s: %g average |", status_str, average);
392 for (i = 0; i < values_num; i++)
393 printf (" %s=%g;;;;", values_names[i], values[i]);
396 return (status_code);
397 } /* int do_check_con_average */
399 static int do_check_con_sum (size_t values_num,
400 double *values, char **values_names)
405 const char *status_str = "UNKNOWN";
406 int status_code = RET_UNKNOWN;
410 for (i = 0; i < values_num; i++)
412 if (!isnan (values[i]))
421 printf ("WARNING: No defined values found\n");
422 return (RET_WARNING);
425 if (match_range (&range_critical_g, total) != 0)
427 status_str = "CRITICAL";
428 status_code = RET_CRITICAL;
430 else if (match_range (&range_warning_g, total) != 0)
432 status_str = "WARNING";
433 status_code = RET_WARNING;
438 status_code = RET_OKAY;
441 printf ("%s: %g sum |", status_str, total);
442 for (i = 0; i < values_num; i++)
443 printf (" %s=%g;;;;", values_names[i], values[i]);
446 return (status_code);
447 } /* int do_check_con_sum */
449 static int do_check (void)
451 lcc_connection_t *connection;
456 char ident_str[1024];
457 lcc_identifier_t ident;
461 snprintf (address, sizeof (address), "unix:%s", socket_file_g);
462 address[sizeof (address) - 1] = 0;
464 snprintf (ident_str, sizeof (ident_str), "%s/%s",
465 hostname_g, value_string_g);
466 ident_str[sizeof (ident_str) - 1] = 0;
469 status = lcc_connect (address, &connection);
472 printf ("ERROR: Connecting to daemon at %s failed.\n",
474 return (RET_CRITICAL);
477 memset (&ident, 0, sizeof (ident));
478 status = lcc_string_to_identifier (connection, &ident, ident_str);
481 printf ("ERROR: Creating an identifier failed: %s.\n",
482 lcc_strerror (connection));
483 LCC_DESTROY (connection);
484 return (RET_CRITICAL);
487 status = lcc_getval (connection, &ident,
488 &values_num, &values, &values_names);
491 printf ("ERROR: Retrieving values from the daemon failed: %s.\n",
492 lcc_strerror (connection));
493 LCC_DESTROY (connection);
494 return (RET_CRITICAL);
497 LCC_DESTROY (connection);
499 status = filter_ds (&values_num, &values, &values_names);
500 if (status != RET_OKAY)
503 status = RET_UNKNOWN;
504 if (consolitation_g == CON_NONE)
505 status = do_check_con_none (values_num, values, values_names);
506 else if (consolitation_g == CON_AVERAGE)
507 status = do_check_con_average (values_num, values, values_names);
508 else if (consolitation_g == CON_SUM)
509 status = do_check_con_sum (values_num, values, values_names);
512 if (values_names != NULL)
513 for (i = 0; i < values_num; i++)
514 free (values_names[i]);
520 int main (int argc, char **argv)
522 range_critical_g.min = NAN;
523 range_critical_g.max = NAN;
524 range_critical_g.invert = 0;
526 range_warning_g.min = NAN;
527 range_warning_g.max = NAN;
528 range_warning_g.invert = 0;
534 c = getopt (argc, argv, "w:c:s:n:H:g:d:h");
541 parse_range (optarg, &range_critical_g);
544 parse_range (optarg, &range_warning_g);
547 socket_file_g = optarg;
550 value_string_g = optarg;
556 if (strcasecmp (optarg, "none") == 0)
557 consolitation_g = CON_NONE;
558 else if (strcasecmp (optarg, "average") == 0)
559 consolitation_g = CON_AVERAGE;
560 else if (strcasecmp (optarg, "sum") == 0)
561 consolitation_g = CON_SUM;
568 tmp = (char **) realloc (match_ds_g,
573 fprintf (stderr, "realloc failed: %s\n",
575 return (RET_UNKNOWN);
578 match_ds_g[match_ds_num_g] = cn_strdup (optarg);
579 if (match_ds_g[match_ds_num_g] == NULL)
581 fprintf (stderr, "cn_strdup failed: %s\n",
583 return (RET_UNKNOWN);
593 if ((socket_file_g == NULL) || (value_string_g == NULL)
594 || (hostname_g == NULL))
597 return (do_check ());