]> git.proxmox.com Git - mirror_ubuntu-eoan-kernel.git/blame - drivers/net/wireless/intel/iwlwifi/mvm/sta.c
projects / mirror_ubuntu-eoan-kernel.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
iwlwifi: add the new 8275 series
[mirror_ubuntu-eoan-kernel.git] / drivers / net / wireless / intel / iwlwifi / mvm / sta.c
CommitLineData
8ca151b5
JB		 1/******************************************************************************
2 *
3 * This file is provided under a dual BSD/GPLv2 license. When using or
4 * redistributing this file, you may do so under either license.
5 *
6 * GPL LICENSE SUMMARY
7 *
fa7878e7
AO		 8 * Copyright(c) 2012 - 2015 Intel Corporation. All rights reserved.
9 * Copyright(c) 2013 - 2015 Intel Mobile Communications GmbH
854c5705 10 * Copyright(c) 2016 Intel Deutschland GmbH
8ca151b5
JB		 11 *
12 * This program is free software; you can redistribute it and/or modify
13 * it under the terms of version 2 of the GNU General Public License as
14 * published by the Free Software Foundation.
15 *
16 * This program is distributed in the hope that it will be useful, but
17 * WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
19 * General Public License for more details.
20 *
21 * You should have received a copy of the GNU General Public License
22 * along with this program; if not, write to the Free Software
23 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110,
24 * USA
25 *
26 * The full GNU General Public License is included in this distribution
410dc5aa 27 * in the file called COPYING.
8ca151b5
JB		 28 *
29 * Contact Information:
cb2f8277 30 * Intel Linux Wireless <linuxwifi@intel.com>
8ca151b5
JB		 31 * Intel Corporation, 5200 N.E. Elam Young Parkway, Hillsboro, OR 97124-6497
32 *
33 * BSD LICENSE
34 *
fa7878e7
AO		 35 * Copyright(c) 2012 - 2015 Intel Corporation. All rights reserved.
36 * Copyright(c) 2013 - 2015 Intel Mobile Communications GmbH
854c5705 37 * Copyright(c) 2016 Intel Deutschland GmbH
8ca151b5
JB		 38 * All rights reserved.
39 *
40 * Redistribution and use in source and binary forms, with or without
41 * modification, are permitted provided that the following conditions
42 * are met:
43 *
44 * * Redistributions of source code must retain the above copyright
45 * notice, this list of conditions and the following disclaimer.
46 * * Redistributions in binary form must reproduce the above copyright
47 * notice, this list of conditions and the following disclaimer in
48 * the documentation and/or other materials provided with the
49 * distribution.
50 * * Neither the name Intel Corporation nor the names of its
51 * contributors may be used to endorse or promote products derived
52 * from this software without specific prior written permission.
53 *
54 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
55 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
56 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
57 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
58 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
59 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
60 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
61 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
62 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
63 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
64 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
65 *
66 *****************************************************************************/
67#include <net/mac80211.h>
68
69#include "mvm.h"
70#include "sta.h"
9ee718aa 71#include "rs.h"
8ca151b5 72
854c5705
SS		 73/*
74 * New version of ADD_STA_sta command added new fields at the end of the
75 * structure, so sending the size of the relevant API's structure is enough to
76 * support both API versions.
77 */
78static inline int iwl_mvm_add_sta_cmd_size(struct iwl_mvm *mvm)
79{
80 return iwl_mvm_has_new_rx_api(mvm) ?
81 sizeof(struct iwl_mvm_add_sta_cmd) :
82 sizeof(struct iwl_mvm_add_sta_cmd_v7);
83}
84
b92e661b
EP		 85static int iwl_mvm_find_free_sta_id(struct iwl_mvm *mvm,
86 enum nl80211_iftype iftype)
8ca151b5
JB		 87{
88 int sta_id;
b92e661b 89 u32 reserved_ids = 0;
8ca151b5 90
b92e661b 91 BUILD_BUG_ON(IWL_MVM_STATION_COUNT > 32);
8ca151b5
JB		 92 WARN_ON_ONCE(test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status));
93
94 lockdep_assert_held(&mvm->mutex);
95
b92e661b
EP		 96 /* d0i3/d3 assumes the AP's sta_id (of sta vif) is 0. reserve it. */
97 if (iftype != NL80211_IFTYPE_STATION)
98 reserved_ids = BIT(0);
99
8ca151b5 100 /* Don't take rcu_read_lock() since we are protected by mvm->mutex */
b92e661b
EP		 101 for (sta_id = 0; sta_id < IWL_MVM_STATION_COUNT; sta_id++) {
102 if (BIT(sta_id) & reserved_ids)
103 continue;
104
8ca151b5
JB		 105 if (!rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
106 lockdep_is_held(&mvm->mutex)))
107 return sta_id;
b92e661b 108 }
8ca151b5
JB		 109 return IWL_MVM_STATION_COUNT;
110}
111
7a453973
JB		 112/* send station add/update command to firmware */
113int iwl_mvm_sta_send_to_fw(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
24afba76 114 bool update, unsigned int flags)
8ca151b5 115{
9d8ce6af 116 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
4b8265ab
EG		 117 struct iwl_mvm_add_sta_cmd add_sta_cmd = {
118 .sta_id = mvm_sta->sta_id,
119 .mac_id_n_color = cpu_to_le32(mvm_sta->mac_id_n_color),
120 .add_modify = update ? 1 : 0,
121 .station_flags_msk = cpu_to_le32(STA_FLG_FAT_EN_MSK |
122 STA_FLG_MIMO_EN_MSK),
cf0cda19 123 .tid_disable_tx = cpu_to_le16(mvm_sta->tid_disable_agg),
4b8265ab 124 };
8ca151b5
JB		 125 int ret;
126 u32 status;
127 u32 agg_size = 0, mpdu_dens = 0;
128
24afba76 129 if (!update || (flags & STA_MODIFY_QUEUES)) {
7a453973
JB		 130 add_sta_cmd.tfd_queue_msk = cpu_to_le32(mvm_sta->tfd_queue_msk);
131 memcpy(&add_sta_cmd.addr, sta->addr, ETH_ALEN);
24afba76
LK		 132
133 if (flags & STA_MODIFY_QUEUES)
134 add_sta_cmd.modify_mask |= STA_MODIFY_QUEUES;
7a453973 135 }
5bc5aaad
JB		 136
137 switch (sta->bandwidth) {
138 case IEEE80211_STA_RX_BW_160:
139 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_FAT_EN_160MHZ);
140 /* fall through */
141 case IEEE80211_STA_RX_BW_80:
142 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_FAT_EN_80MHZ);
143 /* fall through */
144 case IEEE80211_STA_RX_BW_40:
145 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_FAT_EN_40MHZ);
146 /* fall through */
147 case IEEE80211_STA_RX_BW_20:
148 if (sta->ht_cap.ht_supported)
149 add_sta_cmd.station_flags |=
150 cpu_to_le32(STA_FLG_FAT_EN_20MHZ);
151 break;
152 }
153
154 switch (sta->rx_nss) {
155 case 1:
156 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_MIMO_EN_SISO);
157 break;
158 case 2:
159 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_MIMO_EN_MIMO2);
160 break;
161 case 3 ... 8:
162 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_MIMO_EN_MIMO3);
163 break;
164 }
165
166 switch (sta->smps_mode) {
167 case IEEE80211_SMPS_AUTOMATIC:
168 case IEEE80211_SMPS_NUM_MODES:
169 WARN_ON(1);
170 break;
171 case IEEE80211_SMPS_STATIC:
172 /* override NSS */
173 add_sta_cmd.station_flags &= ~cpu_to_le32(STA_FLG_MIMO_EN_MSK);
174 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_MIMO_EN_SISO);
175 break;
176 case IEEE80211_SMPS_DYNAMIC:
177 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_RTS_MIMO_PROT);
178 break;
179 case IEEE80211_SMPS_OFF:
180 /* nothing */
181 break;
182 }
8ca151b5
JB		 183
184 if (sta->ht_cap.ht_supported) {
185 add_sta_cmd.station_flags_msk |=
186 cpu_to_le32(STA_FLG_MAX_AGG_SIZE_MSK |
187 STA_FLG_AGG_MPDU_DENS_MSK);
188
189 mpdu_dens = sta->ht_cap.ampdu_density;
190 }
191
192 if (sta->vht_cap.vht_supported) {
193 agg_size = sta->vht_cap.cap &
194 IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
195 agg_size >>=
196 IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
197 } else if (sta->ht_cap.ht_supported) {
198 agg_size = sta->ht_cap.ampdu_factor;
199 }
200
201 add_sta_cmd.station_flags |=
202 cpu_to_le32(agg_size << STA_FLG_MAX_AGG_SIZE_SHIFT);
203 add_sta_cmd.station_flags |=
204 cpu_to_le32(mpdu_dens << STA_FLG_AGG_MPDU_DENS_SHIFT);
205
206 status = ADD_STA_SUCCESS;
854c5705
SS		 207 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA,
208 iwl_mvm_add_sta_cmd_size(mvm),
f9dc0004 209 &add_sta_cmd, &status);
8ca151b5
JB		 210 if (ret)
211 return ret;
212
837c4da9 213 switch (status & IWL_ADD_STA_STATUS_MASK) {
8ca151b5
JB		 214 case ADD_STA_SUCCESS:
215 IWL_DEBUG_ASSOC(mvm, "ADD_STA PASSED\n");
216 break;
217 default:
218 ret = -EIO;
219 IWL_ERR(mvm, "ADD_STA failed\n");
220 break;
221 }
222
223 return ret;
224}
225
10b2b201
SS		 226static void iwl_mvm_rx_agg_session_expired(unsigned long data)
227{
228 struct iwl_mvm_baid_data __rcu **rcu_ptr = (void *)data;
229 struct iwl_mvm_baid_data *ba_data;
230 struct ieee80211_sta *sta;
231 struct iwl_mvm_sta *mvm_sta;
232 unsigned long timeout;
233
234 rcu_read_lock();
235
236 ba_data = rcu_dereference(*rcu_ptr);
237
238 if (WARN_ON(!ba_data))
239 goto unlock;
240
241 if (!ba_data->timeout)
242 goto unlock;
243
244 timeout = ba_data->last_rx + TU_TO_JIFFIES(ba_data->timeout * 2);
245 if (time_is_after_jiffies(timeout)) {
246 mod_timer(&ba_data->session_timer, timeout);
247 goto unlock;
248 }
249
250 /* Timer expired */
251 sta = rcu_dereference(ba_data->mvm->fw_id_to_mac_id[ba_data->sta_id]);
252 mvm_sta = iwl_mvm_sta_from_mac80211(sta);
253 ieee80211_stop_rx_ba_session_offl(mvm_sta->vif,
254 sta->addr, ba_data->tid);
255unlock:
256 rcu_read_unlock();
257}
258
a0f6bf2a
AN		 259static int iwl_mvm_tdls_sta_init(struct iwl_mvm *mvm,
260 struct ieee80211_sta *sta)
261{
262 unsigned long used_hw_queues;
263 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
5d42e7b2
EG		 264 unsigned int wdg_timeout =
265 iwl_mvm_get_wd_timeout(mvm, NULL, true, false);
a0f6bf2a
AN		 266 u32 ac;
267
268 lockdep_assert_held(&mvm->mutex);
269
270 used_hw_queues = iwl_mvm_get_used_hw_queues(mvm, NULL);
271
272 /* Find available queues, and allocate them to the ACs */
273 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
274 u8 queue = find_first_zero_bit(&used_hw_queues,
275 mvm->first_agg_queue);
276
277 if (queue >= mvm->first_agg_queue) {
278 IWL_ERR(mvm, "Failed to allocate STA queue\n");
279 return -EBUSY;
280 }
281
282 __set_bit(queue, &used_hw_queues);
283 mvmsta->hw_queue[ac] = queue;
284 }
285
286 /* Found a place for all queues - enable them */
287 for (ac = 0; ac < IEEE80211_NUM_ACS; ac++) {
288 iwl_mvm_enable_ac_txq(mvm, mvmsta->hw_queue[ac],
4ecafae9 289 mvmsta->hw_queue[ac],
5c1156ef
LK		 290 iwl_mvm_ac_to_tx_fifo[ac], 0,
291 wdg_timeout);
a0f6bf2a
AN		 292 mvmsta->tfd_queue_msk |= BIT(mvmsta->hw_queue[ac]);
293 }
294
295 return 0;
296}
297
298static void iwl_mvm_tdls_sta_deinit(struct iwl_mvm *mvm,
299 struct ieee80211_sta *sta)
300{
301 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
302 unsigned long sta_msk;
303 int i;
304
305 lockdep_assert_held(&mvm->mutex);
306
307 /* disable the TDLS STA-specific queues */
308 sta_msk = mvmsta->tfd_queue_msk;
a4ca3ed4 309 for_each_set_bit(i, &sta_msk, sizeof(sta_msk) * BITS_PER_BYTE)
06ecdba3 310 iwl_mvm_disable_txq(mvm, i, i, IWL_MAX_TID_COUNT, 0);
a0f6bf2a
AN		 311}
312
9794c64f
LK		 313/* Disable aggregations for a bitmap of TIDs for a given station */
314static int iwl_mvm_invalidate_sta_queue(struct iwl_mvm *mvm, int queue,
315 unsigned long disable_agg_tids,
316 bool remove_queue)
317{
318 struct iwl_mvm_add_sta_cmd cmd = {};
319 struct ieee80211_sta *sta;
320 struct iwl_mvm_sta *mvmsta;
321 u32 status;
322 u8 sta_id;
323 int ret;
324
325 spin_lock_bh(&mvm->queue_info_lock);
326 sta_id = mvm->queue_info[queue].ra_sta_id;
327 spin_unlock_bh(&mvm->queue_info_lock);
328
329 rcu_read_lock();
330
331 sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
332
333 if (WARN_ON_ONCE(IS_ERR_OR_NULL(sta))) {
334 rcu_read_unlock();
335 return -EINVAL;
336 }
337
338 mvmsta = iwl_mvm_sta_from_mac80211(sta);
339
340 mvmsta->tid_disable_agg |= disable_agg_tids;
341
342 cmd.mac_id_n_color = cpu_to_le32(mvmsta->mac_id_n_color);
343 cmd.sta_id = mvmsta->sta_id;
344 cmd.add_modify = STA_MODE_MODIFY;
345 cmd.modify_mask = STA_MODIFY_QUEUES;
346 if (disable_agg_tids)
347 cmd.modify_mask |= STA_MODIFY_TID_DISABLE_TX;
348 if (remove_queue)
349 cmd.modify_mask |= STA_MODIFY_QUEUE_REMOVAL;
350 cmd.tfd_queue_msk = cpu_to_le32(mvmsta->tfd_queue_msk);
351 cmd.tid_disable_tx = cpu_to_le16(mvmsta->tid_disable_agg);
352
353 rcu_read_unlock();
354
355 /* Notify FW of queue removal from the STA queues */
356 status = ADD_STA_SUCCESS;
357 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA,
358 iwl_mvm_add_sta_cmd_size(mvm),
359 &cmd, &status);
360
361 return ret;
362}
363
42db09c1
LK		 364static int iwl_mvm_get_queue_agg_tids(struct iwl_mvm *mvm, int queue)
365{
366 struct ieee80211_sta *sta;
367 struct iwl_mvm_sta *mvmsta;
368 unsigned long tid_bitmap;
369 unsigned long agg_tids = 0;
370 s8 sta_id;
371 int tid;
372
373 lockdep_assert_held(&mvm->mutex);
374
375 spin_lock_bh(&mvm->queue_info_lock);
376 sta_id = mvm->queue_info[queue].ra_sta_id;
377 tid_bitmap = mvm->queue_info[queue].tid_bitmap;
378 spin_unlock_bh(&mvm->queue_info_lock);
379
380 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
381 lockdep_is_held(&mvm->mutex));
382
383 if (WARN_ON_ONCE(IS_ERR_OR_NULL(sta)))
384 return -EINVAL;
385
386 mvmsta = iwl_mvm_sta_from_mac80211(sta);
387
388 spin_lock_bh(&mvmsta->lock);
389 for_each_set_bit(tid, &tid_bitmap, IWL_MAX_TID_COUNT + 1) {
390 if (mvmsta->tid_data[tid].state == IWL_AGG_ON)
391 agg_tids |= BIT(tid);
392 }
393 spin_unlock_bh(&mvmsta->lock);
394
395 return agg_tids;
396}
397
9794c64f
LK		 398/*
399 * Remove a queue from a station's resources.
400 * Note that this only marks as free. It DOESN'T delete a BA agreement, and
401 * doesn't disable the queue
402 */
403static int iwl_mvm_remove_sta_queue_marking(struct iwl_mvm *mvm, int queue)
404{
405 struct ieee80211_sta *sta;
406 struct iwl_mvm_sta *mvmsta;
407 unsigned long tid_bitmap;
408 unsigned long disable_agg_tids = 0;
409 u8 sta_id;
410 int tid;
411
412 lockdep_assert_held(&mvm->mutex);
413
414 spin_lock_bh(&mvm->queue_info_lock);
415 sta_id = mvm->queue_info[queue].ra_sta_id;
416 tid_bitmap = mvm->queue_info[queue].tid_bitmap;
417 spin_unlock_bh(&mvm->queue_info_lock);
418
419 rcu_read_lock();
420
421 sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
422
423 if (WARN_ON_ONCE(IS_ERR_OR_NULL(sta))) {
424 rcu_read_unlock();
425 return 0;
426 }
427
428 mvmsta = iwl_mvm_sta_from_mac80211(sta);
429
430 spin_lock_bh(&mvmsta->lock);
42db09c1 431 /* Unmap MAC queues and TIDs from this queue */
9794c64f 432 for_each_set_bit(tid, &tid_bitmap, IWL_MAX_TID_COUNT + 1) {
9794c64f
LK		 433 if (mvmsta->tid_data[tid].state == IWL_AGG_ON)
434 disable_agg_tids |= BIT(tid);
42db09c1 435 mvmsta->tid_data[tid].txq_id = IEEE80211_INVAL_HW_QUEUE;
9794c64f 436 }
9794c64f 437
42db09c1 438 mvmsta->tfd_queue_msk &= ~BIT(queue); /* Don't use this queue anymore */
9794c64f
LK		 439 spin_unlock_bh(&mvmsta->lock);
440
441 rcu_read_unlock();
442
42db09c1 443 spin_lock_bh(&mvm->queue_info_lock);
9794c64f
LK		 444 /* Unmap MAC queues and TIDs from this queue */
445 mvm->queue_info[queue].hw_queue_to_mac80211 = 0;
446 mvm->queue_info[queue].hw_queue_refcount = 0;
447 mvm->queue_info[queue].tid_bitmap = 0;
42db09c1 448 spin_unlock_bh(&mvm->queue_info_lock);
9794c64f
LK		 449
450 return disable_agg_tids;
451}
452
42db09c1
LK		 453static int iwl_mvm_get_shared_queue(struct iwl_mvm *mvm,
454 unsigned long tfd_queue_mask, u8 ac)
455{
456 int queue = 0;
457 u8 ac_to_queue[IEEE80211_NUM_ACS];
458 int i;
459
460 lockdep_assert_held(&mvm->queue_info_lock);
461
462 memset(&ac_to_queue, IEEE80211_INVAL_HW_QUEUE, sizeof(ac_to_queue));
463
464 /* See what ACs the existing queues for this STA have */
465 for_each_set_bit(i, &tfd_queue_mask, IWL_MVM_DQA_MAX_DATA_QUEUE) {
466 /* Only DATA queues can be shared */
467 if (i < IWL_MVM_DQA_MIN_DATA_QUEUE &&
468 i != IWL_MVM_DQA_BSS_CLIENT_QUEUE)
469 continue;
470
9f9af3d7
LK		 471 /* Don't try and take queues being reconfigured */
472 if (mvm->queue_info[queue].status ==
473 IWL_MVM_QUEUE_RECONFIGURING)
474 continue;
475
42db09c1
LK		 476 ac_to_queue[mvm->queue_info[i].mac80211_ac] = i;
477 }
478
479 /*
480 * The queue to share is chosen only from DATA queues as follows (in
481 * descending priority):
482 * 1. An AC_BE queue
483 * 2. Same AC queue
484 * 3. Highest AC queue that is lower than new AC
485 * 4. Any existing AC (there always is at least 1 DATA queue)
486 */
487
488 /* Priority 1: An AC_BE queue */
489 if (ac_to_queue[IEEE80211_AC_BE] != IEEE80211_INVAL_HW_QUEUE)
490 queue = ac_to_queue[IEEE80211_AC_BE];
491 /* Priority 2: Same AC queue */
492 else if (ac_to_queue[ac] != IEEE80211_INVAL_HW_QUEUE)
493 queue = ac_to_queue[ac];
494 /* Priority 3a: If new AC is VO and VI exists - use VI */
495 else if (ac == IEEE80211_AC_VO &&
496 ac_to_queue[IEEE80211_AC_VI] != IEEE80211_INVAL_HW_QUEUE)
497 queue = ac_to_queue[IEEE80211_AC_VI];
498 /* Priority 3b: No BE so only AC less than the new one is BK */
499 else if (ac_to_queue[IEEE80211_AC_BK] != IEEE80211_INVAL_HW_QUEUE)
500 queue = ac_to_queue[IEEE80211_AC_BK];
501 /* Priority 4a: No BE nor BK - use VI if exists */
502 else if (ac_to_queue[IEEE80211_AC_VI] != IEEE80211_INVAL_HW_QUEUE)
503 queue = ac_to_queue[IEEE80211_AC_VI];
504 /* Priority 4b: No BE, BK nor VI - use VO if exists */
505 else if (ac_to_queue[IEEE80211_AC_VO] != IEEE80211_INVAL_HW_QUEUE)
506 queue = ac_to_queue[IEEE80211_AC_VO];
507
508 /* Make sure queue found (or not) is legal */
9f9af3d7
LK		 509 if (!iwl_mvm_is_dqa_data_queue(mvm, queue) &&
510 !iwl_mvm_is_dqa_mgmt_queue(mvm, queue) &&
511 (queue != IWL_MVM_DQA_BSS_CLIENT_QUEUE)) {
42db09c1 512 IWL_ERR(mvm, "No DATA queues available to share\n");
9f9af3d7
LK		 513 return -ENOSPC;
514 }
515
516 /* Make sure the queue isn't in the middle of being reconfigured */
517 if (mvm->queue_info[queue].status == IWL_MVM_QUEUE_RECONFIGURING) {
518 IWL_ERR(mvm,
519 "TXQ %d is in the middle of re-config - try again\n",
520 queue);
521 return -EBUSY;
42db09c1
LK		 522 }
523
524 return queue;
525}
526
58f2cc57 527/*
9f9af3d7
LK		 528 * If a given queue has a higher AC than the TID stream that is being compared
529 * to, the queue needs to be redirected to the lower AC. This function does that
58f2cc57
LK		 530 * in such a case, otherwise - if no redirection required - it does nothing,
531 * unless the %force param is true.
532 */
9f9af3d7
LK		 533int iwl_mvm_scd_queue_redirect(struct iwl_mvm *mvm, int queue, int tid,
534 int ac, int ssn, unsigned int wdg_timeout,
535 bool force)
58f2cc57
LK		 536{
537 struct iwl_scd_txq_cfg_cmd cmd = {
538 .scd_queue = queue,
f7c692de 539 .action = SCD_CFG_DISABLE_QUEUE,
58f2cc57
LK		 540 };
541 bool shared_queue;
542 unsigned long mq;
543 int ret;
544
545 /*
546 * If the AC is lower than current one - FIFO needs to be redirected to
547 * the lowest one of the streams in the queue. Check if this is needed
548 * here.
549 * Notice that the enum ieee80211_ac_numbers is "flipped", so BK is with
550 * value 3 and VO with value 0, so to check if ac X is lower than ac Y
551 * we need to check if the numerical value of X is LARGER than of Y.
552 */
553 spin_lock_bh(&mvm->queue_info_lock);
554 if (ac <= mvm->queue_info[queue].mac80211_ac && !force) {
555 spin_unlock_bh(&mvm->queue_info_lock);
556
557 IWL_DEBUG_TX_QUEUES(mvm,
558 "No redirection needed on TXQ #%d\n",
559 queue);
560 return 0;
561 }
562
563 cmd.sta_id = mvm->queue_info[queue].ra_sta_id;
564 cmd.tx_fifo = iwl_mvm_ac_to_tx_fifo[mvm->queue_info[queue].mac80211_ac];
edbe961c 565 cmd.tid = mvm->queue_info[queue].txq_tid;
58f2cc57
LK		 566 mq = mvm->queue_info[queue].hw_queue_to_mac80211;
567 shared_queue = (mvm->queue_info[queue].hw_queue_refcount > 1);
568 spin_unlock_bh(&mvm->queue_info_lock);
569
9f9af3d7 570 IWL_DEBUG_TX_QUEUES(mvm, "Redirecting TXQ #%d to FIFO #%d\n",
58f2cc57
LK		 571 queue, iwl_mvm_ac_to_tx_fifo[ac]);
572
573 /* Stop MAC queues and wait for this queue to empty */
574 iwl_mvm_stop_mac_queues(mvm, mq);
575 ret = iwl_trans_wait_tx_queue_empty(mvm->trans, BIT(queue));
576 if (ret) {
577 IWL_ERR(mvm, "Error draining queue %d before reconfig\n",
578 queue);
579 ret = -EIO;
580 goto out;
581 }
582
583 /* Before redirecting the queue we need to de-activate it */
584 iwl_trans_txq_disable(mvm->trans, queue, false);
585 ret = iwl_mvm_send_cmd_pdu(mvm, SCD_QUEUE_CFG, 0, sizeof(cmd), &cmd);
586 if (ret)
587 IWL_ERR(mvm, "Failed SCD disable TXQ %d (ret=%d)\n", queue,
588 ret);
589
590 /* Make sure the SCD wrptr is correctly set before reconfiguring */
ca3b9c6b 591 iwl_trans_txq_enable_cfg(mvm->trans, queue, ssn, NULL, wdg_timeout);
58f2cc57 592
edbe961c
LK		 593 /* Update the TID "owner" of the queue */
594 spin_lock_bh(&mvm->queue_info_lock);
595 mvm->queue_info[queue].txq_tid = tid;
596 spin_unlock_bh(&mvm->queue_info_lock);
597
58f2cc57
LK		 598 /* TODO: Work-around SCD bug when moving back by multiples of 0x40 */
599
600 /* Redirect to lower AC */
601 iwl_mvm_reconfig_scd(mvm, queue, iwl_mvm_ac_to_tx_fifo[ac],
602 cmd.sta_id, tid, LINK_QUAL_AGG_FRAME_LIMIT_DEF,
603 ssn);
604
605 /* Update AC marking of the queue */
606 spin_lock_bh(&mvm->queue_info_lock);
607 mvm->queue_info[queue].mac80211_ac = ac;
608 spin_unlock_bh(&mvm->queue_info_lock);
609
610 /*
611 * Mark queue as shared in transport if shared
612 * Note this has to be done after queue enablement because enablement
613 * can also set this value, and there is no indication there to shared
614 * queues
615 */
616 if (shared_queue)
617 iwl_trans_txq_set_shared_mode(mvm->trans, queue, true);
618
619out:
620 /* Continue using the MAC queues */
621 iwl_mvm_start_mac_queues(mvm, mq);
622
623 return ret;
624}
625
24afba76
LK		 626static int iwl_mvm_sta_alloc_queue(struct iwl_mvm *mvm,
627 struct ieee80211_sta *sta, u8 ac, int tid,
628 struct ieee80211_hdr *hdr)
629{
630 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
631 struct iwl_trans_txq_scd_cfg cfg = {
632 .fifo = iwl_mvm_ac_to_tx_fifo[ac],
633 .sta_id = mvmsta->sta_id,
634 .tid = tid,
635 .frame_limit = IWL_FRAME_LIMIT,
636 };
637 unsigned int wdg_timeout =
638 iwl_mvm_get_wd_timeout(mvm, mvmsta->vif, false, false);
639 u8 mac_queue = mvmsta->vif->hw_queue[ac];
640 int queue = -1;
9794c64f
LK		 641 bool using_inactive_queue = false;
642 unsigned long disable_agg_tids = 0;
643 enum iwl_mvm_agg_state queue_state;
42db09c1 644 bool shared_queue = false;
24afba76 645 int ssn;
42db09c1 646 unsigned long tfd_queue_mask;
cf961e16 647 int ret;
24afba76
LK		 648
649 lockdep_assert_held(&mvm->mutex);
650
42db09c1
LK		 651 spin_lock_bh(&mvmsta->lock);
652 tfd_queue_mask = mvmsta->tfd_queue_msk;
653 spin_unlock_bh(&mvmsta->lock);
654
d2515a99 655 spin_lock_bh(&mvm->queue_info_lock);
24afba76
LK		 656
657 /*
658 * Non-QoS, QoS NDP and MGMT frames should go to a MGMT queue, if one
659 * exists
660 */
661 if (!ieee80211_is_data_qos(hdr->frame_control) ||
662 ieee80211_is_qos_nullfunc(hdr->frame_control)) {
9794c64f
LK		 663 queue = iwl_mvm_find_free_queue(mvm, mvmsta->sta_id,
664 IWL_MVM_DQA_MIN_MGMT_QUEUE,
24afba76
LK		 665 IWL_MVM_DQA_MAX_MGMT_QUEUE);
666 if (queue >= IWL_MVM_DQA_MIN_MGMT_QUEUE)
667 IWL_DEBUG_TX_QUEUES(mvm, "Found free MGMT queue #%d\n",
668 queue);
669
670 /* If no such queue is found, we'll use a DATA queue instead */
671 }
672
9794c64f
LK		 673 if ((queue < 0 && mvmsta->reserved_queue != IEEE80211_INVAL_HW_QUEUE) &&
674 (mvm->queue_info[mvmsta->reserved_queue].status ==
675 IWL_MVM_QUEUE_RESERVED ||
676 mvm->queue_info[mvmsta->reserved_queue].status ==
677 IWL_MVM_QUEUE_INACTIVE)) {
24afba76 678 queue = mvmsta->reserved_queue;
9794c64f 679 mvm->queue_info[queue].reserved = true;
24afba76
LK		 680 IWL_DEBUG_TX_QUEUES(mvm, "Using reserved queue #%d\n", queue);
681 }
682
683 if (queue < 0)
9794c64f
LK		 684 queue = iwl_mvm_find_free_queue(mvm, mvmsta->sta_id,
685 IWL_MVM_DQA_MIN_DATA_QUEUE,
24afba76 686 IWL_MVM_DQA_MAX_DATA_QUEUE);
cf961e16 687
9794c64f
LK		 688 /*
689 * Check if this queue is already allocated but inactive.
690 * In such a case, we'll need to first free this queue before enabling
691 * it again, so we'll mark it as reserved to make sure no new traffic
692 * arrives on it
693 */
694 if (queue > 0 &&
695 mvm->queue_info[queue].status == IWL_MVM_QUEUE_INACTIVE) {
696 mvm->queue_info[queue].status = IWL_MVM_QUEUE_RESERVED;
697 using_inactive_queue = true;
698 IWL_DEBUG_TX_QUEUES(mvm,
699 "Re-assigning TXQ %d: sta_id=%d, tid=%d\n",
700 queue, mvmsta->sta_id, tid);
701 }
702
42db09c1
LK		 703 /* No free queue - we'll have to share */
704 if (queue <= 0) {
705 queue = iwl_mvm_get_shared_queue(mvm, tfd_queue_mask, ac);
706 if (queue > 0) {
707 shared_queue = true;
708 mvm->queue_info[queue].status = IWL_MVM_QUEUE_SHARED;
709 }
710 }
711
cf961e16
LK		 712 /*
713 * Mark TXQ as ready, even though it hasn't been fully configured yet,
714 * to make sure no one else takes it.
715 * This will allow avoiding re-acquiring the lock at the end of the
716 * configuration. On error we'll mark it back as free.
717 */
42db09c1 718 if ((queue > 0) && !shared_queue)
cf961e16 719 mvm->queue_info[queue].status = IWL_MVM_QUEUE_READY;
24afba76 720
d2515a99 721 spin_unlock_bh(&mvm->queue_info_lock);
24afba76 722
42db09c1
LK		 723 /* This shouldn't happen - out of queues */
724 if (WARN_ON(queue <= 0)) {
725 IWL_ERR(mvm, "No available queues for tid %d on sta_id %d\n",
726 tid, cfg.sta_id);
9f9af3d7 727 return queue;
42db09c1 728 }
24afba76
LK		 729
730 /*
731 * Actual en/disablement of aggregations is through the ADD_STA HCMD,
732 * but for configuring the SCD to send A-MPDUs we need to mark the queue
733 * as aggregatable.
734 * Mark all DATA queues as allowing to be aggregated at some point
735 */
d5216a28
LK		 736 cfg.aggregate = (queue >= IWL_MVM_DQA_MIN_DATA_QUEUE ||
737 queue == IWL_MVM_DQA_BSS_CLIENT_QUEUE);
24afba76 738
9794c64f
LK		 739 /*
740 * If this queue was previously inactive (idle) - we need to free it
741 * first
742 */
743 if (using_inactive_queue) {
744 struct iwl_scd_txq_cfg_cmd cmd = {
745 .scd_queue = queue,
f7c692de 746 .action = SCD_CFG_DISABLE_QUEUE,
9794c64f 747 };
93f436e2 748 u8 ac;
9794c64f
LK		 749
750 disable_agg_tids = iwl_mvm_remove_sta_queue_marking(mvm, queue);
751
93f436e2
LK		 752 spin_lock_bh(&mvm->queue_info_lock);
753 ac = mvm->queue_info[queue].mac80211_ac;
754 cmd.sta_id = mvm->queue_info[queue].ra_sta_id;
755 cmd.tx_fifo = iwl_mvm_ac_to_tx_fifo[ac];
edbe961c 756 cmd.tid = mvm->queue_info[queue].txq_tid;
93f436e2
LK		 757 spin_unlock_bh(&mvm->queue_info_lock);
758
9794c64f 759 /* Disable the queue */
8d98ae6e
LK		 760 if (disable_agg_tids)
761 iwl_mvm_invalidate_sta_queue(mvm, queue,
762 disable_agg_tids, false);
9794c64f
LK		 763 iwl_trans_txq_disable(mvm->trans, queue, false);
764 ret = iwl_mvm_send_cmd_pdu(mvm, SCD_QUEUE_CFG, 0, sizeof(cmd),
765 &cmd);
766 if (ret) {
767 IWL_ERR(mvm,
768 "Failed to free inactive queue %d (ret=%d)\n",
769 queue, ret);
770
771 /* Re-mark the inactive queue as inactive */
772 spin_lock_bh(&mvm->queue_info_lock);
773 mvm->queue_info[queue].status = IWL_MVM_QUEUE_INACTIVE;
774 spin_unlock_bh(&mvm->queue_info_lock);
775
776 return ret;
777 }
8d98ae6e
LK		 778
779 /* If TXQ is allocated to another STA, update removal in FW */
780 if (cmd.sta_id != mvmsta->sta_id)
781 iwl_mvm_invalidate_sta_queue(mvm, queue, 0, true);
9794c64f
LK		 782 }
783
42db09c1
LK		 784 IWL_DEBUG_TX_QUEUES(mvm,
785 "Allocating %squeue #%d to sta %d on tid %d\n",
786 shared_queue ? "shared " : "", queue,
787 mvmsta->sta_id, tid);
788
789 if (shared_queue) {
790 /* Disable any open aggs on this queue */
791 disable_agg_tids = iwl_mvm_get_queue_agg_tids(mvm, queue);
792
793 if (disable_agg_tids) {
794 IWL_DEBUG_TX_QUEUES(mvm, "Disabling aggs on queue %d\n",
795 queue);
796 iwl_mvm_invalidate_sta_queue(mvm, queue,
797 disable_agg_tids, false);
798 }
42db09c1 799 }
24afba76
LK		 800
801 ssn = IEEE80211_SEQ_TO_SN(le16_to_cpu(hdr->seq_ctrl));
802 iwl_mvm_enable_txq(mvm, queue, mac_queue, ssn, &cfg,
803 wdg_timeout);
804
58f2cc57
LK		 805 /*
806 * Mark queue as shared in transport if shared
807 * Note this has to be done after queue enablement because enablement
808 * can also set this value, and there is no indication there to shared
809 * queues
810 */
811 if (shared_queue)
812 iwl_trans_txq_set_shared_mode(mvm->trans, queue, true);
813
24afba76
LK		 814 spin_lock_bh(&mvmsta->lock);
815 mvmsta->tid_data[tid].txq_id = queue;
9794c64f 816 mvmsta->tid_data[tid].is_tid_active = true;
24afba76 817 mvmsta->tfd_queue_msk |= BIT(queue);
9794c64f 818 queue_state = mvmsta->tid_data[tid].state;
24afba76
LK		 819
820 if (mvmsta->reserved_queue == queue)
821 mvmsta->reserved_queue = IEEE80211_INVAL_HW_QUEUE;
822 spin_unlock_bh(&mvmsta->lock);
823
42db09c1
LK		 824 if (!shared_queue) {
825 ret = iwl_mvm_sta_send_to_fw(mvm, sta, true, STA_MODIFY_QUEUES);
826 if (ret)
827 goto out_err;
cf961e16 828
42db09c1
LK		 829 /* If we need to re-enable aggregations... */
830 if (queue_state == IWL_AGG_ON) {
831 ret = iwl_mvm_sta_tx_agg(mvm, sta, tid, queue, true);
832 if (ret)
833 goto out_err;
834 }
58f2cc57
LK		 835 } else {
836 /* Redirect queue, if needed */
837 ret = iwl_mvm_scd_queue_redirect(mvm, queue, tid, ac, ssn,
838 wdg_timeout, false);
839 if (ret)
840 goto out_err;
42db09c1 841 }
9794c64f 842
42db09c1 843 return 0;
cf961e16
LK		 844
845out_err:
846 iwl_mvm_disable_txq(mvm, queue, mac_queue, tid, 0);
847
848 return ret;
24afba76
LK		 849}
850
19aefa45
LK		 851static void iwl_mvm_change_queue_owner(struct iwl_mvm *mvm, int queue)
852{
853 struct iwl_scd_txq_cfg_cmd cmd = {
854 .scd_queue = queue,
855 .action = SCD_CFG_UPDATE_QUEUE_TID,
856 };
857 s8 sta_id;
858 int tid;
859 unsigned long tid_bitmap;
860 int ret;
861
862 lockdep_assert_held(&mvm->mutex);
863
864 spin_lock_bh(&mvm->queue_info_lock);
865 sta_id = mvm->queue_info[queue].ra_sta_id;
866 tid_bitmap = mvm->queue_info[queue].tid_bitmap;
867 spin_unlock_bh(&mvm->queue_info_lock);
868
869 if (WARN(!tid_bitmap, "TXQ %d has no tids assigned to it\n", queue))
870 return;
871
872 /* Find any TID for queue */
873 tid = find_first_bit(&tid_bitmap, IWL_MAX_TID_COUNT + 1);
874 cmd.tid = tid;
875 cmd.tx_fifo = iwl_mvm_ac_to_tx_fifo[tid_to_mac80211_ac[tid]];
876
877 ret = iwl_mvm_send_cmd_pdu(mvm, SCD_QUEUE_CFG, 0, sizeof(cmd), &cmd);
878 if (ret)
879 IWL_ERR(mvm, "Failed to update owner of TXQ %d (ret=%d)\n",
880 queue, ret);
881 else
882 IWL_DEBUG_TX_QUEUES(mvm, "Changed TXQ %d ownership to tid %d\n",
883 queue, tid);
884}
885
9f9af3d7
LK		 886static void iwl_mvm_unshare_queue(struct iwl_mvm *mvm, int queue)
887{
888 struct ieee80211_sta *sta;
889 struct iwl_mvm_sta *mvmsta;
890 s8 sta_id;
891 int tid = -1;
892 unsigned long tid_bitmap;
893 unsigned int wdg_timeout;
894 int ssn;
895 int ret = true;
896
897 lockdep_assert_held(&mvm->mutex);
898
899 spin_lock_bh(&mvm->queue_info_lock);
900 sta_id = mvm->queue_info[queue].ra_sta_id;
901 tid_bitmap = mvm->queue_info[queue].tid_bitmap;
902 spin_unlock_bh(&mvm->queue_info_lock);
903
904 /* Find TID for queue, and make sure it is the only one on the queue */
905 tid = find_first_bit(&tid_bitmap, IWL_MAX_TID_COUNT + 1);
906 if (tid_bitmap != BIT(tid)) {
907 IWL_ERR(mvm, "Failed to unshare q %d, active tids=0x%lx\n",
908 queue, tid_bitmap);
909 return;
910 }
911
912 IWL_DEBUG_TX_QUEUES(mvm, "Unsharing TXQ %d, keeping tid %d\n", queue,
913 tid);
914
915 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
916 lockdep_is_held(&mvm->mutex));
917
918 if (WARN_ON_ONCE(IS_ERR_OR_NULL(sta)))
919 return;
920
921 mvmsta = iwl_mvm_sta_from_mac80211(sta);
922 wdg_timeout = iwl_mvm_get_wd_timeout(mvm, mvmsta->vif, false, false);
923
924 ssn = IEEE80211_SEQ_TO_SN(mvmsta->tid_data[tid].seq_number);
925
926 ret = iwl_mvm_scd_queue_redirect(mvm, queue, tid,
927 tid_to_mac80211_ac[tid], ssn,
928 wdg_timeout, true);
929 if (ret) {
930 IWL_ERR(mvm, "Failed to redirect TXQ %d\n", queue);
931 return;
932 }
933
934 /* If aggs should be turned back on - do it */
935 if (mvmsta->tid_data[tid].state == IWL_AGG_ON) {
936 struct iwl_mvm_add_sta_cmd cmd;
937
938 mvmsta->tid_disable_agg &= ~BIT(tid);
939
940 cmd.mac_id_n_color = cpu_to_le32(mvmsta->mac_id_n_color);
941 cmd.sta_id = mvmsta->sta_id;
942 cmd.add_modify = STA_MODE_MODIFY;
943 cmd.modify_mask = STA_MODIFY_TID_DISABLE_TX;
944 cmd.tfd_queue_msk = cpu_to_le32(mvmsta->tfd_queue_msk);
945 cmd.tid_disable_tx = cpu_to_le16(mvmsta->tid_disable_agg);
946
947 ret = iwl_mvm_send_cmd_pdu(mvm, ADD_STA, CMD_ASYNC,
948 iwl_mvm_add_sta_cmd_size(mvm), &cmd);
949 if (!ret) {
950 IWL_DEBUG_TX_QUEUES(mvm,
951 "TXQ #%d is now aggregated again\n",
952 queue);
953
954 /* Mark queue intenally as aggregating again */
955 iwl_trans_txq_set_shared_mode(mvm->trans, queue, false);
956 }
957 }
958
959 spin_lock_bh(&mvm->queue_info_lock);
960 mvm->queue_info[queue].status = IWL_MVM_QUEUE_READY;
961 spin_unlock_bh(&mvm->queue_info_lock);
962}
963
24afba76
LK		 964static inline u8 iwl_mvm_tid_to_ac_queue(int tid)
965{
966 if (tid == IWL_MAX_TID_COUNT)
967 return IEEE80211_AC_VO; /* MGMT */
968
969 return tid_to_mac80211_ac[tid];
970}
971
972static void iwl_mvm_tx_deferred_stream(struct iwl_mvm *mvm,
973 struct ieee80211_sta *sta, int tid)
974{
975 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
976 struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
977 struct sk_buff *skb;
978 struct ieee80211_hdr *hdr;
979 struct sk_buff_head deferred_tx;
980 u8 mac_queue;
981 bool no_queue = false; /* Marks if there is a problem with the queue */
982 u8 ac;
983
984 lockdep_assert_held(&mvm->mutex);
985
986 skb = skb_peek(&tid_data->deferred_tx_frames);
987 if (!skb)
988 return;
989 hdr = (void *)skb->data;
990
991 ac = iwl_mvm_tid_to_ac_queue(tid);
992 mac_queue = IEEE80211_SKB_CB(skb)->hw_queue;
993
994 if (tid_data->txq_id == IEEE80211_INVAL_HW_QUEUE &&
995 iwl_mvm_sta_alloc_queue(mvm, sta, ac, tid, hdr)) {
996 IWL_ERR(mvm,
997 "Can't alloc TXQ for sta %d tid %d - dropping frame\n",
998 mvmsta->sta_id, tid);
999
1000 /*
1001 * Mark queue as problematic so later the deferred traffic is
1002 * freed, as we can do nothing with it
1003 */
1004 no_queue = true;
1005 }
1006
1007 __skb_queue_head_init(&deferred_tx);
1008
d2515a99
LK		 1009 /* Disable bottom-halves when entering TX path */
1010 local_bh_disable();
24afba76
LK		 1011 spin_lock(&mvmsta->lock);
1012 skb_queue_splice_init(&tid_data->deferred_tx_frames, &deferred_tx);
1013 spin_unlock(&mvmsta->lock);
1014
24afba76
LK		 1015 while ((skb = __skb_dequeue(&deferred_tx)))
1016 if (no_queue || iwl_mvm_tx_skb(mvm, skb, sta))
1017 ieee80211_free_txskb(mvm->hw, skb);
1018 local_bh_enable();
1019
1020 /* Wake queue */
1021 iwl_mvm_start_mac_queues(mvm, BIT(mac_queue));
1022}
1023
1024void iwl_mvm_add_new_dqa_stream_wk(struct work_struct *wk)
1025{
1026 struct iwl_mvm *mvm = container_of(wk, struct iwl_mvm,
1027 add_stream_wk);
1028 struct ieee80211_sta *sta;
1029 struct iwl_mvm_sta *mvmsta;
1030 unsigned long deferred_tid_traffic;
9f9af3d7 1031 int queue, sta_id, tid;
24afba76 1032
9794c64f
LK		 1033 /* Check inactivity of queues */
1034 iwl_mvm_inactivity_check(mvm);
1035
24afba76
LK		 1036 mutex_lock(&mvm->mutex);
1037
9f9af3d7
LK		 1038 /* Reconfigure queues requiring reconfiguation */
1039 for (queue = 0; queue < IWL_MAX_HW_QUEUES; queue++) {
1040 bool reconfig;
19aefa45 1041 bool change_owner;
9f9af3d7
LK		 1042
1043 spin_lock_bh(&mvm->queue_info_lock);
1044 reconfig = (mvm->queue_info[queue].status ==
1045 IWL_MVM_QUEUE_RECONFIGURING);
19aefa45
LK		 1046
1047 /*
1048 * We need to take into account a situation in which a TXQ was
1049 * allocated to TID x, and then turned shared by adding TIDs y
1050 * and z. If TID x becomes inactive and is removed from the TXQ,
1051 * ownership must be given to one of the remaining TIDs.
1052 * This is mainly because if TID x continues - a new queue can't
1053 * be allocated for it as long as it is an owner of another TXQ.
1054 */
1055 change_owner = !(mvm->queue_info[queue].tid_bitmap &
1056 BIT(mvm->queue_info[queue].txq_tid)) &&
1057 (mvm->queue_info[queue].status ==
1058 IWL_MVM_QUEUE_SHARED);
9f9af3d7
LK		 1059 spin_unlock_bh(&mvm->queue_info_lock);
1060
1061 if (reconfig)
1062 iwl_mvm_unshare_queue(mvm, queue);
19aefa45
LK		 1063 else if (change_owner)
1064 iwl_mvm_change_queue_owner(mvm, queue);
9f9af3d7
LK		 1065 }
1066
24afba76
LK		 1067 /* Go over all stations with deferred traffic */
1068 for_each_set_bit(sta_id, mvm->sta_deferred_frames,
1069 IWL_MVM_STATION_COUNT) {
1070 clear_bit(sta_id, mvm->sta_deferred_frames);
1071 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
1072 lockdep_is_held(&mvm->mutex));
1073 if (IS_ERR_OR_NULL(sta))
1074 continue;
1075
1076 mvmsta = iwl_mvm_sta_from_mac80211(sta);
1077 deferred_tid_traffic = mvmsta->deferred_traffic_tid_map;
1078
1079 for_each_set_bit(tid, &deferred_tid_traffic,
1080 IWL_MAX_TID_COUNT + 1)
1081 iwl_mvm_tx_deferred_stream(mvm, sta, tid);
1082 }
1083
1084 mutex_unlock(&mvm->mutex);
1085}
1086
1087static int iwl_mvm_reserve_sta_stream(struct iwl_mvm *mvm,
d5216a28
LK		 1088 struct ieee80211_sta *sta,
1089 enum nl80211_iftype vif_type)
24afba76
LK		 1090{
1091 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
1092 int queue;
1093
9794c64f
LK		 1094 /*
1095 * Check for inactive queues, so we don't reach a situation where we
1096 * can't add a STA due to a shortage in queues that doesn't really exist
1097 */
1098 iwl_mvm_inactivity_check(mvm);
1099
24afba76
LK		 1100 spin_lock_bh(&mvm->queue_info_lock);
1101
1102 /* Make sure we have free resources for this STA */
d5216a28
LK		 1103 if (vif_type == NL80211_IFTYPE_STATION && !sta->tdls &&
1104 !mvm->queue_info[IWL_MVM_DQA_BSS_CLIENT_QUEUE].hw_queue_refcount &&
cf961e16
LK		 1105 (mvm->queue_info[IWL_MVM_DQA_BSS_CLIENT_QUEUE].status ==
1106 IWL_MVM_QUEUE_FREE))
d5216a28
LK		 1107 queue = IWL_MVM_DQA_BSS_CLIENT_QUEUE;
1108 else
9794c64f
LK		 1109 queue = iwl_mvm_find_free_queue(mvm, mvmsta->sta_id,
1110 IWL_MVM_DQA_MIN_DATA_QUEUE,
d5216a28 1111 IWL_MVM_DQA_MAX_DATA_QUEUE);
24afba76
LK		 1112 if (queue < 0) {
1113 spin_unlock_bh(&mvm->queue_info_lock);
1114 IWL_ERR(mvm, "No available queues for new station\n");
1115 return -ENOSPC;
1116 }
cf961e16 1117 mvm->queue_info[queue].status = IWL_MVM_QUEUE_RESERVED;
24afba76
LK		 1118
1119 spin_unlock_bh(&mvm->queue_info_lock);
1120
1121 mvmsta->reserved_queue = queue;
1122
1123 IWL_DEBUG_TX_QUEUES(mvm, "Reserving data queue #%d for sta_id %d\n",
1124 queue, mvmsta->sta_id);
1125
1126 return 0;
1127}
1128
8d98ae6e
LK		 1129/*
1130 * In DQA mode, after a HW restart the queues should be allocated as before, in
1131 * order to avoid race conditions when there are shared queues. This function
1132 * does the re-mapping and queue allocation.
1133 *
1134 * Note that re-enabling aggregations isn't done in this function.
1135 */
1136static void iwl_mvm_realloc_queues_after_restart(struct iwl_mvm *mvm,
1137 struct iwl_mvm_sta *mvm_sta)
1138{
1139 unsigned int wdg_timeout =
1140 iwl_mvm_get_wd_timeout(mvm, mvm_sta->vif, false, false);
1141 int i;
1142 struct iwl_trans_txq_scd_cfg cfg = {
1143 .sta_id = mvm_sta->sta_id,
1144 .frame_limit = IWL_FRAME_LIMIT,
1145 };
1146
1147 /* Make sure reserved queue is still marked as such (or allocated) */
1148 mvm->queue_info[mvm_sta->reserved_queue].status =
1149 IWL_MVM_QUEUE_RESERVED;
1150
1151 for (i = 0; i <= IWL_MAX_TID_COUNT; i++) {
1152 struct iwl_mvm_tid_data *tid_data = &mvm_sta->tid_data[i];
1153 int txq_id = tid_data->txq_id;
1154 int ac;
1155 u8 mac_queue;
1156
1157 if (txq_id == IEEE80211_INVAL_HW_QUEUE)
1158 continue;
1159
1160 skb_queue_head_init(&tid_data->deferred_tx_frames);
1161
1162 ac = tid_to_mac80211_ac[i];
1163 mac_queue = mvm_sta->vif->hw_queue[ac];
1164
1165 cfg.tid = i;
1166 cfg.fifo = iwl_mvm_ac_to_tx_fifo[ac];
1167 cfg.aggregate = (txq_id >= IWL_MVM_DQA_MIN_DATA_QUEUE ||
1168 txq_id == IWL_MVM_DQA_BSS_CLIENT_QUEUE);
1169
1170 IWL_DEBUG_TX_QUEUES(mvm,
1171 "Re-mapping sta %d tid %d to queue %d\n",
1172 mvm_sta->sta_id, i, txq_id);
1173
1174 iwl_mvm_enable_txq(mvm, txq_id, mac_queue,
1175 IEEE80211_SEQ_TO_SN(tid_data->seq_number),
1176 &cfg, wdg_timeout);
1177
1178 mvm->queue_info[txq_id].status = IWL_MVM_QUEUE_READY;
1179 }
1180
1181 atomic_set(&mvm->pending_frames[mvm_sta->sta_id], 0);
1182}
1183
8ca151b5
JB		 1184int iwl_mvm_add_sta(struct iwl_mvm *mvm,
1185 struct ieee80211_vif *vif,
1186 struct ieee80211_sta *sta)
1187{
1188 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
9d8ce6af 1189 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
a571f5f6 1190 struct iwl_mvm_rxq_dup_data *dup_data;
8ca151b5
JB		 1191 int i, ret, sta_id;
1192
1193 lockdep_assert_held(&mvm->mutex);
1194
1195 if (!test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status))
b92e661b
EP		 1196 sta_id = iwl_mvm_find_free_sta_id(mvm,
1197 ieee80211_vif_type_p2p(vif));
8ca151b5
JB		 1198 else
1199 sta_id = mvm_sta->sta_id;
1200
36f4631c 1201 if (sta_id == IWL_MVM_STATION_COUNT)
8ca151b5
JB		 1202 return -ENOSPC;
1203
1204 spin_lock_init(&mvm_sta->lock);
1205
8d98ae6e
LK		 1206 /* In DQA mode, if this is a HW restart, re-alloc existing queues */
1207 if (iwl_mvm_is_dqa_supported(mvm) &&
1208 test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
1209 iwl_mvm_realloc_queues_after_restart(mvm, mvm_sta);
1210 goto update_fw;
1211 }
1212
8ca151b5
JB		 1213 mvm_sta->sta_id = sta_id;
1214 mvm_sta->mac_id_n_color = FW_CMD_ID_AND_COLOR(mvmvif->id,
1215 mvmvif->color);
1216 mvm_sta->vif = vif;
1217 mvm_sta->max_agg_bufsize = LINK_QUAL_AGG_FRAME_LIMIT_DEF;
9ee718aa
EL		 1218 mvm_sta->tx_protection = 0;
1219 mvm_sta->tt_tx_protection = false;
8ca151b5
JB		 1220
1221 /* HW restart, don't assume the memory has been zeroed */
e3d4bc8c 1222 atomic_set(&mvm->pending_frames[sta_id], 0);
69191afe 1223 mvm_sta->tid_disable_agg = 0xffff; /* No aggs at first */
8ca151b5 1224 mvm_sta->tfd_queue_msk = 0;
a0f6bf2a 1225
e3118ad7
LK		 1226 /*
1227 * Allocate new queues for a TDLS station, unless we're in DQA mode,
1228 * and then they'll be allocated dynamically
1229 */
1230 if (!iwl_mvm_is_dqa_supported(mvm) && sta->tdls) {
a0f6bf2a
AN		 1231 ret = iwl_mvm_tdls_sta_init(mvm, sta);
1232 if (ret)
1233 return ret;
24afba76 1234 } else if (!iwl_mvm_is_dqa_supported(mvm)) {
a0f6bf2a
AN		 1235 for (i = 0; i < IEEE80211_NUM_ACS; i++)
1236 if (vif->hw_queue[i] != IEEE80211_INVAL_HW_QUEUE)
1237 mvm_sta->tfd_queue_msk |= BIT(vif->hw_queue[i]);
1238 }
8ca151b5 1239
6d9d32b8 1240 /* for HW restart - reset everything but the sequence number */
24afba76 1241 for (i = 0; i <= IWL_MAX_TID_COUNT; i++) {
6d9d32b8
JB		 1242 u16 seq = mvm_sta->tid_data[i].seq_number;
1243 memset(&mvm_sta->tid_data[i], 0, sizeof(mvm_sta->tid_data[i]));
1244 mvm_sta->tid_data[i].seq_number = seq;
24afba76
LK		 1245
1246 if (!iwl_mvm_is_dqa_supported(mvm))
1247 continue;
1248
1249 /*
1250 * Mark all queues for this STA as unallocated and defer TX
1251 * frames until the queue is allocated
1252 */
1253 mvm_sta->tid_data[i].txq_id = IEEE80211_INVAL_HW_QUEUE;
1254 skb_queue_head_init(&mvm_sta->tid_data[i].deferred_tx_frames);
6d9d32b8 1255 }
24afba76 1256 mvm_sta->deferred_traffic_tid_map = 0;
efed6640 1257 mvm_sta->agg_tids = 0;
8ca151b5 1258
a571f5f6
SS		 1259 if (iwl_mvm_has_new_rx_api(mvm) &&
1260 !test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
1261 dup_data = kcalloc(mvm->trans->num_rx_queues,
1262 sizeof(*dup_data),
1263 GFP_KERNEL);
1264 if (!dup_data)
1265 return -ENOMEM;
1266 mvm_sta->dup_data = dup_data;
1267 }
1268
24afba76 1269 if (iwl_mvm_is_dqa_supported(mvm)) {
d5216a28
LK		 1270 ret = iwl_mvm_reserve_sta_stream(mvm, sta,
1271 ieee80211_vif_type_p2p(vif));
24afba76
LK		 1272 if (ret)
1273 goto err;
1274 }
1275
8d98ae6e 1276update_fw:
24afba76 1277 ret = iwl_mvm_sta_send_to_fw(mvm, sta, false, 0);
8ca151b5 1278 if (ret)
a0f6bf2a 1279 goto err;
8ca151b5 1280
9e848010
JB		 1281 if (vif->type == NL80211_IFTYPE_STATION) {
1282 if (!sta->tdls) {
1283 WARN_ON(mvmvif->ap_sta_id != IWL_MVM_STATION_COUNT);
1284 mvmvif->ap_sta_id = sta_id;
1285 } else {
1286 WARN_ON(mvmvif->ap_sta_id == IWL_MVM_STATION_COUNT);
1287 }
1288 }
8ca151b5
JB		 1289
1290 rcu_assign_pointer(mvm->fw_id_to_mac_id[sta_id], sta);
1291
1292 return 0;
a0f6bf2a
AN		 1293
1294err:
e3118ad7
LK		 1295 if (!iwl_mvm_is_dqa_supported(mvm) && sta->tdls)
1296 iwl_mvm_tdls_sta_deinit(mvm, sta);
a0f6bf2a 1297 return ret;
8ca151b5
JB		 1298}
1299
7a453973
JB		 1300int iwl_mvm_update_sta(struct iwl_mvm *mvm,
1301 struct ieee80211_vif *vif,
1302 struct ieee80211_sta *sta)
1303{
24afba76 1304 return iwl_mvm_sta_send_to_fw(mvm, sta, true, 0);
7a453973
JB		 1305}
1306
8ca151b5
JB		 1307int iwl_mvm_drain_sta(struct iwl_mvm *mvm, struct iwl_mvm_sta *mvmsta,
1308 bool drain)
1309{
f9dc0004 1310 struct iwl_mvm_add_sta_cmd cmd = {};
8ca151b5
JB		 1311 int ret;
1312 u32 status;
1313
1314 lockdep_assert_held(&mvm->mutex);
1315
1316 cmd.mac_id_n_color = cpu_to_le32(mvmsta->mac_id_n_color);
1317 cmd.sta_id = mvmsta->sta_id;
1318 cmd.add_modify = STA_MODE_MODIFY;
1319 cmd.station_flags = drain ? cpu_to_le32(STA_FLG_DRAIN_FLOW) : 0;
1320 cmd.station_flags_msk = cpu_to_le32(STA_FLG_DRAIN_FLOW);
1321
1322 status = ADD_STA_SUCCESS;
854c5705
SS		 1323 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA,
1324 iwl_mvm_add_sta_cmd_size(mvm),
f9dc0004 1325 &cmd, &status);
8ca151b5
JB		 1326 if (ret)
1327 return ret;
1328
837c4da9 1329 switch (status & IWL_ADD_STA_STATUS_MASK) {
8ca151b5
JB		 1330 case ADD_STA_SUCCESS:
1331 IWL_DEBUG_INFO(mvm, "Frames for staid %d will drained in fw\n",
1332 mvmsta->sta_id);
1333 break;
1334 default:
1335 ret = -EIO;
1336 IWL_ERR(mvm, "Couldn't drain frames for staid %d\n",
1337 mvmsta->sta_id);
1338 break;
1339 }
1340
1341 return ret;
1342}
1343
1344/*
1345 * Remove a station from the FW table. Before sending the command to remove
1346 * the station validate that the station is indeed known to the driver (sanity
1347 * only).
1348 */
1349static int iwl_mvm_rm_sta_common(struct iwl_mvm *mvm, u8 sta_id)
1350{
1351 struct ieee80211_sta *sta;
1352 struct iwl_mvm_rm_sta_cmd rm_sta_cmd = {
1353 .sta_id = sta_id,
1354 };
1355 int ret;
1356
1357 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
1358 lockdep_is_held(&mvm->mutex));
1359
1360 /* Note: internal stations are marked as error values */
1361 if (!sta) {
1362 IWL_ERR(mvm, "Invalid station id\n");
1363 return -EINVAL;
1364 }
1365
a1022927 1366 ret = iwl_mvm_send_cmd_pdu(mvm, REMOVE_STA, 0,
8ca151b5
JB		 1367 sizeof(rm_sta_cmd), &rm_sta_cmd);
1368 if (ret) {
1369 IWL_ERR(mvm, "Failed to remove station. Id=%d\n", sta_id);
1370 return ret;
1371 }
1372
1373 return 0;
1374}
1375
1376void iwl_mvm_sta_drained_wk(struct work_struct *wk)
1377{
1378 struct iwl_mvm *mvm = container_of(wk, struct iwl_mvm, sta_drained_wk);
1379 u8 sta_id;
1380
1381 /*
1382 * The mutex is needed because of the SYNC cmd, but not only: if the
1383 * work would run concurrently with iwl_mvm_rm_sta, it would run before
1384 * iwl_mvm_rm_sta sets the station as busy, and exit. Then
1385 * iwl_mvm_rm_sta would set the station as busy, and nobody will clean
1386 * that later.
1387 */
1388 mutex_lock(&mvm->mutex);
1389
1390 for_each_set_bit(sta_id, mvm->sta_drained, IWL_MVM_STATION_COUNT) {
1391 int ret;
1392 struct ieee80211_sta *sta =
1393 rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
1394 lockdep_is_held(&mvm->mutex));
1395
1ddbbb0c
JB		 1396 /*
1397 * This station is in use or RCU-removed; the latter happens in
1398 * managed mode, where mac80211 removes the station before we
1399 * can remove it from firmware (we can only do that after the
1400 * MAC is marked unassociated), and possibly while the deauth
1401 * frame to disconnect from the AP is still queued. Then, the
1402 * station pointer is -ENOENT when the last skb is reclaimed.
1403 */
1404 if (!IS_ERR(sta) || PTR_ERR(sta) == -ENOENT)
8ca151b5
JB		 1405 continue;
1406
1407 if (PTR_ERR(sta) == -EINVAL) {
1408 IWL_ERR(mvm, "Drained sta %d, but it is internal?\n",
1409 sta_id);
1410 continue;
1411 }
1412
1413 if (!sta) {
1414 IWL_ERR(mvm, "Drained sta %d, but it was NULL?\n",
1415 sta_id);
1416 continue;
1417 }
1418
1419 WARN_ON(PTR_ERR(sta) != -EBUSY);
1420 /* This station was removed and we waited until it got drained,
1421 * we can now proceed and remove it.
1422 */
1423 ret = iwl_mvm_rm_sta_common(mvm, sta_id);
1424 if (ret) {
1425 IWL_ERR(mvm,
1426 "Couldn't remove sta %d after it was drained\n",
1427 sta_id);
1428 continue;
1429 }
c531c771 1430 RCU_INIT_POINTER(mvm->fw_id_to_mac_id[sta_id], NULL);
8ca151b5 1431 clear_bit(sta_id, mvm->sta_drained);
a0f6bf2a
AN		 1432
1433 if (mvm->tfd_drained[sta_id]) {
1434 unsigned long i, msk = mvm->tfd_drained[sta_id];
1435
a4ca3ed4 1436 for_each_set_bit(i, &msk, sizeof(msk) * BITS_PER_BYTE)
06ecdba3
AN		 1437 iwl_mvm_disable_txq(mvm, i, i,
1438 IWL_MAX_TID_COUNT, 0);
a0f6bf2a
AN		 1439
1440 mvm->tfd_drained[sta_id] = 0;
1441 IWL_DEBUG_TDLS(mvm, "Drained sta %d, with queues %ld\n",
1442 sta_id, msk);
1443 }
8ca151b5
JB		 1444 }
1445
1446 mutex_unlock(&mvm->mutex);
1447}
1448
24afba76
LK		 1449static void iwl_mvm_disable_sta_queues(struct iwl_mvm *mvm,
1450 struct ieee80211_vif *vif,
1451 struct iwl_mvm_sta *mvm_sta)
1452{
1453 int ac;
1454 int i;
1455
1456 lockdep_assert_held(&mvm->mutex);
1457
1458 for (i = 0; i < ARRAY_SIZE(mvm_sta->tid_data); i++) {
1459 if (mvm_sta->tid_data[i].txq_id == IEEE80211_INVAL_HW_QUEUE)
1460 continue;
1461
1462 ac = iwl_mvm_tid_to_ac_queue(i);
1463 iwl_mvm_disable_txq(mvm, mvm_sta->tid_data[i].txq_id,
1464 vif->hw_queue[ac], i, 0);
1465 mvm_sta->tid_data[i].txq_id = IEEE80211_INVAL_HW_QUEUE;
1466 }
1467}
1468
8ca151b5
JB		 1469int iwl_mvm_rm_sta(struct iwl_mvm *mvm,
1470 struct ieee80211_vif *vif,
1471 struct ieee80211_sta *sta)
1472{
1473 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
9d8ce6af 1474 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
8ca151b5
JB		 1475 int ret;
1476
1477 lockdep_assert_held(&mvm->mutex);
1478
a571f5f6
SS		 1479 if (iwl_mvm_has_new_rx_api(mvm))
1480 kfree(mvm_sta->dup_data);
1481
a6f035a0
LK		 1482 if ((vif->type == NL80211_IFTYPE_STATION &&
1483 mvmvif->ap_sta_id == mvm_sta->sta_id) ||
1484 iwl_mvm_is_dqa_supported(mvm)){
fe92e32a
EG		 1485 ret = iwl_mvm_drain_sta(mvm, mvm_sta, true);
1486 if (ret)
1487 return ret;
80d85655 1488 /* flush its queues here since we are freeing mvm_sta */
5888a40c 1489 ret = iwl_mvm_flush_tx_path(mvm, mvm_sta->tfd_queue_msk, 0);
fe92e32a
EG		 1490 if (ret)
1491 return ret;
1492 ret = iwl_trans_wait_tx_queue_empty(mvm->trans,
1493 mvm_sta->tfd_queue_msk);
1494 if (ret)
1495 return ret;
1496 ret = iwl_mvm_drain_sta(mvm, mvm_sta, false);
80d85655 1497
24afba76 1498 /* If DQA is supported - the queues can be disabled now */
a0315dea
LK		 1499 if (iwl_mvm_is_dqa_supported(mvm)) {
1500 u8 reserved_txq = mvm_sta->reserved_queue;
1501 enum iwl_mvm_queue_status *status;
1502
24afba76
LK		 1503 iwl_mvm_disable_sta_queues(mvm, vif, mvm_sta);
1504
a0315dea
LK		 1505 /*
1506 * If no traffic has gone through the reserved TXQ - it
1507 * is still marked as IWL_MVM_QUEUE_RESERVED, and
1508 * should be manually marked as free again
1509 */
1510 spin_lock_bh(&mvm->queue_info_lock);
1511 status = &mvm->queue_info[reserved_txq].status;
1512 if (WARN((*status != IWL_MVM_QUEUE_RESERVED) &&
1513 (*status != IWL_MVM_QUEUE_FREE),
1514 "sta_id %d reserved txq %d status %d",
1515 mvm_sta->sta_id, reserved_txq, *status)) {
1516 spin_unlock_bh(&mvm->queue_info_lock);
1517 return -EINVAL;
1518 }
1519
1520 *status = IWL_MVM_QUEUE_FREE;
1521 spin_unlock_bh(&mvm->queue_info_lock);
1522 }
1523
e3118ad7
LK		 1524 if (vif->type == NL80211_IFTYPE_STATION &&
1525 mvmvif->ap_sta_id == mvm_sta->sta_id) {
1526 /* if associated - we can't remove the AP STA now */
1527 if (vif->bss_conf.assoc)
1528 return ret;
8ca151b5 1529
e3118ad7
LK		 1530 /* unassoc - go ahead - remove the AP STA now */
1531 mvmvif->ap_sta_id = IWL_MVM_STATION_COUNT;
37577fe2 1532
e3118ad7
LK		 1533 /* clear d0i3_ap_sta_id if no longer relevant */
1534 if (mvm->d0i3_ap_sta_id == mvm_sta->sta_id)
1535 mvm->d0i3_ap_sta_id = IWL_MVM_STATION_COUNT;
1536 }
8ca151b5
JB		 1537 }
1538
1d3c3f63
AN		 1539 /*
1540 * This shouldn't happen - the TDLS channel switch should be canceled
1541 * before the STA is removed.
1542 */
1543 if (WARN_ON_ONCE(mvm->tdls_cs.peer.sta_id == mvm_sta->sta_id)) {
1544 mvm->tdls_cs.peer.sta_id = IWL_MVM_STATION_COUNT;
1545 cancel_delayed_work(&mvm->tdls_cs.dwork);
1546 }
1547
e3d4bc8c
EG		 1548 /*
1549 * Make sure that the tx response code sees the station as -EBUSY and
1550 * calls the drain worker.
1551 */
1552 spin_lock_bh(&mvm_sta->lock);
8ca151b5
JB		 1553 /*
1554 * There are frames pending on the AC queues for this station.
1555 * We need to wait until all the frames are drained...
1556 */
e3d4bc8c 1557 if (atomic_read(&mvm->pending_frames[mvm_sta->sta_id])) {
8ca151b5
JB		 1558 rcu_assign_pointer(mvm->fw_id_to_mac_id[mvm_sta->sta_id],
1559 ERR_PTR(-EBUSY));
e3d4bc8c 1560 spin_unlock_bh(&mvm_sta->lock);
a0f6bf2a
AN		 1561
1562 /* disable TDLS sta queues on drain complete */
1563 if (sta->tdls) {
1564 mvm->tfd_drained[mvm_sta->sta_id] =
1565 mvm_sta->tfd_queue_msk;
1566 IWL_DEBUG_TDLS(mvm, "Draining TDLS sta %d\n",
1567 mvm_sta->sta_id);
1568 }
1569
e3d4bc8c 1570 ret = iwl_mvm_drain_sta(mvm, mvm_sta, true);
8ca151b5 1571 } else {
e3d4bc8c 1572 spin_unlock_bh(&mvm_sta->lock);
a0f6bf2a 1573
e3118ad7 1574 if (!iwl_mvm_is_dqa_supported(mvm) && sta->tdls)
a0f6bf2a
AN		 1575 iwl_mvm_tdls_sta_deinit(mvm, sta);
1576
8ca151b5 1577 ret = iwl_mvm_rm_sta_common(mvm, mvm_sta->sta_id);
c531c771 1578 RCU_INIT_POINTER(mvm->fw_id_to_mac_id[mvm_sta->sta_id], NULL);
8ca151b5
JB		 1579 }
1580
1581 return ret;
1582}
1583
1584int iwl_mvm_rm_sta_id(struct iwl_mvm *mvm,
1585 struct ieee80211_vif *vif,
1586 u8 sta_id)
1587{
1588 int ret = iwl_mvm_rm_sta_common(mvm, sta_id);
1589
1590 lockdep_assert_held(&mvm->mutex);
1591
c531c771 1592 RCU_INIT_POINTER(mvm->fw_id_to_mac_id[sta_id], NULL);
8ca151b5
JB		 1593 return ret;
1594}
1595
0e39eb03
CRI		 1596int iwl_mvm_allocate_int_sta(struct iwl_mvm *mvm,
1597 struct iwl_mvm_int_sta *sta,
1598 u32 qmask, enum nl80211_iftype iftype)
8ca151b5
JB		 1599{
1600 if (!test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
b92e661b 1601 sta->sta_id = iwl_mvm_find_free_sta_id(mvm, iftype);
8ca151b5
JB		 1602 if (WARN_ON_ONCE(sta->sta_id == IWL_MVM_STATION_COUNT))
1603 return -ENOSPC;
1604 }
1605
1606 sta->tfd_queue_msk = qmask;
1607
1608 /* put a non-NULL value so iterating over the stations won't stop */
1609 rcu_assign_pointer(mvm->fw_id_to_mac_id[sta->sta_id], ERR_PTR(-EINVAL));
1610 return 0;
1611}
1612
712b24ad
JB		 1613static void iwl_mvm_dealloc_int_sta(struct iwl_mvm *mvm,
1614 struct iwl_mvm_int_sta *sta)
8ca151b5 1615{
c531c771 1616 RCU_INIT_POINTER(mvm->fw_id_to_mac_id[sta->sta_id], NULL);
8ca151b5
JB		 1617 memset(sta, 0, sizeof(struct iwl_mvm_int_sta));
1618 sta->sta_id = IWL_MVM_STATION_COUNT;
1619}
1620
1621static int iwl_mvm_add_int_sta_common(struct iwl_mvm *mvm,
1622 struct iwl_mvm_int_sta *sta,
1623 const u8 *addr,
1624 u16 mac_id, u16 color)
1625{
f9dc0004 1626 struct iwl_mvm_add_sta_cmd cmd;
8ca151b5
JB		 1627 int ret;
1628 u32 status;
1629
1630 lockdep_assert_held(&mvm->mutex);
1631
f9dc0004 1632 memset(&cmd, 0, sizeof(cmd));
8ca151b5
JB		 1633 cmd.sta_id = sta->sta_id;
1634 cmd.mac_id_n_color = cpu_to_le32(FW_CMD_ID_AND_COLOR(mac_id,
1635 color));
1636
1637 cmd.tfd_queue_msk = cpu_to_le32(sta->tfd_queue_msk);
cf0cda19 1638 cmd.tid_disable_tx = cpu_to_le16(0xffff);
8ca151b5
JB		 1639
1640 if (addr)
1641 memcpy(cmd.addr, addr, ETH_ALEN);
1642
854c5705
SS		 1643 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA,
1644 iwl_mvm_add_sta_cmd_size(mvm),
f9dc0004 1645 &cmd, &status);
8ca151b5
JB		 1646 if (ret)
1647 return ret;
1648
837c4da9 1649 switch (status & IWL_ADD_STA_STATUS_MASK) {
8ca151b5
JB		 1650 case ADD_STA_SUCCESS:
1651 IWL_DEBUG_INFO(mvm, "Internal station added.\n");
1652 return 0;
1653 default:
1654 ret = -EIO;
1655 IWL_ERR(mvm, "Add internal station failed, status=0x%x\n",
1656 status);
1657 break;
1658 }
1659 return ret;
1660}
1661
1662int iwl_mvm_add_aux_sta(struct iwl_mvm *mvm)
1663{
4cf677fd
EG		 1664 unsigned int wdg_timeout = iwlmvm_mod_params.tfd_q_hang_detect ?
1665 mvm->cfg->base_params->wd_timeout :
1666 IWL_WATCHDOG_DISABLED;
8ca151b5
JB		 1667 int ret;
1668
1669 lockdep_assert_held(&mvm->mutex);
1670
7da91b0e 1671 /* Map Aux queue to fifo - needs to happen before adding Aux station */
28d0793e
LK		 1672 if (!iwl_mvm_is_dqa_supported(mvm))
1673 iwl_mvm_enable_ac_txq(mvm, mvm->aux_queue, mvm->aux_queue,
1674 IWL_MVM_TX_FIFO_MCAST, 0, wdg_timeout);
7da91b0e
AM		 1675
1676 /* Allocate aux station and assign to it the aux queue */
1677 ret = iwl_mvm_allocate_int_sta(mvm, &mvm->aux_sta, BIT(mvm->aux_queue),
b92e661b 1678 NL80211_IFTYPE_UNSPECIFIED);
8ca151b5
JB		 1679 if (ret)
1680 return ret;
1681
28d0793e
LK		 1682 if (iwl_mvm_is_dqa_supported(mvm)) {
1683 struct iwl_trans_txq_scd_cfg cfg = {
1684 .fifo = IWL_MVM_TX_FIFO_MCAST,
1685 .sta_id = mvm->aux_sta.sta_id,
1686 .tid = IWL_MAX_TID_COUNT,
1687 .aggregate = false,
1688 .frame_limit = IWL_FRAME_LIMIT,
1689 };
1690
1691 iwl_mvm_enable_txq(mvm, mvm->aux_queue, mvm->aux_queue, 0, &cfg,
1692 wdg_timeout);
1693 }
1694
8ca151b5
JB		 1695 ret = iwl_mvm_add_int_sta_common(mvm, &mvm->aux_sta, NULL,
1696 MAC_INDEX_AUX, 0);
1697
1698 if (ret)
1699 iwl_mvm_dealloc_int_sta(mvm, &mvm->aux_sta);
1700 return ret;
1701}
1702
0e39eb03
CRI		 1703int iwl_mvm_add_snif_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
1704{
1705 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1706
1707 lockdep_assert_held(&mvm->mutex);
1708 return iwl_mvm_add_int_sta_common(mvm, &mvm->snif_sta, vif->addr,
1709 mvmvif->id, 0);
1710}
1711
1712int iwl_mvm_rm_snif_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
1713{
1714 int ret;
1715
1716 lockdep_assert_held(&mvm->mutex);
1717
1718 ret = iwl_mvm_rm_sta_common(mvm, mvm->snif_sta.sta_id);
1719 if (ret)
1720 IWL_WARN(mvm, "Failed sending remove station\n");
1721
1722 return ret;
1723}
1724
1725void iwl_mvm_dealloc_snif_sta(struct iwl_mvm *mvm)
1726{
1727 iwl_mvm_dealloc_int_sta(mvm, &mvm->snif_sta);
1728}
1729
712b24ad
JB		 1730void iwl_mvm_del_aux_sta(struct iwl_mvm *mvm)
1731{
1732 lockdep_assert_held(&mvm->mutex);
1733
1734 iwl_mvm_dealloc_int_sta(mvm, &mvm->aux_sta);
1735}
1736
8ca151b5
JB		 1737/*
1738 * Send the add station command for the vif's broadcast station.
1739 * Assumes that the station was already allocated.
1740 *
1741 * @mvm: the mvm component
1742 * @vif: the interface to which the broadcast station is added
1743 * @bsta: the broadcast station to add.
1744 */
013290aa 1745int iwl_mvm_send_add_bcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
8ca151b5
JB		 1746{
1747 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
013290aa 1748 struct iwl_mvm_int_sta *bsta = &mvmvif->bcast_sta;
5023d966 1749 static const u8 _baddr[] = {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF};
a4243402 1750 const u8 *baddr = _baddr;
8ca151b5
JB		 1751
1752 lockdep_assert_held(&mvm->mutex);
1753
de24f638
LK		 1754 if (iwl_mvm_is_dqa_supported(mvm)) {
1755 struct iwl_trans_txq_scd_cfg cfg = {
1756 .fifo = IWL_MVM_TX_FIFO_VO,
1757 .sta_id = mvmvif->bcast_sta.sta_id,
1758 .tid = IWL_MAX_TID_COUNT,
1759 .aggregate = false,
1760 .frame_limit = IWL_FRAME_LIMIT,
1761 };
1762 unsigned int wdg_timeout =
1763 iwl_mvm_get_wd_timeout(mvm, vif, false, false);
1764 int queue;
1765
1766 if ((vif->type == NL80211_IFTYPE_AP) &&
1767 (mvmvif->bcast_sta.tfd_queue_msk &
1768 BIT(IWL_MVM_DQA_AP_PROBE_RESP_QUEUE)))
1769 queue = IWL_MVM_DQA_AP_PROBE_RESP_QUEUE;
4c965139
LK		 1770 else if ((vif->type == NL80211_IFTYPE_P2P_DEVICE) &&
1771 (mvmvif->bcast_sta.tfd_queue_msk &
1772 BIT(IWL_MVM_DQA_P2P_DEVICE_QUEUE)))
1773 queue = IWL_MVM_DQA_P2P_DEVICE_QUEUE;
de24f638
LK		 1774 else if (WARN(1, "Missed required TXQ for adding bcast STA\n"))
1775 return -EINVAL;
1776
1777 iwl_mvm_enable_txq(mvm, queue, vif->hw_queue[0], 0, &cfg,
1778 wdg_timeout);
1779 }
1780
5023d966
JB		 1781 if (vif->type == NL80211_IFTYPE_ADHOC)
1782 baddr = vif->bss_conf.bssid;
1783
8ca151b5
JB		 1784 if (WARN_ON_ONCE(bsta->sta_id == IWL_MVM_STATION_COUNT))
1785 return -ENOSPC;
1786
1787 return iwl_mvm_add_int_sta_common(mvm, bsta, baddr,
1788 mvmvif->id, mvmvif->color);
1789}
1790
1791/* Send the FW a request to remove the station from it's internal data
1792 * structures, but DO NOT remove the entry from the local data structures. */
013290aa 1793int iwl_mvm_send_rm_bcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
8ca151b5 1794{
013290aa 1795 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
8ca151b5
JB		 1796 int ret;
1797
1798 lockdep_assert_held(&mvm->mutex);
1799
013290aa 1800 ret = iwl_mvm_rm_sta_common(mvm, mvmvif->bcast_sta.sta_id);
8ca151b5
JB		 1801 if (ret)
1802 IWL_WARN(mvm, "Failed sending remove station\n");
1803 return ret;
1804}
1805
013290aa
JB		 1806int iwl_mvm_alloc_bcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
1807{
1808 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
de24f638 1809 u32 qmask = 0;
013290aa
JB		 1810
1811 lockdep_assert_held(&mvm->mutex);
1812
de24f638
LK		 1813 if (!iwl_mvm_is_dqa_supported(mvm))
1814 qmask = iwl_mvm_mac_get_queues_mask(vif);
013290aa 1815
de24f638
LK		 1816 if (vif->type == NL80211_IFTYPE_AP) {
1817 /*
1818 * The firmware defines the TFD queue mask to only be relevant
1819 * for *unicast* queues, so the multicast (CAB) queue shouldn't
1820 * be included.
1821 */
013290aa
JB		 1822 qmask &= ~BIT(vif->cab_queue);
1823
de24f638
LK		 1824 if (iwl_mvm_is_dqa_supported(mvm))
1825 qmask |= BIT(IWL_MVM_DQA_AP_PROBE_RESP_QUEUE);
4c965139
LK		 1826 } else if (iwl_mvm_is_dqa_supported(mvm) &&
1827 vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1828 qmask |= BIT(IWL_MVM_DQA_P2P_DEVICE_QUEUE);
de24f638
LK		 1829 }
1830
013290aa
JB		 1831 return iwl_mvm_allocate_int_sta(mvm, &mvmvif->bcast_sta, qmask,
1832 ieee80211_vif_type_p2p(vif));
1833}
1834
8ca151b5
JB		 1835/* Allocate a new station entry for the broadcast station to the given vif,
1836 * and send it to the FW.
1837 * Note that each P2P mac should have its own broadcast station.
1838 *
1839 * @mvm: the mvm component
1840 * @vif: the interface to which the broadcast station is added
1841 * @bsta: the broadcast station to add. */
013290aa 1842int iwl_mvm_add_bcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
8ca151b5
JB		 1843{
1844 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
013290aa 1845 struct iwl_mvm_int_sta *bsta = &mvmvif->bcast_sta;
8ca151b5
JB		 1846 int ret;
1847
1848 lockdep_assert_held(&mvm->mutex);
1849
013290aa 1850 ret = iwl_mvm_alloc_bcast_sta(mvm, vif);
8ca151b5
JB		 1851 if (ret)
1852 return ret;
1853
013290aa 1854 ret = iwl_mvm_send_add_bcast_sta(mvm, vif);
8ca151b5
JB		 1855
1856 if (ret)
1857 iwl_mvm_dealloc_int_sta(mvm, bsta);
013290aa 1858
8ca151b5
JB		 1859 return ret;
1860}
1861
013290aa
JB		 1862void iwl_mvm_dealloc_bcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
1863{
1864 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1865
1866 iwl_mvm_dealloc_int_sta(mvm, &mvmvif->bcast_sta);
1867}
1868
8ca151b5
JB		 1869/*
1870 * Send the FW a request to remove the station from it's internal data
1871 * structures, and in addition remove it from the local data structure.
1872 */
013290aa 1873int iwl_mvm_rm_bcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
8ca151b5
JB		 1874{
1875 int ret;
1876
1877 lockdep_assert_held(&mvm->mutex);
1878
013290aa
JB		 1879 ret = iwl_mvm_send_rm_bcast_sta(mvm, vif);
1880
1881 iwl_mvm_dealloc_bcast_sta(mvm, vif);
8ca151b5 1882
8ca151b5
JB		 1883 return ret;
1884}
1885
113a0447
EG		 1886#define IWL_MAX_RX_BA_SESSIONS 16
1887
b915c101 1888static void iwl_mvm_sync_rxq_del_ba(struct iwl_mvm *mvm, u8 baid)
10b2b201 1889{
b915c101
SS		 1890 struct iwl_mvm_delba_notif notif = {
1891 .metadata.type = IWL_MVM_RXQ_NOTIF_DEL_BA,
1892 .metadata.sync = 1,
1893 .delba.baid = baid,
10b2b201 1894 };
b915c101
SS		 1895 iwl_mvm_sync_rx_queues_internal(mvm, (void *)&notif, sizeof(notif));
1896};
10b2b201 1897
b915c101
SS		 1898static void iwl_mvm_free_reorder(struct iwl_mvm *mvm,
1899 struct iwl_mvm_baid_data *data)
1900{
1901 int i;
1902
1903 iwl_mvm_sync_rxq_del_ba(mvm, data->baid);
1904
1905 for (i = 0; i < mvm->trans->num_rx_queues; i++) {
1906 int j;
1907 struct iwl_mvm_reorder_buffer *reorder_buf =
1908 &data->reorder_buf[i];
1909
0690405f
SS		 1910 spin_lock_bh(&reorder_buf->lock);
1911 if (likely(!reorder_buf->num_stored)) {
1912 spin_unlock_bh(&reorder_buf->lock);
b915c101 1913 continue;
0690405f 1914 }
b915c101
SS		 1915
1916 /*
1917 * This shouldn't happen in regular DELBA since the internal
1918 * delBA notification should trigger a release of all frames in
1919 * the reorder buffer.
1920 */
1921 WARN_ON(1);
1922
1923 for (j = 0; j < reorder_buf->buf_size; j++)
1924 __skb_queue_purge(&reorder_buf->entries[j]);
0690405f
SS		 1925 /*
1926 * Prevent timer re-arm. This prevents a very far fetched case
1927 * where we timed out on the notification. There may be prior
1928 * RX frames pending in the RX queue before the notification
1929 * that might get processed between now and the actual deletion
1930 * and we would re-arm the timer although we are deleting the
1931 * reorder buffer.
1932 */
1933 reorder_buf->removed = true;
1934 spin_unlock_bh(&reorder_buf->lock);
1935 del_timer_sync(&reorder_buf->reorder_timer);
b915c101
SS		 1936 }
1937}
1938
1939static void iwl_mvm_init_reorder_buffer(struct iwl_mvm *mvm,
1940 u32 sta_id,
1941 struct iwl_mvm_baid_data *data,
1942 u16 ssn, u8 buf_size)
1943{
1944 int i;
1945
1946 for (i = 0; i < mvm->trans->num_rx_queues; i++) {
1947 struct iwl_mvm_reorder_buffer *reorder_buf =
1948 &data->reorder_buf[i];
1949 int j;
1950
1951 reorder_buf->num_stored = 0;
1952 reorder_buf->head_sn = ssn;
1953 reorder_buf->buf_size = buf_size;
0690405f
SS		 1954 /* rx reorder timer */
1955 reorder_buf->reorder_timer.function =
1956 iwl_mvm_reorder_timer_expired;
1957 reorder_buf->reorder_timer.data = (unsigned long)reorder_buf;
1958 init_timer(&reorder_buf->reorder_timer);
1959 spin_lock_init(&reorder_buf->lock);
1960 reorder_buf->mvm = mvm;
b915c101
SS		 1961 reorder_buf->queue = i;
1962 reorder_buf->sta_id = sta_id;
1963 for (j = 0; j < reorder_buf->buf_size; j++)
1964 __skb_queue_head_init(&reorder_buf->entries[j]);
1965 }
10b2b201
SS		 1966}
1967
8ca151b5 1968int iwl_mvm_sta_rx_agg(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
10b2b201 1969 int tid, u16 ssn, bool start, u8 buf_size, u16 timeout)
8ca151b5 1970{
9d8ce6af 1971 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
f9dc0004 1972 struct iwl_mvm_add_sta_cmd cmd = {};
10b2b201 1973 struct iwl_mvm_baid_data *baid_data = NULL;
8ca151b5
JB		 1974 int ret;
1975 u32 status;
1976
1977 lockdep_assert_held(&mvm->mutex);
1978
113a0447
EG		 1979 if (start && mvm->rx_ba_sessions >= IWL_MAX_RX_BA_SESSIONS) {
1980 IWL_WARN(mvm, "Not enough RX BA SESSIONS\n");
1981 return -ENOSPC;
1982 }
1983
10b2b201
SS		 1984 if (iwl_mvm_has_new_rx_api(mvm) && start) {
1985 /*
1986 * Allocate here so if allocation fails we can bail out early
1987 * before starting the BA session in the firmware
1988 */
b915c101
SS		 1989 baid_data = kzalloc(sizeof(*baid_data) +
1990 mvm->trans->num_rx_queues *
1991 sizeof(baid_data->reorder_buf[0]),
1992 GFP_KERNEL);
10b2b201
SS		 1993 if (!baid_data)
1994 return -ENOMEM;
1995 }
1996
8ca151b5
JB		 1997 cmd.mac_id_n_color = cpu_to_le32(mvm_sta->mac_id_n_color);
1998 cmd.sta_id = mvm_sta->sta_id;
1999 cmd.add_modify = STA_MODE_MODIFY;
93a42667
EG		 2000 if (start) {
2001 cmd.add_immediate_ba_tid = (u8) tid;
2002 cmd.add_immediate_ba_ssn = cpu_to_le16(ssn);
854c5705 2003 cmd.rx_ba_window = cpu_to_le16((u16)buf_size);
93a42667
EG		 2004 } else {
2005 cmd.remove_immediate_ba_tid = (u8) tid;
2006 }
8ca151b5
JB		 2007 cmd.modify_mask = start ? STA_MODIFY_ADD_BA_TID :
2008 STA_MODIFY_REMOVE_BA_TID;
2009
2010 status = ADD_STA_SUCCESS;
854c5705
SS		 2011 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA,
2012 iwl_mvm_add_sta_cmd_size(mvm),
f9dc0004 2013 &cmd, &status);
8ca151b5 2014 if (ret)
10b2b201 2015 goto out_free;
8ca151b5 2016
837c4da9 2017 switch (status & IWL_ADD_STA_STATUS_MASK) {
8ca151b5 2018 case ADD_STA_SUCCESS:
35263a03
SS		 2019 IWL_DEBUG_HT(mvm, "RX BA Session %sed in fw\n",
2020 start ? "start" : "stopp");
8ca151b5
JB		 2021 break;
2022 case ADD_STA_IMMEDIATE_BA_FAILURE:
2023 IWL_WARN(mvm, "RX BA Session refused by fw\n");
2024 ret = -ENOSPC;
2025 break;
2026 default:
2027 ret = -EIO;
2028 IWL_ERR(mvm, "RX BA Session failed %sing, status 0x%x\n",
2029 start ? "start" : "stopp", status);
2030 break;
2031 }
2032
10b2b201
SS		 2033 if (ret)
2034 goto out_free;
2035
2036 if (start) {
2037 u8 baid;
2038
2039 mvm->rx_ba_sessions++;
2040
2041 if (!iwl_mvm_has_new_rx_api(mvm))
2042 return 0;
2043
2044 if (WARN_ON(!(status & IWL_ADD_STA_BAID_VALID_MASK))) {
2045 ret = -EINVAL;
2046 goto out_free;
2047 }
2048 baid = (u8)((status & IWL_ADD_STA_BAID_MASK) >>
2049 IWL_ADD_STA_BAID_SHIFT);
2050 baid_data->baid = baid;
2051 baid_data->timeout = timeout;
2052 baid_data->last_rx = jiffies;
2053 init_timer(&baid_data->session_timer);
2054 baid_data->session_timer.function =
2055 iwl_mvm_rx_agg_session_expired;
2056 baid_data->session_timer.data =
2057 (unsigned long)&mvm->baid_map[baid];
2058 baid_data->mvm = mvm;
2059 baid_data->tid = tid;
2060 baid_data->sta_id = mvm_sta->sta_id;
2061
2062 mvm_sta->tid_to_baid[tid] = baid;
2063 if (timeout)
2064 mod_timer(&baid_data->session_timer,
2065 TU_TO_EXP_TIME(timeout * 2));
2066
b915c101
SS		 2067 iwl_mvm_init_reorder_buffer(mvm, mvm_sta->sta_id,
2068 baid_data, ssn, buf_size);
10b2b201
SS		 2069 /*
2070 * protect the BA data with RCU to cover a case where our
2071 * internal RX sync mechanism will timeout (not that it's
2072 * supposed to happen) and we will free the session data while
2073 * RX is being processed in parallel
2074 */
35263a03
SS		 2075 IWL_DEBUG_HT(mvm, "Sta %d(%d) is assigned to BAID %d\n",
2076 mvm_sta->sta_id, tid, baid);
10b2b201
SS		 2077 WARN_ON(rcu_access_pointer(mvm->baid_map[baid]));
2078 rcu_assign_pointer(mvm->baid_map[baid], baid_data);
60dec523 2079 } else {
10b2b201
SS		 2080 u8 baid = mvm_sta->tid_to_baid[tid];
2081
60dec523
SS		 2082 if (mvm->rx_ba_sessions > 0)
2083 /* check that restart flow didn't zero the counter */
2084 mvm->rx_ba_sessions--;
10b2b201
SS		 2085 if (!iwl_mvm_has_new_rx_api(mvm))
2086 return 0;
2087
2088 if (WARN_ON(baid == IWL_RX_REORDER_DATA_INVALID_BAID))
2089 return -EINVAL;
2090
2091 baid_data = rcu_access_pointer(mvm->baid_map[baid]);
2092 if (WARN_ON(!baid_data))
2093 return -EINVAL;
2094
2095 /* synchronize all rx queues so we can safely delete */
b915c101 2096 iwl_mvm_free_reorder(mvm, baid_data);
10b2b201 2097 del_timer_sync(&baid_data->session_timer);
10b2b201
SS		 2098 RCU_INIT_POINTER(mvm->baid_map[baid], NULL);
2099 kfree_rcu(baid_data, rcu_head);
35263a03 2100 IWL_DEBUG_HT(mvm, "BAID %d is free\n", baid);
113a0447 2101 }
10b2b201 2102 return 0;
113a0447 2103
10b2b201
SS		 2104out_free:
2105 kfree(baid_data);
8ca151b5
JB		 2106 return ret;
2107}
2108
9794c64f
LK		 2109int iwl_mvm_sta_tx_agg(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
2110 int tid, u8 queue, bool start)
8ca151b5 2111{
9d8ce6af 2112 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
f9dc0004 2113 struct iwl_mvm_add_sta_cmd cmd = {};
8ca151b5
JB		 2114 int ret;
2115 u32 status;
2116
2117 lockdep_assert_held(&mvm->mutex);
2118
2119 if (start) {
2120 mvm_sta->tfd_queue_msk |= BIT(queue);
2121 mvm_sta->tid_disable_agg &= ~BIT(tid);
2122 } else {
cf961e16
LK		 2123 /* In DQA-mode the queue isn't removed on agg termination */
2124 if (!iwl_mvm_is_dqa_supported(mvm))
2125 mvm_sta->tfd_queue_msk &= ~BIT(queue);
8ca151b5
JB		 2126 mvm_sta->tid_disable_agg |= BIT(tid);
2127 }
2128
2129 cmd.mac_id_n_color = cpu_to_le32(mvm_sta->mac_id_n_color);
2130 cmd.sta_id = mvm_sta->sta_id;
2131 cmd.add_modify = STA_MODE_MODIFY;
2132 cmd.modify_mask = STA_MODIFY_QUEUES | STA_MODIFY_TID_DISABLE_TX;
2133 cmd.tfd_queue_msk = cpu_to_le32(mvm_sta->tfd_queue_msk);
2134 cmd.tid_disable_tx = cpu_to_le16(mvm_sta->tid_disable_agg);
2135
2136 status = ADD_STA_SUCCESS;
854c5705
SS		 2137 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA,
2138 iwl_mvm_add_sta_cmd_size(mvm),
f9dc0004 2139 &cmd, &status);
8ca151b5
JB		 2140 if (ret)
2141 return ret;
2142
837c4da9 2143 switch (status & IWL_ADD_STA_STATUS_MASK) {
8ca151b5
JB		 2144 case ADD_STA_SUCCESS:
2145 break;
2146 default:
2147 ret = -EIO;
2148 IWL_ERR(mvm, "TX BA Session failed %sing, status 0x%x\n",
2149 start ? "start" : "stopp", status);
2150 break;
2151 }
2152
2153 return ret;
2154}
2155
b797e3fb 2156const u8 tid_to_mac80211_ac[] = {
8ca151b5
JB		 2157 IEEE80211_AC_BE,
2158 IEEE80211_AC_BK,
2159 IEEE80211_AC_BK,
2160 IEEE80211_AC_BE,
2161 IEEE80211_AC_VI,
2162 IEEE80211_AC_VI,
2163 IEEE80211_AC_VO,
2164 IEEE80211_AC_VO,
9794c64f 2165 IEEE80211_AC_VO, /* We treat MGMT as TID 8, which is set as AC_VO */
8ca151b5
JB		 2166};
2167
3e56eadf
JB		 2168static const u8 tid_to_ucode_ac[] = {
2169 AC_BE,
2170 AC_BK,
2171 AC_BK,
2172 AC_BE,
2173 AC_VI,
2174 AC_VI,
2175 AC_VO,
2176 AC_VO,
2177};
2178
8ca151b5
JB		 2179int iwl_mvm_sta_tx_agg_start(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
2180 struct ieee80211_sta *sta, u16 tid, u16 *ssn)
2181{
5b577a90 2182 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
8ca151b5
JB		 2183 struct iwl_mvm_tid_data *tid_data;
2184 int txq_id;
4ecafae9 2185 int ret;
8ca151b5
JB		 2186
2187 if (WARN_ON_ONCE(tid >= IWL_MAX_TID_COUNT))
2188 return -EINVAL;
2189
2190 if (mvmsta->tid_data[tid].state != IWL_AGG_OFF) {
2191 IWL_ERR(mvm, "Start AGG when state is not IWL_AGG_OFF %d!\n",
2192 mvmsta->tid_data[tid].state);
2193 return -ENXIO;
2194 }
2195
2196 lockdep_assert_held(&mvm->mutex);
2197
b2492501
AN		 2198 spin_lock_bh(&mvmsta->lock);
2199
2200 /* possible race condition - we entered D0i3 while starting agg */
2201 if (test_bit(IWL_MVM_STATUS_IN_D0I3, &mvm->status)) {
2202 spin_unlock_bh(&mvmsta->lock);
2203 IWL_ERR(mvm, "Entered D0i3 while starting Tx agg\n");
2204 return -EIO;
2205 }
2206
9f9af3d7 2207 spin_lock(&mvm->queue_info_lock);
4ecafae9 2208
cf961e16
LK		 2209 /*
2210 * Note the possible cases:
2211 * 1. In DQA mode with an enabled TXQ - TXQ needs to become agg'ed
2212 * 2. Non-DQA mode: the TXQ hasn't yet been enabled, so find a free
2213 * one and mark it as reserved
2214 * 3. In DQA mode, but no traffic yet on this TID: same treatment as in
2215 * non-DQA mode, since the TXQ hasn't yet been allocated
2216 */
2217 txq_id = mvmsta->tid_data[tid].txq_id;
9f9af3d7
LK		 2218 if (iwl_mvm_is_dqa_supported(mvm) &&
2219 unlikely(mvm->queue_info[txq_id].status == IWL_MVM_QUEUE_SHARED)) {
2220 ret = -ENXIO;
2221 IWL_DEBUG_TX_QUEUES(mvm,
2222 "Can't start tid %d agg on shared queue!\n",
2223 tid);
2224 goto release_locks;
2225 } else if (!iwl_mvm_is_dqa_supported(mvm) ||
cf961e16 2226 mvm->queue_info[txq_id].status != IWL_MVM_QUEUE_READY) {
9794c64f
LK		 2227 txq_id = iwl_mvm_find_free_queue(mvm, mvmsta->sta_id,
2228 mvm->first_agg_queue,
cf961e16
LK		 2229 mvm->last_agg_queue);
2230 if (txq_id < 0) {
2231 ret = txq_id;
cf961e16
LK		 2232 IWL_ERR(mvm, "Failed to allocate agg queue\n");
2233 goto release_locks;
2234 }
2235
2236 /* TXQ hasn't yet been enabled, so mark it only as reserved */
2237 mvm->queue_info[txq_id].status = IWL_MVM_QUEUE_RESERVED;
4ecafae9 2238 }
9f9af3d7
LK		 2239
2240 spin_unlock(&mvm->queue_info_lock);
8ca151b5 2241
cf961e16
LK		 2242 IWL_DEBUG_TX_QUEUES(mvm,
2243 "AGG for tid %d will be on queue #%d\n",
2244 tid, txq_id);
2245
8ca151b5 2246 tid_data = &mvmsta->tid_data[tid];
9a886586 2247 tid_data->ssn = IEEE80211_SEQ_TO_SN(tid_data->seq_number);
8ca151b5
JB		 2248 tid_data->txq_id = txq_id;
2249 *ssn = tid_data->ssn;
2250
2251 IWL_DEBUG_TX_QUEUES(mvm,
2252 "Start AGG: sta %d tid %d queue %d - ssn = %d, next_recl = %d\n",
2253 mvmsta->sta_id, tid, txq_id, tid_data->ssn,
2254 tid_data->next_reclaimed);
2255
2256 if (tid_data->ssn == tid_data->next_reclaimed) {
2257 tid_data->state = IWL_AGG_STARTING;
2258 ieee80211_start_tx_ba_cb_irqsafe(vif, sta->addr, tid);
2259 } else {
2260 tid_data->state = IWL_EMPTYING_HW_QUEUE_ADDBA;
2261 }
2262
4ecafae9 2263 ret = 0;
9f9af3d7 2264 goto out;
4ecafae9
LK		 2265
2266release_locks:
9f9af3d7
LK		 2267 spin_unlock(&mvm->queue_info_lock);
2268out:
8ca151b5
JB		 2269 spin_unlock_bh(&mvmsta->lock);
2270
4ecafae9 2271 return ret;
8ca151b5
JB		 2272}
2273
2274int iwl_mvm_sta_tx_agg_oper(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
bb81bb68
EG		 2275 struct ieee80211_sta *sta, u16 tid, u8 buf_size,
2276 bool amsdu)
8ca151b5 2277{
5b577a90 2278 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
8ca151b5 2279 struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
5d42e7b2
EG		 2280 unsigned int wdg_timeout =
2281 iwl_mvm_get_wd_timeout(mvm, vif, sta->tdls, false);
eea76c36 2282 int queue, ret;
cf961e16 2283 bool alloc_queue = true;
9f9af3d7 2284 enum iwl_mvm_queue_status queue_status;
8ca151b5
JB		 2285 u16 ssn;
2286
eea76c36
EG		 2287 struct iwl_trans_txq_scd_cfg cfg = {
2288 .sta_id = mvmsta->sta_id,
2289 .tid = tid,
2290 .frame_limit = buf_size,
2291 .aggregate = true,
2292 };
2293
efed6640
ES		 2294 BUILD_BUG_ON((sizeof(mvmsta->agg_tids) * BITS_PER_BYTE)
2295 != IWL_MAX_TID_COUNT);
2296
8ca151b5
JB		 2297 buf_size = min_t(int, buf_size, LINK_QUAL_AGG_FRAME_LIMIT_DEF);
2298
2299 spin_lock_bh(&mvmsta->lock);
2300 ssn = tid_data->ssn;
2301 queue = tid_data->txq_id;
2302 tid_data->state = IWL_AGG_ON;
efed6640 2303 mvmsta->agg_tids |= BIT(tid);
8ca151b5 2304 tid_data->ssn = 0xffff;
bb81bb68 2305 tid_data->amsdu_in_ampdu_allowed = amsdu;
8ca151b5
JB		 2306 spin_unlock_bh(&mvmsta->lock);
2307
eea76c36 2308 cfg.fifo = iwl_mvm_ac_to_tx_fifo[tid_to_mac80211_ac[tid]];
8ca151b5 2309
9f9af3d7
LK		 2310 spin_lock_bh(&mvm->queue_info_lock);
2311 queue_status = mvm->queue_info[queue].status;
2312 spin_unlock_bh(&mvm->queue_info_lock);
2313
cf961e16
LK		 2314 /* In DQA mode, the existing queue might need to be reconfigured */
2315 if (iwl_mvm_is_dqa_supported(mvm)) {
cf961e16
LK		 2316 /* Maybe there is no need to even alloc a queue... */
2317 if (mvm->queue_info[queue].status == IWL_MVM_QUEUE_READY)
2318 alloc_queue = false;
cf961e16
LK		 2319
2320 /*
2321 * Only reconfig the SCD for the queue if the window size has
2322 * changed from current (become smaller)
2323 */
2324 if (!alloc_queue && buf_size < mvmsta->max_agg_bufsize) {
2325 /*
2326 * If reconfiguring an existing queue, it first must be
2327 * drained
2328 */
2329 ret = iwl_trans_wait_tx_queue_empty(mvm->trans,
2330 BIT(queue));
2331 if (ret) {
2332 IWL_ERR(mvm,
2333 "Error draining queue before reconfig\n");
2334 return ret;
2335 }
2336
2337 ret = iwl_mvm_reconfig_scd(mvm, queue, cfg.fifo,
2338 mvmsta->sta_id, tid,
2339 buf_size, ssn);
2340 if (ret) {
2341 IWL_ERR(mvm,
2342 "Error reconfiguring TXQ #%d\n", queue);
2343 return ret;
2344 }
2345 }
2346 }
2347
2348 if (alloc_queue)
2349 iwl_mvm_enable_txq(mvm, queue,
2350 vif->hw_queue[tid_to_mac80211_ac[tid]], ssn,
2351 &cfg, wdg_timeout);
fa7878e7 2352
9f9af3d7
LK		 2353 /* Send ADD_STA command to enable aggs only if the queue isn't shared */
2354 if (queue_status != IWL_MVM_QUEUE_SHARED) {
2355 ret = iwl_mvm_sta_tx_agg(mvm, sta, tid, queue, true);
2356 if (ret)
2357 return -EIO;
2358 }
8ca151b5 2359
4ecafae9
LK		 2360 /* No need to mark as reserved */
2361 spin_lock_bh(&mvm->queue_info_lock);
cf961e16 2362 mvm->queue_info[queue].status = IWL_MVM_QUEUE_READY;
4ecafae9
LK		 2363 spin_unlock_bh(&mvm->queue_info_lock);
2364
8ca151b5
JB		 2365 /*
2366 * Even though in theory the peer could have different
2367 * aggregation reorder buffer sizes for different sessions,
2368 * our ucode doesn't allow for that and has a global limit
2369 * for each station. Therefore, use the minimum of all the
2370 * aggregation sessions and our default value.
2371 */
2372 mvmsta->max_agg_bufsize =
2373 min(mvmsta->max_agg_bufsize, buf_size);
2374 mvmsta->lq_sta.lq.agg_frame_cnt_limit = mvmsta->max_agg_bufsize;
2375
9ee718aa
EL		 2376 IWL_DEBUG_HT(mvm, "Tx aggregation enabled on ra = %pM tid = %d\n",
2377 sta->addr, tid);
2378
9e680946 2379 return iwl_mvm_send_lq_cmd(mvm, &mvmsta->lq_sta.lq, false);
8ca151b5
JB		 2380}
2381
2382int iwl_mvm_sta_tx_agg_stop(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
2383 struct ieee80211_sta *sta, u16 tid)
2384{
5b577a90 2385 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
8ca151b5
JB		 2386 struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
2387 u16 txq_id;
2388 int err;
2389
f9aa8dd3
EG		 2390 /*
2391 * If mac80211 is cleaning its state, then say that we finished since
2392 * our state has been cleared anyway.
2393 */
2394 if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
2395 ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
2396 return 0;
2397 }
2398
8ca151b5
JB		 2399 spin_lock_bh(&mvmsta->lock);
2400
2401 txq_id = tid_data->txq_id;
2402
2403 IWL_DEBUG_TX_QUEUES(mvm, "Stop AGG: sta %d tid %d q %d state %d\n",
2404 mvmsta->sta_id, tid, txq_id, tid_data->state);
2405
efed6640
ES		 2406 mvmsta->agg_tids &= ~BIT(tid);
2407
4ecafae9 2408 spin_lock_bh(&mvm->queue_info_lock);
cf961e16
LK		 2409 /*
2410 * The TXQ is marked as reserved only if no traffic came through yet
2411 * This means no traffic has been sent on this TID (agg'd or not), so
2412 * we no longer have use for the queue. Since it hasn't even been
2413 * allocated through iwl_mvm_enable_txq, so we can just mark it back as
2414 * free.
2415 */
2416 if (mvm->queue_info[txq_id].status == IWL_MVM_QUEUE_RESERVED)
2417 mvm->queue_info[txq_id].status = IWL_MVM_QUEUE_FREE;
9f9af3d7 2418
4ecafae9
LK		 2419 spin_unlock_bh(&mvm->queue_info_lock);
2420
8ca151b5
JB		 2421 switch (tid_data->state) {
2422 case IWL_AGG_ON:
9a886586 2423 tid_data->ssn = IEEE80211_SEQ_TO_SN(tid_data->seq_number);
8ca151b5
JB		 2424
2425 IWL_DEBUG_TX_QUEUES(mvm,
2426 "ssn = %d, next_recl = %d\n",
2427 tid_data->ssn, tid_data->next_reclaimed);
2428
2429 /* There are still packets for this RA / TID in the HW */
2430 if (tid_data->ssn != tid_data->next_reclaimed) {
2431 tid_data->state = IWL_EMPTYING_HW_QUEUE_DELBA;
2432 err = 0;
2433 break;
2434 }
2435
2436 tid_data->ssn = 0xffff;
f7f89e7b 2437 tid_data->state = IWL_AGG_OFF;
f7f89e7b
JB		 2438 spin_unlock_bh(&mvmsta->lock);
2439
2440 ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
2441
2442 iwl_mvm_sta_tx_agg(mvm, sta, tid, txq_id, false);
2443
cf961e16
LK		 2444 if (!iwl_mvm_is_dqa_supported(mvm)) {
2445 int mac_queue = vif->hw_queue[tid_to_mac80211_ac[tid]];
2446
2447 iwl_mvm_disable_txq(mvm, txq_id, mac_queue, tid, 0);
2448 }
f7f89e7b 2449 return 0;
8ca151b5
JB		 2450 case IWL_AGG_STARTING:
2451 case IWL_EMPTYING_HW_QUEUE_ADDBA:
2452 /*
2453 * The agg session has been stopped before it was set up. This
2454 * can happen when the AddBA timer times out for example.
2455 */
2456
2457 /* No barriers since we are under mutex */
2458 lockdep_assert_held(&mvm->mutex);
8ca151b5
JB		 2459
2460 ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
2461 tid_data->state = IWL_AGG_OFF;
2462 err = 0;
2463 break;
2464 default:
2465 IWL_ERR(mvm,
2466 "Stopping AGG while state not ON or starting for %d on %d (%d)\n",
2467 mvmsta->sta_id, tid, tid_data->state);
2468 IWL_ERR(mvm,
2469 "\ttid_data->txq_id = %d\n", tid_data->txq_id);
2470 err = -EINVAL;
2471 }
2472
2473 spin_unlock_bh(&mvmsta->lock);
2474
2475 return err;
2476}
2477
e3d9e7ce
EG		 2478int iwl_mvm_sta_tx_agg_flush(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
2479 struct ieee80211_sta *sta, u16 tid)
2480{
5b577a90 2481 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
e3d9e7ce
EG		 2482 struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
2483 u16 txq_id;
b6658ff8 2484 enum iwl_mvm_agg_state old_state;
e3d9e7ce
EG		 2485
2486 /*
2487 * First set the agg state to OFF to avoid calling
2488 * ieee80211_stop_tx_ba_cb in iwl_mvm_check_ratid_empty.
2489 */
2490 spin_lock_bh(&mvmsta->lock);
2491 txq_id = tid_data->txq_id;
2492 IWL_DEBUG_TX_QUEUES(mvm, "Flush AGG: sta %d tid %d q %d state %d\n",
2493 mvmsta->sta_id, tid, txq_id, tid_data->state);
b6658ff8 2494 old_state = tid_data->state;
e3d9e7ce 2495 tid_data->state = IWL_AGG_OFF;
efed6640 2496 mvmsta->agg_tids &= ~BIT(tid);
e3d9e7ce
EG		 2497 spin_unlock_bh(&mvmsta->lock);
2498
4ecafae9 2499 spin_lock_bh(&mvm->queue_info_lock);
cf961e16
LK		 2500 /*
2501 * The TXQ is marked as reserved only if no traffic came through yet
2502 * This means no traffic has been sent on this TID (agg'd or not), so
2503 * we no longer have use for the queue. Since it hasn't even been
2504 * allocated through iwl_mvm_enable_txq, so we can just mark it back as
2505 * free.
2506 */
2507 if (mvm->queue_info[txq_id].status == IWL_MVM_QUEUE_RESERVED)
2508 mvm->queue_info[txq_id].status = IWL_MVM_QUEUE_FREE;
4ecafae9
LK		 2509 spin_unlock_bh(&mvm->queue_info_lock);
2510
b6658ff8 2511 if (old_state >= IWL_AGG_ON) {
fe92e32a 2512 iwl_mvm_drain_sta(mvm, mvmsta, true);
5888a40c 2513 if (iwl_mvm_flush_tx_path(mvm, BIT(txq_id), 0))
b6658ff8 2514 IWL_ERR(mvm, "Couldn't flush the AGG queue\n");
fe92e32a
EG		 2515 iwl_trans_wait_tx_queue_empty(mvm->trans,
2516 mvmsta->tfd_queue_msk);
2517 iwl_mvm_drain_sta(mvm, mvmsta, false);
b6658ff8 2518
f7f89e7b
JB		 2519 iwl_mvm_sta_tx_agg(mvm, sta, tid, txq_id, false);
2520
cf961e16
LK		 2521 if (!iwl_mvm_is_dqa_supported(mvm)) {
2522 int mac_queue = vif->hw_queue[tid_to_mac80211_ac[tid]];
2523
2524 iwl_mvm_disable_txq(mvm, tid_data->txq_id, mac_queue,
2525 tid, 0);
2526 }
b6658ff8 2527 }
e3d9e7ce 2528
e3d9e7ce
EG		 2529 return 0;
2530}
2531
8ca151b5
JB		 2532static int iwl_mvm_set_fw_key_idx(struct iwl_mvm *mvm)
2533{
2dc2a15e 2534 int i, max = -1, max_offs = -1;
8ca151b5
JB		 2535
2536 lockdep_assert_held(&mvm->mutex);
2537
2dc2a15e
JB		 2538 /* Pick the unused key offset with the highest 'deleted'
2539 * counter. Every time a key is deleted, all the counters
2540 * are incremented and the one that was just deleted is
2541 * reset to zero. Thus, the highest counter is the one
2542 * that was deleted longest ago. Pick that one.
2543 */
2544 for (i = 0; i < STA_KEY_MAX_NUM; i++) {
2545 if (test_bit(i, mvm->fw_key_table))
2546 continue;
2547 if (mvm->fw_key_deleted[i] > max) {
2548 max = mvm->fw_key_deleted[i];
2549 max_offs = i;
2550 }
2551 }
8ca151b5 2552
2dc2a15e 2553 if (max_offs < 0)
8ca151b5
JB		 2554 return STA_KEY_IDX_INVALID;
2555
2dc2a15e 2556 return max_offs;
8ca151b5
JB		 2557}
2558
5f7a1847
JB		 2559static struct iwl_mvm_sta *iwl_mvm_get_key_sta(struct iwl_mvm *mvm,
2560 struct ieee80211_vif *vif,
2561 struct ieee80211_sta *sta)
8ca151b5 2562{
5b530e95 2563 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
8ca151b5 2564
5f7a1847
JB		 2565 if (sta)
2566 return iwl_mvm_sta_from_mac80211(sta);
8ca151b5
JB		 2567
2568 /*
2569 * The device expects GTKs for station interfaces to be
2570 * installed as GTKs for the AP station. If we have no
2571 * station ID, then use AP's station ID.
2572 */
2573 if (vif->type == NL80211_IFTYPE_STATION &&
9513c5e1
AA		 2574 mvmvif->ap_sta_id != IWL_MVM_STATION_COUNT) {
2575 u8 sta_id = mvmvif->ap_sta_id;
2576
7d6a1ab6
EG		 2577 sta = rcu_dereference_check(mvm->fw_id_to_mac_id[sta_id],
2578 lockdep_is_held(&mvm->mutex));
2579
9513c5e1
AA		 2580 /*
2581 * It is possible that the 'sta' parameter is NULL,
2582 * for example when a GTK is removed - the sta_id will then
2583 * be the AP ID, and no station was passed by mac80211.
2584 */
7d6a1ab6
EG		 2585 if (IS_ERR_OR_NULL(sta))
2586 return NULL;
2587
2588 return iwl_mvm_sta_from_mac80211(sta);
9513c5e1 2589 }
8ca151b5 2590
5f7a1847 2591 return NULL;
8ca151b5
JB		 2592}
2593
2594static int iwl_mvm_send_sta_key(struct iwl_mvm *mvm,
2595 struct iwl_mvm_sta *mvm_sta,
ba3943b0 2596 struct ieee80211_key_conf *keyconf, bool mcast,
d6ee54a9
LC		 2597 u32 tkip_iv32, u16 *tkip_p1k, u32 cmd_flags,
2598 u8 key_offset)
8ca151b5 2599{
5a258aae 2600 struct iwl_mvm_add_sta_key_cmd cmd = {};
f9dc0004 2601 __le16 key_flags;
79920749
JB		 2602 int ret;
2603 u32 status;
8ca151b5
JB		 2604 u16 keyidx;
2605 int i;
2f6319d1 2606 u8 sta_id = mvm_sta->sta_id;
8ca151b5
JB		 2607
2608 keyidx = (keyconf->keyidx << STA_KEY_FLG_KEYID_POS) &
2609 STA_KEY_FLG_KEYID_MSK;
2610 key_flags = cpu_to_le16(keyidx);
2611 key_flags |= cpu_to_le16(STA_KEY_FLG_WEP_KEY_MAP);
2612
2613 switch (keyconf->cipher) {
2614 case WLAN_CIPHER_SUITE_TKIP:
2615 key_flags |= cpu_to_le16(STA_KEY_FLG_TKIP);
5a258aae 2616 cmd.tkip_rx_tsc_byte2 = tkip_iv32;
8ca151b5 2617 for (i = 0; i < 5; i++)
5a258aae
MS		 2618 cmd.tkip_rx_ttak[i] = cpu_to_le16(tkip_p1k[i]);
2619 memcpy(cmd.key, keyconf->key, keyconf->keylen);
8ca151b5
JB		 2620 break;
2621 case WLAN_CIPHER_SUITE_CCMP:
2622 key_flags |= cpu_to_le16(STA_KEY_FLG_CCM);
5a258aae 2623 memcpy(cmd.key, keyconf->key, keyconf->keylen);
8ca151b5 2624 break;
ba3943b0
JB		 2625 case WLAN_CIPHER_SUITE_WEP104:
2626 key_flags |= cpu_to_le16(STA_KEY_FLG_WEP_13BYTES);
aa0cb08b 2627 /* fall through */
ba3943b0
JB		 2628 case WLAN_CIPHER_SUITE_WEP40:
2629 key_flags |= cpu_to_le16(STA_KEY_FLG_WEP);
2630 memcpy(cmd.key + 3, keyconf->key, keyconf->keylen);
2631 break;
2a53d166
AB		 2632 case WLAN_CIPHER_SUITE_GCMP_256:
2633 key_flags |= cpu_to_le16(STA_KEY_FLG_KEY_32BYTES);
2634 /* fall through */
2635 case WLAN_CIPHER_SUITE_GCMP:
2636 key_flags |= cpu_to_le16(STA_KEY_FLG_GCMP);
2637 memcpy(cmd.key, keyconf->key, keyconf->keylen);
2638 break;
8ca151b5 2639 default:
e36e5433
MS		 2640 key_flags |= cpu_to_le16(STA_KEY_FLG_EXT);
2641 memcpy(cmd.key, keyconf->key, keyconf->keylen);
8ca151b5
JB		 2642 }
2643
ba3943b0 2644 if (mcast)
8ca151b5
JB		 2645 key_flags |= cpu_to_le16(STA_KEY_MULTICAST);
2646
d6ee54a9 2647 cmd.key_offset = key_offset;
5a258aae 2648 cmd.key_flags = key_flags;
8ca151b5
JB		 2649 cmd.sta_id = sta_id;
2650
2651 status = ADD_STA_SUCCESS;
a1022927 2652 if (cmd_flags & CMD_ASYNC)
f9dc0004
EG		 2653 ret = iwl_mvm_send_cmd_pdu(mvm, ADD_STA_KEY, CMD_ASYNC,
2654 sizeof(cmd), &cmd);
a1022927
EG		 2655 else
2656 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA_KEY, sizeof(cmd),
2657 &cmd, &status);
8ca151b5
JB		 2658
2659 switch (status) {
2660 case ADD_STA_SUCCESS:
2661 IWL_DEBUG_WEP(mvm, "MODIFY_STA: set dynamic key passed\n");
2662 break;
2663 default:
2664 ret = -EIO;
2665 IWL_ERR(mvm, "MODIFY_STA: set dynamic key failed\n");
2666 break;
2667 }
2668
2669 return ret;
2670}
2671
2672static int iwl_mvm_send_sta_igtk(struct iwl_mvm *mvm,
2673 struct ieee80211_key_conf *keyconf,
2674 u8 sta_id, bool remove_key)
2675{
2676 struct iwl_mvm_mgmt_mcast_key_cmd igtk_cmd = {};
2677
2678 /* verify the key details match the required command's expectations */
8e160ab8
AB		 2679 if (WARN_ON((keyconf->flags & IEEE80211_KEY_FLAG_PAIRWISE) ||
2680 (keyconf->keyidx != 4 && keyconf->keyidx != 5) ||
2681 (keyconf->cipher != WLAN_CIPHER_SUITE_AES_CMAC &&
2682 keyconf->cipher != WLAN_CIPHER_SUITE_BIP_GMAC_128 &&
2683 keyconf->cipher != WLAN_CIPHER_SUITE_BIP_GMAC_256)))
2684 return -EINVAL;
2685
2686 if (WARN_ON(!iwl_mvm_has_new_rx_api(mvm) &&
2687 keyconf->cipher != WLAN_CIPHER_SUITE_AES_CMAC))
8ca151b5
JB		 2688 return -EINVAL;
2689
2690 igtk_cmd.key_id = cpu_to_le32(keyconf->keyidx);
2691 igtk_cmd.sta_id = cpu_to_le32(sta_id);
2692
2693 if (remove_key) {
2694 igtk_cmd.ctrl_flags |= cpu_to_le32(STA_KEY_NOT_VALID);
2695 } else {
2696 struct ieee80211_key_seq seq;
2697 const u8 *pn;
2698
aa950524
AB		 2699 switch (keyconf->cipher) {
2700 case WLAN_CIPHER_SUITE_AES_CMAC:
2701 igtk_cmd.ctrl_flags |= cpu_to_le32(STA_KEY_FLG_CCM);
2702 break;
8e160ab8
AB		 2703 case WLAN_CIPHER_SUITE_BIP_GMAC_128:
2704 case WLAN_CIPHER_SUITE_BIP_GMAC_256:
2705 igtk_cmd.ctrl_flags |= cpu_to_le32(STA_KEY_FLG_GCMP);
2706 break;
aa950524
AB		 2707 default:
2708 return -EINVAL;
2709 }
2710
8e160ab8
AB		 2711 memcpy(igtk_cmd.igtk, keyconf->key, keyconf->keylen);
2712 if (keyconf->cipher == WLAN_CIPHER_SUITE_BIP_GMAC_256)
2713 igtk_cmd.ctrl_flags |=
2714 cpu_to_le32(STA_KEY_FLG_KEY_32BYTES);
8ca151b5
JB		 2715 ieee80211_get_key_rx_seq(keyconf, 0, &seq);
2716 pn = seq.aes_cmac.pn;
2717 igtk_cmd.receive_seq_cnt = cpu_to_le64(((u64) pn[5] << 0) |
2718 ((u64) pn[4] << 8) |
2719 ((u64) pn[3] << 16) |
2720 ((u64) pn[2] << 24) |
2721 ((u64) pn[1] << 32) |
2722 ((u64) pn[0] << 40));
2723 }
2724
2725 IWL_DEBUG_INFO(mvm, "%s igtk for sta %u\n",
2726 remove_key ? "removing" : "installing",
2727 igtk_cmd.sta_id);
2728
8e160ab8
AB		 2729 if (!iwl_mvm_has_new_rx_api(mvm)) {
2730 struct iwl_mvm_mgmt_mcast_key_cmd_v1 igtk_cmd_v1 = {
2731 .ctrl_flags = igtk_cmd.ctrl_flags,
2732 .key_id = igtk_cmd.key_id,
2733 .sta_id = igtk_cmd.sta_id,
2734 .receive_seq_cnt = igtk_cmd.receive_seq_cnt
2735 };
2736
2737 memcpy(igtk_cmd_v1.igtk, igtk_cmd.igtk,
2738 ARRAY_SIZE(igtk_cmd_v1.igtk));
2739 return iwl_mvm_send_cmd_pdu(mvm, MGMT_MCAST_KEY, 0,
2740 sizeof(igtk_cmd_v1), &igtk_cmd_v1);
2741 }
a1022927 2742 return iwl_mvm_send_cmd_pdu(mvm, MGMT_MCAST_KEY, 0,
8ca151b5
JB		 2743 sizeof(igtk_cmd), &igtk_cmd);
2744}
2745
2746
2747static inline u8 *iwl_mvm_get_mac_addr(struct iwl_mvm *mvm,
2748 struct ieee80211_vif *vif,
2749 struct ieee80211_sta *sta)
2750{
5b530e95 2751 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
8ca151b5
JB		 2752
2753 if (sta)
2754 return sta->addr;
2755
2756 if (vif->type == NL80211_IFTYPE_STATION &&
2757 mvmvif->ap_sta_id != IWL_MVM_STATION_COUNT) {
2758 u8 sta_id = mvmvif->ap_sta_id;
2759 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
2760 lockdep_is_held(&mvm->mutex));
2761 return sta->addr;
2762 }
2763
2764
2765 return NULL;
2766}
2767
2f6319d1
JB		 2768static int __iwl_mvm_set_sta_key(struct iwl_mvm *mvm,
2769 struct ieee80211_vif *vif,
2770 struct ieee80211_sta *sta,
ba3943b0 2771 struct ieee80211_key_conf *keyconf,
d6ee54a9 2772 u8 key_offset,
ba3943b0 2773 bool mcast)
2f6319d1
JB		 2774{
2775 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
2776 int ret;
2777 const u8 *addr;
2778 struct ieee80211_key_seq seq;
2779 u16 p1k[5];
2780
2781 switch (keyconf->cipher) {
2782 case WLAN_CIPHER_SUITE_TKIP:
2783 addr = iwl_mvm_get_mac_addr(mvm, vif, sta);
2784 /* get phase 1 key from mac80211 */
2785 ieee80211_get_key_rx_seq(keyconf, 0, &seq);
2786 ieee80211_get_tkip_rx_p1k(keyconf, addr, seq.tkip.iv32, p1k);
ba3943b0 2787 ret = iwl_mvm_send_sta_key(mvm, mvm_sta, keyconf, mcast,
d6ee54a9 2788 seq.tkip.iv32, p1k, 0, key_offset);
2f6319d1
JB		 2789 break;
2790 case WLAN_CIPHER_SUITE_CCMP:
ba3943b0
JB		 2791 case WLAN_CIPHER_SUITE_WEP40:
2792 case WLAN_CIPHER_SUITE_WEP104:
2a53d166
AB		 2793 case WLAN_CIPHER_SUITE_GCMP:
2794 case WLAN_CIPHER_SUITE_GCMP_256:
ba3943b0 2795 ret = iwl_mvm_send_sta_key(mvm, mvm_sta, keyconf, mcast,
d6ee54a9 2796 0, NULL, 0, key_offset);
2f6319d1
JB		 2797 break;
2798 default:
ba3943b0 2799 ret = iwl_mvm_send_sta_key(mvm, mvm_sta, keyconf, mcast,
d6ee54a9 2800 0, NULL, 0, key_offset);
2f6319d1
JB		 2801 }
2802
2803 return ret;
2804}
2805
2806static int __iwl_mvm_remove_sta_key(struct iwl_mvm *mvm, u8 sta_id,
ba3943b0
JB		 2807 struct ieee80211_key_conf *keyconf,
2808 bool mcast)
2f6319d1
JB		 2809{
2810 struct iwl_mvm_add_sta_key_cmd cmd = {};
2811 __le16 key_flags;
2812 int ret;
2813 u32 status;
2814
2815 key_flags = cpu_to_le16((keyconf->keyidx << STA_KEY_FLG_KEYID_POS) &
2816 STA_KEY_FLG_KEYID_MSK);
2817 key_flags |= cpu_to_le16(STA_KEY_FLG_NO_ENC | STA_KEY_FLG_WEP_KEY_MAP);
2818 key_flags |= cpu_to_le16(STA_KEY_NOT_VALID);
2819
ba3943b0 2820 if (mcast)
2f6319d1
JB		 2821 key_flags |= cpu_to_le16(STA_KEY_MULTICAST);
2822
2823 cmd.key_flags = key_flags;
2824 cmd.key_offset = keyconf->hw_key_idx;
2825 cmd.sta_id = sta_id;
2826
2827 status = ADD_STA_SUCCESS;
2828 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA_KEY, sizeof(cmd),
2829 &cmd, &status);
2830
2831 switch (status) {
2832 case ADD_STA_SUCCESS:
2833 IWL_DEBUG_WEP(mvm, "MODIFY_STA: remove sta key passed\n");
2834 break;
2835 default:
2836 ret = -EIO;
2837 IWL_ERR(mvm, "MODIFY_STA: remove sta key failed\n");
2838 break;
2839 }
2840
2841 return ret;
2842}
2843
8ca151b5
JB		 2844int iwl_mvm_set_sta_key(struct iwl_mvm *mvm,
2845 struct ieee80211_vif *vif,
2846 struct ieee80211_sta *sta,
2847 struct ieee80211_key_conf *keyconf,
d6ee54a9 2848 u8 key_offset)
8ca151b5 2849{
ba3943b0 2850 bool mcast = !(keyconf->flags & IEEE80211_KEY_FLAG_PAIRWISE);
5f7a1847 2851 struct iwl_mvm_sta *mvm_sta;
2f6319d1 2852 u8 sta_id;
8ca151b5 2853 int ret;
11828dbc 2854 static const u8 __maybe_unused zero_addr[ETH_ALEN] = {0};
8ca151b5
JB		 2855
2856 lockdep_assert_held(&mvm->mutex);
2857
2858 /* Get the station id from the mvm local station table */
5f7a1847
JB		 2859 mvm_sta = iwl_mvm_get_key_sta(mvm, vif, sta);
2860 if (!mvm_sta) {
2861 IWL_ERR(mvm, "Failed to find station\n");
8ca151b5
JB		 2862 return -EINVAL;
2863 }
5f7a1847 2864 sta_id = mvm_sta->sta_id;
8ca151b5 2865
8e160ab8
AB		 2866 if (keyconf->cipher == WLAN_CIPHER_SUITE_AES_CMAC ||
2867 keyconf->cipher == WLAN_CIPHER_SUITE_BIP_GMAC_128 ||
2868 keyconf->cipher == WLAN_CIPHER_SUITE_BIP_GMAC_256) {
8ca151b5
JB		 2869 ret = iwl_mvm_send_sta_igtk(mvm, keyconf, sta_id, false);
2870 goto end;
2871 }
2872
2873 /*
2874 * It is possible that the 'sta' parameter is NULL, and thus
2875 * there is a need to retrieve the sta from the local station table.
2876 */
2877 if (!sta) {
2878 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
2879 lockdep_is_held(&mvm->mutex));
2880 if (IS_ERR_OR_NULL(sta)) {
2881 IWL_ERR(mvm, "Invalid station id\n");
2882 return -EINVAL;
2883 }
2884 }
2885
2f6319d1 2886 if (WARN_ON_ONCE(iwl_mvm_sta_from_mac80211(sta)->vif != vif))
8ca151b5
JB		 2887 return -EINVAL;
2888
d6ee54a9
LC		 2889 /* If the key_offset is not pre-assigned, we need to find a
2890 * new offset to use. In normal cases, the offset is not
2891 * pre-assigned, but during HW_RESTART we want to reuse the
2892 * same indices, so we pass them when this function is called.
2893 *
2894 * In D3 entry, we need to hardcoded the indices (because the
2895 * firmware hardcodes the PTK offset to 0). In this case, we
2896 * need to make sure we don't overwrite the hw_key_idx in the
2897 * keyconf structure, because otherwise we cannot configure
2898 * the original ones back when resuming.
2899 */
2900 if (key_offset == STA_KEY_IDX_INVALID) {
2901 key_offset = iwl_mvm_set_fw_key_idx(mvm);
2902 if (key_offset == STA_KEY_IDX_INVALID)
8ca151b5 2903 return -ENOSPC;
d6ee54a9 2904 keyconf->hw_key_idx = key_offset;
8ca151b5
JB		 2905 }
2906
d6ee54a9 2907 ret = __iwl_mvm_set_sta_key(mvm, vif, sta, keyconf, key_offset, mcast);
9c3deeb5 2908 if (ret)
ba3943b0 2909 goto end;
ba3943b0
JB		 2910
2911 /*
2912 * For WEP, the same key is used for multicast and unicast. Upload it
2913 * again, using the same key offset, and now pointing the other one
2914 * to the same key slot (offset).
2915 * If this fails, remove the original as well.
2916 */
2917 if (keyconf->cipher == WLAN_CIPHER_SUITE_WEP40 ||
2918 keyconf->cipher == WLAN_CIPHER_SUITE_WEP104) {
d6ee54a9
LC		 2919 ret = __iwl_mvm_set_sta_key(mvm, vif, sta, keyconf,
2920 key_offset, !mcast);
ba3943b0 2921 if (ret) {
ba3943b0 2922 __iwl_mvm_remove_sta_key(mvm, sta_id, keyconf, mcast);
9c3deeb5 2923 goto end;
ba3943b0
JB		 2924 }
2925 }
8ca151b5 2926
9c3deeb5
LC		 2927 __set_bit(key_offset, mvm->fw_key_table);
2928
8ca151b5
JB		 2929end:
2930 IWL_DEBUG_WEP(mvm, "key: cipher=%x len=%d idx=%d sta=%pM ret=%d\n",
2931 keyconf->cipher, keyconf->keylen, keyconf->keyidx,
11828dbc 2932 sta ? sta->addr : zero_addr, ret);
8ca151b5
JB		 2933 return ret;
2934}
2935
2936int iwl_mvm_remove_sta_key(struct iwl_mvm *mvm,
2937 struct ieee80211_vif *vif,
2938 struct ieee80211_sta *sta,
2939 struct ieee80211_key_conf *keyconf)
2940{
ba3943b0 2941 bool mcast = !(keyconf->flags & IEEE80211_KEY_FLAG_PAIRWISE);
5f7a1847
JB		 2942 struct iwl_mvm_sta *mvm_sta;
2943 u8 sta_id = IWL_MVM_STATION_COUNT;
2dc2a15e 2944 int ret, i;
8ca151b5
JB		 2945
2946 lockdep_assert_held(&mvm->mutex);
2947
5f7a1847
JB		 2948 /* Get the station from the mvm local station table */
2949 mvm_sta = iwl_mvm_get_key_sta(mvm, vif, sta);
8ca151b5
JB		 2950
2951 IWL_DEBUG_WEP(mvm, "mvm remove dynamic key: idx=%d sta=%d\n",
2952 keyconf->keyidx, sta_id);
2953
8e160ab8
AB		 2954 if (keyconf->cipher == WLAN_CIPHER_SUITE_AES_CMAC ||
2955 keyconf->cipher == WLAN_CIPHER_SUITE_BIP_GMAC_128 ||
2956 keyconf->cipher == WLAN_CIPHER_SUITE_BIP_GMAC_256)
8ca151b5
JB		 2957 return iwl_mvm_send_sta_igtk(mvm, keyconf, sta_id, true);
2958
2f6319d1 2959 if (!__test_and_clear_bit(keyconf->hw_key_idx, mvm->fw_key_table)) {
8ca151b5
JB		 2960 IWL_ERR(mvm, "offset %d not used in fw key table.\n",
2961 keyconf->hw_key_idx);
2962 return -ENOENT;
2963 }
2964
2dc2a15e
JB		 2965 /* track which key was deleted last */
2966 for (i = 0; i < STA_KEY_MAX_NUM; i++) {
2967 if (mvm->fw_key_deleted[i] < U8_MAX)
2968 mvm->fw_key_deleted[i]++;
2969 }
2970 mvm->fw_key_deleted[keyconf->hw_key_idx] = 0;
2971
5f7a1847 2972 if (!mvm_sta) {
8ca151b5
JB		 2973 IWL_DEBUG_WEP(mvm, "station non-existent, early return.\n");
2974 return 0;
2975 }
2976
5f7a1847
JB		 2977 sta_id = mvm_sta->sta_id;
2978
ba3943b0
JB		 2979 ret = __iwl_mvm_remove_sta_key(mvm, sta_id, keyconf, mcast);
2980 if (ret)
2981 return ret;
2982
2983 /* delete WEP key twice to get rid of (now useless) offset */
2984 if (keyconf->cipher == WLAN_CIPHER_SUITE_WEP40 ||
2985 keyconf->cipher == WLAN_CIPHER_SUITE_WEP104)
2986 ret = __iwl_mvm_remove_sta_key(mvm, sta_id, keyconf, !mcast);
2987
2988 return ret;
8ca151b5
JB		 2989}
2990
2991void iwl_mvm_update_tkip_key(struct iwl_mvm *mvm,
2992 struct ieee80211_vif *vif,
2993 struct ieee80211_key_conf *keyconf,
2994 struct ieee80211_sta *sta, u32 iv32,
2995 u16 *phase1key)
2996{
c3eb536a 2997 struct iwl_mvm_sta *mvm_sta;
ba3943b0 2998 bool mcast = !(keyconf->flags & IEEE80211_KEY_FLAG_PAIRWISE);
8ca151b5 2999
c3eb536a
BL		 3000 rcu_read_lock();
3001
5f7a1847
JB		 3002 mvm_sta = iwl_mvm_get_key_sta(mvm, vif, sta);
3003 if (WARN_ON_ONCE(!mvm_sta))
45854360 3004 goto unlock;
ba3943b0 3005 iwl_mvm_send_sta_key(mvm, mvm_sta, keyconf, mcast,
d6ee54a9 3006 iv32, phase1key, CMD_ASYNC, keyconf->hw_key_idx);
45854360
JB		 3007
3008 unlock:
c3eb536a 3009 rcu_read_unlock();
8ca151b5
JB		 3010}
3011
9cc40712
JB		 3012void iwl_mvm_sta_modify_ps_wake(struct iwl_mvm *mvm,
3013 struct ieee80211_sta *sta)
8ca151b5 3014{
5b577a90 3015 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
f9dc0004 3016 struct iwl_mvm_add_sta_cmd cmd = {
8ca151b5 3017 .add_modify = STA_MODE_MODIFY,
9cc40712 3018 .sta_id = mvmsta->sta_id,
5af01772 3019 .station_flags_msk = cpu_to_le32(STA_FLG_PS),
9cc40712 3020 .mac_id_n_color = cpu_to_le32(mvmsta->mac_id_n_color),
8ca151b5
JB		 3021 };
3022 int ret;
3023
854c5705
SS		 3024 ret = iwl_mvm_send_cmd_pdu(mvm, ADD_STA, CMD_ASYNC,
3025 iwl_mvm_add_sta_cmd_size(mvm), &cmd);
8ca151b5
JB		 3026 if (ret)
3027 IWL_ERR(mvm, "Failed to send ADD_STA command (%d)\n", ret);
3028}
3029
9cc40712
JB		 3030void iwl_mvm_sta_modify_sleep_tx_count(struct iwl_mvm *mvm,
3031 struct ieee80211_sta *sta,
8ca151b5 3032 enum ieee80211_frame_release_type reason,
3e56eadf
JB		 3033 u16 cnt, u16 tids, bool more_data,
3034 bool agg)
8ca151b5 3035{
5b577a90 3036 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
f9dc0004 3037 struct iwl_mvm_add_sta_cmd cmd = {
8ca151b5 3038 .add_modify = STA_MODE_MODIFY,
9cc40712 3039 .sta_id = mvmsta->sta_id,
8ca151b5
JB		 3040 .modify_mask = STA_MODIFY_SLEEPING_STA_TX_COUNT,
3041 .sleep_tx_count = cpu_to_le16(cnt),
9cc40712 3042 .mac_id_n_color = cpu_to_le32(mvmsta->mac_id_n_color),
8ca151b5 3043 };
3e56eadf
JB		 3044 int tid, ret;
3045 unsigned long _tids = tids;
3046
3047 /* convert TIDs to ACs - we don't support TSPEC so that's OK
3048 * Note that this field is reserved and unused by firmware not
3049 * supporting GO uAPSD, so it's safe to always do this.
3050 */
3051 for_each_set_bit(tid, &_tids, IWL_MAX_TID_COUNT)
3052 cmd.awake_acs |= BIT(tid_to_ucode_ac[tid]);
3053
3054 /* If we're releasing frames from aggregation queues then check if the
3055 * all queues combined that we're releasing frames from have
3056 * - more frames than the service period, in which case more_data
3057 * needs to be set
3058 * - fewer than 'cnt' frames, in which case we need to adjust the
3059 * firmware command (but do that unconditionally)
3060 */
3061 if (agg) {
3062 int remaining = cnt;
36be0eb6 3063 int sleep_tx_count;
3e56eadf
JB		 3064
3065 spin_lock_bh(&mvmsta->lock);
3066 for_each_set_bit(tid, &_tids, IWL_MAX_TID_COUNT) {
3067 struct iwl_mvm_tid_data *tid_data;
3068 u16 n_queued;
3069
3070 tid_data = &mvmsta->tid_data[tid];
3071 if (WARN(tid_data->state != IWL_AGG_ON &&
3072 tid_data->state != IWL_EMPTYING_HW_QUEUE_DELBA,
3073 "TID %d state is %d\n",
3074 tid, tid_data->state)) {
3075 spin_unlock_bh(&mvmsta->lock);
3076 ieee80211_sta_eosp(sta);
3077 return;
3078 }
3079
3080 n_queued = iwl_mvm_tid_queued(tid_data);
3081 if (n_queued > remaining) {
3082 more_data = true;
3083 remaining = 0;
3084 break;
3085 }
3086 remaining -= n_queued;
3087 }
36be0eb6
EG		 3088 sleep_tx_count = cnt - remaining;
3089 if (reason == IEEE80211_FRAME_RELEASE_UAPSD)
3090 mvmsta->sleep_tx_count = sleep_tx_count;
3e56eadf
JB		 3091 spin_unlock_bh(&mvmsta->lock);
3092
36be0eb6 3093 cmd.sleep_tx_count = cpu_to_le16(sleep_tx_count);
3e56eadf
JB		 3094 if (WARN_ON(cnt - remaining == 0)) {
3095 ieee80211_sta_eosp(sta);
3096 return;
3097 }
3098 }
3099
3100 /* Note: this is ignored by firmware not supporting GO uAPSD */
3101 if (more_data)
3102 cmd.sleep_state_flags |= cpu_to_le16(STA_SLEEP_STATE_MOREDATA);
3103
3104 if (reason == IEEE80211_FRAME_RELEASE_PSPOLL) {
3105 mvmsta->next_status_eosp = true;
3106 cmd.sleep_state_flags |= cpu_to_le16(STA_SLEEP_STATE_PS_POLL);
3107 } else {
3108 cmd.sleep_state_flags |= cpu_to_le16(STA_SLEEP_STATE_UAPSD);
3109 }
8ca151b5 3110
156f92f2
EG		 3111 /* block the Tx queues until the FW updated the sleep Tx count */
3112 iwl_trans_block_txq_ptrs(mvm->trans, true);
3113
3114 ret = iwl_mvm_send_cmd_pdu(mvm, ADD_STA,
3115 CMD_ASYNC | CMD_WANT_ASYNC_CALLBACK,
854c5705 3116 iwl_mvm_add_sta_cmd_size(mvm), &cmd);
8ca151b5
JB		 3117 if (ret)
3118 IWL_ERR(mvm, "Failed to send ADD_STA command (%d)\n", ret);
3119}
3e56eadf 3120
0416841d
JB		 3121void iwl_mvm_rx_eosp_notif(struct iwl_mvm *mvm,
3122 struct iwl_rx_cmd_buffer *rxb)
3e56eadf
JB		 3123{
3124 struct iwl_rx_packet *pkt = rxb_addr(rxb);
3125 struct iwl_mvm_eosp_notification *notif = (void *)pkt->data;
3126 struct ieee80211_sta *sta;
3127 u32 sta_id = le32_to_cpu(notif->sta_id);
3128
3129 if (WARN_ON_ONCE(sta_id >= IWL_MVM_STATION_COUNT))
0416841d 3130 return;
3e56eadf
JB		 3131
3132 rcu_read_lock();
3133 sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
3134 if (!IS_ERR_OR_NULL(sta))
3135 ieee80211_sta_eosp(sta);
3136 rcu_read_unlock();
3e56eadf 3137}
09b0ce1a
AO		 3138
3139void iwl_mvm_sta_modify_disable_tx(struct iwl_mvm *mvm,
3140 struct iwl_mvm_sta *mvmsta, bool disable)
3141{
3142 struct iwl_mvm_add_sta_cmd cmd = {
3143 .add_modify = STA_MODE_MODIFY,
3144 .sta_id = mvmsta->sta_id,
3145 .station_flags = disable ? cpu_to_le32(STA_FLG_DISABLE_TX) : 0,
3146 .station_flags_msk = cpu_to_le32(STA_FLG_DISABLE_TX),
3147 .mac_id_n_color = cpu_to_le32(mvmsta->mac_id_n_color),
3148 };
3149 int ret;
3150
854c5705
SS		 3151 ret = iwl_mvm_send_cmd_pdu(mvm, ADD_STA, CMD_ASYNC,
3152 iwl_mvm_add_sta_cmd_size(mvm), &cmd);
09b0ce1a
AO		 3153 if (ret)
3154 IWL_ERR(mvm, "Failed to send ADD_STA command (%d)\n", ret);
3155}
003e5236
AO		 3156
3157void iwl_mvm_sta_modify_disable_tx_ap(struct iwl_mvm *mvm,
3158 struct ieee80211_sta *sta,
3159 bool disable)
3160{
3161 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
3162
3163 spin_lock_bh(&mvm_sta->lock);
3164
3165 if (mvm_sta->disable_tx == disable) {
3166 spin_unlock_bh(&mvm_sta->lock);
3167 return;
3168 }
3169
3170 mvm_sta->disable_tx = disable;
3171
3172 /*
0d365ae5
SS		 3173 * Tell mac80211 to start/stop queuing tx for this station,
3174 * but don't stop queuing if there are still pending frames
003e5236
AO		 3175 * for this station.
3176 */
3177 if (disable || !atomic_read(&mvm->pending_frames[mvm_sta->sta_id]))
3178 ieee80211_sta_block_awake(mvm->hw, sta, disable);
3179
3180 iwl_mvm_sta_modify_disable_tx(mvm, mvm_sta, disable);
3181
3182 spin_unlock_bh(&mvm_sta->lock);
3183}
3184
3185void iwl_mvm_modify_all_sta_disable_tx(struct iwl_mvm *mvm,
3186 struct iwl_mvm_vif *mvmvif,
3187 bool disable)
3188{
3189 struct ieee80211_sta *sta;
3190 struct iwl_mvm_sta *mvm_sta;
3191 int i;
3192
3193 lockdep_assert_held(&mvm->mutex);
3194
3195 /* Block/unblock all the stations of the given mvmvif */
3196 for (i = 0; i < IWL_MVM_STATION_COUNT; i++) {
3197 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[i],
3198 lockdep_is_held(&mvm->mutex));
3199 if (IS_ERR_OR_NULL(sta))
3200 continue;
3201
3202 mvm_sta = iwl_mvm_sta_from_mac80211(sta);
3203 if (mvm_sta->mac_id_n_color !=
3204 FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color))
3205 continue;
3206
3207 iwl_mvm_sta_modify_disable_tx_ap(mvm, sta, disable);
3208 }
3209}
dc88b4ba
LC		 3210
3211void iwl_mvm_csa_client_absent(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
3212{
3213 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3214 struct iwl_mvm_sta *mvmsta;
3215
3216 rcu_read_lock();
3217
3218 mvmsta = iwl_mvm_sta_from_staid_rcu(mvm, mvmvif->ap_sta_id);
3219
3220 if (!WARN_ON(!mvmsta))
3221 iwl_mvm_sta_modify_disable_tx(mvm, mvmsta, true);
3222
3223 rcu_read_unlock();
3224}
Ubuntu Eoan Linux kernel mirror