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