struct ieee80211_sta *sta)
 {
        struct mt76_queue *q = dev->q_tx[qid].q;
+       struct mt76_tx_info tx_info = {};
+       int len, n = 0, ret = -ENOMEM;
        struct mt76_queue_entry e;
        struct mt76_txwi_cache *t;
-       struct mt76_queue_buf buf[32];
-       int len, n = 0, ret = -ENOMEM;
        struct sk_buff *iter;
        dma_addr_t addr;
-       u32 tx_info = 0;
 
        t = mt76_get_txwi(dev);
        if (!t) {
        if (dma_mapping_error(dev->dev, addr))
                goto free;
 
-       buf[n].addr = t->dma_addr;
-       buf[n++].len = dev->drv->txwi_size;
-       buf[n].addr = addr;
-       buf[n++].len = len;
+       tx_info.buf[n].addr = t->dma_addr;
+       tx_info.buf[n++].len = dev->drv->txwi_size;
+       tx_info.buf[n].addr = addr;
+       tx_info.buf[n++].len = len;
 
        skb_walk_frags(skb, iter) {
-               if (n == ARRAY_SIZE(buf))
+               if (n == ARRAY_SIZE(tx_info.buf))
                        goto unmap;
 
                addr = dma_map_single(dev->dev, iter->data, iter->len,
                if (dma_mapping_error(dev->dev, addr))
                        goto unmap;
 
-               buf[n].addr = addr;
-               buf[n++].len = iter->len;
+               tx_info.buf[n].addr = addr;
+               tx_info.buf[n++].len = iter->len;
        }
+       tx_info.nbuf = n;
 
        dma_sync_single_for_cpu(dev->dev, t->dma_addr, sizeof(t->txwi),
                                DMA_TO_DEVICE);
        if (ret < 0)
                goto unmap;
 
-       if (q->queued + (n + 1) / 2 >= q->ndesc - 1) {
+       if (q->queued + (tx_info.nbuf + 1) / 2 >= q->ndesc - 1) {
                ret = -ENOMEM;
                goto unmap;
        }
 
-       return mt76_dma_add_buf(dev, q, buf, n, tx_info, skb, t);
+       return mt76_dma_add_buf(dev, q, tx_info.buf, tx_info.nbuf,
+                               tx_info.info, skb, t);
 
 unmap:
        for (n--; n > 0; n--)
-               dma_unmap_single(dev->dev, buf[n].addr, buf[n].len,
-                                DMA_TO_DEVICE);
+               dma_unmap_single(dev->dev, tx_info.buf[n].addr,
+                                tx_info.buf[n].len, DMA_TO_DEVICE);
 
 free:
        e.skb = skb;
 
        int len;
 };
 
+struct mt76_tx_info {
+       struct mt76_queue_buf buf[32];
+       int nbuf;
+       u32 info;
+};
+
 struct mt76u_buf {
        struct mt76_dev *dev;
        struct urb *urb;
        int (*tx_prepare_skb)(struct mt76_dev *dev, void *txwi_ptr,
                              struct sk_buff *skb, enum mt76_txq_id qid,
                              struct mt76_wcid *wcid,
-                             struct ieee80211_sta *sta, u32 *tx_info);
+                             struct ieee80211_sta *sta,
+                             struct mt76_tx_info *tx_info);
 
        void (*tx_complete_skb)(struct mt76_dev *dev, enum mt76_txq_id qid,
                                struct mt76_queue_entry *e);
 
 int mt7603_tx_prepare_skb(struct mt76_dev *mdev, void *txwi_ptr,
                          struct sk_buff *skb, enum mt76_txq_id qid,
                          struct mt76_wcid *wcid, struct ieee80211_sta *sta,
-                         u32 *tx_info)
+                         struct mt76_tx_info *tx_info)
 {
        struct mt7603_dev *dev = container_of(mdev, struct mt7603_dev, mt76);
        struct mt7603_sta *msta = container_of(wcid, struct mt7603_sta, wcid);
 
 int mt7603_tx_prepare_skb(struct mt76_dev *mdev, void *txwi_ptr,
                          struct sk_buff *skb, enum mt76_txq_id qid,
                          struct mt76_wcid *wcid, struct ieee80211_sta *sta,
-                         u32 *tx_info);
+                         struct mt76_tx_info *tx_info);
 
 void mt7603_tx_complete_skb(struct mt76_dev *mdev, enum mt76_txq_id qid,
                            struct mt76_queue_entry *e);
 
 int mt76x02_tx_prepare_skb(struct mt76_dev *mdev, void *txwi,
                           struct sk_buff *skb, enum mt76_txq_id qid,
                           struct mt76_wcid *wcid, struct ieee80211_sta *sta,
-                          u32 *tx_info);
+                          struct mt76_tx_info *tx_info);
 void mt76x02_sw_scan(struct ieee80211_hw *hw, struct ieee80211_vif *vif,
                     const u8 *mac);
 void mt76x02_sw_scan_complete(struct ieee80211_hw *hw,
 
 int mt76x02_tx_prepare_skb(struct mt76_dev *mdev, void *txwi_ptr,
                           struct sk_buff *skb, enum mt76_txq_id qid,
                           struct mt76_wcid *wcid, struct ieee80211_sta *sta,
-                          u32 *tx_info)
+                          struct mt76_tx_info *tx_info)
 {
        struct mt76x02_dev *dev = container_of(mdev, struct mt76x02_dev, mt76);
        struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
        if (pid >= MT_PACKET_ID_FIRST)
                qsel = MT_QSEL_MGMT;
 
-       *tx_info = FIELD_PREP(MT_TXD_INFO_QSEL, qsel) |
-                  MT_TXD_INFO_80211;
+       tx_info->info = FIELD_PREP(MT_TXD_INFO_QSEL, qsel) |
+                       MT_TXD_INFO_80211;
 
        if (!wcid || wcid->hw_key_idx == 0xff || wcid->sw_iv)
-               *tx_info |= MT_TXD_INFO_WIV;
+               tx_info->info |= MT_TXD_INFO_WIV;
 
        return 0;
 }
 
 int mt76x02u_tx_prepare_skb(struct mt76_dev *mdev, void *data,
                            struct sk_buff *skb, enum mt76_txq_id qid,
                            struct mt76_wcid *wcid, struct ieee80211_sta *sta,
-                           u32 *tx_info);
+                           struct mt76_tx_info *tx_info);
 void mt76x02u_tx_complete_skb(struct mt76_dev *mdev, enum mt76_txq_id qid,
                              struct mt76_queue_entry *e);
 #endif /* __MT76x02_USB_H */
 
 int mt76x02u_tx_prepare_skb(struct mt76_dev *mdev, void *data,
                            struct sk_buff *skb, enum mt76_txq_id qid,
                            struct mt76_wcid *wcid, struct ieee80211_sta *sta,
-                           u32 *tx_info)
+                           struct mt76_tx_info *tx_info)
 {
        struct mt76x02_dev *dev = container_of(mdev, struct mt76x02_dev, mt76);
        int pid, len = skb->len, ep = q2ep(mdev->q_tx[qid].q->hw_idx);