From: Suravee Suthikulpanit <suravee.suthikulpa...@amd.com>

Pass amd_iommu structure as one of the parameter to these functions
as its needed to retrieve variable tables inside these functions.

Co-developed-by: Vasant Hegde <vasant.he...@amd.com>
Signed-off-by: Vasant Hegde <vasant.he...@amd.com>
Signed-off-by: Suravee Suthikulpanit <suravee.suthikulpa...@amd.com>
---
 drivers/iommu/amd/iommu.c | 26 +++++++++-----------------
 1 file changed, 9 insertions(+), 17 deletions(-)

diff --git a/drivers/iommu/amd/iommu.c b/drivers/iommu/amd/iommu.c
index 33b69843b860..814b5abe676a 100644
--- a/drivers/iommu/amd/iommu.c
+++ b/drivers/iommu/amd/iommu.c
@@ -2809,21 +2809,17 @@ static int set_remap_table_entry_alias(struct pci_dev 
*pdev, u16 alias,
        return 0;
 }
 
-static struct irq_remap_table *alloc_irq_table(u16 devid, struct pci_dev *pdev)
+static struct irq_remap_table *alloc_irq_table(struct amd_iommu *iommu,
+                                              u16 devid, struct pci_dev *pdev)
 {
        struct irq_remap_table *table = NULL;
        struct irq_remap_table *new_table = NULL;
        struct amd_iommu_pci_seg *pci_seg;
-       struct amd_iommu *iommu;
        unsigned long flags;
        u16 alias;
 
        spin_lock_irqsave(&iommu_table_lock, flags);
 
-       iommu = amd_iommu_rlookup_table[devid];
-       if (!iommu)
-               goto out_unlock;
-
        pci_seg = iommu->pci_seg;
        table = pci_seg->irq_lookup_table[devid];
        if (table)
@@ -2879,18 +2875,14 @@ static struct irq_remap_table *alloc_irq_table(u16 
devid, struct pci_dev *pdev)
        return table;
 }
 
-static int alloc_irq_index(u16 devid, int count, bool align,
-                          struct pci_dev *pdev)
+static int alloc_irq_index(struct amd_iommu *iommu, u16 devid, int count,
+                          bool align, struct pci_dev *pdev)
 {
        struct irq_remap_table *table;
        int index, c, alignment = 1;
        unsigned long flags;
-       struct amd_iommu *iommu = amd_iommu_rlookup_table[devid];
-
-       if (!iommu)
-               return -ENODEV;
 
-       table = alloc_irq_table(devid, pdev);
+       table = alloc_irq_table(iommu, devid, pdev);
        if (!table)
                return -ENODEV;
 
@@ -3262,7 +3254,7 @@ static int irq_remapping_alloc(struct irq_domain *domain, 
unsigned int virq,
        if (info->type == X86_IRQ_ALLOC_TYPE_IOAPIC) {
                struct irq_remap_table *table;
 
-               table = alloc_irq_table(devid, NULL);
+               table = alloc_irq_table(iommu, devid, NULL);
                if (table) {
                        if (!table->min_index) {
                                /*
@@ -3282,10 +3274,10 @@ static int irq_remapping_alloc(struct irq_domain 
*domain, unsigned int virq,
                   info->type == X86_IRQ_ALLOC_TYPE_PCI_MSIX) {
                bool align = (info->type == X86_IRQ_ALLOC_TYPE_PCI_MSI);
 
-               index = alloc_irq_index(devid, nr_irqs, align,
+               index = alloc_irq_index(iommu, devid, nr_irqs, align,
                                        msi_desc_to_pci_dev(info->desc));
        } else {
-               index = alloc_irq_index(devid, nr_irqs, false, NULL);
+               index = alloc_irq_index(iommu, devid, nr_irqs, false, NULL);
        }
 
        if (index < 0) {
@@ -3411,8 +3403,8 @@ static int irq_remapping_select(struct irq_domain *d, 
struct irq_fwspec *fwspec,
 
        if (devid < 0)
                return 0;
+       iommu = __rlookup_amd_iommu((devid >> 16), (devid & 0xffff));
 
-       iommu = amd_iommu_rlookup_table[devid];
        return iommu && iommu->ir_domain == d;
 }
 
-- 
2.27.0

_______________________________________________
iommu mailing list
iommu@lists.linux-foundation.org
https://lists.linuxfoundation.org/mailman/listinfo/iommu

Reply via email to