1/*
2 * Copyright (C) 2014 Felix Fietkau <nbd@openwrt.org>
3 * Copyright (C) 2015 Jakub Kicinski <kubakici@wp.pl>
4 *
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License version 2
7 * as published by the Free Software Foundation
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 * GNU General Public License for more details.
13 */
14
15#include "mt7601u.h"
16#include "trace.h"
17#include <linux/etherdevice.h>
18
19static void
20mt76_mac_process_tx_rate(struct ieee80211_tx_rate *txrate, u16 rate)
21{
22	u8 idx = MT76_GET(MT_TXWI_RATE_MCS, rate);
23
24	txrate->idx = 0;
25	txrate->flags = 0;
26	txrate->count = 1;
27
28	switch (MT76_GET(MT_TXWI_RATE_PHY_MODE, rate)) {
29	case MT_PHY_TYPE_OFDM:
30		txrate->idx = idx + 4;
31		return;
32	case MT_PHY_TYPE_CCK:
33		if (idx >= 8)
34			idx -= 8;
35
36		txrate->idx = idx;
37		return;
38	case MT_PHY_TYPE_HT_GF:
39		txrate->flags |= IEEE80211_TX_RC_GREEN_FIELD;
40		/* fall through */
41	case MT_PHY_TYPE_HT:
42		txrate->flags |= IEEE80211_TX_RC_MCS;
43		txrate->idx = idx;
44		break;
45	default:
46		WARN_ON(1);
47		return;
48	}
49
50	if (MT76_GET(MT_TXWI_RATE_BW, rate) == MT_PHY_BW_40)
51		txrate->flags |= IEEE80211_TX_RC_40_MHZ_WIDTH;
52
53	if (rate & MT_TXWI_RATE_SGI)
54		txrate->flags |= IEEE80211_TX_RC_SHORT_GI;
55}
56
57static void
58mt76_mac_fill_tx_status(struct mt7601u_dev *dev, struct ieee80211_tx_info *info,
59			struct mt76_tx_status *st)
60{
61	struct ieee80211_tx_rate *rate = info->status.rates;
62	int cur_idx, last_rate;
63	int i;
64
65	last_rate = min_t(int, st->retry, IEEE80211_TX_MAX_RATES - 1);
66	mt76_mac_process_tx_rate(&rate[last_rate], st->rate);
67	if (last_rate < IEEE80211_TX_MAX_RATES - 1)
68		rate[last_rate + 1].idx = -1;
69
70	cur_idx = rate[last_rate].idx + st->retry;
71	for (i = 0; i <= last_rate; i++) {
72		rate[i].flags = rate[last_rate].flags;
73		rate[i].idx = max_t(int, 0, cur_idx - i);
74		rate[i].count = 1;
75	}
76
77	if (last_rate > 0)
78		rate[last_rate - 1].count = st->retry + 1 - last_rate;
79
80	info->status.ampdu_len = 1;
81	info->status.ampdu_ack_len = st->success;
82
83	if (st->is_probe)
84		info->flags |= IEEE80211_TX_CTL_RATE_CTRL_PROBE;
85
86	if (st->aggr)
87		info->flags |= IEEE80211_TX_CTL_AMPDU |
88			       IEEE80211_TX_STAT_AMPDU;
89
90	if (!st->ack_req)
91		info->flags |= IEEE80211_TX_CTL_NO_ACK;
92	else if (st->success)
93		info->flags |= IEEE80211_TX_STAT_ACK;
94}
95
96u16 mt76_mac_tx_rate_val(struct mt7601u_dev *dev,
97			 const struct ieee80211_tx_rate *rate, u8 *nss_val)
98{
99	u16 rateval;
100	u8 phy, rate_idx;
101	u8 nss = 1;
102	u8 bw = 0;
103
104	if (rate->flags & IEEE80211_TX_RC_MCS) {
105		rate_idx = rate->idx;
106		nss = 1 + (rate->idx >> 3);
107		phy = MT_PHY_TYPE_HT;
108		if (rate->flags & IEEE80211_TX_RC_GREEN_FIELD)
109			phy = MT_PHY_TYPE_HT_GF;
110		if (rate->flags & IEEE80211_TX_RC_40_MHZ_WIDTH)
111			bw = 1;
112	} else {
113		const struct ieee80211_rate *r;
114		int band = dev->chandef.chan->band;
115		u16 val;
116
117		r = &dev->hw->wiphy->bands[band]->bitrates[rate->idx];
118		if (rate->flags & IEEE80211_TX_RC_USE_SHORT_PREAMBLE)
119			val = r->hw_value_short;
120		else
121			val = r->hw_value;
122
123		phy = val >> 8;
124		rate_idx = val & 0xff;
125		bw = 0;
126	}
127
128	rateval = MT76_SET(MT_RXWI_RATE_MCS, rate_idx);
129	rateval |= MT76_SET(MT_RXWI_RATE_PHY, phy);
130	rateval |= MT76_SET(MT_RXWI_RATE_BW, bw);
131	if (rate->flags & IEEE80211_TX_RC_SHORT_GI)
132		rateval |= MT_RXWI_RATE_SGI;
133
134	*nss_val = nss;
135	return rateval;
136}
137
138void mt76_mac_wcid_set_rate(struct mt7601u_dev *dev, struct mt76_wcid *wcid,
139			    const struct ieee80211_tx_rate *rate)
140{
141	unsigned long flags;
142
143	spin_lock_irqsave(&dev->lock, flags);
144	wcid->tx_rate = mt76_mac_tx_rate_val(dev, rate, &wcid->tx_rate_nss);
145	wcid->tx_rate_set = true;
146	spin_unlock_irqrestore(&dev->lock, flags);
147}
148
149struct mt76_tx_status mt7601u_mac_fetch_tx_status(struct mt7601u_dev *dev)
150{
151	struct mt76_tx_status stat = {};
152	u32 val;
153
154	val = mt7601u_rr(dev, MT_TX_STAT_FIFO);
155	stat.valid = !!(val & MT_TX_STAT_FIFO_VALID);
156	stat.success = !!(val & MT_TX_STAT_FIFO_SUCCESS);
157	stat.aggr = !!(val & MT_TX_STAT_FIFO_AGGR);
158	stat.ack_req = !!(val & MT_TX_STAT_FIFO_ACKREQ);
159	stat.pktid = MT76_GET(MT_TX_STAT_FIFO_PID_TYPE, val);
160	stat.wcid = MT76_GET(MT_TX_STAT_FIFO_WCID, val);
161	stat.rate = MT76_GET(MT_TX_STAT_FIFO_RATE, val);
162
163	return stat;
164}
165
166void mt76_send_tx_status(struct mt7601u_dev *dev, struct mt76_tx_status *stat)
167{
168	struct ieee80211_tx_info info = {};
169	struct ieee80211_sta *sta = NULL;
170	struct mt76_wcid *wcid = NULL;
171	void *msta;
172
173	rcu_read_lock();
174	if (stat->wcid < ARRAY_SIZE(dev->wcid))
175		wcid = rcu_dereference(dev->wcid[stat->wcid]);
176
177	if (wcid) {
178		msta = container_of(wcid, struct mt76_sta, wcid);
179		sta = container_of(msta, struct ieee80211_sta,
180				   drv_priv);
181	}
182
183	mt76_mac_fill_tx_status(dev, &info, stat);
184
185	spin_lock_bh(&dev->mac_lock);
186	ieee80211_tx_status_noskb(dev->hw, sta, &info);
187	spin_unlock_bh(&dev->mac_lock);
188
189	rcu_read_unlock();
190}
191
192void mt7601u_mac_set_protection(struct mt7601u_dev *dev, bool legacy_prot,
193				int ht_mode)
194{
195	int mode = ht_mode & IEEE80211_HT_OP_MODE_PROTECTION;
196	bool non_gf = !!(ht_mode & IEEE80211_HT_OP_MODE_NON_GF_STA_PRSNT);
197	u32 prot[6];
198	bool ht_rts[4] = {};
199	int i;
200
201	prot[0] = MT_PROT_NAV_SHORT |
202		  MT_PROT_TXOP_ALLOW_ALL |
203		  MT_PROT_RTS_THR_EN;
204	prot[1] = prot[0];
205	if (legacy_prot)
206		prot[1] |= MT_PROT_CTRL_CTS2SELF;
207
208	prot[2] = prot[4] = MT_PROT_NAV_SHORT | MT_PROT_TXOP_ALLOW_BW20;
209	prot[3] = prot[5] = MT_PROT_NAV_SHORT | MT_PROT_TXOP_ALLOW_ALL;
210
211	if (legacy_prot) {
212		prot[2] |= MT_PROT_RATE_CCK_11;
213		prot[3] |= MT_PROT_RATE_CCK_11;
214		prot[4] |= MT_PROT_RATE_CCK_11;
215		prot[5] |= MT_PROT_RATE_CCK_11;
216	} else {
217		prot[2] |= MT_PROT_RATE_OFDM_24;
218		prot[3] |= MT_PROT_RATE_DUP_OFDM_24;
219		prot[4] |= MT_PROT_RATE_OFDM_24;
220		prot[5] |= MT_PROT_RATE_DUP_OFDM_24;
221	}
222
223	switch (mode) {
224	case IEEE80211_HT_OP_MODE_PROTECTION_NONE:
225		break;
226
227	case IEEE80211_HT_OP_MODE_PROTECTION_NONMEMBER:
228		ht_rts[0] = ht_rts[1] = ht_rts[2] = ht_rts[3] = true;
229		break;
230
231	case IEEE80211_HT_OP_MODE_PROTECTION_20MHZ:
232		ht_rts[1] = ht_rts[3] = true;
233		break;
234
235	case IEEE80211_HT_OP_MODE_PROTECTION_NONHT_MIXED:
236		ht_rts[0] = ht_rts[1] = ht_rts[2] = ht_rts[3] = true;
237		break;
238	}
239
240	if (non_gf)
241		ht_rts[2] = ht_rts[3] = true;
242
243	for (i = 0; i < 4; i++)
244		if (ht_rts[i])
245			prot[i + 2] |= MT_PROT_CTRL_RTS_CTS;
246
247	for (i = 0; i < 6; i++)
248		mt7601u_wr(dev, MT_CCK_PROT_CFG + i * 4, prot[i]);
249}
250
251void mt7601u_mac_set_short_preamble(struct mt7601u_dev *dev, bool short_preamb)
252{
253	if (short_preamb)
254		mt76_set(dev, MT_AUTO_RSP_CFG, MT_AUTO_RSP_PREAMB_SHORT);
255	else
256		mt76_clear(dev, MT_AUTO_RSP_CFG, MT_AUTO_RSP_PREAMB_SHORT);
257}
258
259void mt7601u_mac_config_tsf(struct mt7601u_dev *dev, bool enable, int interval)
260{
261	u32 val = mt7601u_rr(dev, MT_BEACON_TIME_CFG);
262
263	val &= ~(MT_BEACON_TIME_CFG_TIMER_EN |
264		 MT_BEACON_TIME_CFG_SYNC_MODE |
265		 MT_BEACON_TIME_CFG_TBTT_EN);
266
267	if (!enable) {
268		mt7601u_wr(dev, MT_BEACON_TIME_CFG, val);
269		return;
270	}
271
272	val &= ~MT_BEACON_TIME_CFG_INTVAL;
273	val |= MT76_SET(MT_BEACON_TIME_CFG_INTVAL, interval << 4) |
274		MT_BEACON_TIME_CFG_TIMER_EN |
275		MT_BEACON_TIME_CFG_SYNC_MODE |
276		MT_BEACON_TIME_CFG_TBTT_EN;
277}
278
279static void mt7601u_check_mac_err(struct mt7601u_dev *dev)
280{
281	u32 val = mt7601u_rr(dev, 0x10f4);
282
283	if (!(val & BIT(29)) || !(val & (BIT(7) | BIT(5))))
284		return;
285
286	dev_err(dev->dev, "Error: MAC specific condition occurred\n");
287
288	mt76_set(dev, MT_MAC_SYS_CTRL, MT_MAC_SYS_CTRL_RESET_CSR);
289	udelay(10);
290	mt76_clear(dev, MT_MAC_SYS_CTRL, MT_MAC_SYS_CTRL_RESET_CSR);
291}
292
293void mt7601u_mac_work(struct work_struct *work)
294{
295	struct mt7601u_dev *dev = container_of(work, struct mt7601u_dev,
296					       mac_work.work);
297	struct {
298		u32 addr_base;
299		u32 span;
300		u64 *stat_base;
301	} spans[] = {
302		{ MT_RX_STA_CNT0,	3,	dev->stats.rx_stat },
303		{ MT_TX_STA_CNT0,	3,	dev->stats.tx_stat },
304		{ MT_TX_AGG_STAT,	1,	dev->stats.aggr_stat },
305		{ MT_MPDU_DENSITY_CNT,	1,	dev->stats.zero_len_del },
306		{ MT_TX_AGG_CNT_BASE0,	8,	&dev->stats.aggr_n[0] },
307		{ MT_TX_AGG_CNT_BASE1,	8,	&dev->stats.aggr_n[16] },
308	};
309	u32 sum, n;
310	int i, j, k;
311
312	/* Note: using MCU_RANDOM_READ is actually slower then reading all the
313	 *	 registers by hand.  MCU takes ca. 20ms to complete read of 24
314	 *	 registers while reading them one by one will takes roughly
315	 *	 24*200us =~ 5ms.
316	 */
317
318	k = 0;
319	n = 0;
320	sum = 0;
321	for (i = 0; i < ARRAY_SIZE(spans); i++)
322		for (j = 0; j < spans[i].span; j++) {
323			u32 val = mt7601u_rr(dev, spans[i].addr_base + j * 4);
324
325			spans[i].stat_base[j * 2] += val & 0xffff;
326			spans[i].stat_base[j * 2 + 1] += val >> 16;
327
328			/* Calculate average AMPDU length */
329			if (spans[i].addr_base != MT_TX_AGG_CNT_BASE0 &&
330			    spans[i].addr_base != MT_TX_AGG_CNT_BASE1)
331				continue;
332
333			n += (val >> 16) + (val & 0xffff);
334			sum += (val & 0xffff) * (1 + k * 2) +
335				(val >> 16) * (2 + k * 2);
336			k++;
337		}
338
339	atomic_set(&dev->avg_ampdu_len, n ? DIV_ROUND_CLOSEST(sum, n) : 1);
340
341	mt7601u_check_mac_err(dev);
342
343	ieee80211_queue_delayed_work(dev->hw, &dev->mac_work, 10 * HZ);
344}
345
346void
347mt7601u_mac_wcid_setup(struct mt7601u_dev *dev, u8 idx, u8 vif_idx, u8 *mac)
348{
349	u8 zmac[ETH_ALEN] = {};
350	u32 attr;
351
352	attr = MT76_SET(MT_WCID_ATTR_BSS_IDX, vif_idx & 7) |
353	       MT76_SET(MT_WCID_ATTR_BSS_IDX_EXT, !!(vif_idx & 8));
354
355	mt76_wr(dev, MT_WCID_ATTR(idx), attr);
356
357	if (mac)
358		memcpy(zmac, mac, sizeof(zmac));
359
360	mt7601u_addr_wr(dev, MT_WCID_ADDR(idx), zmac);
361}
362
363void mt7601u_mac_set_ampdu_factor(struct mt7601u_dev *dev)
364{
365	struct ieee80211_sta *sta;
366	struct mt76_wcid *wcid;
367	void *msta;
368	u8 min_factor = 3;
369	int i;
370
371	rcu_read_lock();
372	for (i = 0; i < ARRAY_SIZE(dev->wcid); i++) {
373		wcid = rcu_dereference(dev->wcid[i]);
374		if (!wcid)
375			continue;
376
377		msta = container_of(wcid, struct mt76_sta, wcid);
378		sta = container_of(msta, struct ieee80211_sta, drv_priv);
379
380		min_factor = min(min_factor, sta->ht_cap.ampdu_factor);
381	}
382	rcu_read_unlock();
383
384	mt7601u_wr(dev, MT_MAX_LEN_CFG, 0xa0fff |
385		   MT76_SET(MT_MAX_LEN_CFG_AMPDU, min_factor));
386}
387
388static void
389mt76_mac_process_rate(struct ieee80211_rx_status *status, u16 rate)
390{
391	u8 idx = MT76_GET(MT_RXWI_RATE_MCS, rate);
392
393	switch (MT76_GET(MT_RXWI_RATE_PHY, rate)) {
394	case MT_PHY_TYPE_OFDM:
395		if (WARN_ON(idx >= 8))
396			idx = 0;
397		idx += 4;
398
399		status->rate_idx = idx;
400		return;
401	case MT_PHY_TYPE_CCK:
402		if (idx >= 8) {
403			idx -= 8;
404			status->flag |= RX_FLAG_SHORTPRE;
405		}
406
407		if (WARN_ON(idx >= 4))
408			idx = 0;
409
410		status->rate_idx = idx;
411		return;
412	case MT_PHY_TYPE_HT_GF:
413		status->flag |= RX_FLAG_HT_GF;
414		/* fall through */
415	case MT_PHY_TYPE_HT:
416		status->flag |= RX_FLAG_HT;
417		status->rate_idx = idx;
418		break;
419	default:
420		WARN_ON(1);
421		return;
422	}
423
424	if (rate & MT_RXWI_RATE_SGI)
425		status->flag |= RX_FLAG_SHORT_GI;
426
427	if (rate & MT_RXWI_RATE_STBC)
428		status->flag |= 1 << RX_FLAG_STBC_SHIFT;
429
430	if (rate & MT_RXWI_RATE_BW)
431		status->flag |= RX_FLAG_40MHZ;
432}
433
434static void
435mt7601u_rx_monitor_beacon(struct mt7601u_dev *dev, struct mt7601u_rxwi *rxwi,
436			  u16 rate, int rssi)
437{
438	dev->bcn_freq_off = rxwi->freq_off;
439	dev->bcn_phy_mode = MT76_GET(MT_RXWI_RATE_PHY, rate);
440	dev->avg_rssi = (dev->avg_rssi * 15) / 16 + (rssi << 8);
441}
442
443static int
444mt7601u_rx_is_our_beacon(struct mt7601u_dev *dev, u8 *data)
445{
446	struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)data;
447
448	return ieee80211_is_beacon(hdr->frame_control) &&
449		ether_addr_equal(hdr->addr2, dev->ap_bssid);
450}
451
452u32 mt76_mac_process_rx(struct mt7601u_dev *dev, struct sk_buff *skb,
453			u8 *data, void *rxi)
454{
455	struct ieee80211_rx_status *status = IEEE80211_SKB_RXCB(skb);
456	struct mt7601u_rxwi *rxwi = rxi;
457	u32 len, ctl = le32_to_cpu(rxwi->ctl);
458	u16 rate = le16_to_cpu(rxwi->rate);
459	int rssi;
460
461	len = MT76_GET(MT_RXWI_CTL_MPDU_LEN, ctl);
462	if (len < 10)
463		return 0;
464
465	if (rxwi->rxinfo & cpu_to_le32(MT_RXINFO_DECRYPT)) {
466		status->flag |= RX_FLAG_DECRYPTED;
467		status->flag |= RX_FLAG_IV_STRIPPED | RX_FLAG_MMIC_STRIPPED;
468	}
469
470	status->chains = BIT(0);
471	rssi = mt7601u_phy_get_rssi(dev, rxwi, rate);
472	status->chain_signal[0] = status->signal = rssi;
473	status->freq = dev->chandef.chan->center_freq;
474	status->band = dev->chandef.chan->band;
475
476	mt76_mac_process_rate(status, rate);
477
478	spin_lock_bh(&dev->con_mon_lock);
479	if (mt7601u_rx_is_our_beacon(dev, data))
480		mt7601u_rx_monitor_beacon(dev, rxwi, rate, rssi);
481	else if (rxwi->rxinfo & cpu_to_le32(MT_RXINFO_U2M))
482		dev->avg_rssi = (dev->avg_rssi * 15) / 16 + (rssi << 8);
483	spin_unlock_bh(&dev->con_mon_lock);
484
485	return len;
486}
487
488static enum mt76_cipher_type
489mt76_mac_get_key_info(struct ieee80211_key_conf *key, u8 *key_data)
490{
491	memset(key_data, 0, 32);
492	if (!key)
493		return MT_CIPHER_NONE;
494
495	if (key->keylen > 32)
496		return MT_CIPHER_NONE;
497
498	memcpy(key_data, key->key, key->keylen);
499
500	switch (key->cipher) {
501	case WLAN_CIPHER_SUITE_WEP40:
502		return MT_CIPHER_WEP40;
503	case WLAN_CIPHER_SUITE_WEP104:
504		return MT_CIPHER_WEP104;
505	case WLAN_CIPHER_SUITE_TKIP:
506		return MT_CIPHER_TKIP;
507	case WLAN_CIPHER_SUITE_CCMP:
508		return MT_CIPHER_AES_CCMP;
509	default:
510		return MT_CIPHER_NONE;
511	}
512}
513
514int mt76_mac_wcid_set_key(struct mt7601u_dev *dev, u8 idx,
515			  struct ieee80211_key_conf *key)
516{
517	enum mt76_cipher_type cipher;
518	u8 key_data[32];
519	u8 iv_data[8];
520	u32 val;
521
522	cipher = mt76_mac_get_key_info(key, key_data);
523	if (cipher == MT_CIPHER_NONE && key)
524		return -EINVAL;
525
526	trace_set_key(dev, idx);
527
528	mt7601u_wr_copy(dev, MT_WCID_KEY(idx), key_data, sizeof(key_data));
529
530	memset(iv_data, 0, sizeof(iv_data));
531	if (key) {
532		iv_data[3] = key->keyidx << 6;
533		if (cipher >= MT_CIPHER_TKIP) {
534			/* Note: start with 1 to comply with spec,
535			 *	 (see comment on common/cmm_wpa.c:4291).
536			 */
537			iv_data[0] |= 1;
538			iv_data[3] |= 0x20;
539		}
540	}
541	mt7601u_wr_copy(dev, MT_WCID_IV(idx), iv_data, sizeof(iv_data));
542
543	val = mt7601u_rr(dev, MT_WCID_ATTR(idx));
544	val &= ~MT_WCID_ATTR_PKEY_MODE & ~MT_WCID_ATTR_PKEY_MODE_EXT;
545	val |= MT76_SET(MT_WCID_ATTR_PKEY_MODE, cipher & 7) |
546	       MT76_SET(MT_WCID_ATTR_PKEY_MODE_EXT, cipher >> 3);
547	val &= ~MT_WCID_ATTR_PAIRWISE;
548	val |= MT_WCID_ATTR_PAIRWISE *
549		!!(key && key->flags & IEEE80211_KEY_FLAG_PAIRWISE);
550	mt7601u_wr(dev, MT_WCID_ATTR(idx), val);
551
552	return 0;
553}
554
555int mt76_mac_shared_key_setup(struct mt7601u_dev *dev, u8 vif_idx, u8 key_idx,
556			      struct ieee80211_key_conf *key)
557{
558	enum mt76_cipher_type cipher;
559	u8 key_data[32];
560	u32 val;
561
562	cipher = mt76_mac_get_key_info(key, key_data);
563	if (cipher == MT_CIPHER_NONE && key)
564		return -EINVAL;
565
566	trace_set_shared_key(dev, vif_idx, key_idx);
567
568	mt7601u_wr_copy(dev, MT_SKEY(vif_idx, key_idx),
569			key_data, sizeof(key_data));
570
571	val = mt76_rr(dev, MT_SKEY_MODE(vif_idx));
572	val &= ~(MT_SKEY_MODE_MASK << MT_SKEY_MODE_SHIFT(vif_idx, key_idx));
573	val |= cipher << MT_SKEY_MODE_SHIFT(vif_idx, key_idx);
574	mt76_wr(dev, MT_SKEY_MODE(vif_idx), val);
575
576	return 0;
577}
578