[PATCH] mt76: mt76x02u: update ewma pkt len in mt76x02u_tx_prepare_skb

[Lorenzo Bianconi]

Update ewma packet length in mt76x02u_tx_prepare_skb as it is
done for pci counterpart in order to properly estimate tx time on
current channel

Signed-off-by: Lorenzo Bianconi <lore...@kernel.org>
---
 drivers/net/wireless/mediatek/mt76/mt76x02_usb_core.c | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/drivers/net/wireless/mediatek/mt76/mt76x02_usb_core.c 
b/drivers/net/wireless/mediatek/mt76/mt76x02_usb_core.c
index 4294ffc0478b..d03d3c8e296c 100644
--- a/drivers/net/wireless/mediatek/mt76/mt76x02_usb_core.c
+++ b/drivers/net/wireless/mediatek/mt76/mt76x02_usb_core.c
@@ -120,6 +120,12 @@ int mt76x02u_tx_prepare_skb(struct mt76_dev *mdev, void 
*data,
        if (!wcid || wcid->hw_key_idx == 0xff || wcid->sw_iv)
                flags |= MT_TXD_INFO_WIV;
 
+       if (sta) {
+               struct mt76x02_sta *msta = (struct mt76x02_sta *)sta->drv_priv;
+
+               ewma_pktlen_add(&msta->pktlen, tx_info->skb->len);
+       }
+
        return mt76x02u_skb_dma_info(tx_info->skb, WLAN_PORT, flags);
 }
 EXPORT_SYMBOL_GPL(mt76x02u_tx_prepare_skb);
-- 
2.21.0