mt76: fix TSF value in probe responses

Like beacons, probe responses need a hardware-generated TSF value. Set
the flag that causes the hw to generate it

Signed-off-by: Felix Fietkau <nbd@nbd.name>
Signed-off-by: Kalle Valo <kvalo@codeaurora.org>

1 files changed, 5 insertions, 1 deletions

diff --git a/drivers/net/wireless/mediatek/mt76/mt76x2_mac.c b/drivers/net/wireless/mediatek/mt76/mt76x2_mac.c
index a7416a01baa4..98219b971463 100644
--- a/drivers/net/wireless/mediatek/mt76/mt76x2_mac.c
+++ b/drivers/net/wireless/mediatek/mt76/mt76x2_mac.c
@@ -172,6 +172,7 @@ void mt76x2_mac_write_txwi(struct mt76x2_dev *dev, struct mt76x2_txwi *txwi,
         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
         struct ieee80211_tx_rate *rate = &info->control.rates[0];
         struct ieee80211_key_conf *key = info->control.hw_key;
+        struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
         u16 rate_ht_mask = FIELD_PREP(MT_RXWI_RATE_PHY, BIT(1) | BIT(2));
         u16 txwi_flags = 0;
         u8 nss;
@@ -248,6 +249,10 @@ void mt76x2_mac_write_txwi(struct mt76x2_dev *dev, struct mt76x2_txwi *txwi,
                                     sta->ht_cap.ampdu_density);
         }
 
+        if (ieee80211_is_probe_resp(hdr->frame_control) ||
+            ieee80211_is_beacon(hdr->frame_control))
+                txwi_flags |= MT_TXWI_FLAGS_TS;
+
         txwi->flags |= cpu_to_le16(txwi_flags);
         txwi->len_ctl = cpu_to_le16(skb->len);
 }
@@ -634,7 +639,6 @@ mt76_write_beacon(struct mt76x2_dev *dev, int offset, struct sk_buff *skb)
                 return -ENOSPC;
 
         mt76x2_mac_write_txwi(dev, &txwi, skb, NULL, NULL);
-        txwi.flags |= cpu_to_le16(MT_TXWI_FLAGS_TS);
 
         mt76_wr_copy(dev, offset, &txwi, sizeof(txwi));
         offset += sizeof(txwi);