X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fping.c;h=084e45683b116f28b15ebeddc11c52f218631160;hb=9b626288aff81c68fe19210a58b4a32eeefde9d3;hp=91e0dbcc01072fd7601695349f08684adcd53523;hpb=f17fc1f1e4091ab249f88bfc00109c2a90a9bd6a;p=collectd.git diff --git a/src/ping.c b/src/ping.c index 91e0dbcc..084e4568 100644 --- a/src/ping.c +++ b/src/ping.c @@ -1,11 +1,10 @@ /** * collectd - src/ping.c - * Copyright (C) 2005,2006 Florian octo Forster + * Copyright (C) 2005-2007 Florian octo Forster * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License as published by the - * Free Software Foundation; either version 2 of the License, or (at your - * option) any later version. + * Free Software Foundation; only version 2 of the License is applicable. * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of @@ -24,25 +23,29 @@ #include "common.h" #include "plugin.h" #include "configfile.h" -#include "utils_debug.h" - -#define MODULE_NAME "ping" #include -#include "liboping/liboping.h" - -static pingobj_t *pingobj = NULL; - -static char *file_template = "ping-%s.rrd"; +#include "liboping/oping.h" -static char *ds_def[] = +/* + * Private data types + */ +struct hostlist_s { - "DS:ping:GAUGE:"COLLECTD_HEARTBEAT":0:65535", - NULL + char *host; + int wait_time; + int wait_left; + struct hostlist_s *next; }; -static int ds_num = 1; +typedef struct hostlist_s hostlist_t; + +/* + * Private variables + */ +static pingobj_t *pingobj = NULL; +static hostlist_t *hosts = NULL; -static char *config_keys[] = +static const char *config_keys[] = { "Host", "TTL", @@ -50,38 +53,118 @@ static char *config_keys[] = }; static int config_keys_num = 2; -static void ping_init (void) +/* + * Private functions + */ +static void add_hosts (void) { - return; -} + hostlist_t *hl_this; + hostlist_t *hl_prev; + + hl_this = hosts; + hl_prev = NULL; + while (hl_this != NULL) + { + DEBUG ("ping plugin: host = %s, wait_left = %i, " + "wait_time = %i, next = %p", + hl_this->host, hl_this->wait_left, + hl_this->wait_time, (void *) hl_this->next); -static int ping_config (char *key, char *value) + if (hl_this->wait_left <= 0) + { + if (ping_host_add (pingobj, hl_this->host) == 0) + { + DEBUG ("ping plugin: Successfully added host %s", hl_this->host); + /* Remove the host from the linked list */ + if (hl_prev != NULL) + hl_prev->next = hl_this->next; + else + hosts = hl_this->next; + free (hl_this->host); + free (hl_this); + hl_this = (hl_prev != NULL) ? hl_prev : hosts; + } + else + { + WARNING ("ping plugin: Failed adding host " + "`%s': %s", hl_this->host, + ping_get_error (pingobj)); + hl_this->wait_left = hl_this->wait_time; + hl_this->wait_time *= 2; + if (hl_this->wait_time > 86400) + hl_this->wait_time = 86400; + } + } + else + { + hl_this->wait_left -= interval_g; + } + + if (hl_this != NULL) + { + hl_prev = hl_this; + hl_this = hl_this->next; + } + } +} /* void add_hosts */ + +static int ping_init (void) +{ + if (pingobj == NULL) + return (-1); + + if (hosts != NULL) + add_hosts (); + + return (0); +} /* int ping_init */ + +static int ping_config (const char *key, const char *value) { if (pingobj == NULL) { if ((pingobj = ping_construct ()) == NULL) { - syslog (LOG_ERR, "ping: `ping_construct' failed: %s", - ping_get_error (pingobj)); + ERROR ("ping plugin: `ping_construct' failed."); return (1); } } if (strcasecmp (key, "host") == 0) { - if (ping_host_add (pingobj, value) < 0) + hostlist_t *hl; + char *host; + + if ((hl = (hostlist_t *) malloc (sizeof (hostlist_t))) == NULL) { - syslog (LOG_WARNING, "ping: `ping_host_add' failed: %s", - ping_get_error (pingobj)); + char errbuf[1024]; + ERROR ("ping plugin: malloc failed: %s", + sstrerror (errno, errbuf, + sizeof (errbuf))); return (1); } + if ((host = strdup (value)) == NULL) + { + char errbuf[1024]; + free (hl); + ERROR ("ping plugin: strdup failed: %s", + sstrerror (errno, errbuf, + sizeof (errbuf))); + return (1); + } + + hl->host = host; + hl->wait_time = 2 * interval_g; + hl->wait_left = 0; + hl->next = hosts; + hosts = hl; } else if (strcasecmp (key, "ttl") == 0) { int ttl = atoi (value); - if (ping_setopt (pingobj, PING_DEF_TIMEOUT, (void *) &ttl)) + if (ping_setopt (pingobj, PING_OPT_TTL, (void *) &ttl)) { - syslog (LOG_WARNING, "ping: liboping did not accept the TTL value %i", ttl); + WARNING ("ping: liboping did not accept the TTL value %i", ttl); return (1); } } @@ -93,71 +176,86 @@ static int ping_config (char *key, char *value) return (0); } -static void ping_write (char *host, char *inst, char *val) -{ - char file[512]; - int status; - - status = snprintf (file, 512, file_template, inst); - if (status < 1) - return; - else if (status >= 512) - return; - - rrd_update_file (host, file, val, ds_def, ds_num); -} - -#define BUFSIZE 256 static void ping_submit (char *host, double latency) { - char buf[BUFSIZE]; + value_t values[1]; + value_list_t vl = VALUE_LIST_INIT; + + values[0].gauge = latency; - if (snprintf (buf, BUFSIZE, "%u:%f", (unsigned int) curtime, latency) >= BUFSIZE) - return; + vl.values = values; + vl.values_len = 1; + sstrncpy (vl.host, hostname_g, sizeof (vl.host)); + sstrncpy (vl.plugin, "ping", sizeof (vl.plugin)); + sstrncpy (vl.plugin_instance, "", sizeof (vl.plugin_instance)); + sstrncpy (vl.type_instance, host, sizeof (vl.type_instance)); + sstrncpy (vl.type, "ping", sizeof (vl.type)); - plugin_submit (MODULE_NAME, host, buf); + plugin_dispatch_values (&vl); } -#undef BUFSIZE -static void ping_read (void) +static int ping_read (void) { pingobj_iter_t *iter; - char *host; - double latency; + char host[512]; + double latency; + size_t buf_len; + int number_of_hosts; if (pingobj == NULL) - return; + return (-1); + + if (hosts != NULL) + add_hosts (); if (ping_send (pingobj) < 0) { - syslog (LOG_ERR, "ping: `ping_send' failed: %s", + ERROR ("ping plugin: `ping_send' failed: %s", ping_get_error (pingobj)); - return; + return (-1); } - for (iter = ping_iterator_get (pingobj); iter != NULL; iter = ping_iterator_next (iter)) + number_of_hosts = 0; + for (iter = ping_iterator_get (pingobj); + iter != NULL; + iter = ping_iterator_next (iter)) { - const char *tmp; - - if ((tmp = ping_iterator_get_host (iter)) == NULL) - continue; - if ((host = strdup (tmp)) == NULL) + buf_len = sizeof (host); + if (ping_iterator_get_info (iter, PING_INFO_HOSTNAME, + host, &buf_len)) + { + WARNING ("ping plugin: ping_iterator_get_info " + "(PING_INFO_HOSTNAME) failed."); continue; + } - latency = ping_iterator_get_latency (iter); + buf_len = sizeof (latency); + if (ping_iterator_get_info (iter, PING_INFO_LATENCY, + &latency, &buf_len)) + { + WARNING ("ping plugin: ping_iterator_get_info (%s, " + "PING_INFO_LATENCY) failed.", host); + continue; + } - DBG ("host = %s, latency = %f", host, latency); + DEBUG ("ping plugin: host = %s, latency = %f", host, latency); ping_submit (host, latency); + number_of_hosts++; + } - free (host); host = NULL; + if ((number_of_hosts == 0) && (getuid () != 0)) + { + ERROR ("ping plugin: All hosts failed. Try starting collectd as root."); } -} + + return (number_of_hosts == 0 ? -1 : 0); +} /* int ping_read */ void module_register (void) { - plugin_register (MODULE_NAME, ping_init, ping_read, ping_write); - cf_register (MODULE_NAME, ping_config, config_keys, config_keys_num); -} - -#undef MODULE_NAME + plugin_register_config ("ping", ping_config, + config_keys, config_keys_num); + plugin_register_init ("ping", ping_init); + plugin_register_read ("ping", ping_read); +} /* void module_register */