Get the RX queue inside mt76u_refill_rx. This will allow to reuse
mt76u_urb_alloc for TX allocations.

Signed-off-by: Stanislaw Gruszka <[email protected]>
---
 drivers/net/wireless/mediatek/mt76/usb.c | 19 ++++++++-----------
 1 file changed, 8 insertions(+), 11 deletions(-)

diff --git a/drivers/net/wireless/mediatek/mt76/usb.c 
b/drivers/net/wireless/mediatek/mt76/usb.c
index 48bbb4e3db2f..10507a26d598 100644
--- a/drivers/net/wireless/mediatek/mt76/usb.c
+++ b/drivers/net/wireless/mediatek/mt76/usb.c
@@ -283,12 +283,10 @@ mt76u_set_endpoints(struct usb_interface *intf,
 }
 
 static int
-mt76u_fill_rx_sg(struct mt76_dev *dev, struct urb *urb, int nsgs,
-                gfp_t gfp)
+mt76u_fill_rx_sg(struct mt76_dev *dev, struct mt76_queue *q, struct urb *urb,
+                int nsgs, gfp_t gfp)
 {
-       struct mt76_queue *q = &dev->q_rx[MT_RXQ_MAIN];
        int sglen = SKB_WITH_OVERHEAD(q->buf_size);
-
        int i;
 
        for (i = 0; i < nsgs; i++) {
@@ -321,11 +319,12 @@ mt76u_fill_rx_sg(struct mt76_dev *dev, struct urb *urb, 
int nsgs,
 }
 
 static int
-mt76u_refill_rx(struct mt76_dev *dev, struct mt76_queue *q,
-               struct urb *urb, int nsgs, gfp_t gfp)
+mt76u_refill_rx(struct mt76_dev *dev, struct urb *urb, int nsgs, gfp_t gfp)
 {
+       struct mt76_queue *q = &dev->q_rx[MT_RXQ_MAIN];
+
        if (dev->usb.sg_en) {
-               return mt76u_fill_rx_sg(dev, urb, nsgs, gfp);
+               return mt76u_fill_rx_sg(dev, q, urb, nsgs, gfp);
        } else {
                urb->transfer_buffer_length = SKB_WITH_OVERHEAD(q->buf_size);
                urb->transfer_buffer = page_frag_alloc(&q->rx_page,
@@ -337,7 +336,6 @@ mt76u_refill_rx(struct mt76_dev *dev, struct mt76_queue *q,
 static int
 mt76u_urb_alloc(struct mt76_dev *dev, struct mt76_queue_entry *e)
 {
-       struct mt76_queue *q = &dev->q_rx[MT_RXQ_MAIN];
        struct urb *urb;
 
        urb = usb_alloc_urb(0, GFP_KERNEL);
@@ -354,7 +352,7 @@ mt76u_urb_alloc(struct mt76_dev *dev, struct 
mt76_queue_entry *e)
                sg_init_table(urb->sg, MT_SG_MAX_SIZE);
        }
 
-       return mt76u_refill_rx(dev, q, urb, MT_SG_MAX_SIZE, GFP_KERNEL);
+       return mt76u_refill_rx(dev, urb, MT_SG_MAX_SIZE, GFP_KERNEL);
 }
 
 static void mt76u_urb_free(struct urb *urb)
@@ -521,8 +519,7 @@ static void mt76u_rx_tasklet(unsigned long data)
 
                count = mt76u_process_rx_entry(dev, urb);
                if (count > 0) {
-                       err = mt76u_refill_rx(dev, q, urb, count,
-                                             GFP_ATOMIC);
+                       err = mt76u_refill_rx(dev, urb, count, GFP_ATOMIC);
                        if (err < 0)
                                break;
                }
-- 
2.20.1

Reply via email to