]> Pileus Git - ~andy/linux/blob - net/wireless/sme.c
Merge branch 'nfsd-next' of git://linux-nfs.org/~bfields/linux
[~andy/linux] / net / wireless / sme.c
1 /*
2  * SME code for cfg80211
3  * both driver SME event handling and the SME implementation
4  * (for nl80211's connect() and wext)
5  *
6  * Copyright 2009       Johannes Berg <johannes@sipsolutions.net>
7  * Copyright (C) 2009   Intel Corporation. All rights reserved.
8  */
9
10 #include <linux/etherdevice.h>
11 #include <linux/if_arp.h>
12 #include <linux/slab.h>
13 #include <linux/workqueue.h>
14 #include <linux/wireless.h>
15 #include <linux/export.h>
16 #include <net/iw_handler.h>
17 #include <net/cfg80211.h>
18 #include <net/rtnetlink.h>
19 #include "nl80211.h"
20 #include "reg.h"
21 #include "rdev-ops.h"
22
23 /*
24  * Software SME in cfg80211, using auth/assoc/deauth calls to the
25  * driver. This is is for implementing nl80211's connect/disconnect
26  * and wireless extensions (if configured.)
27  */
28
29 struct cfg80211_conn {
30         struct cfg80211_connect_params params;
31         /* these are sub-states of the _CONNECTING sme_state */
32         enum {
33                 CFG80211_CONN_SCANNING,
34                 CFG80211_CONN_SCAN_AGAIN,
35                 CFG80211_CONN_AUTHENTICATE_NEXT,
36                 CFG80211_CONN_AUTHENTICATING,
37                 CFG80211_CONN_AUTH_FAILED,
38                 CFG80211_CONN_ASSOCIATE_NEXT,
39                 CFG80211_CONN_ASSOCIATING,
40                 CFG80211_CONN_ASSOC_FAILED,
41                 CFG80211_CONN_DEAUTH,
42                 CFG80211_CONN_CONNECTED,
43         } state;
44         u8 bssid[ETH_ALEN], prev_bssid[ETH_ALEN];
45         u8 *ie;
46         size_t ie_len;
47         bool auto_auth, prev_bssid_valid;
48 };
49
50 static void cfg80211_sme_free(struct wireless_dev *wdev)
51 {
52         if (!wdev->conn)
53                 return;
54
55         kfree(wdev->conn->ie);
56         kfree(wdev->conn);
57         wdev->conn = NULL;
58 }
59
60 static int cfg80211_conn_scan(struct wireless_dev *wdev)
61 {
62         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
63         struct cfg80211_scan_request *request;
64         int n_channels, err;
65
66         ASSERT_RTNL();
67         ASSERT_RDEV_LOCK(rdev);
68         ASSERT_WDEV_LOCK(wdev);
69
70         if (rdev->scan_req || rdev->scan_msg)
71                 return -EBUSY;
72
73         if (wdev->conn->params.channel)
74                 n_channels = 1;
75         else
76                 n_channels = ieee80211_get_num_supported_channels(wdev->wiphy);
77
78         request = kzalloc(sizeof(*request) + sizeof(request->ssids[0]) +
79                           sizeof(request->channels[0]) * n_channels,
80                           GFP_KERNEL);
81         if (!request)
82                 return -ENOMEM;
83
84         if (wdev->conn->params.channel)
85                 request->channels[0] = wdev->conn->params.channel;
86         else {
87                 int i = 0, j;
88                 enum ieee80211_band band;
89                 struct ieee80211_supported_band *bands;
90                 struct ieee80211_channel *channel;
91
92                 for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
93                         bands = wdev->wiphy->bands[band];
94                         if (!bands)
95                                 continue;
96                         for (j = 0; j < bands->n_channels; j++) {
97                                 channel = &bands->channels[j];
98                                 if (channel->flags & IEEE80211_CHAN_DISABLED)
99                                         continue;
100                                 request->channels[i++] = channel;
101                         }
102                         request->rates[band] = (1 << bands->n_bitrates) - 1;
103                 }
104                 n_channels = i;
105         }
106         request->n_channels = n_channels;
107         request->ssids = (void *)&request->channels[n_channels];
108         request->n_ssids = 1;
109
110         memcpy(request->ssids[0].ssid, wdev->conn->params.ssid,
111                 wdev->conn->params.ssid_len);
112         request->ssids[0].ssid_len = wdev->conn->params.ssid_len;
113
114         request->wdev = wdev;
115         request->wiphy = &rdev->wiphy;
116         request->scan_start = jiffies;
117
118         rdev->scan_req = request;
119
120         err = rdev_scan(rdev, request);
121         if (!err) {
122                 wdev->conn->state = CFG80211_CONN_SCANNING;
123                 nl80211_send_scan_start(rdev, wdev);
124                 dev_hold(wdev->netdev);
125         } else {
126                 rdev->scan_req = NULL;
127                 kfree(request);
128         }
129         return err;
130 }
131
132 static int cfg80211_conn_do_work(struct wireless_dev *wdev)
133 {
134         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
135         struct cfg80211_connect_params *params;
136         struct cfg80211_assoc_request req = {};
137         int err;
138
139         ASSERT_WDEV_LOCK(wdev);
140
141         if (!wdev->conn)
142                 return 0;
143
144         params = &wdev->conn->params;
145
146         switch (wdev->conn->state) {
147         case CFG80211_CONN_SCANNING:
148                 /* didn't find it during scan ... */
149                 return -ENOENT;
150         case CFG80211_CONN_SCAN_AGAIN:
151                 return cfg80211_conn_scan(wdev);
152         case CFG80211_CONN_AUTHENTICATE_NEXT:
153                 BUG_ON(!rdev->ops->auth);
154                 wdev->conn->state = CFG80211_CONN_AUTHENTICATING;
155                 return cfg80211_mlme_auth(rdev, wdev->netdev,
156                                           params->channel, params->auth_type,
157                                           params->bssid,
158                                           params->ssid, params->ssid_len,
159                                           NULL, 0,
160                                           params->key, params->key_len,
161                                           params->key_idx, NULL, 0);
162         case CFG80211_CONN_AUTH_FAILED:
163                 return -ENOTCONN;
164         case CFG80211_CONN_ASSOCIATE_NEXT:
165                 BUG_ON(!rdev->ops->assoc);
166                 wdev->conn->state = CFG80211_CONN_ASSOCIATING;
167                 if (wdev->conn->prev_bssid_valid)
168                         req.prev_bssid = wdev->conn->prev_bssid;
169                 req.ie = params->ie;
170                 req.ie_len = params->ie_len;
171                 req.use_mfp = params->mfp != NL80211_MFP_NO;
172                 req.crypto = params->crypto;
173                 req.flags = params->flags;
174                 req.ht_capa = params->ht_capa;
175                 req.ht_capa_mask = params->ht_capa_mask;
176                 req.vht_capa = params->vht_capa;
177                 req.vht_capa_mask = params->vht_capa_mask;
178
179                 err = cfg80211_mlme_assoc(rdev, wdev->netdev, params->channel,
180                                           params->bssid, params->ssid,
181                                           params->ssid_len, &req);
182                 if (err)
183                         cfg80211_mlme_deauth(rdev, wdev->netdev, params->bssid,
184                                              NULL, 0,
185                                              WLAN_REASON_DEAUTH_LEAVING,
186                                              false);
187                 return err;
188         case CFG80211_CONN_ASSOC_FAILED:
189                 cfg80211_mlme_deauth(rdev, wdev->netdev, params->bssid,
190                                      NULL, 0,
191                                      WLAN_REASON_DEAUTH_LEAVING, false);
192                 return -ENOTCONN;
193         case CFG80211_CONN_DEAUTH:
194                 cfg80211_mlme_deauth(rdev, wdev->netdev, params->bssid,
195                                      NULL, 0,
196                                      WLAN_REASON_DEAUTH_LEAVING, false);
197                 /* free directly, disconnected event already sent */
198                 cfg80211_sme_free(wdev);
199                 return 0;
200         default:
201                 return 0;
202         }
203 }
204
205 void cfg80211_conn_work(struct work_struct *work)
206 {
207         struct cfg80211_registered_device *rdev =
208                 container_of(work, struct cfg80211_registered_device, conn_work);
209         struct wireless_dev *wdev;
210         u8 bssid_buf[ETH_ALEN], *bssid = NULL;
211
212         rtnl_lock();
213
214         list_for_each_entry(wdev, &rdev->wdev_list, list) {
215                 if (!wdev->netdev)
216                         continue;
217
218                 wdev_lock(wdev);
219                 if (!netif_running(wdev->netdev)) {
220                         wdev_unlock(wdev);
221                         continue;
222                 }
223                 if (!wdev->conn ||
224                     wdev->conn->state == CFG80211_CONN_CONNECTED) {
225                         wdev_unlock(wdev);
226                         continue;
227                 }
228                 if (wdev->conn->params.bssid) {
229                         memcpy(bssid_buf, wdev->conn->params.bssid, ETH_ALEN);
230                         bssid = bssid_buf;
231                 }
232                 if (cfg80211_conn_do_work(wdev)) {
233                         __cfg80211_connect_result(
234                                         wdev->netdev, bssid,
235                                         NULL, 0, NULL, 0,
236                                         WLAN_STATUS_UNSPECIFIED_FAILURE,
237                                         false, NULL);
238                         cfg80211_sme_free(wdev);
239                 }
240                 wdev_unlock(wdev);
241         }
242
243         rtnl_unlock();
244 }
245
246 /* Returned bss is reference counted and must be cleaned up appropriately. */
247 static struct cfg80211_bss *cfg80211_get_conn_bss(struct wireless_dev *wdev)
248 {
249         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
250         struct cfg80211_bss *bss;
251         u16 capa = WLAN_CAPABILITY_ESS;
252
253         ASSERT_WDEV_LOCK(wdev);
254
255         if (wdev->conn->params.privacy)
256                 capa |= WLAN_CAPABILITY_PRIVACY;
257
258         bss = cfg80211_get_bss(wdev->wiphy, wdev->conn->params.channel,
259                                wdev->conn->params.bssid,
260                                wdev->conn->params.ssid,
261                                wdev->conn->params.ssid_len,
262                                WLAN_CAPABILITY_ESS | WLAN_CAPABILITY_PRIVACY,
263                                capa);
264         if (!bss)
265                 return NULL;
266
267         memcpy(wdev->conn->bssid, bss->bssid, ETH_ALEN);
268         wdev->conn->params.bssid = wdev->conn->bssid;
269         wdev->conn->params.channel = bss->channel;
270         wdev->conn->state = CFG80211_CONN_AUTHENTICATE_NEXT;
271         schedule_work(&rdev->conn_work);
272
273         return bss;
274 }
275
276 static void __cfg80211_sme_scan_done(struct net_device *dev)
277 {
278         struct wireless_dev *wdev = dev->ieee80211_ptr;
279         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
280         struct cfg80211_bss *bss;
281
282         ASSERT_WDEV_LOCK(wdev);
283
284         if (!wdev->conn)
285                 return;
286
287         if (wdev->conn->state != CFG80211_CONN_SCANNING &&
288             wdev->conn->state != CFG80211_CONN_SCAN_AGAIN)
289                 return;
290
291         bss = cfg80211_get_conn_bss(wdev);
292         if (bss)
293                 cfg80211_put_bss(&rdev->wiphy, bss);
294         else
295                 schedule_work(&rdev->conn_work);
296 }
297
298 void cfg80211_sme_scan_done(struct net_device *dev)
299 {
300         struct wireless_dev *wdev = dev->ieee80211_ptr;
301
302         wdev_lock(wdev);
303         __cfg80211_sme_scan_done(dev);
304         wdev_unlock(wdev);
305 }
306
307 void cfg80211_sme_rx_auth(struct wireless_dev *wdev, const u8 *buf, size_t len)
308 {
309         struct wiphy *wiphy = wdev->wiphy;
310         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
311         struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)buf;
312         u16 status_code = le16_to_cpu(mgmt->u.auth.status_code);
313
314         ASSERT_WDEV_LOCK(wdev);
315
316         if (!wdev->conn || wdev->conn->state == CFG80211_CONN_CONNECTED)
317                 return;
318
319         if (status_code == WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG &&
320             wdev->conn->auto_auth &&
321             wdev->conn->params.auth_type != NL80211_AUTHTYPE_NETWORK_EAP) {
322                 /* select automatically between only open, shared, leap */
323                 switch (wdev->conn->params.auth_type) {
324                 case NL80211_AUTHTYPE_OPEN_SYSTEM:
325                         if (wdev->connect_keys)
326                                 wdev->conn->params.auth_type =
327                                         NL80211_AUTHTYPE_SHARED_KEY;
328                         else
329                                 wdev->conn->params.auth_type =
330                                         NL80211_AUTHTYPE_NETWORK_EAP;
331                         break;
332                 case NL80211_AUTHTYPE_SHARED_KEY:
333                         wdev->conn->params.auth_type =
334                                 NL80211_AUTHTYPE_NETWORK_EAP;
335                         break;
336                 default:
337                         /* huh? */
338                         wdev->conn->params.auth_type =
339                                 NL80211_AUTHTYPE_OPEN_SYSTEM;
340                         break;
341                 }
342                 wdev->conn->state = CFG80211_CONN_AUTHENTICATE_NEXT;
343                 schedule_work(&rdev->conn_work);
344         } else if (status_code != WLAN_STATUS_SUCCESS) {
345                 __cfg80211_connect_result(wdev->netdev, mgmt->bssid,
346                                           NULL, 0, NULL, 0,
347                                           status_code, false, NULL);
348         } else if (wdev->conn->state == CFG80211_CONN_AUTHENTICATING) {
349                 wdev->conn->state = CFG80211_CONN_ASSOCIATE_NEXT;
350                 schedule_work(&rdev->conn_work);
351         }
352 }
353
354 bool cfg80211_sme_rx_assoc_resp(struct wireless_dev *wdev, u16 status)
355 {
356         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
357
358         if (!wdev->conn)
359                 return false;
360
361         if (status == WLAN_STATUS_SUCCESS) {
362                 wdev->conn->state = CFG80211_CONN_CONNECTED;
363                 return false;
364         }
365
366         if (wdev->conn->prev_bssid_valid) {
367                 /*
368                  * Some stupid APs don't accept reassoc, so we
369                  * need to fall back to trying regular assoc;
370                  * return true so no event is sent to userspace.
371                  */
372                 wdev->conn->prev_bssid_valid = false;
373                 wdev->conn->state = CFG80211_CONN_ASSOCIATE_NEXT;
374                 schedule_work(&rdev->conn_work);
375                 return true;
376         }
377
378         wdev->conn->state = CFG80211_CONN_ASSOC_FAILED;
379         schedule_work(&rdev->conn_work);
380         return false;
381 }
382
383 void cfg80211_sme_deauth(struct wireless_dev *wdev)
384 {
385         cfg80211_sme_free(wdev);
386 }
387
388 void cfg80211_sme_auth_timeout(struct wireless_dev *wdev)
389 {
390         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
391
392         if (!wdev->conn)
393                 return;
394
395         wdev->conn->state = CFG80211_CONN_AUTH_FAILED;
396         schedule_work(&rdev->conn_work);
397 }
398
399 void cfg80211_sme_disassoc(struct wireless_dev *wdev)
400 {
401         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
402
403         if (!wdev->conn)
404                 return;
405
406         wdev->conn->state = CFG80211_CONN_DEAUTH;
407         schedule_work(&rdev->conn_work);
408 }
409
410 void cfg80211_sme_assoc_timeout(struct wireless_dev *wdev)
411 {
412         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
413
414         if (!wdev->conn)
415                 return;
416
417         wdev->conn->state = CFG80211_CONN_ASSOC_FAILED;
418         schedule_work(&rdev->conn_work);
419 }
420
421 static int cfg80211_sme_connect(struct wireless_dev *wdev,
422                                 struct cfg80211_connect_params *connect,
423                                 const u8 *prev_bssid)
424 {
425         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
426         struct cfg80211_bss *bss;
427         int err;
428
429         if (!rdev->ops->auth || !rdev->ops->assoc)
430                 return -EOPNOTSUPP;
431
432         if (wdev->current_bss)
433                 return -EALREADY;
434
435         if (WARN_ON(wdev->conn))
436                 return -EINPROGRESS;
437
438         wdev->conn = kzalloc(sizeof(*wdev->conn), GFP_KERNEL);
439         if (!wdev->conn)
440                 return -ENOMEM;
441
442         /*
443          * Copy all parameters, and treat explicitly IEs, BSSID, SSID.
444          */
445         memcpy(&wdev->conn->params, connect, sizeof(*connect));
446         if (connect->bssid) {
447                 wdev->conn->params.bssid = wdev->conn->bssid;
448                 memcpy(wdev->conn->bssid, connect->bssid, ETH_ALEN);
449         }
450
451         if (connect->ie) {
452                 wdev->conn->ie = kmemdup(connect->ie, connect->ie_len,
453                                         GFP_KERNEL);
454                 wdev->conn->params.ie = wdev->conn->ie;
455                 if (!wdev->conn->ie) {
456                         kfree(wdev->conn);
457                         wdev->conn = NULL;
458                         return -ENOMEM;
459                 }
460         }
461
462         if (connect->auth_type == NL80211_AUTHTYPE_AUTOMATIC) {
463                 wdev->conn->auto_auth = true;
464                 /* start with open system ... should mostly work */
465                 wdev->conn->params.auth_type =
466                         NL80211_AUTHTYPE_OPEN_SYSTEM;
467         } else {
468                 wdev->conn->auto_auth = false;
469         }
470
471         wdev->conn->params.ssid = wdev->ssid;
472         wdev->conn->params.ssid_len = connect->ssid_len;
473
474         /* see if we have the bss already */
475         bss = cfg80211_get_conn_bss(wdev);
476
477         if (prev_bssid) {
478                 memcpy(wdev->conn->prev_bssid, prev_bssid, ETH_ALEN);
479                 wdev->conn->prev_bssid_valid = true;
480         }
481
482         /* we're good if we have a matching bss struct */
483         if (bss) {
484                 wdev->conn->state = CFG80211_CONN_AUTHENTICATE_NEXT;
485                 err = cfg80211_conn_do_work(wdev);
486                 cfg80211_put_bss(wdev->wiphy, bss);
487         } else {
488                 /* otherwise we'll need to scan for the AP first */
489                 err = cfg80211_conn_scan(wdev);
490
491                 /*
492                  * If we can't scan right now, then we need to scan again
493                  * after the current scan finished, since the parameters
494                  * changed (unless we find a good AP anyway).
495                  */
496                 if (err == -EBUSY) {
497                         err = 0;
498                         wdev->conn->state = CFG80211_CONN_SCAN_AGAIN;
499                 }
500         }
501
502         if (err)
503                 cfg80211_sme_free(wdev);
504
505         return err;
506 }
507
508 static int cfg80211_sme_disconnect(struct wireless_dev *wdev, u16 reason)
509 {
510         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
511         int err;
512
513         if (!wdev->conn)
514                 return 0;
515
516         if (!rdev->ops->deauth)
517                 return -EOPNOTSUPP;
518
519         if (wdev->conn->state == CFG80211_CONN_SCANNING ||
520             wdev->conn->state == CFG80211_CONN_SCAN_AGAIN) {
521                 err = 0;
522                 goto out;
523         }
524
525         /* wdev->conn->params.bssid must be set if > SCANNING */
526         err = cfg80211_mlme_deauth(rdev, wdev->netdev,
527                                    wdev->conn->params.bssid,
528                                    NULL, 0, reason, false);
529  out:
530         cfg80211_sme_free(wdev);
531         return err;
532 }
533
534 /*
535  * code shared for in-device and software SME
536  */
537
538 static bool cfg80211_is_all_idle(void)
539 {
540         struct cfg80211_registered_device *rdev;
541         struct wireless_dev *wdev;
542         bool is_all_idle = true;
543
544         /*
545          * All devices must be idle as otherwise if you are actively
546          * scanning some new beacon hints could be learned and would
547          * count as new regulatory hints.
548          */
549         list_for_each_entry(rdev, &cfg80211_rdev_list, list) {
550                 list_for_each_entry(wdev, &rdev->wdev_list, list) {
551                         wdev_lock(wdev);
552                         if (wdev->conn || wdev->current_bss)
553                                 is_all_idle = false;
554                         wdev_unlock(wdev);
555                 }
556         }
557
558         return is_all_idle;
559 }
560
561 static void disconnect_work(struct work_struct *work)
562 {
563         rtnl_lock();
564         if (cfg80211_is_all_idle())
565                 regulatory_hint_disconnect();
566         rtnl_unlock();
567 }
568
569 static DECLARE_WORK(cfg80211_disconnect_work, disconnect_work);
570
571
572 /*
573  * API calls for drivers implementing connect/disconnect and
574  * SME event handling
575  */
576
577 /* This method must consume bss one way or another */
578 void __cfg80211_connect_result(struct net_device *dev, const u8 *bssid,
579                                const u8 *req_ie, size_t req_ie_len,
580                                const u8 *resp_ie, size_t resp_ie_len,
581                                u16 status, bool wextev,
582                                struct cfg80211_bss *bss)
583 {
584         struct wireless_dev *wdev = dev->ieee80211_ptr;
585         const u8 *country_ie;
586 #ifdef CONFIG_CFG80211_WEXT
587         union iwreq_data wrqu;
588 #endif
589
590         ASSERT_WDEV_LOCK(wdev);
591
592         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_STATION &&
593                     wdev->iftype != NL80211_IFTYPE_P2P_CLIENT)) {
594                 cfg80211_put_bss(wdev->wiphy, bss);
595                 return;
596         }
597
598         nl80211_send_connect_result(wiphy_to_dev(wdev->wiphy), dev,
599                                     bssid, req_ie, req_ie_len,
600                                     resp_ie, resp_ie_len,
601                                     status, GFP_KERNEL);
602
603 #ifdef CONFIG_CFG80211_WEXT
604         if (wextev) {
605                 if (req_ie && status == WLAN_STATUS_SUCCESS) {
606                         memset(&wrqu, 0, sizeof(wrqu));
607                         wrqu.data.length = req_ie_len;
608                         wireless_send_event(dev, IWEVASSOCREQIE, &wrqu, req_ie);
609                 }
610
611                 if (resp_ie && status == WLAN_STATUS_SUCCESS) {
612                         memset(&wrqu, 0, sizeof(wrqu));
613                         wrqu.data.length = resp_ie_len;
614                         wireless_send_event(dev, IWEVASSOCRESPIE, &wrqu, resp_ie);
615                 }
616
617                 memset(&wrqu, 0, sizeof(wrqu));
618                 wrqu.ap_addr.sa_family = ARPHRD_ETHER;
619                 if (bssid && status == WLAN_STATUS_SUCCESS) {
620                         memcpy(wrqu.ap_addr.sa_data, bssid, ETH_ALEN);
621                         memcpy(wdev->wext.prev_bssid, bssid, ETH_ALEN);
622                         wdev->wext.prev_bssid_valid = true;
623                 }
624                 wireless_send_event(dev, SIOCGIWAP, &wrqu, NULL);
625         }
626 #endif
627
628         if (!bss && (status == WLAN_STATUS_SUCCESS)) {
629                 WARN_ON_ONCE(!wiphy_to_dev(wdev->wiphy)->ops->connect);
630                 bss = cfg80211_get_bss(wdev->wiphy, NULL, bssid,
631                                        wdev->ssid, wdev->ssid_len,
632                                        WLAN_CAPABILITY_ESS,
633                                        WLAN_CAPABILITY_ESS);
634                 if (bss)
635                         cfg80211_hold_bss(bss_from_pub(bss));
636         }
637
638         if (wdev->current_bss) {
639                 cfg80211_unhold_bss(wdev->current_bss);
640                 cfg80211_put_bss(wdev->wiphy, &wdev->current_bss->pub);
641                 wdev->current_bss = NULL;
642         }
643
644         if (status != WLAN_STATUS_SUCCESS) {
645                 kfree(wdev->connect_keys);
646                 wdev->connect_keys = NULL;
647                 wdev->ssid_len = 0;
648                 if (bss) {
649                         cfg80211_unhold_bss(bss_from_pub(bss));
650                         cfg80211_put_bss(wdev->wiphy, bss);
651                 }
652                 return;
653         }
654
655         if (WARN_ON(!bss))
656                 return;
657
658         wdev->current_bss = bss_from_pub(bss);
659
660         cfg80211_upload_connect_keys(wdev);
661
662         rcu_read_lock();
663         country_ie = ieee80211_bss_get_ie(bss, WLAN_EID_COUNTRY);
664         if (!country_ie) {
665                 rcu_read_unlock();
666                 return;
667         }
668
669         country_ie = kmemdup(country_ie, 2 + country_ie[1], GFP_ATOMIC);
670         rcu_read_unlock();
671
672         if (!country_ie)
673                 return;
674
675         /*
676          * ieee80211_bss_get_ie() ensures we can access:
677          * - country_ie + 2, the start of the country ie data, and
678          * - and country_ie[1] which is the IE length
679          */
680         regulatory_hint_country_ie(wdev->wiphy, bss->channel->band,
681                                    country_ie + 2, country_ie[1]);
682         kfree(country_ie);
683 }
684
685 void cfg80211_connect_result(struct net_device *dev, const u8 *bssid,
686                              const u8 *req_ie, size_t req_ie_len,
687                              const u8 *resp_ie, size_t resp_ie_len,
688                              u16 status, gfp_t gfp)
689 {
690         struct wireless_dev *wdev = dev->ieee80211_ptr;
691         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
692         struct cfg80211_event *ev;
693         unsigned long flags;
694
695         ev = kzalloc(sizeof(*ev) + req_ie_len + resp_ie_len, gfp);
696         if (!ev)
697                 return;
698
699         ev->type = EVENT_CONNECT_RESULT;
700         if (bssid)
701                 memcpy(ev->cr.bssid, bssid, ETH_ALEN);
702         if (req_ie_len) {
703                 ev->cr.req_ie = ((u8 *)ev) + sizeof(*ev);
704                 ev->cr.req_ie_len = req_ie_len;
705                 memcpy((void *)ev->cr.req_ie, req_ie, req_ie_len);
706         }
707         if (resp_ie_len) {
708                 ev->cr.resp_ie = ((u8 *)ev) + sizeof(*ev) + req_ie_len;
709                 ev->cr.resp_ie_len = resp_ie_len;
710                 memcpy((void *)ev->cr.resp_ie, resp_ie, resp_ie_len);
711         }
712         ev->cr.status = status;
713
714         spin_lock_irqsave(&wdev->event_lock, flags);
715         list_add_tail(&ev->list, &wdev->event_list);
716         spin_unlock_irqrestore(&wdev->event_lock, flags);
717         queue_work(cfg80211_wq, &rdev->event_work);
718 }
719 EXPORT_SYMBOL(cfg80211_connect_result);
720
721 /* Consumes bss object one way or another */
722 void __cfg80211_roamed(struct wireless_dev *wdev,
723                        struct cfg80211_bss *bss,
724                        const u8 *req_ie, size_t req_ie_len,
725                        const u8 *resp_ie, size_t resp_ie_len)
726 {
727 #ifdef CONFIG_CFG80211_WEXT
728         union iwreq_data wrqu;
729 #endif
730         ASSERT_WDEV_LOCK(wdev);
731
732         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_STATION &&
733                     wdev->iftype != NL80211_IFTYPE_P2P_CLIENT))
734                 goto out;
735
736         if (WARN_ON(!wdev->current_bss))
737                 goto out;
738
739         cfg80211_unhold_bss(wdev->current_bss);
740         cfg80211_put_bss(wdev->wiphy, &wdev->current_bss->pub);
741         wdev->current_bss = NULL;
742
743         cfg80211_hold_bss(bss_from_pub(bss));
744         wdev->current_bss = bss_from_pub(bss);
745
746         nl80211_send_roamed(wiphy_to_dev(wdev->wiphy), wdev->netdev, bss->bssid,
747                             req_ie, req_ie_len, resp_ie, resp_ie_len,
748                             GFP_KERNEL);
749
750 #ifdef CONFIG_CFG80211_WEXT
751         if (req_ie) {
752                 memset(&wrqu, 0, sizeof(wrqu));
753                 wrqu.data.length = req_ie_len;
754                 wireless_send_event(wdev->netdev, IWEVASSOCREQIE,
755                                     &wrqu, req_ie);
756         }
757
758         if (resp_ie) {
759                 memset(&wrqu, 0, sizeof(wrqu));
760                 wrqu.data.length = resp_ie_len;
761                 wireless_send_event(wdev->netdev, IWEVASSOCRESPIE,
762                                     &wrqu, resp_ie);
763         }
764
765         memset(&wrqu, 0, sizeof(wrqu));
766         wrqu.ap_addr.sa_family = ARPHRD_ETHER;
767         memcpy(wrqu.ap_addr.sa_data, bss->bssid, ETH_ALEN);
768         memcpy(wdev->wext.prev_bssid, bss->bssid, ETH_ALEN);
769         wdev->wext.prev_bssid_valid = true;
770         wireless_send_event(wdev->netdev, SIOCGIWAP, &wrqu, NULL);
771 #endif
772
773         return;
774 out:
775         cfg80211_put_bss(wdev->wiphy, bss);
776 }
777
778 void cfg80211_roamed(struct net_device *dev,
779                      struct ieee80211_channel *channel,
780                      const u8 *bssid,
781                      const u8 *req_ie, size_t req_ie_len,
782                      const u8 *resp_ie, size_t resp_ie_len, gfp_t gfp)
783 {
784         struct wireless_dev *wdev = dev->ieee80211_ptr;
785         struct cfg80211_bss *bss;
786
787         bss = cfg80211_get_bss(wdev->wiphy, channel, bssid, wdev->ssid,
788                                wdev->ssid_len, WLAN_CAPABILITY_ESS,
789                                WLAN_CAPABILITY_ESS);
790         if (WARN_ON(!bss))
791                 return;
792
793         cfg80211_roamed_bss(dev, bss, req_ie, req_ie_len, resp_ie,
794                             resp_ie_len, gfp);
795 }
796 EXPORT_SYMBOL(cfg80211_roamed);
797
798 /* Consumes bss object one way or another */
799 void cfg80211_roamed_bss(struct net_device *dev,
800                          struct cfg80211_bss *bss, const u8 *req_ie,
801                          size_t req_ie_len, const u8 *resp_ie,
802                          size_t resp_ie_len, gfp_t gfp)
803 {
804         struct wireless_dev *wdev = dev->ieee80211_ptr;
805         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
806         struct cfg80211_event *ev;
807         unsigned long flags;
808
809         if (WARN_ON(!bss))
810                 return;
811
812         ev = kzalloc(sizeof(*ev) + req_ie_len + resp_ie_len, gfp);
813         if (!ev) {
814                 cfg80211_put_bss(wdev->wiphy, bss);
815                 return;
816         }
817
818         ev->type = EVENT_ROAMED;
819         ev->rm.req_ie = ((u8 *)ev) + sizeof(*ev);
820         ev->rm.req_ie_len = req_ie_len;
821         memcpy((void *)ev->rm.req_ie, req_ie, req_ie_len);
822         ev->rm.resp_ie = ((u8 *)ev) + sizeof(*ev) + req_ie_len;
823         ev->rm.resp_ie_len = resp_ie_len;
824         memcpy((void *)ev->rm.resp_ie, resp_ie, resp_ie_len);
825         ev->rm.bss = bss;
826
827         spin_lock_irqsave(&wdev->event_lock, flags);
828         list_add_tail(&ev->list, &wdev->event_list);
829         spin_unlock_irqrestore(&wdev->event_lock, flags);
830         queue_work(cfg80211_wq, &rdev->event_work);
831 }
832 EXPORT_SYMBOL(cfg80211_roamed_bss);
833
834 void __cfg80211_disconnected(struct net_device *dev, const u8 *ie,
835                              size_t ie_len, u16 reason, bool from_ap)
836 {
837         struct wireless_dev *wdev = dev->ieee80211_ptr;
838         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
839         int i;
840 #ifdef CONFIG_CFG80211_WEXT
841         union iwreq_data wrqu;
842 #endif
843
844         ASSERT_WDEV_LOCK(wdev);
845
846         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_STATION &&
847                     wdev->iftype != NL80211_IFTYPE_P2P_CLIENT))
848                 return;
849
850         if (wdev->current_bss) {
851                 cfg80211_unhold_bss(wdev->current_bss);
852                 cfg80211_put_bss(wdev->wiphy, &wdev->current_bss->pub);
853         }
854
855         wdev->current_bss = NULL;
856         wdev->ssid_len = 0;
857
858         nl80211_send_disconnected(rdev, dev, reason, ie, ie_len, from_ap);
859
860         /*
861          * Delete all the keys ... pairwise keys can't really
862          * exist any more anyway, but default keys might.
863          */
864         if (rdev->ops->del_key)
865                 for (i = 0; i < 6; i++)
866                         rdev_del_key(rdev, dev, i, false, NULL);
867
868         rdev_set_qos_map(rdev, dev, NULL);
869
870 #ifdef CONFIG_CFG80211_WEXT
871         memset(&wrqu, 0, sizeof(wrqu));
872         wrqu.ap_addr.sa_family = ARPHRD_ETHER;
873         wireless_send_event(dev, SIOCGIWAP, &wrqu, NULL);
874         wdev->wext.connect.ssid_len = 0;
875 #endif
876
877         schedule_work(&cfg80211_disconnect_work);
878 }
879
880 void cfg80211_disconnected(struct net_device *dev, u16 reason,
881                            u8 *ie, size_t ie_len, gfp_t gfp)
882 {
883         struct wireless_dev *wdev = dev->ieee80211_ptr;
884         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
885         struct cfg80211_event *ev;
886         unsigned long flags;
887
888         ev = kzalloc(sizeof(*ev) + ie_len, gfp);
889         if (!ev)
890                 return;
891
892         ev->type = EVENT_DISCONNECTED;
893         ev->dc.ie = ((u8 *)ev) + sizeof(*ev);
894         ev->dc.ie_len = ie_len;
895         memcpy((void *)ev->dc.ie, ie, ie_len);
896         ev->dc.reason = reason;
897
898         spin_lock_irqsave(&wdev->event_lock, flags);
899         list_add_tail(&ev->list, &wdev->event_list);
900         spin_unlock_irqrestore(&wdev->event_lock, flags);
901         queue_work(cfg80211_wq, &rdev->event_work);
902 }
903 EXPORT_SYMBOL(cfg80211_disconnected);
904
905 /*
906  * API calls for nl80211/wext compatibility code
907  */
908 int cfg80211_connect(struct cfg80211_registered_device *rdev,
909                      struct net_device *dev,
910                      struct cfg80211_connect_params *connect,
911                      struct cfg80211_cached_keys *connkeys,
912                      const u8 *prev_bssid)
913 {
914         struct wireless_dev *wdev = dev->ieee80211_ptr;
915         int err;
916
917         ASSERT_WDEV_LOCK(wdev);
918
919         if (WARN_ON(wdev->connect_keys)) {
920                 kfree(wdev->connect_keys);
921                 wdev->connect_keys = NULL;
922         }
923
924         cfg80211_oper_and_ht_capa(&connect->ht_capa_mask,
925                                   rdev->wiphy.ht_capa_mod_mask);
926
927         if (connkeys && connkeys->def >= 0) {
928                 int idx;
929                 u32 cipher;
930
931                 idx = connkeys->def;
932                 cipher = connkeys->params[idx].cipher;
933                 /* If given a WEP key we may need it for shared key auth */
934                 if (cipher == WLAN_CIPHER_SUITE_WEP40 ||
935                     cipher == WLAN_CIPHER_SUITE_WEP104) {
936                         connect->key_idx = idx;
937                         connect->key = connkeys->params[idx].key;
938                         connect->key_len = connkeys->params[idx].key_len;
939
940                         /*
941                          * If ciphers are not set (e.g. when going through
942                          * iwconfig), we have to set them appropriately here.
943                          */
944                         if (connect->crypto.cipher_group == 0)
945                                 connect->crypto.cipher_group = cipher;
946
947                         if (connect->crypto.n_ciphers_pairwise == 0) {
948                                 connect->crypto.n_ciphers_pairwise = 1;
949                                 connect->crypto.ciphers_pairwise[0] = cipher;
950                         }
951                 }
952         }
953
954         wdev->connect_keys = connkeys;
955         memcpy(wdev->ssid, connect->ssid, connect->ssid_len);
956         wdev->ssid_len = connect->ssid_len;
957
958         if (!rdev->ops->connect)
959                 err = cfg80211_sme_connect(wdev, connect, prev_bssid);
960         else
961                 err = rdev_connect(rdev, dev, connect);
962
963         if (err) {
964                 wdev->connect_keys = NULL;
965                 wdev->ssid_len = 0;
966                 return err;
967         }
968
969         return 0;
970 }
971
972 int cfg80211_disconnect(struct cfg80211_registered_device *rdev,
973                         struct net_device *dev, u16 reason, bool wextev)
974 {
975         struct wireless_dev *wdev = dev->ieee80211_ptr;
976         int err = 0;
977
978         ASSERT_WDEV_LOCK(wdev);
979
980         kfree(wdev->connect_keys);
981         wdev->connect_keys = NULL;
982
983         if (wdev->conn)
984                 err = cfg80211_sme_disconnect(wdev, reason);
985         else if (!rdev->ops->disconnect)
986                 cfg80211_mlme_down(rdev, dev);
987         else if (wdev->current_bss)
988                 err = rdev_disconnect(rdev, dev, reason);
989
990         return err;
991 }