Loading...
Note: File does not exist in v3.15.
1// SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2/*
3 * Copyright (C) 2012-2015, 2018-2022 Intel Corporation
4 * Copyright (C) 2013-2015 Intel Mobile Communications GmbH
5 * Copyright (C) 2016-2017 Intel Deutschland GmbH
6 */
7#include <net/mac80211.h>
8
9#include "mvm.h"
10#include "sta.h"
11#include "rs.h"
12
13/*
14 * New version of ADD_STA_sta command added new fields at the end of the
15 * structure, so sending the size of the relevant API's structure is enough to
16 * support both API versions.
17 */
18static inline int iwl_mvm_add_sta_cmd_size(struct iwl_mvm *mvm)
19{
20 if (iwl_mvm_has_new_rx_api(mvm) ||
21 fw_has_api(&mvm->fw->ucode_capa, IWL_UCODE_TLV_API_STA_TYPE))
22 return sizeof(struct iwl_mvm_add_sta_cmd);
23 else
24 return sizeof(struct iwl_mvm_add_sta_cmd_v7);
25}
26
27static int iwl_mvm_find_free_sta_id(struct iwl_mvm *mvm,
28 enum nl80211_iftype iftype)
29{
30 int sta_id;
31 u32 reserved_ids = 0;
32
33 BUILD_BUG_ON(IWL_MVM_STATION_COUNT_MAX > 32);
34 WARN_ON_ONCE(test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status));
35
36 lockdep_assert_held(&mvm->mutex);
37
38 /* d0i3/d3 assumes the AP's sta_id (of sta vif) is 0. reserve it. */
39 if (iftype != NL80211_IFTYPE_STATION)
40 reserved_ids = BIT(0);
41
42 /* Don't take rcu_read_lock() since we are protected by mvm->mutex */
43 for (sta_id = 0; sta_id < mvm->fw->ucode_capa.num_stations; sta_id++) {
44 if (BIT(sta_id) & reserved_ids)
45 continue;
46
47 if (!rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
48 lockdep_is_held(&mvm->mutex)))
49 return sta_id;
50 }
51 return IWL_MVM_INVALID_STA;
52}
53
54/* send station add/update command to firmware */
55int iwl_mvm_sta_send_to_fw(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
56 bool update, unsigned int flags)
57{
58 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
59 struct iwl_mvm_add_sta_cmd add_sta_cmd = {
60 .sta_id = mvm_sta->sta_id,
61 .mac_id_n_color = cpu_to_le32(mvm_sta->mac_id_n_color),
62 .add_modify = update ? 1 : 0,
63 .station_flags_msk = cpu_to_le32(STA_FLG_FAT_EN_MSK |
64 STA_FLG_MIMO_EN_MSK |
65 STA_FLG_RTS_MIMO_PROT),
66 .tid_disable_tx = cpu_to_le16(mvm_sta->tid_disable_agg),
67 };
68 int ret;
69 u32 status;
70 u32 agg_size = 0, mpdu_dens = 0;
71
72 if (fw_has_api(&mvm->fw->ucode_capa, IWL_UCODE_TLV_API_STA_TYPE))
73 add_sta_cmd.station_type = mvm_sta->sta_type;
74
75 if (!update || (flags & STA_MODIFY_QUEUES)) {
76 memcpy(&add_sta_cmd.addr, sta->addr, ETH_ALEN);
77
78 if (!iwl_mvm_has_new_tx_api(mvm)) {
79 add_sta_cmd.tfd_queue_msk =
80 cpu_to_le32(mvm_sta->tfd_queue_msk);
81
82 if (flags & STA_MODIFY_QUEUES)
83 add_sta_cmd.modify_mask |= STA_MODIFY_QUEUES;
84 } else {
85 WARN_ON(flags & STA_MODIFY_QUEUES);
86 }
87 }
88
89 switch (sta->deflink.bandwidth) {
90 case IEEE80211_STA_RX_BW_320:
91 case IEEE80211_STA_RX_BW_160:
92 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_FAT_EN_160MHZ);
93 fallthrough;
94 case IEEE80211_STA_RX_BW_80:
95 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_FAT_EN_80MHZ);
96 fallthrough;
97 case IEEE80211_STA_RX_BW_40:
98 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_FAT_EN_40MHZ);
99 fallthrough;
100 case IEEE80211_STA_RX_BW_20:
101 if (sta->deflink.ht_cap.ht_supported)
102 add_sta_cmd.station_flags |=
103 cpu_to_le32(STA_FLG_FAT_EN_20MHZ);
104 break;
105 }
106
107 switch (sta->deflink.rx_nss) {
108 case 1:
109 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_MIMO_EN_SISO);
110 break;
111 case 2:
112 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_MIMO_EN_MIMO2);
113 break;
114 case 3 ... 8:
115 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_MIMO_EN_MIMO3);
116 break;
117 }
118
119 switch (sta->deflink.smps_mode) {
120 case IEEE80211_SMPS_AUTOMATIC:
121 case IEEE80211_SMPS_NUM_MODES:
122 WARN_ON(1);
123 break;
124 case IEEE80211_SMPS_STATIC:
125 /* override NSS */
126 add_sta_cmd.station_flags &= ~cpu_to_le32(STA_FLG_MIMO_EN_MSK);
127 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_MIMO_EN_SISO);
128 break;
129 case IEEE80211_SMPS_DYNAMIC:
130 add_sta_cmd.station_flags |= cpu_to_le32(STA_FLG_RTS_MIMO_PROT);
131 break;
132 case IEEE80211_SMPS_OFF:
133 /* nothing */
134 break;
135 }
136
137 if (sta->deflink.ht_cap.ht_supported) {
138 add_sta_cmd.station_flags_msk |=
139 cpu_to_le32(STA_FLG_MAX_AGG_SIZE_MSK |
140 STA_FLG_AGG_MPDU_DENS_MSK);
141
142 mpdu_dens = sta->deflink.ht_cap.ampdu_density;
143 }
144
145 if (mvm_sta->vif->bss_conf.chandef.chan->band == NL80211_BAND_6GHZ) {
146 add_sta_cmd.station_flags_msk |=
147 cpu_to_le32(STA_FLG_MAX_AGG_SIZE_MSK |
148 STA_FLG_AGG_MPDU_DENS_MSK);
149
150 mpdu_dens = le16_get_bits(sta->deflink.he_6ghz_capa.capa,
151 IEEE80211_HE_6GHZ_CAP_MIN_MPDU_START);
152 agg_size = le16_get_bits(sta->deflink.he_6ghz_capa.capa,
153 IEEE80211_HE_6GHZ_CAP_MAX_AMPDU_LEN_EXP);
154 } else if (sta->deflink.vht_cap.vht_supported) {
155 agg_size = sta->deflink.vht_cap.cap &
156 IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_MASK;
157 agg_size >>=
158 IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT;
159 } else if (sta->deflink.ht_cap.ht_supported) {
160 agg_size = sta->deflink.ht_cap.ampdu_factor;
161 }
162
163 /* D6.0 10.12.2 A-MPDU length limit rules
164 * A STA indicates the maximum length of the A-MPDU preEOF padding
165 * that it can receive in an HE PPDU in the Maximum A-MPDU Length
166 * Exponent field in its HT Capabilities, VHT Capabilities,
167 * and HE 6 GHz Band Capabilities elements (if present) and the
168 * Maximum AMPDU Length Exponent Extension field in its HE
169 * Capabilities element
170 */
171 if (sta->deflink.he_cap.has_he)
172 agg_size += u8_get_bits(sta->deflink.he_cap.he_cap_elem.mac_cap_info[3],
173 IEEE80211_HE_MAC_CAP3_MAX_AMPDU_LEN_EXP_MASK);
174
175 /* Limit to max A-MPDU supported by FW */
176 if (agg_size > (STA_FLG_MAX_AGG_SIZE_4M >> STA_FLG_MAX_AGG_SIZE_SHIFT))
177 agg_size = (STA_FLG_MAX_AGG_SIZE_4M >>
178 STA_FLG_MAX_AGG_SIZE_SHIFT);
179
180 add_sta_cmd.station_flags |=
181 cpu_to_le32(agg_size << STA_FLG_MAX_AGG_SIZE_SHIFT);
182 add_sta_cmd.station_flags |=
183 cpu_to_le32(mpdu_dens << STA_FLG_AGG_MPDU_DENS_SHIFT);
184 if (mvm_sta->sta_state >= IEEE80211_STA_ASSOC)
185 add_sta_cmd.assoc_id = cpu_to_le16(sta->aid);
186
187 if (sta->wme) {
188 add_sta_cmd.modify_mask |= STA_MODIFY_UAPSD_ACS;
189
190 if (sta->uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_BK)
191 add_sta_cmd.uapsd_acs |= BIT(AC_BK);
192 if (sta->uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_BE)
193 add_sta_cmd.uapsd_acs |= BIT(AC_BE);
194 if (sta->uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_VI)
195 add_sta_cmd.uapsd_acs |= BIT(AC_VI);
196 if (sta->uapsd_queues & IEEE80211_WMM_IE_STA_QOSINFO_AC_VO)
197 add_sta_cmd.uapsd_acs |= BIT(AC_VO);
198 add_sta_cmd.uapsd_acs |= add_sta_cmd.uapsd_acs << 4;
199 add_sta_cmd.sp_length = sta->max_sp ? sta->max_sp * 2 : 128;
200 }
201
202 status = ADD_STA_SUCCESS;
203 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA,
204 iwl_mvm_add_sta_cmd_size(mvm),
205 &add_sta_cmd, &status);
206 if (ret)
207 return ret;
208
209 switch (status & IWL_ADD_STA_STATUS_MASK) {
210 case ADD_STA_SUCCESS:
211 IWL_DEBUG_ASSOC(mvm, "ADD_STA PASSED\n");
212 break;
213 default:
214 ret = -EIO;
215 IWL_ERR(mvm, "ADD_STA failed\n");
216 break;
217 }
218
219 return ret;
220}
221
222static void iwl_mvm_rx_agg_session_expired(struct timer_list *t)
223{
224 struct iwl_mvm_baid_data *data =
225 from_timer(data, t, session_timer);
226 struct iwl_mvm_baid_data __rcu **rcu_ptr = data->rcu_ptr;
227 struct iwl_mvm_baid_data *ba_data;
228 struct ieee80211_sta *sta;
229 struct iwl_mvm_sta *mvm_sta;
230 unsigned long timeout;
231
232 rcu_read_lock();
233
234 ba_data = rcu_dereference(*rcu_ptr);
235
236 if (WARN_ON(!ba_data))
237 goto unlock;
238
239 if (!ba_data->timeout)
240 goto unlock;
241
242 timeout = ba_data->last_rx + TU_TO_JIFFIES(ba_data->timeout * 2);
243 if (time_is_after_jiffies(timeout)) {
244 mod_timer(&ba_data->session_timer, timeout);
245 goto unlock;
246 }
247
248 /* Timer expired */
249 sta = rcu_dereference(ba_data->mvm->fw_id_to_mac_id[ba_data->sta_id]);
250
251 /*
252 * sta should be valid unless the following happens:
253 * The firmware asserts which triggers a reconfig flow, but
254 * the reconfig fails before we set the pointer to sta into
255 * the fw_id_to_mac_id pointer table. Mac80211 can't stop
256 * A-MDPU and hence the timer continues to run. Then, the
257 * timer expires and sta is NULL.
258 */
259 if (!sta)
260 goto unlock;
261
262 mvm_sta = iwl_mvm_sta_from_mac80211(sta);
263 ieee80211_rx_ba_timer_expired(mvm_sta->vif,
264 sta->addr, ba_data->tid);
265unlock:
266 rcu_read_unlock();
267}
268
269/* Disable aggregations for a bitmap of TIDs for a given station */
270static int iwl_mvm_invalidate_sta_queue(struct iwl_mvm *mvm, int queue,
271 unsigned long disable_agg_tids,
272 bool remove_queue)
273{
274 struct iwl_mvm_add_sta_cmd cmd = {};
275 struct ieee80211_sta *sta;
276 struct iwl_mvm_sta *mvmsta;
277 u32 status;
278 u8 sta_id;
279
280 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
281 return -EINVAL;
282
283 sta_id = mvm->queue_info[queue].ra_sta_id;
284
285 rcu_read_lock();
286
287 sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
288
289 if (WARN_ON_ONCE(IS_ERR_OR_NULL(sta))) {
290 rcu_read_unlock();
291 return -EINVAL;
292 }
293
294 mvmsta = iwl_mvm_sta_from_mac80211(sta);
295
296 mvmsta->tid_disable_agg |= disable_agg_tids;
297
298 cmd.mac_id_n_color = cpu_to_le32(mvmsta->mac_id_n_color);
299 cmd.sta_id = mvmsta->sta_id;
300 cmd.add_modify = STA_MODE_MODIFY;
301 cmd.modify_mask = STA_MODIFY_QUEUES;
302 if (disable_agg_tids)
303 cmd.modify_mask |= STA_MODIFY_TID_DISABLE_TX;
304 if (remove_queue)
305 cmd.modify_mask |= STA_MODIFY_QUEUE_REMOVAL;
306 cmd.tfd_queue_msk = cpu_to_le32(mvmsta->tfd_queue_msk);
307 cmd.tid_disable_tx = cpu_to_le16(mvmsta->tid_disable_agg);
308
309 rcu_read_unlock();
310
311 /* Notify FW of queue removal from the STA queues */
312 status = ADD_STA_SUCCESS;
313 return iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA,
314 iwl_mvm_add_sta_cmd_size(mvm),
315 &cmd, &status);
316}
317
318static int iwl_mvm_disable_txq(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
319 int sta_id, u16 *queueptr, u8 tid)
320{
321 int queue = *queueptr;
322 struct iwl_scd_txq_cfg_cmd cmd = {
323 .scd_queue = queue,
324 .action = SCD_CFG_DISABLE_QUEUE,
325 };
326 int ret;
327
328 lockdep_assert_held(&mvm->mutex);
329
330 if (iwl_mvm_has_new_tx_api(mvm)) {
331 if (mvm->sta_remove_requires_queue_remove) {
332 u32 cmd_id = WIDE_ID(DATA_PATH_GROUP,
333 SCD_QUEUE_CONFIG_CMD);
334 struct iwl_scd_queue_cfg_cmd remove_cmd = {
335 .operation = cpu_to_le32(IWL_SCD_QUEUE_REMOVE),
336 .u.remove.tid = cpu_to_le32(tid),
337 .u.remove.sta_mask = cpu_to_le32(BIT(sta_id)),
338 };
339
340 ret = iwl_mvm_send_cmd_pdu(mvm, cmd_id, 0,
341 sizeof(remove_cmd),
342 &remove_cmd);
343 } else {
344 ret = 0;
345 }
346
347 iwl_trans_txq_free(mvm->trans, queue);
348 *queueptr = IWL_MVM_INVALID_QUEUE;
349
350 return ret;
351 }
352
353 if (WARN_ON(mvm->queue_info[queue].tid_bitmap == 0))
354 return 0;
355
356 mvm->queue_info[queue].tid_bitmap &= ~BIT(tid);
357
358 cmd.action = mvm->queue_info[queue].tid_bitmap ?
359 SCD_CFG_ENABLE_QUEUE : SCD_CFG_DISABLE_QUEUE;
360 if (cmd.action == SCD_CFG_DISABLE_QUEUE)
361 mvm->queue_info[queue].status = IWL_MVM_QUEUE_FREE;
362
363 IWL_DEBUG_TX_QUEUES(mvm,
364 "Disabling TXQ #%d tids=0x%x\n",
365 queue,
366 mvm->queue_info[queue].tid_bitmap);
367
368 /* If the queue is still enabled - nothing left to do in this func */
369 if (cmd.action == SCD_CFG_ENABLE_QUEUE)
370 return 0;
371
372 cmd.sta_id = mvm->queue_info[queue].ra_sta_id;
373 cmd.tid = mvm->queue_info[queue].txq_tid;
374
375 /* Make sure queue info is correct even though we overwrite it */
376 WARN(mvm->queue_info[queue].tid_bitmap,
377 "TXQ #%d info out-of-sync - tids=0x%x\n",
378 queue, mvm->queue_info[queue].tid_bitmap);
379
380 /* If we are here - the queue is freed and we can zero out these vals */
381 mvm->queue_info[queue].tid_bitmap = 0;
382
383 if (sta) {
384 struct iwl_mvm_txq *mvmtxq =
385 iwl_mvm_txq_from_tid(sta, tid);
386
387 mvmtxq->txq_id = IWL_MVM_INVALID_QUEUE;
388 list_del_init(&mvmtxq->list);
389 }
390
391 /* Regardless if this is a reserved TXQ for a STA - mark it as false */
392 mvm->queue_info[queue].reserved = false;
393
394 iwl_trans_txq_disable(mvm->trans, queue, false);
395 ret = iwl_mvm_send_cmd_pdu(mvm, SCD_QUEUE_CFG, 0,
396 sizeof(struct iwl_scd_txq_cfg_cmd), &cmd);
397
398 if (ret)
399 IWL_ERR(mvm, "Failed to disable queue %d (ret=%d)\n",
400 queue, ret);
401 return ret;
402}
403
404static int iwl_mvm_get_queue_agg_tids(struct iwl_mvm *mvm, int queue)
405{
406 struct ieee80211_sta *sta;
407 struct iwl_mvm_sta *mvmsta;
408 unsigned long tid_bitmap;
409 unsigned long agg_tids = 0;
410 u8 sta_id;
411 int tid;
412
413 lockdep_assert_held(&mvm->mutex);
414
415 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
416 return -EINVAL;
417
418 sta_id = mvm->queue_info[queue].ra_sta_id;
419 tid_bitmap = mvm->queue_info[queue].tid_bitmap;
420
421 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
422 lockdep_is_held(&mvm->mutex));
423
424 if (WARN_ON_ONCE(IS_ERR_OR_NULL(sta)))
425 return -EINVAL;
426
427 mvmsta = iwl_mvm_sta_from_mac80211(sta);
428
429 spin_lock_bh(&mvmsta->lock);
430 for_each_set_bit(tid, &tid_bitmap, IWL_MAX_TID_COUNT + 1) {
431 if (mvmsta->tid_data[tid].state == IWL_AGG_ON)
432 agg_tids |= BIT(tid);
433 }
434 spin_unlock_bh(&mvmsta->lock);
435
436 return agg_tids;
437}
438
439/*
440 * Remove a queue from a station's resources.
441 * Note that this only marks as free. It DOESN'T delete a BA agreement, and
442 * doesn't disable the queue
443 */
444static int iwl_mvm_remove_sta_queue_marking(struct iwl_mvm *mvm, int queue)
445{
446 struct ieee80211_sta *sta;
447 struct iwl_mvm_sta *mvmsta;
448 unsigned long tid_bitmap;
449 unsigned long disable_agg_tids = 0;
450 u8 sta_id;
451 int tid;
452
453 lockdep_assert_held(&mvm->mutex);
454
455 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
456 return -EINVAL;
457
458 sta_id = mvm->queue_info[queue].ra_sta_id;
459 tid_bitmap = mvm->queue_info[queue].tid_bitmap;
460
461 rcu_read_lock();
462
463 sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
464
465 if (WARN_ON_ONCE(IS_ERR_OR_NULL(sta))) {
466 rcu_read_unlock();
467 return 0;
468 }
469
470 mvmsta = iwl_mvm_sta_from_mac80211(sta);
471
472 spin_lock_bh(&mvmsta->lock);
473 /* Unmap MAC queues and TIDs from this queue */
474 for_each_set_bit(tid, &tid_bitmap, IWL_MAX_TID_COUNT + 1) {
475 struct iwl_mvm_txq *mvmtxq =
476 iwl_mvm_txq_from_tid(sta, tid);
477
478 if (mvmsta->tid_data[tid].state == IWL_AGG_ON)
479 disable_agg_tids |= BIT(tid);
480 mvmsta->tid_data[tid].txq_id = IWL_MVM_INVALID_QUEUE;
481
482 mvmtxq->txq_id = IWL_MVM_INVALID_QUEUE;
483 list_del_init(&mvmtxq->list);
484 }
485
486 mvmsta->tfd_queue_msk &= ~BIT(queue); /* Don't use this queue anymore */
487 spin_unlock_bh(&mvmsta->lock);
488
489 rcu_read_unlock();
490
491 /*
492 * The TX path may have been using this TXQ_ID from the tid_data,
493 * so make sure it's no longer running so that we can safely reuse
494 * this TXQ later. We've set all the TIDs to IWL_MVM_INVALID_QUEUE
495 * above, but nothing guarantees we've stopped using them. Thus,
496 * without this, we could get to iwl_mvm_disable_txq() and remove
497 * the queue while still sending frames to it.
498 */
499 synchronize_net();
500
501 return disable_agg_tids;
502}
503
504static int iwl_mvm_free_inactive_queue(struct iwl_mvm *mvm, int queue,
505 struct ieee80211_sta *old_sta,
506 u8 new_sta_id)
507{
508 struct iwl_mvm_sta *mvmsta;
509 u8 sta_id, tid;
510 unsigned long disable_agg_tids = 0;
511 bool same_sta;
512 u16 queue_tmp = queue;
513 int ret;
514
515 lockdep_assert_held(&mvm->mutex);
516
517 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
518 return -EINVAL;
519
520 sta_id = mvm->queue_info[queue].ra_sta_id;
521 tid = mvm->queue_info[queue].txq_tid;
522
523 same_sta = sta_id == new_sta_id;
524
525 mvmsta = iwl_mvm_sta_from_staid_protected(mvm, sta_id);
526 if (WARN_ON(!mvmsta))
527 return -EINVAL;
528
529 disable_agg_tids = iwl_mvm_remove_sta_queue_marking(mvm, queue);
530 /* Disable the queue */
531 if (disable_agg_tids)
532 iwl_mvm_invalidate_sta_queue(mvm, queue,
533 disable_agg_tids, false);
534
535 ret = iwl_mvm_disable_txq(mvm, old_sta, sta_id, &queue_tmp, tid);
536 if (ret) {
537 IWL_ERR(mvm,
538 "Failed to free inactive queue %d (ret=%d)\n",
539 queue, ret);
540
541 return ret;
542 }
543
544 /* If TXQ is allocated to another STA, update removal in FW */
545 if (!same_sta)
546 iwl_mvm_invalidate_sta_queue(mvm, queue, 0, true);
547
548 return 0;
549}
550
551static int iwl_mvm_get_shared_queue(struct iwl_mvm *mvm,
552 unsigned long tfd_queue_mask, u8 ac)
553{
554 int queue = 0;
555 u8 ac_to_queue[IEEE80211_NUM_ACS];
556 int i;
557
558 /*
559 * This protects us against grabbing a queue that's being reconfigured
560 * by the inactivity checker.
561 */
562 lockdep_assert_held(&mvm->mutex);
563
564 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
565 return -EINVAL;
566
567 memset(&ac_to_queue, IEEE80211_INVAL_HW_QUEUE, sizeof(ac_to_queue));
568
569 /* See what ACs the existing queues for this STA have */
570 for_each_set_bit(i, &tfd_queue_mask, IWL_MVM_DQA_MAX_DATA_QUEUE) {
571 /* Only DATA queues can be shared */
572 if (i < IWL_MVM_DQA_MIN_DATA_QUEUE &&
573 i != IWL_MVM_DQA_BSS_CLIENT_QUEUE)
574 continue;
575
576 ac_to_queue[mvm->queue_info[i].mac80211_ac] = i;
577 }
578
579 /*
580 * The queue to share is chosen only from DATA queues as follows (in
581 * descending priority):
582 * 1. An AC_BE queue
583 * 2. Same AC queue
584 * 3. Highest AC queue that is lower than new AC
585 * 4. Any existing AC (there always is at least 1 DATA queue)
586 */
587
588 /* Priority 1: An AC_BE queue */
589 if (ac_to_queue[IEEE80211_AC_BE] != IEEE80211_INVAL_HW_QUEUE)
590 queue = ac_to_queue[IEEE80211_AC_BE];
591 /* Priority 2: Same AC queue */
592 else if (ac_to_queue[ac] != IEEE80211_INVAL_HW_QUEUE)
593 queue = ac_to_queue[ac];
594 /* Priority 3a: If new AC is VO and VI exists - use VI */
595 else if (ac == IEEE80211_AC_VO &&
596 ac_to_queue[IEEE80211_AC_VI] != IEEE80211_INVAL_HW_QUEUE)
597 queue = ac_to_queue[IEEE80211_AC_VI];
598 /* Priority 3b: No BE so only AC less than the new one is BK */
599 else if (ac_to_queue[IEEE80211_AC_BK] != IEEE80211_INVAL_HW_QUEUE)
600 queue = ac_to_queue[IEEE80211_AC_BK];
601 /* Priority 4a: No BE nor BK - use VI if exists */
602 else if (ac_to_queue[IEEE80211_AC_VI] != IEEE80211_INVAL_HW_QUEUE)
603 queue = ac_to_queue[IEEE80211_AC_VI];
604 /* Priority 4b: No BE, BK nor VI - use VO if exists */
605 else if (ac_to_queue[IEEE80211_AC_VO] != IEEE80211_INVAL_HW_QUEUE)
606 queue = ac_to_queue[IEEE80211_AC_VO];
607
608 /* Make sure queue found (or not) is legal */
609 if (!iwl_mvm_is_dqa_data_queue(mvm, queue) &&
610 !iwl_mvm_is_dqa_mgmt_queue(mvm, queue) &&
611 (queue != IWL_MVM_DQA_BSS_CLIENT_QUEUE)) {
612 IWL_ERR(mvm, "No DATA queues available to share\n");
613 return -ENOSPC;
614 }
615
616 return queue;
617}
618
619/* Re-configure the SCD for a queue that has already been configured */
620static int iwl_mvm_reconfig_scd(struct iwl_mvm *mvm, int queue, int fifo,
621 int sta_id, int tid, int frame_limit, u16 ssn)
622{
623 struct iwl_scd_txq_cfg_cmd cmd = {
624 .scd_queue = queue,
625 .action = SCD_CFG_ENABLE_QUEUE,
626 .window = frame_limit,
627 .sta_id = sta_id,
628 .ssn = cpu_to_le16(ssn),
629 .tx_fifo = fifo,
630 .aggregate = (queue >= IWL_MVM_DQA_MIN_DATA_QUEUE ||
631 queue == IWL_MVM_DQA_BSS_CLIENT_QUEUE),
632 .tid = tid,
633 };
634 int ret;
635
636 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
637 return -EINVAL;
638
639 if (WARN(mvm->queue_info[queue].tid_bitmap == 0,
640 "Trying to reconfig unallocated queue %d\n", queue))
641 return -ENXIO;
642
643 IWL_DEBUG_TX_QUEUES(mvm, "Reconfig SCD for TXQ #%d\n", queue);
644
645 ret = iwl_mvm_send_cmd_pdu(mvm, SCD_QUEUE_CFG, 0, sizeof(cmd), &cmd);
646 WARN_ONCE(ret, "Failed to re-configure queue %d on FIFO %d, ret=%d\n",
647 queue, fifo, ret);
648
649 return ret;
650}
651
652/*
653 * If a given queue has a higher AC than the TID stream that is being compared
654 * to, the queue needs to be redirected to the lower AC. This function does that
655 * in such a case, otherwise - if no redirection required - it does nothing,
656 * unless the %force param is true.
657 */
658static int iwl_mvm_redirect_queue(struct iwl_mvm *mvm, int queue, int tid,
659 int ac, int ssn, unsigned int wdg_timeout,
660 bool force, struct iwl_mvm_txq *txq)
661{
662 struct iwl_scd_txq_cfg_cmd cmd = {
663 .scd_queue = queue,
664 .action = SCD_CFG_DISABLE_QUEUE,
665 };
666 bool shared_queue;
667 int ret;
668
669 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
670 return -EINVAL;
671
672 /*
673 * If the AC is lower than current one - FIFO needs to be redirected to
674 * the lowest one of the streams in the queue. Check if this is needed
675 * here.
676 * Notice that the enum ieee80211_ac_numbers is "flipped", so BK is with
677 * value 3 and VO with value 0, so to check if ac X is lower than ac Y
678 * we need to check if the numerical value of X is LARGER than of Y.
679 */
680 if (ac <= mvm->queue_info[queue].mac80211_ac && !force) {
681 IWL_DEBUG_TX_QUEUES(mvm,
682 "No redirection needed on TXQ #%d\n",
683 queue);
684 return 0;
685 }
686
687 cmd.sta_id = mvm->queue_info[queue].ra_sta_id;
688 cmd.tx_fifo = iwl_mvm_ac_to_tx_fifo[mvm->queue_info[queue].mac80211_ac];
689 cmd.tid = mvm->queue_info[queue].txq_tid;
690 shared_queue = hweight16(mvm->queue_info[queue].tid_bitmap) > 1;
691
692 IWL_DEBUG_TX_QUEUES(mvm, "Redirecting TXQ #%d to FIFO #%d\n",
693 queue, iwl_mvm_ac_to_tx_fifo[ac]);
694
695 /* Stop the queue and wait for it to empty */
696 txq->stopped = true;
697
698 ret = iwl_trans_wait_tx_queues_empty(mvm->trans, BIT(queue));
699 if (ret) {
700 IWL_ERR(mvm, "Error draining queue %d before reconfig\n",
701 queue);
702 ret = -EIO;
703 goto out;
704 }
705
706 /* Before redirecting the queue we need to de-activate it */
707 iwl_trans_txq_disable(mvm->trans, queue, false);
708 ret = iwl_mvm_send_cmd_pdu(mvm, SCD_QUEUE_CFG, 0, sizeof(cmd), &cmd);
709 if (ret)
710 IWL_ERR(mvm, "Failed SCD disable TXQ %d (ret=%d)\n", queue,
711 ret);
712
713 /* Make sure the SCD wrptr is correctly set before reconfiguring */
714 iwl_trans_txq_enable_cfg(mvm->trans, queue, ssn, NULL, wdg_timeout);
715
716 /* Update the TID "owner" of the queue */
717 mvm->queue_info[queue].txq_tid = tid;
718
719 /* TODO: Work-around SCD bug when moving back by multiples of 0x40 */
720
721 /* Redirect to lower AC */
722 iwl_mvm_reconfig_scd(mvm, queue, iwl_mvm_ac_to_tx_fifo[ac],
723 cmd.sta_id, tid, IWL_FRAME_LIMIT, ssn);
724
725 /* Update AC marking of the queue */
726 mvm->queue_info[queue].mac80211_ac = ac;
727
728 /*
729 * Mark queue as shared in transport if shared
730 * Note this has to be done after queue enablement because enablement
731 * can also set this value, and there is no indication there to shared
732 * queues
733 */
734 if (shared_queue)
735 iwl_trans_txq_set_shared_mode(mvm->trans, queue, true);
736
737out:
738 /* Continue using the queue */
739 txq->stopped = false;
740
741 return ret;
742}
743
744static int iwl_mvm_find_free_queue(struct iwl_mvm *mvm, u8 sta_id,
745 u8 minq, u8 maxq)
746{
747 int i;
748
749 lockdep_assert_held(&mvm->mutex);
750
751 if (WARN(maxq >= mvm->trans->trans_cfg->base_params->num_of_queues,
752 "max queue %d >= num_of_queues (%d)", maxq,
753 mvm->trans->trans_cfg->base_params->num_of_queues))
754 maxq = mvm->trans->trans_cfg->base_params->num_of_queues - 1;
755
756 /* This should not be hit with new TX path */
757 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
758 return -ENOSPC;
759
760 /* Start by looking for a free queue */
761 for (i = minq; i <= maxq; i++)
762 if (mvm->queue_info[i].tid_bitmap == 0 &&
763 mvm->queue_info[i].status == IWL_MVM_QUEUE_FREE)
764 return i;
765
766 return -ENOSPC;
767}
768
769static int iwl_mvm_tvqm_enable_txq(struct iwl_mvm *mvm,
770 u8 sta_id, u8 tid, unsigned int timeout)
771{
772 int queue, size;
773
774 if (tid == IWL_MAX_TID_COUNT) {
775 tid = IWL_MGMT_TID;
776 size = max_t(u32, IWL_MGMT_QUEUE_SIZE,
777 mvm->trans->cfg->min_txq_size);
778 } else {
779 struct ieee80211_sta *sta;
780
781 rcu_read_lock();
782 sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
783
784 /* this queue isn't used for traffic (cab_queue) */
785 if (IS_ERR_OR_NULL(sta)) {
786 size = IWL_MGMT_QUEUE_SIZE;
787 } else if (sta->deflink.he_cap.has_he) {
788 /* support for 256 ba size */
789 size = IWL_DEFAULT_QUEUE_SIZE_HE;
790 } else {
791 size = IWL_DEFAULT_QUEUE_SIZE;
792 }
793
794 rcu_read_unlock();
795 }
796
797 /* take the min with bc tbl entries allowed */
798 size = min_t(u32, size, mvm->trans->txqs.bc_tbl_size / sizeof(u16));
799
800 /* size needs to be power of 2 values for calculating read/write pointers */
801 size = rounddown_pow_of_two(size);
802
803 do {
804 queue = iwl_trans_txq_alloc(mvm->trans, 0, BIT(sta_id),
805 tid, size, timeout);
806
807 if (queue < 0)
808 IWL_DEBUG_TX_QUEUES(mvm,
809 "Failed allocating TXQ of size %d for sta %d tid %d, ret: %d\n",
810 size, sta_id, tid, queue);
811 size /= 2;
812 } while (queue < 0 && size >= 16);
813
814 if (queue < 0)
815 return queue;
816
817 IWL_DEBUG_TX_QUEUES(mvm, "Enabling TXQ #%d for sta %d tid %d\n",
818 queue, sta_id, tid);
819
820 return queue;
821}
822
823static int iwl_mvm_sta_alloc_queue_tvqm(struct iwl_mvm *mvm,
824 struct ieee80211_sta *sta, u8 ac,
825 int tid)
826{
827 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
828 struct iwl_mvm_txq *mvmtxq =
829 iwl_mvm_txq_from_tid(sta, tid);
830 unsigned int wdg_timeout =
831 iwl_mvm_get_wd_timeout(mvm, mvmsta->vif, false, false);
832 int queue = -1;
833
834 lockdep_assert_held(&mvm->mutex);
835
836 IWL_DEBUG_TX_QUEUES(mvm,
837 "Allocating queue for sta %d on tid %d\n",
838 mvmsta->sta_id, tid);
839 queue = iwl_mvm_tvqm_enable_txq(mvm, mvmsta->sta_id, tid, wdg_timeout);
840 if (queue < 0)
841 return queue;
842
843 mvmtxq->txq_id = queue;
844 mvm->tvqm_info[queue].txq_tid = tid;
845 mvm->tvqm_info[queue].sta_id = mvmsta->sta_id;
846
847 IWL_DEBUG_TX_QUEUES(mvm, "Allocated queue is %d\n", queue);
848
849 spin_lock_bh(&mvmsta->lock);
850 mvmsta->tid_data[tid].txq_id = queue;
851 spin_unlock_bh(&mvmsta->lock);
852
853 return 0;
854}
855
856static bool iwl_mvm_update_txq_mapping(struct iwl_mvm *mvm,
857 struct ieee80211_sta *sta,
858 int queue, u8 sta_id, u8 tid)
859{
860 bool enable_queue = true;
861
862 /* Make sure this TID isn't already enabled */
863 if (mvm->queue_info[queue].tid_bitmap & BIT(tid)) {
864 IWL_ERR(mvm, "Trying to enable TXQ %d with existing TID %d\n",
865 queue, tid);
866 return false;
867 }
868
869 /* Update mappings and refcounts */
870 if (mvm->queue_info[queue].tid_bitmap)
871 enable_queue = false;
872
873 mvm->queue_info[queue].tid_bitmap |= BIT(tid);
874 mvm->queue_info[queue].ra_sta_id = sta_id;
875
876 if (enable_queue) {
877 if (tid != IWL_MAX_TID_COUNT)
878 mvm->queue_info[queue].mac80211_ac =
879 tid_to_mac80211_ac[tid];
880 else
881 mvm->queue_info[queue].mac80211_ac = IEEE80211_AC_VO;
882
883 mvm->queue_info[queue].txq_tid = tid;
884 }
885
886 if (sta) {
887 struct iwl_mvm_txq *mvmtxq =
888 iwl_mvm_txq_from_tid(sta, tid);
889
890 mvmtxq->txq_id = queue;
891 }
892
893 IWL_DEBUG_TX_QUEUES(mvm,
894 "Enabling TXQ #%d tids=0x%x\n",
895 queue, mvm->queue_info[queue].tid_bitmap);
896
897 return enable_queue;
898}
899
900static bool iwl_mvm_enable_txq(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
901 int queue, u16 ssn,
902 const struct iwl_trans_txq_scd_cfg *cfg,
903 unsigned int wdg_timeout)
904{
905 struct iwl_scd_txq_cfg_cmd cmd = {
906 .scd_queue = queue,
907 .action = SCD_CFG_ENABLE_QUEUE,
908 .window = cfg->frame_limit,
909 .sta_id = cfg->sta_id,
910 .ssn = cpu_to_le16(ssn),
911 .tx_fifo = cfg->fifo,
912 .aggregate = cfg->aggregate,
913 .tid = cfg->tid,
914 };
915 bool inc_ssn;
916
917 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
918 return false;
919
920 /* Send the enabling command if we need to */
921 if (!iwl_mvm_update_txq_mapping(mvm, sta, queue, cfg->sta_id, cfg->tid))
922 return false;
923
924 inc_ssn = iwl_trans_txq_enable_cfg(mvm->trans, queue, ssn,
925 NULL, wdg_timeout);
926 if (inc_ssn)
927 le16_add_cpu(&cmd.ssn, 1);
928
929 WARN(iwl_mvm_send_cmd_pdu(mvm, SCD_QUEUE_CFG, 0, sizeof(cmd), &cmd),
930 "Failed to configure queue %d on FIFO %d\n", queue, cfg->fifo);
931
932 return inc_ssn;
933}
934
935static void iwl_mvm_change_queue_tid(struct iwl_mvm *mvm, int queue)
936{
937 struct iwl_scd_txq_cfg_cmd cmd = {
938 .scd_queue = queue,
939 .action = SCD_CFG_UPDATE_QUEUE_TID,
940 };
941 int tid;
942 unsigned long tid_bitmap;
943 int ret;
944
945 lockdep_assert_held(&mvm->mutex);
946
947 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
948 return;
949
950 tid_bitmap = mvm->queue_info[queue].tid_bitmap;
951
952 if (WARN(!tid_bitmap, "TXQ %d has no tids assigned to it\n", queue))
953 return;
954
955 /* Find any TID for queue */
956 tid = find_first_bit(&tid_bitmap, IWL_MAX_TID_COUNT + 1);
957 cmd.tid = tid;
958 cmd.tx_fifo = iwl_mvm_ac_to_tx_fifo[tid_to_mac80211_ac[tid]];
959
960 ret = iwl_mvm_send_cmd_pdu(mvm, SCD_QUEUE_CFG, 0, sizeof(cmd), &cmd);
961 if (ret) {
962 IWL_ERR(mvm, "Failed to update owner of TXQ %d (ret=%d)\n",
963 queue, ret);
964 return;
965 }
966
967 mvm->queue_info[queue].txq_tid = tid;
968 IWL_DEBUG_TX_QUEUES(mvm, "Changed TXQ %d ownership to tid %d\n",
969 queue, tid);
970}
971
972static void iwl_mvm_unshare_queue(struct iwl_mvm *mvm, int queue)
973{
974 struct ieee80211_sta *sta;
975 struct iwl_mvm_sta *mvmsta;
976 u8 sta_id;
977 int tid = -1;
978 unsigned long tid_bitmap;
979 unsigned int wdg_timeout;
980 int ssn;
981 int ret = true;
982
983 /* queue sharing is disabled on new TX path */
984 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
985 return;
986
987 lockdep_assert_held(&mvm->mutex);
988
989 sta_id = mvm->queue_info[queue].ra_sta_id;
990 tid_bitmap = mvm->queue_info[queue].tid_bitmap;
991
992 /* Find TID for queue, and make sure it is the only one on the queue */
993 tid = find_first_bit(&tid_bitmap, IWL_MAX_TID_COUNT + 1);
994 if (tid_bitmap != BIT(tid)) {
995 IWL_ERR(mvm, "Failed to unshare q %d, active tids=0x%lx\n",
996 queue, tid_bitmap);
997 return;
998 }
999
1000 IWL_DEBUG_TX_QUEUES(mvm, "Unsharing TXQ %d, keeping tid %d\n", queue,
1001 tid);
1002
1003 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
1004 lockdep_is_held(&mvm->mutex));
1005
1006 if (WARN_ON_ONCE(IS_ERR_OR_NULL(sta)))
1007 return;
1008
1009 mvmsta = iwl_mvm_sta_from_mac80211(sta);
1010 wdg_timeout = iwl_mvm_get_wd_timeout(mvm, mvmsta->vif, false, false);
1011
1012 ssn = IEEE80211_SEQ_TO_SN(mvmsta->tid_data[tid].seq_number);
1013
1014 ret = iwl_mvm_redirect_queue(mvm, queue, tid,
1015 tid_to_mac80211_ac[tid], ssn,
1016 wdg_timeout, true,
1017 iwl_mvm_txq_from_tid(sta, tid));
1018 if (ret) {
1019 IWL_ERR(mvm, "Failed to redirect TXQ %d\n", queue);
1020 return;
1021 }
1022
1023 /* If aggs should be turned back on - do it */
1024 if (mvmsta->tid_data[tid].state == IWL_AGG_ON) {
1025 struct iwl_mvm_add_sta_cmd cmd = {0};
1026
1027 mvmsta->tid_disable_agg &= ~BIT(tid);
1028
1029 cmd.mac_id_n_color = cpu_to_le32(mvmsta->mac_id_n_color);
1030 cmd.sta_id = mvmsta->sta_id;
1031 cmd.add_modify = STA_MODE_MODIFY;
1032 cmd.modify_mask = STA_MODIFY_TID_DISABLE_TX;
1033 cmd.tfd_queue_msk = cpu_to_le32(mvmsta->tfd_queue_msk);
1034 cmd.tid_disable_tx = cpu_to_le16(mvmsta->tid_disable_agg);
1035
1036 ret = iwl_mvm_send_cmd_pdu(mvm, ADD_STA, CMD_ASYNC,
1037 iwl_mvm_add_sta_cmd_size(mvm), &cmd);
1038 if (!ret) {
1039 IWL_DEBUG_TX_QUEUES(mvm,
1040 "TXQ #%d is now aggregated again\n",
1041 queue);
1042
1043 /* Mark queue intenally as aggregating again */
1044 iwl_trans_txq_set_shared_mode(mvm->trans, queue, false);
1045 }
1046 }
1047
1048 mvm->queue_info[queue].status = IWL_MVM_QUEUE_READY;
1049}
1050
1051/*
1052 * Remove inactive TIDs of a given queue.
1053 * If all queue TIDs are inactive - mark the queue as inactive
1054 * If only some the queue TIDs are inactive - unmap them from the queue
1055 *
1056 * Returns %true if all TIDs were removed and the queue could be reused.
1057 */
1058static bool iwl_mvm_remove_inactive_tids(struct iwl_mvm *mvm,
1059 struct iwl_mvm_sta *mvmsta, int queue,
1060 unsigned long tid_bitmap,
1061 unsigned long *unshare_queues,
1062 unsigned long *changetid_queues)
1063{
1064 unsigned int tid;
1065
1066 lockdep_assert_held(&mvmsta->lock);
1067 lockdep_assert_held(&mvm->mutex);
1068
1069 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
1070 return false;
1071
1072 /* Go over all non-active TIDs, incl. IWL_MAX_TID_COUNT (for mgmt) */
1073 for_each_set_bit(tid, &tid_bitmap, IWL_MAX_TID_COUNT + 1) {
1074 /* If some TFDs are still queued - don't mark TID as inactive */
1075 if (iwl_mvm_tid_queued(mvm, &mvmsta->tid_data[tid]))
1076 tid_bitmap &= ~BIT(tid);
1077
1078 /* Don't mark as inactive any TID that has an active BA */
1079 if (mvmsta->tid_data[tid].state != IWL_AGG_OFF)
1080 tid_bitmap &= ~BIT(tid);
1081 }
1082
1083 /* If all TIDs in the queue are inactive - return it can be reused */
1084 if (tid_bitmap == mvm->queue_info[queue].tid_bitmap) {
1085 IWL_DEBUG_TX_QUEUES(mvm, "Queue %d is inactive\n", queue);
1086 return true;
1087 }
1088
1089 /*
1090 * If we are here, this is a shared queue and not all TIDs timed-out.
1091 * Remove the ones that did.
1092 */
1093 for_each_set_bit(tid, &tid_bitmap, IWL_MAX_TID_COUNT + 1) {
1094 u16 q_tid_bitmap;
1095
1096 mvmsta->tid_data[tid].txq_id = IWL_MVM_INVALID_QUEUE;
1097 mvm->queue_info[queue].tid_bitmap &= ~BIT(tid);
1098
1099 q_tid_bitmap = mvm->queue_info[queue].tid_bitmap;
1100
1101 /*
1102 * We need to take into account a situation in which a TXQ was
1103 * allocated to TID x, and then turned shared by adding TIDs y
1104 * and z. If TID x becomes inactive and is removed from the TXQ,
1105 * ownership must be given to one of the remaining TIDs.
1106 * This is mainly because if TID x continues - a new queue can't
1107 * be allocated for it as long as it is an owner of another TXQ.
1108 *
1109 * Mark this queue in the right bitmap, we'll send the command
1110 * to the firmware later.
1111 */
1112 if (!(q_tid_bitmap & BIT(mvm->queue_info[queue].txq_tid)))
1113 set_bit(queue, changetid_queues);
1114
1115 IWL_DEBUG_TX_QUEUES(mvm,
1116 "Removing inactive TID %d from shared Q:%d\n",
1117 tid, queue);
1118 }
1119
1120 IWL_DEBUG_TX_QUEUES(mvm,
1121 "TXQ #%d left with tid bitmap 0x%x\n", queue,
1122 mvm->queue_info[queue].tid_bitmap);
1123
1124 /*
1125 * There may be different TIDs with the same mac queues, so make
1126 * sure all TIDs have existing corresponding mac queues enabled
1127 */
1128 tid_bitmap = mvm->queue_info[queue].tid_bitmap;
1129
1130 /* If the queue is marked as shared - "unshare" it */
1131 if (hweight16(mvm->queue_info[queue].tid_bitmap) == 1 &&
1132 mvm->queue_info[queue].status == IWL_MVM_QUEUE_SHARED) {
1133 IWL_DEBUG_TX_QUEUES(mvm, "Marking Q:%d for reconfig\n",
1134 queue);
1135 set_bit(queue, unshare_queues);
1136 }
1137
1138 return false;
1139}
1140
1141/*
1142 * Check for inactivity - this includes checking if any queue
1143 * can be unshared and finding one (and only one) that can be
1144 * reused.
1145 * This function is also invoked as a sort of clean-up task,
1146 * in which case @alloc_for_sta is IWL_MVM_INVALID_STA.
1147 *
1148 * Returns the queue number, or -ENOSPC.
1149 */
1150static int iwl_mvm_inactivity_check(struct iwl_mvm *mvm, u8 alloc_for_sta)
1151{
1152 unsigned long now = jiffies;
1153 unsigned long unshare_queues = 0;
1154 unsigned long changetid_queues = 0;
1155 int i, ret, free_queue = -ENOSPC;
1156 struct ieee80211_sta *queue_owner = NULL;
1157
1158 lockdep_assert_held(&mvm->mutex);
1159
1160 if (iwl_mvm_has_new_tx_api(mvm))
1161 return -ENOSPC;
1162
1163 rcu_read_lock();
1164
1165 /* we skip the CMD queue below by starting at 1 */
1166 BUILD_BUG_ON(IWL_MVM_DQA_CMD_QUEUE != 0);
1167
1168 for (i = 1; i < IWL_MAX_HW_QUEUES; i++) {
1169 struct ieee80211_sta *sta;
1170 struct iwl_mvm_sta *mvmsta;
1171 u8 sta_id;
1172 int tid;
1173 unsigned long inactive_tid_bitmap = 0;
1174 unsigned long queue_tid_bitmap;
1175
1176 queue_tid_bitmap = mvm->queue_info[i].tid_bitmap;
1177 if (!queue_tid_bitmap)
1178 continue;
1179
1180 /* If TXQ isn't in active use anyway - nothing to do here... */
1181 if (mvm->queue_info[i].status != IWL_MVM_QUEUE_READY &&
1182 mvm->queue_info[i].status != IWL_MVM_QUEUE_SHARED)
1183 continue;
1184
1185 /* Check to see if there are inactive TIDs on this queue */
1186 for_each_set_bit(tid, &queue_tid_bitmap,
1187 IWL_MAX_TID_COUNT + 1) {
1188 if (time_after(mvm->queue_info[i].last_frame_time[tid] +
1189 IWL_MVM_DQA_QUEUE_TIMEOUT, now))
1190 continue;
1191
1192 inactive_tid_bitmap |= BIT(tid);
1193 }
1194
1195 /* If all TIDs are active - finish check on this queue */
1196 if (!inactive_tid_bitmap)
1197 continue;
1198
1199 /*
1200 * If we are here - the queue hadn't been served recently and is
1201 * in use
1202 */
1203
1204 sta_id = mvm->queue_info[i].ra_sta_id;
1205 sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
1206
1207 /*
1208 * If the STA doesn't exist anymore, it isn't an error. It could
1209 * be that it was removed since getting the queues, and in this
1210 * case it should've inactivated its queues anyway.
1211 */
1212 if (IS_ERR_OR_NULL(sta))
1213 continue;
1214
1215 mvmsta = iwl_mvm_sta_from_mac80211(sta);
1216
1217 spin_lock_bh(&mvmsta->lock);
1218 ret = iwl_mvm_remove_inactive_tids(mvm, mvmsta, i,
1219 inactive_tid_bitmap,
1220 &unshare_queues,
1221 &changetid_queues);
1222 if (ret && free_queue < 0) {
1223 queue_owner = sta;
1224 free_queue = i;
1225 }
1226 /* only unlock sta lock - we still need the queue info lock */
1227 spin_unlock_bh(&mvmsta->lock);
1228 }
1229
1230
1231 /* Reconfigure queues requiring reconfiguation */
1232 for_each_set_bit(i, &unshare_queues, IWL_MAX_HW_QUEUES)
1233 iwl_mvm_unshare_queue(mvm, i);
1234 for_each_set_bit(i, &changetid_queues, IWL_MAX_HW_QUEUES)
1235 iwl_mvm_change_queue_tid(mvm, i);
1236
1237 rcu_read_unlock();
1238
1239 if (free_queue >= 0 && alloc_for_sta != IWL_MVM_INVALID_STA) {
1240 ret = iwl_mvm_free_inactive_queue(mvm, free_queue, queue_owner,
1241 alloc_for_sta);
1242 if (ret)
1243 return ret;
1244 }
1245
1246 return free_queue;
1247}
1248
1249static int iwl_mvm_sta_alloc_queue(struct iwl_mvm *mvm,
1250 struct ieee80211_sta *sta, u8 ac, int tid)
1251{
1252 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
1253 struct iwl_trans_txq_scd_cfg cfg = {
1254 .fifo = iwl_mvm_mac_ac_to_tx_fifo(mvm, ac),
1255 .sta_id = mvmsta->sta_id,
1256 .tid = tid,
1257 .frame_limit = IWL_FRAME_LIMIT,
1258 };
1259 unsigned int wdg_timeout =
1260 iwl_mvm_get_wd_timeout(mvm, mvmsta->vif, false, false);
1261 int queue = -1;
1262 u16 queue_tmp;
1263 unsigned long disable_agg_tids = 0;
1264 enum iwl_mvm_agg_state queue_state;
1265 bool shared_queue = false, inc_ssn;
1266 int ssn;
1267 unsigned long tfd_queue_mask;
1268 int ret;
1269
1270 lockdep_assert_held(&mvm->mutex);
1271
1272 if (iwl_mvm_has_new_tx_api(mvm))
1273 return iwl_mvm_sta_alloc_queue_tvqm(mvm, sta, ac, tid);
1274
1275 spin_lock_bh(&mvmsta->lock);
1276 tfd_queue_mask = mvmsta->tfd_queue_msk;
1277 ssn = IEEE80211_SEQ_TO_SN(mvmsta->tid_data[tid].seq_number);
1278 spin_unlock_bh(&mvmsta->lock);
1279
1280 if (tid == IWL_MAX_TID_COUNT) {
1281 queue = iwl_mvm_find_free_queue(mvm, mvmsta->sta_id,
1282 IWL_MVM_DQA_MIN_MGMT_QUEUE,
1283 IWL_MVM_DQA_MAX_MGMT_QUEUE);
1284 if (queue >= IWL_MVM_DQA_MIN_MGMT_QUEUE)
1285 IWL_DEBUG_TX_QUEUES(mvm, "Found free MGMT queue #%d\n",
1286 queue);
1287
1288 /* If no such queue is found, we'll use a DATA queue instead */
1289 }
1290
1291 if ((queue < 0 && mvmsta->reserved_queue != IEEE80211_INVAL_HW_QUEUE) &&
1292 (mvm->queue_info[mvmsta->reserved_queue].status ==
1293 IWL_MVM_QUEUE_RESERVED)) {
1294 queue = mvmsta->reserved_queue;
1295 mvm->queue_info[queue].reserved = true;
1296 IWL_DEBUG_TX_QUEUES(mvm, "Using reserved queue #%d\n", queue);
1297 }
1298
1299 if (queue < 0)
1300 queue = iwl_mvm_find_free_queue(mvm, mvmsta->sta_id,
1301 IWL_MVM_DQA_MIN_DATA_QUEUE,
1302 IWL_MVM_DQA_MAX_DATA_QUEUE);
1303 if (queue < 0) {
1304 /* try harder - perhaps kill an inactive queue */
1305 queue = iwl_mvm_inactivity_check(mvm, mvmsta->sta_id);
1306 }
1307
1308 /* No free queue - we'll have to share */
1309 if (queue <= 0) {
1310 queue = iwl_mvm_get_shared_queue(mvm, tfd_queue_mask, ac);
1311 if (queue > 0) {
1312 shared_queue = true;
1313 mvm->queue_info[queue].status = IWL_MVM_QUEUE_SHARED;
1314 }
1315 }
1316
1317 /*
1318 * Mark TXQ as ready, even though it hasn't been fully configured yet,
1319 * to make sure no one else takes it.
1320 * This will allow avoiding re-acquiring the lock at the end of the
1321 * configuration. On error we'll mark it back as free.
1322 */
1323 if (queue > 0 && !shared_queue)
1324 mvm->queue_info[queue].status = IWL_MVM_QUEUE_READY;
1325
1326 /* This shouldn't happen - out of queues */
1327 if (WARN_ON(queue <= 0)) {
1328 IWL_ERR(mvm, "No available queues for tid %d on sta_id %d\n",
1329 tid, cfg.sta_id);
1330 return queue;
1331 }
1332
1333 /*
1334 * Actual en/disablement of aggregations is through the ADD_STA HCMD,
1335 * but for configuring the SCD to send A-MPDUs we need to mark the queue
1336 * as aggregatable.
1337 * Mark all DATA queues as allowing to be aggregated at some point
1338 */
1339 cfg.aggregate = (queue >= IWL_MVM_DQA_MIN_DATA_QUEUE ||
1340 queue == IWL_MVM_DQA_BSS_CLIENT_QUEUE);
1341
1342 IWL_DEBUG_TX_QUEUES(mvm,
1343 "Allocating %squeue #%d to sta %d on tid %d\n",
1344 shared_queue ? "shared " : "", queue,
1345 mvmsta->sta_id, tid);
1346
1347 if (shared_queue) {
1348 /* Disable any open aggs on this queue */
1349 disable_agg_tids = iwl_mvm_get_queue_agg_tids(mvm, queue);
1350
1351 if (disable_agg_tids) {
1352 IWL_DEBUG_TX_QUEUES(mvm, "Disabling aggs on queue %d\n",
1353 queue);
1354 iwl_mvm_invalidate_sta_queue(mvm, queue,
1355 disable_agg_tids, false);
1356 }
1357 }
1358
1359 inc_ssn = iwl_mvm_enable_txq(mvm, sta, queue, ssn, &cfg, wdg_timeout);
1360
1361 /*
1362 * Mark queue as shared in transport if shared
1363 * Note this has to be done after queue enablement because enablement
1364 * can also set this value, and there is no indication there to shared
1365 * queues
1366 */
1367 if (shared_queue)
1368 iwl_trans_txq_set_shared_mode(mvm->trans, queue, true);
1369
1370 spin_lock_bh(&mvmsta->lock);
1371 /*
1372 * This looks racy, but it is not. We have only one packet for
1373 * this ra/tid in our Tx path since we stop the Qdisc when we
1374 * need to allocate a new TFD queue.
1375 */
1376 if (inc_ssn) {
1377 mvmsta->tid_data[tid].seq_number += 0x10;
1378 ssn = (ssn + 1) & IEEE80211_SCTL_SEQ;
1379 }
1380 mvmsta->tid_data[tid].txq_id = queue;
1381 mvmsta->tfd_queue_msk |= BIT(queue);
1382 queue_state = mvmsta->tid_data[tid].state;
1383
1384 if (mvmsta->reserved_queue == queue)
1385 mvmsta->reserved_queue = IEEE80211_INVAL_HW_QUEUE;
1386 spin_unlock_bh(&mvmsta->lock);
1387
1388 if (!shared_queue) {
1389 ret = iwl_mvm_sta_send_to_fw(mvm, sta, true, STA_MODIFY_QUEUES);
1390 if (ret)
1391 goto out_err;
1392
1393 /* If we need to re-enable aggregations... */
1394 if (queue_state == IWL_AGG_ON) {
1395 ret = iwl_mvm_sta_tx_agg(mvm, sta, tid, queue, true);
1396 if (ret)
1397 goto out_err;
1398 }
1399 } else {
1400 /* Redirect queue, if needed */
1401 ret = iwl_mvm_redirect_queue(mvm, queue, tid, ac, ssn,
1402 wdg_timeout, false,
1403 iwl_mvm_txq_from_tid(sta, tid));
1404 if (ret)
1405 goto out_err;
1406 }
1407
1408 return 0;
1409
1410out_err:
1411 queue_tmp = queue;
1412 iwl_mvm_disable_txq(mvm, sta, mvmsta->sta_id, &queue_tmp, tid);
1413
1414 return ret;
1415}
1416
1417void iwl_mvm_add_new_dqa_stream_wk(struct work_struct *wk)
1418{
1419 struct iwl_mvm *mvm = container_of(wk, struct iwl_mvm,
1420 add_stream_wk);
1421
1422 mutex_lock(&mvm->mutex);
1423
1424 iwl_mvm_inactivity_check(mvm, IWL_MVM_INVALID_STA);
1425
1426 while (!list_empty(&mvm->add_stream_txqs)) {
1427 struct iwl_mvm_txq *mvmtxq;
1428 struct ieee80211_txq *txq;
1429 u8 tid;
1430
1431 mvmtxq = list_first_entry(&mvm->add_stream_txqs,
1432 struct iwl_mvm_txq, list);
1433
1434 txq = container_of((void *)mvmtxq, struct ieee80211_txq,
1435 drv_priv);
1436 tid = txq->tid;
1437 if (tid == IEEE80211_NUM_TIDS)
1438 tid = IWL_MAX_TID_COUNT;
1439
1440 /*
1441 * We can't really do much here, but if this fails we can't
1442 * transmit anyway - so just don't transmit the frame etc.
1443 * and let them back up ... we've tried our best to allocate
1444 * a queue in the function itself.
1445 */
1446 if (iwl_mvm_sta_alloc_queue(mvm, txq->sta, txq->ac, tid)) {
1447 list_del_init(&mvmtxq->list);
1448 continue;
1449 }
1450
1451 list_del_init(&mvmtxq->list);
1452 local_bh_disable();
1453 iwl_mvm_mac_itxq_xmit(mvm->hw, txq);
1454 local_bh_enable();
1455 }
1456
1457 mutex_unlock(&mvm->mutex);
1458}
1459
1460static int iwl_mvm_reserve_sta_stream(struct iwl_mvm *mvm,
1461 struct ieee80211_sta *sta,
1462 enum nl80211_iftype vif_type)
1463{
1464 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
1465 int queue;
1466
1467 /* queue reserving is disabled on new TX path */
1468 if (WARN_ON(iwl_mvm_has_new_tx_api(mvm)))
1469 return 0;
1470
1471 /* run the general cleanup/unsharing of queues */
1472 iwl_mvm_inactivity_check(mvm, IWL_MVM_INVALID_STA);
1473
1474 /* Make sure we have free resources for this STA */
1475 if (vif_type == NL80211_IFTYPE_STATION && !sta->tdls &&
1476 !mvm->queue_info[IWL_MVM_DQA_BSS_CLIENT_QUEUE].tid_bitmap &&
1477 (mvm->queue_info[IWL_MVM_DQA_BSS_CLIENT_QUEUE].status ==
1478 IWL_MVM_QUEUE_FREE))
1479 queue = IWL_MVM_DQA_BSS_CLIENT_QUEUE;
1480 else
1481 queue = iwl_mvm_find_free_queue(mvm, mvmsta->sta_id,
1482 IWL_MVM_DQA_MIN_DATA_QUEUE,
1483 IWL_MVM_DQA_MAX_DATA_QUEUE);
1484 if (queue < 0) {
1485 /* try again - this time kick out a queue if needed */
1486 queue = iwl_mvm_inactivity_check(mvm, mvmsta->sta_id);
1487 if (queue < 0) {
1488 IWL_ERR(mvm, "No available queues for new station\n");
1489 return -ENOSPC;
1490 }
1491 }
1492 mvm->queue_info[queue].status = IWL_MVM_QUEUE_RESERVED;
1493
1494 mvmsta->reserved_queue = queue;
1495
1496 IWL_DEBUG_TX_QUEUES(mvm, "Reserving data queue #%d for sta_id %d\n",
1497 queue, mvmsta->sta_id);
1498
1499 return 0;
1500}
1501
1502/*
1503 * In DQA mode, after a HW restart the queues should be allocated as before, in
1504 * order to avoid race conditions when there are shared queues. This function
1505 * does the re-mapping and queue allocation.
1506 *
1507 * Note that re-enabling aggregations isn't done in this function.
1508 */
1509static void iwl_mvm_realloc_queues_after_restart(struct iwl_mvm *mvm,
1510 struct ieee80211_sta *sta)
1511{
1512 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
1513 unsigned int wdg =
1514 iwl_mvm_get_wd_timeout(mvm, mvm_sta->vif, false, false);
1515 int i;
1516 struct iwl_trans_txq_scd_cfg cfg = {
1517 .sta_id = mvm_sta->sta_id,
1518 .frame_limit = IWL_FRAME_LIMIT,
1519 };
1520
1521 /* Make sure reserved queue is still marked as such (if allocated) */
1522 if (mvm_sta->reserved_queue != IEEE80211_INVAL_HW_QUEUE)
1523 mvm->queue_info[mvm_sta->reserved_queue].status =
1524 IWL_MVM_QUEUE_RESERVED;
1525
1526 for (i = 0; i <= IWL_MAX_TID_COUNT; i++) {
1527 struct iwl_mvm_tid_data *tid_data = &mvm_sta->tid_data[i];
1528 int txq_id = tid_data->txq_id;
1529 int ac;
1530
1531 if (txq_id == IWL_MVM_INVALID_QUEUE)
1532 continue;
1533
1534 ac = tid_to_mac80211_ac[i];
1535
1536 if (iwl_mvm_has_new_tx_api(mvm)) {
1537 IWL_DEBUG_TX_QUEUES(mvm,
1538 "Re-mapping sta %d tid %d\n",
1539 mvm_sta->sta_id, i);
1540 txq_id = iwl_mvm_tvqm_enable_txq(mvm, mvm_sta->sta_id,
1541 i, wdg);
1542 /*
1543 * on failures, just set it to IWL_MVM_INVALID_QUEUE
1544 * to try again later, we have no other good way of
1545 * failing here
1546 */
1547 if (txq_id < 0)
1548 txq_id = IWL_MVM_INVALID_QUEUE;
1549 tid_data->txq_id = txq_id;
1550
1551 /*
1552 * Since we don't set the seq number after reset, and HW
1553 * sets it now, FW reset will cause the seq num to start
1554 * at 0 again, so driver will need to update it
1555 * internally as well, so it keeps in sync with real val
1556 */
1557 tid_data->seq_number = 0;
1558 } else {
1559 u16 seq = IEEE80211_SEQ_TO_SN(tid_data->seq_number);
1560
1561 cfg.tid = i;
1562 cfg.fifo = iwl_mvm_mac_ac_to_tx_fifo(mvm, ac);
1563 cfg.aggregate = (txq_id >= IWL_MVM_DQA_MIN_DATA_QUEUE ||
1564 txq_id ==
1565 IWL_MVM_DQA_BSS_CLIENT_QUEUE);
1566
1567 IWL_DEBUG_TX_QUEUES(mvm,
1568 "Re-mapping sta %d tid %d to queue %d\n",
1569 mvm_sta->sta_id, i, txq_id);
1570
1571 iwl_mvm_enable_txq(mvm, sta, txq_id, seq, &cfg, wdg);
1572 mvm->queue_info[txq_id].status = IWL_MVM_QUEUE_READY;
1573 }
1574 }
1575}
1576
1577static int iwl_mvm_add_int_sta_common(struct iwl_mvm *mvm,
1578 struct iwl_mvm_int_sta *sta,
1579 const u8 *addr,
1580 u16 mac_id, u16 color)
1581{
1582 struct iwl_mvm_add_sta_cmd cmd;
1583 int ret;
1584 u32 status = ADD_STA_SUCCESS;
1585
1586 lockdep_assert_held(&mvm->mutex);
1587
1588 memset(&cmd, 0, sizeof(cmd));
1589 cmd.sta_id = sta->sta_id;
1590
1591 if (iwl_fw_lookup_cmd_ver(mvm->fw, ADD_STA, 0) >= 12 &&
1592 sta->type == IWL_STA_AUX_ACTIVITY)
1593 cmd.mac_id_n_color = cpu_to_le32(mac_id);
1594 else
1595 cmd.mac_id_n_color = cpu_to_le32(FW_CMD_ID_AND_COLOR(mac_id,
1596 color));
1597
1598 if (fw_has_api(&mvm->fw->ucode_capa, IWL_UCODE_TLV_API_STA_TYPE))
1599 cmd.station_type = sta->type;
1600
1601 if (!iwl_mvm_has_new_tx_api(mvm))
1602 cmd.tfd_queue_msk = cpu_to_le32(sta->tfd_queue_msk);
1603 cmd.tid_disable_tx = cpu_to_le16(0xffff);
1604
1605 if (addr)
1606 memcpy(cmd.addr, addr, ETH_ALEN);
1607
1608 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA,
1609 iwl_mvm_add_sta_cmd_size(mvm),
1610 &cmd, &status);
1611 if (ret)
1612 return ret;
1613
1614 switch (status & IWL_ADD_STA_STATUS_MASK) {
1615 case ADD_STA_SUCCESS:
1616 IWL_DEBUG_INFO(mvm, "Internal station added.\n");
1617 return 0;
1618 default:
1619 ret = -EIO;
1620 IWL_ERR(mvm, "Add internal station failed, status=0x%x\n",
1621 status);
1622 break;
1623 }
1624 return ret;
1625}
1626
1627int iwl_mvm_add_sta(struct iwl_mvm *mvm,
1628 struct ieee80211_vif *vif,
1629 struct ieee80211_sta *sta)
1630{
1631 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1632 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
1633 struct iwl_mvm_rxq_dup_data *dup_data;
1634 int i, ret, sta_id;
1635 bool sta_update = false;
1636 unsigned int sta_flags = 0;
1637
1638 lockdep_assert_held(&mvm->mutex);
1639
1640 if (!test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status))
1641 sta_id = iwl_mvm_find_free_sta_id(mvm,
1642 ieee80211_vif_type_p2p(vif));
1643 else
1644 sta_id = mvm_sta->sta_id;
1645
1646 if (sta_id == IWL_MVM_INVALID_STA)
1647 return -ENOSPC;
1648
1649 spin_lock_init(&mvm_sta->lock);
1650
1651 /* if this is a HW restart re-alloc existing queues */
1652 if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
1653 struct iwl_mvm_int_sta tmp_sta = {
1654 .sta_id = sta_id,
1655 .type = mvm_sta->sta_type,
1656 };
1657
1658 /*
1659 * First add an empty station since allocating
1660 * a queue requires a valid station
1661 */
1662 ret = iwl_mvm_add_int_sta_common(mvm, &tmp_sta, sta->addr,
1663 mvmvif->id, mvmvif->color);
1664 if (ret)
1665 goto err;
1666
1667 iwl_mvm_realloc_queues_after_restart(mvm, sta);
1668 sta_update = true;
1669 sta_flags = iwl_mvm_has_new_tx_api(mvm) ? 0 : STA_MODIFY_QUEUES;
1670 goto update_fw;
1671 }
1672
1673 mvm_sta->sta_id = sta_id;
1674 mvm_sta->mac_id_n_color = FW_CMD_ID_AND_COLOR(mvmvif->id,
1675 mvmvif->color);
1676 mvm_sta->vif = vif;
1677 if (!mvm->trans->trans_cfg->gen2)
1678 mvm_sta->max_agg_bufsize = LINK_QUAL_AGG_FRAME_LIMIT_DEF;
1679 else
1680 mvm_sta->max_agg_bufsize = LINK_QUAL_AGG_FRAME_LIMIT_GEN2_DEF;
1681 mvm_sta->tx_protection = 0;
1682 mvm_sta->tt_tx_protection = false;
1683 mvm_sta->sta_type = sta->tdls ? IWL_STA_TDLS_LINK : IWL_STA_LINK;
1684
1685 /* HW restart, don't assume the memory has been zeroed */
1686 mvm_sta->tid_disable_agg = 0xffff; /* No aggs at first */
1687 mvm_sta->tfd_queue_msk = 0;
1688
1689 /* for HW restart - reset everything but the sequence number */
1690 for (i = 0; i <= IWL_MAX_TID_COUNT; i++) {
1691 u16 seq = mvm_sta->tid_data[i].seq_number;
1692 memset(&mvm_sta->tid_data[i], 0, sizeof(mvm_sta->tid_data[i]));
1693 mvm_sta->tid_data[i].seq_number = seq;
1694
1695 /*
1696 * Mark all queues for this STA as unallocated and defer TX
1697 * frames until the queue is allocated
1698 */
1699 mvm_sta->tid_data[i].txq_id = IWL_MVM_INVALID_QUEUE;
1700 }
1701
1702 for (i = 0; i < ARRAY_SIZE(sta->txq); i++) {
1703 struct iwl_mvm_txq *mvmtxq =
1704 iwl_mvm_txq_from_mac80211(sta->txq[i]);
1705
1706 mvmtxq->txq_id = IWL_MVM_INVALID_QUEUE;
1707 INIT_LIST_HEAD(&mvmtxq->list);
1708 atomic_set(&mvmtxq->tx_request, 0);
1709 }
1710
1711 mvm_sta->agg_tids = 0;
1712
1713 if (iwl_mvm_has_new_rx_api(mvm) &&
1714 !test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
1715 int q;
1716
1717 dup_data = kcalloc(mvm->trans->num_rx_queues,
1718 sizeof(*dup_data), GFP_KERNEL);
1719 if (!dup_data)
1720 return -ENOMEM;
1721 /*
1722 * Initialize all the last_seq values to 0xffff which can never
1723 * compare equal to the frame's seq_ctrl in the check in
1724 * iwl_mvm_is_dup() since the lower 4 bits are the fragment
1725 * number and fragmented packets don't reach that function.
1726 *
1727 * This thus allows receiving a packet with seqno 0 and the
1728 * retry bit set as the very first packet on a new TID.
1729 */
1730 for (q = 0; q < mvm->trans->num_rx_queues; q++)
1731 memset(dup_data[q].last_seq, 0xff,
1732 sizeof(dup_data[q].last_seq));
1733 mvm_sta->dup_data = dup_data;
1734 }
1735
1736 if (!iwl_mvm_has_new_tx_api(mvm)) {
1737 ret = iwl_mvm_reserve_sta_stream(mvm, sta,
1738 ieee80211_vif_type_p2p(vif));
1739 if (ret)
1740 goto err;
1741 }
1742
1743 /*
1744 * if rs is registered with mac80211, then "add station" will be handled
1745 * via the corresponding ops, otherwise need to notify rate scaling here
1746 */
1747 if (iwl_mvm_has_tlc_offload(mvm))
1748 iwl_mvm_rs_add_sta(mvm, mvm_sta);
1749 else
1750 spin_lock_init(&mvm_sta->lq_sta.rs_drv.pers.lock);
1751
1752 iwl_mvm_toggle_tx_ant(mvm, &mvm_sta->tx_ant);
1753
1754update_fw:
1755 ret = iwl_mvm_sta_send_to_fw(mvm, sta, sta_update, sta_flags);
1756 if (ret)
1757 goto err;
1758
1759 if (vif->type == NL80211_IFTYPE_STATION) {
1760 if (!sta->tdls) {
1761 WARN_ON(mvmvif->ap_sta_id != IWL_MVM_INVALID_STA);
1762 mvmvif->ap_sta_id = sta_id;
1763 } else {
1764 WARN_ON(mvmvif->ap_sta_id == IWL_MVM_INVALID_STA);
1765 }
1766 }
1767
1768 rcu_assign_pointer(mvm->fw_id_to_mac_id[sta_id], sta);
1769
1770 return 0;
1771
1772err:
1773 return ret;
1774}
1775
1776int iwl_mvm_drain_sta(struct iwl_mvm *mvm, struct iwl_mvm_sta *mvmsta,
1777 bool drain)
1778{
1779 struct iwl_mvm_add_sta_cmd cmd = {};
1780 int ret;
1781 u32 status;
1782
1783 lockdep_assert_held(&mvm->mutex);
1784
1785 cmd.mac_id_n_color = cpu_to_le32(mvmsta->mac_id_n_color);
1786 cmd.sta_id = mvmsta->sta_id;
1787 cmd.add_modify = STA_MODE_MODIFY;
1788 cmd.station_flags = drain ? cpu_to_le32(STA_FLG_DRAIN_FLOW) : 0;
1789 cmd.station_flags_msk = cpu_to_le32(STA_FLG_DRAIN_FLOW);
1790
1791 status = ADD_STA_SUCCESS;
1792 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA,
1793 iwl_mvm_add_sta_cmd_size(mvm),
1794 &cmd, &status);
1795 if (ret)
1796 return ret;
1797
1798 switch (status & IWL_ADD_STA_STATUS_MASK) {
1799 case ADD_STA_SUCCESS:
1800 IWL_DEBUG_INFO(mvm, "Frames for staid %d will drained in fw\n",
1801 mvmsta->sta_id);
1802 break;
1803 default:
1804 ret = -EIO;
1805 IWL_ERR(mvm, "Couldn't drain frames for staid %d\n",
1806 mvmsta->sta_id);
1807 break;
1808 }
1809
1810 return ret;
1811}
1812
1813/*
1814 * Remove a station from the FW table. Before sending the command to remove
1815 * the station validate that the station is indeed known to the driver (sanity
1816 * only).
1817 */
1818static int iwl_mvm_rm_sta_common(struct iwl_mvm *mvm, u8 sta_id)
1819{
1820 struct ieee80211_sta *sta;
1821 struct iwl_mvm_rm_sta_cmd rm_sta_cmd = {
1822 .sta_id = sta_id,
1823 };
1824 int ret;
1825
1826 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
1827 lockdep_is_held(&mvm->mutex));
1828
1829 /* Note: internal stations are marked as error values */
1830 if (!sta) {
1831 IWL_ERR(mvm, "Invalid station id\n");
1832 return -EINVAL;
1833 }
1834
1835 ret = iwl_mvm_send_cmd_pdu(mvm, REMOVE_STA, 0,
1836 sizeof(rm_sta_cmd), &rm_sta_cmd);
1837 if (ret) {
1838 IWL_ERR(mvm, "Failed to remove station. Id=%d\n", sta_id);
1839 return ret;
1840 }
1841
1842 return 0;
1843}
1844
1845static void iwl_mvm_disable_sta_queues(struct iwl_mvm *mvm,
1846 struct ieee80211_vif *vif,
1847 struct ieee80211_sta *sta)
1848{
1849 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
1850 int i;
1851
1852 lockdep_assert_held(&mvm->mutex);
1853
1854 for (i = 0; i < ARRAY_SIZE(mvm_sta->tid_data); i++) {
1855 if (mvm_sta->tid_data[i].txq_id == IWL_MVM_INVALID_QUEUE)
1856 continue;
1857
1858 iwl_mvm_disable_txq(mvm, sta, mvm_sta->sta_id,
1859 &mvm_sta->tid_data[i].txq_id, i);
1860 mvm_sta->tid_data[i].txq_id = IWL_MVM_INVALID_QUEUE;
1861 }
1862
1863 for (i = 0; i < ARRAY_SIZE(sta->txq); i++) {
1864 struct iwl_mvm_txq *mvmtxq =
1865 iwl_mvm_txq_from_mac80211(sta->txq[i]);
1866
1867 mvmtxq->txq_id = IWL_MVM_INVALID_QUEUE;
1868 list_del_init(&mvmtxq->list);
1869 }
1870}
1871
1872int iwl_mvm_wait_sta_queues_empty(struct iwl_mvm *mvm,
1873 struct iwl_mvm_sta *mvm_sta)
1874{
1875 int i;
1876
1877 for (i = 0; i < ARRAY_SIZE(mvm_sta->tid_data); i++) {
1878 u16 txq_id;
1879 int ret;
1880
1881 spin_lock_bh(&mvm_sta->lock);
1882 txq_id = mvm_sta->tid_data[i].txq_id;
1883 spin_unlock_bh(&mvm_sta->lock);
1884
1885 if (txq_id == IWL_MVM_INVALID_QUEUE)
1886 continue;
1887
1888 ret = iwl_trans_wait_txq_empty(mvm->trans, txq_id);
1889 if (ret)
1890 return ret;
1891 }
1892
1893 return 0;
1894}
1895
1896int iwl_mvm_rm_sta(struct iwl_mvm *mvm,
1897 struct ieee80211_vif *vif,
1898 struct ieee80211_sta *sta)
1899{
1900 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1901 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
1902 u8 sta_id = mvm_sta->sta_id;
1903 int ret;
1904
1905 lockdep_assert_held(&mvm->mutex);
1906
1907 if (iwl_mvm_has_new_rx_api(mvm))
1908 kfree(mvm_sta->dup_data);
1909
1910 ret = iwl_mvm_drain_sta(mvm, mvm_sta, true);
1911 if (ret)
1912 return ret;
1913
1914 /* flush its queues here since we are freeing mvm_sta */
1915 ret = iwl_mvm_flush_sta(mvm, mvm_sta, false);
1916 if (ret)
1917 return ret;
1918 if (iwl_mvm_has_new_tx_api(mvm)) {
1919 ret = iwl_mvm_wait_sta_queues_empty(mvm, mvm_sta);
1920 } else {
1921 u32 q_mask = mvm_sta->tfd_queue_msk;
1922
1923 ret = iwl_trans_wait_tx_queues_empty(mvm->trans,
1924 q_mask);
1925 }
1926 if (ret)
1927 return ret;
1928
1929 ret = iwl_mvm_drain_sta(mvm, mvm_sta, false);
1930
1931 iwl_mvm_disable_sta_queues(mvm, vif, sta);
1932
1933 /* If there is a TXQ still marked as reserved - free it */
1934 if (mvm_sta->reserved_queue != IEEE80211_INVAL_HW_QUEUE) {
1935 u8 reserved_txq = mvm_sta->reserved_queue;
1936 enum iwl_mvm_queue_status *status;
1937
1938 /*
1939 * If no traffic has gone through the reserved TXQ - it
1940 * is still marked as IWL_MVM_QUEUE_RESERVED, and
1941 * should be manually marked as free again
1942 */
1943 status = &mvm->queue_info[reserved_txq].status;
1944 if (WARN((*status != IWL_MVM_QUEUE_RESERVED) &&
1945 (*status != IWL_MVM_QUEUE_FREE),
1946 "sta_id %d reserved txq %d status %d",
1947 sta_id, reserved_txq, *status))
1948 return -EINVAL;
1949
1950 *status = IWL_MVM_QUEUE_FREE;
1951 }
1952
1953 if (vif->type == NL80211_IFTYPE_STATION &&
1954 mvmvif->ap_sta_id == sta_id) {
1955 /* if associated - we can't remove the AP STA now */
1956 if (vif->cfg.assoc)
1957 return ret;
1958
1959 /* first remove remaining keys */
1960 iwl_mvm_sec_key_remove_ap(mvm, vif);
1961
1962 /* unassoc - go ahead - remove the AP STA now */
1963 mvmvif->ap_sta_id = IWL_MVM_INVALID_STA;
1964 }
1965
1966 /*
1967 * This shouldn't happen - the TDLS channel switch should be canceled
1968 * before the STA is removed.
1969 */
1970 if (WARN_ON_ONCE(mvm->tdls_cs.peer.sta_id == sta_id)) {
1971 mvm->tdls_cs.peer.sta_id = IWL_MVM_INVALID_STA;
1972 cancel_delayed_work(&mvm->tdls_cs.dwork);
1973 }
1974
1975 /*
1976 * Make sure that the tx response code sees the station as -EBUSY and
1977 * calls the drain worker.
1978 */
1979 spin_lock_bh(&mvm_sta->lock);
1980 spin_unlock_bh(&mvm_sta->lock);
1981
1982 ret = iwl_mvm_rm_sta_common(mvm, mvm_sta->sta_id);
1983 RCU_INIT_POINTER(mvm->fw_id_to_mac_id[mvm_sta->sta_id], NULL);
1984
1985 return ret;
1986}
1987
1988int iwl_mvm_rm_sta_id(struct iwl_mvm *mvm,
1989 struct ieee80211_vif *vif,
1990 u8 sta_id)
1991{
1992 int ret = iwl_mvm_rm_sta_common(mvm, sta_id);
1993
1994 lockdep_assert_held(&mvm->mutex);
1995
1996 RCU_INIT_POINTER(mvm->fw_id_to_mac_id[sta_id], NULL);
1997 return ret;
1998}
1999
2000int iwl_mvm_allocate_int_sta(struct iwl_mvm *mvm,
2001 struct iwl_mvm_int_sta *sta,
2002 u32 qmask, enum nl80211_iftype iftype,
2003 enum iwl_sta_type type)
2004{
2005 if (!test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status) ||
2006 sta->sta_id == IWL_MVM_INVALID_STA) {
2007 sta->sta_id = iwl_mvm_find_free_sta_id(mvm, iftype);
2008 if (WARN_ON_ONCE(sta->sta_id == IWL_MVM_INVALID_STA))
2009 return -ENOSPC;
2010 }
2011
2012 sta->tfd_queue_msk = qmask;
2013 sta->type = type;
2014
2015 /* put a non-NULL value so iterating over the stations won't stop */
2016 rcu_assign_pointer(mvm->fw_id_to_mac_id[sta->sta_id], ERR_PTR(-EINVAL));
2017 return 0;
2018}
2019
2020void iwl_mvm_dealloc_int_sta(struct iwl_mvm *mvm, struct iwl_mvm_int_sta *sta)
2021{
2022 RCU_INIT_POINTER(mvm->fw_id_to_mac_id[sta->sta_id], NULL);
2023 memset(sta, 0, sizeof(struct iwl_mvm_int_sta));
2024 sta->sta_id = IWL_MVM_INVALID_STA;
2025}
2026
2027static void iwl_mvm_enable_aux_snif_queue(struct iwl_mvm *mvm, u16 queue,
2028 u8 sta_id, u8 fifo)
2029{
2030 unsigned int wdg_timeout =
2031 mvm->trans->trans_cfg->base_params->wd_timeout;
2032 struct iwl_trans_txq_scd_cfg cfg = {
2033 .fifo = fifo,
2034 .sta_id = sta_id,
2035 .tid = IWL_MAX_TID_COUNT,
2036 .aggregate = false,
2037 .frame_limit = IWL_FRAME_LIMIT,
2038 };
2039
2040 WARN_ON(iwl_mvm_has_new_tx_api(mvm));
2041
2042 iwl_mvm_enable_txq(mvm, NULL, queue, 0, &cfg, wdg_timeout);
2043}
2044
2045static int iwl_mvm_enable_aux_snif_queue_tvqm(struct iwl_mvm *mvm, u8 sta_id)
2046{
2047 unsigned int wdg_timeout =
2048 mvm->trans->trans_cfg->base_params->wd_timeout;
2049
2050 WARN_ON(!iwl_mvm_has_new_tx_api(mvm));
2051
2052 return iwl_mvm_tvqm_enable_txq(mvm, sta_id, IWL_MAX_TID_COUNT,
2053 wdg_timeout);
2054}
2055
2056static int iwl_mvm_add_int_sta_with_queue(struct iwl_mvm *mvm, int macidx,
2057 int maccolor, u8 *addr,
2058 struct iwl_mvm_int_sta *sta,
2059 u16 *queue, int fifo)
2060{
2061 int ret;
2062
2063 /* Map queue to fifo - needs to happen before adding station */
2064 if (!iwl_mvm_has_new_tx_api(mvm))
2065 iwl_mvm_enable_aux_snif_queue(mvm, *queue, sta->sta_id, fifo);
2066
2067 ret = iwl_mvm_add_int_sta_common(mvm, sta, addr, macidx, maccolor);
2068 if (ret) {
2069 if (!iwl_mvm_has_new_tx_api(mvm))
2070 iwl_mvm_disable_txq(mvm, NULL, sta->sta_id, queue,
2071 IWL_MAX_TID_COUNT);
2072 return ret;
2073 }
2074
2075 /*
2076 * For 22000 firmware and on we cannot add queue to a station unknown
2077 * to firmware so enable queue here - after the station was added
2078 */
2079 if (iwl_mvm_has_new_tx_api(mvm)) {
2080 int txq;
2081
2082 txq = iwl_mvm_enable_aux_snif_queue_tvqm(mvm, sta->sta_id);
2083 if (txq < 0) {
2084 iwl_mvm_rm_sta_common(mvm, sta->sta_id);
2085 return txq;
2086 }
2087
2088 *queue = txq;
2089 }
2090
2091 return 0;
2092}
2093
2094int iwl_mvm_add_aux_sta(struct iwl_mvm *mvm, u32 lmac_id)
2095{
2096 int ret;
2097
2098 lockdep_assert_held(&mvm->mutex);
2099
2100 /* Allocate aux station and assign to it the aux queue */
2101 ret = iwl_mvm_allocate_int_sta(mvm, &mvm->aux_sta, BIT(mvm->aux_queue),
2102 NL80211_IFTYPE_UNSPECIFIED,
2103 IWL_STA_AUX_ACTIVITY);
2104 if (ret)
2105 return ret;
2106
2107 /*
2108 * In CDB NICs we need to specify which lmac to use for aux activity
2109 * using the mac_id argument place to send lmac_id to the function
2110 */
2111 ret = iwl_mvm_add_int_sta_with_queue(mvm, lmac_id, 0, NULL,
2112 &mvm->aux_sta, &mvm->aux_queue,
2113 IWL_MVM_TX_FIFO_MCAST);
2114 if (ret) {
2115 iwl_mvm_dealloc_int_sta(mvm, &mvm->aux_sta);
2116 return ret;
2117 }
2118
2119 return 0;
2120}
2121
2122int iwl_mvm_add_snif_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
2123{
2124 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2125
2126 lockdep_assert_held(&mvm->mutex);
2127
2128 return iwl_mvm_add_int_sta_with_queue(mvm, mvmvif->id, mvmvif->color,
2129 NULL, &mvm->snif_sta,
2130 &mvm->snif_queue,
2131 IWL_MVM_TX_FIFO_BE);
2132}
2133
2134int iwl_mvm_rm_snif_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
2135{
2136 int ret;
2137
2138 lockdep_assert_held(&mvm->mutex);
2139
2140 if (WARN_ON_ONCE(mvm->snif_sta.sta_id == IWL_MVM_INVALID_STA))
2141 return -EINVAL;
2142
2143 iwl_mvm_disable_txq(mvm, NULL, mvm->snif_sta.sta_id,
2144 &mvm->snif_queue, IWL_MAX_TID_COUNT);
2145 ret = iwl_mvm_rm_sta_common(mvm, mvm->snif_sta.sta_id);
2146 if (ret)
2147 IWL_WARN(mvm, "Failed sending remove station\n");
2148
2149 return ret;
2150}
2151
2152int iwl_mvm_rm_aux_sta(struct iwl_mvm *mvm)
2153{
2154 int ret;
2155
2156 lockdep_assert_held(&mvm->mutex);
2157
2158 if (WARN_ON_ONCE(mvm->aux_sta.sta_id == IWL_MVM_INVALID_STA))
2159 return -EINVAL;
2160
2161 iwl_mvm_disable_txq(mvm, NULL, mvm->aux_sta.sta_id,
2162 &mvm->aux_queue, IWL_MAX_TID_COUNT);
2163 ret = iwl_mvm_rm_sta_common(mvm, mvm->aux_sta.sta_id);
2164 if (ret)
2165 IWL_WARN(mvm, "Failed sending remove station\n");
2166 iwl_mvm_dealloc_int_sta(mvm, &mvm->aux_sta);
2167
2168 return ret;
2169}
2170
2171void iwl_mvm_dealloc_snif_sta(struct iwl_mvm *mvm)
2172{
2173 iwl_mvm_dealloc_int_sta(mvm, &mvm->snif_sta);
2174}
2175
2176/*
2177 * Send the add station command for the vif's broadcast station.
2178 * Assumes that the station was already allocated.
2179 *
2180 * @mvm: the mvm component
2181 * @vif: the interface to which the broadcast station is added
2182 * @bsta: the broadcast station to add.
2183 */
2184int iwl_mvm_send_add_bcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
2185{
2186 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2187 struct iwl_mvm_int_sta *bsta = &mvmvif->bcast_sta;
2188 static const u8 _baddr[] = {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF};
2189 const u8 *baddr = _baddr;
2190 int queue;
2191 int ret;
2192 unsigned int wdg_timeout =
2193 iwl_mvm_get_wd_timeout(mvm, vif, false, false);
2194 struct iwl_trans_txq_scd_cfg cfg = {
2195 .fifo = IWL_MVM_TX_FIFO_VO,
2196 .sta_id = mvmvif->bcast_sta.sta_id,
2197 .tid = IWL_MAX_TID_COUNT,
2198 .aggregate = false,
2199 .frame_limit = IWL_FRAME_LIMIT,
2200 };
2201
2202 lockdep_assert_held(&mvm->mutex);
2203
2204 if (!iwl_mvm_has_new_tx_api(mvm)) {
2205 if (vif->type == NL80211_IFTYPE_AP ||
2206 vif->type == NL80211_IFTYPE_ADHOC) {
2207 queue = mvm->probe_queue;
2208 } else if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
2209 queue = mvm->p2p_dev_queue;
2210 } else {
2211 WARN(1, "Missing required TXQ for adding bcast STA\n");
2212 return -EINVAL;
2213 }
2214
2215 bsta->tfd_queue_msk |= BIT(queue);
2216
2217 iwl_mvm_enable_txq(mvm, NULL, queue, 0, &cfg, wdg_timeout);
2218 }
2219
2220 if (vif->type == NL80211_IFTYPE_ADHOC)
2221 baddr = vif->bss_conf.bssid;
2222
2223 if (WARN_ON_ONCE(bsta->sta_id == IWL_MVM_INVALID_STA))
2224 return -ENOSPC;
2225
2226 ret = iwl_mvm_add_int_sta_common(mvm, bsta, baddr,
2227 mvmvif->id, mvmvif->color);
2228 if (ret)
2229 return ret;
2230
2231 /*
2232 * For 22000 firmware and on we cannot add queue to a station unknown
2233 * to firmware so enable queue here - after the station was added
2234 */
2235 if (iwl_mvm_has_new_tx_api(mvm)) {
2236 queue = iwl_mvm_tvqm_enable_txq(mvm, bsta->sta_id,
2237 IWL_MAX_TID_COUNT,
2238 wdg_timeout);
2239 if (queue < 0) {
2240 iwl_mvm_rm_sta_common(mvm, bsta->sta_id);
2241 return queue;
2242 }
2243
2244 if (vif->type == NL80211_IFTYPE_AP ||
2245 vif->type == NL80211_IFTYPE_ADHOC)
2246 mvm->probe_queue = queue;
2247 else if (vif->type == NL80211_IFTYPE_P2P_DEVICE)
2248 mvm->p2p_dev_queue = queue;
2249 }
2250
2251 return 0;
2252}
2253
2254static void iwl_mvm_free_bcast_sta_queues(struct iwl_mvm *mvm,
2255 struct ieee80211_vif *vif)
2256{
2257 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2258 u16 *queueptr, queue;
2259
2260 lockdep_assert_held(&mvm->mutex);
2261
2262 iwl_mvm_flush_sta(mvm, &mvmvif->bcast_sta, true);
2263
2264 switch (vif->type) {
2265 case NL80211_IFTYPE_AP:
2266 case NL80211_IFTYPE_ADHOC:
2267 queueptr = &mvm->probe_queue;
2268 break;
2269 case NL80211_IFTYPE_P2P_DEVICE:
2270 queueptr = &mvm->p2p_dev_queue;
2271 break;
2272 default:
2273 WARN(1, "Can't free bcast queue on vif type %d\n",
2274 vif->type);
2275 return;
2276 }
2277
2278 queue = *queueptr;
2279 iwl_mvm_disable_txq(mvm, NULL, mvmvif->bcast_sta.sta_id,
2280 queueptr, IWL_MAX_TID_COUNT);
2281 if (iwl_mvm_has_new_tx_api(mvm))
2282 return;
2283
2284 WARN_ON(!(mvmvif->bcast_sta.tfd_queue_msk & BIT(queue)));
2285 mvmvif->bcast_sta.tfd_queue_msk &= ~BIT(queue);
2286}
2287
2288/* Send the FW a request to remove the station from it's internal data
2289 * structures, but DO NOT remove the entry from the local data structures. */
2290int iwl_mvm_send_rm_bcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
2291{
2292 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2293 int ret;
2294
2295 lockdep_assert_held(&mvm->mutex);
2296
2297 iwl_mvm_free_bcast_sta_queues(mvm, vif);
2298
2299 ret = iwl_mvm_rm_sta_common(mvm, mvmvif->bcast_sta.sta_id);
2300 if (ret)
2301 IWL_WARN(mvm, "Failed sending remove station\n");
2302 return ret;
2303}
2304
2305int iwl_mvm_alloc_bcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
2306{
2307 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2308
2309 lockdep_assert_held(&mvm->mutex);
2310
2311 return iwl_mvm_allocate_int_sta(mvm, &mvmvif->bcast_sta, 0,
2312 ieee80211_vif_type_p2p(vif),
2313 IWL_STA_GENERAL_PURPOSE);
2314}
2315
2316/* Allocate a new station entry for the broadcast station to the given vif,
2317 * and send it to the FW.
2318 * Note that each P2P mac should have its own broadcast station.
2319 *
2320 * @mvm: the mvm component
2321 * @vif: the interface to which the broadcast station is added
2322 * @bsta: the broadcast station to add. */
2323int iwl_mvm_add_p2p_bcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
2324{
2325 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2326 struct iwl_mvm_int_sta *bsta = &mvmvif->bcast_sta;
2327 int ret;
2328
2329 lockdep_assert_held(&mvm->mutex);
2330
2331 ret = iwl_mvm_alloc_bcast_sta(mvm, vif);
2332 if (ret)
2333 return ret;
2334
2335 ret = iwl_mvm_send_add_bcast_sta(mvm, vif);
2336
2337 if (ret)
2338 iwl_mvm_dealloc_int_sta(mvm, bsta);
2339
2340 return ret;
2341}
2342
2343void iwl_mvm_dealloc_bcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
2344{
2345 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2346
2347 iwl_mvm_dealloc_int_sta(mvm, &mvmvif->bcast_sta);
2348}
2349
2350/*
2351 * Send the FW a request to remove the station from it's internal data
2352 * structures, and in addition remove it from the local data structure.
2353 */
2354int iwl_mvm_rm_p2p_bcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
2355{
2356 int ret;
2357
2358 lockdep_assert_held(&mvm->mutex);
2359
2360 ret = iwl_mvm_send_rm_bcast_sta(mvm, vif);
2361
2362 iwl_mvm_dealloc_bcast_sta(mvm, vif);
2363
2364 return ret;
2365}
2366
2367/*
2368 * Allocate a new station entry for the multicast station to the given vif,
2369 * and send it to the FW.
2370 * Note that each AP/GO mac should have its own multicast station.
2371 *
2372 * @mvm: the mvm component
2373 * @vif: the interface to which the multicast station is added
2374 */
2375int iwl_mvm_add_mcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
2376{
2377 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2378 struct iwl_mvm_int_sta *msta = &mvmvif->mcast_sta;
2379 static const u8 _maddr[] = {0x03, 0x00, 0x00, 0x00, 0x00, 0x00};
2380 const u8 *maddr = _maddr;
2381 struct iwl_trans_txq_scd_cfg cfg = {
2382 .fifo = vif->type == NL80211_IFTYPE_AP ?
2383 IWL_MVM_TX_FIFO_MCAST : IWL_MVM_TX_FIFO_BE,
2384 .sta_id = msta->sta_id,
2385 .tid = 0,
2386 .aggregate = false,
2387 .frame_limit = IWL_FRAME_LIMIT,
2388 };
2389 unsigned int timeout = iwl_mvm_get_wd_timeout(mvm, vif, false, false);
2390 int ret;
2391
2392 lockdep_assert_held(&mvm->mutex);
2393
2394 if (WARN_ON(vif->type != NL80211_IFTYPE_AP &&
2395 vif->type != NL80211_IFTYPE_ADHOC))
2396 return -ENOTSUPP;
2397
2398 /*
2399 * In IBSS, ieee80211_check_queues() sets the cab_queue to be
2400 * invalid, so make sure we use the queue we want.
2401 * Note that this is done here as we want to avoid making DQA
2402 * changes in mac80211 layer.
2403 */
2404 if (vif->type == NL80211_IFTYPE_ADHOC)
2405 mvmvif->cab_queue = IWL_MVM_DQA_GCAST_QUEUE;
2406
2407 /*
2408 * While in previous FWs we had to exclude cab queue from TFD queue
2409 * mask, now it is needed as any other queue.
2410 */
2411 if (!iwl_mvm_has_new_tx_api(mvm) &&
2412 fw_has_api(&mvm->fw->ucode_capa, IWL_UCODE_TLV_API_STA_TYPE)) {
2413 iwl_mvm_enable_txq(mvm, NULL, mvmvif->cab_queue, 0, &cfg,
2414 timeout);
2415 msta->tfd_queue_msk |= BIT(mvmvif->cab_queue);
2416 }
2417 ret = iwl_mvm_add_int_sta_common(mvm, msta, maddr,
2418 mvmvif->id, mvmvif->color);
2419 if (ret)
2420 goto err;
2421
2422 /*
2423 * Enable cab queue after the ADD_STA command is sent.
2424 * This is needed for 22000 firmware which won't accept SCD_QUEUE_CFG
2425 * command with unknown station id, and for FW that doesn't support
2426 * station API since the cab queue is not included in the
2427 * tfd_queue_mask.
2428 */
2429 if (iwl_mvm_has_new_tx_api(mvm)) {
2430 int queue = iwl_mvm_tvqm_enable_txq(mvm, msta->sta_id,
2431 0,
2432 timeout);
2433 if (queue < 0) {
2434 ret = queue;
2435 goto err;
2436 }
2437 mvmvif->cab_queue = queue;
2438 } else if (!fw_has_api(&mvm->fw->ucode_capa,
2439 IWL_UCODE_TLV_API_STA_TYPE))
2440 iwl_mvm_enable_txq(mvm, NULL, mvmvif->cab_queue, 0, &cfg,
2441 timeout);
2442
2443 return 0;
2444err:
2445 iwl_mvm_dealloc_int_sta(mvm, msta);
2446 return ret;
2447}
2448
2449static int __iwl_mvm_remove_sta_key(struct iwl_mvm *mvm, u8 sta_id,
2450 struct ieee80211_key_conf *keyconf,
2451 bool mcast)
2452{
2453 union {
2454 struct iwl_mvm_add_sta_key_cmd_v1 cmd_v1;
2455 struct iwl_mvm_add_sta_key_cmd cmd;
2456 } u = {};
2457 bool new_api = fw_has_api(&mvm->fw->ucode_capa,
2458 IWL_UCODE_TLV_API_TKIP_MIC_KEYS);
2459 __le16 key_flags;
2460 int ret, size;
2461 u32 status;
2462
2463 /* This is a valid situation for GTK removal */
2464 if (sta_id == IWL_MVM_INVALID_STA)
2465 return 0;
2466
2467 key_flags = cpu_to_le16((keyconf->keyidx << STA_KEY_FLG_KEYID_POS) &
2468 STA_KEY_FLG_KEYID_MSK);
2469 key_flags |= cpu_to_le16(STA_KEY_FLG_NO_ENC | STA_KEY_FLG_WEP_KEY_MAP);
2470 key_flags |= cpu_to_le16(STA_KEY_NOT_VALID);
2471
2472 if (mcast)
2473 key_flags |= cpu_to_le16(STA_KEY_MULTICAST);
2474
2475 /*
2476 * The fields assigned here are in the same location at the start
2477 * of the command, so we can do this union trick.
2478 */
2479 u.cmd.common.key_flags = key_flags;
2480 u.cmd.common.key_offset = keyconf->hw_key_idx;
2481 u.cmd.common.sta_id = sta_id;
2482
2483 size = new_api ? sizeof(u.cmd) : sizeof(u.cmd_v1);
2484
2485 status = ADD_STA_SUCCESS;
2486 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA_KEY, size, &u.cmd,
2487 &status);
2488
2489 switch (status) {
2490 case ADD_STA_SUCCESS:
2491 IWL_DEBUG_WEP(mvm, "MODIFY_STA: remove sta key passed\n");
2492 break;
2493 default:
2494 ret = -EIO;
2495 IWL_ERR(mvm, "MODIFY_STA: remove sta key failed\n");
2496 break;
2497 }
2498
2499 return ret;
2500}
2501
2502/*
2503 * Send the FW a request to remove the station from it's internal data
2504 * structures, and in addition remove it from the local data structure.
2505 */
2506int iwl_mvm_rm_mcast_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
2507{
2508 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
2509 int ret;
2510
2511 lockdep_assert_held(&mvm->mutex);
2512
2513 iwl_mvm_flush_sta(mvm, &mvmvif->mcast_sta, true);
2514
2515 iwl_mvm_disable_txq(mvm, NULL, mvmvif->mcast_sta.sta_id,
2516 &mvmvif->cab_queue, 0);
2517
2518 ret = iwl_mvm_rm_sta_common(mvm, mvmvif->mcast_sta.sta_id);
2519 if (ret)
2520 IWL_WARN(mvm, "Failed sending remove station\n");
2521
2522 return ret;
2523}
2524
2525static void iwl_mvm_sync_rxq_del_ba(struct iwl_mvm *mvm, u8 baid)
2526{
2527 struct iwl_mvm_delba_data notif = {
2528 .baid = baid,
2529 };
2530
2531 iwl_mvm_sync_rx_queues_internal(mvm, IWL_MVM_RXQ_NOTIF_DEL_BA, true,
2532 ¬if, sizeof(notif));
2533};
2534
2535static void iwl_mvm_free_reorder(struct iwl_mvm *mvm,
2536 struct iwl_mvm_baid_data *data)
2537{
2538 int i;
2539
2540 iwl_mvm_sync_rxq_del_ba(mvm, data->baid);
2541
2542 for (i = 0; i < mvm->trans->num_rx_queues; i++) {
2543 int j;
2544 struct iwl_mvm_reorder_buffer *reorder_buf =
2545 &data->reorder_buf[i];
2546 struct iwl_mvm_reorder_buf_entry *entries =
2547 &data->entries[i * data->entries_per_queue];
2548
2549 spin_lock_bh(&reorder_buf->lock);
2550 if (likely(!reorder_buf->num_stored)) {
2551 spin_unlock_bh(&reorder_buf->lock);
2552 continue;
2553 }
2554
2555 /*
2556 * This shouldn't happen in regular DELBA since the internal
2557 * delBA notification should trigger a release of all frames in
2558 * the reorder buffer.
2559 */
2560 WARN_ON(1);
2561
2562 for (j = 0; j < reorder_buf->buf_size; j++)
2563 __skb_queue_purge(&entries[j].e.frames);
2564 /*
2565 * Prevent timer re-arm. This prevents a very far fetched case
2566 * where we timed out on the notification. There may be prior
2567 * RX frames pending in the RX queue before the notification
2568 * that might get processed between now and the actual deletion
2569 * and we would re-arm the timer although we are deleting the
2570 * reorder buffer.
2571 */
2572 reorder_buf->removed = true;
2573 spin_unlock_bh(&reorder_buf->lock);
2574 del_timer_sync(&reorder_buf->reorder_timer);
2575 }
2576}
2577
2578static void iwl_mvm_init_reorder_buffer(struct iwl_mvm *mvm,
2579 struct iwl_mvm_baid_data *data,
2580 u16 ssn, u16 buf_size)
2581{
2582 int i;
2583
2584 for (i = 0; i < mvm->trans->num_rx_queues; i++) {
2585 struct iwl_mvm_reorder_buffer *reorder_buf =
2586 &data->reorder_buf[i];
2587 struct iwl_mvm_reorder_buf_entry *entries =
2588 &data->entries[i * data->entries_per_queue];
2589 int j;
2590
2591 reorder_buf->num_stored = 0;
2592 reorder_buf->head_sn = ssn;
2593 reorder_buf->buf_size = buf_size;
2594 /* rx reorder timer */
2595 timer_setup(&reorder_buf->reorder_timer,
2596 iwl_mvm_reorder_timer_expired, 0);
2597 spin_lock_init(&reorder_buf->lock);
2598 reorder_buf->mvm = mvm;
2599 reorder_buf->queue = i;
2600 reorder_buf->valid = false;
2601 for (j = 0; j < reorder_buf->buf_size; j++)
2602 __skb_queue_head_init(&entries[j].e.frames);
2603 }
2604}
2605
2606static int iwl_mvm_fw_baid_op_sta(struct iwl_mvm *mvm,
2607 struct iwl_mvm_sta *mvm_sta,
2608 bool start, int tid, u16 ssn,
2609 u16 buf_size)
2610{
2611 struct iwl_mvm_add_sta_cmd cmd = {
2612 .mac_id_n_color = cpu_to_le32(mvm_sta->mac_id_n_color),
2613 .sta_id = mvm_sta->sta_id,
2614 .add_modify = STA_MODE_MODIFY,
2615 };
2616 u32 status;
2617 int ret;
2618
2619 if (start) {
2620 cmd.add_immediate_ba_tid = tid;
2621 cmd.add_immediate_ba_ssn = cpu_to_le16(ssn);
2622 cmd.rx_ba_window = cpu_to_le16(buf_size);
2623 cmd.modify_mask = STA_MODIFY_ADD_BA_TID;
2624 } else {
2625 cmd.remove_immediate_ba_tid = tid;
2626 cmd.modify_mask = STA_MODIFY_REMOVE_BA_TID;
2627 }
2628
2629 status = ADD_STA_SUCCESS;
2630 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA,
2631 iwl_mvm_add_sta_cmd_size(mvm),
2632 &cmd, &status);
2633 if (ret)
2634 return ret;
2635
2636 switch (status & IWL_ADD_STA_STATUS_MASK) {
2637 case ADD_STA_SUCCESS:
2638 IWL_DEBUG_HT(mvm, "RX BA Session %sed in fw\n",
2639 start ? "start" : "stopp");
2640 if (WARN_ON(start && iwl_mvm_has_new_rx_api(mvm) &&
2641 !(status & IWL_ADD_STA_BAID_VALID_MASK)))
2642 return -EINVAL;
2643 return u32_get_bits(status, IWL_ADD_STA_BAID_MASK);
2644 case ADD_STA_IMMEDIATE_BA_FAILURE:
2645 IWL_WARN(mvm, "RX BA Session refused by fw\n");
2646 return -ENOSPC;
2647 default:
2648 IWL_ERR(mvm, "RX BA Session failed %sing, status 0x%x\n",
2649 start ? "start" : "stopp", status);
2650 return -EIO;
2651 }
2652}
2653
2654static int iwl_mvm_fw_baid_op_cmd(struct iwl_mvm *mvm,
2655 struct iwl_mvm_sta *mvm_sta,
2656 bool start, int tid, u16 ssn,
2657 u16 buf_size, int baid)
2658{
2659 struct iwl_rx_baid_cfg_cmd cmd = {
2660 .action = start ? cpu_to_le32(IWL_RX_BAID_ACTION_ADD) :
2661 cpu_to_le32(IWL_RX_BAID_ACTION_REMOVE),
2662 };
2663 u32 cmd_id = WIDE_ID(DATA_PATH_GROUP, RX_BAID_ALLOCATION_CONFIG_CMD);
2664 int ret;
2665
2666 BUILD_BUG_ON(sizeof(struct iwl_rx_baid_cfg_resp) != sizeof(baid));
2667
2668 if (start) {
2669 cmd.alloc.sta_id_mask = cpu_to_le32(BIT(mvm_sta->sta_id));
2670 cmd.alloc.tid = tid;
2671 cmd.alloc.ssn = cpu_to_le16(ssn);
2672 cmd.alloc.win_size = cpu_to_le16(buf_size);
2673 baid = -EIO;
2674 } else if (iwl_fw_lookup_cmd_ver(mvm->fw, cmd_id, 1) == 1) {
2675 cmd.remove_v1.baid = cpu_to_le32(baid);
2676 BUILD_BUG_ON(sizeof(cmd.remove_v1) > sizeof(cmd.remove));
2677 } else {
2678 cmd.remove.sta_id_mask = cpu_to_le32(BIT(mvm_sta->sta_id));
2679 cmd.remove.tid = cpu_to_le32(tid);
2680 }
2681
2682 ret = iwl_mvm_send_cmd_pdu_status(mvm, cmd_id, sizeof(cmd),
2683 &cmd, &baid);
2684 if (ret)
2685 return ret;
2686
2687 if (!start) {
2688 /* ignore firmware baid on remove */
2689 baid = 0;
2690 }
2691
2692 IWL_DEBUG_HT(mvm, "RX BA Session %sed in fw\n",
2693 start ? "start" : "stopp");
2694
2695 if (baid < 0 || baid >= ARRAY_SIZE(mvm->baid_map))
2696 return -EINVAL;
2697
2698 return baid;
2699}
2700
2701static int iwl_mvm_fw_baid_op(struct iwl_mvm *mvm, struct iwl_mvm_sta *mvm_sta,
2702 bool start, int tid, u16 ssn, u16 buf_size,
2703 int baid)
2704{
2705 if (fw_has_capa(&mvm->fw->ucode_capa,
2706 IWL_UCODE_TLV_CAPA_BAID_ML_SUPPORT))
2707 return iwl_mvm_fw_baid_op_cmd(mvm, mvm_sta, start,
2708 tid, ssn, buf_size, baid);
2709
2710 return iwl_mvm_fw_baid_op_sta(mvm, mvm_sta, start,
2711 tid, ssn, buf_size);
2712}
2713
2714int iwl_mvm_sta_rx_agg(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
2715 int tid, u16 ssn, bool start, u16 buf_size, u16 timeout)
2716{
2717 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
2718 struct iwl_mvm_baid_data *baid_data = NULL;
2719 int ret, baid;
2720 u32 max_ba_id_sessions = iwl_mvm_has_new_tx_api(mvm) ? IWL_MAX_BAID :
2721 IWL_MAX_BAID_OLD;
2722
2723 lockdep_assert_held(&mvm->mutex);
2724
2725 if (start && mvm->rx_ba_sessions >= max_ba_id_sessions) {
2726 IWL_WARN(mvm, "Not enough RX BA SESSIONS\n");
2727 return -ENOSPC;
2728 }
2729
2730 if (iwl_mvm_has_new_rx_api(mvm) && start) {
2731 u16 reorder_buf_size = buf_size * sizeof(baid_data->entries[0]);
2732
2733 /* sparse doesn't like the __align() so don't check */
2734#ifndef __CHECKER__
2735 /*
2736 * The division below will be OK if either the cache line size
2737 * can be divided by the entry size (ALIGN will round up) or if
2738 * if the entry size can be divided by the cache line size, in
2739 * which case the ALIGN() will do nothing.
2740 */
2741 BUILD_BUG_ON(SMP_CACHE_BYTES % sizeof(baid_data->entries[0]) &&
2742 sizeof(baid_data->entries[0]) % SMP_CACHE_BYTES);
2743#endif
2744
2745 /*
2746 * Upward align the reorder buffer size to fill an entire cache
2747 * line for each queue, to avoid sharing cache lines between
2748 * different queues.
2749 */
2750 reorder_buf_size = ALIGN(reorder_buf_size, SMP_CACHE_BYTES);
2751
2752 /*
2753 * Allocate here so if allocation fails we can bail out early
2754 * before starting the BA session in the firmware
2755 */
2756 baid_data = kzalloc(sizeof(*baid_data) +
2757 mvm->trans->num_rx_queues *
2758 reorder_buf_size,
2759 GFP_KERNEL);
2760 if (!baid_data)
2761 return -ENOMEM;
2762
2763 /*
2764 * This division is why we need the above BUILD_BUG_ON(),
2765 * if that doesn't hold then this will not be right.
2766 */
2767 baid_data->entries_per_queue =
2768 reorder_buf_size / sizeof(baid_data->entries[0]);
2769 }
2770
2771 if (iwl_mvm_has_new_rx_api(mvm) && !start) {
2772 baid = mvm_sta->tid_to_baid[tid];
2773 } else {
2774 /* we don't really need it in this case */
2775 baid = -1;
2776 }
2777
2778 /* Don't send command to remove (start=0) BAID during restart */
2779 if (start || !test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status))
2780 baid = iwl_mvm_fw_baid_op(mvm, mvm_sta, start, tid, ssn, buf_size,
2781 baid);
2782
2783 if (baid < 0) {
2784 ret = baid;
2785 goto out_free;
2786 }
2787
2788 if (start) {
2789 mvm->rx_ba_sessions++;
2790
2791 if (!iwl_mvm_has_new_rx_api(mvm))
2792 return 0;
2793
2794 baid_data->baid = baid;
2795 baid_data->timeout = timeout;
2796 baid_data->last_rx = jiffies;
2797 baid_data->rcu_ptr = &mvm->baid_map[baid];
2798 timer_setup(&baid_data->session_timer,
2799 iwl_mvm_rx_agg_session_expired, 0);
2800 baid_data->mvm = mvm;
2801 baid_data->tid = tid;
2802 baid_data->sta_id = mvm_sta->sta_id;
2803
2804 mvm_sta->tid_to_baid[tid] = baid;
2805 if (timeout)
2806 mod_timer(&baid_data->session_timer,
2807 TU_TO_EXP_TIME(timeout * 2));
2808
2809 iwl_mvm_init_reorder_buffer(mvm, baid_data, ssn, buf_size);
2810 /*
2811 * protect the BA data with RCU to cover a case where our
2812 * internal RX sync mechanism will timeout (not that it's
2813 * supposed to happen) and we will free the session data while
2814 * RX is being processed in parallel
2815 */
2816 IWL_DEBUG_HT(mvm, "Sta %d(%d) is assigned to BAID %d\n",
2817 mvm_sta->sta_id, tid, baid);
2818 WARN_ON(rcu_access_pointer(mvm->baid_map[baid]));
2819 rcu_assign_pointer(mvm->baid_map[baid], baid_data);
2820 } else {
2821 baid = mvm_sta->tid_to_baid[tid];
2822
2823 if (mvm->rx_ba_sessions > 0)
2824 /* check that restart flow didn't zero the counter */
2825 mvm->rx_ba_sessions--;
2826 if (!iwl_mvm_has_new_rx_api(mvm))
2827 return 0;
2828
2829 if (WARN_ON(baid == IWL_RX_REORDER_DATA_INVALID_BAID))
2830 return -EINVAL;
2831
2832 baid_data = rcu_access_pointer(mvm->baid_map[baid]);
2833 if (WARN_ON(!baid_data))
2834 return -EINVAL;
2835
2836 /* synchronize all rx queues so we can safely delete */
2837 iwl_mvm_free_reorder(mvm, baid_data);
2838 timer_shutdown_sync(&baid_data->session_timer);
2839 RCU_INIT_POINTER(mvm->baid_map[baid], NULL);
2840 kfree_rcu(baid_data, rcu_head);
2841 IWL_DEBUG_HT(mvm, "BAID %d is free\n", baid);
2842
2843 /*
2844 * After we've deleted it, do another queue sync
2845 * so if an IWL_MVM_RXQ_NSSN_SYNC was concurrently
2846 * running it won't find a new session in the old
2847 * BAID. It can find the NULL pointer for the BAID,
2848 * but we must not have it find a different session.
2849 */
2850 iwl_mvm_sync_rx_queues_internal(mvm, IWL_MVM_RXQ_EMPTY,
2851 true, NULL, 0);
2852 }
2853 return 0;
2854
2855out_free:
2856 kfree(baid_data);
2857 return ret;
2858}
2859
2860int iwl_mvm_sta_tx_agg(struct iwl_mvm *mvm, struct ieee80211_sta *sta,
2861 int tid, u8 queue, bool start)
2862{
2863 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
2864 struct iwl_mvm_add_sta_cmd cmd = {};
2865 int ret;
2866 u32 status;
2867
2868 lockdep_assert_held(&mvm->mutex);
2869
2870 if (start) {
2871 mvm_sta->tfd_queue_msk |= BIT(queue);
2872 mvm_sta->tid_disable_agg &= ~BIT(tid);
2873 } else {
2874 /* In DQA-mode the queue isn't removed on agg termination */
2875 mvm_sta->tid_disable_agg |= BIT(tid);
2876 }
2877
2878 cmd.mac_id_n_color = cpu_to_le32(mvm_sta->mac_id_n_color);
2879 cmd.sta_id = mvm_sta->sta_id;
2880 cmd.add_modify = STA_MODE_MODIFY;
2881 if (!iwl_mvm_has_new_tx_api(mvm))
2882 cmd.modify_mask = STA_MODIFY_QUEUES;
2883 cmd.modify_mask |= STA_MODIFY_TID_DISABLE_TX;
2884 cmd.tfd_queue_msk = cpu_to_le32(mvm_sta->tfd_queue_msk);
2885 cmd.tid_disable_tx = cpu_to_le16(mvm_sta->tid_disable_agg);
2886
2887 status = ADD_STA_SUCCESS;
2888 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA,
2889 iwl_mvm_add_sta_cmd_size(mvm),
2890 &cmd, &status);
2891 if (ret)
2892 return ret;
2893
2894 switch (status & IWL_ADD_STA_STATUS_MASK) {
2895 case ADD_STA_SUCCESS:
2896 break;
2897 default:
2898 ret = -EIO;
2899 IWL_ERR(mvm, "TX BA Session failed %sing, status 0x%x\n",
2900 start ? "start" : "stopp", status);
2901 break;
2902 }
2903
2904 return ret;
2905}
2906
2907const u8 tid_to_mac80211_ac[] = {
2908 IEEE80211_AC_BE,
2909 IEEE80211_AC_BK,
2910 IEEE80211_AC_BK,
2911 IEEE80211_AC_BE,
2912 IEEE80211_AC_VI,
2913 IEEE80211_AC_VI,
2914 IEEE80211_AC_VO,
2915 IEEE80211_AC_VO,
2916 IEEE80211_AC_VO, /* We treat MGMT as TID 8, which is set as AC_VO */
2917};
2918
2919static const u8 tid_to_ucode_ac[] = {
2920 AC_BE,
2921 AC_BK,
2922 AC_BK,
2923 AC_BE,
2924 AC_VI,
2925 AC_VI,
2926 AC_VO,
2927 AC_VO,
2928};
2929
2930int iwl_mvm_sta_tx_agg_start(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
2931 struct ieee80211_sta *sta, u16 tid, u16 *ssn)
2932{
2933 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
2934 struct iwl_mvm_tid_data *tid_data;
2935 u16 normalized_ssn;
2936 u16 txq_id;
2937 int ret;
2938
2939 if (WARN_ON_ONCE(tid >= IWL_MAX_TID_COUNT))
2940 return -EINVAL;
2941
2942 if (mvmsta->tid_data[tid].state != IWL_AGG_QUEUED &&
2943 mvmsta->tid_data[tid].state != IWL_AGG_OFF) {
2944 IWL_ERR(mvm,
2945 "Start AGG when state is not IWL_AGG_QUEUED or IWL_AGG_OFF %d!\n",
2946 mvmsta->tid_data[tid].state);
2947 return -ENXIO;
2948 }
2949
2950 lockdep_assert_held(&mvm->mutex);
2951
2952 if (mvmsta->tid_data[tid].txq_id == IWL_MVM_INVALID_QUEUE &&
2953 iwl_mvm_has_new_tx_api(mvm)) {
2954 u8 ac = tid_to_mac80211_ac[tid];
2955
2956 ret = iwl_mvm_sta_alloc_queue_tvqm(mvm, sta, ac, tid);
2957 if (ret)
2958 return ret;
2959 }
2960
2961 spin_lock_bh(&mvmsta->lock);
2962
2963 /*
2964 * Note the possible cases:
2965 * 1. An enabled TXQ - TXQ needs to become agg'ed
2966 * 2. The TXQ hasn't yet been enabled, so find a free one and mark
2967 * it as reserved
2968 */
2969 txq_id = mvmsta->tid_data[tid].txq_id;
2970 if (txq_id == IWL_MVM_INVALID_QUEUE) {
2971 ret = iwl_mvm_find_free_queue(mvm, mvmsta->sta_id,
2972 IWL_MVM_DQA_MIN_DATA_QUEUE,
2973 IWL_MVM_DQA_MAX_DATA_QUEUE);
2974 if (ret < 0) {
2975 IWL_ERR(mvm, "Failed to allocate agg queue\n");
2976 goto out;
2977 }
2978
2979 txq_id = ret;
2980
2981 /* TXQ hasn't yet been enabled, so mark it only as reserved */
2982 mvm->queue_info[txq_id].status = IWL_MVM_QUEUE_RESERVED;
2983 } else if (WARN_ON(txq_id >= IWL_MAX_HW_QUEUES)) {
2984 ret = -ENXIO;
2985 IWL_ERR(mvm, "tid_id %d out of range (0, %d)!\n",
2986 tid, IWL_MAX_HW_QUEUES - 1);
2987 goto out;
2988
2989 } else if (unlikely(mvm->queue_info[txq_id].status ==
2990 IWL_MVM_QUEUE_SHARED)) {
2991 ret = -ENXIO;
2992 IWL_DEBUG_TX_QUEUES(mvm,
2993 "Can't start tid %d agg on shared queue!\n",
2994 tid);
2995 goto out;
2996 }
2997
2998 IWL_DEBUG_TX_QUEUES(mvm,
2999 "AGG for tid %d will be on queue #%d\n",
3000 tid, txq_id);
3001
3002 tid_data = &mvmsta->tid_data[tid];
3003 tid_data->ssn = IEEE80211_SEQ_TO_SN(tid_data->seq_number);
3004 tid_data->txq_id = txq_id;
3005 *ssn = tid_data->ssn;
3006
3007 IWL_DEBUG_TX_QUEUES(mvm,
3008 "Start AGG: sta %d tid %d queue %d - ssn = %d, next_recl = %d\n",
3009 mvmsta->sta_id, tid, txq_id, tid_data->ssn,
3010 tid_data->next_reclaimed);
3011
3012 /*
3013 * In 22000 HW, the next_reclaimed index is only 8 bit, so we'll need
3014 * to align the wrap around of ssn so we compare relevant values.
3015 */
3016 normalized_ssn = tid_data->ssn;
3017 if (mvm->trans->trans_cfg->gen2)
3018 normalized_ssn &= 0xff;
3019
3020 if (normalized_ssn == tid_data->next_reclaimed) {
3021 tid_data->state = IWL_AGG_STARTING;
3022 ret = IEEE80211_AMPDU_TX_START_IMMEDIATE;
3023 } else {
3024 tid_data->state = IWL_EMPTYING_HW_QUEUE_ADDBA;
3025 ret = IEEE80211_AMPDU_TX_START_DELAY_ADDBA;
3026 }
3027
3028out:
3029 spin_unlock_bh(&mvmsta->lock);
3030
3031 return ret;
3032}
3033
3034int iwl_mvm_sta_tx_agg_oper(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
3035 struct ieee80211_sta *sta, u16 tid, u16 buf_size,
3036 bool amsdu)
3037{
3038 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
3039 struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
3040 unsigned int wdg_timeout =
3041 iwl_mvm_get_wd_timeout(mvm, vif, sta->tdls, false);
3042 int queue, ret;
3043 bool alloc_queue = true;
3044 enum iwl_mvm_queue_status queue_status;
3045 u16 ssn;
3046
3047 struct iwl_trans_txq_scd_cfg cfg = {
3048 .sta_id = mvmsta->sta_id,
3049 .tid = tid,
3050 .frame_limit = buf_size,
3051 .aggregate = true,
3052 };
3053
3054 /*
3055 * When FW supports TLC_OFFLOAD, it also implements Tx aggregation
3056 * manager, so this function should never be called in this case.
3057 */
3058 if (WARN_ON_ONCE(iwl_mvm_has_tlc_offload(mvm)))
3059 return -EINVAL;
3060
3061 BUILD_BUG_ON((sizeof(mvmsta->agg_tids) * BITS_PER_BYTE)
3062 != IWL_MAX_TID_COUNT);
3063
3064 spin_lock_bh(&mvmsta->lock);
3065 ssn = tid_data->ssn;
3066 queue = tid_data->txq_id;
3067 tid_data->state = IWL_AGG_ON;
3068 mvmsta->agg_tids |= BIT(tid);
3069 tid_data->ssn = 0xffff;
3070 tid_data->amsdu_in_ampdu_allowed = amsdu;
3071 spin_unlock_bh(&mvmsta->lock);
3072
3073 if (iwl_mvm_has_new_tx_api(mvm)) {
3074 /*
3075 * If there is no queue for this tid, iwl_mvm_sta_tx_agg_start()
3076 * would have failed, so if we are here there is no need to
3077 * allocate a queue.
3078 * However, if aggregation size is different than the default
3079 * size, the scheduler should be reconfigured.
3080 * We cannot do this with the new TX API, so return unsupported
3081 * for now, until it will be offloaded to firmware..
3082 * Note that if SCD default value changes - this condition
3083 * should be updated as well.
3084 */
3085 if (buf_size < IWL_FRAME_LIMIT)
3086 return -ENOTSUPP;
3087
3088 ret = iwl_mvm_sta_tx_agg(mvm, sta, tid, queue, true);
3089 if (ret)
3090 return -EIO;
3091 goto out;
3092 }
3093
3094 cfg.fifo = iwl_mvm_ac_to_tx_fifo[tid_to_mac80211_ac[tid]];
3095
3096 queue_status = mvm->queue_info[queue].status;
3097
3098 /* Maybe there is no need to even alloc a queue... */
3099 if (mvm->queue_info[queue].status == IWL_MVM_QUEUE_READY)
3100 alloc_queue = false;
3101
3102 /*
3103 * Only reconfig the SCD for the queue if the window size has
3104 * changed from current (become smaller)
3105 */
3106 if (!alloc_queue && buf_size < IWL_FRAME_LIMIT) {
3107 /*
3108 * If reconfiguring an existing queue, it first must be
3109 * drained
3110 */
3111 ret = iwl_trans_wait_tx_queues_empty(mvm->trans,
3112 BIT(queue));
3113 if (ret) {
3114 IWL_ERR(mvm,
3115 "Error draining queue before reconfig\n");
3116 return ret;
3117 }
3118
3119 ret = iwl_mvm_reconfig_scd(mvm, queue, cfg.fifo,
3120 mvmsta->sta_id, tid,
3121 buf_size, ssn);
3122 if (ret) {
3123 IWL_ERR(mvm,
3124 "Error reconfiguring TXQ #%d\n", queue);
3125 return ret;
3126 }
3127 }
3128
3129 if (alloc_queue)
3130 iwl_mvm_enable_txq(mvm, sta, queue, ssn,
3131 &cfg, wdg_timeout);
3132
3133 /* Send ADD_STA command to enable aggs only if the queue isn't shared */
3134 if (queue_status != IWL_MVM_QUEUE_SHARED) {
3135 ret = iwl_mvm_sta_tx_agg(mvm, sta, tid, queue, true);
3136 if (ret)
3137 return -EIO;
3138 }
3139
3140 /* No need to mark as reserved */
3141 mvm->queue_info[queue].status = IWL_MVM_QUEUE_READY;
3142
3143out:
3144 /*
3145 * Even though in theory the peer could have different
3146 * aggregation reorder buffer sizes for different sessions,
3147 * our ucode doesn't allow for that and has a global limit
3148 * for each station. Therefore, use the minimum of all the
3149 * aggregation sessions and our default value.
3150 */
3151 mvmsta->max_agg_bufsize =
3152 min(mvmsta->max_agg_bufsize, buf_size);
3153 mvmsta->lq_sta.rs_drv.lq.agg_frame_cnt_limit = mvmsta->max_agg_bufsize;
3154
3155 IWL_DEBUG_HT(mvm, "Tx aggregation enabled on ra = %pM tid = %d\n",
3156 sta->addr, tid);
3157
3158 return iwl_mvm_send_lq_cmd(mvm, &mvmsta->lq_sta.rs_drv.lq);
3159}
3160
3161static void iwl_mvm_unreserve_agg_queue(struct iwl_mvm *mvm,
3162 struct iwl_mvm_sta *mvmsta,
3163 struct iwl_mvm_tid_data *tid_data)
3164{
3165 u16 txq_id = tid_data->txq_id;
3166
3167 lockdep_assert_held(&mvm->mutex);
3168
3169 if (iwl_mvm_has_new_tx_api(mvm))
3170 return;
3171
3172 /*
3173 * The TXQ is marked as reserved only if no traffic came through yet
3174 * This means no traffic has been sent on this TID (agg'd or not), so
3175 * we no longer have use for the queue. Since it hasn't even been
3176 * allocated through iwl_mvm_enable_txq, so we can just mark it back as
3177 * free.
3178 */
3179 if (mvm->queue_info[txq_id].status == IWL_MVM_QUEUE_RESERVED) {
3180 mvm->queue_info[txq_id].status = IWL_MVM_QUEUE_FREE;
3181 tid_data->txq_id = IWL_MVM_INVALID_QUEUE;
3182 }
3183}
3184
3185int iwl_mvm_sta_tx_agg_stop(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
3186 struct ieee80211_sta *sta, u16 tid)
3187{
3188 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
3189 struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
3190 u16 txq_id;
3191 int err;
3192
3193 /*
3194 * If mac80211 is cleaning its state, then say that we finished since
3195 * our state has been cleared anyway.
3196 */
3197 if (test_bit(IWL_MVM_STATUS_IN_HW_RESTART, &mvm->status)) {
3198 ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
3199 return 0;
3200 }
3201
3202 spin_lock_bh(&mvmsta->lock);
3203
3204 txq_id = tid_data->txq_id;
3205
3206 IWL_DEBUG_TX_QUEUES(mvm, "Stop AGG: sta %d tid %d q %d state %d\n",
3207 mvmsta->sta_id, tid, txq_id, tid_data->state);
3208
3209 mvmsta->agg_tids &= ~BIT(tid);
3210
3211 iwl_mvm_unreserve_agg_queue(mvm, mvmsta, tid_data);
3212
3213 switch (tid_data->state) {
3214 case IWL_AGG_ON:
3215 tid_data->ssn = IEEE80211_SEQ_TO_SN(tid_data->seq_number);
3216
3217 IWL_DEBUG_TX_QUEUES(mvm,
3218 "ssn = %d, next_recl = %d\n",
3219 tid_data->ssn, tid_data->next_reclaimed);
3220
3221 tid_data->ssn = 0xffff;
3222 tid_data->state = IWL_AGG_OFF;
3223 spin_unlock_bh(&mvmsta->lock);
3224
3225 ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
3226
3227 iwl_mvm_sta_tx_agg(mvm, sta, tid, txq_id, false);
3228 return 0;
3229 case IWL_AGG_STARTING:
3230 case IWL_EMPTYING_HW_QUEUE_ADDBA:
3231 /*
3232 * The agg session has been stopped before it was set up. This
3233 * can happen when the AddBA timer times out for example.
3234 */
3235
3236 /* No barriers since we are under mutex */
3237 lockdep_assert_held(&mvm->mutex);
3238
3239 ieee80211_stop_tx_ba_cb_irqsafe(vif, sta->addr, tid);
3240 tid_data->state = IWL_AGG_OFF;
3241 err = 0;
3242 break;
3243 default:
3244 IWL_ERR(mvm,
3245 "Stopping AGG while state not ON or starting for %d on %d (%d)\n",
3246 mvmsta->sta_id, tid, tid_data->state);
3247 IWL_ERR(mvm,
3248 "\ttid_data->txq_id = %d\n", tid_data->txq_id);
3249 err = -EINVAL;
3250 }
3251
3252 spin_unlock_bh(&mvmsta->lock);
3253
3254 return err;
3255}
3256
3257int iwl_mvm_sta_tx_agg_flush(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
3258 struct ieee80211_sta *sta, u16 tid)
3259{
3260 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
3261 struct iwl_mvm_tid_data *tid_data = &mvmsta->tid_data[tid];
3262 u16 txq_id;
3263 enum iwl_mvm_agg_state old_state;
3264
3265 /*
3266 * First set the agg state to OFF to avoid calling
3267 * ieee80211_stop_tx_ba_cb in iwl_mvm_check_ratid_empty.
3268 */
3269 spin_lock_bh(&mvmsta->lock);
3270 txq_id = tid_data->txq_id;
3271 IWL_DEBUG_TX_QUEUES(mvm, "Flush AGG: sta %d tid %d q %d state %d\n",
3272 mvmsta->sta_id, tid, txq_id, tid_data->state);
3273 old_state = tid_data->state;
3274 tid_data->state = IWL_AGG_OFF;
3275 mvmsta->agg_tids &= ~BIT(tid);
3276 spin_unlock_bh(&mvmsta->lock);
3277
3278 iwl_mvm_unreserve_agg_queue(mvm, mvmsta, tid_data);
3279
3280 if (old_state >= IWL_AGG_ON) {
3281 iwl_mvm_drain_sta(mvm, mvmsta, true);
3282
3283 if (iwl_mvm_has_new_tx_api(mvm)) {
3284 if (iwl_mvm_flush_sta_tids(mvm, mvmsta->sta_id,
3285 BIT(tid)))
3286 IWL_ERR(mvm, "Couldn't flush the AGG queue\n");
3287 iwl_trans_wait_txq_empty(mvm->trans, txq_id);
3288 } else {
3289 if (iwl_mvm_flush_tx_path(mvm, BIT(txq_id)))
3290 IWL_ERR(mvm, "Couldn't flush the AGG queue\n");
3291 iwl_trans_wait_tx_queues_empty(mvm->trans, BIT(txq_id));
3292 }
3293
3294 iwl_mvm_drain_sta(mvm, mvmsta, false);
3295
3296 iwl_mvm_sta_tx_agg(mvm, sta, tid, txq_id, false);
3297 }
3298
3299 return 0;
3300}
3301
3302static int iwl_mvm_set_fw_key_idx(struct iwl_mvm *mvm)
3303{
3304 int i, max = -1, max_offs = -1;
3305
3306 lockdep_assert_held(&mvm->mutex);
3307
3308 /* Pick the unused key offset with the highest 'deleted'
3309 * counter. Every time a key is deleted, all the counters
3310 * are incremented and the one that was just deleted is
3311 * reset to zero. Thus, the highest counter is the one
3312 * that was deleted longest ago. Pick that one.
3313 */
3314 for (i = 0; i < STA_KEY_MAX_NUM; i++) {
3315 if (test_bit(i, mvm->fw_key_table))
3316 continue;
3317 if (mvm->fw_key_deleted[i] > max) {
3318 max = mvm->fw_key_deleted[i];
3319 max_offs = i;
3320 }
3321 }
3322
3323 if (max_offs < 0)
3324 return STA_KEY_IDX_INVALID;
3325
3326 return max_offs;
3327}
3328
3329static struct iwl_mvm_sta *iwl_mvm_get_key_sta(struct iwl_mvm *mvm,
3330 struct ieee80211_vif *vif,
3331 struct ieee80211_sta *sta)
3332{
3333 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3334
3335 if (sta)
3336 return iwl_mvm_sta_from_mac80211(sta);
3337
3338 /*
3339 * The device expects GTKs for station interfaces to be
3340 * installed as GTKs for the AP station. If we have no
3341 * station ID, then use AP's station ID.
3342 */
3343 if (vif->type == NL80211_IFTYPE_STATION &&
3344 mvmvif->ap_sta_id != IWL_MVM_INVALID_STA) {
3345 u8 sta_id = mvmvif->ap_sta_id;
3346
3347 sta = rcu_dereference_check(mvm->fw_id_to_mac_id[sta_id],
3348 lockdep_is_held(&mvm->mutex));
3349
3350 /*
3351 * It is possible that the 'sta' parameter is NULL,
3352 * for example when a GTK is removed - the sta_id will then
3353 * be the AP ID, and no station was passed by mac80211.
3354 */
3355 if (IS_ERR_OR_NULL(sta))
3356 return NULL;
3357
3358 return iwl_mvm_sta_from_mac80211(sta);
3359 }
3360
3361 return NULL;
3362}
3363
3364static int iwl_mvm_pn_cmp(const u8 *pn1, const u8 *pn2, int len)
3365{
3366 int i;
3367
3368 for (i = len - 1; i >= 0; i--) {
3369 if (pn1[i] > pn2[i])
3370 return 1;
3371 if (pn1[i] < pn2[i])
3372 return -1;
3373 }
3374
3375 return 0;
3376}
3377
3378static int iwl_mvm_send_sta_key(struct iwl_mvm *mvm,
3379 u32 sta_id,
3380 struct ieee80211_key_conf *key, bool mcast,
3381 u32 tkip_iv32, u16 *tkip_p1k, u32 cmd_flags,
3382 u8 key_offset, bool mfp)
3383{
3384 union {
3385 struct iwl_mvm_add_sta_key_cmd_v1 cmd_v1;
3386 struct iwl_mvm_add_sta_key_cmd cmd;
3387 } u = {};
3388 __le16 key_flags;
3389 int ret;
3390 u32 status;
3391 u16 keyidx;
3392 u64 pn = 0;
3393 int i, size;
3394 bool new_api = fw_has_api(&mvm->fw->ucode_capa,
3395 IWL_UCODE_TLV_API_TKIP_MIC_KEYS);
3396 int api_ver = iwl_fw_lookup_cmd_ver(mvm->fw, ADD_STA_KEY,
3397 new_api ? 2 : 1);
3398
3399 if (sta_id == IWL_MVM_INVALID_STA)
3400 return -EINVAL;
3401
3402 keyidx = (key->keyidx << STA_KEY_FLG_KEYID_POS) &
3403 STA_KEY_FLG_KEYID_MSK;
3404 key_flags = cpu_to_le16(keyidx);
3405 key_flags |= cpu_to_le16(STA_KEY_FLG_WEP_KEY_MAP);
3406
3407 switch (key->cipher) {
3408 case WLAN_CIPHER_SUITE_TKIP:
3409 key_flags |= cpu_to_le16(STA_KEY_FLG_TKIP);
3410 if (api_ver >= 2) {
3411 memcpy((void *)&u.cmd.tx_mic_key,
3412 &key->key[NL80211_TKIP_DATA_OFFSET_TX_MIC_KEY],
3413 IWL_MIC_KEY_SIZE);
3414
3415 memcpy((void *)&u.cmd.rx_mic_key,
3416 &key->key[NL80211_TKIP_DATA_OFFSET_RX_MIC_KEY],
3417 IWL_MIC_KEY_SIZE);
3418 pn = atomic64_read(&key->tx_pn);
3419
3420 } else {
3421 u.cmd_v1.tkip_rx_tsc_byte2 = tkip_iv32;
3422 for (i = 0; i < 5; i++)
3423 u.cmd_v1.tkip_rx_ttak[i] =
3424 cpu_to_le16(tkip_p1k[i]);
3425 }
3426 memcpy(u.cmd.common.key, key->key, key->keylen);
3427 break;
3428 case WLAN_CIPHER_SUITE_CCMP:
3429 key_flags |= cpu_to_le16(STA_KEY_FLG_CCM);
3430 memcpy(u.cmd.common.key, key->key, key->keylen);
3431 if (api_ver >= 2)
3432 pn = atomic64_read(&key->tx_pn);
3433 break;
3434 case WLAN_CIPHER_SUITE_WEP104:
3435 key_flags |= cpu_to_le16(STA_KEY_FLG_WEP_13BYTES);
3436 fallthrough;
3437 case WLAN_CIPHER_SUITE_WEP40:
3438 key_flags |= cpu_to_le16(STA_KEY_FLG_WEP);
3439 memcpy(u.cmd.common.key + 3, key->key, key->keylen);
3440 break;
3441 case WLAN_CIPHER_SUITE_GCMP_256:
3442 key_flags |= cpu_to_le16(STA_KEY_FLG_KEY_32BYTES);
3443 fallthrough;
3444 case WLAN_CIPHER_SUITE_GCMP:
3445 key_flags |= cpu_to_le16(STA_KEY_FLG_GCMP);
3446 memcpy(u.cmd.common.key, key->key, key->keylen);
3447 if (api_ver >= 2)
3448 pn = atomic64_read(&key->tx_pn);
3449 break;
3450 default:
3451 key_flags |= cpu_to_le16(STA_KEY_FLG_EXT);
3452 memcpy(u.cmd.common.key, key->key, key->keylen);
3453 }
3454
3455 if (mcast)
3456 key_flags |= cpu_to_le16(STA_KEY_MULTICAST);
3457 if (mfp)
3458 key_flags |= cpu_to_le16(STA_KEY_MFP);
3459
3460 u.cmd.common.key_offset = key_offset;
3461 u.cmd.common.key_flags = key_flags;
3462 u.cmd.common.sta_id = sta_id;
3463
3464 if (key->cipher == WLAN_CIPHER_SUITE_TKIP)
3465 i = 0;
3466 else
3467 i = -1;
3468
3469 for (; i < IEEE80211_NUM_TIDS; i++) {
3470 struct ieee80211_key_seq seq = {};
3471 u8 _rx_pn[IEEE80211_MAX_PN_LEN] = {}, *rx_pn = _rx_pn;
3472 int rx_pn_len = 8;
3473 /* there's a hole at 2/3 in FW format depending on version */
3474 int hole = api_ver >= 3 ? 0 : 2;
3475
3476 ieee80211_get_key_rx_seq(key, i, &seq);
3477
3478 if (key->cipher == WLAN_CIPHER_SUITE_TKIP) {
3479 rx_pn[0] = seq.tkip.iv16;
3480 rx_pn[1] = seq.tkip.iv16 >> 8;
3481 rx_pn[2 + hole] = seq.tkip.iv32;
3482 rx_pn[3 + hole] = seq.tkip.iv32 >> 8;
3483 rx_pn[4 + hole] = seq.tkip.iv32 >> 16;
3484 rx_pn[5 + hole] = seq.tkip.iv32 >> 24;
3485 } else if (key_flags & cpu_to_le16(STA_KEY_FLG_EXT)) {
3486 rx_pn = seq.hw.seq;
3487 rx_pn_len = seq.hw.seq_len;
3488 } else {
3489 rx_pn[0] = seq.ccmp.pn[0];
3490 rx_pn[1] = seq.ccmp.pn[1];
3491 rx_pn[2 + hole] = seq.ccmp.pn[2];
3492 rx_pn[3 + hole] = seq.ccmp.pn[3];
3493 rx_pn[4 + hole] = seq.ccmp.pn[4];
3494 rx_pn[5 + hole] = seq.ccmp.pn[5];
3495 }
3496
3497 if (iwl_mvm_pn_cmp(rx_pn, (u8 *)&u.cmd.common.rx_secur_seq_cnt,
3498 rx_pn_len) > 0)
3499 memcpy(&u.cmd.common.rx_secur_seq_cnt, rx_pn,
3500 rx_pn_len);
3501 }
3502
3503 if (api_ver >= 2) {
3504 u.cmd.transmit_seq_cnt = cpu_to_le64(pn);
3505 size = sizeof(u.cmd);
3506 } else {
3507 size = sizeof(u.cmd_v1);
3508 }
3509
3510 status = ADD_STA_SUCCESS;
3511 if (cmd_flags & CMD_ASYNC)
3512 ret = iwl_mvm_send_cmd_pdu(mvm, ADD_STA_KEY, CMD_ASYNC, size,
3513 &u.cmd);
3514 else
3515 ret = iwl_mvm_send_cmd_pdu_status(mvm, ADD_STA_KEY, size,
3516 &u.cmd, &status);
3517
3518 switch (status) {
3519 case ADD_STA_SUCCESS:
3520 IWL_DEBUG_WEP(mvm, "MODIFY_STA: set dynamic key passed\n");
3521 break;
3522 default:
3523 ret = -EIO;
3524 IWL_ERR(mvm, "MODIFY_STA: set dynamic key failed\n");
3525 break;
3526 }
3527
3528 return ret;
3529}
3530
3531static int iwl_mvm_send_sta_igtk(struct iwl_mvm *mvm,
3532 struct ieee80211_key_conf *keyconf,
3533 u8 sta_id, bool remove_key)
3534{
3535 struct iwl_mvm_mgmt_mcast_key_cmd igtk_cmd = {};
3536
3537 /* verify the key details match the required command's expectations */
3538 if (WARN_ON((keyconf->flags & IEEE80211_KEY_FLAG_PAIRWISE) ||
3539 (keyconf->keyidx != 4 && keyconf->keyidx != 5 &&
3540 keyconf->keyidx != 6 && keyconf->keyidx != 7) ||
3541 (keyconf->cipher != WLAN_CIPHER_SUITE_AES_CMAC &&
3542 keyconf->cipher != WLAN_CIPHER_SUITE_BIP_GMAC_128 &&
3543 keyconf->cipher != WLAN_CIPHER_SUITE_BIP_GMAC_256)))
3544 return -EINVAL;
3545
3546 if (WARN_ON(!iwl_mvm_has_new_rx_api(mvm) &&
3547 keyconf->cipher != WLAN_CIPHER_SUITE_AES_CMAC))
3548 return -EINVAL;
3549
3550 igtk_cmd.key_id = cpu_to_le32(keyconf->keyidx);
3551 igtk_cmd.sta_id = cpu_to_le32(sta_id);
3552
3553 if (remove_key) {
3554 /* This is a valid situation for IGTK */
3555 if (sta_id == IWL_MVM_INVALID_STA)
3556 return 0;
3557
3558 igtk_cmd.ctrl_flags |= cpu_to_le32(STA_KEY_NOT_VALID);
3559 } else {
3560 struct ieee80211_key_seq seq;
3561 const u8 *pn;
3562
3563 switch (keyconf->cipher) {
3564 case WLAN_CIPHER_SUITE_AES_CMAC:
3565 igtk_cmd.ctrl_flags |= cpu_to_le32(STA_KEY_FLG_CCM);
3566 break;
3567 case WLAN_CIPHER_SUITE_BIP_GMAC_128:
3568 case WLAN_CIPHER_SUITE_BIP_GMAC_256:
3569 igtk_cmd.ctrl_flags |= cpu_to_le32(STA_KEY_FLG_GCMP);
3570 break;
3571 default:
3572 return -EINVAL;
3573 }
3574
3575 memcpy(igtk_cmd.igtk, keyconf->key, keyconf->keylen);
3576 if (keyconf->cipher == WLAN_CIPHER_SUITE_BIP_GMAC_256)
3577 igtk_cmd.ctrl_flags |=
3578 cpu_to_le32(STA_KEY_FLG_KEY_32BYTES);
3579 ieee80211_get_key_rx_seq(keyconf, 0, &seq);
3580 pn = seq.aes_cmac.pn;
3581 igtk_cmd.receive_seq_cnt = cpu_to_le64(((u64) pn[5] << 0) |
3582 ((u64) pn[4] << 8) |
3583 ((u64) pn[3] << 16) |
3584 ((u64) pn[2] << 24) |
3585 ((u64) pn[1] << 32) |
3586 ((u64) pn[0] << 40));
3587 }
3588
3589 IWL_DEBUG_INFO(mvm, "%s %sIGTK (%d) for sta %u\n",
3590 remove_key ? "removing" : "installing",
3591 keyconf->keyidx >= 6 ? "B" : "",
3592 keyconf->keyidx, igtk_cmd.sta_id);
3593
3594 if (!iwl_mvm_has_new_rx_api(mvm)) {
3595 struct iwl_mvm_mgmt_mcast_key_cmd_v1 igtk_cmd_v1 = {
3596 .ctrl_flags = igtk_cmd.ctrl_flags,
3597 .key_id = igtk_cmd.key_id,
3598 .sta_id = igtk_cmd.sta_id,
3599 .receive_seq_cnt = igtk_cmd.receive_seq_cnt
3600 };
3601
3602 memcpy(igtk_cmd_v1.igtk, igtk_cmd.igtk,
3603 ARRAY_SIZE(igtk_cmd_v1.igtk));
3604 return iwl_mvm_send_cmd_pdu(mvm, MGMT_MCAST_KEY, 0,
3605 sizeof(igtk_cmd_v1), &igtk_cmd_v1);
3606 }
3607 return iwl_mvm_send_cmd_pdu(mvm, MGMT_MCAST_KEY, 0,
3608 sizeof(igtk_cmd), &igtk_cmd);
3609}
3610
3611
3612static inline u8 *iwl_mvm_get_mac_addr(struct iwl_mvm *mvm,
3613 struct ieee80211_vif *vif,
3614 struct ieee80211_sta *sta)
3615{
3616 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3617
3618 if (sta)
3619 return sta->addr;
3620
3621 if (vif->type == NL80211_IFTYPE_STATION &&
3622 mvmvif->ap_sta_id != IWL_MVM_INVALID_STA) {
3623 u8 sta_id = mvmvif->ap_sta_id;
3624 sta = rcu_dereference_protected(mvm->fw_id_to_mac_id[sta_id],
3625 lockdep_is_held(&mvm->mutex));
3626 return sta->addr;
3627 }
3628
3629
3630 return NULL;
3631}
3632
3633static int __iwl_mvm_set_sta_key(struct iwl_mvm *mvm,
3634 struct ieee80211_vif *vif,
3635 struct ieee80211_sta *sta,
3636 struct ieee80211_key_conf *keyconf,
3637 u8 key_offset,
3638 bool mcast)
3639{
3640 const u8 *addr;
3641 struct ieee80211_key_seq seq;
3642 u16 p1k[5];
3643 u32 sta_id;
3644 bool mfp = false;
3645
3646 if (sta) {
3647 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
3648
3649 sta_id = mvm_sta->sta_id;
3650 mfp = sta->mfp;
3651 } else if (vif->type == NL80211_IFTYPE_AP &&
3652 !(keyconf->flags & IEEE80211_KEY_FLAG_PAIRWISE)) {
3653 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3654
3655 sta_id = mvmvif->mcast_sta.sta_id;
3656 } else {
3657 IWL_ERR(mvm, "Failed to find station id\n");
3658 return -EINVAL;
3659 }
3660
3661 if (keyconf->cipher == WLAN_CIPHER_SUITE_TKIP) {
3662 addr = iwl_mvm_get_mac_addr(mvm, vif, sta);
3663 /* get phase 1 key from mac80211 */
3664 ieee80211_get_key_rx_seq(keyconf, 0, &seq);
3665 ieee80211_get_tkip_rx_p1k(keyconf, addr, seq.tkip.iv32, p1k);
3666
3667 return iwl_mvm_send_sta_key(mvm, sta_id, keyconf, mcast,
3668 seq.tkip.iv32, p1k, 0, key_offset,
3669 mfp);
3670 }
3671
3672 return iwl_mvm_send_sta_key(mvm, sta_id, keyconf, mcast,
3673 0, NULL, 0, key_offset, mfp);
3674}
3675
3676int iwl_mvm_set_sta_key(struct iwl_mvm *mvm,
3677 struct ieee80211_vif *vif,
3678 struct ieee80211_sta *sta,
3679 struct ieee80211_key_conf *keyconf,
3680 u8 key_offset)
3681{
3682 bool mcast = !(keyconf->flags & IEEE80211_KEY_FLAG_PAIRWISE);
3683 struct iwl_mvm_sta *mvm_sta;
3684 u8 sta_id = IWL_MVM_INVALID_STA;
3685 int ret;
3686 static const u8 __maybe_unused zero_addr[ETH_ALEN] = {0};
3687
3688 lockdep_assert_held(&mvm->mutex);
3689
3690 if (vif->type != NL80211_IFTYPE_AP ||
3691 keyconf->flags & IEEE80211_KEY_FLAG_PAIRWISE) {
3692 /* Get the station id from the mvm local station table */
3693 mvm_sta = iwl_mvm_get_key_sta(mvm, vif, sta);
3694 if (!mvm_sta) {
3695 IWL_ERR(mvm, "Failed to find station\n");
3696 return -EINVAL;
3697 }
3698 sta_id = mvm_sta->sta_id;
3699
3700 /*
3701 * It is possible that the 'sta' parameter is NULL, and thus
3702 * there is a need to retrieve the sta from the local station
3703 * table.
3704 */
3705 if (!sta) {
3706 sta = rcu_dereference_protected(
3707 mvm->fw_id_to_mac_id[sta_id],
3708 lockdep_is_held(&mvm->mutex));
3709 if (IS_ERR_OR_NULL(sta)) {
3710 IWL_ERR(mvm, "Invalid station id\n");
3711 return -EINVAL;
3712 }
3713 }
3714
3715 if (WARN_ON_ONCE(iwl_mvm_sta_from_mac80211(sta)->vif != vif))
3716 return -EINVAL;
3717 } else {
3718 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
3719
3720 sta_id = mvmvif->mcast_sta.sta_id;
3721 }
3722
3723 if (keyconf->cipher == WLAN_CIPHER_SUITE_AES_CMAC ||
3724 keyconf->cipher == WLAN_CIPHER_SUITE_BIP_GMAC_128 ||
3725 keyconf->cipher == WLAN_CIPHER_SUITE_BIP_GMAC_256) {
3726 ret = iwl_mvm_send_sta_igtk(mvm, keyconf, sta_id, false);
3727 goto end;
3728 }
3729
3730 /* If the key_offset is not pre-assigned, we need to find a
3731 * new offset to use. In normal cases, the offset is not
3732 * pre-assigned, but during HW_RESTART we want to reuse the
3733 * same indices, so we pass them when this function is called.
3734 *
3735 * In D3 entry, we need to hardcoded the indices (because the
3736 * firmware hardcodes the PTK offset to 0). In this case, we
3737 * need to make sure we don't overwrite the hw_key_idx in the
3738 * keyconf structure, because otherwise we cannot configure
3739 * the original ones back when resuming.
3740 */
3741 if (key_offset == STA_KEY_IDX_INVALID) {
3742 key_offset = iwl_mvm_set_fw_key_idx(mvm);
3743 if (key_offset == STA_KEY_IDX_INVALID)
3744 return -ENOSPC;
3745 keyconf->hw_key_idx = key_offset;
3746 }
3747
3748 ret = __iwl_mvm_set_sta_key(mvm, vif, sta, keyconf, key_offset, mcast);
3749 if (ret)
3750 goto end;
3751
3752 /*
3753 * For WEP, the same key is used for multicast and unicast. Upload it
3754 * again, using the same key offset, and now pointing the other one
3755 * to the same key slot (offset).
3756 * If this fails, remove the original as well.
3757 */
3758 if ((keyconf->cipher == WLAN_CIPHER_SUITE_WEP40 ||
3759 keyconf->cipher == WLAN_CIPHER_SUITE_WEP104) &&
3760 sta) {
3761 ret = __iwl_mvm_set_sta_key(mvm, vif, sta, keyconf,
3762 key_offset, !mcast);
3763 if (ret) {
3764 __iwl_mvm_remove_sta_key(mvm, sta_id, keyconf, mcast);
3765 goto end;
3766 }
3767 }
3768
3769 __set_bit(key_offset, mvm->fw_key_table);
3770
3771end:
3772 IWL_DEBUG_WEP(mvm, "key: cipher=%x len=%d idx=%d sta=%pM ret=%d\n",
3773 keyconf->cipher, keyconf->keylen, keyconf->keyidx,
3774 sta ? sta->addr : zero_addr, ret);
3775 return ret;
3776}
3777
3778int iwl_mvm_remove_sta_key(struct iwl_mvm *mvm,
3779 struct ieee80211_vif *vif,
3780 struct ieee80211_sta *sta,
3781 struct ieee80211_key_conf *keyconf)
3782{
3783 bool mcast = !(keyconf->flags & IEEE80211_KEY_FLAG_PAIRWISE);
3784 struct iwl_mvm_sta *mvm_sta;
3785 u8 sta_id = IWL_MVM_INVALID_STA;
3786 int ret, i;
3787
3788 lockdep_assert_held(&mvm->mutex);
3789
3790 /* Get the station from the mvm local station table */
3791 mvm_sta = iwl_mvm_get_key_sta(mvm, vif, sta);
3792 if (mvm_sta)
3793 sta_id = mvm_sta->sta_id;
3794 else if (!sta && vif->type == NL80211_IFTYPE_AP && mcast)
3795 sta_id = iwl_mvm_vif_from_mac80211(vif)->mcast_sta.sta_id;
3796
3797
3798 IWL_DEBUG_WEP(mvm, "mvm remove dynamic key: idx=%d sta=%d\n",
3799 keyconf->keyidx, sta_id);
3800
3801 if (keyconf->cipher == WLAN_CIPHER_SUITE_AES_CMAC ||
3802 keyconf->cipher == WLAN_CIPHER_SUITE_BIP_GMAC_128 ||
3803 keyconf->cipher == WLAN_CIPHER_SUITE_BIP_GMAC_256)
3804 return iwl_mvm_send_sta_igtk(mvm, keyconf, sta_id, true);
3805
3806 if (!__test_and_clear_bit(keyconf->hw_key_idx, mvm->fw_key_table)) {
3807 IWL_ERR(mvm, "offset %d not used in fw key table.\n",
3808 keyconf->hw_key_idx);
3809 return -ENOENT;
3810 }
3811
3812 /* track which key was deleted last */
3813 for (i = 0; i < STA_KEY_MAX_NUM; i++) {
3814 if (mvm->fw_key_deleted[i] < U8_MAX)
3815 mvm->fw_key_deleted[i]++;
3816 }
3817 mvm->fw_key_deleted[keyconf->hw_key_idx] = 0;
3818
3819 if (sta && !mvm_sta) {
3820 IWL_DEBUG_WEP(mvm, "station non-existent, early return.\n");
3821 return 0;
3822 }
3823
3824 ret = __iwl_mvm_remove_sta_key(mvm, sta_id, keyconf, mcast);
3825 if (ret)
3826 return ret;
3827
3828 /* delete WEP key twice to get rid of (now useless) offset */
3829 if (keyconf->cipher == WLAN_CIPHER_SUITE_WEP40 ||
3830 keyconf->cipher == WLAN_CIPHER_SUITE_WEP104)
3831 ret = __iwl_mvm_remove_sta_key(mvm, sta_id, keyconf, !mcast);
3832
3833 return ret;
3834}
3835
3836void iwl_mvm_update_tkip_key(struct iwl_mvm *mvm,
3837 struct ieee80211_vif *vif,
3838 struct ieee80211_key_conf *keyconf,
3839 struct ieee80211_sta *sta, u32 iv32,
3840 u16 *phase1key)
3841{
3842 struct iwl_mvm_sta *mvm_sta;
3843 bool mcast = !(keyconf->flags & IEEE80211_KEY_FLAG_PAIRWISE);
3844 bool mfp = sta ? sta->mfp : false;
3845
3846 rcu_read_lock();
3847
3848 mvm_sta = iwl_mvm_get_key_sta(mvm, vif, sta);
3849 if (WARN_ON_ONCE(!mvm_sta))
3850 goto unlock;
3851 iwl_mvm_send_sta_key(mvm, mvm_sta->sta_id, keyconf, mcast,
3852 iv32, phase1key, CMD_ASYNC, keyconf->hw_key_idx,
3853 mfp);
3854
3855 unlock:
3856 rcu_read_unlock();
3857}
3858
3859void iwl_mvm_sta_modify_ps_wake(struct iwl_mvm *mvm,
3860 struct ieee80211_sta *sta)
3861{
3862 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
3863 struct iwl_mvm_add_sta_cmd cmd = {
3864 .add_modify = STA_MODE_MODIFY,
3865 .sta_id = mvmsta->sta_id,
3866 .station_flags_msk = cpu_to_le32(STA_FLG_PS),
3867 .mac_id_n_color = cpu_to_le32(mvmsta->mac_id_n_color),
3868 };
3869 int ret;
3870
3871 ret = iwl_mvm_send_cmd_pdu(mvm, ADD_STA, CMD_ASYNC,
3872 iwl_mvm_add_sta_cmd_size(mvm), &cmd);
3873 if (ret)
3874 IWL_ERR(mvm, "Failed to send ADD_STA command (%d)\n", ret);
3875}
3876
3877void iwl_mvm_sta_modify_sleep_tx_count(struct iwl_mvm *mvm,
3878 struct ieee80211_sta *sta,
3879 enum ieee80211_frame_release_type reason,
3880 u16 cnt, u16 tids, bool more_data,
3881 bool single_sta_queue)
3882{
3883 struct iwl_mvm_sta *mvmsta = iwl_mvm_sta_from_mac80211(sta);
3884 struct iwl_mvm_add_sta_cmd cmd = {
3885 .add_modify = STA_MODE_MODIFY,
3886 .sta_id = mvmsta->sta_id,
3887 .modify_mask = STA_MODIFY_SLEEPING_STA_TX_COUNT,
3888 .sleep_tx_count = cpu_to_le16(cnt),
3889 .mac_id_n_color = cpu_to_le32(mvmsta->mac_id_n_color),
3890 };
3891 int tid, ret;
3892 unsigned long _tids = tids;
3893
3894 /* convert TIDs to ACs - we don't support TSPEC so that's OK
3895 * Note that this field is reserved and unused by firmware not
3896 * supporting GO uAPSD, so it's safe to always do this.
3897 */
3898 for_each_set_bit(tid, &_tids, IWL_MAX_TID_COUNT)
3899 cmd.awake_acs |= BIT(tid_to_ucode_ac[tid]);
3900
3901 /* If we're releasing frames from aggregation or dqa queues then check
3902 * if all the queues that we're releasing frames from, combined, have:
3903 * - more frames than the service period, in which case more_data
3904 * needs to be set
3905 * - fewer than 'cnt' frames, in which case we need to adjust the
3906 * firmware command (but do that unconditionally)
3907 */
3908 if (single_sta_queue) {
3909 int remaining = cnt;
3910 int sleep_tx_count;
3911
3912 spin_lock_bh(&mvmsta->lock);
3913 for_each_set_bit(tid, &_tids, IWL_MAX_TID_COUNT) {
3914 struct iwl_mvm_tid_data *tid_data;
3915 u16 n_queued;
3916
3917 tid_data = &mvmsta->tid_data[tid];
3918
3919 n_queued = iwl_mvm_tid_queued(mvm, tid_data);
3920 if (n_queued > remaining) {
3921 more_data = true;
3922 remaining = 0;
3923 break;
3924 }
3925 remaining -= n_queued;
3926 }
3927 sleep_tx_count = cnt - remaining;
3928 if (reason == IEEE80211_FRAME_RELEASE_UAPSD)
3929 mvmsta->sleep_tx_count = sleep_tx_count;
3930 spin_unlock_bh(&mvmsta->lock);
3931
3932 cmd.sleep_tx_count = cpu_to_le16(sleep_tx_count);
3933 if (WARN_ON(cnt - remaining == 0)) {
3934 ieee80211_sta_eosp(sta);
3935 return;
3936 }
3937 }
3938
3939 /* Note: this is ignored by firmware not supporting GO uAPSD */
3940 if (more_data)
3941 cmd.sleep_state_flags |= STA_SLEEP_STATE_MOREDATA;
3942
3943 if (reason == IEEE80211_FRAME_RELEASE_PSPOLL) {
3944 mvmsta->next_status_eosp = true;
3945 cmd.sleep_state_flags |= STA_SLEEP_STATE_PS_POLL;
3946 } else {
3947 cmd.sleep_state_flags |= STA_SLEEP_STATE_UAPSD;
3948 }
3949
3950 /* block the Tx queues until the FW updated the sleep Tx count */
3951 iwl_trans_block_txq_ptrs(mvm->trans, true);
3952
3953 ret = iwl_mvm_send_cmd_pdu(mvm, ADD_STA,
3954 CMD_ASYNC | CMD_WANT_ASYNC_CALLBACK,
3955 iwl_mvm_add_sta_cmd_size(mvm), &cmd);
3956 if (ret)
3957 IWL_ERR(mvm, "Failed to send ADD_STA command (%d)\n", ret);
3958}
3959
3960void iwl_mvm_rx_eosp_notif(struct iwl_mvm *mvm,
3961 struct iwl_rx_cmd_buffer *rxb)
3962{
3963 struct iwl_rx_packet *pkt = rxb_addr(rxb);
3964 struct iwl_mvm_eosp_notification *notif = (void *)pkt->data;
3965 struct ieee80211_sta *sta;
3966 u32 sta_id = le32_to_cpu(notif->sta_id);
3967
3968 if (WARN_ON_ONCE(sta_id >= mvm->fw->ucode_capa.num_stations))
3969 return;
3970
3971 rcu_read_lock();
3972 sta = rcu_dereference(mvm->fw_id_to_mac_id[sta_id]);
3973 if (!IS_ERR_OR_NULL(sta))
3974 ieee80211_sta_eosp(sta);
3975 rcu_read_unlock();
3976}
3977
3978void iwl_mvm_sta_modify_disable_tx(struct iwl_mvm *mvm,
3979 struct iwl_mvm_sta *mvmsta, bool disable)
3980{
3981 struct iwl_mvm_add_sta_cmd cmd = {
3982 .add_modify = STA_MODE_MODIFY,
3983 .sta_id = mvmsta->sta_id,
3984 .station_flags = disable ? cpu_to_le32(STA_FLG_DISABLE_TX) : 0,
3985 .station_flags_msk = cpu_to_le32(STA_FLG_DISABLE_TX),
3986 .mac_id_n_color = cpu_to_le32(mvmsta->mac_id_n_color),
3987 };
3988 int ret;
3989
3990 ret = iwl_mvm_send_cmd_pdu(mvm, ADD_STA, CMD_ASYNC,
3991 iwl_mvm_add_sta_cmd_size(mvm), &cmd);
3992 if (ret)
3993 IWL_ERR(mvm, "Failed to send ADD_STA command (%d)\n", ret);
3994}
3995
3996void iwl_mvm_sta_modify_disable_tx_ap(struct iwl_mvm *mvm,
3997 struct ieee80211_sta *sta,
3998 bool disable)
3999{
4000 struct iwl_mvm_sta *mvm_sta = iwl_mvm_sta_from_mac80211(sta);
4001
4002 spin_lock_bh(&mvm_sta->lock);
4003
4004 if (mvm_sta->disable_tx == disable) {
4005 spin_unlock_bh(&mvm_sta->lock);
4006 return;
4007 }
4008
4009 mvm_sta->disable_tx = disable;
4010
4011 /*
4012 * If sta PS state is handled by mac80211, tell it to start/stop
4013 * queuing tx for this station.
4014 */
4015 if (!ieee80211_hw_check(mvm->hw, AP_LINK_PS))
4016 ieee80211_sta_block_awake(mvm->hw, sta, disable);
4017
4018 iwl_mvm_sta_modify_disable_tx(mvm, mvm_sta, disable);
4019
4020 spin_unlock_bh(&mvm_sta->lock);
4021}
4022
4023static void iwl_mvm_int_sta_modify_disable_tx(struct iwl_mvm *mvm,
4024 struct iwl_mvm_vif *mvmvif,
4025 struct iwl_mvm_int_sta *sta,
4026 bool disable)
4027{
4028 u32 id = FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color);
4029 struct iwl_mvm_add_sta_cmd cmd = {
4030 .add_modify = STA_MODE_MODIFY,
4031 .sta_id = sta->sta_id,
4032 .station_flags = disable ? cpu_to_le32(STA_FLG_DISABLE_TX) : 0,
4033 .station_flags_msk = cpu_to_le32(STA_FLG_DISABLE_TX),
4034 .mac_id_n_color = cpu_to_le32(id),
4035 };
4036 int ret;
4037
4038 ret = iwl_mvm_send_cmd_pdu(mvm, ADD_STA, CMD_ASYNC,
4039 iwl_mvm_add_sta_cmd_size(mvm), &cmd);
4040 if (ret)
4041 IWL_ERR(mvm, "Failed to send ADD_STA command (%d)\n", ret);
4042}
4043
4044void iwl_mvm_modify_all_sta_disable_tx(struct iwl_mvm *mvm,
4045 struct iwl_mvm_vif *mvmvif,
4046 bool disable)
4047{
4048 struct ieee80211_sta *sta;
4049 struct iwl_mvm_sta *mvm_sta;
4050 int i;
4051
4052 rcu_read_lock();
4053
4054 /* Block/unblock all the stations of the given mvmvif */
4055 for (i = 0; i < mvm->fw->ucode_capa.num_stations; i++) {
4056 sta = rcu_dereference(mvm->fw_id_to_mac_id[i]);
4057 if (IS_ERR_OR_NULL(sta))
4058 continue;
4059
4060 mvm_sta = iwl_mvm_sta_from_mac80211(sta);
4061 if (mvm_sta->mac_id_n_color !=
4062 FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color))
4063 continue;
4064
4065 iwl_mvm_sta_modify_disable_tx_ap(mvm, sta, disable);
4066 }
4067
4068 rcu_read_unlock();
4069
4070 if (!fw_has_api(&mvm->fw->ucode_capa, IWL_UCODE_TLV_API_STA_TYPE))
4071 return;
4072
4073 /* Need to block/unblock also multicast station */
4074 if (mvmvif->mcast_sta.sta_id != IWL_MVM_INVALID_STA)
4075 iwl_mvm_int_sta_modify_disable_tx(mvm, mvmvif,
4076 &mvmvif->mcast_sta, disable);
4077
4078 /*
4079 * Only unblock the broadcast station (FW blocks it for immediate
4080 * quiet, not the driver)
4081 */
4082 if (!disable && mvmvif->bcast_sta.sta_id != IWL_MVM_INVALID_STA)
4083 iwl_mvm_int_sta_modify_disable_tx(mvm, mvmvif,
4084 &mvmvif->bcast_sta, disable);
4085}
4086
4087void iwl_mvm_csa_client_absent(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
4088{
4089 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
4090 struct iwl_mvm_sta *mvmsta;
4091
4092 rcu_read_lock();
4093
4094 mvmsta = iwl_mvm_sta_from_staid_rcu(mvm, mvmvif->ap_sta_id);
4095
4096 if (mvmsta)
4097 iwl_mvm_sta_modify_disable_tx(mvm, mvmsta, true);
4098
4099 rcu_read_unlock();
4100}
4101
4102u16 iwl_mvm_tid_queued(struct iwl_mvm *mvm, struct iwl_mvm_tid_data *tid_data)
4103{
4104 u16 sn = IEEE80211_SEQ_TO_SN(tid_data->seq_number);
4105
4106 /*
4107 * In 22000 HW, the next_reclaimed index is only 8 bit, so we'll need
4108 * to align the wrap around of ssn so we compare relevant values.
4109 */
4110 if (mvm->trans->trans_cfg->gen2)
4111 sn &= 0xff;
4112
4113 return ieee80211_sn_sub(sn, tid_data->next_reclaimed);
4114}
4115
4116int iwl_mvm_add_pasn_sta(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
4117 struct iwl_mvm_int_sta *sta, u8 *addr, u32 cipher,
4118 u8 *key, u32 key_len)
4119{
4120 int ret;
4121 u16 queue;
4122 struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
4123 struct ieee80211_key_conf *keyconf;
4124
4125 ret = iwl_mvm_allocate_int_sta(mvm, sta, 0,
4126 NL80211_IFTYPE_UNSPECIFIED,
4127 IWL_STA_LINK);
4128 if (ret)
4129 return ret;
4130
4131 ret = iwl_mvm_add_int_sta_with_queue(mvm, mvmvif->id, mvmvif->color,
4132 addr, sta, &queue,
4133 IWL_MVM_TX_FIFO_BE);
4134 if (ret)
4135 goto out;
4136
4137 keyconf = kzalloc(sizeof(*keyconf) + key_len, GFP_KERNEL);
4138 if (!keyconf) {
4139 ret = -ENOBUFS;
4140 goto out;
4141 }
4142
4143 keyconf->cipher = cipher;
4144 memcpy(keyconf->key, key, key_len);
4145 keyconf->keylen = key_len;
4146
4147 ret = iwl_mvm_send_sta_key(mvm, sta->sta_id, keyconf, false,
4148 0, NULL, 0, 0, true);
4149 kfree(keyconf);
4150 return 0;
4151out:
4152 iwl_mvm_dealloc_int_sta(mvm, sta);
4153 return ret;
4154}
4155
4156void iwl_mvm_cancel_channel_switch(struct iwl_mvm *mvm,
4157 struct ieee80211_vif *vif,
4158 u32 mac_id)
4159{
4160 struct iwl_cancel_channel_switch_cmd cancel_channel_switch_cmd = {
4161 .mac_id = cpu_to_le32(mac_id),
4162 };
4163 int ret;
4164
4165 ret = iwl_mvm_send_cmd_pdu(mvm,
4166 WIDE_ID(MAC_CONF_GROUP, CANCEL_CHANNEL_SWITCH_CMD),
4167 CMD_ASYNC,
4168 sizeof(cancel_channel_switch_cmd),
4169 &cancel_channel_switch_cmd);
4170 if (ret)
4171 IWL_ERR(mvm, "Failed to cancel the channel switch\n");
4172}