X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fping.c;h=27a65faafc36f23a14118ac61a04353acc48704f;hb=b5c5890955fa19651ad8b3f48d99364d270a0d8f;hp=d418232f1498ef5986e53021c6952f1d19cf4a38;hpb=69a36d8615e8f6b07fc24ff4937e5f4a1c60e012;p=collectd.git diff --git a/src/ping.c b/src/ping.c index d418232f..27a65faa 100644 --- a/src/ping.c +++ b/src/ping.c @@ -1,11 +1,10 @@ /** * collectd - src/ping.c - * Copyright (C) 2005 Florian octo Forster + * Copyright (C) 2005,2006 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 @@ -20,116 +19,240 @@ * Florian octo Forster **/ -#include "ping.h" - -#if COLLECT_PING -#define MODULE_NAME "ping" - -#include "plugin.h" +#include "collectd.h" #include "common.h" +#include "plugin.h" +#include "configfile.h" #include -#include "libping/ping.h" +#include "liboping/oping.h" -extern char *pinghosts[MAX_PINGHOSTS]; -extern int num_pinghosts; -static int pingerrors[MAX_PINGHOSTS]; +/* + * Private data types + */ +struct hostlist_s +{ + char *host; + int wait_time; + int wait_left; + struct hostlist_s *next; +}; +typedef struct hostlist_s hostlist_t; -static char *file_template = "ping-%s.rrd"; +/* + * Private variables + */ +static pingobj_t *pingobj = NULL; +static hostlist_t *hosts = NULL; -static char *ds_def[] = +static const char *config_keys[] = { - "DS:ping:GAUGE:25:0:65535", + "Host", + "TTL", NULL }; -static int ds_num = 1; +static int config_keys_num = 2; -void ping_init (void) +/* + * Private functions + */ +static void add_hosts (void) { - int i; + 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); + + 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; + } - for (i = 0; i < num_pinghosts; i++) - pingerrors[i] = 0; + if (hl_this != NULL) + { + hl_prev = hl_this; + hl_this = hl_this->next; + } + } +} + +static int ping_init (void) +{ + if (hosts != NULL) + add_hosts (); - return; + return (0); } -void ping_write (char *host, char *inst, char *val) +static int ping_config (const char *key, const char *value) { - char file[512]; - int status; + if (pingobj == NULL) + { + if ((pingobj = ping_construct ()) == NULL) + { + ERROR ("ping plugin: `ping_construct' failed."); + return (1); + } + } + + if (strcasecmp (key, "host") == 0) + { + hostlist_t *hl; + char *host; + + if ((hl = (hostlist_t *) malloc (sizeof (hostlist_t))) == NULL) + { + 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); + } - status = snprintf (file, 512, file_template, inst); - if (status < 1) - return; - else if (status >= 512) - return; + 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)) + { + WARNING ("ping: liboping did not accept the TTL value %i", ttl); + return (1); + } + } + else + { + return (-1); + } - rrd_update_file (host, file, val, ds_def, ds_num); + return (0); } -#define BUFSIZE 256 -void ping_submit (int ping_time, char *host) +static void ping_submit (char *host, double latency) { - char buf[BUFSIZE]; + value_t values[1]; + value_list_t vl = VALUE_LIST_INIT; - if (snprintf (buf, BUFSIZE, "%u:%u", (unsigned int) curtime, ping_time) >= BUFSIZE) - return; + values[0].gauge = latency; - plugin_submit (MODULE_NAME, host, buf); + vl.values = values; + vl.values_len = 1; + vl.time = time (NULL); + strcpy (vl.host, hostname_g); + strcpy (vl.plugin, "ping"); + strcpy (vl.plugin_instance, ""); + strncpy (vl.type_instance, host, sizeof (vl.type_instance)); + + plugin_dispatch_values ("ping", &vl); } -#undef BUFSIZE -void ping_read (void) +static int ping_read (void) { - int ping; - int i; + pingobj_iter_t *iter; + + char host[512]; + double latency; + size_t buf_len; + int number_of_hosts; + + if (pingobj == NULL) + return (-1); + + if (hosts != NULL) + add_hosts (); + + if (ping_send (pingobj) < 0) + { + ERROR ("ping plugin: `ping_send' failed: %s", + ping_get_error (pingobj)); + return (-1); + } - for (i = 0; i < num_pinghosts; i++) + number_of_hosts = 0; + for (iter = ping_iterator_get (pingobj); + iter != NULL; + iter = ping_iterator_next (iter)) { - if (pingerrors[i] & 0x30) + 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; - - ping = tpinghost (pinghosts[i]); + } - switch (ping) + buf_len = sizeof (latency); + if (ping_iterator_get_info (iter, PING_INFO_LATENCY, + &latency, &buf_len)) { - case 0: - if (!(pingerrors[i] & 0x01)) - syslog (LOG_WARNING, "ping %s: Connection timed out.", pinghosts[i]); - pingerrors[i] |= 0x01; - break; - - case -1: - if (!(pingerrors[i] & 0x02)) - syslog (LOG_WARNING, "ping %s: Host or service is not reachable.", pinghosts[i]); - pingerrors[i] |= 0x02; - break; - - case -2: - syslog (LOG_ERR, "ping %s: Socket error. Ping will be disabled.", pinghosts[i]); - pingerrors[i] |= 0x10; - break; - - case -3: - if (!(pingerrors[i] & 0x04)) - syslog (LOG_WARNING, "ping %s: Connection refused.", pinghosts[i]); - pingerrors[i] |= 0x04; - break; - - default: - if (pingerrors[i] != 0x00) - syslog (LOG_NOTICE, "ping %s: Back to normal: %ims.", pinghosts[i], ping); - pingerrors[i] = 0x00; - ping_submit (ping, pinghosts[i]); - } /* switch (ping) */ - } /* for (i = 0; i < num_pinghosts; i++) */ -} + WARNING ("ping plugin: ping_iterator_get_info (%s, " + "PING_INFO_LATENCY) failed.", host); + continue; + } + + DEBUG ("ping plugin: host = %s, latency = %f", host, latency); + ping_submit (host, latency); + number_of_hosts++; + } + + 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); -} - -#undef MODULE_NAME -#endif /* COLLECT_PING */ + 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 */