Added a config option to the traffic plugin to ignore certain interfaces. Thanks...

diff --git a/src/traffic.c b/src/traffic.c
index 07c9ecf..d6ab873 100644 (file)
--- a/src/traffic.c
+++ b/src/traffic.c
@@ -18,11 +18,13 @@
  *
  * Authors:
  *   Florian octo Forster <octo at verplant.org>
+ *   Sune Marcher <sm at flork.dk>
  **/
 
 #include "collectd.h"
 #include "common.h"
 #include "plugin.h"
+#include "configfile.h"
 
 #if HAVE_SYS_TYPES_H
 #  include <sys/types.h>
@@ -55,12 +57,22 @@
 
 #define BUFSIZE 512
 
+/*
+ * (Module-)Global variables
+ */
 /* TODO: Move this to `interface-%s/<blah>.rrd' in version 4. */
 static char *bytes_file   = "traffic-%s.rrd";
 static char *packets_file = "if_packets-%s.rrd";
 static char *errors_file  = "if_errors-%s.rrd";
 /* TODO: Maybe implement multicast and broadcast counters */
 
+static char *config_keys[] =
+{
+       "Ignore",
+       NULL
+};
+static int config_keys_num = 1;
+
 static char *bytes_ds_def[] =
 {
        "DS:incoming:COUNTER:"COLLECTD_HEARTBEAT":0:U",
@@ -85,6 +97,9 @@ static char *errors_ds_def[] =
 };
 static int errors_ds_num = 2;
 
+static char **if_ignore_list = NULL;
+static int    if_ignore_list_num = 0;
+
 #ifdef HAVE_LIBKSTAT
 #define MAX_NUMIF 256
 extern kstat_ctl_t *kc;
@@ -92,6 +107,33 @@ static kstat_t *ksp[MAX_NUMIF];
 static int numif = 0;
 #endif /* HAVE_LIBKSTAT */
 
+static int traffic_config (char *key, char *value)
+{
+       char **temp;
+
+       if (strcasecmp (key, "Ignore") != 0)
+               return (-1);
+
+       temp = (char **) realloc (if_ignore_list, (if_ignore_list_num + 1) * sizeof (char *));
+       if (temp == NULL)
+       {
+               syslog (LOG_EMERG, "Cannot allocate more memory.");
+               return (1);
+       }
+       if_ignore_list = temp;
+
+       if ((if_ignore_list[if_ignore_list_num] = strdup (value)) == NULL)
+       {
+               syslog (LOG_EMERG, "Cannot allocate memory.");
+               return (1);
+       }
+       if_ignore_list_num++;
+
+       syslog (LOG_NOTICE, "traffic: Ignoring interface `%s'", value);
+
+       return (0);
+}
+
 static void traffic_init (void)
 {
 #if HAVE_GETIFADDRS
@@ -134,6 +176,21 @@ static void traffic_init (void)
        return;
 }
 
+/*
+ * Check if this interface/instance should be ignored. This is called from
+ * both, `submit' and `write' to give client and server the ability to ignore
+ * certain stuff..
+ */
+static int check_ignore_if (const char *interface)
+{
+       int i;
+
+       for (i = 0; i < if_ignore_list_num; i++)
+               if (strcasecmp (interface, if_ignore_list[i]) == 0)
+                       return (1);
+       return (0);
+}
+
 static void generic_write (char *host, char *inst, char *val,
                char *file_template,
                char **ds_def, int ds_num)
@@ -141,6 +198,9 @@ static void generic_write (char *host, char *inst, char *val,
        char file[512];
        int status;
 
+       if (check_ignore_if (inst))
+               return;
+
        status = snprintf (file, BUFSIZE, file_template, inst);
        if (status < 1)
                return;
@@ -166,16 +226,23 @@ static void errors_write (char *host, char *inst, char *val)
 }
 
 #if TRAFFIC_HAVE_READ
-static void bytes_submit (char *device,
-               unsigned long long incoming,
-               unsigned long long outgoing)
+static void bytes_submit (char *dev,
+               unsigned long long rx,
+               unsigned long long tx)
 {
-       char buf[BUFSIZE];
+       char buf[512];
+       int  status;
+
+       if (check_ignore_if (dev))
+               return;
 
-       if (snprintf (buf, BUFSIZE, "%u:%lld:%lld", (unsigned int) curtime, incoming, outgoing) >= BUFSIZE)
+       status = snprintf (buf, 512, "%u:%lld:%lld",
+                               (unsigned int) curtime,
+                               rx, tx);
+       if ((status >= 512) || (status < 1))
                return;
 
-       plugin_submit (MODULE_NAME, device, buf);
+       plugin_submit (MODULE_NAME, dev, buf);
 }
 
 #if HAVE_GETIFADDRS || HAVE_LIBKSTAT
@@ -186,6 +253,9 @@ static void packets_submit (char *dev,
        char buf[512];
        int  status;
 
+       if (check_ignore_if (dev))
+               return;
+
        status = snprintf (buf, 512, "%u:%lld:%lld",
                        (unsigned int) curtime,
                        rx, tx);
@@ -201,6 +271,9 @@ static void errors_submit (char *dev,
        char buf[512];
        int  status;
 
+       if (check_ignore_if (dev))
+               return;
+
        status = snprintf (buf, 512, "%u:%lld:%lld",
                        (unsigned int) curtime,
                        rx, tx);
@@ -356,6 +429,7 @@ void module_register (void)
        plugin_register (MODULE_NAME, traffic_init, traffic_read, bytes_write);
        plugin_register ("if_packets", NULL, NULL, packets_write);
        plugin_register ("if_errors",  NULL, NULL, errors_write);
+       cf_register (MODULE_NAME, traffic_config, config_keys, config_keys_num);
 }
 
 #undef BUFSIZE