]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - tools/tools/net80211/wlanstats/main.c
sys/{x86,amd64}: remove one of doubled ;s
[FreeBSD/FreeBSD.git] / tools / tools / net80211 / wlanstats / main.c
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, MERCHANTIBILITY
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$
30  */
31
32 /*
33  * wlanstats [-i interface]
34  * (default interface is wlan0).
35  */
36
37 #include <sys/param.h>
38 #include <sys/socket.h>
39
40 #include <net/ethernet.h>
41 #include <net80211/_ieee80211.h>
42
43 #include <err.h>
44 #include <signal.h>
45 #include <stdio.h>
46 #include <stdlib.h>
47 #include <strings.h>
48 #include <unistd.h>
49
50 #include "wlanstats.h"
51
52 static struct {
53         const char *tag;
54         const char *fmt;
55 } tags[] = {
56   { "default",
57     "input,rx_mgmt,output,rx_badkeyid,scan_active,scan_bg,bmiss,rssi,noise,rate"
58   },
59   { "ampdu",
60     "input,output,ampdu_reorder,ampdu_oor,rx_dup,ampdu_flush,ampdu_move,"
61     "ampdu_drop,ampdu_bar,ampdu_baroow,ampdu_barmove,ampdu_bartx,"
62     "ampdu_bartxfail,ampdu_bartxretry,rssi,rate"
63   },
64   {
65     "amsdu",
66     "input,output,amsdu_tooshort,amsdu_split,amsdu_decap,amsdu_encap,rssi,rate"
67   },
68 };
69
70 static const char *
71 getfmt(const char *tag)
72 {
73         int i;
74         for (i = 0; i < nitems(tags); i++)
75                 if (strcasecmp(tags[i].tag, tag) == 0)
76                         return tags[i].fmt;
77         return tag;
78 }
79
80 static int signalled;
81
82 static void
83 catchalarm(int signo __unused)
84 {
85         signalled = 1;
86 }
87
88 #if 0
89 static void
90 print_sta_stats(FILE *fd, const u_int8_t macaddr[IEEE80211_ADDR_LEN])
91 {
92 #define STAT(x,fmt) \
93         if (ns->ns_##x) { fprintf(fd, "%s" #x " " fmt, sep, ns->ns_##x); sep = " "; }
94         struct ieee80211req ireq;
95         struct ieee80211req_sta_stats stats;
96         const struct ieee80211_nodestats *ns = &stats.is_stats;
97         const char *sep;
98
99         (void) memset(&ireq, 0, sizeof(ireq));
100         (void) strncpy(ireq.i_name, ifr.ifr_name, sizeof(ireq.i_name));
101         ireq.i_type = IEEE80211_IOC_STA_STATS;
102         ireq.i_data = &stats;
103         ireq.i_len = sizeof(stats);
104         memcpy(stats.is_u.macaddr, macaddr, IEEE80211_ADDR_LEN);
105         if (ioctl(s, SIOCG80211, &ireq) < 0)
106                 err(1, "unable to get station stats for %s",
107                         ether_ntoa((const struct ether_addr*) macaddr));
108
109         fprintf(fd, "%s:\n", ether_ntoa((const struct ether_addr*) macaddr));
110
111         sep = "\t";
112         STAT(rx_data, "%u");
113         STAT(rx_mgmt, "%u");
114         STAT(rx_ctrl, "%u");
115         STAT(rx_beacons, "%u");
116         STAT(rx_proberesp, "%u");
117         STAT(rx_ucast, "%u");
118         STAT(rx_mcast, "%u");
119         STAT(rx_bytes, "%llu");
120         STAT(rx_dup, "%u");
121         STAT(rx_noprivacy, "%u");
122         STAT(rx_wepfail, "%u");
123         STAT(rx_demicfail, "%u");
124         STAT(rx_decap, "%u");
125         STAT(rx_defrag, "%u");
126         STAT(rx_disassoc, "%u");
127         STAT(rx_deauth, "%u");
128         STAT(rx_decryptcrc, "%u");
129         STAT(rx_unauth, "%u");
130         STAT(rx_unencrypted, "%u");
131         fprintf(fd, "\n");
132
133         sep = "\t";
134         STAT(tx_data, "%u");
135         STAT(tx_mgmt, "%u");
136         STAT(tx_probereq, "%u");
137         STAT(tx_ucast, "%u");
138         STAT(tx_mcast, "%u");
139         STAT(tx_bytes, "%llu");
140         STAT(tx_novlantag, "%u");
141         STAT(tx_vlanmismatch, "%u");
142         fprintf(fd, "\n");
143
144         sep = "\t";
145         STAT(tx_assoc, "%u");
146         STAT(tx_assoc_fail, "%u");
147         STAT(tx_auth, "%u");
148         STAT(tx_auth_fail, "%u");
149         STAT(tx_deauth, "%u");
150         STAT(tx_deauth_code, "%llu");
151         STAT(tx_disassoc, "%u");
152         STAT(tx_disassoc_code, "%u");
153         fprintf(fd, "\n");
154
155 #undef STAT
156 }
157 #endif
158
159 void
160 usage(void) {
161         printf("wlanstats: [-ah] [-i ifname] [-l] [-m station MAC address] [-o fmt] [interval]\n");
162 }
163
164 int
165 main(int argc, char *argv[])
166 {
167         struct wlanstatfoo *wf;
168         struct ether_addr *ea;
169         const uint8_t *mac = NULL;
170         const char *ifname;
171         int allnodes = 0;
172         int c, mode;
173
174         ifname = getenv("WLAN");
175         if (ifname == NULL)
176                 ifname = "wlan0";
177         wf = wlanstats_new(ifname, getfmt("default"));
178         while ((c = getopt(argc, argv, "ahi:lm:o:")) != -1) {
179                 switch (c) {
180                 case 'a':
181                         allnodes++;
182                         break;
183                 case 'h':
184                         usage();
185                         exit(0);
186                 case 'i':
187                         wf->setifname(wf, optarg);
188                         break;
189                 case 'l':
190                         wf->print_fields(wf, stdout);
191                         return 0;
192                 case 'm':
193                         ea = ether_aton(optarg);
194                         if (!ea)
195                                 errx(1, "%s: invalid ethernet address", optarg);
196                         mac = ea->octet;
197                         break;
198                 case 'o':
199                         wf->setfmt(wf, getfmt(optarg));
200                         break;
201                 default:
202                         usage();
203                         exit(1);
204                         /*NOTREACHED*/
205                 }
206         }
207         argc -= optind;
208         argv += optind;
209
210         mode = wf->getopmode(wf);
211         wf->setstamac(wf, mac);
212
213         if (argc > 0) {
214                 u_long interval = strtoul(argv[0], NULL, 0);
215                 int line, omask;
216
217                 if (interval < 1)
218                         interval = 1;
219                 signal(SIGALRM, catchalarm);
220                 signalled = 0;
221                 alarm(interval);
222         banner:
223                 wf->print_header(wf, stdout);
224                 line = 0;
225         loop:
226                 if (line != 0) {
227                         wf->collect_cur(wf);
228                         wf->print_current(wf, stdout);
229                         wf->update_tot(wf);
230                 } else {
231                         wf->collect_tot(wf);
232                         wf->print_total(wf, stdout);
233                 }
234                 fflush(stdout);
235                 omask = sigblock(sigmask(SIGALRM));
236                 if (!signalled)
237                         sigpause(0);
238                 sigsetmask(omask);
239                 signalled = 0;
240                 alarm(interval);
241                 line++;
242                 /* refresh every display in case sta roams */
243                 if (mac == NULL && mode == IEEE80211_M_STA)
244                         wf->setstamac(wf, NULL);
245                 if (line == 21)         /* XXX tty line count */
246                         goto banner;
247                 else
248                         goto loop;
249                 /*NOTREACHED*/
250 #if 0
251         } else if (allnodes) {
252                 struct ieee80211req_sta_info *si;
253                 union {
254                         struct ieee80211req_sta_req req;
255                         uint8_t buf[24*1024];
256                 } u;
257                 uint8_t *cp;
258                 struct ieee80211req ireq;
259                 int len;
260
261                 /*
262                  * Retrieve station/neighbor table and print stats for each.
263                  */
264                 (void) memset(&ireq, 0, sizeof(ireq));
265                 (void) strncpy(ireq.i_name, ifr.ifr_name, sizeof(ireq.i_name));
266                 ireq.i_type = IEEE80211_IOC_STA_INFO;
267                 memset(&u.req.macaddr, 0xff, sizeof(u.req.macaddr));
268                 ireq.i_data = &u;
269                 ireq.i_len = sizeof(u);
270                 if (ioctl(s, SIOCG80211, &ireq) < 0)
271                         err(1, "unable to get station information");
272                 len = ireq.i_len;
273                 if (len >= sizeof(struct ieee80211req_sta_info)) {
274                         cp = u.req.info;
275                         do {
276                                 si = (struct ieee80211req_sta_info *) cp;
277                                 if (si->isi_len < sizeof(*si))
278                                         break;
279                                 print_sta_stats(stdout, si->isi_macaddr);
280                                 cp += si->isi_len, len -= si->isi_len;
281                         } while (len >= sizeof(struct ieee80211req_sta_info));
282                 }
283 #endif
284         } else {
285                 wf->collect_tot(wf);
286                 wf->print_verbose(wf, stdout);
287         }
288         return 0;
289 }