16c92544dSBjoern A. Zeeb // SPDX-License-Identifier: ISC
26c92544dSBjoern A. Zeeb 
36c92544dSBjoern A. Zeeb #include "mt7603.h"
46c92544dSBjoern A. Zeeb 
56c92544dSBjoern A. Zeeb struct beacon_bc_data {
66c92544dSBjoern A. Zeeb 	struct mt7603_dev *dev;
76c92544dSBjoern A. Zeeb 	struct sk_buff_head q;
86c92544dSBjoern A. Zeeb 	struct sk_buff *tail[MT7603_MAX_INTERFACES];
96c92544dSBjoern A. Zeeb 	int count[MT7603_MAX_INTERFACES];
106c92544dSBjoern A. Zeeb };
116c92544dSBjoern A. Zeeb 
126c92544dSBjoern A. Zeeb static void
mt7603_update_beacon_iter(void * priv,u8 * mac,struct ieee80211_vif * vif)136c92544dSBjoern A. Zeeb mt7603_update_beacon_iter(void *priv, u8 *mac, struct ieee80211_vif *vif)
146c92544dSBjoern A. Zeeb {
156c92544dSBjoern A. Zeeb 	struct mt7603_dev *dev = (struct mt7603_dev *)priv;
166c92544dSBjoern A. Zeeb 	struct mt76_dev *mdev = &dev->mt76;
176c92544dSBjoern A. Zeeb 	struct mt7603_vif *mvif = (struct mt7603_vif *)vif->drv_priv;
186c92544dSBjoern A. Zeeb 	struct sk_buff *skb = NULL;
196c92544dSBjoern A. Zeeb 
206c92544dSBjoern A. Zeeb 	if (!(mdev->beacon_mask & BIT(mvif->idx)))
216c92544dSBjoern A. Zeeb 		return;
226c92544dSBjoern A. Zeeb 
236c92544dSBjoern A. Zeeb 	skb = ieee80211_beacon_get(mt76_hw(dev), vif, 0);
246c92544dSBjoern A. Zeeb 	if (!skb)
256c92544dSBjoern A. Zeeb 		return;
266c92544dSBjoern A. Zeeb 
276c92544dSBjoern A. Zeeb 	mt76_tx_queue_skb(dev, dev->mphy.q_tx[MT_TXQ_BEACON],
286c92544dSBjoern A. Zeeb 			  MT_TXQ_BEACON, skb, &mvif->sta.wcid, NULL);
296c92544dSBjoern A. Zeeb 
306c92544dSBjoern A. Zeeb 	spin_lock_bh(&dev->ps_lock);
316c92544dSBjoern A. Zeeb 	mt76_wr(dev, MT_DMA_FQCR0, MT_DMA_FQCR0_BUSY |
326c92544dSBjoern A. Zeeb 		FIELD_PREP(MT_DMA_FQCR0_TARGET_WCID, mvif->sta.wcid.idx) |
336c92544dSBjoern A. Zeeb 		FIELD_PREP(MT_DMA_FQCR0_TARGET_QID,
346c92544dSBjoern A. Zeeb 			   dev->mphy.q_tx[MT_TXQ_CAB]->hw_idx) |
356c92544dSBjoern A. Zeeb 		FIELD_PREP(MT_DMA_FQCR0_DEST_PORT_ID, 3) |
366c92544dSBjoern A. Zeeb 		FIELD_PREP(MT_DMA_FQCR0_DEST_QUEUE_ID, 8));
376c92544dSBjoern A. Zeeb 
386c92544dSBjoern A. Zeeb 	if (!mt76_poll(dev, MT_DMA_FQCR0, MT_DMA_FQCR0_BUSY, 0, 5000))
396c92544dSBjoern A. Zeeb 		dev->beacon_check = MT7603_WATCHDOG_TIMEOUT;
406c92544dSBjoern A. Zeeb 
416c92544dSBjoern A. Zeeb 	spin_unlock_bh(&dev->ps_lock);
426c92544dSBjoern A. Zeeb }
436c92544dSBjoern A. Zeeb 
446c92544dSBjoern A. Zeeb static void
mt7603_add_buffered_bc(void * priv,u8 * mac,struct ieee80211_vif * vif)456c92544dSBjoern A. Zeeb mt7603_add_buffered_bc(void *priv, u8 *mac, struct ieee80211_vif *vif)
466c92544dSBjoern A. Zeeb {
476c92544dSBjoern A. Zeeb 	struct beacon_bc_data *data = priv;
486c92544dSBjoern A. Zeeb 	struct mt7603_dev *dev = data->dev;
496c92544dSBjoern A. Zeeb 	struct mt7603_vif *mvif = (struct mt7603_vif *)vif->drv_priv;
506c92544dSBjoern A. Zeeb 	struct ieee80211_tx_info *info;
516c92544dSBjoern A. Zeeb 	struct sk_buff *skb;
526c92544dSBjoern A. Zeeb 
536c92544dSBjoern A. Zeeb 	if (!(dev->mt76.beacon_mask & BIT(mvif->idx)))
546c92544dSBjoern A. Zeeb 		return;
556c92544dSBjoern A. Zeeb 
566c92544dSBjoern A. Zeeb 	skb = ieee80211_get_buffered_bc(mt76_hw(dev), vif);
576c92544dSBjoern A. Zeeb 	if (!skb)
586c92544dSBjoern A. Zeeb 		return;
596c92544dSBjoern A. Zeeb 
606c92544dSBjoern A. Zeeb 	info = IEEE80211_SKB_CB(skb);
616c92544dSBjoern A. Zeeb 	info->control.vif = vif;
626c92544dSBjoern A. Zeeb 	info->flags |= IEEE80211_TX_CTL_ASSIGN_SEQ;
636c92544dSBjoern A. Zeeb 	mt76_skb_set_moredata(skb, true);
646c92544dSBjoern A. Zeeb 	__skb_queue_tail(&data->q, skb);
656c92544dSBjoern A. Zeeb 	data->tail[mvif->idx] = skb;
666c92544dSBjoern A. Zeeb 	data->count[mvif->idx]++;
676c92544dSBjoern A. Zeeb }
686c92544dSBjoern A. Zeeb 
mt7603_pre_tbtt_tasklet(struct tasklet_struct * t)696c92544dSBjoern A. Zeeb void mt7603_pre_tbtt_tasklet(struct tasklet_struct *t)
706c92544dSBjoern A. Zeeb {
716c92544dSBjoern A. Zeeb 	struct mt7603_dev *dev = from_tasklet(dev, t, mt76.pre_tbtt_tasklet);
726c92544dSBjoern A. Zeeb 	struct mt76_dev *mdev = &dev->mt76;
736c92544dSBjoern A. Zeeb 	struct mt76_queue *q;
746c92544dSBjoern A. Zeeb 	struct beacon_bc_data data = {};
756c92544dSBjoern A. Zeeb 	struct sk_buff *skb;
766c92544dSBjoern A. Zeeb 	int i, nframes;
776c92544dSBjoern A. Zeeb 
786c92544dSBjoern A. Zeeb 	if (mt76_hw(dev)->conf.flags & IEEE80211_CONF_OFFCHANNEL)
796c92544dSBjoern A. Zeeb 		return;
806c92544dSBjoern A. Zeeb 
816c92544dSBjoern A. Zeeb 	data.dev = dev;
826c92544dSBjoern A. Zeeb 	__skb_queue_head_init(&data.q);
836c92544dSBjoern A. Zeeb 
846c92544dSBjoern A. Zeeb 	q = dev->mphy.q_tx[MT_TXQ_BEACON];
856c92544dSBjoern A. Zeeb 	spin_lock(&q->lock);
866c92544dSBjoern A. Zeeb 	ieee80211_iterate_active_interfaces_atomic(mt76_hw(dev),
876c92544dSBjoern A. Zeeb 		IEEE80211_IFACE_ITER_RESUME_ALL,
886c92544dSBjoern A. Zeeb 		mt7603_update_beacon_iter, dev);
896c92544dSBjoern A. Zeeb 	mt76_queue_kick(dev, q);
906c92544dSBjoern A. Zeeb 	spin_unlock(&q->lock);
916c92544dSBjoern A. Zeeb 
926c92544dSBjoern A. Zeeb 	/* Flush all previous CAB queue packets */
936c92544dSBjoern A. Zeeb 	mt76_wr(dev, MT_WF_ARB_CAB_FLUSH, GENMASK(30, 16) | BIT(0));
946c92544dSBjoern A. Zeeb 
956c92544dSBjoern A. Zeeb 	mt76_queue_tx_cleanup(dev, dev->mphy.q_tx[MT_TXQ_CAB], false);
966c92544dSBjoern A. Zeeb 
976c92544dSBjoern A. Zeeb 	mt76_csa_check(mdev);
986c92544dSBjoern A. Zeeb 	if (mdev->csa_complete)
996c92544dSBjoern A. Zeeb 		goto out;
1006c92544dSBjoern A. Zeeb 
1016c92544dSBjoern A. Zeeb 	q = dev->mphy.q_tx[MT_TXQ_CAB];
1026c92544dSBjoern A. Zeeb 	do {
1036c92544dSBjoern A. Zeeb 		nframes = skb_queue_len(&data.q);
1046c92544dSBjoern A. Zeeb 		ieee80211_iterate_active_interfaces_atomic(mt76_hw(dev),
1056c92544dSBjoern A. Zeeb 			IEEE80211_IFACE_ITER_RESUME_ALL,
1066c92544dSBjoern A. Zeeb 			mt7603_add_buffered_bc, &data);
1076c92544dSBjoern A. Zeeb 	} while (nframes != skb_queue_len(&data.q) &&
1086c92544dSBjoern A. Zeeb 		 skb_queue_len(&data.q) < 8);
1096c92544dSBjoern A. Zeeb 
1106c92544dSBjoern A. Zeeb 	if (skb_queue_empty(&data.q))
1116c92544dSBjoern A. Zeeb 		goto out;
1126c92544dSBjoern A. Zeeb 
1136c92544dSBjoern A. Zeeb 	for (i = 0; i < ARRAY_SIZE(data.tail); i++) {
1146c92544dSBjoern A. Zeeb 		if (!data.tail[i])
1156c92544dSBjoern A. Zeeb 			continue;
1166c92544dSBjoern A. Zeeb 
1176c92544dSBjoern A. Zeeb 		mt76_skb_set_moredata(data.tail[i], false);
1186c92544dSBjoern A. Zeeb 	}
1196c92544dSBjoern A. Zeeb 
1206c92544dSBjoern A. Zeeb 	spin_lock(&q->lock);
1216c92544dSBjoern A. Zeeb 	while ((skb = __skb_dequeue(&data.q)) != NULL) {
1226c92544dSBjoern A. Zeeb 		struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
1236c92544dSBjoern A. Zeeb 		struct ieee80211_vif *vif = info->control.vif;
1246c92544dSBjoern A. Zeeb 		struct mt7603_vif *mvif = (struct mt7603_vif *)vif->drv_priv;
1256c92544dSBjoern A. Zeeb 
1266c92544dSBjoern A. Zeeb 		mt76_tx_queue_skb(dev, q, MT_TXQ_CAB, skb, &mvif->sta.wcid, NULL);
1276c92544dSBjoern A. Zeeb 	}
1286c92544dSBjoern A. Zeeb 	mt76_queue_kick(dev, q);
1296c92544dSBjoern A. Zeeb 	spin_unlock(&q->lock);
1306c92544dSBjoern A. Zeeb 
1316c92544dSBjoern A. Zeeb 	for (i = 0; i < ARRAY_SIZE(data.count); i++)
1326c92544dSBjoern A. Zeeb 		mt76_wr(dev, MT_WF_ARB_CAB_COUNT_B0_REG(i),
1336c92544dSBjoern A. Zeeb 			data.count[i] << MT_WF_ARB_CAB_COUNT_B0_SHIFT(i));
1346c92544dSBjoern A. Zeeb 
1356c92544dSBjoern A. Zeeb 	mt76_wr(dev, MT_WF_ARB_CAB_START,
1366c92544dSBjoern A. Zeeb 		MT_WF_ARB_CAB_START_BSSn(0) |
1376c92544dSBjoern A. Zeeb 		(MT_WF_ARB_CAB_START_BSS0n(1) *
1386c92544dSBjoern A. Zeeb 		 ((1 << (MT7603_MAX_INTERFACES - 1)) - 1)));
1396c92544dSBjoern A. Zeeb 
1406c92544dSBjoern A. Zeeb out:
1416c92544dSBjoern A. Zeeb 	mt76_queue_tx_cleanup(dev, dev->mphy.q_tx[MT_TXQ_BEACON], false);
1426c92544dSBjoern A. Zeeb 	if (dev->mphy.q_tx[MT_TXQ_BEACON]->queued > hweight8(mdev->beacon_mask))
1436c92544dSBjoern A. Zeeb 		dev->beacon_check++;
1446c92544dSBjoern A. Zeeb }
1456c92544dSBjoern A. Zeeb 
mt7603_beacon_set_timer(struct mt7603_dev * dev,int idx,int intval)1466c92544dSBjoern A. Zeeb void mt7603_beacon_set_timer(struct mt7603_dev *dev, int idx, int intval)
1476c92544dSBjoern A. Zeeb {
1486c92544dSBjoern A. Zeeb 	u32 pre_tbtt = MT7603_PRE_TBTT_TIME / 64;
1496c92544dSBjoern A. Zeeb 
1506c92544dSBjoern A. Zeeb 	if (idx >= 0) {
1516c92544dSBjoern A. Zeeb 		if (intval)
1526c92544dSBjoern A. Zeeb 			dev->mt76.beacon_mask |= BIT(idx);
1536c92544dSBjoern A. Zeeb 		else
1546c92544dSBjoern A. Zeeb 			dev->mt76.beacon_mask &= ~BIT(idx);
1556c92544dSBjoern A. Zeeb 	}
1566c92544dSBjoern A. Zeeb 
1576c92544dSBjoern A. Zeeb 	if (!dev->mt76.beacon_mask || (!intval && idx < 0)) {
1586c92544dSBjoern A. Zeeb 		mt7603_irq_disable(dev, MT_INT_MAC_IRQ3);
1596c92544dSBjoern A. Zeeb 		mt76_clear(dev, MT_ARB_SCR, MT_ARB_SCR_BCNQ_OPMODE_MASK);
1606c92544dSBjoern A. Zeeb 		mt76_wr(dev, MT_HW_INT_MASK(3), 0);
1616c92544dSBjoern A. Zeeb 		return;
1626c92544dSBjoern A. Zeeb 	}
1636c92544dSBjoern A. Zeeb 
164cbb3ec25SBjoern A. Zeeb 	if (intval)
1656c92544dSBjoern A. Zeeb 		dev->mt76.beacon_int = intval;
1666c92544dSBjoern A. Zeeb 	mt76_wr(dev, MT_TBTT,
1676c92544dSBjoern A. Zeeb 		FIELD_PREP(MT_TBTT_PERIOD, intval) | MT_TBTT_CAL_ENABLE);
1686c92544dSBjoern A. Zeeb 
1696c92544dSBjoern A. Zeeb 	mt76_wr(dev, MT_TBTT_TIMER_CFG, 0x99); /* start timer */
1706c92544dSBjoern A. Zeeb 
1716c92544dSBjoern A. Zeeb 	mt76_rmw_field(dev, MT_ARB_SCR, MT_ARB_SCR_BCNQ_OPMODE_MASK,
1726c92544dSBjoern A. Zeeb 		       MT_BCNQ_OPMODE_AP);
1736c92544dSBjoern A. Zeeb 	mt76_clear(dev, MT_ARB_SCR, MT_ARB_SCR_TBTT_BCN_PRIO);
1746c92544dSBjoern A. Zeeb 	mt76_set(dev, MT_ARB_SCR, MT_ARB_SCR_TBTT_BCAST_PRIO);
1756c92544dSBjoern A. Zeeb 
1766c92544dSBjoern A. Zeeb 	mt76_wr(dev, MT_PRE_TBTT, pre_tbtt);
1776c92544dSBjoern A. Zeeb 
1786c92544dSBjoern A. Zeeb 	mt76_set(dev, MT_HW_INT_MASK(3),
1796c92544dSBjoern A. Zeeb 		 MT_HW_INT3_PRE_TBTT0 | MT_HW_INT3_TBTT0);
1806c92544dSBjoern A. Zeeb 
1816c92544dSBjoern A. Zeeb 	mt76_set(dev, MT_WF_ARB_BCN_START,
1826c92544dSBjoern A. Zeeb 		 MT_WF_ARB_BCN_START_BSSn(0) |
1836c92544dSBjoern A. Zeeb 		 ((dev->mt76.beacon_mask >> 1) *
1846c92544dSBjoern A. Zeeb 		  MT_WF_ARB_BCN_START_BSS0n(1)));
1856c92544dSBjoern A. Zeeb 	mt7603_irq_enable(dev, MT_INT_MAC_IRQ3);
1866c92544dSBjoern A. Zeeb 
1876c92544dSBjoern A. Zeeb 	if (dev->mt76.beacon_mask & ~BIT(0))
1886c92544dSBjoern A. Zeeb 		mt76_set(dev, MT_LPON_SBTOR(0), MT_LPON_SBTOR_SUB_BSS_EN);
1896c92544dSBjoern A. Zeeb 	else
1906c92544dSBjoern A. Zeeb 		mt76_clear(dev, MT_LPON_SBTOR(0), MT_LPON_SBTOR_SUB_BSS_EN);
1916c92544dSBjoern A. Zeeb }
192