[PATCH] vp_vdpa: fix id_table array not null terminated error

[Xiaoguang Wang]

Allocate one extra virtio_device_id as null terminator, otherwise
vdpa_mgmtdev_get_classes() may iterate multiple times and visit
undefined memory.

Fixes: ffbda8e9df10 ("vdpa/vp_vdpa : add vdpa tool support in vp_vdpa")
Cc: sta...@vger.kernel.org
Signed-off-by: Xiaoguang Wang <lege.w...@jaguarmicro.com>
---
 drivers/vdpa/virtio_pci/vp_vdpa.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/drivers/vdpa/virtio_pci/vp_vdpa.c 
b/drivers/vdpa/virtio_pci/vp_vdpa.c
index ac4ab22f7d8b..74cc4ed77cc4 100644
--- a/drivers/vdpa/virtio_pci/vp_vdpa.c
+++ b/drivers/vdpa/virtio_pci/vp_vdpa.c
@@ -612,7 +612,11 @@ static int vp_vdpa_probe(struct pci_dev *pdev, const 
struct pci_device_id *id)
                goto mdev_err;
        }
 
-       mdev_id = kzalloc(sizeof(struct virtio_device_id), GFP_KERNEL);
+       /*
+        * id_table should be a null terminated array.
+        * See vdpa_mgmtdev_get_classes().
+        */
+       mdev_id = kzalloc(sizeof(struct virtio_device_id) * 2, GFP_KERNEL);
        if (!mdev_id) {
                err = -ENOMEM;
                goto mdev_id_err;
-- 
2.40.1