src/conntrack.c

   1 /**
   2  * collectd - src/conntrack.c
   3  * Copyright (C) 2009  Tomasz Pala
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License as published by the
   7  * Free Software Foundation; only version 2 of the License is applicable.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  17  *
  18  * Authors:
  19  *   Tomasz Pala <gotar at pld-linux.org>
  20  * based on entropy.c by:
  21  *   Florian octo Forster <octo at collectd.org>
  22  **/
  23
  24 #include "collectd.h"
  25 #include "common.h"
  26 #include "plugin.h"
  27
  28 #if !KERNEL_LINUX
  29 # error "No applicable input method."
  30 #endif
  31
  32 #define CONNTRACK_FILE "/proc/sys/net/netfilter/nf_conntrack_count"
  33 #define CONNTRACK_MAX_FILE "/proc/sys/net/netfilter/nf_conntrack_max"
  34 #define CONNTRACK_FILE_OLD "/proc/sys/net/ipv4/netfilter/ip_conntrack_count"
  35 #define CONNTRACK_MAX_FILE_OLD "/proc/sys/net/ipv4/netfilter/ip_conntrack_max"
  36
  37 static const char *config_keys[] =
  38 {
  39         "OldFiles"
  40 };
  41 static int config_keys_num = STATIC_ARRAY_SIZE (config_keys);
  42 /*
  43     Each table/chain combo that will be queried goes into this list
  44 */
  45
  46 static int old_files = 0;
  47
  48 static int conntrack_config(const char *key, const char *value)
  49 {
  50     if (strcmp(key, "OldFiles") == 0)
  51         old_files = 1;
  52
  53     return 0;
  54 }
  55
  56 static void conntrack_submit (const char *type, const char *type_instance,
  57                               value_t conntrack)
  58 {
  59         value_list_t vl = VALUE_LIST_INIT;
  60
  61         vl.values = &conntrack;
  62         vl.values_len = 1;
  63         sstrncpy (vl.host, hostname_g, sizeof (vl.host));
  64         sstrncpy (vl.plugin, "conntrack", sizeof (vl.plugin));
  65         sstrncpy (vl.type, type, sizeof (vl.type));
  66         if (type_instance != NULL)
  67                 sstrncpy (vl.type_instance, type_instance,
  68                           sizeof (vl.type_instance));
  69
  70         plugin_dispatch_values (&vl);
  71 } /* static void conntrack_submit */
  72
  73 static int conntrack_read (void)
  74 {
  75         value_t conntrack, conntrack_max, conntrack_pct;
  76         FILE *fh;
  77         char buffer[64];
  78         size_t buffer_len;
  79
  80         fh = fopen (old_files?CONNTRACK_FILE_OLD:CONNTRACK_FILE, "r");
  81         if (fh == NULL)
  82                 return (-1);
  83
  84         memset (buffer, 0, sizeof (buffer));
  85         if (fgets (buffer, sizeof (buffer), fh) == NULL)
  86         {
  87                 fclose (fh);
  88                 return (-1);
  89         }
  90         fclose (fh);
  91
  92         /* strip trailing newline. */
  93         buffer_len = strlen (buffer);
  94         while ((buffer_len > 0) && isspace ((int) buffer[buffer_len - 1]))
  95         {
  96                 buffer[buffer_len - 1] = 0;
  97                 buffer_len--;
  98         }
  99
 100         if (parse_value (buffer, &conntrack, DS_TYPE_GAUGE) != 0)
 101                 return (-1);
 102
 103         conntrack_submit ("conntrack", NULL, conntrack);
 104
 105         fh = fopen (old_files?CONNTRACK_MAX_FILE_OLD:CONNTRACK_MAX_FILE, "r");
 106         if (fh == NULL)
 107                 return (-1);
 108
 109         memset (buffer, 0, sizeof (buffer));
 110         if (fgets (buffer, sizeof (buffer), fh) == NULL)
 111         {
 112                 fclose (fh);
 113                 return (-1);
 114         }
 115         fclose (fh);
 116
 117         /* strip trailing newline. */
 118         buffer_len = strlen (buffer);
 119         while ((buffer_len > 0) && isspace ((int) buffer[buffer_len - 1]))
 120         {
 121                 buffer[buffer_len - 1] = 0;
 122                 buffer_len--;
 123         }
 124
 125         if (parse_value (buffer, &conntrack_max, DS_TYPE_GAUGE) != 0)
 126                 return (-1);
 127
 128         conntrack_submit ("conntrack", "max", conntrack_max);
 129         conntrack_pct.gauge = (conntrack.gauge / conntrack_max.gauge) * 100;
 130         conntrack_submit ("percent", "used", conntrack_pct);
 131
 132
 133         return (0);
 134 } /* static int conntrack_read */
 135
 136 void module_register (void)
 137 {
 138     plugin_register_config ("conntrack", conntrack_config,
 139                             config_keys, config_keys_num);
 140         plugin_register_read ("conntrack", conntrack_read);
 141 } /* void module_register */