iommu/amd: Update irq_remapping_alloc to use IOMMU lookup helper function

To allow IOMMU rlookup using both PCI segment and device ID.

Co-developed-by: Vasant Hegde <vasant.hegde@amd.com>
Signed-off-by: Vasant Hegde <vasant.hegde@amd.com>
Signed-off-by: Suravee Suthikulpanit <suravee.suthikulpanit@amd.com>
Link: https://lore.kernel.org/r/20220706113825.25582-17-vasant.hegde@amd.com
Signed-off-by: Joerg Roedel <jroedel@suse.de>

diff --git a/drivers/iommu/amd/iommu.c b/drivers/iommu/amd/iommu.c
index 19db4d5..5e4648c 100644 (file)
--- a/drivers/iommu/amd/iommu.c
+++ b/drivers/iommu/amd/iommu.c
@@ -3246,8 +3246,9 @@ static int irq_remapping_alloc(struct irq_domain *domain, unsigned int virq,
        struct irq_alloc_info *info = arg;
        struct irq_data *irq_data;
        struct amd_ir_data *data = NULL;
+       struct amd_iommu *iommu;
        struct irq_cfg *cfg;
-       int i, ret, devid;
+       int i, ret, devid, seg, sbdf;
        int index;
 
        if (!info)
@@ -3263,8 +3264,14 @@ static int irq_remapping_alloc(struct irq_domain *domain, unsigned int virq,
        if (info->type == X86_IRQ_ALLOC_TYPE_PCI_MSI)
                info->flags &= ~X86_IRQ_ALLOC_CONTIGUOUS_VECTORS;
 
-       devid = get_devid(info);
-       if (devid < 0)
+       sbdf = get_devid(info);
+       if (sbdf < 0)
+               return -EINVAL;
+
+       seg = PCI_SBDF_TO_SEGID(sbdf);
+       devid = PCI_SBDF_TO_DEVID(sbdf);
+       iommu = __rlookup_amd_iommu(seg, devid);
+       if (!iommu)
                return -EINVAL;
 
        ret = irq_domain_alloc_irqs_parent(domain, virq, nr_irqs, arg);
@@ -3273,7 +3280,6 @@ 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;
-               struct amd_iommu *iommu;
 
                table = alloc_irq_table(devid, NULL);
                if (table) {
@@ -3283,7 +3289,6 @@ static int irq_remapping_alloc(struct irq_domain *domain, unsigned int virq,
                                 * interrupts.
                                 */
                                table->min_index = 32;
-                               iommu = amd_iommu_rlookup_table[devid];
                                for (i = 0; i < 32; ++i)
                                        iommu->irte_ops->set_allocated(table, i);
                        }