[PATCH] vfio: Set PCI_DEV_FLAGS_ASSIGNED when assigning device to guest

[Bandan Das <bsd@redhat.com>]

Some drivers such as ixgbe rely on pci_vfs_assigned() to prevent
disabling sr-iov when vfs are still assigned during hotplug
event or module removal. Set and unset PCI_DEV_FLAGS_ASSIGNED
appropriately

Signed-off-by: Bandan Das <bsd@redhat.com>
---
 drivers/vfio/pci/vfio_pci.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/drivers/vfio/pci/vfio_pci.c b/drivers/vfio/pci/vfio_pci.c
index 7ba0424..7cc7ed6 100644
--- a/drivers/vfio/pci/vfio_pci.c
+++ b/drivers/vfio/pci/vfio_pci.c
@@ -90,6 +90,7 @@ static int vfio_pci_enable(struct vfio_pci_device *vdev)
 		vdev->has_vga = true;
 #endif
 
+	pdev->dev_flags |= PCI_DEV_FLAGS_ASSIGNED;
 	return 0;
 }
 
@@ -149,6 +150,7 @@ static void vfio_pci_disable(struct vfio_pci_device *vdev)
 				__func__, dev_name(&pdev->dev), ret);
 	}
 
+	pdev->dev_flags &= ~PCI_DEV_FLAGS_ASSIGNED;
 	pci_restore_state(pdev);
 }
 
-- 
1.8.3.1