162306a36Sopenharmony_ci// SPDX-License-Identifier: ISC
262306a36Sopenharmony_ci
362306a36Sopenharmony_ci#include "mt7603.h"
462306a36Sopenharmony_ci#include "mac.h"
562306a36Sopenharmony_ci#include "../dma.h"
662306a36Sopenharmony_ci
762306a36Sopenharmony_cistatic void
862306a36Sopenharmony_cimt7603_rx_loopback_skb(struct mt7603_dev *dev, struct sk_buff *skb)
962306a36Sopenharmony_ci{
1062306a36Sopenharmony_ci	static const u8 tid_to_ac[8] = {
1162306a36Sopenharmony_ci		IEEE80211_AC_BE,
1262306a36Sopenharmony_ci		IEEE80211_AC_BK,
1362306a36Sopenharmony_ci		IEEE80211_AC_BK,
1462306a36Sopenharmony_ci		IEEE80211_AC_BE,
1562306a36Sopenharmony_ci		IEEE80211_AC_VI,
1662306a36Sopenharmony_ci		IEEE80211_AC_VI,
1762306a36Sopenharmony_ci		IEEE80211_AC_VO,
1862306a36Sopenharmony_ci		IEEE80211_AC_VO
1962306a36Sopenharmony_ci	};
2062306a36Sopenharmony_ci	__le32 *txd = (__le32 *)skb->data;
2162306a36Sopenharmony_ci	struct ieee80211_hdr *hdr;
2262306a36Sopenharmony_ci	struct ieee80211_sta *sta;
2362306a36Sopenharmony_ci	struct mt7603_sta *msta;
2462306a36Sopenharmony_ci	struct mt76_wcid *wcid;
2562306a36Sopenharmony_ci	void *priv;
2662306a36Sopenharmony_ci	int idx;
2762306a36Sopenharmony_ci	u32 val;
2862306a36Sopenharmony_ci	u8 tid = 0;
2962306a36Sopenharmony_ci
3062306a36Sopenharmony_ci	if (skb->len < MT_TXD_SIZE + sizeof(struct ieee80211_hdr))
3162306a36Sopenharmony_ci		goto free;
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_ci	val = le32_to_cpu(txd[1]);
3462306a36Sopenharmony_ci	idx = FIELD_GET(MT_TXD1_WLAN_IDX, val);
3562306a36Sopenharmony_ci	skb->priority = FIELD_GET(MT_TXD1_TID, val);
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_ci	if (idx >= MT7603_WTBL_STA - 1)
3862306a36Sopenharmony_ci		goto free;
3962306a36Sopenharmony_ci
4062306a36Sopenharmony_ci	wcid = rcu_dereference(dev->mt76.wcid[idx]);
4162306a36Sopenharmony_ci	if (!wcid)
4262306a36Sopenharmony_ci		goto free;
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_ci	priv = msta = container_of(wcid, struct mt7603_sta, wcid);
4562306a36Sopenharmony_ci	val = le32_to_cpu(txd[0]);
4662306a36Sopenharmony_ci	val &= ~(MT_TXD0_P_IDX | MT_TXD0_Q_IDX);
4762306a36Sopenharmony_ci	val |= FIELD_PREP(MT_TXD0_Q_IDX, MT_TX_HW_QUEUE_MGMT);
4862306a36Sopenharmony_ci	txd[0] = cpu_to_le32(val);
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_ci	sta = container_of(priv, struct ieee80211_sta, drv_priv);
5162306a36Sopenharmony_ci	hdr = (struct ieee80211_hdr *)&skb->data[MT_TXD_SIZE];
5262306a36Sopenharmony_ci	if (ieee80211_is_data_qos(hdr->frame_control))
5362306a36Sopenharmony_ci		tid = *ieee80211_get_qos_ctl(hdr) &
5462306a36Sopenharmony_ci		      IEEE80211_QOS_CTL_TAG1D_MASK;
5562306a36Sopenharmony_ci	skb_set_queue_mapping(skb, tid_to_ac[tid]);
5662306a36Sopenharmony_ci	ieee80211_sta_set_buffered(sta, tid, true);
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_ci	spin_lock_bh(&dev->ps_lock);
5962306a36Sopenharmony_ci	__skb_queue_tail(&msta->psq, skb);
6062306a36Sopenharmony_ci	if (skb_queue_len(&msta->psq) >= 64) {
6162306a36Sopenharmony_ci		skb = __skb_dequeue(&msta->psq);
6262306a36Sopenharmony_ci		dev_kfree_skb(skb);
6362306a36Sopenharmony_ci	}
6462306a36Sopenharmony_ci	spin_unlock_bh(&dev->ps_lock);
6562306a36Sopenharmony_ci	return;
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_cifree:
6862306a36Sopenharmony_ci	dev_kfree_skb(skb);
6962306a36Sopenharmony_ci}
7062306a36Sopenharmony_ci
7162306a36Sopenharmony_civoid mt7603_queue_rx_skb(struct mt76_dev *mdev, enum mt76_rxq_id q,
7262306a36Sopenharmony_ci			 struct sk_buff *skb, u32 *info)
7362306a36Sopenharmony_ci{
7462306a36Sopenharmony_ci	struct mt7603_dev *dev = container_of(mdev, struct mt7603_dev, mt76);
7562306a36Sopenharmony_ci	__le32 *rxd = (__le32 *)skb->data;
7662306a36Sopenharmony_ci	__le32 *end = (__le32 *)&skb->data[skb->len];
7762306a36Sopenharmony_ci	enum rx_pkt_type type;
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_ci	type = le32_get_bits(rxd[0], MT_RXD0_PKT_TYPE);
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ci	if (q == MT_RXQ_MCU) {
8262306a36Sopenharmony_ci		if (type == PKT_TYPE_RX_EVENT)
8362306a36Sopenharmony_ci			mt76_mcu_rx_event(&dev->mt76, skb);
8462306a36Sopenharmony_ci		else
8562306a36Sopenharmony_ci			mt7603_rx_loopback_skb(dev, skb);
8662306a36Sopenharmony_ci		return;
8762306a36Sopenharmony_ci	}
8862306a36Sopenharmony_ci
8962306a36Sopenharmony_ci	switch (type) {
9062306a36Sopenharmony_ci	case PKT_TYPE_TXS:
9162306a36Sopenharmony_ci		for (rxd++; rxd + 5 <= end; rxd += 5)
9262306a36Sopenharmony_ci			mt7603_mac_add_txs(dev, rxd);
9362306a36Sopenharmony_ci		dev_kfree_skb(skb);
9462306a36Sopenharmony_ci		break;
9562306a36Sopenharmony_ci	case PKT_TYPE_RX_EVENT:
9662306a36Sopenharmony_ci		mt76_mcu_rx_event(&dev->mt76, skb);
9762306a36Sopenharmony_ci		return;
9862306a36Sopenharmony_ci	case PKT_TYPE_NORMAL:
9962306a36Sopenharmony_ci		if (mt7603_mac_fill_rx(dev, skb) == 0) {
10062306a36Sopenharmony_ci			mt76_rx(&dev->mt76, q, skb);
10162306a36Sopenharmony_ci			return;
10262306a36Sopenharmony_ci		}
10362306a36Sopenharmony_ci		fallthrough;
10462306a36Sopenharmony_ci	default:
10562306a36Sopenharmony_ci		dev_kfree_skb(skb);
10662306a36Sopenharmony_ci		break;
10762306a36Sopenharmony_ci	}
10862306a36Sopenharmony_ci}
10962306a36Sopenharmony_ci
11062306a36Sopenharmony_cistatic int
11162306a36Sopenharmony_cimt7603_init_rx_queue(struct mt7603_dev *dev, struct mt76_queue *q,
11262306a36Sopenharmony_ci		     int idx, int n_desc, int bufsize)
11362306a36Sopenharmony_ci{
11462306a36Sopenharmony_ci	int err;
11562306a36Sopenharmony_ci
11662306a36Sopenharmony_ci	err = mt76_queue_alloc(dev, q, idx, n_desc, bufsize,
11762306a36Sopenharmony_ci			       MT_RX_RING_BASE);
11862306a36Sopenharmony_ci	if (err < 0)
11962306a36Sopenharmony_ci		return err;
12062306a36Sopenharmony_ci
12162306a36Sopenharmony_ci	mt7603_irq_enable(dev, MT_INT_RX_DONE(idx));
12262306a36Sopenharmony_ci
12362306a36Sopenharmony_ci	return 0;
12462306a36Sopenharmony_ci}
12562306a36Sopenharmony_ci
12662306a36Sopenharmony_cistatic int mt7603_poll_tx(struct napi_struct *napi, int budget)
12762306a36Sopenharmony_ci{
12862306a36Sopenharmony_ci	struct mt7603_dev *dev;
12962306a36Sopenharmony_ci	int i;
13062306a36Sopenharmony_ci
13162306a36Sopenharmony_ci	dev = container_of(napi, struct mt7603_dev, mt76.tx_napi);
13262306a36Sopenharmony_ci	dev->tx_dma_check = 0;
13362306a36Sopenharmony_ci
13462306a36Sopenharmony_ci	mt76_queue_tx_cleanup(dev, dev->mt76.q_mcu[MT_MCUQ_WM], false);
13562306a36Sopenharmony_ci	for (i = MT_TXQ_PSD; i >= 0; i--)
13662306a36Sopenharmony_ci		mt76_queue_tx_cleanup(dev, dev->mphy.q_tx[i], false);
13762306a36Sopenharmony_ci
13862306a36Sopenharmony_ci	if (napi_complete_done(napi, 0))
13962306a36Sopenharmony_ci		mt7603_irq_enable(dev, MT_INT_TX_DONE_ALL);
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_ci	mt76_queue_tx_cleanup(dev, dev->mt76.q_mcu[MT_MCUQ_WM], false);
14262306a36Sopenharmony_ci	for (i = MT_TXQ_PSD; i >= 0; i--)
14362306a36Sopenharmony_ci		mt76_queue_tx_cleanup(dev, dev->mphy.q_tx[i], false);
14462306a36Sopenharmony_ci
14562306a36Sopenharmony_ci	mt7603_mac_sta_poll(dev);
14662306a36Sopenharmony_ci
14762306a36Sopenharmony_ci	mt76_worker_schedule(&dev->mt76.tx_worker);
14862306a36Sopenharmony_ci
14962306a36Sopenharmony_ci	return 0;
15062306a36Sopenharmony_ci}
15162306a36Sopenharmony_ci
15262306a36Sopenharmony_ciint mt7603_dma_init(struct mt7603_dev *dev)
15362306a36Sopenharmony_ci{
15462306a36Sopenharmony_ci	static const u8 wmm_queue_map[] = {
15562306a36Sopenharmony_ci		[IEEE80211_AC_BK] = 0,
15662306a36Sopenharmony_ci		[IEEE80211_AC_BE] = 1,
15762306a36Sopenharmony_ci		[IEEE80211_AC_VI] = 2,
15862306a36Sopenharmony_ci		[IEEE80211_AC_VO] = 3,
15962306a36Sopenharmony_ci	};
16062306a36Sopenharmony_ci	int ret;
16162306a36Sopenharmony_ci	int i;
16262306a36Sopenharmony_ci
16362306a36Sopenharmony_ci	mt76_dma_attach(&dev->mt76);
16462306a36Sopenharmony_ci
16562306a36Sopenharmony_ci	mt76_clear(dev, MT_WPDMA_GLO_CFG,
16662306a36Sopenharmony_ci		   MT_WPDMA_GLO_CFG_TX_DMA_EN |
16762306a36Sopenharmony_ci		   MT_WPDMA_GLO_CFG_RX_DMA_EN |
16862306a36Sopenharmony_ci		   MT_WPDMA_GLO_CFG_DMA_BURST_SIZE |
16962306a36Sopenharmony_ci		   MT_WPDMA_GLO_CFG_TX_WRITEBACK_DONE);
17062306a36Sopenharmony_ci
17162306a36Sopenharmony_ci	mt76_wr(dev, MT_WPDMA_RST_IDX, ~0);
17262306a36Sopenharmony_ci	mt7603_pse_client_reset(dev);
17362306a36Sopenharmony_ci
17462306a36Sopenharmony_ci	for (i = 0; i < ARRAY_SIZE(wmm_queue_map); i++) {
17562306a36Sopenharmony_ci		ret = mt76_init_tx_queue(&dev->mphy, i, wmm_queue_map[i],
17662306a36Sopenharmony_ci					 MT7603_TX_RING_SIZE, MT_TX_RING_BASE, 0);
17762306a36Sopenharmony_ci		if (ret)
17862306a36Sopenharmony_ci			return ret;
17962306a36Sopenharmony_ci	}
18062306a36Sopenharmony_ci
18162306a36Sopenharmony_ci	ret = mt76_init_tx_queue(&dev->mphy, MT_TXQ_PSD, MT_TX_HW_QUEUE_MGMT,
18262306a36Sopenharmony_ci				 MT7603_PSD_RING_SIZE, MT_TX_RING_BASE, 0);
18362306a36Sopenharmony_ci	if (ret)
18462306a36Sopenharmony_ci		return ret;
18562306a36Sopenharmony_ci
18662306a36Sopenharmony_ci	ret = mt76_init_mcu_queue(&dev->mt76, MT_MCUQ_WM, MT_TX_HW_QUEUE_MCU,
18762306a36Sopenharmony_ci				  MT_MCU_RING_SIZE, MT_TX_RING_BASE);
18862306a36Sopenharmony_ci	if (ret)
18962306a36Sopenharmony_ci		return ret;
19062306a36Sopenharmony_ci
19162306a36Sopenharmony_ci	ret = mt76_init_tx_queue(&dev->mphy, MT_TXQ_BEACON, MT_TX_HW_QUEUE_BCN,
19262306a36Sopenharmony_ci				 MT_MCU_RING_SIZE, MT_TX_RING_BASE, 0);
19362306a36Sopenharmony_ci	if (ret)
19462306a36Sopenharmony_ci		return ret;
19562306a36Sopenharmony_ci
19662306a36Sopenharmony_ci	ret = mt76_init_tx_queue(&dev->mphy, MT_TXQ_CAB, MT_TX_HW_QUEUE_BMC,
19762306a36Sopenharmony_ci				 MT_MCU_RING_SIZE, MT_TX_RING_BASE, 0);
19862306a36Sopenharmony_ci	if (ret)
19962306a36Sopenharmony_ci		return ret;
20062306a36Sopenharmony_ci
20162306a36Sopenharmony_ci	mt7603_irq_enable(dev,
20262306a36Sopenharmony_ci			  MT_INT_TX_DONE(IEEE80211_AC_VO) |
20362306a36Sopenharmony_ci			  MT_INT_TX_DONE(IEEE80211_AC_VI) |
20462306a36Sopenharmony_ci			  MT_INT_TX_DONE(IEEE80211_AC_BE) |
20562306a36Sopenharmony_ci			  MT_INT_TX_DONE(IEEE80211_AC_BK) |
20662306a36Sopenharmony_ci			  MT_INT_TX_DONE(MT_TX_HW_QUEUE_MGMT) |
20762306a36Sopenharmony_ci			  MT_INT_TX_DONE(MT_TX_HW_QUEUE_MCU) |
20862306a36Sopenharmony_ci			  MT_INT_TX_DONE(MT_TX_HW_QUEUE_BCN) |
20962306a36Sopenharmony_ci			  MT_INT_TX_DONE(MT_TX_HW_QUEUE_BMC));
21062306a36Sopenharmony_ci
21162306a36Sopenharmony_ci	ret = mt7603_init_rx_queue(dev, &dev->mt76.q_rx[MT_RXQ_MCU], 1,
21262306a36Sopenharmony_ci				   MT7603_MCU_RX_RING_SIZE, MT_RX_BUF_SIZE);
21362306a36Sopenharmony_ci	if (ret)
21462306a36Sopenharmony_ci		return ret;
21562306a36Sopenharmony_ci
21662306a36Sopenharmony_ci	ret = mt7603_init_rx_queue(dev, &dev->mt76.q_rx[MT_RXQ_MAIN], 0,
21762306a36Sopenharmony_ci				   MT7603_RX_RING_SIZE, MT_RX_BUF_SIZE);
21862306a36Sopenharmony_ci	if (ret)
21962306a36Sopenharmony_ci		return ret;
22062306a36Sopenharmony_ci
22162306a36Sopenharmony_ci	mt76_wr(dev, MT_DELAY_INT_CFG, 0);
22262306a36Sopenharmony_ci	ret = mt76_init_queues(dev, mt76_dma_rx_poll);
22362306a36Sopenharmony_ci	if (ret)
22462306a36Sopenharmony_ci		return ret;
22562306a36Sopenharmony_ci
22662306a36Sopenharmony_ci	netif_napi_add_tx(&dev->mt76.tx_napi_dev, &dev->mt76.tx_napi,
22762306a36Sopenharmony_ci			  mt7603_poll_tx);
22862306a36Sopenharmony_ci	napi_enable(&dev->mt76.tx_napi);
22962306a36Sopenharmony_ci
23062306a36Sopenharmony_ci	return 0;
23162306a36Sopenharmony_ci}
23262306a36Sopenharmony_ci
23362306a36Sopenharmony_civoid mt7603_dma_cleanup(struct mt7603_dev *dev)
23462306a36Sopenharmony_ci{
23562306a36Sopenharmony_ci	mt76_clear(dev, MT_WPDMA_GLO_CFG,
23662306a36Sopenharmony_ci		   MT_WPDMA_GLO_CFG_TX_DMA_EN |
23762306a36Sopenharmony_ci		   MT_WPDMA_GLO_CFG_RX_DMA_EN |
23862306a36Sopenharmony_ci		   MT_WPDMA_GLO_CFG_TX_WRITEBACK_DONE);
23962306a36Sopenharmony_ci
24062306a36Sopenharmony_ci	mt76_dma_cleanup(&dev->mt76);
24162306a36Sopenharmony_ci}
242