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