X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fcollectd-nagios.c;h=15252b19a1ba10fc8d3e52ec384ab9f6a2903be3;hb=2ca211b327b5e1689937a93ac6a4d0b9754d3627;hp=0646c4c1a121902d49d8673bb32ea4c25c0b16fc;hpb=5394b98619328ad1e5a7e8b847f973e8cff6543c;p=collectd.git diff --git a/src/collectd-nagios.c b/src/collectd-nagios.c index 0646c4c1..15252b19 100644 --- a/src/collectd-nagios.c +++ b/src/collectd-nagios.c @@ -99,6 +99,7 @@ #define CON_NONE 0 #define CON_AVERAGE 1 #define CON_SUM 2 +#define CON_PERCENTAGE 3 struct range_s { @@ -138,19 +139,75 @@ static char *cn_strdup (const char *str) /* {{{ */ return (ret); } /* }}} char *cn_strdup */ -static int ignore_ds (const char *name) +static int filter_ds (size_t *values_num, + double **values, char ***values_names) { - int i; + gauge_t *new_values; + char **new_names; + + size_t i; if (match_ds_g == NULL) - return (0); + return (RET_OKAY); + + new_values = (gauge_t *)calloc (match_ds_num_g, sizeof (*new_values)); + if (new_values == NULL) + { + fprintf (stderr, "malloc failed: %s\n", strerror (errno)); + return (RET_UNKNOWN); + } + + new_names = (char **)calloc (match_ds_num_g, sizeof (*new_names)); + if (new_names == NULL) + { + fprintf (stderr, "malloc failed: %s\n", strerror (errno)); + free (new_values); + return (RET_UNKNOWN); + } for (i = 0; i < match_ds_num_g; i++) - if (strcasecmp (match_ds_g[i], name) == 0) - return (0); + { + size_t j; - return (1); -} /* int ignore_ds */ + /* match_ds_g keeps pointers into argv but the names will be freed */ + new_names[i] = cn_strdup (match_ds_g[i]); + if (new_names[i] == NULL) + { + fprintf (stderr, "cn_strdup failed: %s\n", strerror (errno)); + free (new_values); + for (j = 0; j < i; j++) + free (new_names[j]); + free (new_names); + return (RET_UNKNOWN); + } + + for (j = 0; j < *values_num; j++) + if (strcasecmp (new_names[i], (*values_names)[j]) == 0) + break; + + if (j == *values_num) + { + printf ("ERROR: DS `%s' is not available.\n", new_names[i]); + free (new_values); + for (j = 0; j <= i; j++) + free (new_names[j]); + free (new_names); + return (RET_CRITICAL); + } + + new_values[i] = (*values)[j]; + } + + free (*values); + for (i = 0; i < *values_num; i++) + free ((*values_names)[i]); + free (*values_names); + + *values = new_values; + *values_names = new_names; + *values_num = match_ds_num_g; + return (RET_OKAY); +} /* int filter_ds */ static void parse_range (char *string, range_t *range) { @@ -231,7 +288,7 @@ static void usage (const char *name) exit (1); } /* void usage */ -static int do_check_con_none (int values_num, +static int do_check_con_none (size_t values_num, double *values, char **values_names) { int num_critical = 0; @@ -243,9 +300,6 @@ static int do_check_con_none (int values_num, for (i = 0; i < values_num; i++) { - if (ignore_ds (values_names[i])) - continue; - if (isnan (values[i])) num_warning++; else if (match_range (&range_critical_g, values[i]) != 0) @@ -290,7 +344,7 @@ static int do_check_con_none (int values_num, return (status_code); } /* int do_check_con_none */ -static int do_check_con_average (int values_num, +static int do_check_con_average (size_t values_num, double *values, char **values_names) { int i; @@ -304,9 +358,6 @@ static int do_check_con_average (int values_num, total_num = 0; for (i = 0; i < values_num; i++) { - if (ignore_ds (values_names[i])) - continue; - if (!isnan (values[i])) { total += values[i]; @@ -346,7 +397,8 @@ static int do_check_con_average (int values_num, return (status_code); } /* int do_check_con_average */ -static int do_check_con_sum (int values_num, double *values, char **values_names) +static int do_check_con_sum (size_t values_num, + double *values, char **values_names) { int i; double total; @@ -358,9 +410,6 @@ static int do_check_con_sum (int values_num, double *values, char **values_names total_num = 0; for (i = 0; i < values_num; i++) { - if (ignore_ds (values_names[i])) - continue; - if (!isnan (values[i])) { total += values[i]; @@ -398,6 +447,56 @@ static int do_check_con_sum (int values_num, double *values, char **values_names return (status_code); } /* int do_check_con_sum */ +static int do_check_con_percentage (size_t values_num, + double *values, char **values_names) +{ + int i; + double sum = 0.0; + double percentage; + + const char *status_str = "UNKNOWN"; + int status_code = RET_UNKNOWN; + + if ((values_num < 1) || (isnan (values[0]))) + { + printf ("WARNING: The first value is not defined\n"); + return (RET_WARNING); + } + + for (i = 0; i < values_num; i++) + if (!isnan (values[i])) + sum += values[i]; + + if (sum == 0.0) + { + printf ("WARNING: Values sum up to zero\n"); + return (RET_WARNING); + } + + percentage = 100.0 * values[0] / sum; + + if (match_range (&range_critical_g, percentage) != 0) + { + status_str = "CRITICAL"; + status_code = RET_CRITICAL; + } + else if (match_range (&range_warning_g, percentage) != 0) + { + status_str = "WARNING"; + status_code = RET_WARNING; + } + else + { + status_str = "OKAY"; + status_code = RET_OKAY; + } + + printf ("%s: %lf percent |", status_str, percentage); + for (i = 0; i < values_num; i++) + printf (" %s=%lf;;;;", values_names[i], values[i]); + return (status_code); +} /* int do_check_con_percentage */ + static int do_check (void) { lcc_connection_t *connection; @@ -448,6 +547,10 @@ static int do_check (void) LCC_DESTROY (connection); + status = filter_ds (&values_num, &values, &values_names); + if (status != RET_OKAY) + return (status); + status = RET_UNKNOWN; if (consolitation_g == CON_NONE) status = do_check_con_none (values_num, values, values_names); @@ -455,6 +558,8 @@ static int do_check (void) status = do_check_con_average (values_num, values, values_names); else if (consolitation_g == CON_SUM) status = do_check_con_sum (values_num, values, values_names); + else if (consolitation_g == CON_PERCENTAGE) + status = do_check_con_percentage (values_num, values, values_names); free (values); if (values_names != NULL) @@ -507,6 +612,8 @@ int main (int argc, char **argv) consolitation_g = CON_AVERAGE; else if (strcasecmp (optarg, "sum") == 0) consolitation_g = CON_SUM; + else if (strcasecmp (optarg, "percentage") == 0) + consolitation_g = CON_PERCENTAGE; else usage (argv[0]); break;