2331139fcfe75a0d33273ff04f7fec5d9a10ad26
[wpasupplicant] / src / drivers / driver_nl80211.c
1 /*
2  * WPA Supplicant - driver interaction with Linux nl80211/cfg80211
3  * Copyright (c) 2003-2008, Jouni Malinen <j@w1.fi>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation.
8  *
9  * Alternatively, this software may be distributed under the terms of BSD
10  * license.
11  *
12  * See README and COPYING for more details.
13  */
14
15 #include "includes.h"
16 #include <sys/ioctl.h>
17 #include <net/if_arp.h>
18 #include <netlink/genl/genl.h>
19 #include <netlink/genl/family.h>
20 #include <netlink/genl/ctrl.h>
21 #include "nl80211_copy.h"
22 #ifdef CONFIG_CLIENT_MLME
23 #include <netpacket/packet.h>
24 #include <linux/if_ether.h>
25 #include "radiotap.h"
26 #include "radiotap_iter.h"
27 #endif /* CONFIG_CLIENT_MLME */
28
29 #include "wireless_copy.h"
30 #include "common.h"
31 #include "driver.h"
32 #include "eloop.h"
33 #include "ieee802_11_defs.h"
34
35 #ifndef IFF_LOWER_UP
36 #define IFF_LOWER_UP   0x10000         /* driver signals L1 up         */
37 #endif
38 #ifndef IFF_DORMANT
39 #define IFF_DORMANT    0x20000         /* driver signals dormant       */
40 #endif
41
42 #ifndef IF_OPER_DORMANT
43 #define IF_OPER_DORMANT 5
44 #endif
45 #ifndef IF_OPER_UP
46 #define IF_OPER_UP 6
47 #endif
48
49
50 struct wpa_driver_nl80211_data {
51         void *ctx;
52         int event_sock;
53         int ioctl_sock;
54         char ifname[IFNAMSIZ + 1];
55         int ifindex;
56         int if_removed;
57         u8 *assoc_req_ies;
58         size_t assoc_req_ies_len;
59         u8 *assoc_resp_ies;
60         size_t assoc_resp_ies_len;
61         struct wpa_driver_capa capa;
62         int has_capability;
63         int we_version_compiled;
64
65         /* for set_auth_alg fallback */
66         int use_crypt;
67         int auth_alg_fallback;
68
69         int operstate;
70
71         char mlmedev[IFNAMSIZ + 1];
72
73         int scan_complete_events;
74
75         struct nl_handle *nl_handle;
76         struct nl_cache *nl_cache;
77         struct nl_cb *nl_cb;
78         struct genl_family *nl80211;
79
80 #ifdef CONFIG_CLIENT_MLME
81         int monitor_sock; /* socket for monitor */
82         int monitor_ifidx;
83 #endif /* CONFIG_CLIENT_MLME */
84 };
85
86
87 static void wpa_driver_nl80211_scan_timeout(void *eloop_ctx,
88                                             void *timeout_ctx);
89 static int wpa_driver_nl80211_set_mode(void *priv, int mode);
90 static int wpa_driver_nl80211_flush_pmkid(void *priv);
91 static int wpa_driver_nl80211_get_range(void *priv);
92 static void
93 wpa_driver_nl80211_finish_drv_init(struct wpa_driver_nl80211_data *drv);
94
95
96 /* nl80211 code */
97 static int ack_handler(struct nl_msg *msg, void *arg)
98 {
99         int *err = arg;
100         *err = 0;
101         return NL_STOP;
102 }
103
104 static int finish_handler(struct nl_msg *msg, void *arg)
105 {
106         int *ret = arg;
107         *ret = 0;
108         return NL_SKIP;
109 }
110
111 static int error_handler(struct sockaddr_nl *nla, struct nlmsgerr *err,
112                          void *arg)
113 {
114         int *ret = arg;
115         *ret = err->error;
116         return NL_SKIP;
117 }
118
119 static int send_and_recv_msgs(struct wpa_driver_nl80211_data *drv,
120                               struct nl_msg *msg,
121                               int (*valid_handler)(struct nl_msg *, void *),
122                               void *valid_data)
123 {
124         struct nl_cb *cb;
125         int err = -ENOMEM;
126
127         cb = nl_cb_clone(drv->nl_cb);
128         if (!cb)
129                 goto out;
130
131         err = nl_send_auto_complete(drv->nl_handle, msg);
132         if (err < 0)
133                 goto out;
134
135         err = 1;
136
137         nl_cb_err(cb, NL_CB_CUSTOM, error_handler, &err);
138         nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, finish_handler, &err);
139         nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ack_handler, &err);
140
141         if (valid_handler)
142                 nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM,
143                           valid_handler, valid_data);
144
145         while (err > 0)
146                 nl_recvmsgs(drv->nl_handle, cb);
147  out:
148         nl_cb_put(cb);
149         nlmsg_free(msg);
150         return err;
151 }
152
153
154 static int wpa_driver_nl80211_send_oper_ifla(
155         struct wpa_driver_nl80211_data *drv,
156         int linkmode, int operstate)
157 {
158         struct {
159                 struct nlmsghdr hdr;
160                 struct ifinfomsg ifinfo;
161                 char opts[16];
162         } req;
163         struct rtattr *rta;
164         static int nl_seq;
165         ssize_t ret;
166
167         os_memset(&req, 0, sizeof(req));
168
169         req.hdr.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifinfomsg));
170         req.hdr.nlmsg_type = RTM_SETLINK;
171         req.hdr.nlmsg_flags = NLM_F_REQUEST;
172         req.hdr.nlmsg_seq = ++nl_seq;
173         req.hdr.nlmsg_pid = 0;
174
175         req.ifinfo.ifi_family = AF_UNSPEC;
176         req.ifinfo.ifi_type = 0;
177         req.ifinfo.ifi_index = drv->ifindex;
178         req.ifinfo.ifi_flags = 0;
179         req.ifinfo.ifi_change = 0;
180
181         if (linkmode != -1) {
182                 rta = (struct rtattr *)
183                         ((char *) &req + NLMSG_ALIGN(req.hdr.nlmsg_len));
184                 rta->rta_type = IFLA_LINKMODE;
185                 rta->rta_len = RTA_LENGTH(sizeof(char));
186                 *((char *) RTA_DATA(rta)) = linkmode;
187                 req.hdr.nlmsg_len = NLMSG_ALIGN(req.hdr.nlmsg_len) +
188                         RTA_LENGTH(sizeof(char));
189         }
190         if (operstate != -1) {
191                 rta = (struct rtattr *)
192                         ((char *) &req + NLMSG_ALIGN(req.hdr.nlmsg_len));
193                 rta->rta_type = IFLA_OPERSTATE;
194                 rta->rta_len = RTA_LENGTH(sizeof(char));
195                 *((char *) RTA_DATA(rta)) = operstate;
196                 req.hdr.nlmsg_len = NLMSG_ALIGN(req.hdr.nlmsg_len) +
197                         RTA_LENGTH(sizeof(char));
198         }
199
200         wpa_printf(MSG_DEBUG, "WEXT: Operstate: linkmode=%d, operstate=%d",
201                    linkmode, operstate);
202
203         ret = send(drv->event_sock, &req, req.hdr.nlmsg_len, 0);
204         if (ret < 0) {
205                 wpa_printf(MSG_DEBUG, "WEXT: Sending operstate IFLA failed: "
206                            "%s (assume operstate is not supported)",
207                            strerror(errno));
208         }
209
210         return ret < 0 ? -1 : 0;
211 }
212
213
214 static int wpa_driver_nl80211_set_auth_param(
215         struct wpa_driver_nl80211_data *drv, int idx, u32 value)
216 {
217         struct iwreq iwr;
218         int ret = 0;
219
220         os_memset(&iwr, 0, sizeof(iwr));
221         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
222         iwr.u.param.flags = idx & IW_AUTH_INDEX;
223         iwr.u.param.value = value;
224
225         if (ioctl(drv->ioctl_sock, SIOCSIWAUTH, &iwr) < 0) {
226                 if (errno != EOPNOTSUPP) {
227                         wpa_printf(MSG_DEBUG, "WEXT: SIOCSIWAUTH(param %d "
228                                    "value 0x%x) failed: %s)",
229                                    idx, value, strerror(errno));
230                 }
231                 ret = errno == EOPNOTSUPP ? -2 : -1;
232         }
233
234         return ret;
235 }
236
237
238 static int wpa_driver_nl80211_get_bssid(void *priv, u8 *bssid)
239 {
240         struct wpa_driver_nl80211_data *drv = priv;
241         struct iwreq iwr;
242         int ret = 0;
243
244         os_memset(&iwr, 0, sizeof(iwr));
245         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
246
247         if (ioctl(drv->ioctl_sock, SIOCGIWAP, &iwr) < 0) {
248                 perror("ioctl[SIOCGIWAP]");
249                 ret = -1;
250         }
251         os_memcpy(bssid, iwr.u.ap_addr.sa_data, ETH_ALEN);
252
253         return ret;
254 }
255
256
257 static int wpa_driver_nl80211_set_bssid(void *priv, const u8 *bssid)
258 {
259         struct wpa_driver_nl80211_data *drv = priv;
260         struct iwreq iwr;
261         int ret = 0;
262
263         os_memset(&iwr, 0, sizeof(iwr));
264         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
265         iwr.u.ap_addr.sa_family = ARPHRD_ETHER;
266         if (bssid)
267                 os_memcpy(iwr.u.ap_addr.sa_data, bssid, ETH_ALEN);
268         else
269                 os_memset(iwr.u.ap_addr.sa_data, 0, ETH_ALEN);
270
271         if (ioctl(drv->ioctl_sock, SIOCSIWAP, &iwr) < 0) {
272                 perror("ioctl[SIOCSIWAP]");
273                 ret = -1;
274         }
275
276         return ret;
277 }
278
279
280 static int wpa_driver_nl80211_get_ssid(void *priv, u8 *ssid)
281 {
282         struct wpa_driver_nl80211_data *drv = priv;
283         struct iwreq iwr;
284         int ret = 0;
285
286         os_memset(&iwr, 0, sizeof(iwr));
287         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
288         iwr.u.essid.pointer = (caddr_t) ssid;
289         iwr.u.essid.length = 32;
290
291         if (ioctl(drv->ioctl_sock, SIOCGIWESSID, &iwr) < 0) {
292                 perror("ioctl[SIOCGIWESSID]");
293                 ret = -1;
294         } else {
295                 ret = iwr.u.essid.length;
296                 if (ret > 32)
297                         ret = 32;
298                 /* Some drivers include nul termination in the SSID, so let's
299                  * remove it here before further processing. WE-21 changes this
300                  * to explicitly require the length _not_ to include nul
301                  * termination. */
302                 if (ret > 0 && ssid[ret - 1] == '\0' &&
303                     drv->we_version_compiled < 21)
304                         ret--;
305         }
306
307         return ret;
308 }
309
310
311 static int wpa_driver_nl80211_set_ssid(void *priv, const u8 *ssid,
312                                        size_t ssid_len)
313 {
314         struct wpa_driver_nl80211_data *drv = priv;
315         struct iwreq iwr;
316         int ret = 0;
317         char buf[33];
318
319         if (ssid_len > 32)
320                 return -1;
321
322         os_memset(&iwr, 0, sizeof(iwr));
323         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
324         /* flags: 1 = ESSID is active, 0 = not (promiscuous) */
325         iwr.u.essid.flags = (ssid_len != 0);
326         os_memset(buf, 0, sizeof(buf));
327         os_memcpy(buf, ssid, ssid_len);
328         iwr.u.essid.pointer = (caddr_t) buf;
329         if (drv->we_version_compiled < 21) {
330                 /* For historic reasons, set SSID length to include one extra
331                  * character, C string nul termination, even though SSID is
332                  * really an octet string that should not be presented as a C
333                  * string. Some Linux drivers decrement the length by one and
334                  * can thus end up missing the last octet of the SSID if the
335                  * length is not incremented here. WE-21 changes this to
336                  * explicitly require the length _not_ to include nul
337                  * termination. */
338                 if (ssid_len)
339                         ssid_len++;
340         }
341         iwr.u.essid.length = ssid_len;
342
343         if (ioctl(drv->ioctl_sock, SIOCSIWESSID, &iwr) < 0) {
344                 perror("ioctl[SIOCSIWESSID]");
345                 ret = -1;
346         }
347
348         return ret;
349 }
350
351
352 static int wpa_driver_nl80211_set_freq(void *priv, int freq)
353 {
354         struct wpa_driver_nl80211_data *drv = priv;
355         struct iwreq iwr;
356         int ret = 0;
357
358         os_memset(&iwr, 0, sizeof(iwr));
359         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
360         iwr.u.freq.m = freq * 100000;
361         iwr.u.freq.e = 1;
362
363         if (ioctl(drv->ioctl_sock, SIOCSIWFREQ, &iwr) < 0) {
364                 perror("ioctl[SIOCSIWFREQ]");
365                 ret = -1;
366         }
367
368         return ret;
369 }
370
371
372 static void
373 wpa_driver_nl80211_event_wireless_custom(void *ctx, char *custom)
374 {
375         union wpa_event_data data;
376
377         wpa_printf(MSG_MSGDUMP, "WEXT: Custom wireless event: '%s'",
378                    custom);
379
380         os_memset(&data, 0, sizeof(data));
381         /* Host AP driver */
382         if (os_strncmp(custom, "MLME-MICHAELMICFAILURE.indication", 33) == 0) {
383                 data.michael_mic_failure.unicast =
384                         os_strstr(custom, " unicast ") != NULL;
385                 /* TODO: parse parameters(?) */
386                 wpa_supplicant_event(ctx, EVENT_MICHAEL_MIC_FAILURE, &data);
387         } else if (os_strncmp(custom, "ASSOCINFO(ReqIEs=", 17) == 0) {
388                 char *spos;
389                 int bytes;
390
391                 spos = custom + 17;
392
393                 bytes = strspn(spos, "0123456789abcdefABCDEF");
394                 if (!bytes || (bytes & 1))
395                         return;
396                 bytes /= 2;
397
398                 data.assoc_info.req_ies = os_malloc(bytes);
399                 if (data.assoc_info.req_ies == NULL)
400                         return;
401
402                 data.assoc_info.req_ies_len = bytes;
403                 hexstr2bin(spos, data.assoc_info.req_ies, bytes);
404
405                 spos += bytes * 2;
406
407                 data.assoc_info.resp_ies = NULL;
408                 data.assoc_info.resp_ies_len = 0;
409
410                 if (os_strncmp(spos, " RespIEs=", 9) == 0) {
411                         spos += 9;
412
413                         bytes = strspn(spos, "0123456789abcdefABCDEF");
414                         if (!bytes || (bytes & 1))
415                                 goto done;
416                         bytes /= 2;
417
418                         data.assoc_info.resp_ies = os_malloc(bytes);
419                         if (data.assoc_info.resp_ies == NULL)
420                                 goto done;
421
422                         data.assoc_info.resp_ies_len = bytes;
423                         hexstr2bin(spos, data.assoc_info.resp_ies, bytes);
424                 }
425
426                 wpa_supplicant_event(ctx, EVENT_ASSOCINFO, &data);
427
428         done:
429                 os_free(data.assoc_info.resp_ies);
430                 os_free(data.assoc_info.req_ies);
431 #ifdef CONFIG_PEERKEY
432         } else if (os_strncmp(custom, "STKSTART.request=", 17) == 0) {
433                 if (hwaddr_aton(custom + 17, data.stkstart.peer)) {
434                         wpa_printf(MSG_DEBUG, "WEXT: unrecognized "
435                                    "STKSTART.request '%s'", custom + 17);
436                         return;
437                 }
438                 wpa_supplicant_event(ctx, EVENT_STKSTART, &data);
439 #endif /* CONFIG_PEERKEY */
440         }
441 }
442
443
444 static int wpa_driver_nl80211_event_wireless_michaelmicfailure(
445         void *ctx, const char *ev, size_t len)
446 {
447         const struct iw_michaelmicfailure *mic;
448         union wpa_event_data data;
449
450         if (len < sizeof(*mic))
451                 return -1;
452
453         mic = (const struct iw_michaelmicfailure *) ev;
454
455         wpa_printf(MSG_DEBUG, "Michael MIC failure wireless event: "
456                    "flags=0x%x src_addr=" MACSTR, mic->flags,
457                    MAC2STR(mic->src_addr.sa_data));
458
459         os_memset(&data, 0, sizeof(data));
460         data.michael_mic_failure.unicast = !(mic->flags & IW_MICFAILURE_GROUP);
461         wpa_supplicant_event(ctx, EVENT_MICHAEL_MIC_FAILURE, &data);
462
463         return 0;
464 }
465
466
467 static int wpa_driver_nl80211_event_wireless_pmkidcand(
468         struct wpa_driver_nl80211_data *drv, const char *ev, size_t len)
469 {
470         const struct iw_pmkid_cand *cand;
471         union wpa_event_data data;
472         const u8 *addr;
473
474         if (len < sizeof(*cand))
475                 return -1;
476
477         cand = (const struct iw_pmkid_cand *) ev;
478         addr = (const u8 *) cand->bssid.sa_data;
479
480         wpa_printf(MSG_DEBUG, "PMKID candidate wireless event: "
481                    "flags=0x%x index=%d bssid=" MACSTR, cand->flags,
482                    cand->index, MAC2STR(addr));
483
484         os_memset(&data, 0, sizeof(data));
485         os_memcpy(data.pmkid_candidate.bssid, addr, ETH_ALEN);
486         data.pmkid_candidate.index = cand->index;
487         data.pmkid_candidate.preauth = cand->flags & IW_PMKID_CAND_PREAUTH;
488         wpa_supplicant_event(drv->ctx, EVENT_PMKID_CANDIDATE, &data);
489
490         return 0;
491 }
492
493
494 static int wpa_driver_nl80211_event_wireless_assocreqie(
495         struct wpa_driver_nl80211_data *drv, const char *ev, int len)
496 {
497         if (len < 0)
498                 return -1;
499
500         wpa_hexdump(MSG_DEBUG, "AssocReq IE wireless event", (const u8 *) ev,
501                     len);
502         os_free(drv->assoc_req_ies);
503         drv->assoc_req_ies = os_malloc(len);
504         if (drv->assoc_req_ies == NULL) {
505                 drv->assoc_req_ies_len = 0;
506                 return -1;
507         }
508         os_memcpy(drv->assoc_req_ies, ev, len);
509         drv->assoc_req_ies_len = len;
510
511         return 0;
512 }
513
514
515 static int wpa_driver_nl80211_event_wireless_assocrespie(
516         struct wpa_driver_nl80211_data *drv, const char *ev, int len)
517 {
518         if (len < 0)
519                 return -1;
520
521         wpa_hexdump(MSG_DEBUG, "AssocResp IE wireless event", (const u8 *) ev,
522                     len);
523         os_free(drv->assoc_resp_ies);
524         drv->assoc_resp_ies = os_malloc(len);
525         if (drv->assoc_resp_ies == NULL) {
526                 drv->assoc_resp_ies_len = 0;
527                 return -1;
528         }
529         os_memcpy(drv->assoc_resp_ies, ev, len);
530         drv->assoc_resp_ies_len = len;
531
532         return 0;
533 }
534
535
536 static void wpa_driver_nl80211_event_assoc_ies(struct wpa_driver_nl80211_data *drv)
537 {
538         union wpa_event_data data;
539
540         if (drv->assoc_req_ies == NULL && drv->assoc_resp_ies == NULL)
541                 return;
542
543         os_memset(&data, 0, sizeof(data));
544         if (drv->assoc_req_ies) {
545                 data.assoc_info.req_ies = drv->assoc_req_ies;
546                 drv->assoc_req_ies = NULL;
547                 data.assoc_info.req_ies_len = drv->assoc_req_ies_len;
548         }
549         if (drv->assoc_resp_ies) {
550                 data.assoc_info.resp_ies = drv->assoc_resp_ies;
551                 drv->assoc_resp_ies = NULL;
552                 data.assoc_info.resp_ies_len = drv->assoc_resp_ies_len;
553         }
554
555         wpa_supplicant_event(drv->ctx, EVENT_ASSOCINFO, &data);
556
557         os_free(data.assoc_info.req_ies);
558         os_free(data.assoc_info.resp_ies);
559 }
560
561
562 static void wpa_driver_nl80211_event_wireless(struct wpa_driver_nl80211_data *drv,
563                                            void *ctx, char *data, int len)
564 {
565         struct iw_event iwe_buf, *iwe = &iwe_buf;
566         char *pos, *end, *custom, *buf;
567
568         pos = data;
569         end = data + len;
570
571         while (pos + IW_EV_LCP_LEN <= end) {
572                 /* Event data may be unaligned, so make a local, aligned copy
573                  * before processing. */
574                 os_memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
575                 wpa_printf(MSG_DEBUG, "Wireless event: cmd=0x%x len=%d",
576                            iwe->cmd, iwe->len);
577                 if (iwe->len <= IW_EV_LCP_LEN)
578                         return;
579
580                 custom = pos + IW_EV_POINT_LEN;
581                 if (drv->we_version_compiled > 18 &&
582                     (iwe->cmd == IWEVMICHAELMICFAILURE ||
583                      iwe->cmd == IWEVCUSTOM ||
584                      iwe->cmd == IWEVASSOCREQIE ||
585                      iwe->cmd == IWEVASSOCRESPIE ||
586                      iwe->cmd == IWEVPMKIDCAND)) {
587                         /* WE-19 removed the pointer from struct iw_point */
588                         char *dpos = (char *) &iwe_buf.u.data.length;
589                         int dlen = dpos - (char *) &iwe_buf;
590                         os_memcpy(dpos, pos + IW_EV_LCP_LEN,
591                                   sizeof(struct iw_event) - dlen);
592                 } else {
593                         os_memcpy(&iwe_buf, pos, sizeof(struct iw_event));
594                         custom += IW_EV_POINT_OFF;
595                 }
596
597                 switch (iwe->cmd) {
598                 case SIOCGIWAP:
599                         wpa_printf(MSG_DEBUG, "Wireless event: new AP: "
600                                    MACSTR,
601                                    MAC2STR((u8 *) iwe->u.ap_addr.sa_data));
602                         if (is_zero_ether_addr(
603                                     (const u8 *) iwe->u.ap_addr.sa_data) ||
604                             os_memcmp(iwe->u.ap_addr.sa_data,
605                                       "\x44\x44\x44\x44\x44\x44", ETH_ALEN) ==
606                             0) {
607                                 os_free(drv->assoc_req_ies);
608                                 drv->assoc_req_ies = NULL;
609                                 os_free(drv->assoc_resp_ies);
610                                 drv->assoc_resp_ies = NULL;
611                                 wpa_supplicant_event(ctx, EVENT_DISASSOC,
612                                                      NULL);
613                         
614                         } else {
615                                 wpa_driver_nl80211_event_assoc_ies(drv);
616                                 wpa_supplicant_event(ctx, EVENT_ASSOC, NULL);
617                         }
618                         break;
619                 case IWEVMICHAELMICFAILURE:
620                         wpa_driver_nl80211_event_wireless_michaelmicfailure(
621                                 ctx, custom, iwe->u.data.length);
622                         break;
623                 case IWEVCUSTOM:
624                         if (custom + iwe->u.data.length > end)
625                                 return;
626                         buf = os_malloc(iwe->u.data.length + 1);
627                         if (buf == NULL)
628                                 return;
629                         os_memcpy(buf, custom, iwe->u.data.length);
630                         buf[iwe->u.data.length] = '\0';
631                         wpa_driver_nl80211_event_wireless_custom(ctx, buf);
632                         os_free(buf);
633                         break;
634                 case SIOCGIWSCAN:
635                         drv->scan_complete_events = 1;
636                         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout,
637                                              drv, ctx);
638                         wpa_supplicant_event(ctx, EVENT_SCAN_RESULTS, NULL);
639                         break;
640                 case IWEVASSOCREQIE:
641                         wpa_driver_nl80211_event_wireless_assocreqie(
642                                 drv, custom, iwe->u.data.length);
643                         break;
644                 case IWEVASSOCRESPIE:
645                         wpa_driver_nl80211_event_wireless_assocrespie(
646                                 drv, custom, iwe->u.data.length);
647                         break;
648                 case IWEVPMKIDCAND:
649                         wpa_driver_nl80211_event_wireless_pmkidcand(
650                                 drv, custom, iwe->u.data.length);
651                         break;
652                 }
653
654                 pos += iwe->len;
655         }
656 }
657
658
659 static void wpa_driver_nl80211_event_link(struct wpa_driver_nl80211_data *drv,
660                                           void *ctx, char *buf, size_t len,
661                                           int del)
662 {
663         union wpa_event_data event;
664
665         os_memset(&event, 0, sizeof(event));
666         if (len > sizeof(event.interface_status.ifname))
667                 len = sizeof(event.interface_status.ifname) - 1;
668         os_memcpy(event.interface_status.ifname, buf, len);
669         event.interface_status.ievent = del ? EVENT_INTERFACE_REMOVED :
670                 EVENT_INTERFACE_ADDED;
671
672         wpa_printf(MSG_DEBUG, "RTM_%sLINK, IFLA_IFNAME: Interface '%s' %s",
673                    del ? "DEL" : "NEW",
674                    event.interface_status.ifname,
675                    del ? "removed" : "added");
676
677         if (os_strcmp(drv->ifname, event.interface_status.ifname) == 0) {
678                 if (del)
679                         drv->if_removed = 1;
680                 else
681                         drv->if_removed = 0;
682         }
683
684         wpa_supplicant_event(ctx, EVENT_INTERFACE_STATUS, &event);
685 }
686
687
688 static int wpa_driver_nl80211_own_ifname(struct wpa_driver_nl80211_data *drv,
689                                          struct nlmsghdr *h)
690 {
691         struct ifinfomsg *ifi;
692         int attrlen, _nlmsg_len, rta_len;
693         struct rtattr *attr;
694
695         ifi = NLMSG_DATA(h);
696
697         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
698
699         attrlen = h->nlmsg_len - _nlmsg_len;
700         if (attrlen < 0)
701                 return 0;
702
703         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
704
705         rta_len = RTA_ALIGN(sizeof(struct rtattr));
706         while (RTA_OK(attr, attrlen)) {
707                 if (attr->rta_type == IFLA_IFNAME) {
708                         if (os_strcmp(((char *) attr) + rta_len, drv->ifname)
709                             == 0)
710                                 return 1;
711                         else
712                                 break;
713                 }
714                 attr = RTA_NEXT(attr, attrlen);
715         }
716
717         return 0;
718 }
719
720
721 static int wpa_driver_nl80211_own_ifindex(struct wpa_driver_nl80211_data *drv,
722                                           int ifindex, struct nlmsghdr *h)
723 {
724         if (drv->ifindex == ifindex)
725                 return 1;
726
727         if (drv->if_removed && wpa_driver_nl80211_own_ifname(drv, h)) {
728                 drv->ifindex = if_nametoindex(drv->ifname);
729                 wpa_printf(MSG_DEBUG, "nl80211: Update ifindex for a removed "
730                            "interface");
731                 wpa_driver_nl80211_finish_drv_init(drv);
732                 return 1;
733         }
734
735         return 0;
736 }
737
738
739 static void wpa_driver_nl80211_event_rtm_newlink(struct wpa_driver_nl80211_data *drv,
740                                               void *ctx, struct nlmsghdr *h,
741                                               size_t len)
742 {
743         struct ifinfomsg *ifi;
744         int attrlen, _nlmsg_len, rta_len;
745         struct rtattr * attr;
746
747         if (len < sizeof(*ifi))
748                 return;
749
750         ifi = NLMSG_DATA(h);
751
752         if (!wpa_driver_nl80211_own_ifindex(drv, ifi->ifi_index, h)) {
753                 wpa_printf(MSG_DEBUG, "Ignore event for foreign ifindex %d",
754                            ifi->ifi_index);
755                 return;
756         }
757
758         wpa_printf(MSG_DEBUG, "RTM_NEWLINK: operstate=%d ifi_flags=0x%x "
759                    "(%s%s%s%s)",
760                    drv->operstate, ifi->ifi_flags,
761                    (ifi->ifi_flags & IFF_UP) ? "[UP]" : "",
762                    (ifi->ifi_flags & IFF_RUNNING) ? "[RUNNING]" : "",
763                    (ifi->ifi_flags & IFF_LOWER_UP) ? "[LOWER_UP]" : "",
764                    (ifi->ifi_flags & IFF_DORMANT) ? "[DORMANT]" : "");
765         /*
766          * Some drivers send the association event before the operup event--in
767          * this case, lifting operstate in wpa_driver_nl80211_set_operstate()
768          * fails. This will hit us when wpa_supplicant does not need to do
769          * IEEE 802.1X authentication
770          */
771         if (drv->operstate == 1 &&
772             (ifi->ifi_flags & (IFF_LOWER_UP | IFF_DORMANT)) == IFF_LOWER_UP &&
773             !(ifi->ifi_flags & IFF_RUNNING))
774                 wpa_driver_nl80211_send_oper_ifla(drv, -1, IF_OPER_UP);
775
776         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
777
778         attrlen = h->nlmsg_len - _nlmsg_len;
779         if (attrlen < 0)
780                 return;
781
782         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
783
784         rta_len = RTA_ALIGN(sizeof(struct rtattr));
785         while (RTA_OK(attr, attrlen)) {
786                 if (attr->rta_type == IFLA_WIRELESS) {
787                         wpa_driver_nl80211_event_wireless(
788                                 drv, ctx, ((char *) attr) + rta_len,
789                                 attr->rta_len - rta_len);
790                 } else if (attr->rta_type == IFLA_IFNAME) {
791                         wpa_driver_nl80211_event_link(
792                                 drv, ctx,
793                                 ((char *) attr) + rta_len,
794                                 attr->rta_len - rta_len, 0);
795                 }
796                 attr = RTA_NEXT(attr, attrlen);
797         }
798 }
799
800
801 static void wpa_driver_nl80211_event_rtm_dellink(struct wpa_driver_nl80211_data *drv,
802                                               void *ctx, struct nlmsghdr *h,
803                                               size_t len)
804 {
805         struct ifinfomsg *ifi;
806         int attrlen, _nlmsg_len, rta_len;
807         struct rtattr * attr;
808
809         if (len < sizeof(*ifi))
810                 return;
811
812         ifi = NLMSG_DATA(h);
813
814         _nlmsg_len = NLMSG_ALIGN(sizeof(struct ifinfomsg));
815
816         attrlen = h->nlmsg_len - _nlmsg_len;
817         if (attrlen < 0)
818                 return;
819
820         attr = (struct rtattr *) (((char *) ifi) + _nlmsg_len);
821
822         rta_len = RTA_ALIGN(sizeof(struct rtattr));
823         while (RTA_OK(attr, attrlen)) {
824                 if (attr->rta_type == IFLA_IFNAME) {
825                         wpa_driver_nl80211_event_link(
826                                 drv, ctx,
827                                 ((char *) attr) + rta_len,
828                                 attr->rta_len - rta_len, 1);
829                 }
830                 attr = RTA_NEXT(attr, attrlen);
831         }
832 }
833
834
835 static void wpa_driver_nl80211_event_receive(int sock, void *eloop_ctx,
836                                           void *sock_ctx)
837 {
838         char buf[8192];
839         int left;
840         struct sockaddr_nl from;
841         socklen_t fromlen;
842         struct nlmsghdr *h;
843         int max_events = 10;
844
845 try_again:
846         fromlen = sizeof(from);
847         left = recvfrom(sock, buf, sizeof(buf), MSG_DONTWAIT,
848                         (struct sockaddr *) &from, &fromlen);
849         if (left < 0) {
850                 if (errno != EINTR && errno != EAGAIN)
851                         perror("recvfrom(netlink)");
852                 return;
853         }
854
855         h = (struct nlmsghdr *) buf;
856         while (left >= (int) sizeof(*h)) {
857                 int len, plen;
858
859                 len = h->nlmsg_len;
860                 plen = len - sizeof(*h);
861                 if (len > left || plen < 0) {
862                         wpa_printf(MSG_DEBUG, "Malformed netlink message: "
863                                    "len=%d left=%d plen=%d",
864                                    len, left, plen);
865                         break;
866                 }
867
868                 switch (h->nlmsg_type) {
869                 case RTM_NEWLINK:
870                         wpa_driver_nl80211_event_rtm_newlink(eloop_ctx, sock_ctx,
871                                                           h, plen);
872                         break;
873                 case RTM_DELLINK:
874                         wpa_driver_nl80211_event_rtm_dellink(eloop_ctx, sock_ctx,
875                                                           h, plen);
876                         break;
877                 }
878
879                 len = NLMSG_ALIGN(len);
880                 left -= len;
881                 h = (struct nlmsghdr *) ((char *) h + len);
882         }
883
884         if (left > 0) {
885                 wpa_printf(MSG_DEBUG, "%d extra bytes in the end of netlink "
886                            "message", left);
887         }
888
889         if (--max_events > 0) {
890                 /*
891                  * Try to receive all events in one eloop call in order to
892                  * limit race condition on cases where AssocInfo event, Assoc
893                  * event, and EAPOL frames are received more or less at the
894                  * same time. We want to process the event messages first
895                  * before starting EAPOL processing.
896                  */
897                 goto try_again;
898         }
899 }
900
901
902 static int wpa_driver_nl80211_get_ifflags_ifname(struct wpa_driver_nl80211_data *drv,
903                                               const char *ifname, int *flags)
904 {
905         struct ifreq ifr;
906
907         os_memset(&ifr, 0, sizeof(ifr));
908         os_strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
909         if (ioctl(drv->ioctl_sock, SIOCGIFFLAGS, (caddr_t) &ifr) < 0) {
910                 perror("ioctl[SIOCGIFFLAGS]");
911                 return -1;
912         }
913         *flags = ifr.ifr_flags & 0xffff;
914         return 0;
915 }
916
917
918 /**
919  * wpa_driver_nl80211_get_ifflags - Get interface flags (SIOCGIFFLAGS)
920  * @drv: driver_nl80211 private data
921  * @flags: Pointer to returned flags value
922  * Returns: 0 on success, -1 on failure
923  */
924 static int wpa_driver_nl80211_get_ifflags(struct wpa_driver_nl80211_data *drv,
925                                           int *flags)
926 {
927         return wpa_driver_nl80211_get_ifflags_ifname(drv, drv->ifname, flags);
928 }
929
930
931 static int wpa_driver_nl80211_set_ifflags_ifname(
932         struct wpa_driver_nl80211_data *drv,
933         const char *ifname, int flags)
934 {
935         struct ifreq ifr;
936
937         os_memset(&ifr, 0, sizeof(ifr));
938         os_strlcpy(ifr.ifr_name, ifname, IFNAMSIZ);
939         ifr.ifr_flags = flags & 0xffff;
940         if (ioctl(drv->ioctl_sock, SIOCSIFFLAGS, (caddr_t) &ifr) < 0) {
941                 perror("SIOCSIFFLAGS");
942                 return -1;
943         }
944         return 0;
945 }
946
947
948 /**
949  * wpa_driver_nl80211_set_ifflags - Set interface flags (SIOCSIFFLAGS)
950  * @drv: driver_nl80211 private data
951  * @flags: New value for flags
952  * Returns: 0 on success, -1 on failure
953  */
954 static int wpa_driver_nl80211_set_ifflags(struct wpa_driver_nl80211_data *drv,
955                                           int flags)
956 {
957         return wpa_driver_nl80211_set_ifflags_ifname(drv, drv->ifname, flags);
958 }
959
960
961 /**
962  * wpa_driver_nl80211_set_country - ask nl80211 to set the regulatory domain
963  * @priv: driver_nl80211 private data
964  * @alpha2_arg: country to which to switch to
965  * Returns: 0 on success, -1 on failure
966  *
967  * This asks nl80211 to set the regulatory domain for given
968  * country ISO / IEC alpha2.
969  */
970 static int wpa_driver_nl80211_set_country(void *priv, const char *alpha2_arg)
971 {
972         struct wpa_driver_nl80211_data *drv = priv;
973         char alpha2[3];
974         struct nl_msg *msg;
975
976         msg = nlmsg_alloc();
977         if (!msg)
978                 goto nla_put_failure;
979
980         alpha2[0] = alpha2_arg[0];
981         alpha2[1] = alpha2_arg[1];
982         alpha2[2] = '\0';
983
984         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
985                     0, NL80211_CMD_REQ_SET_REG, 0);
986
987         NLA_PUT_STRING(msg, NL80211_ATTR_REG_ALPHA2, alpha2);
988         if (send_and_recv_msgs(drv, msg, NULL, NULL))
989                 return -EINVAL;
990         return 0;
991 nla_put_failure:
992         return -EINVAL;
993 }
994
995
996 #ifdef CONFIG_CLIENT_MLME
997
998 static int nl80211_set_vif(struct wpa_driver_nl80211_data *drv,
999                            int drop_unencrypted, int userspace_mlme)
1000 {
1001 #ifdef NL80211_CMD_SET_VIF
1002         struct nl_msg *msg;
1003         int ret = -1;
1004
1005         msg = nlmsg_alloc();
1006         if (!msg)
1007                 return -ENOMEM;
1008
1009         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
1010                     NL80211_CMD_SET_VIF, 0);
1011
1012         if (drop_unencrypted >= 0)
1013                 NLA_PUT_U8(msg, NL80211_ATTR_VIF_DROP_UNENCRYPTED,
1014                            drop_unencrypted);
1015         if (userspace_mlme >= 0)
1016                 NLA_PUT_U8(msg, NL80211_ATTR_VIF_USERSPACE_MLME,
1017                            userspace_mlme);
1018
1019         ret = 0;
1020
1021         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
1022
1023         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1024         return ret;
1025
1026 nla_put_failure:
1027         return -ENOBUFS;
1028 #else /* NL80211_CMD_SET_VIF */
1029         return -1;
1030 #endif /* NL80211_CMD_SET_VIF */
1031 }
1032
1033
1034 static int wpa_driver_nl80211_set_userspace_mlme(
1035         struct wpa_driver_nl80211_data *drv, int enabled)
1036 {
1037         return nl80211_set_vif(drv, -1, enabled);
1038 }
1039
1040
1041 static void nl80211_remove_iface(struct wpa_driver_nl80211_data *drv,
1042                                  int ifidx)
1043 {
1044         struct nl_msg *msg;
1045
1046         msg = nlmsg_alloc();
1047         if (!msg)
1048                 goto nla_put_failure;
1049
1050         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1051                     0, NL80211_CMD_DEL_INTERFACE, 0);
1052         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, ifidx);
1053         if (send_and_recv_msgs(drv, msg, NULL, NULL) == 0)
1054                 return;
1055 nla_put_failure:
1056         wpa_printf(MSG_ERROR, "nl80211: Failed to remove interface.");
1057 }
1058
1059
1060 static int nl80211_create_iface(struct wpa_driver_nl80211_data *drv,
1061                                 const char *ifname, enum nl80211_iftype iftype)
1062 {
1063         struct nl_msg *msg, *flags = NULL;
1064         int ifidx, err;
1065         int ret = -ENOBUFS;
1066
1067         msg = nlmsg_alloc();
1068         if (!msg)
1069                 return -1;
1070
1071         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
1072                     0, NL80211_CMD_NEW_INTERFACE, 0);
1073         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, if_nametoindex(drv->ifname));
1074         NLA_PUT_STRING(msg, NL80211_ATTR_IFNAME, ifname);
1075         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE, iftype);
1076
1077         if (iftype == NL80211_IFTYPE_MONITOR) {
1078                 flags = nlmsg_alloc();
1079                 if (!flags)
1080                         goto nla_put_failure;
1081
1082                 NLA_PUT_FLAG(flags, NL80211_MNTR_FLAG_COOK_FRAMES);
1083
1084                 err = nla_put_nested(msg, NL80211_ATTR_MNTR_FLAGS, flags);
1085
1086                 nlmsg_free(flags);
1087
1088                 if (err)
1089                         goto nla_put_failure;
1090         }
1091
1092         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
1093         if (ret) {
1094         nla_put_failure:
1095                 wpa_printf(MSG_ERROR, "nl80211: Failed to create interface %d",
1096                            ret);
1097                 return ret;
1098         }
1099
1100         ifidx = if_nametoindex(ifname);
1101         if (ifidx <= 0)
1102                 return -1;
1103
1104         return ifidx;
1105 }
1106
1107
1108 static void handle_monitor_read(int sock, void *eloop_ctx, void *sock_ctx)
1109 {
1110         struct wpa_driver_nl80211_data *drv = eloop_ctx;
1111         int len;
1112         unsigned char buf[3000];
1113         struct ieee80211_radiotap_iterator iter;
1114         int ret;
1115         int injected = 0, failed = 0, rxflags = 0;
1116         struct ieee80211_rx_status rx_status;
1117
1118         len = recv(sock, buf, sizeof(buf), 0);
1119         if (len < 0) {
1120                 perror("recv");
1121                 return;
1122         }
1123
1124         if (ieee80211_radiotap_iterator_init(&iter, (void *) buf, len)) {
1125                 wpa_printf(MSG_DEBUG, "nl80211: received invalid radiotap "
1126                            "frame");
1127                 return;
1128         }
1129
1130         os_memset(&rx_status, 0, sizeof(rx_status));
1131
1132         while (1) {
1133                 ret = ieee80211_radiotap_iterator_next(&iter);
1134                 if (ret == -ENOENT)
1135                         break;
1136                 if (ret) {
1137                         wpa_printf(MSG_DEBUG, "nl80211: received invalid "
1138                                    "radiotap frame (%d)", ret);
1139                         return;
1140                 }
1141                 switch (iter.this_arg_index) {
1142                 case IEEE80211_RADIOTAP_FLAGS:
1143                         if (*iter.this_arg & IEEE80211_RADIOTAP_F_FCS)
1144                                 len -= 4;
1145                         break;
1146                 case IEEE80211_RADIOTAP_RX_FLAGS:
1147                         rxflags = 1;
1148                         break;
1149                 case IEEE80211_RADIOTAP_TX_FLAGS:
1150                         injected = 1;
1151                         failed = le_to_host16((*(u16 *) iter.this_arg)) &
1152                                 IEEE80211_RADIOTAP_F_TX_FAIL;
1153                         break;
1154                 case IEEE80211_RADIOTAP_DATA_RETRIES:
1155                         break;
1156                 case IEEE80211_RADIOTAP_CHANNEL:
1157                         /* TODO convert from freq/flags to channel number
1158                          * rx_status.channel = XXX;
1159                         */
1160                         break;
1161                 case IEEE80211_RADIOTAP_RATE:
1162                         break;
1163                 case IEEE80211_RADIOTAP_DB_ANTSIGNAL:
1164                         rx_status.ssi = *iter.this_arg;
1165                         break;
1166                 }
1167         }
1168
1169         if (rxflags && injected)
1170                 return;
1171
1172         if (!injected) {
1173                 wpa_supplicant_sta_rx(drv->ctx, buf + iter.max_length,
1174                                       len - iter.max_length, &rx_status);
1175         } else if (failed) {
1176                 /* TX failure callback */
1177         } else {
1178                 /* TX success (ACK) callback */
1179         }
1180 }
1181
1182
1183 static int wpa_driver_nl80211_create_monitor_interface(
1184         struct wpa_driver_nl80211_data *drv)
1185 {
1186         char buf[IFNAMSIZ];
1187         struct sockaddr_ll ll;
1188         int optval, flags;
1189         socklen_t optlen;
1190
1191         os_snprintf(buf, IFNAMSIZ, "mon.%s", drv->ifname);
1192         buf[IFNAMSIZ - 1] = '\0';
1193
1194         drv->monitor_ifidx =
1195                 nl80211_create_iface(drv, buf, NL80211_IFTYPE_MONITOR);
1196
1197         if (drv->monitor_ifidx < 0)
1198                 return -1;
1199
1200         if (wpa_driver_nl80211_get_ifflags_ifname(drv, buf, &flags) != 0 ||
1201             wpa_driver_nl80211_set_ifflags_ifname(drv, buf, flags | IFF_UP) !=
1202             0) {
1203                 wpa_printf(MSG_ERROR, "nl80211: Could not set interface '%s' "
1204                            "UP", buf);
1205                 goto error;
1206         }
1207
1208         os_memset(&ll, 0, sizeof(ll));
1209         ll.sll_family = AF_PACKET;
1210         ll.sll_ifindex = drv->monitor_ifidx;
1211         drv->monitor_sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));
1212         if (drv->monitor_sock < 0) {
1213                 perror("socket[PF_PACKET,SOCK_RAW]");
1214                 goto error;
1215         }
1216
1217         if (bind(drv->monitor_sock, (struct sockaddr *) &ll,
1218                  sizeof(ll)) < 0) {
1219                 perror("monitor socket bind");
1220                 goto error;
1221         }
1222
1223         optlen = sizeof(optval);
1224         optval = 20;
1225         if (setsockopt
1226             (drv->monitor_sock, SOL_SOCKET, SO_PRIORITY, &optval, optlen)) {
1227                 perror("Failed to set socket priority");
1228                 goto error;
1229         }
1230
1231         if (eloop_register_read_sock(drv->monitor_sock, handle_monitor_read,
1232                                      drv, NULL)) {
1233                 wpa_printf(MSG_ERROR, "nl80211: Could not register monitor "
1234                            "read socket");
1235                 goto error;
1236         }
1237
1238         return 0;
1239
1240  error:
1241         nl80211_remove_iface(drv, drv->monitor_ifidx);
1242         return -1;
1243 }
1244
1245 #endif /* CONFIG_CLIENT_MLME */
1246
1247
1248 /**
1249  * wpa_driver_nl80211_init - Initialize nl80211 driver interface
1250  * @ctx: context to be used when calling wpa_supplicant functions,
1251  * e.g., wpa_supplicant_event()
1252  * @ifname: interface name, e.g., wlan0
1253  * Returns: Pointer to private data, %NULL on failure
1254  */
1255 static void * wpa_driver_nl80211_init(void *ctx, const char *ifname)
1256 {
1257         int s;
1258         struct sockaddr_nl local;
1259         struct wpa_driver_nl80211_data *drv;
1260
1261         drv = os_zalloc(sizeof(*drv));
1262         if (drv == NULL)
1263                 return NULL;
1264         drv->ctx = ctx;
1265         os_strlcpy(drv->ifname, ifname, sizeof(drv->ifname));
1266
1267         drv->nl_cb = nl_cb_alloc(NL_CB_DEFAULT);
1268         if (drv->nl_cb == NULL) {
1269                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
1270                            "callbacks");
1271                 goto err1;
1272         }
1273
1274         drv->nl_handle = nl_handle_alloc_cb(drv->nl_cb);
1275         if (drv->nl_handle == NULL) {
1276                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate netlink "
1277                            "callbacks");
1278                 goto err2;
1279         }
1280
1281         if (genl_connect(drv->nl_handle)) {
1282                 wpa_printf(MSG_ERROR, "nl80211: Failed to connect to generic "
1283                            "netlink");
1284                 goto err3;
1285         }
1286
1287         drv->nl_cache = genl_ctrl_alloc_cache(drv->nl_handle);
1288         if (drv->nl_cache == NULL) {
1289                 wpa_printf(MSG_ERROR, "nl80211: Failed to allocate generic "
1290                            "netlink cache");
1291                 goto err3;
1292         }
1293
1294         drv->nl80211 = genl_ctrl_search_by_name(drv->nl_cache, "nl80211");
1295         if (drv->nl80211 == NULL) {
1296                 wpa_printf(MSG_ERROR, "nl80211: 'nl80211' generic netlink not "
1297                            "found");
1298                 goto err4;
1299         }
1300
1301         drv->ioctl_sock = socket(PF_INET, SOCK_DGRAM, 0);
1302         if (drv->ioctl_sock < 0) {
1303                 perror("socket(PF_INET,SOCK_DGRAM)");
1304                 goto err5;
1305         }
1306
1307         s = socket(PF_NETLINK, SOCK_RAW, NETLINK_ROUTE);
1308         if (s < 0) {
1309                 perror("socket(PF_NETLINK,SOCK_RAW,NETLINK_ROUTE)");
1310                 goto err6;
1311         }
1312
1313         os_memset(&local, 0, sizeof(local));
1314         local.nl_family = AF_NETLINK;
1315         local.nl_groups = RTMGRP_LINK;
1316         if (bind(s, (struct sockaddr *) &local, sizeof(local)) < 0) {
1317                 perror("bind(netlink)");
1318                 close(s);
1319                 goto err6;
1320         }
1321
1322         eloop_register_read_sock(s, wpa_driver_nl80211_event_receive, drv,
1323                                  ctx);
1324         drv->event_sock = s;
1325
1326         wpa_driver_nl80211_finish_drv_init(drv);
1327
1328         return drv;
1329
1330 err6:
1331         close(drv->ioctl_sock);
1332 err5:
1333         genl_family_put(drv->nl80211);
1334 err4:
1335         nl_cache_free(drv->nl_cache);
1336 err3:
1337         nl_handle_destroy(drv->nl_handle);
1338 err2:
1339         nl_cb_put(drv->nl_cb);
1340 err1:
1341         os_free(drv);
1342         return NULL;
1343 }
1344
1345
1346 static void
1347 wpa_driver_nl80211_finish_drv_init(struct wpa_driver_nl80211_data *drv)
1348 {
1349         int flags;
1350
1351         if (wpa_driver_nl80211_get_ifflags(drv, &flags) != 0)
1352                 printf("Could not get interface '%s' flags\n", drv->ifname);
1353         else if (!(flags & IFF_UP)) {
1354                 if (wpa_driver_nl80211_set_ifflags(drv, flags | IFF_UP) != 0) {
1355                         printf("Could not set interface '%s' UP\n",
1356                                drv->ifname);
1357                 } else {
1358                         /*
1359                          * Wait some time to allow driver to initialize before
1360                          * starting configuring the driver. This seems to be
1361                          * needed at least some drivers that load firmware etc.
1362                          * when the interface is set up.
1363                          */
1364                         wpa_printf(MSG_DEBUG, "Interface %s set UP - waiting "
1365                                    "a second for the driver to complete "
1366                                    "initialization", drv->ifname);
1367                         sleep(1);
1368                 }
1369         }
1370
1371         /*
1372          * Make sure that the driver does not have any obsolete PMKID entries.
1373          */
1374         wpa_driver_nl80211_flush_pmkid(drv);
1375
1376         if (wpa_driver_nl80211_set_mode(drv, 0) < 0) {
1377                 printf("Could not configure driver to use managed mode\n");
1378         }
1379
1380         wpa_driver_nl80211_get_range(drv);
1381
1382         drv->ifindex = if_nametoindex(drv->ifname);
1383
1384         wpa_driver_nl80211_send_oper_ifla(drv, 1, IF_OPER_DORMANT);
1385 }
1386
1387
1388 /**
1389  * wpa_driver_nl80211_deinit - Deinitialize nl80211 driver interface
1390  * @priv: Pointer to private nl80211 data from wpa_driver_nl80211_init()
1391  *
1392  * Shut down driver interface and processing of driver events. Free
1393  * private data buffer if one was allocated in wpa_driver_nl80211_init().
1394  */
1395 static void wpa_driver_nl80211_deinit(void *priv)
1396 {
1397         struct wpa_driver_nl80211_data *drv = priv;
1398         int flags;
1399
1400 #ifdef CONFIG_CLIENT_MLME
1401         if (drv->monitor_sock >= 0) {
1402                 eloop_unregister_read_sock(drv->monitor_sock);
1403                 close(drv->monitor_sock);
1404         }
1405         if (drv->monitor_ifidx > 0)
1406                 nl80211_remove_iface(drv, drv->monitor_ifidx);
1407         if (drv->capa.flags & WPA_DRIVER_FLAGS_USER_SPACE_MLME)
1408                 wpa_driver_nl80211_set_userspace_mlme(drv, 0);
1409 #endif /* CONFIG_CLIENT_MLME */
1410
1411         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
1412
1413         /*
1414          * Clear possibly configured driver parameters in order to make it
1415          * easier to use the driver after wpa_supplicant has been terminated.
1416          */
1417         (void) wpa_driver_nl80211_set_bssid(drv,
1418                                          (u8 *) "\x00\x00\x00\x00\x00\x00");
1419
1420         wpa_driver_nl80211_send_oper_ifla(priv, 0, IF_OPER_UP);
1421
1422         eloop_unregister_read_sock(drv->event_sock);
1423
1424         if (wpa_driver_nl80211_get_ifflags(drv, &flags) == 0)
1425                 (void) wpa_driver_nl80211_set_ifflags(drv, flags & ~IFF_UP);
1426
1427         close(drv->event_sock);
1428         close(drv->ioctl_sock);
1429         os_free(drv->assoc_req_ies);
1430         os_free(drv->assoc_resp_ies);
1431
1432         genl_family_put(drv->nl80211);
1433         nl_cache_free(drv->nl_cache);
1434         nl_handle_destroy(drv->nl_handle);
1435         nl_cb_put(drv->nl_cb);
1436
1437         os_free(drv);
1438 }
1439
1440
1441 /**
1442  * wpa_driver_nl80211_scan_timeout - Scan timeout to report scan completion
1443  * @eloop_ctx: Unused
1444  * @timeout_ctx: ctx argument given to wpa_driver_nl80211_init()
1445  *
1446  * This function can be used as registered timeout when starting a scan to
1447  * generate a scan completed event if the driver does not report this.
1448  */
1449 static void wpa_driver_nl80211_scan_timeout(void *eloop_ctx, void *timeout_ctx)
1450 {
1451         wpa_printf(MSG_DEBUG, "Scan timeout - try to get results");
1452         wpa_supplicant_event(timeout_ctx, EVENT_SCAN_RESULTS, NULL);
1453 }
1454
1455
1456 /**
1457  * wpa_driver_nl80211_scan - Request the driver to initiate scan
1458  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
1459  * @ssid: Specific SSID to scan for (ProbeReq) or %NULL to scan for
1460  *      all SSIDs (either active scan with broadcast SSID or passive
1461  *      scan
1462  * @ssid_len: Length of the SSID
1463  * Returns: 0 on success, -1 on failure
1464  */
1465 static int wpa_driver_nl80211_scan(void *priv, const u8 *ssid, size_t ssid_len)
1466 {
1467         struct wpa_driver_nl80211_data *drv = priv;
1468         struct iwreq iwr;
1469         int ret = 0, timeout;
1470         struct iw_scan_req req;
1471
1472         if (ssid_len > IW_ESSID_MAX_SIZE) {
1473                 wpa_printf(MSG_DEBUG, "%s: too long SSID (%lu)",
1474                            __FUNCTION__, (unsigned long) ssid_len);
1475                 return -1;
1476         }
1477
1478         os_memset(&iwr, 0, sizeof(iwr));
1479         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
1480
1481         if (ssid && ssid_len) {
1482                 os_memset(&req, 0, sizeof(req));
1483                 req.essid_len = ssid_len;
1484                 req.bssid.sa_family = ARPHRD_ETHER;
1485                 os_memset(req.bssid.sa_data, 0xff, ETH_ALEN);
1486                 os_memcpy(req.essid, ssid, ssid_len);
1487                 iwr.u.data.pointer = (caddr_t) &req;
1488                 iwr.u.data.length = sizeof(req);
1489                 iwr.u.data.flags = IW_SCAN_THIS_ESSID;
1490         }
1491
1492         if (ioctl(drv->ioctl_sock, SIOCSIWSCAN, &iwr) < 0) {
1493                 perror("ioctl[SIOCSIWSCAN]");
1494                 ret = -1;
1495         }
1496
1497         /* Not all drivers generate "scan completed" wireless event, so try to
1498          * read results after a timeout. */
1499         timeout = 5;
1500         if (drv->scan_complete_events) {
1501                 /*
1502                  * The driver seems to deliver SIOCGIWSCAN events to notify
1503                  * when scan is complete, so use longer timeout to avoid race
1504                  * conditions with scanning and following association request.
1505                  */
1506                 timeout = 30;
1507         }
1508         wpa_printf(MSG_DEBUG, "Scan requested (ret=%d) - scan timeout %d "
1509                    "seconds", ret, timeout);
1510         eloop_cancel_timeout(wpa_driver_nl80211_scan_timeout, drv, drv->ctx);
1511         eloop_register_timeout(timeout, 0, wpa_driver_nl80211_scan_timeout, drv,
1512                                drv->ctx);
1513
1514         return ret;
1515 }
1516
1517
1518 static u8 * wpa_driver_nl80211_giwscan(struct wpa_driver_nl80211_data *drv,
1519                                     size_t *len)
1520 {
1521         struct iwreq iwr;
1522         u8 *res_buf;
1523         size_t res_buf_len;
1524
1525         res_buf_len = IW_SCAN_MAX_DATA;
1526         for (;;) {
1527                 res_buf = os_malloc(res_buf_len);
1528                 if (res_buf == NULL)
1529                         return NULL;
1530                 os_memset(&iwr, 0, sizeof(iwr));
1531                 os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
1532                 iwr.u.data.pointer = res_buf;
1533                 iwr.u.data.length = res_buf_len;
1534
1535                 if (ioctl(drv->ioctl_sock, SIOCGIWSCAN, &iwr) == 0)
1536                         break;
1537
1538                 if (errno == E2BIG && res_buf_len < 100000) {
1539                         os_free(res_buf);
1540                         res_buf = NULL;
1541                         res_buf_len *= 2;
1542                         wpa_printf(MSG_DEBUG, "Scan results did not fit - "
1543                                    "trying larger buffer (%lu bytes)",
1544                                    (unsigned long) res_buf_len);
1545                 } else {
1546                         perror("ioctl[SIOCGIWSCAN]");
1547                         os_free(res_buf);
1548                         return NULL;
1549                 }
1550         }
1551
1552         if (iwr.u.data.length > res_buf_len) {
1553                 os_free(res_buf);
1554                 return NULL;
1555         }
1556         *len = iwr.u.data.length;
1557
1558         return res_buf;
1559 }
1560
1561
1562 /*
1563  * Data structure for collecting WEXT scan results. This is needed to allow
1564  * the various methods of reporting IEs to be combined into a single IE buffer.
1565  */
1566 struct wext_scan_data {
1567         struct wpa_scan_res res;
1568         u8 *ie;
1569         size_t ie_len;
1570         u8 ssid[32];
1571         size_t ssid_len;
1572         int maxrate;
1573 };
1574
1575
1576 static void wext_get_scan_mode(struct iw_event *iwe,
1577                                struct wext_scan_data *res)
1578 {
1579         if (iwe->u.mode == IW_MODE_ADHOC)
1580                 res->res.caps |= IEEE80211_CAP_IBSS;
1581         else if (iwe->u.mode == IW_MODE_MASTER || iwe->u.mode == IW_MODE_INFRA)
1582                 res->res.caps |= IEEE80211_CAP_ESS;
1583 }
1584
1585
1586 static void wext_get_scan_ssid(struct iw_event *iwe,
1587                                struct wext_scan_data *res, char *custom,
1588                                char *end)
1589 {
1590         int ssid_len = iwe->u.essid.length;
1591         if (custom + ssid_len > end)
1592                 return;
1593         if (iwe->u.essid.flags &&
1594             ssid_len > 0 &&
1595             ssid_len <= IW_ESSID_MAX_SIZE) {
1596                 os_memcpy(res->ssid, custom, ssid_len);
1597                 res->ssid_len = ssid_len;
1598         }
1599 }
1600
1601
1602 static void wext_get_scan_freq(struct iw_event *iwe,
1603                                struct wext_scan_data *res)
1604 {
1605         int divi = 1000000, i;
1606
1607         if (iwe->u.freq.e == 0) {
1608                 /*
1609                  * Some drivers do not report frequency, but a channel.
1610                  * Try to map this to frequency by assuming they are using
1611                  * IEEE 802.11b/g.  But don't overwrite a previously parsed
1612                  * frequency if the driver sends both frequency and channel,
1613                  * since the driver may be sending an A-band channel that we
1614                  * don't handle here.
1615                  */
1616
1617                 if (res->res.freq)
1618                         return;
1619
1620                 if (iwe->u.freq.m >= 1 && iwe->u.freq.m <= 13) {
1621                         res->res.freq = 2407 + 5 * iwe->u.freq.m;
1622                         return;
1623                 } else if (iwe->u.freq.m == 14) {
1624                         res->res.freq = 2484;
1625                         return;
1626                 }
1627         }
1628
1629         if (iwe->u.freq.e > 6) {
1630                 wpa_printf(MSG_DEBUG, "Invalid freq in scan results (BSSID="
1631                            MACSTR " m=%d e=%d)",
1632                            MAC2STR(res->res.bssid), iwe->u.freq.m,
1633                            iwe->u.freq.e);
1634                 return;
1635         }
1636
1637         for (i = 0; i < iwe->u.freq.e; i++)
1638                 divi /= 10;
1639         res->res.freq = iwe->u.freq.m / divi;
1640 }
1641
1642
1643 static void wext_get_scan_qual(struct iw_event *iwe,
1644                                struct wext_scan_data *res)
1645 {
1646         res->res.qual = iwe->u.qual.qual;
1647         res->res.noise = iwe->u.qual.noise;
1648         res->res.level = iwe->u.qual.level;
1649 }
1650
1651
1652 static void wext_get_scan_encode(struct iw_event *iwe,
1653                                  struct wext_scan_data *res)
1654 {
1655         if (!(iwe->u.data.flags & IW_ENCODE_DISABLED))
1656                 res->res.caps |= IEEE80211_CAP_PRIVACY;
1657 }
1658
1659
1660 static void wext_get_scan_rate(struct iw_event *iwe,
1661                                struct wext_scan_data *res, char *pos,
1662                                char *end)
1663 {
1664         int maxrate;
1665         char *custom = pos + IW_EV_LCP_LEN;
1666         struct iw_param p;
1667         size_t clen;
1668
1669         clen = iwe->len;
1670         if (custom + clen > end)
1671                 return;
1672         maxrate = 0;
1673         while (((ssize_t) clen) >= (ssize_t) sizeof(struct iw_param)) {
1674                 /* Note: may be misaligned, make a local, aligned copy */
1675                 os_memcpy(&p, custom, sizeof(struct iw_param));
1676                 if (p.value > maxrate)
1677                         maxrate = p.value;
1678                 clen -= sizeof(struct iw_param);
1679                 custom += sizeof(struct iw_param);
1680         }
1681
1682         /* Convert the maxrate from WE-style (b/s units) to
1683          * 802.11 rates (500000 b/s units).
1684          */
1685         res->maxrate = maxrate / 500000;
1686 }
1687
1688
1689 static void wext_get_scan_iwevgenie(struct iw_event *iwe,
1690                                     struct wext_scan_data *res, char *custom,
1691                                     char *end)
1692 {
1693         char *genie, *gpos, *gend;
1694         u8 *tmp;
1695
1696         if (iwe->u.data.length == 0)
1697                 return;
1698
1699         gpos = genie = custom;
1700         gend = genie + iwe->u.data.length;
1701         if (gend > end) {
1702                 wpa_printf(MSG_INFO, "IWEVGENIE overflow");
1703                 return;
1704         }
1705
1706         tmp = os_realloc(res->ie, res->ie_len + gend - gpos);
1707         if (tmp == NULL)
1708                 return;
1709         os_memcpy(tmp + res->ie_len, gpos, gend - gpos);
1710         res->ie = tmp;
1711         res->ie_len += gend - gpos;
1712 }
1713
1714
1715 static void wext_get_scan_custom(struct iw_event *iwe,
1716                                  struct wext_scan_data *res, char *custom,
1717                                  char *end)
1718 {
1719         size_t clen;
1720         u8 *tmp;
1721
1722         clen = iwe->u.data.length;
1723         if (custom + clen > end)
1724                 return;
1725
1726         if (clen > 7 && os_strncmp(custom, "wpa_ie=", 7) == 0) {
1727                 char *spos;
1728                 int bytes;
1729                 spos = custom + 7;
1730                 bytes = custom + clen - spos;
1731                 if (bytes & 1 || bytes == 0)
1732                         return;
1733                 bytes /= 2;
1734                 tmp = os_realloc(res->ie, res->ie_len + bytes);
1735                 if (tmp == NULL)
1736                         return;
1737                 hexstr2bin(spos, tmp + res->ie_len, bytes);
1738                 res->ie = tmp;
1739                 res->ie_len += bytes;
1740         } else if (clen > 7 && os_strncmp(custom, "rsn_ie=", 7) == 0) {
1741                 char *spos;
1742                 int bytes;
1743                 spos = custom + 7;
1744                 bytes = custom + clen - spos;
1745                 if (bytes & 1 || bytes == 0)
1746                         return;
1747                 bytes /= 2;
1748                 tmp = os_realloc(res->ie, res->ie_len + bytes);
1749                 if (tmp == NULL)
1750                         return;
1751                 hexstr2bin(spos, tmp + res->ie_len, bytes);
1752                 res->ie = tmp;
1753                 res->ie_len += bytes;
1754         } else if (clen > 4 && os_strncmp(custom, "tsf=", 4) == 0) {
1755                 char *spos;
1756                 int bytes;
1757                 u8 bin[8];
1758                 spos = custom + 4;
1759                 bytes = custom + clen - spos;
1760                 if (bytes != 16) {
1761                         wpa_printf(MSG_INFO, "Invalid TSF length (%d)", bytes);
1762                         return;
1763                 }
1764                 bytes /= 2;
1765                 hexstr2bin(spos, bin, bytes);
1766                 res->res.tsf += WPA_GET_BE64(bin);
1767         }
1768 }
1769
1770
1771 static int wext_19_iw_point(struct wpa_driver_nl80211_data *drv, u16 cmd)
1772 {
1773         return drv->we_version_compiled > 18 &&
1774                 (cmd == SIOCGIWESSID || cmd == SIOCGIWENCODE ||
1775                  cmd == IWEVGENIE || cmd == IWEVCUSTOM);
1776 }
1777
1778
1779 static void wpa_driver_nl80211_add_scan_entry(struct wpa_scan_results *res,
1780                                            struct wext_scan_data *data)
1781 {
1782         struct wpa_scan_res **tmp;
1783         struct wpa_scan_res *r;
1784         size_t extra_len;
1785         u8 *pos, *end, *ssid_ie = NULL, *rate_ie = NULL;
1786
1787         /* Figure out whether we need to fake any IEs */
1788         pos = data->ie;
1789         end = pos + data->ie_len;
1790         while (pos && pos + 1 < end) {
1791                 if (pos + 2 + pos[1] > end)
1792                         break;
1793                 if (pos[0] == WLAN_EID_SSID)
1794                         ssid_ie = pos;
1795                 else if (pos[0] == WLAN_EID_SUPP_RATES)
1796                         rate_ie = pos;
1797                 else if (pos[0] == WLAN_EID_EXT_SUPP_RATES)
1798                         rate_ie = pos;
1799                 pos += 2 + pos[1];
1800         }
1801
1802         extra_len = 0;
1803         if (ssid_ie == NULL)
1804                 extra_len += 2 + data->ssid_len;
1805         if (rate_ie == NULL && data->maxrate)
1806                 extra_len += 3;
1807
1808         r = os_zalloc(sizeof(*r) + extra_len + data->ie_len);
1809         if (r == NULL)
1810                 return;
1811         os_memcpy(r, &data->res, sizeof(*r));
1812         r->ie_len = extra_len + data->ie_len;
1813         pos = (u8 *) (r + 1);
1814         if (ssid_ie == NULL) {
1815                 /*
1816                  * Generate a fake SSID IE since the driver did not report
1817                  * a full IE list.
1818                  */
1819                 *pos++ = WLAN_EID_SSID;
1820                 *pos++ = data->ssid_len;
1821                 os_memcpy(pos, data->ssid, data->ssid_len);
1822                 pos += data->ssid_len;
1823         }
1824         if (rate_ie == NULL && data->maxrate) {
1825                 /*
1826                  * Generate a fake Supported Rates IE since the driver did not
1827                  * report a full IE list.
1828                  */
1829                 *pos++ = WLAN_EID_SUPP_RATES;
1830                 *pos++ = 1;
1831                 *pos++ = data->maxrate;
1832         }
1833         if (data->ie)
1834                 os_memcpy(pos, data->ie, data->ie_len);
1835
1836         tmp = os_realloc(res->res,
1837                          (res->num + 1) * sizeof(struct wpa_scan_res *));
1838         if (tmp == NULL) {
1839                 os_free(r);
1840                 return;
1841         }
1842         tmp[res->num++] = r;
1843         res->res = tmp;
1844 }
1845                                       
1846
1847 /**
1848  * wpa_driver_nl80211_get_scan_results - Fetch the latest scan results
1849  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
1850  * Returns: Scan results on success, -1 on failure
1851  */
1852 static struct wpa_scan_results *
1853 wpa_driver_nl80211_get_scan_results(void *priv)
1854 {
1855         struct wpa_driver_nl80211_data *drv = priv;
1856         size_t ap_num = 0, len;
1857         int first;
1858         u8 *res_buf;
1859         struct iw_event iwe_buf, *iwe = &iwe_buf;
1860         char *pos, *end, *custom;
1861         struct wpa_scan_results *res;
1862         struct wext_scan_data data;
1863
1864         res_buf = wpa_driver_nl80211_giwscan(drv, &len);
1865         if (res_buf == NULL)
1866                 return NULL;
1867
1868         ap_num = 0;
1869         first = 1;
1870
1871         res = os_zalloc(sizeof(*res));
1872         if (res == NULL) {
1873                 os_free(res_buf);
1874                 return NULL;
1875         }
1876
1877         pos = (char *) res_buf;
1878         end = (char *) res_buf + len;
1879         os_memset(&data, 0, sizeof(data));
1880
1881         while (pos + IW_EV_LCP_LEN <= end) {
1882                 /* Event data may be unaligned, so make a local, aligned copy
1883                  * before processing. */
1884                 os_memcpy(&iwe_buf, pos, IW_EV_LCP_LEN);
1885                 if (iwe->len <= IW_EV_LCP_LEN)
1886                         break;
1887
1888                 custom = pos + IW_EV_POINT_LEN;
1889                 if (wext_19_iw_point(drv, iwe->cmd)) {
1890                         /* WE-19 removed the pointer from struct iw_point */
1891                         char *dpos = (char *) &iwe_buf.u.data.length;
1892                         int dlen = dpos - (char *) &iwe_buf;
1893                         os_memcpy(dpos, pos + IW_EV_LCP_LEN,
1894                                   sizeof(struct iw_event) - dlen);
1895                 } else {
1896                         os_memcpy(&iwe_buf, pos, sizeof(struct iw_event));
1897                         custom += IW_EV_POINT_OFF;
1898                 }
1899
1900                 switch (iwe->cmd) {
1901                 case SIOCGIWAP:
1902                         if (!first)
1903                                 wpa_driver_nl80211_add_scan_entry(res, &data);
1904                         first = 0;
1905                         os_free(data.ie);
1906                         os_memset(&data, 0, sizeof(data));
1907                         os_memcpy(data.res.bssid,
1908                                   iwe->u.ap_addr.sa_data, ETH_ALEN);
1909                         break;
1910                 case SIOCGIWMODE:
1911                         wext_get_scan_mode(iwe, &data);
1912                         break;
1913                 case SIOCGIWESSID:
1914                         wext_get_scan_ssid(iwe, &data, custom, end);
1915                         break;
1916                 case SIOCGIWFREQ:
1917                         wext_get_scan_freq(iwe, &data);
1918                         break;
1919                 case IWEVQUAL:
1920                         wext_get_scan_qual(iwe, &data);
1921                         break;
1922                 case SIOCGIWENCODE:
1923                         wext_get_scan_encode(iwe, &data);
1924                         break;
1925                 case SIOCGIWRATE:
1926                         wext_get_scan_rate(iwe, &data, pos, end);
1927                         break;
1928                 case IWEVGENIE:
1929                         wext_get_scan_iwevgenie(iwe, &data, custom, end);
1930                         break;
1931                 case IWEVCUSTOM:
1932                         wext_get_scan_custom(iwe, &data, custom, end);
1933                         break;
1934                 }
1935
1936                 pos += iwe->len;
1937         }
1938         os_free(res_buf);
1939         res_buf = NULL;
1940         if (!first)
1941                 wpa_driver_nl80211_add_scan_entry(res, &data);
1942         os_free(data.ie);
1943
1944         wpa_printf(MSG_DEBUG, "Received %lu bytes of scan results (%lu BSSes)",
1945                    (unsigned long) len, (unsigned long) res->num);
1946
1947         return res;
1948 }
1949
1950
1951 static int wpa_driver_nl80211_get_range(void *priv)
1952 {
1953         struct wpa_driver_nl80211_data *drv = priv;
1954         struct iw_range *range;
1955         struct iwreq iwr;
1956         int minlen;
1957         size_t buflen;
1958
1959         /*
1960          * Use larger buffer than struct iw_range in order to allow the
1961          * structure to grow in the future.
1962          */
1963         buflen = sizeof(struct iw_range) + 500;
1964         range = os_zalloc(buflen);
1965         if (range == NULL)
1966                 return -1;
1967
1968         os_memset(&iwr, 0, sizeof(iwr));
1969         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
1970         iwr.u.data.pointer = (caddr_t) range;
1971         iwr.u.data.length = buflen;
1972
1973         minlen = ((char *) &range->enc_capa) - (char *) range +
1974                 sizeof(range->enc_capa);
1975
1976         if (ioctl(drv->ioctl_sock, SIOCGIWRANGE, &iwr) < 0) {
1977                 perror("ioctl[SIOCGIWRANGE]");
1978                 os_free(range);
1979                 return -1;
1980         } else if (iwr.u.data.length >= minlen &&
1981                    range->we_version_compiled >= 18) {
1982                 wpa_printf(MSG_DEBUG, "SIOCGIWRANGE: WE(compiled)=%d "
1983                            "WE(source)=%d enc_capa=0x%x",
1984                            range->we_version_compiled,
1985                            range->we_version_source,
1986                            range->enc_capa);
1987                 drv->has_capability = 1;
1988                 drv->we_version_compiled = range->we_version_compiled;
1989                 if (range->enc_capa & IW_ENC_CAPA_WPA) {
1990                         drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA |
1991                                 WPA_DRIVER_CAPA_KEY_MGMT_WPA_PSK;
1992                 }
1993                 if (range->enc_capa & IW_ENC_CAPA_WPA2) {
1994                         drv->capa.key_mgmt |= WPA_DRIVER_CAPA_KEY_MGMT_WPA2 |
1995                                 WPA_DRIVER_CAPA_KEY_MGMT_WPA2_PSK;
1996                 }
1997                 drv->capa.enc |= WPA_DRIVER_CAPA_ENC_WEP40 |
1998                         WPA_DRIVER_CAPA_ENC_WEP104;
1999                 if (range->enc_capa & IW_ENC_CAPA_CIPHER_TKIP)
2000                         drv->capa.enc |= WPA_DRIVER_CAPA_ENC_TKIP;
2001                 if (range->enc_capa & IW_ENC_CAPA_CIPHER_CCMP)
2002                         drv->capa.enc |= WPA_DRIVER_CAPA_ENC_CCMP;
2003                 wpa_printf(MSG_DEBUG, "  capabilities: key_mgmt 0x%x enc 0x%x",
2004                            drv->capa.key_mgmt, drv->capa.enc);
2005         } else {
2006                 wpa_printf(MSG_DEBUG, "SIOCGIWRANGE: too old (short) data - "
2007                            "assuming WPA is not supported");
2008         }
2009
2010         os_free(range);
2011         return 0;
2012 }
2013
2014
2015 static int wpa_driver_nl80211_set_wpa(void *priv, int enabled)
2016 {
2017         struct wpa_driver_nl80211_data *drv = priv;
2018         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
2019
2020         return wpa_driver_nl80211_set_auth_param(drv, IW_AUTH_WPA_ENABLED,
2021                                               enabled);
2022 }
2023
2024
2025 static int wpa_driver_nl80211_set_key(void *priv, wpa_alg alg,
2026                                       const u8 *addr, int key_idx,
2027                                       int set_tx, const u8 *seq,
2028                                       size_t seq_len,
2029                                       const u8 *key, size_t key_len)
2030 {
2031         struct wpa_driver_nl80211_data *drv = priv;
2032         int err;
2033         struct nl_msg *msg;
2034
2035         wpa_printf(MSG_DEBUG, "%s: alg=%d addr=%p key_idx=%d set_tx=%d "
2036                    "seq_len=%lu key_len=%lu",
2037                    __func__, alg, addr, key_idx, set_tx,
2038                    (unsigned long) seq_len, (unsigned long) key_len);
2039
2040         msg = nlmsg_alloc();
2041         if (msg == NULL)
2042                 return -1;
2043
2044         if (alg == WPA_ALG_NONE) {
2045                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
2046                             NL80211_CMD_DEL_KEY, 0);
2047         } else {
2048                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0, 0,
2049                             NL80211_CMD_NEW_KEY, 0);
2050                 NLA_PUT(msg, NL80211_ATTR_KEY_DATA, key_len, key);
2051                 switch (alg) {
2052                 case WPA_ALG_WEP:
2053                         if (key_len == 5)
2054                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
2055                                             0x000FAC01);
2056                         else
2057                                 NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER,
2058                                             0x000FAC05);
2059                         break;
2060                 case WPA_ALG_TKIP:
2061                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC02);
2062                         break;
2063                 case WPA_ALG_CCMP:
2064                         NLA_PUT_U32(msg, NL80211_ATTR_KEY_CIPHER, 0x000FAC04);
2065                         break;
2066                 default:
2067                         nlmsg_free(msg);
2068                         return -1;
2069                 }
2070         }
2071
2072         if (addr && os_memcmp(addr, "\xff\xff\xff\xff\xff\xff", ETH_ALEN) != 0)
2073         {
2074                 wpa_printf(MSG_DEBUG, "   addr=" MACSTR, MAC2STR(addr));
2075                 NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
2076         }
2077         NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
2078         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2079
2080         err = send_and_recv_msgs(drv, msg, NULL, NULL);
2081         if (err) {
2082                 wpa_printf(MSG_DEBUG, "nl80211: set_key failed; err=%d", err);
2083                 return -1;
2084         }
2085
2086         if (set_tx && alg != WPA_ALG_NONE) {
2087                 msg = nlmsg_alloc();
2088                 if (msg == NULL)
2089                         return -1;
2090
2091                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2092                             0, NL80211_CMD_SET_KEY, 0);
2093                 NLA_PUT_U8(msg, NL80211_ATTR_KEY_IDX, key_idx);
2094                 NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2095                 NLA_PUT_FLAG(msg, NL80211_ATTR_KEY_DEFAULT);
2096
2097                 err = send_and_recv_msgs(drv, msg, NULL, NULL);
2098                 if (err) {
2099                         wpa_printf(MSG_DEBUG, "nl80211: set default key "
2100                                    "failed; err=%d", err);
2101                         return -1;
2102                 }
2103         }
2104
2105         return 0;
2106
2107 nla_put_failure:
2108         return -ENOBUFS;
2109 }
2110
2111
2112 static int wpa_driver_nl80211_set_countermeasures(void *priv,
2113                                                int enabled)
2114 {
2115         struct wpa_driver_nl80211_data *drv = priv;
2116         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
2117         return wpa_driver_nl80211_set_auth_param(drv,
2118                                               IW_AUTH_TKIP_COUNTERMEASURES,
2119                                               enabled);
2120 }
2121
2122
2123 static int wpa_driver_nl80211_set_drop_unencrypted(void *priv,
2124                                                 int enabled)
2125 {
2126         struct wpa_driver_nl80211_data *drv = priv;
2127         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
2128         drv->use_crypt = enabled;
2129         return wpa_driver_nl80211_set_auth_param(drv, IW_AUTH_DROP_UNENCRYPTED,
2130                                               enabled);
2131 }
2132
2133
2134 static int wpa_driver_nl80211_mlme(struct wpa_driver_nl80211_data *drv,
2135                                 const u8 *addr, int cmd, int reason_code)
2136 {
2137         struct iwreq iwr;
2138         struct iw_mlme mlme;
2139         int ret = 0;
2140
2141         os_memset(&iwr, 0, sizeof(iwr));
2142         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
2143         os_memset(&mlme, 0, sizeof(mlme));
2144         mlme.cmd = cmd;
2145         mlme.reason_code = reason_code;
2146         mlme.addr.sa_family = ARPHRD_ETHER;
2147         os_memcpy(mlme.addr.sa_data, addr, ETH_ALEN);
2148         iwr.u.data.pointer = (caddr_t) &mlme;
2149         iwr.u.data.length = sizeof(mlme);
2150
2151         if (ioctl(drv->ioctl_sock, SIOCSIWMLME, &iwr) < 0) {
2152                 perror("ioctl[SIOCSIWMLME]");
2153                 ret = -1;
2154         }
2155
2156         return ret;
2157 }
2158
2159
2160 static int wpa_driver_nl80211_deauthenticate(void *priv, const u8 *addr,
2161                                           int reason_code)
2162 {
2163         struct wpa_driver_nl80211_data *drv = priv;
2164         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
2165         return wpa_driver_nl80211_mlme(drv, addr, IW_MLME_DEAUTH, reason_code);
2166 }
2167
2168
2169 static int wpa_driver_nl80211_disassociate(void *priv, const u8 *addr,
2170                                         int reason_code)
2171 {
2172         struct wpa_driver_nl80211_data *drv = priv;
2173         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
2174         return wpa_driver_nl80211_mlme(drv, addr, IW_MLME_DISASSOC,
2175                                     reason_code);
2176 }
2177
2178
2179 static int wpa_driver_nl80211_set_gen_ie(void *priv, const u8 *ie,
2180                                       size_t ie_len)
2181 {
2182         struct wpa_driver_nl80211_data *drv = priv;
2183         struct iwreq iwr;
2184         int ret = 0;
2185
2186         os_memset(&iwr, 0, sizeof(iwr));
2187         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
2188         iwr.u.data.pointer = (caddr_t) ie;
2189         iwr.u.data.length = ie_len;
2190
2191         if (ioctl(drv->ioctl_sock, SIOCSIWGENIE, &iwr) < 0) {
2192                 perror("ioctl[SIOCSIWGENIE]");
2193                 ret = -1;
2194         }
2195
2196         return ret;
2197 }
2198
2199
2200 static int wpa_driver_nl80211_cipher2wext(int cipher)
2201 {
2202         switch (cipher) {
2203         case CIPHER_NONE:
2204                 return IW_AUTH_CIPHER_NONE;
2205         case CIPHER_WEP40:
2206                 return IW_AUTH_CIPHER_WEP40;
2207         case CIPHER_TKIP:
2208                 return IW_AUTH_CIPHER_TKIP;
2209         case CIPHER_CCMP:
2210                 return IW_AUTH_CIPHER_CCMP;
2211         case CIPHER_WEP104:
2212                 return IW_AUTH_CIPHER_WEP104;
2213         default:
2214                 return 0;
2215         }
2216 }
2217
2218
2219 static int wpa_driver_nl80211_keymgmt2wext(int keymgmt)
2220 {
2221         switch (keymgmt) {
2222         case KEY_MGMT_802_1X:
2223         case KEY_MGMT_802_1X_NO_WPA:
2224                 return IW_AUTH_KEY_MGMT_802_1X;
2225         case KEY_MGMT_PSK:
2226                 return IW_AUTH_KEY_MGMT_PSK;
2227         default:
2228                 return 0;
2229         }
2230 }
2231
2232
2233 static int
2234 wpa_driver_nl80211_auth_alg_fallback(struct wpa_driver_nl80211_data *drv,
2235                                   struct wpa_driver_associate_params *params)
2236 {
2237         struct iwreq iwr;
2238         int ret = 0;
2239
2240         wpa_printf(MSG_DEBUG, "WEXT: Driver did not support "
2241                    "SIOCSIWAUTH for AUTH_ALG, trying SIOCSIWENCODE");
2242
2243         os_memset(&iwr, 0, sizeof(iwr));
2244         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
2245         /* Just changing mode, not actual keys */
2246         iwr.u.encoding.flags = 0;
2247         iwr.u.encoding.pointer = (caddr_t) NULL;
2248         iwr.u.encoding.length = 0;
2249
2250         /*
2251          * Note: IW_ENCODE_{OPEN,RESTRICTED} can be interpreted to mean two
2252          * different things. Here they are used to indicate Open System vs.
2253          * Shared Key authentication algorithm. However, some drivers may use
2254          * them to select between open/restricted WEP encrypted (open = allow
2255          * both unencrypted and encrypted frames; restricted = only allow
2256          * encrypted frames).
2257          */
2258
2259         if (!drv->use_crypt) {
2260                 iwr.u.encoding.flags |= IW_ENCODE_DISABLED;
2261         } else {
2262                 if (params->auth_alg & AUTH_ALG_OPEN_SYSTEM)
2263                         iwr.u.encoding.flags |= IW_ENCODE_OPEN;
2264                 if (params->auth_alg & AUTH_ALG_SHARED_KEY)
2265                         iwr.u.encoding.flags |= IW_ENCODE_RESTRICTED;
2266         }
2267
2268         if (ioctl(drv->ioctl_sock, SIOCSIWENCODE, &iwr) < 0) {
2269                 perror("ioctl[SIOCSIWENCODE]");
2270                 ret = -1;
2271         }
2272
2273         return ret;
2274 }
2275
2276
2277 static int wpa_driver_nl80211_associate(
2278         void *priv, struct wpa_driver_associate_params *params)
2279 {
2280         struct wpa_driver_nl80211_data *drv = priv;
2281         int ret = 0;
2282         int allow_unencrypted_eapol;
2283         int value;
2284
2285         wpa_printf(MSG_DEBUG, "%s", __FUNCTION__);
2286
2287         /*
2288          * If the driver did not support SIOCSIWAUTH, fallback to
2289          * SIOCSIWENCODE here.
2290          */
2291         if (drv->auth_alg_fallback &&
2292             wpa_driver_nl80211_auth_alg_fallback(drv, params) < 0)
2293                 ret = -1;
2294
2295         if (!params->bssid &&
2296             wpa_driver_nl80211_set_bssid(drv, NULL) < 0)
2297                 ret = -1;
2298
2299         /* TODO: should consider getting wpa version and cipher/key_mgmt suites
2300          * from configuration, not from here, where only the selected suite is
2301          * available */
2302         if (wpa_driver_nl80211_set_gen_ie(drv, params->wpa_ie, params->wpa_ie_len)
2303             < 0)
2304                 ret = -1;
2305         if (params->wpa_ie == NULL || params->wpa_ie_len == 0)
2306                 value = IW_AUTH_WPA_VERSION_DISABLED;
2307         else if (params->wpa_ie[0] == WLAN_EID_RSN)
2308                 value = IW_AUTH_WPA_VERSION_WPA2;
2309         else
2310                 value = IW_AUTH_WPA_VERSION_WPA;
2311         if (wpa_driver_nl80211_set_auth_param(drv,
2312                                            IW_AUTH_WPA_VERSION, value) < 0)
2313                 ret = -1;
2314         value = wpa_driver_nl80211_cipher2wext(params->pairwise_suite);
2315         if (wpa_driver_nl80211_set_auth_param(drv,
2316                                            IW_AUTH_CIPHER_PAIRWISE, value) < 0)
2317                 ret = -1;
2318         value = wpa_driver_nl80211_cipher2wext(params->group_suite);
2319         if (wpa_driver_nl80211_set_auth_param(drv,
2320                                            IW_AUTH_CIPHER_GROUP, value) < 0)
2321                 ret = -1;
2322         value = wpa_driver_nl80211_keymgmt2wext(params->key_mgmt_suite);
2323         if (wpa_driver_nl80211_set_auth_param(drv,
2324                                            IW_AUTH_KEY_MGMT, value) < 0)
2325                 ret = -1;
2326         value = params->key_mgmt_suite != KEY_MGMT_NONE ||
2327                 params->pairwise_suite != CIPHER_NONE ||
2328                 params->group_suite != CIPHER_NONE ||
2329                 params->wpa_ie_len;
2330         if (wpa_driver_nl80211_set_auth_param(drv,
2331                                            IW_AUTH_PRIVACY_INVOKED, value) < 0)
2332                 ret = -1;
2333
2334         /* Allow unencrypted EAPOL messages even if pairwise keys are set when
2335          * not using WPA. IEEE 802.1X specifies that these frames are not
2336          * encrypted, but WPA encrypts them when pairwise keys are in use. */
2337         if (params->key_mgmt_suite == KEY_MGMT_802_1X ||
2338             params->key_mgmt_suite == KEY_MGMT_PSK)
2339                 allow_unencrypted_eapol = 0;
2340         else
2341                 allow_unencrypted_eapol = 1;
2342         
2343         if (wpa_driver_nl80211_set_auth_param(drv,
2344                                            IW_AUTH_RX_UNENCRYPTED_EAPOL,
2345                                            allow_unencrypted_eapol) < 0)
2346                 ret = -1;
2347         if (params->freq && wpa_driver_nl80211_set_freq(drv, params->freq) < 0)
2348                 ret = -1;
2349         if (wpa_driver_nl80211_set_ssid(drv, params->ssid, params->ssid_len) < 0)
2350                 ret = -1;
2351         if (params->bssid &&
2352             wpa_driver_nl80211_set_bssid(drv, params->bssid) < 0)
2353                 ret = -1;
2354
2355         return ret;
2356 }
2357
2358
2359 static int wpa_driver_nl80211_set_auth_alg(void *priv, int auth_alg)
2360 {
2361         struct wpa_driver_nl80211_data *drv = priv;
2362         int algs = 0, res;
2363
2364         if (auth_alg & AUTH_ALG_OPEN_SYSTEM)
2365                 algs |= IW_AUTH_ALG_OPEN_SYSTEM;
2366         if (auth_alg & AUTH_ALG_SHARED_KEY)
2367                 algs |= IW_AUTH_ALG_SHARED_KEY;
2368         if (auth_alg & AUTH_ALG_LEAP)
2369                 algs |= IW_AUTH_ALG_LEAP;
2370         if (algs == 0) {
2371                 /* at least one algorithm should be set */
2372                 algs = IW_AUTH_ALG_OPEN_SYSTEM;
2373         }
2374
2375         res = wpa_driver_nl80211_set_auth_param(drv, IW_AUTH_80211_AUTH_ALG,
2376                                              algs);
2377         drv->auth_alg_fallback = res == -2;
2378         return res;
2379 }
2380
2381
2382 /**
2383  * wpa_driver_nl80211_set_mode - Set wireless mode (infra/adhoc), SIOCSIWMODE
2384  * @priv: Pointer to private wext data from wpa_driver_nl80211_init()
2385  * @mode: 0 = infra/BSS (associate with an AP), 1 = adhoc/IBSS
2386  * Returns: 0 on success, -1 on failure
2387  */
2388 static int wpa_driver_nl80211_set_mode(void *priv, int mode)
2389 {
2390         struct wpa_driver_nl80211_data *drv = priv;
2391         int ret = -1, flags;
2392         struct nl_msg *msg;
2393
2394         msg = nlmsg_alloc();
2395         if (!msg)
2396                 return -1;
2397
2398         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2399                     0, NL80211_CMD_SET_INTERFACE, 0);
2400         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2401         NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE,
2402                     mode ? NL80211_IFTYPE_ADHOC : NL80211_IFTYPE_STATION);
2403
2404         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2405         if (!ret)
2406                 return 0;
2407         else
2408                 goto try_again;
2409
2410 nla_put_failure:
2411         wpa_printf(MSG_ERROR, "nl80211: Failed to set interface mode");
2412         return -1;
2413
2414 try_again:
2415         /* mac80211 doesn't allow mode changes while the device is up, so
2416          * take the device down, try to set the mode again, and bring the
2417          * device back up.
2418          */
2419         if (wpa_driver_nl80211_get_ifflags(drv, &flags) == 0) {
2420                 (void) wpa_driver_nl80211_set_ifflags(drv, flags & ~IFF_UP);
2421
2422                 /* Try to set the mode again while the interface is down */
2423                 msg = nlmsg_alloc();
2424                 if (!msg)
2425                         return -1;
2426
2427                 genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2428                             0, NL80211_CMD_SET_INTERFACE, 0);
2429                 NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2430                 NLA_PUT_U32(msg, NL80211_ATTR_IFTYPE,
2431                             mode ? NL80211_IFTYPE_ADHOC :
2432                             NL80211_IFTYPE_STATION);
2433                 ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2434                 if (ret) {
2435                         wpa_printf(MSG_ERROR, "Failed to set interface %s "
2436                                    "mode", drv->ifname);
2437                 }
2438
2439                 /* Ignore return value of get_ifflags to ensure that the device
2440                  * is always up like it was before this function was called.
2441                  */
2442                 (void) wpa_driver_nl80211_get_ifflags(drv, &flags);
2443                 (void) wpa_driver_nl80211_set_ifflags(drv, flags | IFF_UP);
2444         }
2445
2446         return ret;
2447 }
2448
2449
2450 static int wpa_driver_nl80211_pmksa(struct wpa_driver_nl80211_data *drv,
2451                                  u32 cmd, const u8 *bssid, const u8 *pmkid)
2452 {
2453         struct iwreq iwr;
2454         struct iw_pmksa pmksa;
2455         int ret = 0;
2456
2457         os_memset(&iwr, 0, sizeof(iwr));
2458         os_strlcpy(iwr.ifr_name, drv->ifname, IFNAMSIZ);
2459         os_memset(&pmksa, 0, sizeof(pmksa));
2460         pmksa.cmd = cmd;
2461         pmksa.bssid.sa_family = ARPHRD_ETHER;
2462         if (bssid)
2463                 os_memcpy(pmksa.bssid.sa_data, bssid, ETH_ALEN);
2464         if (pmkid)
2465                 os_memcpy(pmksa.pmkid, pmkid, IW_PMKID_LEN);
2466         iwr.u.data.pointer = (caddr_t) &pmksa;
2467         iwr.u.data.length = sizeof(pmksa);
2468
2469         if (ioctl(drv->ioctl_sock, SIOCSIWPMKSA, &iwr) < 0) {
2470                 if (errno != EOPNOTSUPP)
2471                         perror("ioctl[SIOCSIWPMKSA]");
2472                 ret = -1;
2473         }
2474
2475         return ret;
2476 }
2477
2478
2479 static int wpa_driver_nl80211_add_pmkid(void *priv, const u8 *bssid,
2480                                      const u8 *pmkid)
2481 {
2482         struct wpa_driver_nl80211_data *drv = priv;
2483         return wpa_driver_nl80211_pmksa(drv, IW_PMKSA_ADD, bssid, pmkid);
2484 }
2485
2486
2487 static int wpa_driver_nl80211_remove_pmkid(void *priv, const u8 *bssid,
2488                                         const u8 *pmkid)
2489 {
2490         struct wpa_driver_nl80211_data *drv = priv;
2491         return wpa_driver_nl80211_pmksa(drv, IW_PMKSA_REMOVE, bssid, pmkid);
2492 }
2493
2494
2495 static int wpa_driver_nl80211_flush_pmkid(void *priv)
2496 {
2497         struct wpa_driver_nl80211_data *drv = priv;
2498         return wpa_driver_nl80211_pmksa(drv, IW_PMKSA_FLUSH, NULL, NULL);
2499 }
2500
2501
2502 static int wpa_driver_nl80211_get_capa(void *priv,
2503                                        struct wpa_driver_capa *capa)
2504 {
2505         struct wpa_driver_nl80211_data *drv = priv;
2506         if (!drv->has_capability)
2507                 return -1;
2508         os_memcpy(capa, &drv->capa, sizeof(*capa));
2509         return 0;
2510 }
2511
2512
2513 static int wpa_driver_nl80211_set_operstate(void *priv, int state)
2514 {
2515         struct wpa_driver_nl80211_data *drv = priv;
2516
2517         wpa_printf(MSG_DEBUG, "%s: operstate %d->%d (%s)",
2518                    __func__, drv->operstate, state, state ? "UP" : "DORMANT");
2519         drv->operstate = state;
2520         return wpa_driver_nl80211_send_oper_ifla(
2521                 drv, -1, state ? IF_OPER_UP : IF_OPER_DORMANT);
2522 }
2523
2524
2525 #ifdef CONFIG_CLIENT_MLME
2526 static int wpa_driver_nl80211_open_mlme(struct wpa_driver_nl80211_data *drv)
2527 {
2528         if (wpa_driver_nl80211_set_userspace_mlme(drv, 1) < 0) {
2529                 wpa_printf(MSG_ERROR, "nl80211: Failed to enable userspace "
2530                            "MLME");
2531                 return -1;
2532         }
2533         if (wpa_driver_nl80211_create_monitor_interface(drv)) {
2534                 wpa_printf(MSG_ERROR, "nl80211: Failed to create monitor "
2535                            "interface");
2536                 return -1;
2537         }
2538         return 0;
2539 }
2540 #endif /* CONFIG_CLIENT_MLME */
2541
2542
2543 static int wpa_driver_nl80211_set_param(void *priv, const char *param)
2544 {
2545 #ifdef CONFIG_CLIENT_MLME
2546         struct wpa_driver_nl80211_data *drv = priv;
2547
2548         if (param == NULL)
2549                 return 0;
2550
2551         wpa_printf(MSG_DEBUG, "%s: param='%s'", __func__, param);
2552
2553         if (os_strstr(param, "use_mlme=1")) {
2554                 wpa_printf(MSG_DEBUG, "nl80211: Using user space MLME");
2555                 drv->capa.flags |= WPA_DRIVER_FLAGS_USER_SPACE_MLME;
2556
2557                 if (wpa_driver_nl80211_open_mlme(drv))
2558                         return -1;
2559         }
2560 #endif /* CONFIG_CLIENT_MLME */
2561
2562         return 0;
2563 }
2564
2565
2566 #ifdef CONFIG_CLIENT_MLME
2567
2568 struct phy_info_arg {
2569         u16 *num_modes;
2570         struct wpa_hw_modes *modes;
2571 };
2572
2573
2574 static int phy_info_handler(struct nl_msg *msg, void *arg)
2575 {
2576         struct nlattr *tb_msg[NL80211_ATTR_MAX + 1];
2577         struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg));
2578         struct phy_info_arg *phy_info = arg;
2579
2580         struct nlattr *tb_band[NL80211_BAND_ATTR_MAX + 1];
2581
2582         struct nlattr *tb_freq[NL80211_FREQUENCY_ATTR_MAX + 1];
2583         static struct nla_policy freq_policy[NL80211_FREQUENCY_ATTR_MAX + 1]
2584                 = {
2585                 [NL80211_FREQUENCY_ATTR_FREQ] = { .type = NLA_U32 },
2586                 [NL80211_FREQUENCY_ATTR_DISABLED] = { .type = NLA_FLAG },
2587                 [NL80211_FREQUENCY_ATTR_PASSIVE_SCAN] = { .type = NLA_FLAG },
2588                 [NL80211_FREQUENCY_ATTR_NO_IBSS] = { .type = NLA_FLAG },
2589                 [NL80211_FREQUENCY_ATTR_RADAR] = { .type = NLA_FLAG },
2590         };
2591
2592         struct nlattr *tb_rate[NL80211_BITRATE_ATTR_MAX + 1];
2593         static struct nla_policy rate_policy[NL80211_BITRATE_ATTR_MAX + 1] = {
2594                 [NL80211_BITRATE_ATTR_RATE] = { .type = NLA_U32 },
2595                 [NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE] =
2596                 { .type = NLA_FLAG },
2597         };
2598
2599         struct nlattr *nl_band;
2600         struct nlattr *nl_freq;
2601         struct nlattr *nl_rate;
2602         int rem_band, rem_freq, rem_rate;
2603         struct wpa_hw_modes *mode;
2604         int idx, mode_is_set;
2605
2606         nla_parse(tb_msg, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
2607                   genlmsg_attrlen(gnlh, 0), NULL);
2608
2609         if (!tb_msg[NL80211_ATTR_WIPHY_BANDS])
2610                 return NL_SKIP;
2611
2612         nla_for_each_nested(nl_band, tb_msg[NL80211_ATTR_WIPHY_BANDS],
2613                             rem_band) {
2614                 mode = os_realloc(phy_info->modes,
2615                                   (*phy_info->num_modes + 1) * sizeof(*mode));
2616                 if (!mode)
2617                         return NL_SKIP;
2618                 phy_info->modes = mode;
2619
2620                 mode_is_set = 0;
2621
2622                 mode = &phy_info->modes[*(phy_info->num_modes)];
2623                 os_memset(mode, 0, sizeof(*mode));
2624                 *(phy_info->num_modes) += 1;
2625
2626                 nla_parse(tb_band, NL80211_BAND_ATTR_MAX, nla_data(nl_band),
2627                           nla_len(nl_band), NULL);
2628
2629                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS],
2630                                     rem_freq) {
2631                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
2632                                   nla_data(nl_freq), nla_len(nl_freq),
2633                                   freq_policy);
2634                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
2635                                 continue;
2636                         mode->num_channels++;
2637                 }
2638
2639                 mode->channels = os_zalloc(mode->num_channels *
2640                                            sizeof(struct wpa_channel_data));
2641                 if (!mode->channels)
2642                         return NL_SKIP;
2643
2644                 idx = 0;
2645
2646                 nla_for_each_nested(nl_freq, tb_band[NL80211_BAND_ATTR_FREQS],
2647                                     rem_freq) {
2648                         nla_parse(tb_freq, NL80211_FREQUENCY_ATTR_MAX,
2649                                   nla_data(nl_freq), nla_len(nl_freq),
2650                                   freq_policy);
2651                         if (!tb_freq[NL80211_FREQUENCY_ATTR_FREQ])
2652                                 continue;
2653
2654                         mode->channels[idx].freq = nla_get_u32(
2655                                 tb_freq[NL80211_FREQUENCY_ATTR_FREQ]);
2656                         mode->channels[idx].flag |= WPA_CHAN_W_SCAN |
2657                                 WPA_CHAN_W_ACTIVE_SCAN |
2658                                 WPA_CHAN_W_IBSS;
2659
2660                         if (!mode_is_set) {
2661                                 /* crude heuristic */
2662                                 if (mode->channels[idx].freq < 4000)
2663                                         mode->mode = WPA_MODE_IEEE80211B;
2664                                 else
2665                                         mode->mode = WPA_MODE_IEEE80211A;
2666                                 mode_is_set = 1;
2667                         }
2668
2669                         /* crude heuristic */
2670                         if (mode->channels[idx].freq < 4000) {
2671                                 if (mode->channels[idx].freq == 2848)
2672                                         mode->channels[idx].chan = 14;
2673                                 else
2674                                         mode->channels[idx].chan =
2675                                                 (mode->channels[idx].freq -
2676                                                  2407) / 5;
2677                         } else
2678                                 mode->channels[idx].chan =
2679                                         mode->channels[idx].freq / 5 - 1000;
2680
2681                         if (tb_freq[NL80211_FREQUENCY_ATTR_DISABLED])
2682                                 mode->channels[idx].flag &= ~WPA_CHAN_W_SCAN;
2683                         if (tb_freq[NL80211_FREQUENCY_ATTR_PASSIVE_SCAN])
2684                                 mode->channels[idx].flag &=
2685                                         ~WPA_CHAN_W_ACTIVE_SCAN;
2686                         if (tb_freq[NL80211_FREQUENCY_ATTR_NO_IBSS])
2687                                 mode->channels[idx].flag &= ~WPA_CHAN_W_IBSS;
2688                         idx++;
2689                 }
2690
2691                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES],
2692                                     rem_rate) {
2693                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
2694                                   nla_data(nl_rate), nla_len(nl_rate),
2695                                   rate_policy);
2696                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
2697                                 continue;
2698                         mode->num_rates++;
2699                 }
2700
2701                 mode->rates = os_zalloc(mode->num_rates *
2702                                         sizeof(struct wpa_rate_data));
2703                 if (!mode->rates)
2704                         return NL_SKIP;
2705
2706                 idx = 0;
2707
2708                 nla_for_each_nested(nl_rate, tb_band[NL80211_BAND_ATTR_RATES],
2709                                     rem_rate) {
2710                         nla_parse(tb_rate, NL80211_BITRATE_ATTR_MAX,
2711                                   nla_data(nl_rate), nla_len(nl_rate),
2712                                   rate_policy);
2713                         if (!tb_rate[NL80211_BITRATE_ATTR_RATE])
2714                                 continue;
2715                         mode->rates[idx].rate = nla_get_u32(
2716                                 tb_rate[NL80211_BITRATE_ATTR_RATE]);
2717
2718                         /* crude heuristic */
2719                         if (mode->mode == WPA_MODE_IEEE80211B &&
2720                             mode->rates[idx].rate > 200)
2721                                 mode->mode = WPA_MODE_IEEE80211G;
2722
2723                         if (tb_rate[NL80211_BITRATE_ATTR_2GHZ_SHORTPREAMBLE])
2724                                 mode->rates[idx].flags |= WPA_RATE_PREAMBLE2;
2725
2726                         idx++;
2727                 }
2728         }
2729
2730         return NL_SKIP;
2731 }
2732
2733
2734 static struct wpa_hw_modes *
2735 wpa_driver_nl80211_get_hw_feature_data(void *priv, u16 *num_modes, u16 *flags)
2736 {
2737         struct wpa_driver_nl80211_data *drv = priv;
2738         struct nl_msg *msg;
2739         struct phy_info_arg result = {
2740                 .num_modes = num_modes,
2741                 .modes = NULL,
2742         };
2743
2744         *num_modes = 0;
2745         *flags = 0;
2746
2747         msg = nlmsg_alloc();
2748         if (!msg)
2749                 return NULL;
2750
2751         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2752                     0, NL80211_CMD_GET_WIPHY, 0);
2753
2754         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2755
2756         if (send_and_recv_msgs(drv, msg, phy_info_handler, &result) == 0)
2757                 return result.modes;
2758 nla_put_failure:
2759         return NULL;
2760 }
2761
2762
2763 static int wpa_driver_nl80211_set_channel(void *priv, wpa_hw_mode phymode,
2764                                           int chan, int freq)
2765 {
2766         return wpa_driver_nl80211_set_freq(priv, freq);
2767 }
2768
2769
2770 static int wpa_driver_nl80211_send_mlme(void *priv, const u8 *data,
2771                                         size_t data_len)
2772 {
2773         struct wpa_driver_nl80211_data *drv = priv;
2774         __u8 rtap_hdr[] = {
2775                 0x00, 0x00, /* radiotap version */
2776                 0x0e, 0x00, /* radiotap length */
2777                 0x02, 0xc0, 0x00, 0x00, /* bmap: flags, tx and rx flags */
2778                 0x0c,       /* F_WEP | F_FRAG (encrypt/fragment if required) */
2779                 0x00,       /* padding */
2780                 0x00, 0x00, /* RX and TX flags to indicate that */
2781                 0x00, 0x00, /* this is the injected frame directly */
2782         };
2783         struct iovec iov[2] = {
2784                 {
2785                         .iov_base = &rtap_hdr,
2786                         .iov_len = sizeof(rtap_hdr),
2787                 },
2788                 {
2789                         .iov_base = (void *) data,
2790                         .iov_len = data_len,
2791                 }
2792         };
2793         struct msghdr msg = {
2794                 .msg_name = NULL,
2795                 .msg_namelen = 0,
2796                 .msg_iov = iov,
2797                 .msg_iovlen = 2,
2798                 .msg_control = NULL,
2799                 .msg_controllen = 0,
2800                 .msg_flags = 0,
2801         };
2802
2803         if (sendmsg(drv->monitor_sock, &msg, 0) < 0) {
2804                 perror("send[MLME]");
2805                 return -1;
2806         }
2807
2808         return 0;
2809 }
2810
2811
2812 static int wpa_driver_nl80211_mlme_add_sta(void *priv, const u8 *addr,
2813                                            const u8 *supp_rates,
2814                                            size_t supp_rates_len)
2815 {
2816         struct wpa_driver_nl80211_data *drv = priv;
2817         struct nl_msg *msg;
2818         int ret = -1;
2819
2820         msg = nlmsg_alloc();
2821         if (!msg)
2822                 return -ENOMEM;
2823
2824         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2825                     0, NL80211_CMD_NEW_STATION, 0);
2826
2827         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2828         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
2829         /* TODO: Get proper Association ID and listen interval */
2830         NLA_PUT_U16(msg, NL80211_ATTR_STA_AID, 1);
2831         NLA_PUT(msg, NL80211_ATTR_STA_SUPPORTED_RATES, supp_rates_len,
2832                 supp_rates);
2833         NLA_PUT_U16(msg, NL80211_ATTR_STA_LISTEN_INTERVAL, 1);
2834
2835         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2836         /* ignore EEXIST, this happens if a STA associates while associated */
2837         if (ret == -EEXIST || ret >= 0)
2838                 ret = 0;
2839
2840 nla_put_failure:
2841         return ret;
2842 }
2843
2844
2845 static int wpa_driver_nl80211_mlme_remove_sta(void *priv, const u8 *addr)
2846 {
2847         struct wpa_driver_nl80211_data *drv = priv;
2848         struct nl_msg *msg;
2849         int ret = -1;
2850
2851         msg = nlmsg_alloc();
2852         if (!msg)
2853                 return -ENOMEM;
2854
2855         genlmsg_put(msg, 0, 0, genl_family_get_id(drv->nl80211), 0,
2856                     0, NL80211_CMD_DEL_STATION, 0);
2857
2858         NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, drv->ifindex);
2859         NLA_PUT(msg, NL80211_ATTR_MAC, ETH_ALEN, addr);
2860
2861         ret = 0;
2862
2863         ret = send_and_recv_msgs(drv, msg, NULL, NULL);
2864         return ret;
2865
2866 nla_put_failure:
2867         return -ENOBUFS;
2868 }
2869
2870 #endif /* CONFIG_CLIENT_MLME */
2871
2872
2873 const struct wpa_driver_ops wpa_driver_nl80211_ops = {
2874         .name = "nl80211",
2875         .desc = "Linux nl80211/cfg80211",
2876         .get_bssid = wpa_driver_nl80211_get_bssid,
2877         .get_ssid = wpa_driver_nl80211_get_ssid,
2878         .set_wpa = wpa_driver_nl80211_set_wpa,
2879         .set_key = wpa_driver_nl80211_set_key,
2880         .set_countermeasures = wpa_driver_nl80211_set_countermeasures,
2881         .set_drop_unencrypted = wpa_driver_nl80211_set_drop_unencrypted,
2882         .scan = wpa_driver_nl80211_scan,
2883         .get_scan_results2 = wpa_driver_nl80211_get_scan_results,
2884         .deauthenticate = wpa_driver_nl80211_deauthenticate,
2885         .disassociate = wpa_driver_nl80211_disassociate,
2886         .set_mode = wpa_driver_nl80211_set_mode,
2887         .associate = wpa_driver_nl80211_associate,
2888         .set_auth_alg = wpa_driver_nl80211_set_auth_alg,
2889         .init = wpa_driver_nl80211_init,
2890         .deinit = wpa_driver_nl80211_deinit,
2891         .set_param = wpa_driver_nl80211_set_param,
2892         .add_pmkid = wpa_driver_nl80211_add_pmkid,
2893         .remove_pmkid = wpa_driver_nl80211_remove_pmkid,
2894         .flush_pmkid = wpa_driver_nl80211_flush_pmkid,
2895         .get_capa = wpa_driver_nl80211_get_capa,
2896         .set_operstate = wpa_driver_nl80211_set_operstate,
2897         .set_country = wpa_driver_nl80211_set_country,
2898 #ifdef CONFIG_CLIENT_MLME
2899         .get_hw_feature_data = wpa_driver_nl80211_get_hw_feature_data,
2900         .set_channel = wpa_driver_nl80211_set_channel,
2901         .set_ssid = wpa_driver_nl80211_set_ssid,
2902         .set_bssid = wpa_driver_nl80211_set_bssid,
2903         .send_mlme = wpa_driver_nl80211_send_mlme,
2904         .mlme_add_sta = wpa_driver_nl80211_mlme_add_sta,
2905         .mlme_remove_sta = wpa_driver_nl80211_mlme_remove_sta,
2906 #endif /* CONFIG_CLIENT_MLME */
2907 };