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