src/pf.c

   1 /*
   2  * Copyright (c) 2010 Pierre-Yves Ritschard <pyr@openbsd.org>
   3  * Copyright (c) 2011 Stefan Rinkes <stefan.rinkes@gmail.org>
   4  *
   5  * Permission to use, copy, modify, and distribute this software for any
   6  * purpose with or without fee is hereby granted, provided that the above
   7  * copyright notice and this permission notice appear in all copies.
   8  *
   9  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  10  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  11  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  12  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  13  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  14  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  15  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  16  */
  17
  18 #include "pfcommon.h"
  19
  20 static int       pf_init(void);
  21 static int       pf_read(void);
  22 static void      submit_counter(const char *, const char *, counter_t, int);
  23
  24 char            *pf_device = "/dev/pf";
  25
  26 int
  27 pf_init(void)
  28 {
  29         struct pf_status        status;
  30         int                     pfdev = -1;
  31
  32         if ((pfdev = open(pf_device, O_RDONLY)) == -1) {
  33                 ERROR("unable to open %s", pf_device);
  34                 return (-1);
  35         }
  36
  37         if (ioctl(pfdev, DIOCGETSTATUS, &status) == -1) {
  38                 ERROR("DIOCGETSTATUS: %i", pfdev);
  39                 close(pfdev);
  40                 return (-1);
  41         }
  42
  43         close(pfdev);
  44         if (!status.running)
  45                 return (-1);
  46
  47         return (0);
  48 }
  49
  50 int
  51 pf_read(void)
  52 {
  53         struct pf_status        status;
  54         int                     pfdev = -1;
  55         int                     i;
  56
  57         char            *cnames[] = PFRES_NAMES;
  58         char            *lnames[] = LCNT_NAMES;
  59         char            *names[] = { "searches", "inserts", "removals" };
  60
  61         if ((pfdev = open(pf_device, O_RDONLY)) == -1) {
  62                 ERROR("unable to open %s", pf_device);
  63                 return (-1);
  64         }
  65
  66         if (ioctl(pfdev, DIOCGETSTATUS, &status) == -1) {
  67                 ERROR("DIOCGETSTATUS: %i", pfdev);
  68                 close(pfdev);
  69                 return (-1);
  70         }
  71
  72         close(pfdev);
  73
  74         for (i = 0; i < PFRES_MAX; i++)
  75                 submit_counter("pf_counters", cnames[i], status.counters[i], 0);
  76         for (i = 0; i < LCNT_MAX; i++)
  77                 submit_counter("pf_limits", lnames[i], status.lcounters[i], 0);
  78         for (i = 0; i < FCNT_MAX; i++)
  79                 submit_counter("pf_state", names[i], status.fcounters[i], 0);
  80         for (i = 0; i < SCNT_MAX; i++)
  81                 submit_counter("pf_source", names[i], status.scounters[i], 0);
  82
  83         submit_counter("pf_states", "current", status.states, 1);
  84
  85         return (0);
  86 }
  87
  88 void
  89 submit_counter(const char *type, const char *inst, counter_t val, int usegauge)
  90 {
  91 #ifndef TEST
  92         value_t         values[1];
  93         value_list_t    vl = VALUE_LIST_INIT;
  94
  95         if (usegauge)
  96                 values[0].gauge = val;
  97         else
  98                 values[0].counter = val;
  99
 100         vl.values = values;
 101         vl.values_len = 1;
 102         sstrncpy (vl.host, hostname_g, sizeof (vl.host));
 103         sstrncpy (vl.plugin, "pf", sizeof (vl.plugin));
 104         sstrncpy (vl.type, type, sizeof(vl.type));
 105         sstrncpy (vl.type_instance, inst, sizeof(vl.type_instance));
 106         plugin_dispatch_values(&vl);
 107 #else
 108         printf("%s.%s: %lld\n", type, inst, val);
 109 #endif
 110 }
 111
 112 #ifdef TEST
 113 int
 114 main(int argc, char *argv[])
 115 {
 116         if (pf_init())
 117                 err(1, "pf_init");
 118         if (pf_read())
 119                 err(1, "pf_read");
 120         return (0);
 121 }
 122 #else
 123 void module_register(void) {
 124         plugin_register_init("pf", pf_init);
 125         plugin_register_read("pf", pf_read);
 126 }
 127 #endif