Mark some functions in userland __printflike.
[dragonfly.git] / usr.sbin / 802_11 / ndis_events / ndis_events.c
CommitLineData
efba76b4
SW
1/*-
2 * Copyright (c) 2005
3 * Bill Paul <wpaul@windriver.com>. 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 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 * must display the following acknowledgement:
15 * This product includes software developed by Bill Paul.
16 * 4. Neither the name of the author nor the names of any co-contributors
17 * may be used to endorse or promote products derived from this software
18 * without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY Bill Paul AND CONTRIBUTORS ``AS IS'' AND
21 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23 * ARE DISCLAIMED. IN NO EVENT SHALL Bill Paul OR THE VOICES IN HIS HEAD
24 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
25 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
30 * THE POSSIBILITY OF SUCH DAMAGE.
31 *
32 * $FreeBSD: src/usr.sbin/wpa/ndis_events/ndis_events.c,v 1.8 2011/06/24 07:05:20 kevlo Exp $
33 */
34
35/*
36 * This program simulates the behavior of the ndis_events utility
37 * supplied with wpa_supplicant for Windows. The original utility
38 * is designed to translate Windows WMI events. We don't have WMI,
39 * but we need to supply certain event info to wpa_supplicant in
40 * order to make WPA2 work correctly, so we fake up the interface.
41 */
42
43#include <sys/types.h>
44#include <sys/param.h>
45#include <sys/socket.h>
46#include <sys/ioctl.h>
47#include <sys/errno.h>
48#include <sys/sysctl.h>
49#include <net/if.h>
50#include <net/if_dl.h>
51#include <net/if_var.h>
52
53#include <netinet/in.h>
54#include <arpa/inet.h>
55#include <netdb.h>
56#include <net/route.h>
57
58#include <stdio.h>
59#include <string.h>
60#include <stdlib.h>
61#include <unistd.h>
62#include <err.h>
63#include <syslog.h>
64#include <stdarg.h>
65
66static int verbose = 0;
67static int debug = 0;
68static int all_events = 0;
69
70#define PROGNAME "ndis_events"
71
72#define WPA_SUPPLICANT_PORT 9876
73#define NDIS_INDICATION_LEN 2048
74
75#define EVENT_CONNECT 0
76#define EVENT_DISCONNECT 1
77#define EVENT_MEDIA_SPECIFIC 2
78
79#define NDIS_STATUS_MEDIA_CONNECT 0x4001000B
80#define NDIS_STATUS_MEDIA_DISCONNECT 0x4001000C
81#define NDIS_STATUS_MEDIA_SPECIFIC_INDICATION 0x40010012
82
83struct ndis_evt {
84 uint32_t ne_sts;
85 uint32_t ne_len;
86#ifdef notdef
87 char ne_buf[1];
88#endif
89};
90
f7eac9d2 91static void dbgmsg(const char *, ...) __printflike(1, 2);
efba76b4
SW
92static int find_ifname(int, char *);
93static int announce_event(char *, int, struct sockaddr_in *);
94static void usage(void);
95
96static void
97dbgmsg(const char *fmt, ...)
98{
99 va_list ap;
100
101 va_start(ap, fmt);
102 if (debug)
103 vwarnx(fmt, ap);
104 else
105 vsyslog(LOG_ERR, fmt, ap);
106 va_end(ap);
107
108 return;
109}
110
111static int
112find_ifname(int idx, char *name)
113{
114 int mib[6];
115 size_t needed;
116 struct if_msghdr *ifm;
117 struct sockaddr_dl *sdl;
118 char *buf, *lim, *next;
119
120 needed = 0;
121 mib[0] = CTL_NET;
122 mib[1] = PF_ROUTE;
123 mib[2] = 0; /* protocol */
124 mib[3] = 0; /* wildcard address family */
125 mib[4] = NET_RT_IFLIST;
126 mib[5] = 0; /* no flags */
127
128 if (sysctl (mib, 6, NULL, &needed, NULL, 0) < 0)
129 return(EIO);
130
131 buf = malloc (needed);
132 if (buf == NULL)
133 return(ENOMEM);
134
135 if (sysctl (mib, 6, buf, &needed, NULL, 0) < 0) {
136 free(buf);
137 return(EIO);
138 }
139
140 lim = buf + needed;
141
142 next = buf;
143 while (next < lim) {
144 ifm = (struct if_msghdr *)next;
145 if (ifm->ifm_type == RTM_IFINFO) {
146 sdl = (struct sockaddr_dl *)(ifm + 1);
147 if (ifm->ifm_index == idx) {
148 strncpy(name, sdl->sdl_data, sdl->sdl_nlen);
149 name[sdl->sdl_nlen] = '\0';
150 free (buf);
151 return (0);
152 }
153 }
154 next += ifm->ifm_msglen;
155 }
156
157 free (buf);
158
159 return(ENOENT);
160}
161
162static int
163announce_event(char *ifname, int sock, struct sockaddr_in *dst)
164{
165 int s;
166 char indication[NDIS_INDICATION_LEN];
167 struct ifreq ifr;
168 struct ndis_evt *e;
169 char buf[512], *pos, *end;
170 int len, type, _type;
171
172 s = socket(PF_INET, SOCK_DGRAM, 0);
173
174 if (s < 0) {
175 dbgmsg("socket creation failed");
176 return(EINVAL);
177 }
178
179 bzero((char *)&ifr, sizeof(ifr));
180 e = (struct ndis_evt *)indication;
181 e->ne_len = NDIS_INDICATION_LEN - sizeof(struct ndis_evt);
182
183 strlcpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name));
184 ifr.ifr_data = indication;
185
186 if (ioctl(s, SIOCGPRIVATE_0, &ifr) < 0) {
187 close(s);
188 if (verbose) {
189 if (errno == ENOENT)
190 dbgmsg("drained all events from %s",
96ea1ce3 191 ifname);
efba76b4
SW
192 else
193 dbgmsg("failed to read event info from %s: %d",
194 ifname, errno);
195 }
196 return(ENOENT);
197 }
198
199 if (e->ne_sts == NDIS_STATUS_MEDIA_CONNECT) {
200 type = EVENT_CONNECT;
201 if (verbose)
202 dbgmsg("Received a connect event for %s", ifname);
203 if (!all_events) {
204 close(s);
205 return(0);
206 }
8533325e 207 } else if (e->ne_sts == NDIS_STATUS_MEDIA_DISCONNECT) {
efba76b4
SW
208 type = EVENT_DISCONNECT;
209 if (verbose)
210 dbgmsg("Received a disconnect event for %s", ifname);
211 if (!all_events) {
212 close(s);
213 return(0);
214 }
8533325e 215 } else if (e->ne_sts == NDIS_STATUS_MEDIA_SPECIFIC_INDICATION) {
efba76b4
SW
216 type = EVENT_MEDIA_SPECIFIC;
217 if (verbose)
218 dbgmsg("Received a media-specific event for %s",
219 ifname);
8533325e
SW
220 } else {
221 dbgmsg("unknown event %u\n", e->ne_sts);
222 return(EINVAL);
efba76b4
SW
223 }
224
225 end = buf + sizeof(buf);
226 _type = (int) type;
227 memcpy(buf, &_type, sizeof(_type));
228 pos = buf + sizeof(_type);
229
230 len = snprintf(pos + 1, end - pos - 1, "%s", ifname);
231 if (len < 0) {
232 close(s);
233 return(ENOSPC);
234 }
235 if (len > 255)
236 len = 255;
237 *pos = (unsigned char) len;
238 pos += 1 + len;
239 if (e->ne_len) {
240 if (e->ne_len > 255 || 1 + e->ne_len > (u_int)(end - pos)) {
241 dbgmsg("Not enough room for send_event data (%d)\n",
242 e->ne_len);
243 close(s);
244 return(ENOSPC);
245 }
246 *pos++ = (unsigned char) e->ne_len;
247 memcpy(pos, (indication) + sizeof(struct ndis_evt), e->ne_len);
248 pos += e->ne_len;
249 }
250
251 len = sendto(sock, buf, pos - buf, 0, (struct sockaddr *) dst,
252 sizeof(struct sockaddr_in));
253
254 close(s);
255 return(0);
256}
257
258static void
259usage(void)
260{
261 fprintf(stderr, "Usage: ndis_events [-a] [-d] [-v]\n");
262 exit(1);
263}
264
265int
266main(int argc, char *argv[])
267{
268 int s, r, n;
269 struct sockaddr_in sin;
270 char msg[NDIS_INDICATION_LEN];
271 struct rt_msghdr *rtm;
272 struct if_msghdr *ifm;
273 char ifname[IFNAMSIZ];
274 int ch;
275
276 while ((ch = getopt(argc, argv, "dva")) != -1) {
277 switch(ch) {
278 case 'd':
279 debug++;
280 break;
281 case 'v':
282 verbose++;
283 break;
284 case 'a':
285 all_events++;
286 break;
287 default:
288 usage();
289 break;
290 }
291 }
292
293 if (!debug && daemon(0, 0))
294 err(1, "failed to daemonize ourselves");
295
296 if (!debug)
297 openlog(PROGNAME, LOG_PID | LOG_CONS, LOG_DAEMON);
298
299 bzero((char *)&sin, sizeof(sin));
300
301 /* Create a datagram socket. */
302
303 s = socket(PF_INET, SOCK_DGRAM, 0);
304 if (s < 0) {
305 dbgmsg("socket creation failed");
306 exit(1);
307 }
308
309 sin.sin_family = AF_INET;
310 sin.sin_addr.s_addr = inet_addr("127.0.0.1");
311 sin.sin_port = htons(WPA_SUPPLICANT_PORT);
312
313 /* Create a routing socket. */
314
315 r = socket (PF_ROUTE, SOCK_RAW, 0);
316 if (r < 0) {
317 dbgmsg("routing socket creation failed");
318 exit(1);
319 }
320
321 /* Now sit and spin, waiting for events. */
322
323 if (verbose)
324 dbgmsg("Listening for events");
325
326 while (1) {
327 n = read(r, msg, NDIS_INDICATION_LEN);
3d67fe3c
SW
328 if (n < 0) {
329 dbgmsg("read failed");
330 exit(1);
331 }
efba76b4
SW
332 rtm = (struct rt_msghdr *)msg;
333 if (rtm->rtm_type != RTM_IFINFO)
334 continue;
335 ifm = (struct if_msghdr *)msg;
336 if (find_ifname(ifm->ifm_index, ifname))
337 continue;
338 if (strstr(ifname, "ndis")) {
339 while(announce_event(ifname, s, &sin) == 0)
340 ;
341 } else {
342 if (verbose)
343 dbgmsg("Skipping ifinfo message from %s",
344 ifname);
345 }
346 }
347
348 /* NOTREACHED */
349 exit(0);
350}