1
2
3
4
5#include <linux/of.h>
6#include "mt76.h"
7
8#define CHAN2G(_idx, _freq) { \
9 .band = NL80211_BAND_2GHZ, \
10 .center_freq = (_freq), \
11 .hw_value = (_idx), \
12 .max_power = 30, \
13}
14
15#define CHAN5G(_idx, _freq) { \
16 .band = NL80211_BAND_5GHZ, \
17 .center_freq = (_freq), \
18 .hw_value = (_idx), \
19 .max_power = 30, \
20}
21
22static const struct ieee80211_channel mt76_channels_2ghz[] = {
23 CHAN2G(1, 2412),
24 CHAN2G(2, 2417),
25 CHAN2G(3, 2422),
26 CHAN2G(4, 2427),
27 CHAN2G(5, 2432),
28 CHAN2G(6, 2437),
29 CHAN2G(7, 2442),
30 CHAN2G(8, 2447),
31 CHAN2G(9, 2452),
32 CHAN2G(10, 2457),
33 CHAN2G(11, 2462),
34 CHAN2G(12, 2467),
35 CHAN2G(13, 2472),
36 CHAN2G(14, 2484),
37};
38
39static const struct ieee80211_channel mt76_channels_5ghz[] = {
40 CHAN5G(36, 5180),
41 CHAN5G(40, 5200),
42 CHAN5G(44, 5220),
43 CHAN5G(48, 5240),
44
45 CHAN5G(52, 5260),
46 CHAN5G(56, 5280),
47 CHAN5G(60, 5300),
48 CHAN5G(64, 5320),
49
50 CHAN5G(100, 5500),
51 CHAN5G(104, 5520),
52 CHAN5G(108, 5540),
53 CHAN5G(112, 5560),
54 CHAN5G(116, 5580),
55 CHAN5G(120, 5600),
56 CHAN5G(124, 5620),
57 CHAN5G(128, 5640),
58 CHAN5G(132, 5660),
59 CHAN5G(136, 5680),
60 CHAN5G(140, 5700),
61
62 CHAN5G(149, 5745),
63 CHAN5G(153, 5765),
64 CHAN5G(157, 5785),
65 CHAN5G(161, 5805),
66 CHAN5G(165, 5825),
67};
68
69static const struct ieee80211_tpt_blink mt76_tpt_blink[] = {
70 { .throughput = 0 * 1024, .blink_time = 334 },
71 { .throughput = 1 * 1024, .blink_time = 260 },
72 { .throughput = 5 * 1024, .blink_time = 220 },
73 { .throughput = 10 * 1024, .blink_time = 190 },
74 { .throughput = 20 * 1024, .blink_time = 170 },
75 { .throughput = 50 * 1024, .blink_time = 150 },
76 { .throughput = 70 * 1024, .blink_time = 130 },
77 { .throughput = 100 * 1024, .blink_time = 110 },
78 { .throughput = 200 * 1024, .blink_time = 80 },
79 { .throughput = 300 * 1024, .blink_time = 50 },
80};
81
82static int mt76_led_init(struct mt76_dev *dev)
83{
84 struct device_node *np = dev->dev->of_node;
85 struct ieee80211_hw *hw = dev->hw;
86 int led_pin;
87
88 if (!dev->led_cdev.brightness_set && !dev->led_cdev.blink_set)
89 return 0;
90
91 snprintf(dev->led_name, sizeof(dev->led_name),
92 "mt76-%s", wiphy_name(hw->wiphy));
93
94 dev->led_cdev.name = dev->led_name;
95 dev->led_cdev.default_trigger =
96 ieee80211_create_tpt_led_trigger(hw,
97 IEEE80211_TPT_LEDTRIG_FL_RADIO,
98 mt76_tpt_blink,
99 ARRAY_SIZE(mt76_tpt_blink));
100
101 np = of_get_child_by_name(np, "led");
102 if (np) {
103 if (!of_property_read_u32(np, "led-sources", &led_pin))
104 dev->led_pin = led_pin;
105 dev->led_al = of_property_read_bool(np, "led-active-low");
106 }
107
108 return devm_led_classdev_register(dev->dev, &dev->led_cdev);
109}
110
111static void mt76_init_stream_cap(struct mt76_dev *dev,
112 struct ieee80211_supported_band *sband,
113 bool vht)
114{
115 struct ieee80211_sta_ht_cap *ht_cap = &sband->ht_cap;
116 int i, nstream = hweight8(dev->antenna_mask);
117 struct ieee80211_sta_vht_cap *vht_cap;
118 u16 mcs_map = 0;
119
120 if (nstream > 1)
121 ht_cap->cap |= IEEE80211_HT_CAP_TX_STBC;
122 else
123 ht_cap->cap &= ~IEEE80211_HT_CAP_TX_STBC;
124
125 for (i = 0; i < IEEE80211_HT_MCS_MASK_LEN; i++)
126 ht_cap->mcs.rx_mask[i] = i < nstream ? 0xff : 0;
127
128 if (!vht)
129 return;
130
131 vht_cap = &sband->vht_cap;
132 if (nstream > 1)
133 vht_cap->cap |= IEEE80211_VHT_CAP_TXSTBC;
134 else
135 vht_cap->cap &= ~IEEE80211_VHT_CAP_TXSTBC;
136
137 for (i = 0; i < 8; i++) {
138 if (i < nstream)
139 mcs_map |= (IEEE80211_VHT_MCS_SUPPORT_0_9 << (i * 2));
140 else
141 mcs_map |=
142 (IEEE80211_VHT_MCS_NOT_SUPPORTED << (i * 2));
143 }
144 vht_cap->vht_mcs.rx_mcs_map = cpu_to_le16(mcs_map);
145 vht_cap->vht_mcs.tx_mcs_map = cpu_to_le16(mcs_map);
146}
147
148void mt76_set_stream_caps(struct mt76_dev *dev, bool vht)
149{
150 if (dev->cap.has_2ghz)
151 mt76_init_stream_cap(dev, &dev->sband_2g.sband, false);
152 if (dev->cap.has_5ghz)
153 mt76_init_stream_cap(dev, &dev->sband_5g.sband, vht);
154}
155EXPORT_SYMBOL_GPL(mt76_set_stream_caps);
156
157static int
158mt76_init_sband(struct mt76_dev *dev, struct mt76_sband *msband,
159 const struct ieee80211_channel *chan, int n_chan,
160 struct ieee80211_rate *rates, int n_rates, bool vht)
161{
162 struct ieee80211_supported_band *sband = &msband->sband;
163 struct ieee80211_sta_ht_cap *ht_cap;
164 struct ieee80211_sta_vht_cap *vht_cap;
165 void *chanlist;
166 int size;
167
168 size = n_chan * sizeof(*chan);
169 chanlist = devm_kmemdup(dev->dev, chan, size, GFP_KERNEL);
170 if (!chanlist)
171 return -ENOMEM;
172
173 msband->chan = devm_kcalloc(dev->dev, n_chan, sizeof(*msband->chan),
174 GFP_KERNEL);
175 if (!msband->chan)
176 return -ENOMEM;
177
178 sband->channels = chanlist;
179 sband->n_channels = n_chan;
180 sband->bitrates = rates;
181 sband->n_bitrates = n_rates;
182 dev->chandef.chan = &sband->channels[0];
183
184 ht_cap = &sband->ht_cap;
185 ht_cap->ht_supported = true;
186 ht_cap->cap |= IEEE80211_HT_CAP_SUP_WIDTH_20_40 |
187 IEEE80211_HT_CAP_GRN_FLD |
188 IEEE80211_HT_CAP_SGI_20 |
189 IEEE80211_HT_CAP_SGI_40 |
190 (1 << IEEE80211_HT_CAP_RX_STBC_SHIFT);
191
192 ht_cap->mcs.tx_params = IEEE80211_HT_MCS_TX_DEFINED;
193 ht_cap->ampdu_factor = IEEE80211_HT_MAX_AMPDU_64K;
194 ht_cap->ampdu_density = IEEE80211_HT_MPDU_DENSITY_4;
195
196 mt76_init_stream_cap(dev, sband, vht);
197
198 if (!vht)
199 return 0;
200
201 vht_cap = &sband->vht_cap;
202 vht_cap->vht_supported = true;
203 vht_cap->cap |= IEEE80211_VHT_CAP_RXLDPC |
204 IEEE80211_VHT_CAP_RXSTBC_1 |
205 IEEE80211_VHT_CAP_SHORT_GI_80 |
206 IEEE80211_VHT_CAP_RX_ANTENNA_PATTERN |
207 IEEE80211_VHT_CAP_TX_ANTENNA_PATTERN |
208 (3 << IEEE80211_VHT_CAP_MAX_A_MPDU_LENGTH_EXPONENT_SHIFT);
209
210 return 0;
211}
212
213static int
214mt76_init_sband_2g(struct mt76_dev *dev, struct ieee80211_rate *rates,
215 int n_rates)
216{
217 dev->hw->wiphy->bands[NL80211_BAND_2GHZ] = &dev->sband_2g.sband;
218
219 return mt76_init_sband(dev, &dev->sband_2g,
220 mt76_channels_2ghz,
221 ARRAY_SIZE(mt76_channels_2ghz),
222 rates, n_rates, false);
223}
224
225static int
226mt76_init_sband_5g(struct mt76_dev *dev, struct ieee80211_rate *rates,
227 int n_rates, bool vht)
228{
229 dev->hw->wiphy->bands[NL80211_BAND_5GHZ] = &dev->sband_5g.sband;
230
231 return mt76_init_sband(dev, &dev->sband_5g,
232 mt76_channels_5ghz,
233 ARRAY_SIZE(mt76_channels_5ghz),
234 rates, n_rates, vht);
235}
236
237static void
238mt76_check_sband(struct mt76_dev *dev, int band)
239{
240 struct ieee80211_supported_band *sband = dev->hw->wiphy->bands[band];
241 bool found = false;
242 int i;
243
244 if (!sband)
245 return;
246
247 for (i = 0; i < sband->n_channels; i++) {
248 if (sband->channels[i].flags & IEEE80211_CHAN_DISABLED)
249 continue;
250
251 found = true;
252 break;
253 }
254
255 if (found)
256 return;
257
258 sband->n_channels = 0;
259 dev->hw->wiphy->bands[band] = NULL;
260}
261
262struct mt76_dev *
263mt76_alloc_device(struct device *pdev, unsigned int size,
264 const struct ieee80211_ops *ops,
265 const struct mt76_driver_ops *drv_ops)
266{
267 struct ieee80211_hw *hw;
268 struct mt76_dev *dev;
269
270 hw = ieee80211_alloc_hw(size, ops);
271 if (!hw)
272 return NULL;
273
274 dev = hw->priv;
275 dev->hw = hw;
276 dev->dev = pdev;
277 dev->drv = drv_ops;
278
279 spin_lock_init(&dev->rx_lock);
280 spin_lock_init(&dev->lock);
281 spin_lock_init(&dev->cc_lock);
282 mutex_init(&dev->mutex);
283 init_waitqueue_head(&dev->tx_wait);
284 skb_queue_head_init(&dev->status_list);
285
286 tasklet_init(&dev->tx_tasklet, mt76_tx_tasklet, (unsigned long)dev);
287
288 return dev;
289}
290EXPORT_SYMBOL_GPL(mt76_alloc_device);
291
292int mt76_register_device(struct mt76_dev *dev, bool vht,
293 struct ieee80211_rate *rates, int n_rates)
294{
295 struct ieee80211_hw *hw = dev->hw;
296 struct wiphy *wiphy = hw->wiphy;
297 int ret;
298
299 dev_set_drvdata(dev->dev, dev);
300
301 INIT_LIST_HEAD(&dev->txwi_cache);
302
303 SET_IEEE80211_DEV(hw, dev->dev);
304 SET_IEEE80211_PERM_ADDR(hw, dev->macaddr);
305
306 wiphy->features |= NL80211_FEATURE_ACTIVE_MONITOR;
307
308 wiphy_ext_feature_set(wiphy, NL80211_EXT_FEATURE_CQM_RSSI_LIST);
309
310 wiphy->available_antennas_tx = dev->antenna_mask;
311 wiphy->available_antennas_rx = dev->antenna_mask;
312
313 hw->txq_data_size = sizeof(struct mt76_txq);
314 hw->max_tx_fragments = 16;
315
316 ieee80211_hw_set(hw, SIGNAL_DBM);
317 ieee80211_hw_set(hw, PS_NULLFUNC_STACK);
318 ieee80211_hw_set(hw, HOST_BROADCAST_PS_BUFFERING);
319 ieee80211_hw_set(hw, AMPDU_AGGREGATION);
320 ieee80211_hw_set(hw, SUPPORTS_RC_TABLE);
321 ieee80211_hw_set(hw, SUPPORT_FAST_XMIT);
322 ieee80211_hw_set(hw, SUPPORTS_CLONED_SKBS);
323 ieee80211_hw_set(hw, SUPPORTS_AMSDU_IN_AMPDU);
324 ieee80211_hw_set(hw, TX_AMSDU);
325 ieee80211_hw_set(hw, TX_FRAG_LIST);
326 ieee80211_hw_set(hw, MFP_CAPABLE);
327 ieee80211_hw_set(hw, AP_LINK_PS);
328 ieee80211_hw_set(hw, REPORTS_TX_ACK_STATUS);
329 ieee80211_hw_set(hw, NEEDS_UNIQUE_STA_ADDR);
330
331 wiphy->flags |= WIPHY_FLAG_IBSS_RSN;
332
333 if (dev->cap.has_2ghz) {
334 ret = mt76_init_sband_2g(dev, rates, n_rates);
335 if (ret)
336 return ret;
337 }
338
339 if (dev->cap.has_5ghz) {
340 ret = mt76_init_sband_5g(dev, rates + 4, n_rates - 4, vht);
341 if (ret)
342 return ret;
343 }
344
345 wiphy_read_of_freq_limits(dev->hw->wiphy);
346 mt76_check_sband(dev, NL80211_BAND_2GHZ);
347 mt76_check_sband(dev, NL80211_BAND_5GHZ);
348
349 if (IS_ENABLED(CONFIG_MT76_LEDS)) {
350 ret = mt76_led_init(dev);
351 if (ret)
352 return ret;
353 }
354
355 return ieee80211_register_hw(hw);
356}
357EXPORT_SYMBOL_GPL(mt76_register_device);
358
359void mt76_unregister_device(struct mt76_dev *dev)
360{
361 struct ieee80211_hw *hw = dev->hw;
362
363 mt76_tx_status_check(dev, NULL, true);
364 ieee80211_unregister_hw(hw);
365}
366EXPORT_SYMBOL_GPL(mt76_unregister_device);
367
368void mt76_free_device(struct mt76_dev *dev)
369{
370 mt76_tx_free(dev);
371 ieee80211_free_hw(dev->hw);
372}
373EXPORT_SYMBOL_GPL(mt76_free_device);
374
375void mt76_rx(struct mt76_dev *dev, enum mt76_rxq_id q, struct sk_buff *skb)
376{
377 if (!test_bit(MT76_STATE_RUNNING, &dev->state)) {
378 dev_kfree_skb(skb);
379 return;
380 }
381
382 __skb_queue_tail(&dev->rx_skb[q], skb);
383}
384EXPORT_SYMBOL_GPL(mt76_rx);
385
386bool mt76_has_tx_pending(struct mt76_dev *dev)
387{
388 struct mt76_queue *q;
389 int i;
390
391 for (i = 0; i < ARRAY_SIZE(dev->q_tx); i++) {
392 q = dev->q_tx[i].q;
393 if (q && q->queued)
394 return true;
395 }
396
397 return false;
398}
399EXPORT_SYMBOL_GPL(mt76_has_tx_pending);
400
401void mt76_set_channel(struct mt76_dev *dev)
402{
403 struct ieee80211_hw *hw = dev->hw;
404 struct cfg80211_chan_def *chandef = &hw->conf.chandef;
405 struct mt76_channel_state *state;
406 bool offchannel = hw->conf.flags & IEEE80211_CONF_OFFCHANNEL;
407 int timeout = HZ / 5;
408
409 wait_event_timeout(dev->tx_wait, !mt76_has_tx_pending(dev), timeout);
410
411 if (dev->drv->update_survey)
412 dev->drv->update_survey(dev);
413
414 dev->chandef = *chandef;
415
416 if (!offchannel)
417 dev->main_chan = chandef->chan;
418
419 if (chandef->chan != dev->main_chan) {
420 state = mt76_channel_state(dev, chandef->chan);
421 memset(state, 0, sizeof(*state));
422 }
423}
424EXPORT_SYMBOL_GPL(mt76_set_channel);
425
426int mt76_get_survey(struct ieee80211_hw *hw, int idx,
427 struct survey_info *survey)
428{
429 struct mt76_dev *dev = hw->priv;
430 struct mt76_sband *sband;
431 struct ieee80211_channel *chan;
432 struct mt76_channel_state *state;
433 int ret = 0;
434
435 if (idx == 0 && dev->drv->update_survey)
436 dev->drv->update_survey(dev);
437
438 sband = &dev->sband_2g;
439 if (idx >= sband->sband.n_channels) {
440 idx -= sband->sband.n_channels;
441 sband = &dev->sband_5g;
442 }
443
444 if (idx >= sband->sband.n_channels)
445 return -ENOENT;
446
447 chan = &sband->sband.channels[idx];
448 state = mt76_channel_state(dev, chan);
449
450 memset(survey, 0, sizeof(*survey));
451 survey->channel = chan;
452 survey->filled = SURVEY_INFO_TIME | SURVEY_INFO_TIME_BUSY;
453 if (chan == dev->main_chan)
454 survey->filled |= SURVEY_INFO_IN_USE;
455
456 spin_lock_bh(&dev->cc_lock);
457 survey->time = div_u64(state->cc_active, 1000);
458 survey->time_busy = div_u64(state->cc_busy, 1000);
459 spin_unlock_bh(&dev->cc_lock);
460
461 return ret;
462}
463EXPORT_SYMBOL_GPL(mt76_get_survey);
464
465void mt76_wcid_key_setup(struct mt76_dev *dev, struct mt76_wcid *wcid,
466 struct ieee80211_key_conf *key)
467{
468 struct ieee80211_key_seq seq;
469 int i;
470
471 wcid->rx_check_pn = false;
472
473 if (!key)
474 return;
475
476 if (key->cipher != WLAN_CIPHER_SUITE_CCMP)
477 return;
478
479 wcid->rx_check_pn = true;
480 for (i = 0; i < IEEE80211_NUM_TIDS; i++) {
481 ieee80211_get_key_rx_seq(key, i, &seq);
482 memcpy(wcid->rx_key_pn[i], seq.ccmp.pn, sizeof(seq.ccmp.pn));
483 }
484}
485EXPORT_SYMBOL(mt76_wcid_key_setup);
486
487static struct ieee80211_sta *mt76_rx_convert(struct sk_buff *skb)
488{
489 struct ieee80211_rx_status *status = IEEE80211_SKB_RXCB(skb);
490 struct mt76_rx_status mstat;
491
492 mstat = *((struct mt76_rx_status *)skb->cb);
493 memset(status, 0, sizeof(*status));
494
495 status->flag = mstat.flag;
496 status->freq = mstat.freq;
497 status->enc_flags = mstat.enc_flags;
498 status->encoding = mstat.encoding;
499 status->bw = mstat.bw;
500 status->rate_idx = mstat.rate_idx;
501 status->nss = mstat.nss;
502 status->band = mstat.band;
503 status->signal = mstat.signal;
504 status->chains = mstat.chains;
505
506 BUILD_BUG_ON(sizeof(mstat) > sizeof(skb->cb));
507 BUILD_BUG_ON(sizeof(status->chain_signal) !=
508 sizeof(mstat.chain_signal));
509 memcpy(status->chain_signal, mstat.chain_signal,
510 sizeof(mstat.chain_signal));
511
512 return wcid_to_sta(mstat.wcid);
513}
514
515static int
516mt76_check_ccmp_pn(struct sk_buff *skb)
517{
518 struct mt76_rx_status *status = (struct mt76_rx_status *)skb->cb;
519 struct mt76_wcid *wcid = status->wcid;
520 struct ieee80211_hdr *hdr;
521 int ret;
522
523 if (!(status->flag & RX_FLAG_DECRYPTED))
524 return 0;
525
526 if (!wcid || !wcid->rx_check_pn)
527 return 0;
528
529 if (!(status->flag & RX_FLAG_IV_STRIPPED)) {
530
531
532
533
534 hdr = (struct ieee80211_hdr *)skb->data;
535 if (ieee80211_is_frag(hdr) &&
536 !ieee80211_is_first_frag(hdr->frame_control))
537 return 0;
538 }
539
540 BUILD_BUG_ON(sizeof(status->iv) != sizeof(wcid->rx_key_pn[0]));
541 ret = memcmp(status->iv, wcid->rx_key_pn[status->tid],
542 sizeof(status->iv));
543 if (ret <= 0)
544 return -EINVAL;
545
546 memcpy(wcid->rx_key_pn[status->tid], status->iv, sizeof(status->iv));
547
548 if (status->flag & RX_FLAG_IV_STRIPPED)
549 status->flag |= RX_FLAG_PN_VALIDATED;
550
551 return 0;
552}
553
554static void
555mt76_check_sta(struct mt76_dev *dev, struct sk_buff *skb)
556{
557 struct mt76_rx_status *status = (struct mt76_rx_status *)skb->cb;
558 struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
559 struct ieee80211_sta *sta;
560 struct mt76_wcid *wcid = status->wcid;
561 bool ps;
562 int i;
563
564 if (ieee80211_is_pspoll(hdr->frame_control) && !wcid) {
565 sta = ieee80211_find_sta_by_ifaddr(dev->hw, hdr->addr2, NULL);
566 if (sta)
567 wcid = status->wcid = (struct mt76_wcid *)sta->drv_priv;
568 }
569
570 if (!wcid || !wcid->sta)
571 return;
572
573 sta = container_of((void *)wcid, struct ieee80211_sta, drv_priv);
574
575 if (status->signal <= 0)
576 ewma_signal_add(&wcid->rssi, -status->signal);
577
578 wcid->inactive_count = 0;
579
580 if (!test_bit(MT_WCID_FLAG_CHECK_PS, &wcid->flags))
581 return;
582
583 if (ieee80211_is_pspoll(hdr->frame_control)) {
584 ieee80211_sta_pspoll(sta);
585 return;
586 }
587
588 if (ieee80211_has_morefrags(hdr->frame_control) ||
589 !(ieee80211_is_mgmt(hdr->frame_control) ||
590 ieee80211_is_data(hdr->frame_control)))
591 return;
592
593 ps = ieee80211_has_pm(hdr->frame_control);
594
595 if (ps && (ieee80211_is_data_qos(hdr->frame_control) ||
596 ieee80211_is_qos_nullfunc(hdr->frame_control)))
597 ieee80211_sta_uapsd_trigger(sta, status->tid);
598
599 if (!!test_bit(MT_WCID_FLAG_PS, &wcid->flags) == ps)
600 return;
601
602 if (ps)
603 set_bit(MT_WCID_FLAG_PS, &wcid->flags);
604 else
605 clear_bit(MT_WCID_FLAG_PS, &wcid->flags);
606
607 dev->drv->sta_ps(dev, sta, ps);
608 ieee80211_sta_ps_transition(sta, ps);
609
610 if (ps)
611 return;
612
613 for (i = 0; i < ARRAY_SIZE(sta->txq); i++) {
614 struct mt76_txq *mtxq;
615
616 if (!sta->txq[i])
617 continue;
618
619 mtxq = (struct mt76_txq *)sta->txq[i]->drv_priv;
620 if (!skb_queue_empty(&mtxq->retry_q))
621 ieee80211_schedule_txq(dev->hw, sta->txq[i]);
622 }
623}
624
625void mt76_rx_complete(struct mt76_dev *dev, struct sk_buff_head *frames,
626 struct napi_struct *napi)
627{
628 struct ieee80211_sta *sta;
629 struct sk_buff *skb;
630
631 spin_lock(&dev->rx_lock);
632 while ((skb = __skb_dequeue(frames)) != NULL) {
633 if (mt76_check_ccmp_pn(skb)) {
634 dev_kfree_skb(skb);
635 continue;
636 }
637
638 sta = mt76_rx_convert(skb);
639 ieee80211_rx_napi(dev->hw, sta, skb, napi);
640 }
641 spin_unlock(&dev->rx_lock);
642}
643
644void mt76_rx_poll_complete(struct mt76_dev *dev, enum mt76_rxq_id q,
645 struct napi_struct *napi)
646{
647 struct sk_buff_head frames;
648 struct sk_buff *skb;
649
650 __skb_queue_head_init(&frames);
651
652 while ((skb = __skb_dequeue(&dev->rx_skb[q])) != NULL) {
653 mt76_check_sta(dev, skb);
654 mt76_rx_aggr_reorder(skb, &frames);
655 }
656
657 mt76_rx_complete(dev, &frames, napi);
658}
659EXPORT_SYMBOL_GPL(mt76_rx_poll_complete);
660
661static int
662mt76_sta_add(struct mt76_dev *dev, struct ieee80211_vif *vif,
663 struct ieee80211_sta *sta)
664{
665 struct mt76_wcid *wcid = (struct mt76_wcid *)sta->drv_priv;
666 int ret;
667 int i;
668
669 mutex_lock(&dev->mutex);
670
671 ret = dev->drv->sta_add(dev, vif, sta);
672 if (ret)
673 goto out;
674
675 for (i = 0; i < ARRAY_SIZE(sta->txq); i++) {
676 struct mt76_txq *mtxq;
677
678 if (!sta->txq[i])
679 continue;
680
681 mtxq = (struct mt76_txq *)sta->txq[i]->drv_priv;
682 mtxq->wcid = wcid;
683
684 mt76_txq_init(dev, sta->txq[i]);
685 }
686
687 ewma_signal_init(&wcid->rssi);
688 rcu_assign_pointer(dev->wcid[wcid->idx], wcid);
689
690out:
691 mutex_unlock(&dev->mutex);
692
693 return ret;
694}
695
696void __mt76_sta_remove(struct mt76_dev *dev, struct ieee80211_vif *vif,
697 struct ieee80211_sta *sta)
698{
699 struct mt76_wcid *wcid = (struct mt76_wcid *)sta->drv_priv;
700 int i, idx = wcid->idx;
701
702 rcu_assign_pointer(dev->wcid[idx], NULL);
703 synchronize_rcu();
704
705 for (i = 0; i < ARRAY_SIZE(wcid->aggr); i++)
706 mt76_rx_aggr_stop(dev, wcid, i);
707
708 if (dev->drv->sta_remove)
709 dev->drv->sta_remove(dev, vif, sta);
710
711 mt76_tx_status_check(dev, wcid, true);
712 for (i = 0; i < ARRAY_SIZE(sta->txq); i++)
713 mt76_txq_remove(dev, sta->txq[i]);
714 mt76_wcid_free(dev->wcid_mask, idx);
715}
716EXPORT_SYMBOL_GPL(__mt76_sta_remove);
717
718static void
719mt76_sta_remove(struct mt76_dev *dev, struct ieee80211_vif *vif,
720 struct ieee80211_sta *sta)
721{
722 mutex_lock(&dev->mutex);
723 __mt76_sta_remove(dev, vif, sta);
724 mutex_unlock(&dev->mutex);
725}
726
727int mt76_sta_state(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
728 struct ieee80211_sta *sta,
729 enum ieee80211_sta_state old_state,
730 enum ieee80211_sta_state new_state)
731{
732 struct mt76_dev *dev = hw->priv;
733
734 if (old_state == IEEE80211_STA_NOTEXIST &&
735 new_state == IEEE80211_STA_NONE)
736 return mt76_sta_add(dev, vif, sta);
737
738 if (old_state == IEEE80211_STA_AUTH &&
739 new_state == IEEE80211_STA_ASSOC &&
740 dev->drv->sta_assoc)
741 dev->drv->sta_assoc(dev, vif, sta);
742
743 if (old_state == IEEE80211_STA_NONE &&
744 new_state == IEEE80211_STA_NOTEXIST)
745 mt76_sta_remove(dev, vif, sta);
746
747 return 0;
748}
749EXPORT_SYMBOL_GPL(mt76_sta_state);
750
751int mt76_get_txpower(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
752 int *dbm)
753{
754 struct mt76_dev *dev = hw->priv;
755 int n_chains = hweight8(dev->antenna_mask);
756
757 *dbm = DIV_ROUND_UP(dev->txpower_cur, 2);
758
759
760
761
762 switch (n_chains) {
763 case 4:
764 *dbm += 6;
765 break;
766 case 3:
767 *dbm += 4;
768 break;
769 case 2:
770 *dbm += 3;
771 break;
772 default:
773 break;
774 }
775
776 return 0;
777}
778EXPORT_SYMBOL_GPL(mt76_get_txpower);
779
780static void
781__mt76_csa_finish(void *priv, u8 *mac, struct ieee80211_vif *vif)
782{
783 if (vif->csa_active && ieee80211_csa_is_complete(vif))
784 ieee80211_csa_finish(vif);
785}
786
787void mt76_csa_finish(struct mt76_dev *dev)
788{
789 if (!dev->csa_complete)
790 return;
791
792 ieee80211_iterate_active_interfaces_atomic(dev->hw,
793 IEEE80211_IFACE_ITER_RESUME_ALL,
794 __mt76_csa_finish, dev);
795
796 dev->csa_complete = 0;
797}
798EXPORT_SYMBOL_GPL(mt76_csa_finish);
799
800static void
801__mt76_csa_check(void *priv, u8 *mac, struct ieee80211_vif *vif)
802{
803 struct mt76_dev *dev = priv;
804
805 if (!vif->csa_active)
806 return;
807
808 dev->csa_complete |= ieee80211_csa_is_complete(vif);
809}
810
811void mt76_csa_check(struct mt76_dev *dev)
812{
813 ieee80211_iterate_active_interfaces_atomic(dev->hw,
814 IEEE80211_IFACE_ITER_RESUME_ALL,
815 __mt76_csa_check, dev);
816}
817EXPORT_SYMBOL_GPL(mt76_csa_check);
818
819int
820mt76_set_tim(struct ieee80211_hw *hw, struct ieee80211_sta *sta, bool set)
821{
822 return 0;
823}
824EXPORT_SYMBOL_GPL(mt76_set_tim);
825
826void mt76_insert_ccmp_hdr(struct sk_buff *skb, u8 key_id)
827{
828 struct mt76_rx_status *status = (struct mt76_rx_status *)skb->cb;
829 int hdr_len = ieee80211_get_hdrlen_from_skb(skb);
830 u8 *hdr, *pn = status->iv;
831
832 __skb_push(skb, 8);
833 memmove(skb->data, skb->data + 8, hdr_len);
834 hdr = skb->data + hdr_len;
835
836 hdr[0] = pn[5];
837 hdr[1] = pn[4];
838 hdr[2] = 0;
839 hdr[3] = 0x20 | (key_id << 6);
840 hdr[4] = pn[3];
841 hdr[5] = pn[2];
842 hdr[6] = pn[1];
843 hdr[7] = pn[0];
844
845 status->flag &= ~RX_FLAG_IV_STRIPPED;
846}
847EXPORT_SYMBOL_GPL(mt76_insert_ccmp_hdr);
848
849int mt76_get_rate(struct mt76_dev *dev,
850 struct ieee80211_supported_band *sband,
851 int idx, bool cck)
852{
853 int i, offset = 0, len = sband->n_bitrates;
854
855 if (cck) {
856 if (sband == &dev->sband_5g.sband)
857 return 0;
858
859 idx &= ~BIT(2);
860 } else if (sband == &dev->sband_2g.sband) {
861 offset = 4;
862 }
863
864 for (i = offset; i < len; i++) {
865 if ((sband->bitrates[i].hw_value & GENMASK(7, 0)) == idx)
866 return i;
867 }
868
869 return 0;
870}
871EXPORT_SYMBOL_GPL(mt76_get_rate);
872
873void mt76_sw_scan(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
874 const u8 *mac)
875{
876 struct mt76_dev *dev = hw->priv;
877
878 set_bit(MT76_SCANNING, &dev->state);
879}
880EXPORT_SYMBOL_GPL(mt76_sw_scan);
881
882void mt76_sw_scan_complete(struct ieee80211_hw *hw, struct ieee80211_vif *vif)
883{
884 struct mt76_dev *dev = hw->priv;
885
886 clear_bit(MT76_SCANNING, &dev->state);
887}
888EXPORT_SYMBOL_GPL(mt76_sw_scan_complete);
889
