1 files changed, 382 insertions, 13 deletions

diff --git a/net/wireless/sme.c b/net/wireless/sme.c
index fc117031d0bb..3abb04729873 100644
--- a/net/wireless/sme.c
+++ b/net/wireless/sme.c
@@ -12,6 +12,266 @@
 #include <net/rtnetlink.h>
 #include "nl80211.h"
 
+struct cfg80211_conn {
+        struct cfg80211_connect_params params;
+        /* these are sub-states of the _CONNECTING sme_state */
+        enum {
+                CFG80211_CONN_IDLE,
+                CFG80211_CONN_SCANNING,
+                CFG80211_CONN_SCAN_AGAIN,
+                CFG80211_CONN_AUTHENTICATE_NEXT,
+                CFG80211_CONN_AUTHENTICATING,
+                CFG80211_CONN_ASSOCIATE_NEXT,
+                CFG80211_CONN_ASSOCIATING,
+        } state;
+        u8 bssid[ETH_ALEN];
+        u8 *ie;
+        size_t ie_len;
+        bool auto_auth;
+};
+
+
+static int cfg80211_conn_scan(struct wireless_dev *wdev)
+{
+        struct cfg80211_registered_device *drv = wiphy_to_dev(wdev->wiphy);
+        struct cfg80211_scan_request *request;
+        int n_channels, err;
+
+        ASSERT_RTNL();
+
+        if (drv->scan_req)
+                return -EBUSY;
+
+        if (wdev->conn->params.channel) {
+                n_channels = 1;
+        } else {
+                enum ieee80211_band band;
+                n_channels = 0;
+
+                for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
+                        if (!wdev->wiphy->bands[band])
+                                continue;
+                        n_channels += wdev->wiphy->bands[band]->n_channels;
+                }
+        }
+        request = kzalloc(sizeof(*request) + sizeof(request->ssids[0]) +
+                          sizeof(request->channels[0]) * n_channels,
+                          GFP_KERNEL);
+        if (!request)
+                return -ENOMEM;
+
+        request->channels = (void *)((char *)request + sizeof(*request));
+        if (wdev->conn->params.channel)
+                request->channels[0] = wdev->conn->params.channel;
+        else {
+                int i = 0, j;
+                enum ieee80211_band band;
+
+                for (band = 0; band < IEEE80211_NUM_BANDS; band++) {
+                        if (!wdev->wiphy->bands[band])
+                                continue;
+                        for (j = 0; j < wdev->wiphy->bands[band]->n_channels;
+                             i++, j++)
+                                request->channels[i] =
+                                        &wdev->wiphy->bands[band]->channels[j];
+                }
+        }
+        request->n_channels = n_channels;
+        request->ssids = (void *)(request->channels + n_channels);
+        request->n_ssids = 1;
+
+        memcpy(request->ssids[0].ssid, wdev->conn->params.ssid,
+                wdev->conn->params.ssid_len);
+        request->ssids[0].ssid_len = wdev->conn->params.ssid_len;
+
+        request->ifidx = wdev->netdev->ifindex;
+        request->wiphy = &drv->wiphy;
+
+        drv->scan_req = request;
+
+        err = drv->ops->scan(wdev->wiphy, wdev->netdev, request);
+        if (!err) {
+                wdev->conn->state = CFG80211_CONN_SCANNING;
+                nl80211_send_scan_start(drv, wdev->netdev);
+        } else {
+                drv->scan_req = NULL;
+                kfree(request);
+        }
+        return err;
+}
+
+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;
+
+        memset(&u, 0, sizeof(u));
+
+        if (!wdev->conn)
+                return 0;
+
+        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;
+                wdev->conn->state = CFG80211_CONN_AUTHENTICATING;
+                BUG_ON(!drv->ops->auth);
+                return drv->ops->auth(wdev->wiphy, wdev->netdev, &u.auth_req);
+        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));
+                wdev->conn->state = CFG80211_CONN_ASSOCIATING;
+                BUG_ON(!drv->ops->assoc);
+                return drv->ops->assoc(wdev->wiphy, wdev->netdev,
+                                        &u.assoc_req);
+        default:
+                return 0;
+        }
+}
+
+void cfg80211_conn_work(struct work_struct *work)
+{
+        struct cfg80211_registered_device *drv =
+                container_of(work, struct cfg80211_registered_device, conn_work);
+        struct wireless_dev *wdev;
+
+        rtnl_lock();
+        mutex_lock(&drv->devlist_mtx);
+
+        list_for_each_entry(wdev, &drv->netdev_list, list) {
+                if (!netif_running(wdev->netdev))
+                        continue;
+                if (wdev->sme_state != CFG80211_SME_CONNECTING)
+                        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);
+        }
+
+        mutex_unlock(&drv->devlist_mtx);
+        rtnl_unlock();
+}
+
+static bool cfg80211_get_conn_bss(struct wireless_dev *wdev)
+{
+        struct cfg80211_registered_device *drv = wiphy_to_dev(wdev->wiphy);
+        struct cfg80211_bss *bss;
+        u16 capa = WLAN_CAPABILITY_ESS;
+
+        if (wdev->conn->params.privacy)
+                capa |= WLAN_CAPABILITY_PRIVACY;
+
+        bss = cfg80211_get_bss(wdev->wiphy, NULL, wdev->conn->params.bssid,
+                               wdev->conn->params.ssid,
+                               wdev->conn->params.ssid_len,
+                               WLAN_CAPABILITY_ESS | WLAN_CAPABILITY_PRIVACY,
+                               capa);
+
+        if (!bss)
+                return false;
+
+        memcpy(wdev->conn->bssid, bss->bssid, ETH_ALEN);
+        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);
+
+        cfg80211_put_bss(bss);
+        return true;
+}
+
+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);
+
+        if (wdev->sme_state != CFG80211_SME_CONNECTING)
+                return;
+
+        if (WARN_ON(!wdev->conn))
+                return;
+
+        if (wdev->conn->state != CFG80211_CONN_SCANNING &&
+            wdev->conn->state != CFG80211_CONN_SCAN_AGAIN)
+                return;
+
+        if (!cfg80211_get_conn_bss(wdev)) {
+                /* not found */
+                if (wdev->conn->state == CFG80211_CONN_SCAN_AGAIN)
+                        schedule_work(&drv->conn_work);
+                else
+                        cfg80211_connect_result(dev, wdev->conn->params.bssid,
+                                                NULL, 0, NULL, 0,
+                                                WLAN_STATUS_UNSPECIFIED_FAILURE,
+                                                GFP_ATOMIC);
+                return;
+        }
+}
+
+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;
+        struct cfg80211_registered_device *rdev = wiphy_to_dev(wiphy);
+        struct ieee80211_mgmt *mgmt = (struct ieee80211_mgmt *)buf;
+        u16 status_code = le16_to_cpu(mgmt->u.auth.status_code);
+
+        /* should only RX auth frames when connecting */
+        if (wdev->sme_state != CFG80211_SME_CONNECTING)
+                return;
+
+        if (WARN_ON(!wdev->conn))
+                return;
+
+        if (status_code == WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG &&
+            wdev->conn->auto_auth &&
+            wdev->conn->params.auth_type != NL80211_AUTHTYPE_NETWORK_EAP) {
+                /* 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;
+                        break;
+                case NL80211_AUTHTYPE_SHARED_KEY:
+                        wdev->conn->params.auth_type =
+                                NL80211_AUTHTYPE_NETWORK_EAP;
+                        break;
+                default:
+                        /* huh? */
+                        wdev->conn->params.auth_type =
+                                NL80211_AUTHTYPE_OPEN_SYSTEM;
+                        break;
+                }
+                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 &&
+                 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,
@@ -27,7 +287,7 @@ void cfg80211_connect_result(struct net_device *dev, const u8 *bssid,
         if (WARN_ON(wdev->iftype != NL80211_IFTYPE_STATION))
                 return;
 
-        if (WARN_ON(wdev->sme_state != CFG80211_SME_CONNECTING))
+        if (wdev->sme_state != CFG80211_SME_CONNECTING)
                 return;
 
         if (wdev->current_bss) {
@@ -53,6 +313,9 @@ void cfg80211_connect_result(struct net_device *dev, const u8 *bssid,
                 wdev->sme_state = CFG80211_SME_IDLE;
         }
 
+        if (wdev->conn)
+                wdev->conn->state = CFG80211_CONN_IDLE;
+
         nl80211_send_connect_result(wiphy_to_dev(wdev->wiphy), dev, bssid,
                                     req_ie, req_ie_len, resp_ie, resp_ie_len,
                                     status, gfp);
@@ -72,7 +335,7 @@ void cfg80211_connect_result(struct net_device *dev, const u8 *bssid,
 
         memset(&wrqu, 0, sizeof(wrqu));
         wrqu.ap_addr.sa_family = ARPHRD_ETHER;
-        if (bssid)
+        if (bssid && status == WLAN_STATUS_SUCCESS)
                 memcpy(wrqu.ap_addr.sa_data, bssid, ETH_ALEN);
         wireless_send_event(dev, SIOCGIWAP, &wrqu, NULL);
 #endif
@@ -138,9 +401,8 @@ void cfg80211_roamed(struct net_device *dev, const u8 *bssid,
 }
 EXPORT_SYMBOL(cfg80211_roamed);
 
-static void __cfg80211_disconnected(struct net_device *dev, gfp_t gfp,
-                                    u8 *ie, size_t ie_len, u16 reason,
-                                    bool from_ap)
+void __cfg80211_disconnected(struct net_device *dev, gfp_t gfp, u8 *ie,
+                             size_t ie_len, u16 reason, bool from_ap)
 {
         struct wireless_dev *wdev = dev->ieee80211_ptr;
 #ifdef CONFIG_WIRELESS_EXT
@@ -161,6 +423,11 @@ static void __cfg80211_disconnected(struct net_device *dev, gfp_t gfp,
         wdev->current_bss = NULL;
         wdev->sme_state = CFG80211_SME_IDLE;
 
+        if (wdev->conn) {
+                kfree(wdev->conn->ie);
+                wdev->conn->ie = NULL;
+        }
+
         nl80211_send_disconnected(wiphy_to_dev(wdev->wiphy), dev,
                                   reason, ie, ie_len, from_ap, gfp);
 
@@ -174,7 +441,7 @@ static void __cfg80211_disconnected(struct net_device *dev, gfp_t gfp,
 void cfg80211_disconnected(struct net_device *dev, u16 reason,
                            u8 *ie, size_t ie_len, gfp_t gfp)
 {
-        __cfg80211_disconnected(dev, reason, ie, ie_len, true, gfp);
+        __cfg80211_disconnected(dev, gfp, ie, ie_len, reason, true);
 }
 EXPORT_SYMBOL(cfg80211_disconnected);
 
@@ -189,7 +456,74 @@ int cfg80211_connect(struct cfg80211_registered_device *rdev,
                 return -EALREADY;
 
         if (!rdev->ops->connect) {
-                return -EOPNOTSUPP;
+                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));
+
+                /*
+                 * Copy all parameters, and treat explicitly IEs, BSSID, SSID.
+                 */
+                memcpy(&wdev->conn->params, connect, sizeof(*connect));
+                if (connect->bssid) {
+                        wdev->conn->params.bssid = wdev->conn->bssid;
+                        memcpy(wdev->conn->bssid, connect->bssid, ETH_ALEN);
+                }
+
+                if (connect->ie) {
+                        wdev->conn->ie = kmemdup(connect->ie, connect->ie_len,
+                                                GFP_KERNEL);
+                        wdev->conn->params.ie = wdev->conn->ie;
+                        if (!wdev->conn->ie)
+                                return -ENOMEM;
+                }
+
+                if (connect->auth_type == NL80211_AUTHTYPE_AUTOMATIC) {
+                        wdev->conn->auto_auth = true;
+                        /* start with open system ... should mostly work */
+                        wdev->conn->params.auth_type =
+                                NL80211_AUTHTYPE_OPEN_SYSTEM;
+                } else {
+                        wdev->conn->auto_auth = false;
+                }
+
+                memcpy(wdev->ssid, connect->ssid, connect->ssid_len);
+                wdev->ssid_len = connect->ssid_len;
+                wdev->conn->params.ssid = wdev->ssid;
+                wdev->conn->params.ssid_len = connect->ssid_len;
+
+                /* don't care about result -- but fill bssid & channel */
+                if (!wdev->conn->params.bssid || !wdev->conn->params.channel)
+                        cfg80211_get_conn_bss(wdev);
+
+                wdev->sme_state = CFG80211_SME_CONNECTING;
+
+                /* we're good if we have both BSSID and channel */
+                if (wdev->conn->params.bssid && wdev->conn->params.channel) {
+                        wdev->conn->state = CFG80211_CONN_AUTHENTICATE_NEXT;
+                        err = cfg80211_conn_do_work(wdev);
+                } else {
+                        /* otherwise we'll need to scan for the AP first */
+                        err = cfg80211_conn_scan(wdev);
+                        /*
+                         * If we can't scan right now, then we need to scan again
+                         * after the current scan finished, since the parameters
+                         * changed (unless we find a good AP anyway).
+                         */
+                        if (err == -EBUSY) {
+                                err = 0;
+                                wdev->conn->state = CFG80211_CONN_SCAN_AGAIN;
+                        }
+                }
+                if (err)
+                        wdev->sme_state = CFG80211_SME_IDLE;
+
+                return err;
         } else {
                 wdev->sme_state = CFG80211_SME_CONNECTING;
                 err = rdev->ops->connect(&rdev->wiphy, dev, connect);
@@ -197,28 +531,63 @@ int cfg80211_connect(struct cfg80211_registered_device *rdev,
                         wdev->sme_state = CFG80211_SME_IDLE;
                         return err;
                 }
-        }
 
-        memcpy(wdev->ssid, connect->ssid, connect->ssid_len);
-        wdev->ssid_len = connect->ssid_len;
+                memcpy(wdev->ssid, connect->ssid, connect->ssid_len);
+                wdev->ssid_len = connect->ssid_len;
 
-        return 0;
+                return 0;
+        }
 }
 
 int cfg80211_disconnect(struct cfg80211_registered_device *rdev,
                         struct net_device *dev, u16 reason)
 {
+        struct wireless_dev *wdev = dev->ieee80211_ptr;
         int err;
 
+        if (wdev->sme_state == CFG80211_SME_IDLE)
+                return -EINVAL;
+
         if (!rdev->ops->disconnect) {
-                return -EOPNOTSUPP;
+                struct cfg80211_deauth_request deauth;
+                u8 bssid[ETH_ALEN];
+
+                /* internal bug. */
+                if (WARN_ON(!wdev->conn))
+                        return -EINVAL;
+
+                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;
+                        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);
+                if (err)
+                        return err;
         } else {
                 err = rdev->ops->disconnect(&rdev->wiphy, dev, reason);
                 if (err)
                         return err;
         }
 
-        __cfg80211_disconnected(dev, 0, NULL, 0, false, GFP_KERNEL);
+        if (wdev->sme_state == CFG80211_SME_CONNECTED)
+                __cfg80211_disconnected(dev, GFP_KERNEL, 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);
 
         return 0;
 }