Extract the initialization and destruction actions
of dim for use in the next patch.

Signed-off-by: Heng Qi <hen...@linux.alibaba.com>
---
 drivers/net/virtio_net.c | 38 +++++++++++++++++++++++++++-----------
 1 file changed, 27 insertions(+), 11 deletions(-)

diff --git a/drivers/net/virtio_net.c b/drivers/net/virtio_net.c
index 115c3c5414f2..858d7ab89e34 100644
--- a/drivers/net/virtio_net.c
+++ b/drivers/net/virtio_net.c
@@ -2274,6 +2274,13 @@ static int virtnet_enable_queue_pair(struct virtnet_info 
*vi, int qp_index)
        return err;
 }
 
+static void virtnet_dim_clean(struct virtnet_info *vi,
+                             int start_qnum, int end_qnum)
+{
+       for (; start_qnum <= end_qnum; start_qnum++)
+               cancel_work_sync(&vi->rq[start_qnum].dim.work);
+}
+
 static int virtnet_open(struct net_device *dev)
 {
        struct virtnet_info *vi = netdev_priv(dev);
@@ -2297,11 +2304,9 @@ static int virtnet_open(struct net_device *dev)
 err_enable_qp:
        disable_delayed_refill(vi);
        cancel_delayed_work_sync(&vi->refill);
-
-       for (i--; i >= 0; i--) {
+       virtnet_dim_clean(vi, 0, i - 1);
+       for (i--; i >= 0; i--)
                virtnet_disable_queue_pair(vi, i);
-               cancel_work_sync(&vi->rq[i].dim.work);
-       }
 
        return err;
 }
@@ -2466,7 +2471,7 @@ static int virtnet_rx_resize(struct virtnet_info *vi,
 
        if (running) {
                napi_disable(&rq->napi);
-               cancel_work_sync(&rq->dim.work);
+               virtnet_dim_clean(vi, qindex, qindex);
        }
 
        err = virtqueue_resize(rq->vq, ring_num, virtnet_rq_unmap_free_buf);
@@ -2716,10 +2721,9 @@ static int virtnet_close(struct net_device *dev)
        /* Make sure refill_work doesn't re-enable napi! */
        cancel_delayed_work_sync(&vi->refill);
 
-       for (i = 0; i < vi->max_queue_pairs; i++) {
+       virtnet_dim_clean(vi, 0, vi->max_queue_pairs - 1);
+       for (i = 0; i < vi->max_queue_pairs; i++)
                virtnet_disable_queue_pair(vi, i);
-               cancel_work_sync(&vi->rq[i].dim.work);
-       }
 
        return 0;
 }
@@ -4434,6 +4438,19 @@ static int virtnet_find_vqs(struct virtnet_info *vi)
        return ret;
 }
 
+static void virtnet_dim_init(struct virtnet_info *vi)
+{
+       int i;
+
+       if (!virtio_has_feature(vi->vdev, VIRTIO_NET_F_VQ_NOTF_COAL))
+               return;
+
+       for (i = 0; i < vi->max_queue_pairs; i++) {
+               INIT_WORK(&vi->rq[i].dim.work, virtnet_rx_dim_work);
+               vi->rq[i].dim.mode = DIM_CQ_PERIOD_MODE_START_FROM_EQE;
+       }
+}
+
 static int virtnet_alloc_queues(struct virtnet_info *vi)
 {
        int i;
@@ -4461,9 +4478,6 @@ static int virtnet_alloc_queues(struct virtnet_info *vi)
                                         virtnet_poll_tx,
                                         napi_tx ? napi_weight : 0);
 
-               INIT_WORK(&vi->rq[i].dim.work, virtnet_rx_dim_work);
-               vi->rq[i].dim.mode = DIM_CQ_PERIOD_MODE_START_FROM_EQE;
-
                sg_init_table(vi->rq[i].sg, ARRAY_SIZE(vi->rq[i].sg));
                ewma_pkt_len_init(&vi->rq[i].mrg_avg_pkt_len);
                sg_init_table(vi->sq[i].sg, ARRAY_SIZE(vi->sq[i].sg));
@@ -4873,6 +4887,8 @@ static int virtnet_probe(struct virtio_device *vdev)
 
        virtio_device_ready(vdev);
 
+       virtnet_dim_init(vi);
+
        _virtnet_set_queues(vi, vi->curr_queue_pairs);
 
        /* a random MAC address has been assigned, notify the device.
-- 
2.32.0.3.g01195cf9f


Reply via email to