usr.sbin/wpa/ndis_events/ndis_events.c

   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
  33 #include <sys/cdefs.h>
  34 __FBSDID("$FreeBSD$");
  35
  36 /*
  37  * This program simulates the behavior of the ndis_events utility
  38  * supplied with wpa_supplicant for Windows. The original utility
  39  * is designed to translate Windows WMI events. We don't have WMI,
  40  * but we need to supply certain event info to wpa_supplicant in
  41  * order to make WPA2 work correctly, so we fake up the interface.
  42  */
  43
  44 #include <sys/types.h>
  45 #include <sys/cdefs.h>
  46 #include <sys/param.h>
  47 #include <sys/socket.h>
  48 #include <sys/ioctl.h>
  49 #include <sys/socket.h>
  50 #include <sys/errno.h>
  51 #include <sys/sysctl.h>
  52 #include <net/if.h>
  53 #include <net/if_dl.h>
  54 #include <net/if_var.h>
  55
  56 #include <netinet/in.h>
  57 #include <arpa/inet.h>
  58 #include <netdb.h>
  59 #include <net/route.h>
  60
  61 #include <stdio.h>
  62 #include <string.h>
  63 #include <stdlib.h>
  64 #include <unistd.h>
  65 #include <err.h>
  66 #include <syslog.h>
  67 #include <stdarg.h>
  68
  69 static int verbose = 0;
  70 static int debug = 0;
  71 static int all_events = 0;
  72
  73 #define PROGNAME "ndis_events"
  74
  75 #define WPA_SUPPLICANT_PORT     9876
  76 #define NDIS_INDICATION_LEN     2048
  77
  78 #define EVENT_CONNECT           0
  79 #define EVENT_DISCONNECT        1
  80 #define EVENT_MEDIA_SPECIFIC    2
  81
  82 #define NDIS_STATUS_MEDIA_CONNECT               0x4001000B
  83 #define NDIS_STATUS_MEDIA_DISCONNECT            0x4001000C
  84 #define NDIS_STATUS_MEDIA_SPECIFIC_INDICATION   0x40010012
  85
  86 struct ndis_evt {
  87         uint32_t                ne_sts;
  88         uint32_t                ne_len;
  89 #ifdef notdef
  90         char                    ne_buf[1];
  91 #endif
  92 };
  93
  94 static int find_ifname(int, char *);
  95 static int announce_event(char *, int, struct sockaddr_in *);
  96 static void usage(void);
  97
  98 static void
  99 dbgmsg(const char *fmt, ...)
 100 {
 101         va_list                 ap;
 102
 103         va_start(ap, fmt);
 104         if (debug)
 105                 vwarnx(fmt, ap);
 106         else
 107                 vsyslog(LOG_ERR, fmt, ap);
 108         va_end(ap);
 109
 110         return;
 111 }
 112
 113 static int
 114 find_ifname(idx, name)
 115         int                     idx;
 116         char                    *name;
 117 {
 118         int                     mib[6];
 119         size_t                  needed;
 120         struct if_msghdr        *ifm;
 121         struct sockaddr_dl      *sdl;
 122         char                    *buf, *lim, *next;
 123
 124         needed = 0;
 125         mib[0] = CTL_NET;
 126         mib[1] = PF_ROUTE;
 127         mib[2] = 0;             /* protocol */
 128         mib[3] = 0;             /* wildcard address family */
 129         mib[4] = NET_RT_IFLIST;
 130         mib[5] = 0;             /* no flags */
 131
 132         if (sysctl (mib, 6, NULL, &needed, NULL, 0) < 0)
 133                 return(EIO);
 134
 135         buf = malloc (needed);
 136         if (buf == NULL)
 137                 return(ENOMEM);
 138
 139         if (sysctl (mib, 6, buf, &needed, NULL, 0) < 0) {
 140                 free(buf);
 141                 return(EIO);
 142         }
 143
 144         lim = buf + needed;
 145
 146         next = buf;
 147         while (next < lim) {
 148                 ifm = (struct if_msghdr *)next;
 149                 if (ifm->ifm_type == RTM_IFINFO) {
 150                         sdl = (struct sockaddr_dl *)(ifm + 1);
 151                         if (ifm->ifm_index == idx) {
 152                                 strncpy(name, sdl->sdl_data, sdl->sdl_nlen);
 153                                 name[sdl->sdl_nlen] = '\0';
 154                                 free (buf);
 155                                 return (0);
 156                         }
 157                 }
 158                 next += ifm->ifm_msglen;
 159         }
 160
 161         free (buf);
 162
 163         return(ENOENT);
 164 }
 165
 166 static int
 167 announce_event(ifname, sock, dst)
 168         char                    *ifname;
 169         int                     sock;
 170         struct sockaddr_in      *dst;
 171 {
 172         int                     s;
 173         char                    indication[NDIS_INDICATION_LEN];
 174         struct ifreq            ifr;
 175         struct ndis_evt         *e;
 176         char                    buf[512], *pos, *end;
 177         int                     len, type, _type;
 178
 179         s = socket(PF_INET, SOCK_DGRAM, 0);
 180
 181         if (s < 0) {
 182                 dbgmsg("socket creation failed");
 183                 return(EINVAL);
 184         }
 185
 186         bzero((char *)&ifr, sizeof(ifr));
 187         e = (struct ndis_evt *)indication;
 188         e->ne_len = NDIS_INDICATION_LEN - sizeof(struct ndis_evt);
 189
 190         strlcpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name));
 191         ifr.ifr_data = indication;
 192
 193         if (ioctl(s, SIOCGPRIVATE_0, &ifr) < 0) {
 194                 close(s);
 195                 if (verbose) {
 196                         if (errno == ENOENT)
 197                                 dbgmsg("drained all events from %s",
 198                                     ifname, errno);
 199                         else
 200                                 dbgmsg("failed to read event info from %s: %d",
 201                                     ifname, errno);
 202                 }
 203                 return(ENOENT);
 204         }
 205
 206         if (e->ne_sts == NDIS_STATUS_MEDIA_CONNECT) {
 207                 type = EVENT_CONNECT;
 208                 if (verbose)
 209                         dbgmsg("Received a connect event for %s", ifname);
 210                 if (!all_events) {
 211                         close(s);
 212                         return(0);
 213                 }
 214         }
 215         if (e->ne_sts == NDIS_STATUS_MEDIA_DISCONNECT) {
 216                 type = EVENT_DISCONNECT;
 217                 if (verbose)
 218                         dbgmsg("Received a disconnect event for %s", ifname);
 219                 if (!all_events) {
 220                         close(s);
 221                         return(0);
 222                 }
 223         }
 224         if (e->ne_sts == NDIS_STATUS_MEDIA_SPECIFIC_INDICATION) {
 225                 type = EVENT_MEDIA_SPECIFIC;
 226                 if (verbose)
 227                         dbgmsg("Received a media-specific event for %s",
 228                             ifname);
 229         }
 230
 231         end = buf + sizeof(buf);
 232         _type = (int) type;
 233         memcpy(buf, &_type, sizeof(_type));
 234         pos = buf + sizeof(_type);
 235
 236         len = snprintf(pos + 1, end - pos - 1, "%s", ifname);
 237         if (len < 0) {
 238                 close(s);
 239                 return(ENOSPC);
 240         }
 241         if (len > 255)
 242                 len = 255;
 243         *pos = (unsigned char) len;
 244         pos += 1 + len;
 245         if (e->ne_len) {
 246                 if (e->ne_len > 255 || 1 + e->ne_len > end - pos) {
 247                         dbgmsg("Not enough room for send_event data (%d)\n",
 248                             e->ne_len);
 249                         close(s);
 250                         return(ENOSPC);
 251                 }
 252                 *pos++ = (unsigned char) e->ne_len;
 253                 memcpy(pos, (indication) + sizeof(struct ndis_evt), e->ne_len);
 254                 pos += e->ne_len;
 255         }
 256
 257         len = sendto(sock, buf, pos - buf, 0, (struct sockaddr *) dst,
 258             sizeof(struct sockaddr_in));
 259
 260         close(s);
 261         return(0);
 262 }
 263
 264 static void
 265 usage()
 266 {
 267         fprintf(stderr, "Usage: ndis_events [-a] [-d] [-v]\n");
 268         exit(1);
 269 }
 270
 271 int
 272 main(argc, argv)
 273         int                     argc;
 274         char                    *argv[];
 275 {
 276         int                     s, r, n;
 277         struct sockaddr_in      sin;
 278         char                    msg[NDIS_INDICATION_LEN];
 279         struct rt_msghdr        *rtm;
 280         struct if_msghdr        *ifm;
 281         char                    ifname[IFNAMSIZ];
 282         int                     ch;
 283
 284         while ((ch = getopt(argc, argv, "dva")) != -1) {
 285                 switch(ch) {
 286                 case 'd':
 287                         debug++;
 288                         break;
 289                 case 'v':
 290                         verbose++;
 291                         break;
 292                 case 'a':
 293                         all_events++;
 294                         break;
 295                 default:
 296                         usage();
 297                         break;
 298                 }
 299         }
 300
 301         if (!debug && daemon(0, 0))
 302                 err(1, "failed to daemonize ourselves");
 303
 304         if (!debug)
 305                 openlog(PROGNAME, LOG_PID | LOG_CONS, LOG_DAEMON);
 306
 307         bzero((char *)&sin, sizeof(sin));
 308
 309         /* Create a datagram  socket. */
 310
 311         s = socket(PF_INET, SOCK_DGRAM, 0);
 312         if (s < 0) {
 313                 dbgmsg("socket creation failed");
 314                 exit(1);
 315         }
 316
 317         sin.sin_family = AF_INET;
 318         sin.sin_addr.s_addr = inet_addr("127.0.0.1");
 319         sin.sin_port = htons(WPA_SUPPLICANT_PORT);
 320
 321         /* Create a routing socket. */
 322
 323         r = socket (PF_ROUTE, SOCK_RAW, 0);
 324         if (r < 0) {
 325                 dbgmsg("routing socket creation failed");
 326                 exit(1);
 327         }
 328
 329         /* Now sit and spin, waiting for events. */
 330
 331         if (verbose)
 332                 dbgmsg("Listening for events");
 333
 334         while (1) {
 335                 n = read(r, msg, NDIS_INDICATION_LEN);
 336                 rtm = (struct rt_msghdr *)msg;
 337                 if (rtm->rtm_type != RTM_IFINFO)
 338                         continue;
 339                 ifm = (struct if_msghdr *)msg;
 340                 if (find_ifname(ifm->ifm_index, ifname))
 341                         continue;
 342                 if (strstr(ifname, "ndis")) {
 343                         while(announce_event(ifname, s, &sin) == 0)
 344                                 ;
 345                 } else {
 346                         if (verbose)
 347                                 dbgmsg("Skipping ifinfo message from %s",
 348                                     ifname);
 349                 }
 350         }
 351
 352         /* NOTREACHED */
 353         exit(0);
 354 }