X-Git-Url: http://pileus.org/git/?a=blobdiff_plain;f=virt%2Fkvm%2Fiommu.c;h=967aba133a624c38dee5b61cbb9b1376b767f228;hb=f793f2961170c0b49c1650e69e7825484159ce62;hp=78c80f67f535f97055d312aa946c109bcfdcd424;hpb=259a53407d9ae0c109902669c6f2f88a389dd163;p=~andy%2Flinux

diff --git a/virt/kvm/iommu.c b/virt/kvm/iommu.c
index 78c80f67f53..967aba133a6 100644
--- a/virt/kvm/iommu.c
+++ b/virt/kvm/iommu.c
@@ -187,6 +187,8 @@ int kvm_assign_device(struct kvm *kvm,
 			goto out_unmap;
 	}
 
+	pdev->dev_flags |= PCI_DEV_FLAGS_ASSIGNED;
+
 	printk(KERN_DEBUG "assign device %x:%x:%x.%x\n",
 		assigned_dev->host_segnr,
 		assigned_dev->host_busnr,
@@ -215,6 +217,8 @@ int kvm_deassign_device(struct kvm *kvm,
 
 	iommu_detach_device(domain, &pdev->dev);
 
+	pdev->dev_flags &= ~PCI_DEV_FLAGS_ASSIGNED;
+
 	printk(KERN_DEBUG "deassign device %x:%x:%x.%x\n",
 		assigned_dev->host_segnr,
 		assigned_dev->host_busnr,