cfg80211: rework key operation

[GitHub/mt8127/android_kernel_alcatel_ttab.git] / net / wireless / ibss.c

diff --git a/net/wireless/ibss.c b/net/wireless/ibss.c
index 99ef9364b7e88c06d3062c3f0adce9f8570deddb..9394e78cd11fe0544a11329e41091976f3b46bfc 100644 (file)
--- a/net/wireless/ibss.c
+++ b/net/wireless/ibss.c
@@ -39,6 +39,8 @@ void __cfg80211_ibss_joined(struct net_device *dev, const u8 *bssid)
        cfg80211_hold_bss(bss_from_pub(bss));
        wdev->current_bss = bss_from_pub(bss);
 
+       cfg80211_upload_connect_keys(wdev);
+
        nl80211_send_ibss_bssid(wiphy_to_dev(wdev->wiphy), dev, bssid,
                                GFP_KERNEL);
 #ifdef CONFIG_WIRELESS_EXT
@@ -71,7 +73,8 @@ EXPORT_SYMBOL(cfg80211_ibss_joined);
 
 int __cfg80211_join_ibss(struct cfg80211_registered_device *rdev,
                         struct net_device *dev,
-                        struct cfg80211_ibss_params *params)
+                        struct cfg80211_ibss_params *params,
+                        struct cfg80211_cached_keys *connkeys)
 {
        struct wireless_dev *wdev = dev->ieee80211_ptr;
        int err;
@@ -81,13 +84,18 @@ int __cfg80211_join_ibss(struct cfg80211_registered_device *rdev,
        if (wdev->ssid_len)
                return -EALREADY;
 
+       if (WARN_ON(wdev->connect_keys))
+               kfree(wdev->connect_keys);
+       wdev->connect_keys = connkeys;
+
 #ifdef CONFIG_WIRELESS_EXT
        wdev->wext.ibss.channel = params->channel;
 #endif
        err = rdev->ops->join_ibss(&rdev->wiphy, dev, params);
-
-       if (err)
+       if (err) {
+               wdev->connect_keys = NULL;
                return err;
+       }
 
        memcpy(wdev->ssid, params->ssid, params->ssid_len);
        wdev->ssid_len = params->ssid_len;
@@ -97,13 +105,14 @@ int __cfg80211_join_ibss(struct cfg80211_registered_device *rdev,
 
 int cfg80211_join_ibss(struct cfg80211_registered_device *rdev,
                       struct net_device *dev,
-                      struct cfg80211_ibss_params *params)
+                      struct cfg80211_ibss_params *params,
+                      struct cfg80211_cached_keys *connkeys)
 {
        struct wireless_dev *wdev = dev->ieee80211_ptr;
        int err;
 
        wdev_lock(wdev);
-       err = __cfg80211_join_ibss(rdev, dev, params);
+       err = __cfg80211_join_ibss(rdev, dev, params, connkeys);
        wdev_unlock(wdev);
 
        return err;
@@ -112,9 +121,22 @@ int cfg80211_join_ibss(struct cfg80211_registered_device *rdev,
 static void __cfg80211_clear_ibss(struct net_device *dev, bool nowext)
 {
        struct wireless_dev *wdev = dev->ieee80211_ptr;
+       struct cfg80211_registered_device *rdev = wiphy_to_dev(wdev->wiphy);
+       int i;
 
        ASSERT_WDEV_LOCK(wdev);
 
+       kfree(wdev->connect_keys);
+       wdev->connect_keys = NULL;
+
+       /*
+        * 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);
+
        if (wdev->current_bss) {
                cfg80211_unhold_bss(wdev->current_bss);
                cfg80211_put_bss(&wdev->current_bss->pub);
@@ -172,11 +194,14 @@ int cfg80211_leave_ibss(struct cfg80211_registered_device *rdev,
 }
 
 #ifdef CONFIG_WIRELESS_EXT
-static int cfg80211_ibss_wext_join(struct cfg80211_registered_device *rdev,
-                                  struct wireless_dev *wdev)
+int cfg80211_ibss_wext_join(struct cfg80211_registered_device *rdev,
+                           struct wireless_dev *wdev)
 {
+       struct cfg80211_cached_keys *ck = NULL;
        enum ieee80211_band band;
-       int i;
+       int i, err;
+
+       ASSERT_WDEV_LOCK(wdev);
 
        if (!wdev->wext.ibss.beacon_interval)
                wdev->wext.ibss.beacon_interval = 100;
@@ -216,8 +241,24 @@ static int cfg80211_ibss_wext_join(struct cfg80211_registered_device *rdev,
        if (!netif_running(wdev->netdev))
                return 0;
 
-       return cfg80211_join_ibss(wiphy_to_dev(wdev->wiphy),
-                                 wdev->netdev, &wdev->wext.ibss);
+       if (wdev->wext.keys)
+               wdev->wext.keys->def = wdev->wext.default_key;
+
+       wdev->wext.ibss.privacy = wdev->wext.default_key != -1;
+
+       if (wdev->wext.keys) {
+               ck = kmemdup(wdev->wext.keys, sizeof(*ck), GFP_KERNEL);
+               if (!ck)
+                       return -ENOMEM;
+               for (i = 0; i < 6; i++)
+                       ck->params[i].key = ck->data[i];
+       }
+       err = __cfg80211_join_ibss(rdev, wdev->netdev,
+                                  &wdev->wext.ibss, ck);
+       if (err)
+               kfree(ck);
+
+       return err;
 }
 
 int cfg80211_ibss_wext_siwfreq(struct net_device *dev,
@@ -265,7 +306,11 @@ int cfg80211_ibss_wext_siwfreq(struct net_device *dev,
                wdev->wext.ibss.channel_fixed = false;
        }
 
-       return cfg80211_ibss_wext_join(wiphy_to_dev(wdev->wiphy), wdev);
+       wdev_lock(wdev);
+       err = cfg80211_ibss_wext_join(wiphy_to_dev(wdev->wiphy), wdev);
+       wdev_unlock(wdev);
+
+       return err;
 }
 /* temporary symbol - mark GPL - in the future the handler won't be */
 EXPORT_SYMBOL_GPL(cfg80211_ibss_wext_siwfreq);
@@ -333,7 +378,11 @@ int cfg80211_ibss_wext_siwessid(struct net_device *dev,
        memcpy(wdev->wext.ibss.ssid, ssid, len);
        wdev->wext.ibss.ssid_len = len;
 
-       return cfg80211_ibss_wext_join(wiphy_to_dev(wdev->wiphy), wdev);
+       wdev_lock(wdev);
+       err = cfg80211_ibss_wext_join(wiphy_to_dev(wdev->wiphy), wdev);
+       wdev_unlock(wdev);
+
+       return err;
 }
 /* temporary symbol - mark GPL - in the future the handler won't be */
 EXPORT_SYMBOL_GPL(cfg80211_ibss_wext_siwessid);
@@ -414,7 +463,11 @@ int cfg80211_ibss_wext_siwap(struct net_device *dev,
        } else
                wdev->wext.ibss.bssid = NULL;
 
-       return cfg80211_ibss_wext_join(wiphy_to_dev(wdev->wiphy), wdev);
+       wdev_lock(wdev);
+       err = cfg80211_ibss_wext_join(wiphy_to_dev(wdev->wiphy), wdev);
+       wdev_unlock(wdev);
+
+       return err;
 }
 /* temporary symbol - mark GPL - in the future the handler won't be */
 EXPORT_SYMBOL_GPL(cfg80211_ibss_wext_siwap);