On Sun, 2019-02-10 at 22:06 +0100, Alexander Wetzel wrote: > > - Enforce cipher does not change when replacing a key. is that actually required somehow? > + * @EXT_SET_KEY: a new key must be set but is only valid for decryption > + * @EXT_KEY_RX_TX: a key installed with @EXT_SET_KEY is becoming the > + * designated Rx/Tx key for the station Not sure I like the EXT_SET_KEY. There's also no "designated Rx key", is there? It's always selected by key ID. How about SET_KEY_RXONLY and SET_KEY_TX or something like that? > +static int ieee80211_set_tx_key(struct ieee80211_sub_if_data *sdata, > + const u8 *mac_addr, u8 key_idx) > +{ > + struct ieee80211_local *local = sdata->local; > + struct ieee80211_key *key; > + struct sta_info *sta; > + int ret; > + > + if (!wiphy_ext_feature_isset(local->hw.wiphy, > + NL80211_EXT_FEATURE_EXT_KEY_ID)) > + return -EINVAL; You set this, wouldn't it make more sense to check EXT_KEY_ID_NATIVE? Or maybe this is because of the next patch? > + sta = sta_info_get_bss(sdata, mac_addr); > + > + if (!sta) > + return -EINVAL; > + > + if (sta->ptk_idx == key_idx) > + return 0; > + > + mutex_lock(&local->key_mtx); > + key = key_mtx_dereference(local, sta->ptk[key_idx]); > + > + if (key && key->flags & KEY_FLAG_RX_ONLY) do you even need the flag? Isn't it equivalent to checking sta->ptk_idx != key->idx or so? Less data to maintain would be better. > + bool ext_native = ieee80211_hw_check(&local->hw, EXT_KEY_ID_NATIVE); you sort of only need this in the next patch, but I guess it doesn't matter that much > +int ieee80211_key_activate_tx(struct ieee80211_key *key) > +{ > + struct ieee80211_sub_if_data *sdata = key->sdata; > + struct sta_info *sta = key->sta; > + struct ieee80211_local *local = key->local; > + struct ieee80211_key *old; > + int ret; > + > + assert_key_lock(local); > + > + key->flags &= ~KEY_FLAG_RX_ONLY; > + > + if (!(key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE) || > + key->conf.flags & (IEEE80211_KEY_FLAG_GENERATE_MMIC | > + IEEE80211_KEY_FLAG_PUT_MIC_SPACE | > + IEEE80211_KEY_FLAG_RESERVE_TAILROOM)) > + increment_tailroom_need_count(sdata); > + > + if (key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE) { > + ret = drv_set_key(local, EXT_KEY_RX_TX, sdata, > + &sta->sta, &key->conf); > + if (ret) { > + sdata_err(sdata, > + "failed to activate key for Tx (%d, %pM)\n", > + key->conf.keyidx, sta->sta.addr); > + return ret; You've already cleared the RX_ONLY flag, which gets you inconsistent data now. > + } > + } > + > + old = key_mtx_dereference(local, sta->ptk[sta->ptk_idx]); > + sta->ptk_idx = key->conf.keyidx; but you set this only here. > - /* Stop TX till we are on the new key */ > + /* Stop Tx till we are on the new key */ Uh, I had to read that three times ... please don't make changes like that? :) > old_key->flags |= KEY_FLAG_TAINTED; > ieee80211_clear_fast_xmit(sta); > > - /* Aggregation sessions during rekey are complicated due to the > + /* Aggregation sessions during rekey are complicated by the similarly here, please don't make drive-by comment wording issues (also, I'm not sure I agree - the old version just treats "complicated" as an adjective, you treat it as a verb, but ultimately doesn't really matter? > #define NUM_DEFAULT_KEYS 4 > #define NUM_DEFAULT_MGMT_KEYS 2 > +#define INVALID_PTK_KEYIDX 2 /* Existing key slot never used by PTK keys */ We could also use something obviously wrong like 0xff? > +++ b/net/mac80211/tx.c > @@ -3000,23 +3000,15 @@ void ieee80211_check_fast_xmit(struct sta_info *sta) > switch (build.key->conf.cipher) { > case WLAN_CIPHER_SUITE_CCMP: > case WLAN_CIPHER_SUITE_CCMP_256: > - /* add fixed key ID */ > - if (gen_iv) { > - (build.hdr + build.hdr_len)[3] = > - 0x20 | (build.key->conf.keyidx << 6); > + if (gen_iv) > build.pn_offs = build.hdr_len; > - } > if (gen_iv || iv_spc) > build.hdr_len += IEEE80211_CCMP_HDR_LEN; > break; > case WLAN_CIPHER_SUITE_GCMP: > case WLAN_CIPHER_SUITE_GCMP_256: > - /* add fixed key ID */ > - if (gen_iv) { > - (build.hdr + build.hdr_len)[3] = > - 0x20 | (build.key->conf.keyidx << 6); > + if (gen_iv) > build.pn_offs = build.hdr_len; > - } > if (gen_iv || iv_spc) > build.hdr_len += IEEE80211_GCMP_HDR_LEN; > break; > @@ -3383,6 +3375,7 @@ static void ieee80211_xmit_fast_finish(struct ieee80211_sub_if_data *sdata, > pn = atomic64_inc_return(&key->conf.tx_pn); > crypto_hdr[0] = pn; > crypto_hdr[1] = pn >> 8; > + crypto_hdr[3] = 0x20 | (key->conf.keyidx << 6); > crypto_hdr[4] = pn >> 16; > crypto_hdr[5] = pn >> 24; > crypto_hdr[6] = pn >> 32; This shouldn't be needed, you do update the fast TX cache when changing the key? johannes