iommu/vt-d: Pass device_domain_info to __dmar_remove_one_dev_info

This struct contains all necessary information for the
function already. Also handle the info->dev == NULL case
while at it.

Signed-off-by: Joerg Roedel <jroedel@suse.de>

diff --git a/drivers/iommu/intel-iommu.c b/drivers/iommu/intel-iommu.c
index d55ef9df0879e6c19f7da5bc0425beaaf5393b4d..18e44421a862f99b64ef0568529b233cdded056a 100644 (file)
--- a/drivers/iommu/intel-iommu.c
+++ b/drivers/iommu/intel-iommu.c
@@ -473,10 +473,9 @@ static void domain_exit(struct dmar_domain *domain);
 static void domain_remove_dev_info(struct dmar_domain *domain);
 static void dmar_remove_one_dev_info(struct dmar_domain *domain,
                                     struct device *dev);
+static void __dmar_remove_one_dev_info(struct device_domain_info *info);
 static void domain_context_clear(struct intel_iommu *iommu,
                                 struct device *dev);
-static void __dmar_remove_one_dev_info(struct dmar_domain *domain,
-                                      struct device *dev);
 static int domain_detach_iommu(struct dmar_domain *domain,
                               struct intel_iommu *iommu);
 
@@ -2216,7 +2215,7 @@ static void domain_remove_dev_info(struct dmar_domain *domain)
 
        spin_lock_irqsave(&device_domain_lock, flags);
        list_for_each_entry_safe(info, tmp, &domain->devices, link)
-               __dmar_remove_one_dev_info(domain, info->dev);
+               __dmar_remove_one_dev_info(info);
        spin_unlock_irqrestore(&device_domain_lock, flags);
 }
 
@@ -4538,43 +4537,41 @@ static void domain_context_clear(struct intel_iommu *iommu, struct device *dev)
        pci_for_each_dma_alias(to_pci_dev(dev), &domain_context_clear_one_cb, iommu);
 }
 
-static void __dmar_remove_one_dev_info(struct dmar_domain *domain,
-                                      struct device *dev)
+static void __dmar_remove_one_dev_info(struct device_domain_info *info)
 {
-       struct device_domain_info *info;
        struct intel_iommu *iommu;
        unsigned long flags;
-       u8 bus, devfn;
 
        assert_spin_locked(&device_domain_lock);
 
-       iommu = device_to_iommu(dev, &bus, &devfn);
-       if (!iommu)
+       if (WARN_ON(!info))
                return;
 
-       info = dev->archdata.iommu;
+       iommu = info->iommu;
 
-       if (WARN_ON(!info))
-               return;
+       if (info->dev) {
+               iommu_disable_dev_iotlb(info);
+               domain_context_clear(iommu, info->dev);
+       }
 
        unlink_domain_info(info);
 
-       iommu_disable_dev_iotlb(info);
-       domain_context_clear(iommu, dev);
-       free_devinfo_mem(info);
-
        spin_lock_irqsave(&iommu->lock, flags);
-       domain_detach_iommu(domain, iommu);
+       domain_detach_iommu(info->domain, iommu);
        spin_unlock_irqrestore(&iommu->lock, flags);
+
+       free_devinfo_mem(info);
 }
 
 static void dmar_remove_one_dev_info(struct dmar_domain *domain,
                                     struct device *dev)
 {
+       struct device_domain_info *info;
        unsigned long flags;
 
        spin_lock_irqsave(&device_domain_lock, flags);
-       __dmar_remove_one_dev_info(domain, dev);
+       info = dev->archdata.iommu;
+       __dmar_remove_one_dev_info(info);
        spin_unlock_irqrestore(&device_domain_lock, flags);
 }