setup_vq replaces vring_create_virtuque() with vring_setup_virtqueue() to
support the need to enable reset vq.

Signed-off-by: Xuan Zhuo <xuanz...@linux.alibaba.com>
---
 drivers/virtio/virtio_pci_common.c |  8 ++++----
 drivers/virtio/virtio_pci_common.h |  4 +++-
 drivers/virtio/virtio_pci_legacy.c |  5 +++--
 drivers/virtio/virtio_pci_modern.c | 20 ++++++++++++++------
 4 files changed, 24 insertions(+), 13 deletions(-)

diff --git a/drivers/virtio/virtio_pci_common.c 
b/drivers/virtio/virtio_pci_common.c
index 6b2573ec1ae8..c02936d29a31 100644
--- a/drivers/virtio/virtio_pci_common.c
+++ b/drivers/virtio/virtio_pci_common.c
@@ -209,7 +209,7 @@ static struct virtqueue *vp_setup_vq(struct virtio_device 
*vdev, unsigned index,
                                     void (*callback)(struct virtqueue *vq),
                                     const char *name,
                                     bool ctx,
-                                    u16 msix_vec)
+                                    u16 msix_vec, u16 num)
 {
        struct virtio_pci_device *vp_dev = to_vp_device(vdev);
        struct virtio_pci_vq_info *info = kmalloc(sizeof *info, GFP_KERNEL);
@@ -221,7 +221,7 @@ static struct virtqueue *vp_setup_vq(struct virtio_device 
*vdev, unsigned index,
                return ERR_PTR(-ENOMEM);
 
        vq = vp_dev->setup_vq(vp_dev, info, index, callback, name, ctx,
-                             msix_vec);
+                             msix_vec, NULL, num);
        if (IS_ERR(vq))
                goto out_info;
 
@@ -368,7 +368,7 @@ static int vp_find_vqs_msix(struct virtio_device *vdev, 
unsigned nvqs,
                        msix_vec = VP_MSIX_VQ_VECTOR;
                vqs[i] = vp_setup_vq(vdev, queue_idx++, callbacks[i], names[i],
                                     ctx ? ctx[i] : false,
-                                    msix_vec);
+                                    msix_vec, 0);
                if (IS_ERR(vqs[i])) {
                        err = PTR_ERR(vqs[i]);
                        goto error_find;
@@ -423,7 +423,7 @@ static int vp_find_vqs_intx(struct virtio_device *vdev, 
unsigned nvqs,
                }
                vqs[i] = vp_setup_vq(vdev, queue_idx++, callbacks[i], names[i],
                                     ctx ? ctx[i] : false,
-                                    VIRTIO_MSI_NO_VECTOR);
+                                    VIRTIO_MSI_NO_VECTOR, 0);
                if (IS_ERR(vqs[i])) {
                        err = PTR_ERR(vqs[i]);
                        goto out_del_vqs;
diff --git a/drivers/virtio/virtio_pci_common.h 
b/drivers/virtio/virtio_pci_common.h
index 392d990b7c73..65db92245e41 100644
--- a/drivers/virtio/virtio_pci_common.h
+++ b/drivers/virtio/virtio_pci_common.h
@@ -84,7 +84,9 @@ struct virtio_pci_device {
                                      void (*callback)(struct virtqueue *vq),
                                      const char *name,
                                      bool ctx,
-                                     u16 msix_vec);
+                                     u16 msix_vec,
+                                     struct virtqueue *vq,
+                                     u16 num);
        void (*del_vq)(struct virtio_pci_vq_info *info);
 
        u16 (*config_vector)(struct virtio_pci_device *vp_dev, u16 vector);
diff --git a/drivers/virtio/virtio_pci_legacy.c 
b/drivers/virtio/virtio_pci_legacy.c
index b3f8128b7983..9bc41b764624 100644
--- a/drivers/virtio/virtio_pci_legacy.c
+++ b/drivers/virtio/virtio_pci_legacy.c
@@ -113,9 +113,10 @@ static struct virtqueue *setup_vq(struct virtio_pci_device 
*vp_dev,
                                  void (*callback)(struct virtqueue *vq),
                                  const char *name,
                                  bool ctx,
-                                 u16 msix_vec)
+                                 u16 msix_vec,
+                                 struct virtqueue *vq,
+                                 u16 ring_num)
 {
-       struct virtqueue *vq;
        u16 num;
        int err;
        u64 q_pfn;
diff --git a/drivers/virtio/virtio_pci_modern.c 
b/drivers/virtio/virtio_pci_modern.c
index 5455bc041fb6..2ce58de549de 100644
--- a/drivers/virtio/virtio_pci_modern.c
+++ b/drivers/virtio/virtio_pci_modern.c
@@ -187,11 +187,12 @@ static struct virtqueue *setup_vq(struct 
virtio_pci_device *vp_dev,
                                  void (*callback)(struct virtqueue *vq),
                                  const char *name,
                                  bool ctx,
-                                 u16 msix_vec)
+                                 u16 msix_vec,
+                                 struct virtqueue *vq,
+                                 u16 ring_num)
 {
 
        struct virtio_pci_modern_device *mdev = &vp_dev->mdev;
-       struct virtqueue *vq;
        u16 num;
        int err;
 
@@ -203,6 +204,13 @@ static struct virtqueue *setup_vq(struct virtio_pci_device 
*vp_dev,
        if (!num || vp_modern_get_queue_enable(mdev, index))
                return ERR_PTR(-ENOENT);
 
+       if (ring_num) {
+               if (ring_num > num)
+                       return ERR_PTR(-ENOENT);
+
+               num = ring_num;
+       }
+
        if (num & (num - 1)) {
                dev_warn(&vp_dev->pci_dev->dev, "bad queue size %u", num);
                return ERR_PTR(-EINVAL);
@@ -211,10 +219,10 @@ static struct virtqueue *setup_vq(struct 
virtio_pci_device *vp_dev,
        info->msix_vector = msix_vec;
 
        /* create the vring */
-       vq = vring_create_virtqueue(index, num,
-                                   SMP_CACHE_BYTES, &vp_dev->vdev,
-                                   true, true, ctx,
-                                   vp_notify, callback, name);
+       vq = vring_setup_virtqueue(index, num,
+                                  SMP_CACHE_BYTES, &vp_dev->vdev,
+                                  true, true, ctx,
+                                  vp_notify, callback, name, vq);
        if (!vq)
                return ERR_PTR(-ENOMEM);
 
-- 
2.31.0

_______________________________________________
Virtualization mailing list
Virtualization@lists.linux-foundation.org
https://lists.linuxfoundation.org/mailman/listinfo/virtualization

Reply via email to