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_rdev(wdev->wiphy);
  63         struct cfg80211_scan_request *request;
  64         int n_channels, err;
  65
  66         ASSERT_RTNL();
  67         ASSERT_WDEV_LOCK(wdev);
  68
  69         if (rdev->scan_req || rdev->scan_msg)
  70                 return -EBUSY;
  71
  72         if (wdev->conn->params.channel)
  73                 n_channels = 1;
  74         else
  75                 n_channels = ieee80211_get_num_supported_channels(wdev->wiphy);
  76
  77         request = kzalloc(sizeof(*request) + sizeof(request->ssids[0]) +
  78                           sizeof(request->channels[0]) * n_channels,
  79                           GFP_KERNEL);
  80         if (!request)
  81                 return -ENOMEM;
  82
  83         if (wdev->conn->params.channel)
  84                 request->channels[0] = wdev->conn->params.channel;
  85         else {
  86                 int i = 0, j;
  87                 enum ieee80211_band band;
  88                 struct ieee80211_supported_band *bands;
  89                 struct ieee80211_channel *channel;
  90
  91                 for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
  92                         bands = wdev->wiphy->bands[band];
  93                         if (!bands)
  94                                 continue;
  95                         for (j = 0; j < bands->n_channels; j++) {
  96                                 channel = &bands->channels[j];
  97                                 if (channel->flags & IEEE80211_CHAN_DISABLED)
  98                                         continue;
  99                                 request->channels[i++] = channel;
 100                         }
 101                         request->rates[band] = (1 << bands->n_bitrates) - 1;
 102                 }
 103                 n_channels = i;
 104         }
 105         request->n_channels = n_channels;
 106         request->ssids = (void *)&request->channels[n_channels];
 107         request->n_ssids = 1;
 108
 109         memcpy(request->ssids[0].ssid, wdev->conn->params.ssid,
 110                 wdev->conn->params.ssid_len);
 111         request->ssids[0].ssid_len = wdev->conn->params.ssid_len;
 112
 113         request->wdev = wdev;
 114         request->wiphy = &rdev->wiphy;
 115         request->scan_start = jiffies;
 116
 117         rdev->scan_req = request;
 118
 119         err = rdev_scan(rdev, request);
 120         if (!err) {
 121                 wdev->conn->state = CFG80211_CONN_SCANNING;
 122                 nl80211_send_scan_start(rdev, wdev);
 123                 dev_hold(wdev->netdev);
 124         } else {
 125                 rdev->scan_req = NULL;
 126                 kfree(request);
 127         }
 128         return err;
 129 }
 130
 131 static int cfg80211_conn_do_work(struct wireless_dev *wdev)
 132 {
 133         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
 134         struct cfg80211_connect_params *params;
 135         struct cfg80211_assoc_request req = {};
 136         int err;
 137
 138         ASSERT_WDEV_LOCK(wdev);
 139
 140         if (!wdev->conn)
 141                 return 0;
 142
 143         params = &wdev->conn->params;
 144
 145         switch (wdev->conn->state) {
 146         case CFG80211_CONN_SCANNING:
 147                 /* didn't find it during scan ... */
 148                 return -ENOENT;
 149         case CFG80211_CONN_SCAN_AGAIN:
 150                 return cfg80211_conn_scan(wdev);
 151         case CFG80211_CONN_AUTHENTICATE_NEXT:
 152                 if (WARN_ON(!rdev->ops->auth))
 153                         return -EOPNOTSUPP;
 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                 if (WARN_ON(!rdev->ops->assoc))
 166                         return -EOPNOTSUPP;
 167                 wdev->conn->state = CFG80211_CONN_ASSOCIATING;
 168                 if (wdev->conn->prev_bssid_valid)
 169                         req.prev_bssid = wdev->conn->prev_bssid;
 170                 req.ie = params->ie;
 171                 req.ie_len = params->ie_len;
 172                 req.use_mfp = params->mfp != NL80211_MFP_NO;
 173                 req.crypto = params->crypto;
 174                 req.flags = params->flags;
 175                 req.ht_capa = params->ht_capa;
 176                 req.ht_capa_mask = params->ht_capa_mask;
 177                 req.vht_capa = params->vht_capa;
 178                 req.vht_capa_mask = params->vht_capa_mask;
 179
 180                 err = cfg80211_mlme_assoc(rdev, wdev->netdev, params->channel,
 181                                           params->bssid, params->ssid,
 182                                           params->ssid_len, &req);
 183                 if (err)
 184                         cfg80211_mlme_deauth(rdev, wdev->netdev, params->bssid,
 185                                              NULL, 0,
 186                                              WLAN_REASON_DEAUTH_LEAVING,
 187                                              false);
 188                 return err;
 189         case CFG80211_CONN_ASSOC_FAILED:
 190                 cfg80211_mlme_deauth(rdev, wdev->netdev, params->bssid,
 191                                      NULL, 0,
 192                                      WLAN_REASON_DEAUTH_LEAVING, false);
 193                 return -ENOTCONN;
 194         case CFG80211_CONN_DEAUTH:
 195                 cfg80211_mlme_deauth(rdev, wdev->netdev, params->bssid,
 196                                      NULL, 0,
 197                                      WLAN_REASON_DEAUTH_LEAVING, false);
 198                 /* free directly, disconnected event already sent */
 199                 cfg80211_sme_free(wdev);
 200                 return 0;
 201         default:
 202                 return 0;
 203         }
 204 }
 205
 206 void cfg80211_conn_work(struct work_struct *work)
 207 {
 208         struct cfg80211_registered_device *rdev =
 209                 container_of(work, struct cfg80211_registered_device, conn_work);
 210         struct wireless_dev *wdev;
 211         u8 bssid_buf[ETH_ALEN], *bssid = NULL;
 212
 213         rtnl_lock();
 214
 215         list_for_each_entry(wdev, &rdev->wdev_list, list) {
 216                 if (!wdev->netdev)
 217                         continue;
 218
 219                 wdev_lock(wdev);
 220                 if (!netif_running(wdev->netdev)) {
 221                         wdev_unlock(wdev);
 222                         continue;
 223                 }
 224                 if (!wdev->conn ||
 225                     wdev->conn->state == CFG80211_CONN_CONNECTED) {
 226                         wdev_unlock(wdev);
 227                         continue;
 228                 }
 229                 if (wdev->conn->params.bssid) {
 230                         memcpy(bssid_buf, wdev->conn->params.bssid, ETH_ALEN);
 231                         bssid = bssid_buf;
 232                 }
 233                 if (cfg80211_conn_do_work(wdev)) {
 234                         __cfg80211_connect_result(
 235                                         wdev->netdev, bssid,
 236                                         NULL, 0, NULL, 0,
 237                                         WLAN_STATUS_UNSPECIFIED_FAILURE,
 238                                         false, NULL);
 239                         cfg80211_sme_free(wdev);
 240                 }
 241                 wdev_unlock(wdev);
 242         }
 243
 244         rtnl_unlock();
 245 }
 246
 247 /* Returned bss is reference counted and must be cleaned up appropriately. */
 248 static struct cfg80211_bss *cfg80211_get_conn_bss(struct wireless_dev *wdev)
 249 {
 250         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
 251         struct cfg80211_bss *bss;
 252         u16 capa = WLAN_CAPABILITY_ESS;
 253
 254         ASSERT_WDEV_LOCK(wdev);
 255
 256         if (wdev->conn->params.privacy)
 257                 capa |= WLAN_CAPABILITY_PRIVACY;
 258
 259         bss = cfg80211_get_bss(wdev->wiphy, wdev->conn->params.channel,
 260                                wdev->conn->params.bssid,
 261                                wdev->conn->params.ssid,
 262                                wdev->conn->params.ssid_len,
 263                                WLAN_CAPABILITY_ESS | WLAN_CAPABILITY_PRIVACY,
 264                                capa);
 265         if (!bss)
 266                 return NULL;
 267
 268         memcpy(wdev->conn->bssid, bss->bssid, ETH_ALEN);
 269         wdev->conn->params.bssid = wdev->conn->bssid;
 270         wdev->conn->params.channel = bss->channel;
 271         wdev->conn->state = CFG80211_CONN_AUTHENTICATE_NEXT;
 272         schedule_work(&rdev->conn_work);
 273
 274         return bss;
 275 }
 276
 277 static void __cfg80211_sme_scan_done(struct net_device *dev)
 278 {
 279         struct wireless_dev *wdev = dev->ieee80211_ptr;
 280         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
 281         struct cfg80211_bss *bss;
 282
 283         ASSERT_WDEV_LOCK(wdev);
 284
 285         if (!wdev->conn)
 286                 return;
 287
 288         if (wdev->conn->state != CFG80211_CONN_SCANNING &&
 289             wdev->conn->state != CFG80211_CONN_SCAN_AGAIN)
 290                 return;
 291
 292         bss = cfg80211_get_conn_bss(wdev);
 293         if (bss)
 294                 cfg80211_put_bss(&rdev->wiphy, bss);
 295         else
 296                 schedule_work(&rdev->conn_work);
 297 }
 298
 299 void cfg80211_sme_scan_done(struct net_device *dev)
 300 {
 301         struct wireless_dev *wdev = dev->ieee80211_ptr;
 302
 303         wdev_lock(wdev);
 304         __cfg80211_sme_scan_done(dev);
 305         wdev_unlock(wdev);
 306 }
 307
 308 void cfg80211_sme_rx_auth(struct wireless_dev *wdev, const u8 *buf, size_t len)
 309 {
 310         struct wiphy *wiphy = wdev->wiphy;
 311         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wiphy);
 312         struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)buf;
 313         u16 status_code = le16_to_cpu(mgmt->u.auth.status_code);
 314
 315         ASSERT_WDEV_LOCK(wdev);
 316
 317         if (!wdev->conn || wdev->conn->state == CFG80211_CONN_CONNECTED)
 318                 return;
 319
 320         if (status_code == WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG &&
 321             wdev->conn->auto_auth &&
 322             wdev->conn->params.auth_type != NL80211_AUTHTYPE_NETWORK_EAP) {
 323                 /* select automatically between only open, shared, leap */
 324                 switch (wdev->conn->params.auth_type) {
 325                 case NL80211_AUTHTYPE_OPEN_SYSTEM:
 326                         if (wdev->connect_keys)
 327                                 wdev->conn->params.auth_type =
 328                                         NL80211_AUTHTYPE_SHARED_KEY;
 329                         else
 330                                 wdev->conn->params.auth_type =
 331                                         NL80211_AUTHTYPE_NETWORK_EAP;
 332                         break;
 333                 case NL80211_AUTHTYPE_SHARED_KEY:
 334                         wdev->conn->params.auth_type =
 335                                 NL80211_AUTHTYPE_NETWORK_EAP;
 336                         break;
 337                 default:
 338                         /* huh? */
 339                         wdev->conn->params.auth_type =
 340                                 NL80211_AUTHTYPE_OPEN_SYSTEM;
 341                         break;
 342                 }
 343                 wdev->conn->state = CFG80211_CONN_AUTHENTICATE_NEXT;
 344                 schedule_work(&rdev->conn_work);
 345         } else if (status_code != WLAN_STATUS_SUCCESS) {
 346                 __cfg80211_connect_result(wdev->netdev, mgmt->bssid,
 347                                           NULL, 0, NULL, 0,
 348                                           status_code, false, NULL);
 349         } else if (wdev->conn->state == CFG80211_CONN_AUTHENTICATING) {
 350                 wdev->conn->state = CFG80211_CONN_ASSOCIATE_NEXT;
 351                 schedule_work(&rdev->conn_work);
 352         }
 353 }
 354
 355 bool cfg80211_sme_rx_assoc_resp(struct wireless_dev *wdev, u16 status)
 356 {
 357         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
 358
 359         if (!wdev->conn)
 360                 return false;
 361
 362         if (status == WLAN_STATUS_SUCCESS) {
 363                 wdev->conn->state = CFG80211_CONN_CONNECTED;
 364                 return false;
 365         }
 366
 367         if (wdev->conn->prev_bssid_valid) {
 368                 /*
 369                  * Some stupid APs don't accept reassoc, so we
 370                  * need to fall back to trying regular assoc;
 371                  * return true so no event is sent to userspace.
 372                  */
 373                 wdev->conn->prev_bssid_valid = false;
 374                 wdev->conn->state = CFG80211_CONN_ASSOCIATE_NEXT;
 375                 schedule_work(&rdev->conn_work);
 376                 return true;
 377         }
 378
 379         wdev->conn->state = CFG80211_CONN_ASSOC_FAILED;
 380         schedule_work(&rdev->conn_work);
 381         return false;
 382 }
 383
 384 void cfg80211_sme_deauth(struct wireless_dev *wdev)
 385 {
 386         cfg80211_sme_free(wdev);
 387 }
 388
 389 void cfg80211_sme_auth_timeout(struct wireless_dev *wdev)
 390 {
 391         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
 392
 393         if (!wdev->conn)
 394                 return;
 395
 396         wdev->conn->state = CFG80211_CONN_AUTH_FAILED;
 397         schedule_work(&rdev->conn_work);
 398 }
 399
 400 void cfg80211_sme_disassoc(struct wireless_dev *wdev)
 401 {
 402         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
 403
 404         if (!wdev->conn)
 405                 return;
 406
 407         wdev->conn->state = CFG80211_CONN_DEAUTH;
 408         schedule_work(&rdev->conn_work);
 409 }
 410
 411 void cfg80211_sme_assoc_timeout(struct wireless_dev *wdev)
 412 {
 413         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
 414
 415         if (!wdev->conn)
 416                 return;
 417
 418         wdev->conn->state = CFG80211_CONN_ASSOC_FAILED;
 419         schedule_work(&rdev->conn_work);
 420 }
 421
 422 static int cfg80211_sme_connect(struct wireless_dev *wdev,
 423                                 struct cfg80211_connect_params *connect,
 424                                 const u8 *prev_bssid)
 425 {
 426         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
 427         struct cfg80211_bss *bss;
 428         int err;
 429
 430         if (!rdev->ops->auth || !rdev->ops->assoc)
 431                 return -EOPNOTSUPP;
 432
 433         if (wdev->current_bss)
 434                 return -EALREADY;
 435
 436         if (WARN_ON(wdev->conn))
 437                 return -EINPROGRESS;
 438
 439         wdev->conn = kzalloc(sizeof(*wdev->conn), GFP_KERNEL);
 440         if (!wdev->conn)
 441                 return -ENOMEM;
 442
 443         /*
 444          * Copy all parameters, and treat explicitly IEs, BSSID, SSID.
 445          */
 446         memcpy(&wdev->conn->params, connect, sizeof(*connect));
 447         if (connect->bssid) {
 448                 wdev->conn->params.bssid = wdev->conn->bssid;
 449                 memcpy(wdev->conn->bssid, connect->bssid, ETH_ALEN);
 450         }
 451
 452         if (connect->ie) {
 453                 wdev->conn->ie = kmemdup(connect->ie, connect->ie_len,
 454                                         GFP_KERNEL);
 455                 wdev->conn->params.ie = wdev->conn->ie;
 456                 if (!wdev->conn->ie) {
 457                         kfree(wdev->conn);
 458                         wdev->conn = NULL;
 459                         return -ENOMEM;
 460                 }
 461         }
 462
 463         if (connect->auth_type == NL80211_AUTHTYPE_AUTOMATIC) {
 464                 wdev->conn->auto_auth = true;
 465                 /* start with open system ... should mostly work */
 466                 wdev->conn->params.auth_type =
 467                         NL80211_AUTHTYPE_OPEN_SYSTEM;
 468         } else {
 469                 wdev->conn->auto_auth = false;
 470         }
 471
 472         wdev->conn->params.ssid = wdev->ssid;
 473         wdev->conn->params.ssid_len = wdev->ssid_len;
 474
 475         /* see if we have the bss already */
 476         bss = cfg80211_get_conn_bss(wdev);
 477
 478         if (prev_bssid) {
 479                 memcpy(wdev->conn->prev_bssid, prev_bssid, ETH_ALEN);
 480                 wdev->conn->prev_bssid_valid = true;
 481         }
 482
 483         /* we're good if we have a matching bss struct */
 484         if (bss) {
 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_rdev(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_rdev(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_rdev(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_rdev(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_rdev(wdev->wiphy),
 747                             wdev->netdev, bss->bssid,
 748                             req_ie, req_ie_len, resp_ie, resp_ie_len,
 749                             GFP_KERNEL);
 750
 751 #ifdef CONFIG_CFG80211_WEXT
 752         if (req_ie) {
 753                 memset(&wrqu, 0, sizeof(wrqu));
 754                 wrqu.data.length = req_ie_len;
 755                 wireless_send_event(wdev->netdev, IWEVASSOCREQIE,
 756                                     &wrqu, req_ie);
 757         }
 758
 759         if (resp_ie) {
 760                 memset(&wrqu, 0, sizeof(wrqu));
 761                 wrqu.data.length = resp_ie_len;
 762                 wireless_send_event(wdev->netdev, IWEVASSOCRESPIE,
 763                                     &wrqu, resp_ie);
 764         }
 765
 766         memset(&wrqu, 0, sizeof(wrqu));
 767         wrqu.ap_addr.sa_family = ARPHRD_ETHER;
 768         memcpy(wrqu.ap_addr.sa_data, bss->bssid, ETH_ALEN);
 769         memcpy(wdev->wext.prev_bssid, bss->bssid, ETH_ALEN);
 770         wdev->wext.prev_bssid_valid = true;
 771         wireless_send_event(wdev->netdev, SIOCGIWAP, &wrqu, NULL);
 772 #endif
 773
 774         return;
 775 out:
 776         cfg80211_put_bss(wdev->wiphy, bss);
 777 }
 778
 779 void cfg80211_roamed(struct net_device *dev,
 780                      struct ieee80211_channel *channel,
 781                      const u8 *bssid,
 782                      const u8 *req_ie, size_t req_ie_len,
 783                      const u8 *resp_ie, size_t resp_ie_len, gfp_t gfp)
 784 {
 785         struct wireless_dev *wdev = dev->ieee80211_ptr;
 786         struct cfg80211_bss *bss;
 787
 788         bss = cfg80211_get_bss(wdev->wiphy, channel, bssid, wdev->ssid,
 789                                wdev->ssid_len, WLAN_CAPABILITY_ESS,
 790                                WLAN_CAPABILITY_ESS);
 791         if (WARN_ON(!bss))
 792                 return;
 793
 794         cfg80211_roamed_bss(dev, bss, req_ie, req_ie_len, resp_ie,
 795                             resp_ie_len, gfp);
 796 }
 797 EXPORT_SYMBOL(cfg80211_roamed);
 798
 799 /* Consumes bss object one way or another */
 800 void cfg80211_roamed_bss(struct net_device *dev,
 801                          struct cfg80211_bss *bss, const u8 *req_ie,
 802                          size_t req_ie_len, const u8 *resp_ie,
 803                          size_t resp_ie_len, gfp_t gfp)
 804 {
 805         struct wireless_dev *wdev = dev->ieee80211_ptr;
 806         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
 807         struct cfg80211_event *ev;
 808         unsigned long flags;
 809
 810         if (WARN_ON(!bss))
 811                 return;
 812
 813         ev = kzalloc(sizeof(*ev) + req_ie_len + resp_ie_len, gfp);
 814         if (!ev) {
 815                 cfg80211_put_bss(wdev->wiphy, bss);
 816                 return;
 817         }
 818
 819         ev->type = EVENT_ROAMED;
 820         ev->rm.req_ie = ((u8 *)ev) + sizeof(*ev);
 821         ev->rm.req_ie_len = req_ie_len;
 822         memcpy((void *)ev->rm.req_ie, req_ie, req_ie_len);
 823         ev->rm.resp_ie = ((u8 *)ev) + sizeof(*ev) + req_ie_len;
 824         ev->rm.resp_ie_len = resp_ie_len;
 825         memcpy((void *)ev->rm.resp_ie, resp_ie, resp_ie_len);
 826         ev->rm.bss = bss;
 827
 828         spin_lock_irqsave(&wdev->event_lock, flags);
 829         list_add_tail(&ev->list, &wdev->event_list);
 830         spin_unlock_irqrestore(&wdev->event_lock, flags);
 831         queue_work(cfg80211_wq, &rdev->event_work);
 832 }
 833 EXPORT_SYMBOL(cfg80211_roamed_bss);
 834
 835 void __cfg80211_disconnected(struct net_device *dev, const u8 *ie,
 836                              size_t ie_len, u16 reason, bool from_ap)
 837 {
 838         struct wireless_dev *wdev = dev->ieee80211_ptr;
 839         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
 840         int i;
 841 #ifdef CONFIG_CFG80211_WEXT
 842         union iwreq_data wrqu;
 843 #endif
 844
 845         ASSERT_WDEV_LOCK(wdev);
 846
 847         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_STATION &&
 848                     wdev->iftype != NL80211_IFTYPE_P2P_CLIENT))
 849                 return;
 850
 851         if (wdev->current_bss) {
 852                 cfg80211_unhold_bss(wdev->current_bss);
 853                 cfg80211_put_bss(wdev->wiphy, &wdev->current_bss->pub);
 854         }
 855
 856         wdev->current_bss = NULL;
 857         wdev->ssid_len = 0;
 858
 859         nl80211_send_disconnected(rdev, dev, reason, ie, ie_len, from_ap);
 860
 861         /*
 862          * Delete all the keys ... pairwise keys can't really
 863          * exist any more anyway, but default keys might.
 864          */
 865         if (rdev->ops->del_key)
 866                 for (i = 0; i < 6; i++)
 867                         rdev_del_key(rdev, dev, i, false, NULL);
 868
 869         rdev_set_qos_map(rdev, dev, NULL);
 870
 871 #ifdef CONFIG_CFG80211_WEXT
 872         memset(&wrqu, 0, sizeof(wrqu));
 873         wrqu.ap_addr.sa_family = ARPHRD_ETHER;
 874         wireless_send_event(dev, SIOCGIWAP, &wrqu, NULL);
 875         wdev->wext.connect.ssid_len = 0;
 876 #endif
 877
 878         schedule_work(&cfg80211_disconnect_work);
 879 }
 880
 881 void cfg80211_disconnected(struct net_device *dev, u16 reason,
 882                            const u8 *ie, size_t ie_len, gfp_t gfp)
 883 {
 884         struct wireless_dev *wdev = dev->ieee80211_ptr;
 885         struct cfg80211_registered_device *rdev = wiphy_to_rdev(wdev->wiphy);
 886         struct cfg80211_event *ev;
 887         unsigned long flags;
 888
 889         ev = kzalloc(sizeof(*ev) + ie_len, gfp);
 890         if (!ev)
 891                 return;
 892
 893         ev->type = EVENT_DISCONNECTED;
 894         ev->dc.ie = ((u8 *)ev) + sizeof(*ev);
 895         ev->dc.ie_len = ie_len;
 896         memcpy((void *)ev->dc.ie, ie, ie_len);
 897         ev->dc.reason = reason;
 898
 899         spin_lock_irqsave(&wdev->event_lock, flags);
 900         list_add_tail(&ev->list, &wdev->event_list);
 901         spin_unlock_irqrestore(&wdev->event_lock, flags);
 902         queue_work(cfg80211_wq, &rdev->event_work);
 903 }
 904 EXPORT_SYMBOL(cfg80211_disconnected);
 905
 906 /*
 907  * API calls for nl80211/wext compatibility code
 908  */
 909 int cfg80211_connect(struct cfg80211_registered_device *rdev,
 910                      struct net_device *dev,
 911                      struct cfg80211_connect_params *connect,
 912                      struct cfg80211_cached_keys *connkeys,
 913                      const u8 *prev_bssid)
 914 {
 915         struct wireless_dev *wdev = dev->ieee80211_ptr;
 916         int err;
 917
 918         ASSERT_WDEV_LOCK(wdev);
 919
 920         if (WARN_ON(wdev->connect_keys)) {
 921                 kfree(wdev->connect_keys);
 922                 wdev->connect_keys = NULL;
 923         }
 924
 925         cfg80211_oper_and_ht_capa(&connect->ht_capa_mask,
 926                                   rdev->wiphy.ht_capa_mod_mask);
 927
 928         if (connkeys && connkeys->def >= 0) {
 929                 int idx;
 930                 u32 cipher;
 931
 932                 idx = connkeys->def;
 933                 cipher = connkeys->params[idx].cipher;
 934                 /* If given a WEP key we may need it for shared key auth */
 935                 if (cipher == WLAN_CIPHER_SUITE_WEP40 ||
 936                     cipher == WLAN_CIPHER_SUITE_WEP104) {
 937                         connect->key_idx = idx;
 938                         connect->key = connkeys->params[idx].key;
 939                         connect->key_len = connkeys->params[idx].key_len;
 940
 941                         /*
 942                          * If ciphers are not set (e.g. when going through
 943                          * iwconfig), we have to set them appropriately here.
 944                          */
 945                         if (connect->crypto.cipher_group == 0)
 946                                 connect->crypto.cipher_group = cipher;
 947
 948                         if (connect->crypto.n_ciphers_pairwise == 0) {
 949                                 connect->crypto.n_ciphers_pairwise = 1;
 950                                 connect->crypto.ciphers_pairwise[0] = cipher;
 951                         }
 952                 }
 953         }
 954
 955         wdev->connect_keys = connkeys;
 956         memcpy(wdev->ssid, connect->ssid, connect->ssid_len);
 957         wdev->ssid_len = connect->ssid_len;
 958
 959         if (!rdev->ops->connect)
 960                 err = cfg80211_sme_connect(wdev, connect, prev_bssid);
 961         else
 962                 err = rdev_connect(rdev, dev, connect);
 963
 964         if (err) {
 965                 wdev->connect_keys = NULL;
 966                 wdev->ssid_len = 0;
 967                 return err;
 968         }
 969
 970         return 0;
 971 }
 972
 973 int cfg80211_disconnect(struct cfg80211_registered_device *rdev,
 974                         struct net_device *dev, u16 reason, bool wextev)
 975 {
 976         struct wireless_dev *wdev = dev->ieee80211_ptr;
 977         int err = 0;
 978
 979         ASSERT_WDEV_LOCK(wdev);
 980
 981         kfree(wdev->connect_keys);
 982         wdev->connect_keys = NULL;
 983
 984         if (wdev->conn)
 985                 err = cfg80211_sme_disconnect(wdev, reason);
 986         else if (!rdev->ops->disconnect)
 987                 cfg80211_mlme_down(rdev, dev);
 988         else if (wdev->current_bss)
 989                 err = rdev_disconnect(rdev, dev, reason);
 990
 991         return err;
 992 }