17f17b86aSRyder Lee // SPDX-License-Identifier: ISC
2c8846e10SFelix Fietkau 
3c8846e10SFelix Fietkau #include "mt7603.h"
4c8846e10SFelix Fietkau 
5c8846e10SFelix Fietkau struct beacon_bc_data {
6c8846e10SFelix Fietkau 	struct mt7603_dev *dev;
7c8846e10SFelix Fietkau 	struct sk_buff_head q;
8c8846e10SFelix Fietkau 	struct sk_buff *tail[MT7603_MAX_INTERFACES];
9c8846e10SFelix Fietkau 	int count[MT7603_MAX_INTERFACES];
10c8846e10SFelix Fietkau };
11c8846e10SFelix Fietkau 
12c8846e10SFelix Fietkau static void
mt7603_mac_stuck_beacon_recovery(struct mt7603_dev * dev)13*8caa9dd3SFelix Fietkau mt7603_mac_stuck_beacon_recovery(struct mt7603_dev *dev)
14*8caa9dd3SFelix Fietkau {
15*8caa9dd3SFelix Fietkau 	if (dev->beacon_check % 5 != 4)
16*8caa9dd3SFelix Fietkau 		return;
17*8caa9dd3SFelix Fietkau 
18*8caa9dd3SFelix Fietkau 	mt76_clear(dev, MT_WPDMA_GLO_CFG, MT_WPDMA_GLO_CFG_TX_DMA_EN);
19*8caa9dd3SFelix Fietkau 	mt76_set(dev, MT_SCH_4, MT_SCH_4_RESET);
20*8caa9dd3SFelix Fietkau 	mt76_clear(dev, MT_SCH_4, MT_SCH_4_RESET);
21*8caa9dd3SFelix Fietkau 	mt76_set(dev, MT_WPDMA_GLO_CFG, MT_WPDMA_GLO_CFG_TX_DMA_EN);
22*8caa9dd3SFelix Fietkau 
23*8caa9dd3SFelix Fietkau 	mt76_set(dev, MT_WF_CFG_OFF_WOCCR, MT_WF_CFG_OFF_WOCCR_TMAC_GC_DIS);
24*8caa9dd3SFelix Fietkau 	mt76_set(dev, MT_ARB_SCR, MT_ARB_SCR_TX_DISABLE);
25*8caa9dd3SFelix Fietkau 	mt76_clear(dev, MT_ARB_SCR, MT_ARB_SCR_TX_DISABLE);
26*8caa9dd3SFelix Fietkau 	mt76_clear(dev, MT_WF_CFG_OFF_WOCCR, MT_WF_CFG_OFF_WOCCR_TMAC_GC_DIS);
27*8caa9dd3SFelix Fietkau }
28*8caa9dd3SFelix Fietkau 
29*8caa9dd3SFelix Fietkau static void
mt7603_update_beacon_iter(void * priv,u8 * mac,struct ieee80211_vif * vif)30c8846e10SFelix Fietkau mt7603_update_beacon_iter(void *priv, u8 *mac, struct ieee80211_vif *vif)
31c8846e10SFelix Fietkau {
32c8846e10SFelix Fietkau 	struct mt7603_dev *dev = (struct mt7603_dev *)priv;
3389870594SLorenzo Bianconi 	struct mt76_dev *mdev = &dev->mt76;
34c8846e10SFelix Fietkau 	struct mt7603_vif *mvif = (struct mt7603_vif *)vif->drv_priv;
35c8846e10SFelix Fietkau 	struct sk_buff *skb = NULL;
36*8caa9dd3SFelix Fietkau 	u32 om_idx = mvif->idx;
37*8caa9dd3SFelix Fietkau 	u32 val;
38c8846e10SFelix Fietkau 
3989870594SLorenzo Bianconi 	if (!(mdev->beacon_mask & BIT(mvif->idx)))
40c8846e10SFelix Fietkau 		return;
41c8846e10SFelix Fietkau 
426e8912a5SShaul Triebitz 	skb = ieee80211_beacon_get(mt76_hw(dev), vif, 0);
43c8846e10SFelix Fietkau 	if (!skb)
44c8846e10SFelix Fietkau 		return;
45c8846e10SFelix Fietkau 
46*8caa9dd3SFelix Fietkau 	if (om_idx)
47*8caa9dd3SFelix Fietkau 		om_idx |= 0x10;
48*8caa9dd3SFelix Fietkau 	val = MT_DMA_FQCR0_BUSY | MT_DMA_FQCR0_MODE |
49*8caa9dd3SFelix Fietkau 		FIELD_PREP(MT_DMA_FQCR0_TARGET_BSS, om_idx) |
50*8caa9dd3SFelix Fietkau 		FIELD_PREP(MT_DMA_FQCR0_DEST_PORT_ID, 3) |
51*8caa9dd3SFelix Fietkau 		FIELD_PREP(MT_DMA_FQCR0_DEST_QUEUE_ID, 8);
52*8caa9dd3SFelix Fietkau 
53*8caa9dd3SFelix Fietkau 	spin_lock_bh(&dev->ps_lock);
54*8caa9dd3SFelix Fietkau 
55*8caa9dd3SFelix Fietkau 	mt76_wr(dev, MT_DMA_FQCR0, val |
56*8caa9dd3SFelix Fietkau 		FIELD_PREP(MT_DMA_FQCR0_TARGET_QID, MT_TX_HW_QUEUE_BCN));
57*8caa9dd3SFelix Fietkau 	if (!mt76_poll(dev, MT_DMA_FQCR0, MT_DMA_FQCR0_BUSY, 0, 5000)) {
58*8caa9dd3SFelix Fietkau 		dev->beacon_check = MT7603_WATCHDOG_TIMEOUT;
59*8caa9dd3SFelix Fietkau 		goto out;
60*8caa9dd3SFelix Fietkau 	}
61*8caa9dd3SFelix Fietkau 
62*8caa9dd3SFelix Fietkau 	mt76_wr(dev, MT_DMA_FQCR0, val |
63*8caa9dd3SFelix Fietkau 		FIELD_PREP(MT_DMA_FQCR0_TARGET_QID, MT_TX_HW_QUEUE_BMC));
64*8caa9dd3SFelix Fietkau 	if (!mt76_poll(dev, MT_DMA_FQCR0, MT_DMA_FQCR0_BUSY, 0, 5000)) {
65*8caa9dd3SFelix Fietkau 		dev->beacon_check = MT7603_WATCHDOG_TIMEOUT;
66*8caa9dd3SFelix Fietkau 		goto out;
67*8caa9dd3SFelix Fietkau 	}
68*8caa9dd3SFelix Fietkau 
69d08295f5SFelix Fietkau 	mt76_tx_queue_skb(dev, dev->mphy.q_tx[MT_TXQ_BEACON],
70d08295f5SFelix Fietkau 			  MT_TXQ_BEACON, skb, &mvif->sta.wcid, NULL);
71c8846e10SFelix Fietkau 
72*8caa9dd3SFelix Fietkau out:
73c8846e10SFelix Fietkau 	spin_unlock_bh(&dev->ps_lock);
74c8846e10SFelix Fietkau }
75c8846e10SFelix Fietkau 
76c8846e10SFelix Fietkau static void
mt7603_add_buffered_bc(void * priv,u8 * mac,struct ieee80211_vif * vif)77c8846e10SFelix Fietkau mt7603_add_buffered_bc(void *priv, u8 *mac, struct ieee80211_vif *vif)
78c8846e10SFelix Fietkau {
79c8846e10SFelix Fietkau 	struct beacon_bc_data *data = priv;
80c8846e10SFelix Fietkau 	struct mt7603_dev *dev = data->dev;
81c8846e10SFelix Fietkau 	struct mt7603_vif *mvif = (struct mt7603_vif *)vif->drv_priv;
82c8846e10SFelix Fietkau 	struct ieee80211_tx_info *info;
83c8846e10SFelix Fietkau 	struct sk_buff *skb;
84c8846e10SFelix Fietkau 
85c8a04d98SLorenzo Bianconi 	if (!(dev->mt76.beacon_mask & BIT(mvif->idx)))
86c8846e10SFelix Fietkau 		return;
87c8846e10SFelix Fietkau 
88c8846e10SFelix Fietkau 	skb = ieee80211_get_buffered_bc(mt76_hw(dev), vif);
89c8846e10SFelix Fietkau 	if (!skb)
90c8846e10SFelix Fietkau 		return;
91c8846e10SFelix Fietkau 
92c8846e10SFelix Fietkau 	info = IEEE80211_SKB_CB(skb);
93c8846e10SFelix Fietkau 	info->control.vif = vif;
94c8846e10SFelix Fietkau 	info->flags |= IEEE80211_TX_CTL_ASSIGN_SEQ;
95c8846e10SFelix Fietkau 	mt76_skb_set_moredata(skb, true);
96c8846e10SFelix Fietkau 	__skb_queue_tail(&data->q, skb);
97c8846e10SFelix Fietkau 	data->tail[mvif->idx] = skb;
98c8846e10SFelix Fietkau 	data->count[mvif->idx]++;
99c8846e10SFelix Fietkau }
100c8846e10SFelix Fietkau 
mt7603_pre_tbtt_tasklet(struct tasklet_struct * t)1015ee3e780SAllen Pais void mt7603_pre_tbtt_tasklet(struct tasklet_struct *t)
102c8846e10SFelix Fietkau {
1035ee3e780SAllen Pais 	struct mt7603_dev *dev = from_tasklet(dev, t, mt76.pre_tbtt_tasklet);
10489870594SLorenzo Bianconi 	struct mt76_dev *mdev = &dev->mt76;
105c8846e10SFelix Fietkau 	struct mt76_queue *q;
106c8846e10SFelix Fietkau 	struct beacon_bc_data data = {};
107c8846e10SFelix Fietkau 	struct sk_buff *skb;
108c8846e10SFelix Fietkau 	int i, nframes;
109c8846e10SFelix Fietkau 
110bd115805SLorenzo Bianconi 	if (mt76_hw(dev)->conf.flags & IEEE80211_CONF_OFFCHANNEL)
111bd115805SLorenzo Bianconi 		return;
112bd115805SLorenzo Bianconi 
113c8846e10SFelix Fietkau 	data.dev = dev;
114c8846e10SFelix Fietkau 	__skb_queue_head_init(&data.q);
115c8846e10SFelix Fietkau 
116*8caa9dd3SFelix Fietkau 	/* Flush all previous CAB queue packets and beacons */
117*8caa9dd3SFelix Fietkau 	mt76_wr(dev, MT_WF_ARB_CAB_FLUSH, GENMASK(30, 16) | BIT(0));
118*8caa9dd3SFelix Fietkau 
119*8caa9dd3SFelix Fietkau 	mt76_queue_tx_cleanup(dev, dev->mphy.q_tx[MT_TXQ_CAB], false);
120*8caa9dd3SFelix Fietkau 	mt76_queue_tx_cleanup(dev, dev->mphy.q_tx[MT_TXQ_BEACON], false);
121*8caa9dd3SFelix Fietkau 
122*8caa9dd3SFelix Fietkau 	if (dev->mphy.q_tx[MT_TXQ_BEACON]->queued > 0)
123*8caa9dd3SFelix Fietkau 		dev->beacon_check++;
124*8caa9dd3SFelix Fietkau 	else
125*8caa9dd3SFelix Fietkau 		dev->beacon_check = 0;
126*8caa9dd3SFelix Fietkau 	mt7603_mac_stuck_beacon_recovery(dev);
127*8caa9dd3SFelix Fietkau 
12891990519SLorenzo Bianconi 	q = dev->mphy.q_tx[MT_TXQ_BEACON];
129b61cc2a7SYunbo Yu 	spin_lock(&q->lock);
130c8846e10SFelix Fietkau 	ieee80211_iterate_active_interfaces_atomic(mt76_hw(dev),
131c8846e10SFelix Fietkau 		IEEE80211_IFACE_ITER_RESUME_ALL,
132c8846e10SFelix Fietkau 		mt7603_update_beacon_iter, dev);
133c8846e10SFelix Fietkau 	mt76_queue_kick(dev, q);
134b61cc2a7SYunbo Yu 	spin_unlock(&q->lock);
135c8846e10SFelix Fietkau 
13689870594SLorenzo Bianconi 	mt76_csa_check(mdev);
13789870594SLorenzo Bianconi 	if (mdev->csa_complete)
138*8caa9dd3SFelix Fietkau 		return;
139c8846e10SFelix Fietkau 
14091990519SLorenzo Bianconi 	q = dev->mphy.q_tx[MT_TXQ_CAB];
141c8846e10SFelix Fietkau 	do {
142c8846e10SFelix Fietkau 		nframes = skb_queue_len(&data.q);
143c8846e10SFelix Fietkau 		ieee80211_iterate_active_interfaces_atomic(mt76_hw(dev),
144c8846e10SFelix Fietkau 			IEEE80211_IFACE_ITER_RESUME_ALL,
145c8846e10SFelix Fietkau 			mt7603_add_buffered_bc, &data);
146c8846e10SFelix Fietkau 	} while (nframes != skb_queue_len(&data.q) &&
147c8846e10SFelix Fietkau 		 skb_queue_len(&data.q) < 8);
148c8846e10SFelix Fietkau 
149c8846e10SFelix Fietkau 	if (skb_queue_empty(&data.q))
150*8caa9dd3SFelix Fietkau 		return;
151c8846e10SFelix Fietkau 
152c8846e10SFelix Fietkau 	for (i = 0; i < ARRAY_SIZE(data.tail); i++) {
153c8846e10SFelix Fietkau 		if (!data.tail[i])
154c8846e10SFelix Fietkau 			continue;
155c8846e10SFelix Fietkau 
156c8846e10SFelix Fietkau 		mt76_skb_set_moredata(data.tail[i], false);
157c8846e10SFelix Fietkau 	}
158c8846e10SFelix Fietkau 
159b61cc2a7SYunbo Yu 	spin_lock(&q->lock);
160c8846e10SFelix Fietkau 	while ((skb = __skb_dequeue(&data.q)) != NULL) {
161c8846e10SFelix Fietkau 		struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
162c8846e10SFelix Fietkau 		struct ieee80211_vif *vif = info->control.vif;
163c8846e10SFelix Fietkau 		struct mt7603_vif *mvif = (struct mt7603_vif *)vif->drv_priv;
164c8846e10SFelix Fietkau 
165d08295f5SFelix Fietkau 		mt76_tx_queue_skb(dev, q, MT_TXQ_CAB, skb, &mvif->sta.wcid, NULL);
166c8846e10SFelix Fietkau 	}
167c8846e10SFelix Fietkau 	mt76_queue_kick(dev, q);
168b61cc2a7SYunbo Yu 	spin_unlock(&q->lock);
169c8846e10SFelix Fietkau 
170c8846e10SFelix Fietkau 	for (i = 0; i < ARRAY_SIZE(data.count); i++)
171c8846e10SFelix Fietkau 		mt76_wr(dev, MT_WF_ARB_CAB_COUNT_B0_REG(i),
172c8846e10SFelix Fietkau 			data.count[i] << MT_WF_ARB_CAB_COUNT_B0_SHIFT(i));
173c8846e10SFelix Fietkau 
174c8846e10SFelix Fietkau 	mt76_wr(dev, MT_WF_ARB_CAB_START,
175c8846e10SFelix Fietkau 		MT_WF_ARB_CAB_START_BSSn(0) |
176c8846e10SFelix Fietkau 		(MT_WF_ARB_CAB_START_BSS0n(1) *
177c8846e10SFelix Fietkau 		 ((1 << (MT7603_MAX_INTERFACES - 1)) - 1)));
178c8846e10SFelix Fietkau }
179c8846e10SFelix Fietkau 
mt7603_beacon_set_timer(struct mt7603_dev * dev,int idx,int intval)180c8846e10SFelix Fietkau void mt7603_beacon_set_timer(struct mt7603_dev *dev, int idx, int intval)
181c8846e10SFelix Fietkau {
182c8846e10SFelix Fietkau 	u32 pre_tbtt = MT7603_PRE_TBTT_TIME / 64;
183c8846e10SFelix Fietkau 
184c8846e10SFelix Fietkau 	if (idx >= 0) {
185c8846e10SFelix Fietkau 		if (intval)
186c8a04d98SLorenzo Bianconi 			dev->mt76.beacon_mask |= BIT(idx);
187c8846e10SFelix Fietkau 		else
188c8a04d98SLorenzo Bianconi 			dev->mt76.beacon_mask &= ~BIT(idx);
189c8846e10SFelix Fietkau 	}
190c8846e10SFelix Fietkau 
191c8a04d98SLorenzo Bianconi 	if (!dev->mt76.beacon_mask || (!intval && idx < 0)) {
192c8846e10SFelix Fietkau 		mt7603_irq_disable(dev, MT_INT_MAC_IRQ3);
193c8846e10SFelix Fietkau 		mt76_clear(dev, MT_ARB_SCR, MT_ARB_SCR_BCNQ_OPMODE_MASK);
194c8846e10SFelix Fietkau 		mt76_wr(dev, MT_HW_INT_MASK(3), 0);
195c8846e10SFelix Fietkau 		return;
196c8846e10SFelix Fietkau 	}
197c8846e10SFelix Fietkau 
198f090d0caSFelix Fietkau 	if (intval)
1993041c445SLorenzo Bianconi 		dev->mt76.beacon_int = intval;
200c8846e10SFelix Fietkau 	mt76_wr(dev, MT_TBTT,
201c8846e10SFelix Fietkau 		FIELD_PREP(MT_TBTT_PERIOD, intval) | MT_TBTT_CAL_ENABLE);
202c8846e10SFelix Fietkau 
203c8846e10SFelix Fietkau 	mt76_wr(dev, MT_TBTT_TIMER_CFG, 0x99); /* start timer */
204c8846e10SFelix Fietkau 
205c8846e10SFelix Fietkau 	mt76_rmw_field(dev, MT_ARB_SCR, MT_ARB_SCR_BCNQ_OPMODE_MASK,
206c8846e10SFelix Fietkau 		       MT_BCNQ_OPMODE_AP);
207c8846e10SFelix Fietkau 	mt76_clear(dev, MT_ARB_SCR, MT_ARB_SCR_TBTT_BCN_PRIO);
208c8846e10SFelix Fietkau 	mt76_set(dev, MT_ARB_SCR, MT_ARB_SCR_TBTT_BCAST_PRIO);
209c8846e10SFelix Fietkau 
210c8846e10SFelix Fietkau 	mt76_wr(dev, MT_PRE_TBTT, pre_tbtt);
211c8846e10SFelix Fietkau 
212c8846e10SFelix Fietkau 	mt76_set(dev, MT_HW_INT_MASK(3),
213c8846e10SFelix Fietkau 		 MT_HW_INT3_PRE_TBTT0 | MT_HW_INT3_TBTT0);
214c8846e10SFelix Fietkau 
215c8846e10SFelix Fietkau 	mt76_set(dev, MT_WF_ARB_BCN_START,
216c8846e10SFelix Fietkau 		 MT_WF_ARB_BCN_START_BSSn(0) |
217c8a04d98SLorenzo Bianconi 		 ((dev->mt76.beacon_mask >> 1) *
218c8a04d98SLorenzo Bianconi 		  MT_WF_ARB_BCN_START_BSS0n(1)));
219c8846e10SFelix Fietkau 	mt7603_irq_enable(dev, MT_INT_MAC_IRQ3);
220c8846e10SFelix Fietkau 
221c8a04d98SLorenzo Bianconi 	if (dev->mt76.beacon_mask & ~BIT(0))
222c8846e10SFelix Fietkau 		mt76_set(dev, MT_LPON_SBTOR(0), MT_LPON_SBTOR_SUB_BSS_EN);
223c8846e10SFelix Fietkau 	else
224c8846e10SFelix Fietkau 		mt76_clear(dev, MT_LPON_SBTOR(0), MT_LPON_SBTOR_SUB_BSS_EN);
225c8846e10SFelix Fietkau }
226