2 files changed, 9 insertions, 0 deletions

diff --git a/drivers/vfio/pci/vfio_pci.c b/drivers/vfio/pci/vfio_pci.c
index 9a86470f4ed9..0c94204e21f5 100644
--- a/drivers/vfio/pci/vfio_pci.c
+++ b/drivers/vfio/pci/vfio_pci.c
@@ -750,6 +750,12 @@ static long vfio_pci_ioctl(void *device_data,
                         if (ret)
                                 return ret;
 
+                        if (vdev->region[i].ops->add_capability) {
+                                ret = vdev->region[i].ops->add_capability(vdev,
+                                                &vdev->region[i], &caps);
+                                if (ret)
+                                        return ret;
+                        }
                 }
                 }
 
diff --git a/drivers/vfio/pci/vfio_pci_private.h b/drivers/vfio/pci/vfio_pci_private.h
index 86aab05d3d46..93c1738dae11 100644
--- a/drivers/vfio/pci/vfio_pci_private.h
+++ b/drivers/vfio/pci/vfio_pci_private.h
@@ -62,6 +62,9 @@ struct vfio_pci_regops {
         int     (*mmap)(struct vfio_pci_device *vdev,
                         struct vfio_pci_region *region,
                         struct vm_area_struct *vma);
+        int     (*add_capability)(struct vfio_pci_device *vdev,
+                                  struct vfio_pci_region *region,
+                                  struct vfio_info_cap *caps);
 };
 
 struct vfio_pci_region {