1 /*- 2 * Copyright (c) 2002-2007 Sam Leffler, Errno Consulting 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer, 10 * without modification. 11 * 2. Redistributions in binary form must reproduce at minimum a disclaimer 12 * similar to the "NO WARRANTY" disclaimer below ("Disclaimer") and any 13 * redistribution must be conditioned upon including a substantially 14 * similar Disclaimer requirement for further binary redistribution. 15 * 16 * NO WARRANTY 17 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19 * LIMITED TO, THE IMPLIED WARRANTIES OF NONINFRINGEMENT, MERCHANTABILITY 20 * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL 21 * THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR SPECIAL, EXEMPLARY, 22 * OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 23 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 24 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER 25 * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 26 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF 27 * THE POSSIBILITY OF SUCH DAMAGES. 28 * 29 * $FreeBSD: src/tools/tools/net80211/wlanstats/main.c,v 1.6 2009/02/05 22:17:10 sam Exp $ 30 */ 31 32 /* 33 * wlanstats [-i interface] 34 * (default interface is wlan0). 35 */ 36 37 #include <sys/types.h> 38 #include <sys/socket.h> 39 #include <net/ethernet.h> 40 #include <netproto/802_11/_ieee80211.h> 41 42 #include <stdlib.h> 43 #include <stdio.h> 44 #include <signal.h> 45 #include <unistd.h> 46 #include <err.h> 47 #include <strings.h> 48 49 #include "wlanstats.h" 50 51 static struct { 52 const char *tag; 53 const char *fmt; 54 } tags[] = { 55 { "default", 56 "input,rx_mgmt,output,rx_badkeyid,scan_active,scan_bg,bmiss,rssi,noise,rate" 57 }, 58 { "ampdu", 59 "input,output,ampdu_reorder,ampdu_oor,rx_dup,ampdu_flush,ampdu_move," 60 "ampdu_drop,ampdu_bar,ampdu_baroow,ampdu_barmove,rssi,rate" 61 }, 62 }; 63 64 static const char * 65 getfmt(const char *tag) 66 { 67 #define N(a) (sizeof(a)/sizeof(a[0])) 68 int i; 69 for (i = 0; i < N(tags); i++) 70 if (strcasecmp(tags[i].tag, tag) == 0) 71 return tags[i].fmt; 72 return tag; 73 #undef N 74 } 75 76 static int signalled; 77 78 static void 79 catchalarm(int signo __unused) 80 { 81 signalled = 1; 82 } 83 84 #if 0 85 static void 86 print_sta_stats(FILE *fd, const u_int8_t macaddr[IEEE80211_ADDR_LEN]) 87 { 88 #define STAT(x,fmt) \ 89 if (ns->ns_##x) { fprintf(fd, "%s" #x " " fmt, sep, ns->ns_##x); sep = " "; } 90 struct ieee80211req ireq; 91 struct ieee80211req_sta_stats stats; 92 const struct ieee80211_nodestats *ns = &stats.is_stats; 93 const char *sep; 94 95 (void) memset(&ireq, 0, sizeof(ireq)); 96 (void) strncpy(ireq.i_name, ifr.ifr_name, sizeof(ireq.i_name)); 97 ireq.i_type = IEEE80211_IOC_STA_STATS; 98 ireq.i_data = &stats; 99 ireq.i_len = sizeof(stats); 100 memcpy(stats.is_u.macaddr, macaddr, IEEE80211_ADDR_LEN); 101 if (ioctl(s, SIOCG80211, &ireq) < 0) 102 err(1, "unable to get station stats for %s", 103 ether_ntoa((const struct ether_addr*) macaddr)); 104 105 fprintf(fd, "%s:\n", ether_ntoa((const struct ether_addr*) macaddr)); 106 107 sep = "\t"; 108 STAT(rx_data, "%u"); 109 STAT(rx_mgmt, "%u"); 110 STAT(rx_ctrl, "%u"); 111 STAT(rx_beacons, "%u"); 112 STAT(rx_proberesp, "%u"); 113 STAT(rx_ucast, "%u"); 114 STAT(rx_mcast, "%u"); 115 STAT(rx_bytes, "%llu"); 116 STAT(rx_dup, "%u"); 117 STAT(rx_noprivacy, "%u"); 118 STAT(rx_wepfail, "%u"); 119 STAT(rx_demicfail, "%u"); 120 STAT(rx_decap, "%u"); 121 STAT(rx_defrag, "%u"); 122 STAT(rx_disassoc, "%u"); 123 STAT(rx_deauth, "%u"); 124 STAT(rx_decryptcrc, "%u"); 125 STAT(rx_unauth, "%u"); 126 STAT(rx_unencrypted, "%u"); 127 fprintf(fd, "\n"); 128 129 sep = "\t"; 130 STAT(tx_data, "%u"); 131 STAT(tx_mgmt, "%u"); 132 STAT(tx_probereq, "%u"); 133 STAT(tx_ucast, "%u"); 134 STAT(tx_mcast, "%u"); 135 STAT(tx_bytes, "%llu"); 136 STAT(tx_novlantag, "%u"); 137 STAT(tx_vlanmismatch, "%u"); 138 fprintf(fd, "\n"); 139 140 sep = "\t"; 141 STAT(tx_assoc, "%u"); 142 STAT(tx_assoc_fail, "%u"); 143 STAT(tx_auth, "%u"); 144 STAT(tx_auth_fail, "%u"); 145 STAT(tx_deauth, "%u"); 146 STAT(tx_deauth_code, "%llu"); 147 STAT(tx_disassoc, "%u"); 148 STAT(tx_disassoc_code, "%u"); 149 fprintf(fd, "\n"); 150 151 #undef STAT 152 } 153 #endif 154 155 int 156 main(int argc, char *argv[]) 157 { 158 struct wlanstatfoo *wf; 159 struct ether_addr *ea; 160 const uint8_t *mac = NULL; 161 const char *ifname; 162 int allnodes = 0; 163 int c, mode; 164 165 ifname = getenv("WLAN"); 166 if (ifname == NULL) 167 ifname = "wlan0"; 168 wf = wlanstats_new(ifname, getfmt("default")); 169 while ((c = getopt(argc, argv, "ai:lm:o:")) != -1) { 170 switch (c) { 171 case 'a': 172 allnodes++; 173 break; 174 case 'i': 175 wf->setifname(wf, optarg); 176 break; 177 case 'l': 178 wf->print_fields(wf, stdout); 179 return 0; 180 case 'm': 181 ea = ether_aton(optarg); 182 if (!ea) 183 errx(1, "%s: invalid ethernet address", optarg); 184 mac = ea->octet; 185 break; 186 case 'o': 187 wf->setfmt(wf, getfmt(optarg)); 188 break; 189 default: 190 errx(-1, "usage: %s [-a] [-i ifname] [-l] [-o fmt] [interval]\n", argv[0]); 191 /*NOTREACHED*/ 192 } 193 } 194 argc -= optind; 195 argv += optind; 196 197 mode = wf->getopmode(wf); 198 wf->setstamac(wf, mac); 199 200 if (argc > 0) { 201 u_long interval = strtoul(argv[0], NULL, 0); 202 int line, omask; 203 204 if (interval < 1) 205 interval = 1; 206 signal(SIGALRM, catchalarm); 207 signalled = 0; 208 alarm(interval); 209 banner: 210 wf->print_header(wf, stdout); 211 line = 0; 212 loop: 213 if (line != 0) { 214 wf->collect_cur(wf); 215 wf->print_current(wf, stdout); 216 wf->update_tot(wf); 217 } else { 218 wf->collect_tot(wf); 219 wf->print_total(wf, stdout); 220 } 221 fflush(stdout); 222 omask = sigblock(sigmask(SIGALRM)); 223 if (!signalled) 224 sigpause(0); 225 sigsetmask(omask); 226 signalled = 0; 227 alarm(interval); 228 line++; 229 /* refresh every display in case sta roams */ 230 if (mac == NULL && mode == IEEE80211_M_STA) 231 wf->setstamac(wf, NULL); 232 if (line == 21) /* XXX tty line count */ 233 goto banner; 234 else 235 goto loop; 236 /*NOTREACHED*/ 237 #if 0 238 } else if (allnodes) { 239 struct ieee80211req_sta_info *si; 240 union { 241 struct ieee80211req_sta_req req; 242 uint8_t buf[24*1024]; 243 } u; 244 uint8_t *cp; 245 struct ieee80211req ireq; 246 int len; 247 248 /* 249 * Retrieve station/neighbor table and print stats for each. 250 */ 251 (void) memset(&ireq, 0, sizeof(ireq)); 252 (void) strncpy(ireq.i_name, ifr.ifr_name, sizeof(ireq.i_name)); 253 ireq.i_type = IEEE80211_IOC_STA_INFO; 254 memset(&u.req.macaddr, 0xff, sizeof(u.req.macaddr)); 255 ireq.i_data = &u; 256 ireq.i_len = sizeof(u); 257 if (ioctl(s, SIOCG80211, &ireq) < 0) 258 err(1, "unable to get station information"); 259 len = ireq.i_len; 260 if (len >= sizeof(struct ieee80211req_sta_info)) { 261 cp = u.req.info; 262 do { 263 si = (struct ieee80211req_sta_info *) cp; 264 if (si->isi_len < sizeof(*si)) 265 break; 266 print_sta_stats(stdout, si->isi_macaddr); 267 cp += si->isi_len, len -= si->isi_len; 268 } while (len >= sizeof(struct ieee80211req_sta_info)); 269 } 270 #endif 271 } else { 272 wf->collect_tot(wf); 273 wf->print_verbose(wf, stdout); 274 } 275 return 0; 276 } 277