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