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 verplant.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
  34 static void conntrack_submit (value_t conntrack)
  35 {
  36         value_list_t vl = VALUE_LIST_INIT;
  37
  38         vl.values = &conntrack;
  39         vl.values_len = 1;
  40         sstrncpy (vl.host, hostname_g, sizeof (vl.host));
  41         sstrncpy (vl.plugin, "conntrack", sizeof (vl.plugin));
  42         sstrncpy (vl.type, "conntrack", sizeof (vl.type));
  43
  44         plugin_dispatch_values (&vl);
  45 } /* static void conntrack_submit */
  46
  47 static int conntrack_read (void)
  48 {
  49         value_t conntrack;
  50         FILE *fh;
  51         char buffer[64];
  52         size_t buffer_len;
  53
  54         fh = fopen (CONNTRACK_FILE, "r");
  55         if (fh == NULL)
  56                 return (-1);
  57
  58         memset (buffer, 0, sizeof (buffer));
  59         if (fgets (buffer, sizeof (buffer), fh) == NULL)
  60         {
  61                 fclose (fh);
  62                 return (-1);
  63         }
  64         fclose (fh);
  65
  66         /* strip trailing newline. */
  67         buffer_len = strlen (buffer);
  68         while ((buffer_len > 0) && isspace ((int) buffer[buffer_len - 1]))
  69         {
  70                 buffer[buffer_len - 1] = 0;
  71                 buffer_len--;
  72         }
  73
  74         if (parse_value (buffer, &conntrack, DS_TYPE_GAUGE) != 0)
  75                 return (-1);
  76
  77         conntrack_submit (conntrack);
  78
  79         return (0);
  80 } /* static int conntrack_read */
  81
  82 void module_register (void)
  83 {
  84         plugin_register_read ("conntrack", conntrack_read);
  85 } /* void module_register */