X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=src%2Fping.c;h=2a2f03eb39b80f062b652d4240c8fd3cddcee849;hb=ec6fec62b15992b22774233c6a915e46673b25af;hp=d418232f1498ef5986e53021c6952f1d19cf4a38;hpb=69a36d8615e8f6b07fc24ff4937e5f4a1c60e012;p=collectd.git diff --git a/src/ping.c b/src/ping.c index d418232f..2a2f03eb 100644 --- a/src/ping.c +++ b/src/ping.c @@ -1,6 +1,6 @@ /** * 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 @@ -20,41 +20,157 @@ * Florian octo Forster **/ -#include "ping.h" +#include "collectd.h" +#include "common.h" +#include "plugin.h" +#include "configfile.h" +#include "utils_debug.h" -#if COLLECT_PING #define MODULE_NAME "ping" -#include "plugin.h" -#include "common.h" - #include -#include "libping/ping.h" +#include "liboping/oping.h" + +struct hostlist_s +{ + char *host; + int wait_time; + int wait_left; + struct hostlist_s *next; +}; +typedef struct hostlist_s hostlist_t; -extern char *pinghosts[MAX_PINGHOSTS]; -extern int num_pinghosts; -static int pingerrors[MAX_PINGHOSTS]; +static pingobj_t *pingobj = NULL; +static hostlist_t *hosts = NULL; static char *file_template = "ping-%s.rrd"; static char *ds_def[] = { - "DS:ping:GAUGE:25:0:65535", + "DS:ping:GAUGE:"COLLECTD_HEARTBEAT":0:65535", NULL }; static int ds_num = 1; -void ping_init (void) +static char *config_keys[] = +{ + "Host", + "TTL", + NULL +}; +static int config_keys_num = 2; + +static void add_hosts (void) { - int i; + hostlist_t *hl_this; + hostlist_t *hl_prev; - for (i = 0; i < num_pinghosts; i++) - pingerrors[i] = 0; + int step = atoi (COLLECTD_STEP); - return; + hl_this = hosts; + hl_prev = NULL; + while (hl_this != NULL) + { + DBG ("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) + { + DBG ("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 + { + 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 -= step; + } + + if (hl_this != NULL) + { + hl_prev = hl_this; + hl_this = hl_this->next; + } + } } -void ping_write (char *host, char *inst, char *val) +static void ping_init (void) +{ + if (hosts != NULL) + add_hosts (); +} + +static int ping_config (char *key, char *value) +{ + if (pingobj == NULL) + { + if ((pingobj = ping_construct ()) == NULL) + { + syslog (LOG_ERR, "ping: `ping_construct' failed: %s", + ping_get_error (pingobj)); + return (1); + } + } + + if (strcasecmp (key, "host") == 0) + { + hostlist_t *hl; + char *host; + int step = atoi (COLLECTD_STEP); + + if ((hl = (hostlist_t *) malloc (sizeof (hostlist_t))) == NULL) + { + syslog (LOG_ERR, "ping plugin: malloc failed: %s", + strerror (errno)); + return (1); + } + if ((host = strdup (value)) == NULL) + { + free (hl); + syslog (LOG_ERR, "ping plugin: strdup failed: %s", + strerror (errno)); + return (1); + } + + hl->host = host; + hl->wait_time = 2 * step; + 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)) + { + syslog (LOG_WARNING, "ping: liboping did not accept the TTL value %i", ttl); + return (1); + } + } + else + { + return (-1); + } + + return (0); +} + +static void ping_write (char *host, char *inst, char *val) { char file[512]; int status; @@ -69,67 +185,61 @@ void ping_write (char *host, char *inst, char *val) } #define BUFSIZE 256 -void ping_submit (int ping_time, char *host) +static void ping_submit (char *host, double latency) { char buf[BUFSIZE]; - if (snprintf (buf, BUFSIZE, "%u:%u", (unsigned int) curtime, ping_time) >= BUFSIZE) + if (snprintf (buf, BUFSIZE, "%u:%f", (unsigned int) curtime, latency) >= BUFSIZE) return; plugin_submit (MODULE_NAME, host, buf); } #undef BUFSIZE -void ping_read (void) +static void ping_read (void) { - int ping; - int i; + pingobj_iter_t *iter; + + char host[512]; + double latency; + size_t buf_len; + + if (pingobj == NULL) + return; + + if (hosts != NULL) + add_hosts (); - for (i = 0; i < num_pinghosts; i++) + if (ping_send (pingobj) < 0) { - if (pingerrors[i] & 0x30) + syslog (LOG_ERR, "ping: `ping_send' failed: %s", + ping_get_error (pingobj)); + return; + } + + for (iter = ping_iterator_get (pingobj); + iter != NULL; + iter = ping_iterator_next (iter)) + { + buf_len = sizeof (host); + if (ping_iterator_get_info (iter, PING_INFO_HOSTNAME, + host, &buf_len)) continue; - - ping = tpinghost (pinghosts[i]); - switch (ping) - { - 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++) */ + buf_len = sizeof (latency); + if (ping_iterator_get_info (iter, PING_INFO_LATENCY, + &latency, &buf_len)) + continue; + + DBG ("host = %s, latency = %f", host, latency); + ping_submit (host, latency); + } } 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 -#endif /* COLLECT_PING */