cfg80211: fix two buglets
[safe/jmp/linux-2.6] / net / wireless / sme.c
1 /*
2  * SME code for cfg80211's connect emulation.
3  *
4  * Copyright 2009       Johannes Berg <johannes@sipsolutions.net>
5  * Copyright (C) 2009   Intel Corporation. All rights reserved.
6  */
7
8 #include <linux/etherdevice.h>
9 #include <linux/if_arp.h>
10 #include <linux/workqueue.h>
11 #include <net/cfg80211.h>
12 #include <net/rtnetlink.h>
13 #include "nl80211.h"
14
15 struct cfg80211_conn {
16         struct cfg80211_connect_params params;
17         /* these are sub-states of the _CONNECTING sme_state */
18         enum {
19                 CFG80211_CONN_IDLE,
20                 CFG80211_CONN_SCANNING,
21                 CFG80211_CONN_SCAN_AGAIN,
22                 CFG80211_CONN_AUTHENTICATE_NEXT,
23                 CFG80211_CONN_AUTHENTICATING,
24                 CFG80211_CONN_ASSOCIATE_NEXT,
25                 CFG80211_CONN_ASSOCIATING,
26         } state;
27         u8 bssid[ETH_ALEN];
28         u8 *ie;
29         size_t ie_len;
30         bool auto_auth;
31 };
32
33
34 static int cfg80211_conn_scan(struct wireless_dev *wdev)
35 {
36         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
37         struct cfg80211_scan_request *request;
38         int n_channels, err;
39
40         ASSERT_RTNL();
41         ASSERT_RDEV_LOCK(rdev);
42         ASSERT_WDEV_LOCK(wdev);
43
44         if (rdev->scan_req)
45                 return -EBUSY;
46
47         if (wdev->conn->params.channel) {
48                 n_channels = 1;
49         } else {
50                 enum ieee80211_band band;
51                 n_channels = 0;
52
53                 for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
54                         if (!wdev->wiphy->bands[band])
55                                 continue;
56                         n_channels += wdev->wiphy->bands[band]->n_channels;
57                 }
58         }
59         request = kzalloc(sizeof(*request) + sizeof(request->ssids[0]) +
60                           sizeof(request->channels[0]) * n_channels,
61                           GFP_KERNEL);
62         if (!request)
63                 return -ENOMEM;
64
65         request->channels = (void *)((char *)request + sizeof(*request));
66         if (wdev->conn->params.channel)
67                 request->channels[0] = wdev->conn->params.channel;
68         else {
69                 int i = 0, j;
70                 enum ieee80211_band band;
71
72                 for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
73                         if (!wdev->wiphy->bands[band])
74                                 continue;
75                         for (j = 0; j < wdev->wiphy->bands[band]->n_channels;
76                              i++, j++)
77                                 request->channels[i] =
78                                         &wdev->wiphy->bands[band]->channels[j];
79                 }
80         }
81         request->n_channels = n_channels;
82         request->ssids = (void *)(request->channels + n_channels);
83         request->n_ssids = 1;
84
85         memcpy(request->ssids[0].ssid, wdev->conn->params.ssid,
86                 wdev->conn->params.ssid_len);
87         request->ssids[0].ssid_len = wdev->conn->params.ssid_len;
88
89         request->ifidx = wdev->netdev->ifindex;
90         request->wiphy = &rdev->wiphy;
91
92         rdev->scan_req = request;
93
94         err = rdev->ops->scan(wdev->wiphy, wdev->netdev, request);
95         if (!err) {
96                 wdev->conn->state = CFG80211_CONN_SCANNING;
97                 nl80211_send_scan_start(rdev, wdev->netdev);
98         } else {
99                 rdev->scan_req = NULL;
100                 kfree(request);
101         }
102         return err;
103 }
104
105 static int cfg80211_conn_do_work(struct wireless_dev *wdev)
106 {
107         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
108         struct cfg80211_connect_params *params;
109         int err;
110
111         ASSERT_WDEV_LOCK(wdev);
112
113         if (!wdev->conn)
114                 return 0;
115
116         params = &wdev->conn->params;
117
118         switch (wdev->conn->state) {
119         case CFG80211_CONN_SCAN_AGAIN:
120                 return cfg80211_conn_scan(wdev);
121         case CFG80211_CONN_AUTHENTICATE_NEXT:
122                 BUG_ON(!rdev->ops->auth);
123                 wdev->conn->state = CFG80211_CONN_AUTHENTICATING;
124                 return __cfg80211_mlme_auth(rdev, wdev->netdev,
125                                             params->channel, params->auth_type,
126                                             params->bssid,
127                                             params->ssid, params->ssid_len,
128                                             NULL, 0);
129         case CFG80211_CONN_ASSOCIATE_NEXT:
130                 BUG_ON(!rdev->ops->assoc);
131                 wdev->conn->state = CFG80211_CONN_ASSOCIATING;
132                 /*
133                  * We could, later, implement roaming here and then actually
134                  * set prev_bssid to non-NULL. But then we need to be aware
135                  * that some APs don't like that -- so we'd need to retry
136                  * the association.
137                  */
138                 err = __cfg80211_mlme_assoc(rdev, wdev->netdev,
139                                             params->channel, params->bssid,
140                                             NULL,
141                                             params->ssid, params->ssid_len,
142                                             params->ie, params->ie_len,
143                                             false, &params->crypto);
144                 if (err)
145                         __cfg80211_mlme_deauth(rdev, wdev->netdev, params->bssid,
146                                                NULL, 0,
147                                                WLAN_REASON_DEAUTH_LEAVING);
148                 return err;
149         default:
150                 return 0;
151         }
152 }
153
154 void cfg80211_conn_work(struct work_struct *work)
155 {
156         struct cfg80211_registered_device *rdev =
157                 container_of(work, struct cfg80211_registered_device, conn_work);
158         struct wireless_dev *wdev;
159
160         rtnl_lock();
161         cfg80211_lock_rdev(rdev);
162         mutex_lock(&rdev->devlist_mtx);
163
164         list_for_each_entry(wdev, &rdev->netdev_list, list) {
165                 wdev_lock(wdev);
166                 if (!netif_running(wdev->netdev)) {
167                         wdev_unlock(wdev);
168                         continue;
169                 }
170                 if (wdev->sme_state != CFG80211_SME_CONNECTING) {
171                         wdev_unlock(wdev);
172                         continue;
173                 }
174                 if (cfg80211_conn_do_work(wdev))
175                         __cfg80211_connect_result(
176                                         wdev->netdev,
177                                         wdev->conn->params.bssid,
178                                         NULL, 0, NULL, 0,
179                                         WLAN_STATUS_UNSPECIFIED_FAILURE,
180                                         false);
181                 wdev_unlock(wdev);
182         }
183
184         mutex_unlock(&rdev->devlist_mtx);
185         cfg80211_unlock_rdev(rdev);
186         rtnl_unlock();
187 }
188
189 static bool cfg80211_get_conn_bss(struct wireless_dev *wdev)
190 {
191         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
192         struct cfg80211_bss *bss;
193         u16 capa = WLAN_CAPABILITY_ESS;
194
195         ASSERT_WDEV_LOCK(wdev);
196
197         if (wdev->conn->params.privacy)
198                 capa |= WLAN_CAPABILITY_PRIVACY;
199
200         bss = cfg80211_get_bss(wdev->wiphy, NULL, wdev->conn->params.bssid,
201                                wdev->conn->params.ssid,
202                                wdev->conn->params.ssid_len,
203                                WLAN_CAPABILITY_ESS | WLAN_CAPABILITY_PRIVACY,
204                                capa);
205         if (!bss)
206                 return false;
207
208         memcpy(wdev->conn->bssid, bss->bssid, ETH_ALEN);
209         wdev->conn->params.bssid = wdev->conn->bssid;
210         wdev->conn->params.channel = bss->channel;
211         wdev->conn->state = CFG80211_CONN_AUTHENTICATE_NEXT;
212         schedule_work(&rdev->conn_work);
213
214         cfg80211_put_bss(bss);
215         return true;
216 }
217
218 static void __cfg80211_sme_scan_done(struct net_device *dev)
219 {
220         struct wireless_dev *wdev = dev->ieee80211_ptr;
221         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
222
223         ASSERT_WDEV_LOCK(wdev);
224
225         if (wdev->sme_state != CFG80211_SME_CONNECTING)
226                 return;
227
228         if (WARN_ON(!wdev->conn))
229                 return;
230
231         if (wdev->conn->state != CFG80211_CONN_SCANNING &&
232             wdev->conn->state != CFG80211_CONN_SCAN_AGAIN)
233                 return;
234
235         if (!cfg80211_get_conn_bss(wdev)) {
236                 /* not found */
237                 if (wdev->conn->state == CFG80211_CONN_SCAN_AGAIN)
238                         schedule_work(&rdev->conn_work);
239                 else
240                         __cfg80211_connect_result(
241                                         wdev->netdev,
242                                         wdev->conn->params.bssid,
243                                         NULL, 0, NULL, 0,
244                                         WLAN_STATUS_UNSPECIFIED_FAILURE,
245                                         false);
246         }
247 }
248
249 void cfg80211_sme_scan_done(struct net_device *dev)
250 {
251         struct wireless_dev *wdev = dev->ieee80211_ptr;
252
253         wdev_lock(wdev);
254         __cfg80211_sme_scan_done(dev);
255         wdev_unlock(wdev);
256 }
257
258 void cfg80211_sme_rx_auth(struct net_device *dev,
259                           const u8 *buf, size_t len)
260 {
261         struct wireless_dev *wdev = dev->ieee80211_ptr;
262         struct wiphy *wiphy = wdev->wiphy;
263         struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
264         struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)buf;
265         u16 status_code = le16_to_cpu(mgmt->u.auth.status_code);
266
267         ASSERT_WDEV_LOCK(wdev);
268
269         /* should only RX auth frames when connecting */
270         if (wdev->sme_state != CFG80211_SME_CONNECTING)
271                 return;
272
273         if (WARN_ON(!wdev->conn))
274                 return;
275
276         if (status_code == WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG &&
277             wdev->conn->auto_auth &&
278             wdev->conn->params.auth_type != NL80211_AUTHTYPE_NETWORK_EAP) {
279                 /* select automatically between only open, shared, leap */
280                 switch (wdev->conn->params.auth_type) {
281                 case NL80211_AUTHTYPE_OPEN_SYSTEM:
282                         wdev->conn->params.auth_type =
283                                 NL80211_AUTHTYPE_SHARED_KEY;
284                         break;
285                 case NL80211_AUTHTYPE_SHARED_KEY:
286                         wdev->conn->params.auth_type =
287                                 NL80211_AUTHTYPE_NETWORK_EAP;
288                         break;
289                 default:
290                         /* huh? */
291                         wdev->conn->params.auth_type =
292                                 NL80211_AUTHTYPE_OPEN_SYSTEM;
293                         break;
294                 }
295                 wdev->conn->state = CFG80211_CONN_AUTHENTICATE_NEXT;
296                 schedule_work(&rdev->conn_work);
297         } else if (status_code != WLAN_STATUS_SUCCESS) {
298                 __cfg80211_connect_result(dev, mgmt->bssid, NULL, 0, NULL, 0,
299                                           status_code, false);
300         } else if (wdev->sme_state == CFG80211_SME_CONNECTING &&
301                  wdev->conn->state == CFG80211_CONN_AUTHENTICATING) {
302                 wdev->conn->state = CFG80211_CONN_ASSOCIATE_NEXT;
303                 schedule_work(&rdev->conn_work);
304         }
305 }
306
307 void __cfg80211_connect_result(struct net_device *dev, const u8 *bssid,
308                                const u8 *req_ie, size_t req_ie_len,
309                                const u8 *resp_ie, size_t resp_ie_len,
310                                u16 status, bool wextev)
311 {
312         struct wireless_dev *wdev = dev->ieee80211_ptr;
313         struct cfg80211_bss *bss;
314 #ifdef CONFIG_WIRELESS_EXT
315         union iwreq_data wrqu;
316 #endif
317
318         ASSERT_WDEV_LOCK(wdev);
319
320         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_STATION))
321                 return;
322
323         if (wdev->sme_state == CFG80211_SME_CONNECTED)
324                 nl80211_send_roamed(wiphy_to_dev(wdev->wiphy), dev,
325                                     bssid, req_ie, req_ie_len,
326                                     resp_ie, resp_ie_len, GFP_KERNEL);
327         else
328                 nl80211_send_connect_result(wiphy_to_dev(wdev->wiphy), dev,
329                                             bssid, req_ie, req_ie_len,
330                                             resp_ie, resp_ie_len,
331                                             status, GFP_KERNEL);
332
333 #ifdef CONFIG_WIRELESS_EXT
334         if (wextev) {
335                 if (req_ie && status == WLAN_STATUS_SUCCESS) {
336                         memset(&wrqu, 0, sizeof(wrqu));
337                         wrqu.data.length = req_ie_len;
338                         wireless_send_event(dev, IWEVASSOCRESPIE, &wrqu, req_ie);
339                 }
340
341                 if (resp_ie && status == WLAN_STATUS_SUCCESS) {
342                         memset(&wrqu, 0, sizeof(wrqu));
343                         wrqu.data.length = resp_ie_len;
344                         wireless_send_event(dev, IWEVASSOCRESPIE, &wrqu, resp_ie);
345                 }
346
347                 memset(&wrqu, 0, sizeof(wrqu));
348                 wrqu.ap_addr.sa_family = ARPHRD_ETHER;
349                 if (bssid && status == WLAN_STATUS_SUCCESS)
350                         memcpy(wrqu.ap_addr.sa_data, bssid, ETH_ALEN);
351                 wireless_send_event(dev, SIOCGIWAP, &wrqu, NULL);
352         }
353 #endif
354
355         if (status == WLAN_STATUS_SUCCESS &&
356             wdev->sme_state == CFG80211_SME_IDLE) {
357                 wdev->sme_state = CFG80211_SME_CONNECTED;
358                 return;
359         }
360
361         if (wdev->sme_state != CFG80211_SME_CONNECTING)
362                 return;
363
364         if (wdev->current_bss) {
365                 cfg80211_unhold_bss(wdev->current_bss);
366                 cfg80211_put_bss(&wdev->current_bss->pub);
367                 wdev->current_bss = NULL;
368         }
369
370         if (wdev->conn)
371                 wdev->conn->state = CFG80211_CONN_IDLE;
372
373         if (status == WLAN_STATUS_SUCCESS) {
374                 bss = cfg80211_get_bss(wdev->wiphy, NULL, bssid,
375                                        wdev->ssid, wdev->ssid_len,
376                                        WLAN_CAPABILITY_ESS,
377                                        WLAN_CAPABILITY_ESS);
378
379                 if (WARN_ON(!bss))
380                         return;
381
382                 cfg80211_hold_bss(bss_from_pub(bss));
383                 wdev->current_bss = bss_from_pub(bss);
384
385                 wdev->sme_state = CFG80211_SME_CONNECTED;
386         } else {
387                 wdev->sme_state = CFG80211_SME_IDLE;
388                 kfree(wdev->conn);
389                 wdev->conn = NULL;
390         }
391 }
392
393 void cfg80211_connect_result(struct net_device *dev, const u8 *bssid,
394                              const u8 *req_ie, size_t req_ie_len,
395                              const u8 *resp_ie, size_t resp_ie_len,
396                              u16 status, gfp_t gfp)
397 {
398         struct wireless_dev *wdev = dev->ieee80211_ptr;
399         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
400         struct cfg80211_event *ev;
401         unsigned long flags;
402
403         ev = kzalloc(sizeof(*ev) + req_ie_len + resp_ie_len, gfp);
404         if (!ev)
405                 return;
406
407         ev->type = EVENT_CONNECT_RESULT;
408         memcpy(ev->cr.bssid, bssid, ETH_ALEN);
409         ev->cr.req_ie = ((u8 *)ev) + sizeof(*ev);
410         ev->cr.req_ie_len = req_ie_len;
411         memcpy((void *)ev->cr.req_ie, req_ie, req_ie_len);
412         ev->cr.resp_ie = ((u8 *)ev) + sizeof(*ev) + req_ie_len;
413         ev->cr.resp_ie_len = resp_ie_len;
414         memcpy((void *)ev->cr.resp_ie, resp_ie, resp_ie_len);
415         ev->cr.status = status;
416
417         spin_lock_irqsave(&wdev->event_lock, flags);
418         list_add_tail(&ev->list, &wdev->event_list);
419         spin_unlock_irqrestore(&wdev->event_lock, flags);
420         schedule_work(&rdev->event_work);
421 }
422 EXPORT_SYMBOL(cfg80211_connect_result);
423
424 void __cfg80211_roamed(struct wireless_dev *wdev, const u8 *bssid,
425                        const u8 *req_ie, size_t req_ie_len,
426                        const u8 *resp_ie, size_t resp_ie_len)
427 {
428         struct cfg80211_bss *bss;
429 #ifdef CONFIG_WIRELESS_EXT
430         union iwreq_data wrqu;
431 #endif
432
433         ASSERT_WDEV_LOCK(wdev);
434
435         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_STATION))
436                 return;
437
438         if (WARN_ON(wdev->sme_state != CFG80211_SME_CONNECTED))
439                 return;
440
441         /* internal error -- how did we get to CONNECTED w/o BSS? */
442         if (WARN_ON(!wdev->current_bss)) {
443                 return;
444         }
445
446         cfg80211_unhold_bss(wdev->current_bss);
447         cfg80211_put_bss(&wdev->current_bss->pub);
448         wdev->current_bss = NULL;
449
450         bss = cfg80211_get_bss(wdev->wiphy, NULL, bssid,
451                                wdev->ssid, wdev->ssid_len,
452                                WLAN_CAPABILITY_ESS, WLAN_CAPABILITY_ESS);
453
454         if (WARN_ON(!bss))
455                 return;
456
457         cfg80211_hold_bss(bss_from_pub(bss));
458         wdev->current_bss = bss_from_pub(bss);
459
460         nl80211_send_roamed(wiphy_to_dev(wdev->wiphy), wdev->netdev, bssid,
461                             req_ie, req_ie_len, resp_ie, resp_ie_len,
462                             GFP_KERNEL);
463
464 #ifdef CONFIG_WIRELESS_EXT
465         if (req_ie) {
466                 memset(&wrqu, 0, sizeof(wrqu));
467                 wrqu.data.length = req_ie_len;
468                 wireless_send_event(wdev->netdev, IWEVASSOCRESPIE,
469                                     &wrqu, req_ie);
470         }
471
472         if (resp_ie) {
473                 memset(&wrqu, 0, sizeof(wrqu));
474                 wrqu.data.length = resp_ie_len;
475                 wireless_send_event(wdev->netdev, IWEVASSOCRESPIE,
476                                     &wrqu, resp_ie);
477         }
478
479         memset(&wrqu, 0, sizeof(wrqu));
480         wrqu.ap_addr.sa_family = ARPHRD_ETHER;
481         memcpy(wrqu.ap_addr.sa_data, bssid, ETH_ALEN);
482         wireless_send_event(wdev->netdev, SIOCGIWAP, &wrqu, NULL);
483 #endif
484 }
485
486 void cfg80211_roamed(struct net_device *dev, const u8 *bssid,
487                      const u8 *req_ie, size_t req_ie_len,
488                      const u8 *resp_ie, size_t resp_ie_len, gfp_t gfp)
489 {
490         struct wireless_dev *wdev = dev->ieee80211_ptr;
491         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
492         struct cfg80211_event *ev;
493         unsigned long flags;
494
495         ev = kzalloc(sizeof(*ev) + req_ie_len + resp_ie_len, gfp);
496         if (!ev)
497                 return;
498
499         ev->type = EVENT_ROAMED;
500         memcpy(ev->rm.bssid, bssid, ETH_ALEN);
501         ev->rm.req_ie = ((u8 *)ev) + sizeof(*ev);
502         ev->rm.req_ie_len = req_ie_len;
503         memcpy((void *)ev->rm.req_ie, req_ie, req_ie_len);
504         ev->rm.resp_ie = ((u8 *)ev) + sizeof(*ev) + req_ie_len;
505         ev->rm.resp_ie_len = resp_ie_len;
506         memcpy((void *)ev->rm.resp_ie, resp_ie, resp_ie_len);
507
508         spin_lock_irqsave(&wdev->event_lock, flags);
509         list_add_tail(&ev->list, &wdev->event_list);
510         spin_unlock_irqrestore(&wdev->event_lock, flags);
511         schedule_work(&rdev->event_work);
512 }
513 EXPORT_SYMBOL(cfg80211_roamed);
514
515 void __cfg80211_disconnected(struct net_device *dev, const u8 *ie,
516                              size_t ie_len, u16 reason, bool from_ap)
517 {
518         struct wireless_dev *wdev = dev->ieee80211_ptr;
519 #ifdef CONFIG_WIRELESS_EXT
520         union iwreq_data wrqu;
521 #endif
522
523         ASSERT_WDEV_LOCK(wdev);
524
525         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_STATION))
526                 return;
527
528         if (WARN_ON(wdev->sme_state != CFG80211_SME_CONNECTED))
529                 return;
530
531         if (wdev->current_bss) {
532                 cfg80211_unhold_bss(wdev->current_bss);
533                 cfg80211_put_bss(&wdev->current_bss->pub);
534         }
535
536         wdev->current_bss = NULL;
537         wdev->sme_state = CFG80211_SME_IDLE;
538
539         if (wdev->conn) {
540                 kfree(wdev->conn->ie);
541                 wdev->conn->ie = NULL;
542                 kfree(wdev->conn);
543                 wdev->conn = NULL;
544         }
545
546         nl80211_send_disconnected(wiphy_to_dev(wdev->wiphy), dev,
547                                   reason, ie, ie_len, from_ap);
548
549 #ifdef CONFIG_WIRELESS_EXT
550         memset(&wrqu, 0, sizeof(wrqu));
551         wrqu.ap_addr.sa_family = ARPHRD_ETHER;
552         wireless_send_event(dev, SIOCGIWAP, &wrqu, NULL);
553 #endif
554 }
555
556 void cfg80211_disconnected(struct net_device *dev, u16 reason,
557                            u8 *ie, size_t ie_len, gfp_t gfp)
558 {
559         struct wireless_dev *wdev = dev->ieee80211_ptr;
560         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
561         struct cfg80211_event *ev;
562         unsigned long flags;
563
564         ev = kzalloc(sizeof(*ev) + ie_len, gfp);
565         if (!ev)
566                 return;
567
568         ev->type = EVENT_DISCONNECTED;
569         ev->dc.ie = ((u8 *)ev) + sizeof(*ev);
570         ev->dc.ie_len = ie_len;
571         memcpy((void *)ev->dc.ie, ie, ie_len);
572         ev->dc.reason = reason;
573
574         spin_lock_irqsave(&wdev->event_lock, flags);
575         list_add_tail(&ev->list, &wdev->event_list);
576         spin_unlock_irqrestore(&wdev->event_lock, flags);
577         schedule_work(&rdev->event_work);
578 }
579 EXPORT_SYMBOL(cfg80211_disconnected);
580
581 int __cfg80211_connect(struct cfg80211_registered_device *rdev,
582                        struct net_device *dev,
583                        struct cfg80211_connect_params *connect)
584 {
585         struct wireless_dev *wdev = dev->ieee80211_ptr;
586         int err;
587
588         ASSERT_WDEV_LOCK(wdev);
589
590         if (wdev->sme_state != CFG80211_SME_IDLE)
591                 return -EALREADY;
592
593         if (!rdev->ops->connect) {
594                 if (!rdev->ops->auth || !rdev->ops->assoc)
595                         return -EOPNOTSUPP;
596
597                 if (WARN_ON(wdev->conn))
598                         return -EINPROGRESS;
599
600                 wdev->conn = kzalloc(sizeof(*wdev->conn), GFP_KERNEL);
601                 if (!wdev->conn)
602                         return -ENOMEM;
603
604                 /*
605                  * Copy all parameters, and treat explicitly IEs, BSSID, SSID.
606                  */
607                 memcpy(&wdev->conn->params, connect, sizeof(*connect));
608                 if (connect->bssid) {
609                         wdev->conn->params.bssid = wdev->conn->bssid;
610                         memcpy(wdev->conn->bssid, connect->bssid, ETH_ALEN);
611                 }
612
613                 if (connect->ie) {
614                         wdev->conn->ie = kmemdup(connect->ie, connect->ie_len,
615                                                 GFP_KERNEL);
616                         wdev->conn->params.ie = wdev->conn->ie;
617                         if (!wdev->conn->ie) {
618                                 kfree(wdev->conn);
619                                 wdev->conn = NULL;
620                                 return -ENOMEM;
621                         }
622                 }
623
624                 if (connect->auth_type == NL80211_AUTHTYPE_AUTOMATIC) {
625                         wdev->conn->auto_auth = true;
626                         /* start with open system ... should mostly work */
627                         wdev->conn->params.auth_type =
628                                 NL80211_AUTHTYPE_OPEN_SYSTEM;
629                 } else {
630                         wdev->conn->auto_auth = false;
631                 }
632
633                 memcpy(wdev->ssid, connect->ssid, connect->ssid_len);
634                 wdev->ssid_len = connect->ssid_len;
635                 wdev->conn->params.ssid = wdev->ssid;
636                 wdev->conn->params.ssid_len = connect->ssid_len;
637
638                 /* don't care about result -- but fill bssid & channel */
639                 if (!wdev->conn->params.bssid || !wdev->conn->params.channel)
640                         cfg80211_get_conn_bss(wdev);
641
642                 wdev->sme_state = CFG80211_SME_CONNECTING;
643
644                 /* we're good if we have both BSSID and channel */
645                 if (wdev->conn->params.bssid && wdev->conn->params.channel) {
646                         wdev->conn->state = CFG80211_CONN_AUTHENTICATE_NEXT;
647                         err = cfg80211_conn_do_work(wdev);
648                 } else {
649                         /* otherwise we'll need to scan for the AP first */
650                         err = cfg80211_conn_scan(wdev);
651                         /*
652                          * If we can't scan right now, then we need to scan again
653                          * after the current scan finished, since the parameters
654                          * changed (unless we find a good AP anyway).
655                          */
656                         if (err == -EBUSY) {
657                                 err = 0;
658                                 wdev->conn->state = CFG80211_CONN_SCAN_AGAIN;
659                         }
660                 }
661                 if (err) {
662                         kfree(wdev->conn);
663                         wdev->conn = NULL;
664                         wdev->sme_state = CFG80211_SME_IDLE;
665                 }
666
667                 return err;
668         } else {
669                 wdev->sme_state = CFG80211_SME_CONNECTING;
670                 err = rdev->ops->connect(&rdev->wiphy, dev, connect);
671                 if (err) {
672                         wdev->sme_state = CFG80211_SME_IDLE;
673                         return err;
674                 }
675
676                 memcpy(wdev->ssid, connect->ssid, connect->ssid_len);
677                 wdev->ssid_len = connect->ssid_len;
678
679                 return 0;
680         }
681 }
682
683 int cfg80211_connect(struct cfg80211_registered_device *rdev,
684                      struct net_device *dev,
685                      struct cfg80211_connect_params *connect)
686 {
687         int err;
688
689         wdev_lock(dev->ieee80211_ptr);
690         err = __cfg80211_connect(rdev, dev, connect);
691         wdev_unlock(dev->ieee80211_ptr);
692
693         return err;
694 }
695
696 int __cfg80211_disconnect(struct cfg80211_registered_device *rdev,
697                           struct net_device *dev, u16 reason, bool wextev)
698 {
699         struct wireless_dev *wdev = dev->ieee80211_ptr;
700         int err;
701
702         ASSERT_WDEV_LOCK(wdev);
703
704         if (wdev->sme_state == CFG80211_SME_IDLE)
705                 return -EINVAL;
706
707         if (!rdev->ops->disconnect) {
708                 if (!rdev->ops->deauth)
709                         return -EOPNOTSUPP;
710
711                 /* was it connected by userspace SME? */
712                 if (!wdev->conn) {
713                         cfg80211_mlme_down(rdev, dev);
714                         return 0;
715                 }
716
717                 if (wdev->sme_state == CFG80211_SME_CONNECTING &&
718                     (wdev->conn->state == CFG80211_CONN_SCANNING ||
719                      wdev->conn->state == CFG80211_CONN_SCAN_AGAIN)) {
720                         wdev->sme_state = CFG80211_SME_IDLE;
721                         kfree(wdev->conn);
722                         wdev->conn = NULL;
723                         return 0;
724                 }
725
726                 /* wdev->conn->params.bssid must be set if > SCANNING */
727                 err = __cfg80211_mlme_deauth(rdev, dev,
728                                              wdev->conn->params.bssid,
729                                              NULL, 0, reason);
730                 if (err)
731                         return err;
732         } else {
733                 err = rdev->ops->disconnect(&rdev->wiphy, dev, reason);
734                 if (err)
735                         return err;
736         }
737
738         if (wdev->sme_state == CFG80211_SME_CONNECTED)
739                 __cfg80211_disconnected(dev, NULL, 0, 0, false);
740         else if (wdev->sme_state == CFG80211_SME_CONNECTING)
741                 __cfg80211_connect_result(dev, NULL, NULL, 0, NULL, 0,
742                                           WLAN_STATUS_UNSPECIFIED_FAILURE,
743                                           wextev);
744
745         return 0;
746 }
747
748 int cfg80211_disconnect(struct cfg80211_registered_device *rdev,
749                         struct net_device *dev,
750                         u16 reason, bool wextev)
751 {
752         int err;
753
754         wdev_lock(dev->ieee80211_ptr);
755         err = __cfg80211_disconnect(rdev, dev, reason, wextev);
756         wdev_unlock(dev->ieee80211_ptr);
757
758         return err;
759 }
760
761 void cfg80211_sme_disassoc(struct net_device *dev, int idx)
762 {
763         struct wireless_dev *wdev = dev->ieee80211_ptr;
764         struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
765         u8 bssid[ETH_ALEN];
766
767         ASSERT_WDEV_LOCK(wdev);
768
769         if (!wdev->conn)
770                 return;
771
772         if (wdev->conn->state == CFG80211_CONN_IDLE)
773                 return;
774
775         /*
776          * Ok, so the association was made by this SME -- we don't
777          * want it any more so deauthenticate too.
778          */
779
780         if (!wdev->auth_bsses[idx])
781                 return;
782
783         memcpy(bssid, wdev->auth_bsses[idx]->pub.bssid, ETH_ALEN);
784         if (cfg80211_mlme_deauth(rdev, dev, bssid,
785                                  NULL, 0, WLAN_REASON_DEAUTH_LEAVING)) {
786                 /* whatever -- assume gone anyway */
787                 cfg80211_unhold_bss(wdev->auth_bsses[idx]);
788                 cfg80211_put_bss(&wdev->auth_bsses[idx]->pub);
789                 wdev->auth_bsses[idx] = NULL;
790         }
791 }