1 files changed, 4 insertions, 0 deletions

diff --git a/drivers/vfio/pci/vfio_pci.c b/drivers/vfio/pci/vfio_pci.c
index b423a309a6e0..125b58eff936 100644
--- a/drivers/vfio/pci/vfio_pci.c
+++ b/drivers/vfio/pci/vfio_pci.c
@@ -28,6 +28,7 @@
 #include <linux/uaccess.h>
 #include <linux/vfio.h>
 #include <linux/vgaarb.h>
+#include <linux/nospec.h>
 
 #include "vfio_pci_private.h"
 
@@ -727,6 +728,9 @@ static long vfio_pci_ioctl(void *device_data,
                         if (info.index >=
                             VFIO_PCI_NUM_REGIONS + vdev->num_regions)
                                 return -EINVAL;
+                        info.index = array_index_nospec(info.index,
+                                                        VFIO_PCI_NUM_REGIONS +
+                                                        vdev->num_regions);
 
                         i = info.index - VFIO_PCI_NUM_REGIONS;