src/madwifi.c

   1 /**
   2  * collectd - src/madwifi.c
   3  * Copyright (C) 2009  Ondrej 'SanTiago' Zajicek
   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  * Author:
  19  *   Ondrej 'SanTiago' Zajicek <santiago@crfreenet.org>
  20  *
  21  *   based on some code from interfaces.c (collectd) and Madwifi driver
  22  **/
  23
  24 /**
  25  * There are several data streams provided by Madwifi plugin, some are
  26  * connected to network interface, some are connected to each node
  27  * associated to that interface. Nodes represents other sides in
  28  * wireless communication, for example on network interface in AP mode,
  29  * there is one node for each associated station. Node data streams
  30  * contain MAC address of the node as the last part  of the type_instance
  31  * field.
  32  *
  33  * Inteface data streams:
  34  *      ath_nodes       The number of associated nodes
  35  *      ath_stat        Device statistic counters
  36  *
  37  * Node data streams:
  38  *      node_octets     RX and TX data count (octets/bytes)
  39  *      node_rssi       Received RSSI of the node
  40  *      node_tx_rate    Reported TX rate to that node
  41  *      node_stat       Node statistic counters
  42  *
  43  * Both statistic counters have type instances for each counter returned
  44  * by Madwifi. See madwifi.h for content of ieee80211_nodestats,
  45  * ieee80211_stats and ath_stats structures. Type instances use the same
  46  * name as fields in these structures (like ns_rx_dup). Some fields are
  47  * not reported, because they are not counters (like ns_tx_deauth_code
  48  * or ast_tx_rssi). Fields ns_rx_bytes and ns_tx_bytes are reported as
  49  * node_octets data stream instead of type instance of node_stat.
  50  * Statistics are not logged when they are zero.
  51  *
  52  * There are two sets of these counters - the first 'WatchList' is a
  53  * set of counters that are individually logged. The second 'MiscList'
  54  * is a set of counters that are summed together and the sum is logged.
  55  * By default, the most important statistics are in the WatchList and
  56  * many error statistics are in MiscList. There are also many statistics
  57  * that are not in any of these sets, so they are not monitored by default.
  58  * It is possible to alter these lists using configuration options:
  59  *
  60  *      WatchAdd X      Adds X to WachList
  61  *      WatchRemove X   Removes X from WachList
  62  *      WatchSet All    Adds all statistics to WatchList
  63  *      WatchSet None   Removes all statistics from WachList
  64  *
  65  * There are also Misc* variants fo these options, they modifies MiscList
  66  * instead of WatchList.
  67  *
  68  * Example:
  69  *
  70  *      WatchSet None
  71  *      WatchAdd node_octets
  72  *      WatchAdd node_rssi
  73  *      WatchAdd is_rx_acl
  74  *      WatchAdd is_scan_active
  75  *
  76  * That causes that just the four mentioned data streams are logged.
  77  *
  78  *
  79  * By default, madwifi plugin enumerates network interfaces using /sys
  80  * filesystem. Configuration option `Source' can change this to use
  81  * /proc filesystem (which is useful for example when running on Linux
  82  * 2.4). But without /sys filesystem, Madwifi plugin cannot check whether
  83  * given interface is madwifi interface and there are private ioctls used,
  84  * which may do something completely different on non-madwifi devices.
  85  * Therefore, the /proc filesystem should always be used together with option
  86  * `Interface', to limit found interfaces to madwifi interfaces only.
  87  **/
  88
  89 #include "collectd.h"
  90
  91 #include "plugin.h"
  92 #include "utils/common/common.h"
  93 #include "utils/ignorelist/ignorelist.h"
  94
  95 #include <dirent.h>
  96 #include <sys/ioctl.h>
  97
  98 #if !KERNEL_LINUX
  99 #error "No applicable input method."
 100 #endif
 101
 102 #include "madwifi.h"
 103 #include <linux/wireless.h>
 104
 105 struct stat_spec {
 106   uint16_t flags;
 107   uint16_t offset;
 108   const char *name;
 109 };
 110
 111 #define OFFSETOF(s, i) ((size_t) & ((s *)0)->i)
 112
 113 #define FLAG(i) (((uint32_t)1) << ((i) % 32))
 114
 115 #define SPC_STAT 0
 116 #define NOD_STAT 1
 117 #define IFA_STAT 2
 118 #define ATH_STAT 3
 119 #define SRC_MASK 3
 120
 121 /* By default, the item is disabled */
 122 #define D 0
 123
 124 /* By default, the item is logged */
 125 #define LOG 4
 126
 127 /* By default, the item is summed with other such items and logged together */
 128 #define SU 8
 129
 130 #define SS_STAT(flags, name)                                                   \
 131   { flags | SPC_STAT, 0, #name }
 132 #define NS_STAT(flags, name)                                                   \
 133   { flags | NOD_STAT, OFFSETOF(struct ieee80211_nodestats, name), #name }
 134 #define IS_STAT(flags, name)                                                   \
 135   { flags | IFA_STAT, OFFSETOF(struct ieee80211_stats, name), #name }
 136 #define AS_STAT(flags, name)                                                   \
 137   { flags | ATH_STAT, OFFSETOF(struct ath_stats, name), #name }
 138
 139 /*
 140  * (Module-)Global variables
 141  */
 142
 143 /* Indices of special stats in specs array */
 144 #define STAT_NODE_OCTETS 0
 145 #define STAT_NODE_RSSI 1
 146 #define STAT_NODE_TX_RATE 2
 147 #define STAT_ATH_NODES 3
 148 #define STAT_NS_RX_BEACONS 4
 149 #define STAT_AST_ANT_RX 5
 150 #define STAT_AST_ANT_TX 6
 151
 152 static struct stat_spec specs[] = {
 153
 154     /* Special statistics */
 155     SS_STAT(LOG, node_octets),  /* rx and tx data count (bytes) */
 156     SS_STAT(LOG, node_rssi),    /* received RSSI of the node */
 157     SS_STAT(LOG, node_tx_rate), /* used tx rate to the node */
 158     SS_STAT(LOG, ath_nodes),    /* the number of associated nodes */
 159     SS_STAT(D, ns_rx_beacons),  /* rx beacon frames */
 160     SS_STAT(LOG, ast_ant_rx),   /* rx frames with antenna */
 161     SS_STAT(LOG, ast_ant_tx),   /* tx frames with antenna */
 162
 163     /* Node statistics */
 164     NS_STAT(LOG, ns_rx_data),        /* rx data frames */
 165     NS_STAT(LOG, ns_rx_mgmt),        /* rx management frames */
 166     NS_STAT(LOG, ns_rx_ctrl),        /* rx control frames */
 167     NS_STAT(D, ns_rx_ucast),         /* rx unicast frames */
 168     NS_STAT(D, ns_rx_mcast),         /* rx multi/broadcast frames */
 169     NS_STAT(D, ns_rx_proberesp),     /* rx probe response frames */
 170     NS_STAT(LOG, ns_rx_dup),         /* rx discard because it's a dup */
 171     NS_STAT(SU, ns_rx_noprivacy),    /* rx w/ wep but privacy off */
 172     NS_STAT(SU, ns_rx_wepfail),      /* rx wep processing failed */
 173     NS_STAT(SU, ns_rx_demicfail),    /* rx demic failed */
 174     NS_STAT(SU, ns_rx_decap),        /* rx decapsulation failed */
 175     NS_STAT(SU, ns_rx_defrag),       /* rx defragmentation failed */
 176     NS_STAT(D, ns_rx_disassoc),      /* rx disassociation */
 177     NS_STAT(D, ns_rx_deauth),        /* rx deauthentication */
 178     NS_STAT(SU, ns_rx_decryptcrc),   /* rx decrypt failed on crc */
 179     NS_STAT(SU, ns_rx_unauth),       /* rx on unauthorized port */
 180     NS_STAT(SU, ns_rx_unencrypted),  /* rx unecrypted w/ privacy */
 181     NS_STAT(LOG, ns_tx_data),        /* tx data frames */
 182     NS_STAT(LOG, ns_tx_mgmt),        /* tx management frames */
 183     NS_STAT(D, ns_tx_ucast),         /* tx unicast frames */
 184     NS_STAT(D, ns_tx_mcast),         /* tx multi/broadcast frames */
 185     NS_STAT(D, ns_tx_probereq),      /* tx probe request frames */
 186     NS_STAT(D, ns_tx_uapsd),         /* tx on uapsd queue */
 187     NS_STAT(SU, ns_tx_novlantag),    /* tx discard due to no tag */
 188     NS_STAT(SU, ns_tx_vlanmismatch), /* tx discard due to of bad tag */
 189     NS_STAT(D, ns_tx_eosplost),      /* uapsd EOSP retried out */
 190     NS_STAT(D, ns_ps_discard),       /* ps discard due to of age */
 191     NS_STAT(D, ns_uapsd_triggers),   /* uapsd triggers */
 192     NS_STAT(LOG, ns_tx_assoc),       /* [re]associations */
 193     NS_STAT(LOG, ns_tx_auth),        /* [re]authentications */
 194     NS_STAT(D, ns_tx_deauth),        /* deauthentications */
 195     NS_STAT(D, ns_tx_disassoc),      /* disassociations */
 196     NS_STAT(D, ns_psq_drops),        /* power save queue drops */
 197
 198     /* Iface statistics */
 199     IS_STAT(SU, is_rx_badversion),       /* rx frame with bad version */
 200     IS_STAT(SU, is_rx_tooshort),         /* rx frame too short */
 201     IS_STAT(LOG, is_rx_wrongbss),        /* rx from wrong bssid */
 202     IS_STAT(LOG, is_rx_dup),             /* rx discard due to it's a dup */
 203     IS_STAT(SU, is_rx_wrongdir),         /* rx w/ wrong direction */
 204     IS_STAT(D, is_rx_mcastecho),         /* rx discard due to of mcast echo */
 205     IS_STAT(SU, is_rx_notassoc),         /* rx discard due to sta !assoc */
 206     IS_STAT(SU, is_rx_noprivacy),        /* rx w/ wep but privacy off */
 207     IS_STAT(SU, is_rx_unencrypted),      /* rx w/o wep and privacy on */
 208     IS_STAT(SU, is_rx_wepfail),          /* rx wep processing failed */
 209     IS_STAT(SU, is_rx_decap),            /* rx decapsulation failed */
 210     IS_STAT(D, is_rx_mgtdiscard),        /* rx discard mgt frames */
 211     IS_STAT(D, is_rx_ctl),               /* rx discard ctrl frames */
 212     IS_STAT(D, is_rx_beacon),            /* rx beacon frames */
 213     IS_STAT(D, is_rx_rstoobig),          /* rx rate set truncated */
 214     IS_STAT(SU, is_rx_elem_missing),     /* rx required element missing*/
 215     IS_STAT(SU, is_rx_elem_toobig),      /* rx element too big */
 216     IS_STAT(SU, is_rx_elem_toosmall),    /* rx element too small */
 217     IS_STAT(LOG, is_rx_elem_unknown),    /* rx element unknown */
 218     IS_STAT(SU, is_rx_badchan),          /* rx frame w/ invalid chan */
 219     IS_STAT(SU, is_rx_chanmismatch),     /* rx frame chan mismatch */
 220     IS_STAT(SU, is_rx_nodealloc),        /* rx frame dropped */
 221     IS_STAT(LOG, is_rx_ssidmismatch),    /* rx frame ssid mismatch  */
 222     IS_STAT(SU, is_rx_auth_unsupported), /* rx w/ unsupported auth alg */
 223     IS_STAT(SU, is_rx_auth_fail),        /* rx sta auth failure */
 224     IS_STAT(SU, is_rx_auth_countermeasures), /* rx auth discard due to CM */
 225     IS_STAT(SU, is_rx_assoc_bss),            /* rx assoc from wrong bssid */
 226     IS_STAT(SU, is_rx_assoc_notauth),        /* rx assoc w/o auth */
 227     IS_STAT(SU, is_rx_assoc_capmismatch),    /* rx assoc w/ cap mismatch */
 228     IS_STAT(SU, is_rx_assoc_norate),         /* rx assoc w/ no rate match */
 229     IS_STAT(SU, is_rx_assoc_badwpaie),       /* rx assoc w/ bad WPA IE */
 230     IS_STAT(LOG, is_rx_deauth),              /* rx deauthentication */
 231     IS_STAT(LOG, is_rx_disassoc),            /* rx disassociation */
 232     IS_STAT(SU, is_rx_badsubtype),           /* rx frame w/ unknown subtype*/
 233     IS_STAT(SU, is_rx_nobuf),                /* rx failed for lack of buf */
 234     IS_STAT(SU, is_rx_decryptcrc),           /* rx decrypt failed on crc */
 235     IS_STAT(D, is_rx_ahdemo_mgt),            /* rx discard ahdemo mgt frame*/
 236     IS_STAT(SU, is_rx_bad_auth),             /* rx bad auth request */
 237     IS_STAT(SU, is_rx_unauth),               /* rx on unauthorized port */
 238     IS_STAT(SU, is_rx_badkeyid),             /* rx w/ incorrect keyid */
 239     IS_STAT(D, is_rx_ccmpreplay),            /* rx seq# violation (CCMP), */
 240     IS_STAT(D, is_rx_ccmpformat),            /* rx format bad (CCMP), */
 241     IS_STAT(D, is_rx_ccmpmic),               /* rx MIC check failed (CCMP), */
 242     IS_STAT(D, is_rx_tkipreplay),            /* rx seq# violation (TKIP), */
 243     IS_STAT(D, is_rx_tkipformat),            /* rx format bad (TKIP), */
 244     IS_STAT(D, is_rx_tkipmic),               /* rx MIC check failed (TKIP), */
 245     IS_STAT(D, is_rx_tkipicv),               /* rx ICV check failed (TKIP), */
 246     IS_STAT(D, is_rx_badcipher),             /* rx failed due to of key type */
 247     IS_STAT(D, is_rx_nocipherctx),           /* rx failed due to key !setup */
 248     IS_STAT(D, is_rx_acl),               /* rx discard due to of acl policy */
 249     IS_STAT(D, is_rx_ffcnt),             /* rx fast frames */
 250     IS_STAT(SU, is_rx_badathtnl),        /* driver key alloc failed */
 251     IS_STAT(SU, is_tx_nobuf),            /* tx failed for lack of buf */
 252     IS_STAT(SU, is_tx_nonode),           /* tx failed for no node */
 253     IS_STAT(SU, is_tx_unknownmgt),       /* tx of unknown mgt frame */
 254     IS_STAT(SU, is_tx_badcipher),        /* tx failed due to of key type */
 255     IS_STAT(SU, is_tx_nodefkey),         /* tx failed due to no defkey */
 256     IS_STAT(SU, is_tx_noheadroom),       /* tx failed due to no space */
 257     IS_STAT(D, is_tx_ffokcnt),           /* tx fast frames sent success */
 258     IS_STAT(D, is_tx_fferrcnt),          /* tx fast frames sent success */
 259     IS_STAT(D, is_scan_active),          /* active scans started */
 260     IS_STAT(D, is_scan_passive),         /* passive scans started */
 261     IS_STAT(D, is_node_timeout),         /* nodes timed out inactivity */
 262     IS_STAT(D, is_crypto_nomem),         /* no memory for crypto ctx */
 263     IS_STAT(D, is_crypto_tkip),          /* tkip crypto done in s/w */
 264     IS_STAT(D, is_crypto_tkipenmic),     /* tkip en-MIC done in s/w */
 265     IS_STAT(D, is_crypto_tkipdemic),     /* tkip de-MIC done in s/w */
 266     IS_STAT(D, is_crypto_tkipcm),        /* tkip counter measures */
 267     IS_STAT(D, is_crypto_ccmp),          /* ccmp crypto done in s/w */
 268     IS_STAT(D, is_crypto_wep),           /* wep crypto done in s/w */
 269     IS_STAT(D, is_crypto_setkey_cipher), /* cipher rejected key */
 270     IS_STAT(D, is_crypto_setkey_nokey),  /* no key index for setkey */
 271     IS_STAT(D, is_crypto_delkey),        /* driver key delete failed */
 272     IS_STAT(D, is_crypto_badcipher),     /* unknown cipher */
 273     IS_STAT(D, is_crypto_nocipher),      /* cipher not available */
 274     IS_STAT(D, is_crypto_attachfail),    /* cipher attach failed */
 275     IS_STAT(D, is_crypto_swfallback),    /* cipher fallback to s/w */
 276     IS_STAT(D, is_crypto_keyfail),       /* driver key alloc failed */
 277     IS_STAT(D, is_crypto_enmicfail),     /* en-MIC failed */
 278     IS_STAT(SU, is_ibss_capmismatch),    /* merge failed-cap mismatch */
 279     IS_STAT(SU, is_ibss_norate),         /* merge failed-rate mismatch */
 280     IS_STAT(D, is_ps_unassoc),           /* ps-poll for unassoc. sta */
 281     IS_STAT(D, is_ps_badaid),            /* ps-poll w/ incorrect aid */
 282     IS_STAT(D, is_ps_qempty),            /* ps-poll w/ nothing to send */
 283
 284     /* Atheros statistics */
 285     AS_STAT(D, ast_watchdog),     /* device reset by watchdog */
 286     AS_STAT(D, ast_hardware),     /* fatal hardware error interrupts */
 287     AS_STAT(D, ast_bmiss),        /* beacon miss interrupts */
 288     AS_STAT(D, ast_rxorn),        /* rx overrun interrupts */
 289     AS_STAT(D, ast_rxeol),        /* rx eol interrupts */
 290     AS_STAT(D, ast_txurn),        /* tx underrun interrupts */
 291     AS_STAT(D, ast_mib),          /* mib interrupts */
 292     AS_STAT(D, ast_tx_packets),   /* packet sent on the interface */
 293     AS_STAT(D, ast_tx_mgmt),      /* management frames transmitted */
 294     AS_STAT(LOG, ast_tx_discard), /* frames discarded prior to assoc */
 295     AS_STAT(SU, ast_tx_invalid),  /* frames discarded due to is device gone */
 296     AS_STAT(SU, ast_tx_qstop),    /* tx queue stopped because it's full */
 297     AS_STAT(SU, ast_tx_encap),    /* tx encapsulation failed */
 298     AS_STAT(SU, ast_tx_nonode),   /* tx failed due to of no node */
 299     AS_STAT(SU, ast_tx_nobuf),    /* tx failed due to of no tx buffer (data), */
 300     AS_STAT(SU, ast_tx_nobufmgt), /* tx failed due to of no tx buffer (mgmt),*/
 301     AS_STAT(LOG, ast_tx_xretries),   /* tx failed due to of too many retries */
 302     AS_STAT(SU, ast_tx_fifoerr),     /* tx failed due to of FIFO underrun */
 303     AS_STAT(SU, ast_tx_filtered),    /* tx failed due to xmit filtered */
 304     AS_STAT(LOG, ast_tx_shortretry), /* tx on-chip retries (short), */
 305     AS_STAT(LOG, ast_tx_longretry),  /* tx on-chip retries (long), */
 306     AS_STAT(SU, ast_tx_badrate),     /* tx failed due to of bogus xmit rate */
 307     AS_STAT(D, ast_tx_noack),        /* tx frames with no ack marked */
 308     AS_STAT(D, ast_tx_rts),          /* tx frames with rts enabled */
 309     AS_STAT(D, ast_tx_cts),          /* tx frames with cts enabled */
 310     AS_STAT(D, ast_tx_shortpre),     /* tx frames with short preamble */
 311     AS_STAT(LOG, ast_tx_altrate),    /* tx frames with alternate rate */
 312     AS_STAT(D, ast_tx_protect),      /* tx frames with protection */
 313     AS_STAT(SU, ast_rx_orn),         /* rx failed due to of desc overrun */
 314     AS_STAT(LOG, ast_rx_crcerr),     /* rx failed due to of bad CRC */
 315     AS_STAT(SU, ast_rx_fifoerr),     /* rx failed due to of FIFO overrun */
 316     AS_STAT(SU, ast_rx_badcrypt),    /* rx failed due to of decryption */
 317     AS_STAT(SU, ast_rx_badmic),      /* rx failed due to of MIC failure */
 318     AS_STAT(LOG, ast_rx_phyerr),     /* rx PHY error summary count */
 319     AS_STAT(SU, ast_rx_tooshort),    /* rx discarded due to frame too short */
 320     AS_STAT(SU, ast_rx_toobig),      /* rx discarded due to frame too large */
 321     AS_STAT(SU, ast_rx_nobuf),       /* rx setup failed due to of no skbuff */
 322     AS_STAT(D, ast_rx_packets),      /* packet recv on the interface */
 323     AS_STAT(D, ast_rx_mgt),          /* management frames received */
 324     AS_STAT(D, ast_rx_ctl),          /* control frames received */
 325     AS_STAT(D, ast_be_xmit),         /* beacons transmitted */
 326     AS_STAT(SU, ast_be_nobuf),       /* no skbuff available for beacon */
 327     AS_STAT(D, ast_per_cal),         /* periodic calibration calls */
 328     AS_STAT(D, ast_per_calfail),     /* periodic calibration failed */
 329     AS_STAT(D, ast_per_rfgain),      /* periodic calibration rfgain reset */
 330     AS_STAT(D, ast_rate_calls),      /* rate control checks */
 331     AS_STAT(D, ast_rate_raise),      /* rate control raised xmit rate */
 332     AS_STAT(D, ast_rate_drop),       /* rate control dropped xmit rate */
 333     AS_STAT(D, ast_ant_defswitch),   /* rx/default antenna switches */
 334     AS_STAT(D, ast_ant_txswitch)     /* tx antenna switches */
 335 };
 336
 337 /* Bounds between SS, NS, IS and AS stats in stats array */
 338 static int bounds[4];
 339
 340 #define WL_LEN 6
 341 /* Bitmasks for logged and error items */
 342 static uint32_t watch_items[WL_LEN];
 343 static uint32_t misc_items[WL_LEN];
 344
 345 static const char *config_keys[] = {"Interface", "IgnoreSelected", "Source",
 346                                     "WatchAdd",  "WatchRemove",    "WatchSet",
 347                                     "MiscAdd",   "MiscRemove",     "MiscSet"};
 348 static int config_keys_num = STATIC_ARRAY_SIZE(config_keys);
 349
 350 static ignorelist_t *ignorelist;
 351
 352 static int use_sysfs = 1;
 353 static int init_state;
 354
 355 static inline int item_watched(int i) {
 356   assert(i >= 0);
 357   assert((size_t)i < (STATIC_ARRAY_SIZE(watch_items) * 32));
 358   return watch_items[i / 32] & FLAG(i);
 359 }
 360
 361 static inline int item_summed(int i) {
 362   assert(i >= 0);
 363   assert((size_t)i < (STATIC_ARRAY_SIZE(misc_items) * 32));
 364   return misc_items[i / 32] & FLAG(i);
 365 }
 366
 367 static inline void watchlist_add(uint32_t *wl, int item) {
 368   assert(item >= 0);
 369   assert(item < WL_LEN * 32);
 370   wl[item / 32] |= FLAG(item);
 371 }
 372
 373 static inline void watchlist_remove(uint32_t *wl, int item) {
 374   assert(item >= 0);
 375   assert(item < WL_LEN * 32);
 376   wl[item / 32] &= ~FLAG(item);
 377 }
 378
 379 static inline void watchlist_set(uint32_t *wl, uint32_t val) {
 380   for (int i = 0; i < WL_LEN; i++)
 381     wl[i] = val;
 382 }
 383
 384 /* This is horribly inefficient, but it is called only during configuration */
 385 static int watchitem_find(const char *name) {
 386   int max = STATIC_ARRAY_SIZE(specs);
 387
 388   for (int i = 0; i < max; i++)
 389     if (strcasecmp(name, specs[i].name) == 0)
 390       return i;
 391
 392   return -1;
 393 }
 394
 395 /* Collectd hooks */
 396
 397 /* We need init function called before madwifi_config */
 398
 399 static int madwifi_real_init(void) {
 400   size_t max = STATIC_ARRAY_SIZE(specs);
 401
 402   for (size_t i = 0; i < STATIC_ARRAY_SIZE(bounds); i++)
 403     bounds[i] = 0;
 404
 405   watchlist_set(watch_items, 0);
 406   watchlist_set(misc_items, 0);
 407
 408   for (size_t i = 0; i < max; i++) {
 409     bounds[specs[i].flags & SRC_MASK] = i;
 410
 411     if (specs[i].flags & LOG)
 412       watch_items[i / 32] |= FLAG(i);
 413
 414     if (specs[i].flags & SU)
 415       misc_items[i / 32] |= FLAG(i);
 416   }
 417
 418   for (size_t i = 0; i < STATIC_ARRAY_SIZE(bounds); i++)
 419     bounds[i]++;
 420
 421   return 0;
 422 }
 423
 424 static int madwifi_config(const char *key, const char *value) {
 425   if (init_state != 1)
 426     madwifi_real_init();
 427   init_state = 1;
 428
 429   if (ignorelist == NULL)
 430     ignorelist = ignorelist_create(/* invert = */ 1);
 431
 432   if (strcasecmp(key, "Interface") == 0)
 433     ignorelist_add(ignorelist, value);
 434
 435   else if (strcasecmp(key, "IgnoreSelected") == 0)
 436     ignorelist_set_invert(ignorelist, IS_TRUE(value) ? 0 : 1);
 437
 438   else if (strcasecmp(key, "Source") == 0) {
 439     if (strcasecmp(value, "ProcFS") == 0)
 440       use_sysfs = 0;
 441     else if (strcasecmp(value, "SysFS") == 0)
 442       use_sysfs = 1;
 443     else {
 444       ERROR("madwifi plugin: The argument of the `Source' "
 445             "option must either be `SysFS' or "
 446             "`ProcFS'.");
 447       return -1;
 448     }
 449   }
 450
 451   else if (strcasecmp(key, "WatchSet") == 0) {
 452     if (strcasecmp(value, "All") == 0)
 453       watchlist_set(watch_items, 0xFFFFFFFF);
 454     else if (strcasecmp(value, "None") == 0)
 455       watchlist_set(watch_items, 0);
 456     else
 457       return -1;
 458   }
 459
 460   else if (strcasecmp(key, "WatchAdd") == 0) {
 461     int id = watchitem_find(value);
 462
 463     if (id < 0)
 464       return -1;
 465     else
 466       watchlist_add(watch_items, id);
 467   }
 468
 469   else if (strcasecmp(key, "WatchRemove") == 0) {
 470     int id = watchitem_find(value);
 471
 472     if (id < 0)
 473       return -1;
 474     else
 475       watchlist_remove(watch_items, id);
 476   }
 477
 478   else if (strcasecmp(key, "MiscSet") == 0) {
 479     if (strcasecmp(value, "All") == 0)
 480       watchlist_set(misc_items, 0xFFFFFFFF);
 481     else if (strcasecmp(value, "None") == 0)
 482       watchlist_set(misc_items, 0);
 483     else
 484       return -1;
 485   }
 486
 487   else if (strcasecmp(key, "MiscAdd") == 0) {
 488     int id = watchitem_find(value);
 489
 490     if (id < 0)
 491       return -1;
 492     else
 493       watchlist_add(misc_items, id);
 494   }
 495
 496   else if (strcasecmp(key, "MiscRemove") == 0) {
 497     int id = watchitem_find(value);
 498
 499     if (id < 0)
 500       return -1;
 501     else
 502       watchlist_remove(misc_items, id);
 503   }
 504
 505   else
 506     return -1;
 507
 508   return 0;
 509 }
 510
 511 static void submit(const char *dev, const char *type, const char *ti1,
 512                    const char *ti2, value_t *val, size_t len) {
 513   value_list_t vl = VALUE_LIST_INIT;
 514
 515   vl.values = val;
 516   vl.values_len = len;
 517   sstrncpy(vl.plugin, "madwifi", sizeof(vl.plugin));
 518   sstrncpy(vl.plugin_instance, dev, sizeof(vl.plugin_instance));
 519   sstrncpy(vl.type, type, sizeof(vl.type));
 520
 521   if ((ti1 != NULL) && (ti2 != NULL))
 522     snprintf(vl.type_instance, sizeof(vl.type_instance), "%s-%s", ti1, ti2);
 523   else if ((ti1 != NULL) && (ti2 == NULL))
 524     sstrncpy(vl.type_instance, ti1, sizeof(vl.type_instance));
 525
 526   plugin_dispatch_values(&vl);
 527 }
 528
 529 static void submit_derive(const char *dev, const char *type, const char *ti1,
 530                           const char *ti2, derive_t value) {
 531   submit(dev, type, ti1, ti2, &(value_t){.derive = value}, 1);
 532 }
 533
 534 static void submit_derive2(const char *dev, const char *type, const char *ti1,
 535                            const char *ti2, derive_t val1, derive_t val2) {
 536   value_t values[] = {
 537       {.derive = val1},
 538       {.derive = val2},
 539   };
 540
 541   submit(dev, type, ti1, ti2, values, STATIC_ARRAY_SIZE(values));
 542 }
 543
 544 static void submit_gauge(const char *dev, const char *type, const char *ti1,
 545                          const char *ti2, gauge_t value) {
 546   submit(dev, type, ti1, ti2, &(value_t){.gauge = value}, 1);
 547 }
 548
 549 static void submit_antx(const char *dev, const char *name, u_int32_t *vals,
 550                         int vals_num) {
 551   char ti2[16];
 552
 553   for (int i = 0; i < vals_num; i++) {
 554     if (vals[i] == 0)
 555       continue;
 556
 557     snprintf(ti2, sizeof(ti2), "%i", i);
 558     submit_derive(dev, "ath_stat", name, ti2, (derive_t)vals[i]);
 559   }
 560 }
 561
 562 static inline void macaddr_to_str(char *buf, size_t bufsize,
 563                                   const uint8_t mac[IEEE80211_ADDR_LEN]) {
 564   snprintf(buf, bufsize, "%02x:%02x:%02x:%02x:%02x:%02x", mac[0], mac[1],
 565            mac[2], mac[3], mac[4], mac[5]);
 566 }
 567
 568 static void process_stat_struct(int which, const void *ptr, const char *dev,
 569                                 const char *mac, const char *type_name,
 570                                 const char *misc_name) {
 571   uint32_t misc = 0;
 572
 573   assert(which >= 1);
 574   assert(((size_t)which) < STATIC_ARRAY_SIZE(bounds));
 575
 576   for (int i = bounds[which - 1]; i < bounds[which]; i++) {
 577     uint32_t val = *(uint32_t *)(((char *)ptr) + specs[i].offset);
 578
 579     if (item_watched(i) && (val != 0))
 580       submit_derive(dev, type_name, specs[i].name, mac, val);
 581
 582     if (item_summed(i))
 583       misc += val;
 584   }
 585
 586   if (misc != 0)
 587     submit_derive(dev, type_name, misc_name, mac, misc);
 588 }
 589
 590 static int process_athstats(int sk, const char *dev) {
 591   struct ifreq ifr;
 592   struct ath_stats stats;
 593   int status;
 594
 595   sstrncpy(ifr.ifr_name, dev, sizeof(ifr.ifr_name));
 596   ifr.ifr_data = (void *)&stats;
 597   status = ioctl(sk, SIOCGATHSTATS, &ifr);
 598   if (status < 0) {
 599     /* Silent, because not all interfaces support all ioctls. */
 600     DEBUG("madwifi plugin: Sending IO-control "
 601           "SIOCGATHSTATS to device %s "
 602           "failed with status %i.",
 603           dev, status);
 604     return status;
 605   }
 606
 607   /* These stats are handled as a special case, because they are
 608      eight values each */
 609
 610   if (item_watched(STAT_AST_ANT_RX))
 611     submit_antx(dev, "ast_ant_rx", stats.ast_ant_rx,
 612                 STATIC_ARRAY_SIZE(stats.ast_ant_rx));
 613
 614   if (item_watched(STAT_AST_ANT_TX))
 615     submit_antx(dev, "ast_ant_tx", stats.ast_ant_tx,
 616                 STATIC_ARRAY_SIZE(stats.ast_ant_tx));
 617
 618   /* All other ath statistics */
 619   process_stat_struct(ATH_STAT, &stats, dev, NULL, "ath_stat", "ast_misc");
 620   return 0;
 621 }
 622
 623 static int process_80211stats(int sk, const char *dev) {
 624   struct ifreq ifr;
 625   struct ieee80211_stats stats;
 626   int status;
 627
 628   sstrncpy(ifr.ifr_name, dev, sizeof(ifr.ifr_name));
 629   ifr.ifr_data = (void *)&stats;
 630   status = ioctl(sk, SIOCG80211STATS, &ifr);
 631   if (status < 0) {
 632     /* Silent, because not all interfaces support all ioctls. */
 633     DEBUG("madwifi plugin: Sending IO-control "
 634           "SIOCG80211STATS to device %s "
 635           "failed with status %i.",
 636           dev, status);
 637     return status;
 638   }
 639
 640   process_stat_struct(IFA_STAT, &stats, dev, NULL, "ath_stat", "is_misc");
 641   return 0;
 642 }
 643
 644 static int process_station(int sk, const char *dev,
 645                            struct ieee80211req_sta_info *si) {
 646   static char mac[DATA_MAX_NAME_LEN];
 647   struct ieee80211req_sta_stats stats;
 648   const struct ieee80211_nodestats *ns = &stats.is_stats;
 649   int status;
 650
 651   macaddr_to_str(mac, sizeof(mac), si->isi_macaddr);
 652
 653   if (item_watched(STAT_NODE_TX_RATE))
 654     submit_gauge(dev, "node_tx_rate", mac, NULL,
 655                  (si->isi_rates[si->isi_txrate] & IEEE80211_RATE_VAL) / 2);
 656
 657   if (item_watched(STAT_NODE_RSSI))
 658     submit_gauge(dev, "node_rssi", mac, NULL, si->isi_rssi);
 659
 660   struct iwreq iwr = {.u.data.pointer = (void *)&stats,
 661                       .u.data.length = sizeof(stats)};
 662   sstrncpy(iwr.ifr_name, dev, sizeof(iwr.ifr_name));
 663
 664   memcpy(stats.is_u.macaddr, si->isi_macaddr, IEEE80211_ADDR_LEN);
 665   status = ioctl(sk, IEEE80211_IOCTL_STA_STATS, &iwr);
 666   if (status < 0) {
 667     /* Silent, because not all interfaces support all ioctls. */
 668     DEBUG("madwifi plugin: Sending IO-control "
 669           "IEEE80211_IOCTL_STA_STATS to device %s "
 670           "failed with status %i.",
 671           dev, status);
 672     return status;
 673   }
 674
 675   /* These two stats are handled as a special case as they are
 676      a pair of 64bit values */
 677   if (item_watched(STAT_NODE_OCTETS))
 678     submit_derive2(dev, "node_octets", mac, NULL, ns->ns_rx_bytes,
 679                    ns->ns_tx_bytes);
 680
 681   /* This stat is handled as a special case, because it is stored
 682      as uin64_t, but we will ignore upper half */
 683   if (item_watched(STAT_NS_RX_BEACONS))
 684     submit_derive(dev, "node_stat", "ns_rx_beacons", mac,
 685                   (ns->ns_rx_beacons & 0xFFFFFFFF));
 686
 687   /* All other node statistics */
 688   process_stat_struct(NOD_STAT, ns, dev, mac, "node_stat", "ns_misc");
 689   return 0;
 690 }
 691
 692 static int process_stations(int sk, const char *dev) {
 693   uint8_t buf[24 * 1024] = {0};
 694   uint8_t *cp;
 695   int nodes;
 696   size_t len;
 697   int status;
 698
 699   struct iwreq iwr = {.u.data.pointer = (void *)buf,
 700                       .u.data.length = sizeof(buf)};
 701   sstrncpy(iwr.ifr_name, dev, sizeof(iwr.ifr_name));
 702
 703   status = ioctl(sk, IEEE80211_IOCTL_STA_INFO, &iwr);
 704   if (status < 0) {
 705     /* Silent, because not all interfaces support all ioctls. */
 706     DEBUG("madwifi plugin: Sending IO-control "
 707           "IEEE80211_IOCTL_STA_INFO to device %s "
 708           "failed with status %i.",
 709           dev, status);
 710     return status;
 711   }
 712
 713   len = iwr.u.data.length;
 714
 715   cp = buf;
 716   nodes = 0;
 717   while (len >= sizeof(struct ieee80211req_sta_info)) {
 718     struct ieee80211req_sta_info *si = (void *)cp;
 719     process_station(sk, dev, si);
 720     cp += si->isi_len;
 721     len -= si->isi_len;
 722     nodes++;
 723   }
 724
 725   if (item_watched(STAT_ATH_NODES))
 726     submit_gauge(dev, "ath_nodes", NULL, NULL, nodes);
 727   return 0;
 728 }
 729
 730 static int process_device(int sk, const char *dev) {
 731   int num_success = 0;
 732   int status;
 733
 734   status = process_athstats(sk, dev);
 735   if (status == 0)
 736     num_success++;
 737
 738   status = process_80211stats(sk, dev);
 739   if (status == 0)
 740     num_success++;
 741
 742   status = process_stations(sk, dev);
 743   if (status == 0)
 744     num_success++;
 745
 746   return (num_success == 0) ? -1 : 0;
 747 }
 748
 749 static int check_devname(const char *dev) {
 750   char buf[PATH_MAX];
 751   char buf2[PATH_MAX];
 752   int i;
 753
 754   if (dev[0] == '.')
 755     return 0;
 756
 757   snprintf(buf, sizeof(buf), "/sys/class/net/%s/device/driver", dev);
 758   buf[sizeof(buf) - 1] = '\0';
 759
 760   i = readlink(buf, buf2, sizeof(buf2) - 1);
 761   if (i < 0)
 762     return 0;
 763
 764   buf2[i] = '\0';
 765
 766   if (strstr(buf2, "/drivers/ath_") == NULL)
 767     return 0;
 768   return 1;
 769 }
 770
 771 static int sysfs_iterate(int sk) {
 772   struct dirent *de;
 773   DIR *nets;
 774   int status;
 775   int num_success;
 776   int num_fail;
 777
 778   nets = opendir("/sys/class/net/");
 779   if (nets == NULL) {
 780     WARNING("madwifi plugin: opening /sys/class/net failed");
 781     return -1;
 782   }
 783
 784   num_success = 0;
 785   num_fail = 0;
 786   while ((de = readdir(nets))) {
 787     if (check_devname(de->d_name) == 0)
 788       continue;
 789
 790     if (ignorelist_match(ignorelist, de->d_name) != 0)
 791       continue;
 792
 793     status = process_device(sk, de->d_name);
 794     if (status != 0) {
 795       ERROR("madwifi plugin: Processing interface "
 796             "%s failed.",
 797             de->d_name);
 798       num_fail++;
 799     } else {
 800       num_success++;
 801     }
 802   } /* while (readdir) */
 803
 804   closedir(nets);
 805
 806   if ((num_success == 0) && (num_fail != 0))
 807     return -1;
 808   return 0;
 809 }
 810
 811 static int procfs_iterate(int sk) {
 812   char buffer[1024];
 813   char *device, *dummy;
 814   FILE *fh;
 815   int status;
 816   int num_success;
 817   int num_fail;
 818
 819   if ((fh = fopen("/proc/net/dev", "r")) == NULL) {
 820     WARNING("madwifi plugin: opening /proc/net/dev failed");
 821     return -1;
 822   }
 823
 824   num_success = 0;
 825   num_fail = 0;
 826   while (fgets(buffer, sizeof(buffer), fh) != NULL) {
 827     dummy = strchr(buffer, ':');
 828     if (dummy == NULL)
 829       continue;
 830     dummy[0] = 0;
 831
 832     device = buffer;
 833     while (device[0] == ' ')
 834       device++;
 835
 836     if (device[0] == 0)
 837       continue;
 838
 839     if (ignorelist_match(ignorelist, device) != 0)
 840       continue;
 841
 842     status = process_device(sk, device);
 843     if (status != 0) {
 844       ERROR("madwifi plugin: Processing interface "
 845             "%s failed.",
 846             device);
 847       num_fail++;
 848     } else {
 849       num_success++;
 850     }
 851   } /* while (fgets) */
 852
 853   fclose(fh);
 854
 855   if ((num_success == 0) && (num_fail != 0))
 856     return -1;
 857   return 0;
 858 }
 859
 860 static int madwifi_read(void) {
 861   int rv;
 862   int sk;
 863
 864   if (init_state == 0)
 865     madwifi_real_init();
 866   init_state = 2;
 867
 868   sk = socket(AF_INET, SOCK_DGRAM, 0);
 869   if (sk < 0)
 870     return -1;
 871
 872   /* procfs iteration is not safe because it does not check whether given
 873      interface is madwifi interface and there are private ioctls used, which
 874      may do something completely different on non-madwifi devices.
 875      Therefore, it is not used unless explicitly enabled (and should be used
 876      together with ignorelist). */
 877
 878   if (use_sysfs)
 879     rv = sysfs_iterate(sk);
 880   else
 881     rv = procfs_iterate(sk);
 882
 883   close(sk);
 884
 885   return rv;
 886 }
 887
 888 void module_register(void) {
 889   plugin_register_config("madwifi", madwifi_config, config_keys,
 890                          config_keys_num);
 891
 892   plugin_register_read("madwifi", madwifi_read);
 893 }