Merge mt76x{0,2}_mac_wcid_set_key into common code.

Signed-off-by: Stanislaw Gruszka <[email protected]>
---
 drivers/net/wireless/mediatek/mt76/mt76.h          |  3 ++
 drivers/net/wireless/mediatek/mt76/mt76x0/mac.c    | 41 ----------------------
 drivers/net/wireless/mediatek/mt76/mt76x0/mac.h    |  2 --
 drivers/net/wireless/mediatek/mt76/mt76x0/main.c   |  4 +--
 drivers/net/wireless/mediatek/mt76/mt76x2_common.c |  4 +--
 drivers/net/wireless/mediatek/mt76/mt76x2_mac.h    |  2 --
 .../net/wireless/mediatek/mt76/mt76x2_mac_common.c | 29 ---------------
 drivers/net/wireless/mediatek/mt76/mt76xx_mac.c    | 29 +++++++++++++++
 drivers/net/wireless/mediatek/mt76/mt76xx_mac.h    |  2 ++
 9 files changed, 38 insertions(+), 78 deletions(-)

diff --git a/drivers/net/wireless/mediatek/mt76/mt76.h 
b/drivers/net/wireless/mediatek/mt76/mt76.h
index 10959c4669bd..b02b4191ef37 100644
--- a/drivers/net/wireless/mediatek/mt76/mt76.h
+++ b/drivers/net/wireless/mediatek/mt76/mt76.h
@@ -423,6 +423,9 @@ struct mt76_rx_status {
 #define mt76_rmw_field(_dev, _reg, _field, _val)       \
        mt76_rmw(_dev, _reg, _field, FIELD_PREP(_field, _val))
 
+#define __mt76_rmw_field(_dev, _reg, _field, _val)     \
+       __mt76_rmw(_dev, _reg, _field, FIELD_PREP(_field, _val))
+
 #define mt76_hw(dev) (dev)->mt76.hw
 
 bool __mt76_poll(struct mt76_dev *dev, u32 offset, u32 mask, u32 val,
diff --git a/drivers/net/wireless/mediatek/mt76/mt76x0/mac.c 
b/drivers/net/wireless/mediatek/mt76/mt76x0/mac.c
index b026fc178f89..2fcdedd0cadc 100644
--- a/drivers/net/wireless/mediatek/mt76/mt76x0/mac.c
+++ b/drivers/net/wireless/mediatek/mt76/mt76x0/mac.c
@@ -565,44 +565,3 @@ u32 mt76x0_mac_process_rx(struct mt76x0_dev *dev, struct 
sk_buff *skb,
 
        return len;
 }
-
-int mt76x0_mac_wcid_set_key(struct mt76x0_dev *dev, u8 idx,
-                         struct ieee80211_key_conf *key)
-{
-       enum mt76xx_cipher_type cipher;
-       u8 key_data[32];
-       u8 iv_data[8];
-       u32 val;
-
-       cipher = mt76xx_mac_get_key_info(key, key_data);
-       if (cipher == MT_CIPHER_NONE && key)
-               return -EINVAL;
-
-       trace_mt76x0_set_key(&dev->mt76, idx);
-
-       mt76_wr_copy(dev, MT_WCID_KEY(idx), key_data, sizeof(key_data));
-
-       memset(iv_data, 0, sizeof(iv_data));
-       if (key) {
-               iv_data[3] = key->keyidx << 6;
-               if (cipher >= MT_CIPHER_TKIP) {
-                       /* Note: start with 1 to comply with spec,
-                        *       (see comment on common/cmm_wpa.c:4291).
-                        */
-                       iv_data[0] |= 1;
-                       iv_data[3] |= 0x20;
-               }
-       }
-       mt76_wr_copy(dev, MT_WCID_IV(idx), iv_data, sizeof(iv_data));
-
-       val = mt76_rr(dev, MT_WCID_ATTR(idx));
-       val &= ~MT_WCID_ATTR_PKEY_MODE & ~MT_WCID_ATTR_PKEY_MODE_EXT;
-       val |= FIELD_PREP(MT_WCID_ATTR_PKEY_MODE, cipher & 7) |
-              FIELD_PREP(MT_WCID_ATTR_PKEY_MODE_EXT, cipher >> 3);
-       val &= ~MT_WCID_ATTR_PAIRWISE;
-       val |= MT_WCID_ATTR_PAIRWISE *
-               !!(key && key->flags & IEEE80211_KEY_FLAG_PAIRWISE);
-       mt76_wr(dev, MT_WCID_ATTR(idx), val);
-
-       return 0;
-}
diff --git a/drivers/net/wireless/mediatek/mt76/mt76x0/mac.h 
b/drivers/net/wireless/mediatek/mt76/mt76x0/mac.h
index d44ef1e9a1dd..f8074c477044 100644
--- a/drivers/net/wireless/mediatek/mt76/mt76x0/mac.h
+++ b/drivers/net/wireless/mediatek/mt76/mt76x0/mac.h
@@ -138,8 +138,6 @@ struct mt76_txwi {
 
 u32 mt76x0_mac_process_rx(struct mt76x0_dev *dev, struct sk_buff *skb,
                        u8 *data, void *rxi);
-int mt76x0_mac_wcid_set_key(struct mt76x0_dev *dev, u8 idx,
-                         struct ieee80211_key_conf *key);
 void mt76x0_mac_wcid_set_rate(struct mt76x0_dev *dev, struct mt76_wcid *wcid,
                            const struct ieee80211_tx_rate *rate);
 
diff --git a/drivers/net/wireless/mediatek/mt76/mt76x0/main.c 
b/drivers/net/wireless/mediatek/mt76/mt76x0/main.c
index 6fbeae7afb93..a074e877f839 100644
--- a/drivers/net/wireless/mediatek/mt76/mt76x0/main.c
+++ b/drivers/net/wireless/mediatek/mt76/mt76x0/main.c
@@ -259,7 +259,7 @@ static int mt76x0_config(struct ieee80211_hw *hw, u32 
changed)
 
        if (!msta) {
                if (key || wcid->hw_key_idx == idx) {
-                       ret = mt76x0_mac_wcid_set_key(dev, wcid->idx, key);
+                       ret = mt76xx_mac_wcid_set_key(&dev->mt76, wcid->idx, 
key);
                        if (ret)
                                return ret;
                }
@@ -267,7 +267,7 @@ static int mt76x0_config(struct ieee80211_hw *hw, u32 
changed)
                return mt76xx_mac_shared_key_setup(&dev->mt76, mvif->idx, idx, 
key);
        }
 
-       return mt76x0_mac_wcid_set_key(dev, msta->wcid.idx, key);
+       return mt76xx_mac_wcid_set_key(&dev->mt76, msta->wcid.idx, key);
 }
 
 static int mt76x0_set_rts_threshold(struct ieee80211_hw *hw, u32 value)
diff --git a/drivers/net/wireless/mediatek/mt76/mt76x2_common.c 
b/drivers/net/wireless/mediatek/mt76/mt76x2_common.c
index d739f1ea1434..17bdbe5b48a9 100644
--- a/drivers/net/wireless/mediatek/mt76/mt76x2_common.c
+++ b/drivers/net/wireless/mediatek/mt76/mt76x2_common.c
@@ -219,7 +219,7 @@ int mt76x2_set_key(struct ieee80211_hw *hw, enum 
set_key_cmd cmd,
 
        if (!msta) {
                if (key || wcid->hw_key_idx == idx) {
-                       ret = mt76x2_mac_wcid_set_key(dev, wcid->idx, key);
+                       ret = mt76xx_mac_wcid_set_key(&dev->mt76, wcid->idx, 
key);
                        if (ret)
                                return ret;
                }
@@ -227,7 +227,7 @@ int mt76x2_set_key(struct ieee80211_hw *hw, enum 
set_key_cmd cmd,
                return mt76xx_mac_shared_key_setup(&dev->mt76, mvif->idx, idx, 
key);
        }
 
-       return mt76x2_mac_wcid_set_key(dev, msta->wcid.idx, key);
+       return mt76xx_mac_wcid_set_key(&dev->mt76, msta->wcid.idx, key);
 }
 EXPORT_SYMBOL_GPL(mt76x2_set_key);
 
diff --git a/drivers/net/wireless/mediatek/mt76/mt76x2_mac.h 
b/drivers/net/wireless/mediatek/mt76/mt76x2_mac.h
index 947b24718001..2aa885f71df4 100644
--- a/drivers/net/wireless/mediatek/mt76/mt76x2_mac.h
+++ b/drivers/net/wireless/mediatek/mt76/mt76x2_mac.h
@@ -169,8 +169,6 @@ void mt76x2_mac_write_txwi(struct mt76x2_dev *dev, struct 
mt76x2_txwi *txwi,
                           struct sk_buff *skb, struct mt76_wcid *wcid,
                           struct ieee80211_sta *sta, int len);
 void mt76x2_mac_wcid_setup(struct mt76x2_dev *dev, u8 idx, u8 vif_idx, u8 
*mac);
-int mt76x2_mac_wcid_set_key(struct mt76x2_dev *dev, u8 idx,
-                           struct ieee80211_key_conf *key);
 void mt76x2_mac_wcid_set_rate(struct mt76x2_dev *dev, struct mt76_wcid *wcid,
                              const struct ieee80211_tx_rate *rate);
 void mt76x2_mac_wcid_set_drop(struct mt76x2_dev *dev, u8 idx, bool drop);
diff --git a/drivers/net/wireless/mediatek/mt76/mt76x2_mac_common.c 
b/drivers/net/wireless/mediatek/mt76/mt76x2_mac_common.c
index a4099ca34dad..c218434af347 100644
--- a/drivers/net/wireless/mediatek/mt76/mt76x2_mac_common.c
+++ b/drivers/net/wireless/mediatek/mt76/mt76x2_mac_common.c
@@ -229,35 +229,6 @@ void mt76x2_send_tx_status(struct mt76x2_dev *dev,
 }
 EXPORT_SYMBOL_GPL(mt76x2_send_tx_status);
 
-int mt76x2_mac_wcid_set_key(struct mt76x2_dev *dev, u8 idx,
-                           struct ieee80211_key_conf *key)
-{
-       enum mt76xx_cipher_type cipher;
-       u8 key_data[32];
-       u8 iv_data[8];
-
-       cipher = mt76xx_mac_get_key_info(key, key_data);
-       if (cipher == MT_CIPHER_NONE && key)
-               return -EOPNOTSUPP;
-
-       mt76_rmw_field(dev, MT_WCID_ATTR(idx), MT_WCID_ATTR_PKEY_MODE, cipher);
-       mt76_wr_copy(dev, MT_WCID_KEY(idx), key_data, sizeof(key_data));
-
-       memset(iv_data, 0, sizeof(iv_data));
-       if (key) {
-               mt76_rmw_field(dev, MT_WCID_ATTR(idx), MT_WCID_ATTR_PAIRWISE,
-                              !!(key->flags & IEEE80211_KEY_FLAG_PAIRWISE));
-               iv_data[3] = key->keyidx << 6;
-               if (cipher >= MT_CIPHER_TKIP)
-                       iv_data[3] |= 0x20;
-       }
-
-       mt76_wr_copy(dev, MT_WCID_IV(idx), iv_data, sizeof(iv_data));
-
-       return 0;
-}
-EXPORT_SYMBOL_GPL(mt76x2_mac_wcid_set_key);
-
 static __le16
 mt76x2_mac_tx_rate_val(struct mt76x2_dev *dev,
                       const struct ieee80211_tx_rate *rate, u8 *nss_val)
diff --git a/drivers/net/wireless/mediatek/mt76/mt76xx_mac.c 
b/drivers/net/wireless/mediatek/mt76/mt76xx_mac.c
index f10d6f9a0402..4e1f491d48d9 100644
--- a/drivers/net/wireless/mediatek/mt76/mt76xx_mac.c
+++ b/drivers/net/wireless/mediatek/mt76/mt76xx_mac.c
@@ -67,3 +67,32 @@ int mt76xx_mac_shared_key_setup(struct mt76_dev *dev, u8 
vif_idx, u8 key_idx,
        return 0;
 }
 EXPORT_SYMBOL_GPL(mt76xx_mac_shared_key_setup);
+
+int mt76xx_mac_wcid_set_key(struct mt76_dev *dev, u8 idx,
+                           struct ieee80211_key_conf *key)
+{
+       enum mt76xx_cipher_type cipher;
+       u8 key_data[32];
+       u8 iv_data[8];
+
+       cipher = mt76xx_mac_get_key_info(key, key_data);
+       if (cipher == MT_CIPHER_NONE && key)
+               return -EOPNOTSUPP;
+
+       __mt76_wr_copy(dev, MT_WCID_KEY(idx), key_data, sizeof(key_data));
+       __mt76_rmw_field(dev, MT_WCID_ATTR(idx), MT_WCID_ATTR_PKEY_MODE, 
cipher);
+
+       memset(iv_data, 0, sizeof(iv_data));
+       if (key) {
+               __mt76_rmw_field(dev, MT_WCID_ATTR(idx), MT_WCID_ATTR_PAIRWISE,
+                                !!(key->flags & IEEE80211_KEY_FLAG_PAIRWISE));
+               iv_data[3] = key->keyidx << 6;
+               if (cipher >= MT_CIPHER_TKIP)
+                       iv_data[3] |= 0x20;
+       }
+
+       __mt76_wr_copy(dev, MT_WCID_IV(idx), iv_data, sizeof(iv_data));
+
+       return 0;
+}
+EXPORT_SYMBOL_GPL(mt76xx_mac_wcid_set_key);
diff --git a/drivers/net/wireless/mediatek/mt76/mt76xx_mac.h 
b/drivers/net/wireless/mediatek/mt76/mt76xx_mac.h
index 6377b31c4404..3570cb6ddd93 100644
--- a/drivers/net/wireless/mediatek/mt76/mt76xx_mac.h
+++ b/drivers/net/wireless/mediatek/mt76/mt76xx_mac.h
@@ -44,4 +44,6 @@ enum mt76xx_cipher_type
 
 int mt76xx_mac_shared_key_setup(struct mt76_dev *dev, u8 vif_idx, u8 key_idx,
                                struct ieee80211_key_conf *key);
+int mt76xx_mac_wcid_set_key(struct mt76_dev *dev, u8 idx,
+                           struct ieee80211_key_conf *key);
 #endif
-- 
1.9.3

Reply via email to