cfg80211: keep track of current_bss for userspace SME
[safe/jmp/linux-2.6] / net / wireless / sme.c
index 3abb047..3728d2b 100644 (file)
@@ -8,6 +8,8 @@
 #include <linux/etherdevice.h>
 #include <linux/if_arp.h>
 #include <linux/workqueue.h>
+#include <linux/wireless.h>
+#include <net/iw_handler.h>
 #include <net/cfg80211.h>
 #include <net/rtnetlink.h>
 #include "nl80211.h"
@@ -33,13 +35,15 @@ struct cfg80211_conn {
 
 static int cfg80211_conn_scan(struct wireless_dev *wdev)
 {
-       struct cfg80211_registered_device *drv = wiphy_to_dev(wdev->wiphy);
+       struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
        struct cfg80211_scan_request *request;
        int n_channels, err;
 
        ASSERT_RTNL();
+       ASSERT_RDEV_LOCK(rdev);
+       ASSERT_WDEV_LOCK(wdev);
 
-       if (drv->scan_req)
+       if (rdev->scan_req)
                return -EBUSY;
 
        if (wdev->conn->params.channel) {
@@ -84,17 +88,18 @@ static int cfg80211_conn_scan(struct wireless_dev *wdev)
                wdev->conn->params.ssid_len);
        request->ssids[0].ssid_len = wdev->conn->params.ssid_len;
 
-       request->ifidx = wdev->netdev->ifindex;
-       request->wiphy = &drv->wiphy;
+       request->dev = wdev->netdev;
+       request->wiphy = &rdev->wiphy;
 
-       drv->scan_req = request;
+       rdev->scan_req = request;
 
-       err = drv->ops->scan(wdev->wiphy, wdev->netdev, request);
+       err = rdev->ops->scan(wdev->wiphy, wdev->netdev, request);
        if (!err) {
                wdev->conn->state = CFG80211_CONN_SCANNING;
-               nl80211_send_scan_start(drv, wdev->netdev);
+               nl80211_send_scan_start(rdev, wdev->netdev);
+               dev_hold(wdev->netdev);
        } else {
-               drv->scan_req = NULL;
+               rdev->scan_req = NULL;
                kfree(request);
        }
        return err;
@@ -102,45 +107,50 @@ static int cfg80211_conn_scan(struct wireless_dev *wdev)
 
 static int cfg80211_conn_do_work(struct wireless_dev *wdev)
 {
-       struct cfg80211_registered_device *drv = wiphy_to_dev(wdev->wiphy);
-       union {
-               struct cfg80211_auth_request auth_req;
-               struct cfg80211_assoc_request assoc_req;
-       } u;
+       struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
+       struct cfg80211_connect_params *params;
+       int err;
 
-       memset(&u, 0, sizeof(u));
+       ASSERT_WDEV_LOCK(wdev);
 
        if (!wdev->conn)
                return 0;
 
+       params = &wdev->conn->params;
+
        switch (wdev->conn->state) {
        case CFG80211_CONN_SCAN_AGAIN:
                return cfg80211_conn_scan(wdev);
        case CFG80211_CONN_AUTHENTICATE_NEXT:
-               u.auth_req.chan = wdev->conn->params.channel;
-               u.auth_req.peer_addr = wdev->conn->params.bssid;
-               u.auth_req.ssid = wdev->conn->params.ssid;
-               u.auth_req.ssid_len = wdev->conn->params.ssid_len;
-               u.auth_req.auth_type = wdev->conn->params.auth_type;
-               u.auth_req.ie = NULL;
-               u.auth_req.ie_len = 0;
+               BUG_ON(!rdev->ops->auth);
                wdev->conn->state = CFG80211_CONN_AUTHENTICATING;
-               BUG_ON(!drv->ops->auth);
-               return drv->ops->auth(wdev->wiphy, wdev->netdev, &u.auth_req);
+               return __cfg80211_mlme_auth(rdev, wdev->netdev,
+                                           params->channel, params->auth_type,
+                                           params->bssid,
+                                           params->ssid, params->ssid_len,
+                                           NULL, 0,
+                                           params->key, params->key_len,
+                                           params->key_idx);
        case CFG80211_CONN_ASSOCIATE_NEXT:
-               u.assoc_req.chan = wdev->conn->params.channel;
-               u.assoc_req.peer_addr = wdev->conn->params.bssid;
-               u.assoc_req.ssid = wdev->conn->params.ssid;
-               u.assoc_req.ssid_len = wdev->conn->params.ssid_len;
-               u.assoc_req.ie = wdev->conn->params.ie;
-               u.assoc_req.ie_len = wdev->conn->params.ie_len;
-               u.assoc_req.use_mfp = false;
-               memcpy(&u.assoc_req.crypto, &wdev->conn->params.crypto,
-                       sizeof(u.assoc_req.crypto));
+               BUG_ON(!rdev->ops->assoc);
                wdev->conn->state = CFG80211_CONN_ASSOCIATING;
-               BUG_ON(!drv->ops->assoc);
-               return drv->ops->assoc(wdev->wiphy, wdev->netdev,
-                                       &u.assoc_req);
+               /*
+                * We could, later, implement roaming here and then actually
+                * set prev_bssid to non-NULL. But then we need to be aware
+                * that some APs don't like that -- so we'd need to retry
+                * the association.
+                */
+               err = __cfg80211_mlme_assoc(rdev, wdev->netdev,
+                                           params->channel, params->bssid,
+                                           NULL,
+                                           params->ssid, params->ssid_len,
+                                           params->ie, params->ie_len,
+                                           false, &params->crypto);
+               if (err)
+                       __cfg80211_mlme_deauth(rdev, wdev->netdev, params->bssid,
+                                              NULL, 0,
+                                              WLAN_REASON_DEAUTH_LEAVING);
+               return err;
        default:
                return 0;
        }
@@ -148,36 +158,47 @@ static int cfg80211_conn_do_work(struct wireless_dev *wdev)
 
 void cfg80211_conn_work(struct work_struct *work)
 {
-       struct cfg80211_registered_device *drv =
+       struct cfg80211_registered_device *rdev =
                container_of(work, struct cfg80211_registered_device, conn_work);
        struct wireless_dev *wdev;
 
        rtnl_lock();
-       mutex_lock(&drv->devlist_mtx);
+       cfg80211_lock_rdev(rdev);
+       mutex_lock(&rdev->devlist_mtx);
 
-       list_for_each_entry(wdev, &drv->netdev_list, list) {
-               if (!netif_running(wdev->netdev))
+       list_for_each_entry(wdev, &rdev->netdev_list, list) {
+               wdev_lock(wdev);
+               if (!netif_running(wdev->netdev)) {
+                       wdev_unlock(wdev);
                        continue;
-               if (wdev->sme_state != CFG80211_SME_CONNECTING)
+               }
+               if (wdev->sme_state != CFG80211_SME_CONNECTING) {
+                       wdev_unlock(wdev);
                        continue;
+               }
                if (cfg80211_conn_do_work(wdev))
-                       cfg80211_connect_result(wdev->netdev,
-                                               wdev->conn->params.bssid,
-                                               NULL, 0, NULL, 0,
-                                               WLAN_STATUS_UNSPECIFIED_FAILURE,
-                                               GFP_ATOMIC);
+                       __cfg80211_connect_result(
+                                       wdev->netdev,
+                                       wdev->conn->params.bssid,
+                                       NULL, 0, NULL, 0,
+                                       WLAN_STATUS_UNSPECIFIED_FAILURE,
+                                       false, NULL);
+               wdev_unlock(wdev);
        }
 
-       mutex_unlock(&drv->devlist_mtx);
+       mutex_unlock(&rdev->devlist_mtx);
+       cfg80211_unlock_rdev(rdev);
        rtnl_unlock();
 }
 
 static bool cfg80211_get_conn_bss(struct wireless_dev *wdev)
 {
-       struct cfg80211_registered_device *drv = wiphy_to_dev(wdev->wiphy);
+       struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
        struct cfg80211_bss *bss;
        u16 capa = WLAN_CAPABILITY_ESS;
 
+       ASSERT_WDEV_LOCK(wdev);
+
        if (wdev->conn->params.privacy)
                capa |= WLAN_CAPABILITY_PRIVACY;
 
@@ -186,7 +207,6 @@ static bool cfg80211_get_conn_bss(struct wireless_dev *wdev)
                               wdev->conn->params.ssid_len,
                               WLAN_CAPABILITY_ESS | WLAN_CAPABILITY_PRIVACY,
                               capa);
-
        if (!bss)
                return false;
 
@@ -194,21 +214,23 @@ static bool cfg80211_get_conn_bss(struct wireless_dev *wdev)
        wdev->conn->params.bssid = wdev->conn->bssid;
        wdev->conn->params.channel = bss->channel;
        wdev->conn->state = CFG80211_CONN_AUTHENTICATE_NEXT;
-       schedule_work(&drv->conn_work);
+       schedule_work(&rdev->conn_work);
 
        cfg80211_put_bss(bss);
        return true;
 }
 
-void cfg80211_sme_scan_done(struct net_device *dev)
+static void __cfg80211_sme_scan_done(struct net_device *dev)
 {
        struct wireless_dev *wdev = dev->ieee80211_ptr;
-       struct cfg80211_registered_device *drv = wiphy_to_dev(wdev->wiphy);
+       struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
+
+       ASSERT_WDEV_LOCK(wdev);
 
        if (wdev->sme_state != CFG80211_SME_CONNECTING)
                return;
 
-       if (WARN_ON(!wdev->conn))
+       if (!wdev->conn)
                return;
 
        if (wdev->conn->state != CFG80211_CONN_SCANNING &&
@@ -218,17 +240,28 @@ void cfg80211_sme_scan_done(struct net_device *dev)
        if (!cfg80211_get_conn_bss(wdev)) {
                /* not found */
                if (wdev->conn->state == CFG80211_CONN_SCAN_AGAIN)
-                       schedule_work(&drv->conn_work);
+                       schedule_work(&rdev->conn_work);
                else
-                       cfg80211_connect_result(dev, wdev->conn->params.bssid,
-                                               NULL, 0, NULL, 0,
-                                               WLAN_STATUS_UNSPECIFIED_FAILURE,
-                                               GFP_ATOMIC);
-               return;
+                       __cfg80211_connect_result(
+                                       wdev->netdev,
+                                       wdev->conn->params.bssid,
+                                       NULL, 0, NULL, 0,
+                                       WLAN_STATUS_UNSPECIFIED_FAILURE,
+                                       false, NULL);
        }
 }
 
-void cfg80211_sme_rx_auth(struct net_device *dev, const u8 *buf, size_t len)
+void cfg80211_sme_scan_done(struct net_device *dev)
+{
+       struct wireless_dev *wdev = dev->ieee80211_ptr;
+
+       wdev_lock(wdev);
+       __cfg80211_sme_scan_done(dev);
+       wdev_unlock(wdev);
+}
+
+void cfg80211_sme_rx_auth(struct net_device *dev,
+                         const u8 *buf, size_t len)
 {
        struct wireless_dev *wdev = dev->ieee80211_ptr;
        struct wiphy *wiphy = wdev->wiphy;
@@ -236,6 +269,8 @@ void cfg80211_sme_rx_auth(struct net_device *dev, const u8 *buf, size_t len)
        struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)buf;
        u16 status_code = le16_to_cpu(mgmt->u.auth.status_code);
 
+       ASSERT_WDEV_LOCK(wdev);
+
        /* should only RX auth frames when connecting */
        if (wdev->sme_state != CFG80211_SME_CONNECTING)
                return;
@@ -249,8 +284,12 @@ void cfg80211_sme_rx_auth(struct net_device *dev, const u8 *buf, size_t len)
                /* select automatically between only open, shared, leap */
                switch (wdev->conn->params.auth_type) {
                case NL80211_AUTHTYPE_OPEN_SYSTEM:
-                       wdev->conn->params.auth_type =
-                               NL80211_AUTHTYPE_SHARED_KEY;
+                       if (wdev->connect_keys)
+                               wdev->conn->params.auth_type =
+                                       NL80211_AUTHTYPE_SHARED_KEY;
+                       else
+                               wdev->conn->params.auth_type =
+                                       NL80211_AUTHTYPE_NETWORK_EAP;
                        break;
                case NL80211_AUTHTYPE_SHARED_KEY:
                        wdev->conn->params.auth_type =
@@ -264,94 +303,148 @@ void cfg80211_sme_rx_auth(struct net_device *dev, const u8 *buf, size_t len)
                }
                wdev->conn->state = CFG80211_CONN_AUTHENTICATE_NEXT;
                schedule_work(&rdev->conn_work);
-       } else if (status_code != WLAN_STATUS_SUCCESS)
-               wdev->sme_state = CFG80211_SME_IDLE;
-       else if (wdev->sme_state == CFG80211_SME_CONNECTING &&
+       } else if (status_code != WLAN_STATUS_SUCCESS) {
+               __cfg80211_connect_result(dev, mgmt->bssid, NULL, 0, NULL, 0,
+                                         status_code, false, NULL);
+       } else if (wdev->sme_state == CFG80211_SME_CONNECTING &&
                 wdev->conn->state == CFG80211_CONN_AUTHENTICATING) {
                wdev->conn->state = CFG80211_CONN_ASSOCIATE_NEXT;
                schedule_work(&rdev->conn_work);
        }
 }
 
-void cfg80211_connect_result(struct net_device *dev, const u8 *bssid,
-                            const u8 *req_ie, size_t req_ie_len,
-                            const u8 *resp_ie, size_t resp_ie_len,
-                            u16 status, gfp_t gfp)
+void __cfg80211_connect_result(struct net_device *dev, const u8 *bssid,
+                              const u8 *req_ie, size_t req_ie_len,
+                              const u8 *resp_ie, size_t resp_ie_len,
+                              u16 status, bool wextev,
+                              struct cfg80211_bss *bss)
 {
        struct wireless_dev *wdev = dev->ieee80211_ptr;
-       struct cfg80211_bss *bss;
 #ifdef CONFIG_WIRELESS_EXT
        union iwreq_data wrqu;
 #endif
 
+       ASSERT_WDEV_LOCK(wdev);
+
        if (WARN_ON(wdev->iftype != NL80211_IFTYPE_STATION))
                return;
 
-       if (wdev->sme_state != CFG80211_SME_CONNECTING)
-               return;
+       if (wdev->sme_state == CFG80211_SME_CONNECTED)
+               nl80211_send_roamed(wiphy_to_dev(wdev->wiphy), dev,
+                                   bssid, req_ie, req_ie_len,
+                                   resp_ie, resp_ie_len, GFP_KERNEL);
+       else
+               nl80211_send_connect_result(wiphy_to_dev(wdev->wiphy), dev,
+                                           bssid, req_ie, req_ie_len,
+                                           resp_ie, resp_ie_len,
+                                           status, GFP_KERNEL);
+
+#ifdef CONFIG_WIRELESS_EXT
+       if (wextev) {
+               if (req_ie && status == WLAN_STATUS_SUCCESS) {
+                       memset(&wrqu, 0, sizeof(wrqu));
+                       wrqu.data.length = req_ie_len;
+                       wireless_send_event(dev, IWEVASSOCREQIE, &wrqu, req_ie);
+               }
+
+               if (resp_ie && status == WLAN_STATUS_SUCCESS) {
+                       memset(&wrqu, 0, sizeof(wrqu));
+                       wrqu.data.length = resp_ie_len;
+                       wireless_send_event(dev, IWEVASSOCRESPIE, &wrqu, resp_ie);
+               }
+
+               memset(&wrqu, 0, sizeof(wrqu));
+               wrqu.ap_addr.sa_family = ARPHRD_ETHER;
+               if (bssid && status == WLAN_STATUS_SUCCESS)
+                       memcpy(wrqu.ap_addr.sa_data, bssid, ETH_ALEN);
+               wireless_send_event(dev, SIOCGIWAP, &wrqu, NULL);
+       }
+#endif
 
        if (wdev->current_bss) {
                cfg80211_unhold_bss(wdev->current_bss);
-               cfg80211_put_bss(wdev->current_bss);
+               cfg80211_put_bss(&wdev->current_bss->pub);
                wdev->current_bss = NULL;
        }
 
-       if (status == WLAN_STATUS_SUCCESS) {
-               bss = cfg80211_get_bss(wdev->wiphy, NULL, bssid,
-                                      wdev->ssid, wdev->ssid_len,
-                                      WLAN_CAPABILITY_ESS,
-                                      WLAN_CAPABILITY_ESS);
+       if (status == WLAN_STATUS_SUCCESS &&
+           wdev->sme_state == CFG80211_SME_IDLE)
+               goto success;
 
-               if (WARN_ON(!bss))
-                       return;
+       if (wdev->sme_state != CFG80211_SME_CONNECTING)
+               return;
 
-               cfg80211_hold_bss(bss);
-               wdev->current_bss = bss;
+       if (wdev->conn)
+               wdev->conn->state = CFG80211_CONN_IDLE;
 
-               wdev->sme_state = CFG80211_SME_CONNECTED;
-       } else {
+       if (status != WLAN_STATUS_SUCCESS) {
                wdev->sme_state = CFG80211_SME_IDLE;
+               kfree(wdev->conn);
+               wdev->conn = NULL;
+               kfree(wdev->connect_keys);
+               wdev->connect_keys = NULL;
+               return;
        }
 
-       if (wdev->conn)
-               wdev->conn->state = CFG80211_CONN_IDLE;
+ success:
+       if (!bss)
+               bss = cfg80211_get_bss(wdev->wiphy, NULL, bssid,
+                                      wdev->ssid, wdev->ssid_len,
+                                      WLAN_CAPABILITY_ESS,
+                                      WLAN_CAPABILITY_ESS);
 
-       nl80211_send_connect_result(wiphy_to_dev(wdev->wiphy), dev, bssid,
-                                   req_ie, req_ie_len, resp_ie, resp_ie_len,
-                                   status, gfp);
+       if (WARN_ON(!bss))
+               return;
 
-#ifdef CONFIG_WIRELESS_EXT
-       if (req_ie && status == WLAN_STATUS_SUCCESS) {
-               memset(&wrqu, 0, sizeof(wrqu));
-               wrqu.data.length = req_ie_len;
-               wireless_send_event(dev, IWEVASSOCRESPIE, &wrqu, req_ie);
-       }
+       cfg80211_hold_bss(bss_from_pub(bss));
+       wdev->current_bss = bss_from_pub(bss);
 
-       if (resp_ie && status == WLAN_STATUS_SUCCESS) {
-               memset(&wrqu, 0, sizeof(wrqu));
-               wrqu.data.length = resp_ie_len;
-               wireless_send_event(dev, IWEVASSOCRESPIE, &wrqu, resp_ie);
-       }
+       wdev->sme_state = CFG80211_SME_CONNECTED;
+       cfg80211_upload_connect_keys(wdev);
+}
 
-       memset(&wrqu, 0, sizeof(wrqu));
-       wrqu.ap_addr.sa_family = ARPHRD_ETHER;
-       if (bssid && status == WLAN_STATUS_SUCCESS)
-               memcpy(wrqu.ap_addr.sa_data, bssid, ETH_ALEN);
-       wireless_send_event(dev, SIOCGIWAP, &wrqu, NULL);
-#endif
+void cfg80211_connect_result(struct net_device *dev, const u8 *bssid,
+                            const u8 *req_ie, size_t req_ie_len,
+                            const u8 *resp_ie, size_t resp_ie_len,
+                            u16 status, gfp_t gfp)
+{
+       struct wireless_dev *wdev = dev->ieee80211_ptr;
+       struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
+       struct cfg80211_event *ev;
+       unsigned long flags;
+
+       ev = kzalloc(sizeof(*ev) + req_ie_len + resp_ie_len, gfp);
+       if (!ev)
+               return;
+
+       ev->type = EVENT_CONNECT_RESULT;
+       memcpy(ev->cr.bssid, bssid, ETH_ALEN);
+       ev->cr.req_ie = ((u8 *)ev) + sizeof(*ev);
+       ev->cr.req_ie_len = req_ie_len;
+       memcpy((void *)ev->cr.req_ie, req_ie, req_ie_len);
+       ev->cr.resp_ie = ((u8 *)ev) + sizeof(*ev) + req_ie_len;
+       ev->cr.resp_ie_len = resp_ie_len;
+       memcpy((void *)ev->cr.resp_ie, resp_ie, resp_ie_len);
+       ev->cr.status = status;
+
+       spin_lock_irqsave(&wdev->event_lock, flags);
+       list_add_tail(&ev->list, &wdev->event_list);
+       spin_unlock_irqrestore(&wdev->event_lock, flags);
+       schedule_work(&rdev->event_work);
 }
 EXPORT_SYMBOL(cfg80211_connect_result);
 
-void cfg80211_roamed(struct net_device *dev, const u8 *bssid,
-                    const u8 *req_ie, size_t req_ie_len,
-                    const u8 *resp_ie, size_t resp_ie_len, gfp_t gfp)
+void __cfg80211_roamed(struct wireless_dev *wdev, const u8 *bssid,
+                      const u8 *req_ie, size_t req_ie_len,
+                      const u8 *resp_ie, size_t resp_ie_len)
 {
-       struct wireless_dev *wdev = dev->ieee80211_ptr;
        struct cfg80211_bss *bss;
 #ifdef CONFIG_WIRELESS_EXT
        union iwreq_data wrqu;
 #endif
 
+       ASSERT_WDEV_LOCK(wdev);
+
        if (WARN_ON(wdev->iftype != NL80211_IFTYPE_STATION))
                return;
 
@@ -364,7 +457,7 @@ void cfg80211_roamed(struct net_device *dev, const u8 *bssid,
        }
 
        cfg80211_unhold_bss(wdev->current_bss);
-       cfg80211_put_bss(wdev->current_bss);
+       cfg80211_put_bss(&wdev->current_bss->pub);
        wdev->current_bss = NULL;
 
        bss = cfg80211_get_bss(wdev->wiphy, NULL, bssid,
@@ -374,41 +467,76 @@ void cfg80211_roamed(struct net_device *dev, const u8 *bssid,
        if (WARN_ON(!bss))
                return;
 
-       cfg80211_hold_bss(bss);
-       wdev->current_bss = bss;
+       cfg80211_hold_bss(bss_from_pub(bss));
+       wdev->current_bss = bss_from_pub(bss);
 
-       nl80211_send_roamed(wiphy_to_dev(wdev->wiphy), dev, bssid,
-                           req_ie, req_ie_len, resp_ie, resp_ie_len, gfp);
+       nl80211_send_roamed(wiphy_to_dev(wdev->wiphy), wdev->netdev, bssid,
+                           req_ie, req_ie_len, resp_ie, resp_ie_len,
+                           GFP_KERNEL);
 
 #ifdef CONFIG_WIRELESS_EXT
        if (req_ie) {
                memset(&wrqu, 0, sizeof(wrqu));
                wrqu.data.length = req_ie_len;
-               wireless_send_event(dev, IWEVASSOCRESPIE, &wrqu, req_ie);
+               wireless_send_event(wdev->netdev, IWEVASSOCREQIE,
+                                   &wrqu, req_ie);
        }
 
        if (resp_ie) {
                memset(&wrqu, 0, sizeof(wrqu));
                wrqu.data.length = resp_ie_len;
-               wireless_send_event(dev, IWEVASSOCRESPIE, &wrqu, resp_ie);
+               wireless_send_event(wdev->netdev, IWEVASSOCRESPIE,
+                                   &wrqu, resp_ie);
        }
 
        memset(&wrqu, 0, sizeof(wrqu));
        wrqu.ap_addr.sa_family = ARPHRD_ETHER;
        memcpy(wrqu.ap_addr.sa_data, bssid, ETH_ALEN);
-       wireless_send_event(dev, SIOCGIWAP, &wrqu, NULL);
+       wireless_send_event(wdev->netdev, SIOCGIWAP, &wrqu, NULL);
 #endif
 }
+
+void cfg80211_roamed(struct net_device *dev, const u8 *bssid,
+                    const u8 *req_ie, size_t req_ie_len,
+                    const u8 *resp_ie, size_t resp_ie_len, gfp_t gfp)
+{
+       struct wireless_dev *wdev = dev->ieee80211_ptr;
+       struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
+       struct cfg80211_event *ev;
+       unsigned long flags;
+
+       ev = kzalloc(sizeof(*ev) + req_ie_len + resp_ie_len, gfp);
+       if (!ev)
+               return;
+
+       ev->type = EVENT_ROAMED;
+       memcpy(ev->rm.bssid, bssid, ETH_ALEN);
+       ev->rm.req_ie = ((u8 *)ev) + sizeof(*ev);
+       ev->rm.req_ie_len = req_ie_len;
+       memcpy((void *)ev->rm.req_ie, req_ie, req_ie_len);
+       ev->rm.resp_ie = ((u8 *)ev) + sizeof(*ev) + req_ie_len;
+       ev->rm.resp_ie_len = resp_ie_len;
+       memcpy((void *)ev->rm.resp_ie, resp_ie, resp_ie_len);
+
+       spin_lock_irqsave(&wdev->event_lock, flags);
+       list_add_tail(&ev->list, &wdev->event_list);
+       spin_unlock_irqrestore(&wdev->event_lock, flags);
+       schedule_work(&rdev->event_work);
+}
 EXPORT_SYMBOL(cfg80211_roamed);
 
-void __cfg80211_disconnected(struct net_device *dev, gfp_t gfp, u8 *ie,
+void __cfg80211_disconnected(struct net_device *dev, const u8 *ie,
                             size_t ie_len, u16 reason, bool from_ap)
 {
        struct wireless_dev *wdev = dev->ieee80211_ptr;
+       struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
+       int i;
 #ifdef CONFIG_WIRELESS_EXT
        union iwreq_data wrqu;
 #endif
 
+       ASSERT_WDEV_LOCK(wdev);
+
        if (WARN_ON(wdev->iftype != NL80211_IFTYPE_STATION))
                return;
 
@@ -417,7 +545,7 @@ void __cfg80211_disconnected(struct net_device *dev, gfp_t gfp, u8 *ie,
 
        if (wdev->current_bss) {
                cfg80211_unhold_bss(wdev->current_bss);
-               cfg80211_put_bss(wdev->current_bss);
+               cfg80211_put_bss(&wdev->current_bss->pub);
        }
 
        wdev->current_bss = NULL;
@@ -426,10 +554,19 @@ void __cfg80211_disconnected(struct net_device *dev, gfp_t gfp, u8 *ie,
        if (wdev->conn) {
                kfree(wdev->conn->ie);
                wdev->conn->ie = NULL;
+               kfree(wdev->conn);
+               wdev->conn = NULL;
        }
 
-       nl80211_send_disconnected(wiphy_to_dev(wdev->wiphy), dev,
-                                 reason, ie, ie_len, from_ap, gfp);
+       nl80211_send_disconnected(rdev, dev, reason, ie, ie_len, from_ap);
+
+       /*
+        * Delete all the keys ... pairwise keys can't really
+        * exist any more anyway, but default keys might.
+        */
+       if (rdev->ops->del_key)
+               for (i = 0; i < 6; i++)
+                       rdev->ops->del_key(wdev->wiphy, dev, i, NULL);
 
 #ifdef CONFIG_WIRELESS_EXT
        memset(&wrqu, 0, sizeof(wrqu));
@@ -441,30 +578,69 @@ void __cfg80211_disconnected(struct net_device *dev, gfp_t gfp, u8 *ie,
 void cfg80211_disconnected(struct net_device *dev, u16 reason,
                           u8 *ie, size_t ie_len, gfp_t gfp)
 {
-       __cfg80211_disconnected(dev, gfp, ie, ie_len, reason, true);
+       struct wireless_dev *wdev = dev->ieee80211_ptr;
+       struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
+       struct cfg80211_event *ev;
+       unsigned long flags;
+
+       ev = kzalloc(sizeof(*ev) + ie_len, gfp);
+       if (!ev)
+               return;
+
+       ev->type = EVENT_DISCONNECTED;
+       ev->dc.ie = ((u8 *)ev) + sizeof(*ev);
+       ev->dc.ie_len = ie_len;
+       memcpy((void *)ev->dc.ie, ie, ie_len);
+       ev->dc.reason = reason;
+
+       spin_lock_irqsave(&wdev->event_lock, flags);
+       list_add_tail(&ev->list, &wdev->event_list);
+       spin_unlock_irqrestore(&wdev->event_lock, flags);
+       schedule_work(&rdev->event_work);
 }
 EXPORT_SYMBOL(cfg80211_disconnected);
 
-int cfg80211_connect(struct cfg80211_registered_device *rdev,
-                    struct net_device *dev,
-                    struct cfg80211_connect_params *connect)
+int __cfg80211_connect(struct cfg80211_registered_device *rdev,
+                      struct net_device *dev,
+                      struct cfg80211_connect_params *connect,
+                      struct cfg80211_cached_keys *connkeys)
 {
-       int err;
        struct wireless_dev *wdev = dev->ieee80211_ptr;
+       int err;
+
+       ASSERT_WDEV_LOCK(wdev);
 
        if (wdev->sme_state != CFG80211_SME_IDLE)
                return -EALREADY;
 
+       if (WARN_ON(wdev->connect_keys)) {
+               kfree(wdev->connect_keys);
+               wdev->connect_keys = NULL;
+       }
+
+       if (connkeys && connkeys->def >= 0) {
+               int idx;
+
+               idx = connkeys->def;
+               /* If given a WEP key we may need it for shared key auth */
+               if (connkeys->params[idx].cipher == WLAN_CIPHER_SUITE_WEP40 ||
+                   connkeys->params[idx].cipher == WLAN_CIPHER_SUITE_WEP104) {
+                       connect->key_idx = idx;
+                       connect->key = connkeys->params[idx].key;
+                       connect->key_len = connkeys->params[idx].key_len;
+               }
+       }
+
        if (!rdev->ops->connect) {
                if (!rdev->ops->auth || !rdev->ops->assoc)
                        return -EOPNOTSUPP;
 
-               if (!wdev->conn) {
-                       wdev->conn = kzalloc(sizeof(*wdev->conn), GFP_KERNEL);
-                       if (!wdev->conn)
-                               return -ENOMEM;
-               } else
-                       memset(wdev->conn, 0, sizeof(*wdev->conn));
+               if (WARN_ON(wdev->conn))
+                       return -EINPROGRESS;
+
+               wdev->conn = kzalloc(sizeof(*wdev->conn), GFP_KERNEL);
+               if (!wdev->conn)
+                       return -ENOMEM;
 
                /*
                 * Copy all parameters, and treat explicitly IEs, BSSID, SSID.
@@ -479,8 +655,11 @@ int cfg80211_connect(struct cfg80211_registered_device *rdev,
                        wdev->conn->ie = kmemdup(connect->ie, connect->ie_len,
                                                GFP_KERNEL);
                        wdev->conn->params.ie = wdev->conn->ie;
-                       if (!wdev->conn->ie)
+                       if (!wdev->conn->ie) {
+                               kfree(wdev->conn);
+                               wdev->conn = NULL;
                                return -ENOMEM;
+                       }
                }
 
                if (connect->auth_type == NL80211_AUTHTYPE_AUTOMATIC) {
@@ -502,6 +681,7 @@ int cfg80211_connect(struct cfg80211_registered_device *rdev,
                        cfg80211_get_conn_bss(wdev);
 
                wdev->sme_state = CFG80211_SME_CONNECTING;
+               wdev->connect_keys = connkeys;
 
                /* we're good if we have both BSSID and channel */
                if (wdev->conn->params.bssid && wdev->conn->params.channel) {
@@ -520,14 +700,20 @@ int cfg80211_connect(struct cfg80211_registered_device *rdev,
                                wdev->conn->state = CFG80211_CONN_SCAN_AGAIN;
                        }
                }
-               if (err)
+               if (err) {
+                       kfree(wdev->conn);
+                       wdev->conn = NULL;
                        wdev->sme_state = CFG80211_SME_IDLE;
+                       wdev->connect_keys = NULL;
+               }
 
                return err;
        } else {
                wdev->sme_state = CFG80211_SME_CONNECTING;
+               wdev->connect_keys = connkeys;
                err = rdev->ops->connect(&rdev->wiphy, dev, connect);
                if (err) {
+                       wdev->connect_keys = NULL;
                        wdev->sme_state = CFG80211_SME_IDLE;
                        return err;
                }
@@ -539,41 +725,57 @@ int cfg80211_connect(struct cfg80211_registered_device *rdev,
        }
 }
 
-int cfg80211_disconnect(struct cfg80211_registered_device *rdev,
-                       struct net_device *dev, u16 reason)
+int cfg80211_connect(struct cfg80211_registered_device *rdev,
+                    struct net_device *dev,
+                    struct cfg80211_connect_params *connect,
+                    struct cfg80211_cached_keys *connkeys)
+{
+       int err;
+
+       wdev_lock(dev->ieee80211_ptr);
+       err = __cfg80211_connect(rdev, dev, connect, connkeys);
+       wdev_unlock(dev->ieee80211_ptr);
+
+       return err;
+}
+
+int __cfg80211_disconnect(struct cfg80211_registered_device *rdev,
+                         struct net_device *dev, u16 reason, bool wextev)
 {
        struct wireless_dev *wdev = dev->ieee80211_ptr;
        int err;
 
+       ASSERT_WDEV_LOCK(wdev);
+
        if (wdev->sme_state == CFG80211_SME_IDLE)
                return -EINVAL;
 
+       kfree(wdev->connect_keys);
+       wdev->connect_keys = NULL;
+
        if (!rdev->ops->disconnect) {
-               struct cfg80211_deauth_request deauth;
-               u8 bssid[ETH_ALEN];
+               if (!rdev->ops->deauth)
+                       return -EOPNOTSUPP;
 
-               /* internal bug. */
-               if (WARN_ON(!wdev->conn))
-                       return -EINVAL;
+               /* was it connected by userspace SME? */
+               if (!wdev->conn) {
+                       cfg80211_mlme_down(rdev, dev);
+                       return 0;
+               }
 
                if (wdev->sme_state == CFG80211_SME_CONNECTING &&
                    (wdev->conn->state == CFG80211_CONN_SCANNING ||
                     wdev->conn->state == CFG80211_CONN_SCAN_AGAIN)) {
                        wdev->sme_state = CFG80211_SME_IDLE;
+                       kfree(wdev->conn);
+                       wdev->conn = NULL;
                        return 0;
                }
 
-               if (!rdev->ops->deauth)
-                       return -EOPNOTSUPP;
-
-               memset(&deauth, 0, sizeof(deauth));
-
                /* wdev->conn->params.bssid must be set if > SCANNING */
-               memcpy(bssid, wdev->conn->params.bssid, ETH_ALEN);
-               deauth.peer_addr = bssid;
-               deauth.reason_code = reason;
-
-               err = rdev->ops->deauth(&rdev->wiphy, dev, &deauth);
+               err = __cfg80211_mlme_deauth(rdev, dev,
+                                            wdev->conn->params.bssid,
+                                            NULL, 0, reason);
                if (err)
                        return err;
        } else {
@@ -583,11 +785,56 @@ int cfg80211_disconnect(struct cfg80211_registered_device *rdev,
        }
 
        if (wdev->sme_state == CFG80211_SME_CONNECTED)
-               __cfg80211_disconnected(dev, GFP_KERNEL, NULL, 0, 0, false);
+               __cfg80211_disconnected(dev, NULL, 0, 0, false);
        else if (wdev->sme_state == CFG80211_SME_CONNECTING)
-               cfg80211_connect_result(dev, NULL, NULL, 0, NULL, 0,
-                                       WLAN_STATUS_UNSPECIFIED_FAILURE,
-                                       GFP_KERNEL);
+               __cfg80211_connect_result(dev, NULL, NULL, 0, NULL, 0,
+                                         WLAN_STATUS_UNSPECIFIED_FAILURE,
+                                         wextev, NULL);
 
        return 0;
 }
+
+int cfg80211_disconnect(struct cfg80211_registered_device *rdev,
+                       struct net_device *dev,
+                       u16 reason, bool wextev)
+{
+       int err;
+
+       wdev_lock(dev->ieee80211_ptr);
+       err = __cfg80211_disconnect(rdev, dev, reason, wextev);
+       wdev_unlock(dev->ieee80211_ptr);
+
+       return err;
+}
+
+void cfg80211_sme_disassoc(struct net_device *dev, int idx)
+{
+       struct wireless_dev *wdev = dev->ieee80211_ptr;
+       struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
+       u8 bssid[ETH_ALEN];
+
+       ASSERT_WDEV_LOCK(wdev);
+
+       if (!wdev->conn)
+               return;
+
+       if (wdev->conn->state == CFG80211_CONN_IDLE)
+               return;
+
+       /*
+        * Ok, so the association was made by this SME -- we don't
+        * want it any more so deauthenticate too.
+        */
+
+       if (!wdev->auth_bsses[idx])
+               return;
+
+       memcpy(bssid, wdev->auth_bsses[idx]->pub.bssid, ETH_ALEN);
+       if (__cfg80211_mlme_deauth(rdev, dev, bssid,
+                                  NULL, 0, WLAN_REASON_DEAUTH_LEAVING)) {
+               /* whatever -- assume gone anyway */
+               cfg80211_unhold_bss(wdev->auth_bsses[idx]);
+               cfg80211_put_bss(&wdev->auth_bsses[idx]->pub);
+               wdev->auth_bsses[idx] = NULL;
+       }
+}