This will replace global irq lookup table (irq_lookup_table).

Co-developed-by: Suravee Suthikulpanit <suravee.suthikulpa...@amd.com>
Signed-off-by: Suravee Suthikulpanit <suravee.suthikulpa...@amd.com>
Signed-off-by: Vasant Hegde <vasant.he...@amd.com>
---
 drivers/iommu/amd/amd_iommu_types.h |  6 ++++++
 drivers/iommu/amd/init.c            | 27 +++++++++++++++++++++++++++
 2 files changed, 33 insertions(+)

diff --git a/drivers/iommu/amd/amd_iommu_types.h 
b/drivers/iommu/amd/amd_iommu_types.h
index 9c008662be1b..d507c96598a7 100644
--- a/drivers/iommu/amd/amd_iommu_types.h
+++ b/drivers/iommu/amd/amd_iommu_types.h
@@ -557,6 +557,12 @@ struct amd_iommu_pci_seg {
         * device id.
         */
        struct amd_iommu **rlookup_table;
+
+       /*
+        * This table is used to find the irq remapping table for a given
+        * device id quickly.
+        */
+       struct irq_remap_table **irq_lookup_table;
 };
 
 /*
diff --git a/drivers/iommu/amd/init.c b/drivers/iommu/amd/init.c
index a2efc02ba80a..48db6c3164aa 100644
--- a/drivers/iommu/amd/init.c
+++ b/drivers/iommu/amd/init.c
@@ -682,6 +682,26 @@ static inline void free_rlookup_table(struct 
amd_iommu_pci_seg *pci_seg)
        pci_seg->rlookup_table = NULL;
 }
 
+static inline int __init alloc_irq_lookup_table(struct amd_iommu_pci_seg 
*pci_seg)
+{
+       pci_seg->irq_lookup_table = (void *)__get_free_pages(
+                                            GFP_KERNEL | __GFP_ZERO,
+                                            get_order(rlookup_table_size));
+       kmemleak_alloc(pci_seg->irq_lookup_table,
+                      rlookup_table_size, 1, GFP_KERNEL);
+       if (pci_seg->irq_lookup_table == NULL)
+               return -ENOMEM;
+
+       return 0;
+}
+
+static inline void free_irq_lookup_table(struct amd_iommu_pci_seg *pci_seg)
+{
+       kmemleak_free(pci_seg->irq_lookup_table);
+       free_pages((unsigned long)pci_seg->irq_lookup_table,
+                  get_order(rlookup_table_size));
+       pci_seg->irq_lookup_table = NULL;
+}
 
 /*
  * Allocates the command buffer. This buffer is per AMD IOMMU. We can
@@ -1533,6 +1553,7 @@ static void __init free_pci_segment(void)
 
        for_each_pci_segment_safe(pci_seg, next) {
                list_del(&pci_seg->list);
+               free_irq_lookup_table(pci_seg);
                free_rlookup_table(pci_seg);
                free_dev_table(pci_seg);
                kfree(pci_seg);
@@ -2896,6 +2917,7 @@ static int __init early_amd_iommu_init(void)
                amd_iommu_irq_remap = check_ioapic_information();
 
        if (amd_iommu_irq_remap) {
+               struct amd_iommu_pci_seg *pci_seg;
                /*
                 * Interrupt remapping enabled, create kmem_cache for the
                 * remapping tables.
@@ -2912,6 +2934,11 @@ static int __init early_amd_iommu_init(void)
                if (!amd_iommu_irq_cache)
                        goto out;
 
+               for_each_pci_segment(pci_seg) {
+                       if (alloc_irq_lookup_table(pci_seg))
+                               goto out;
+               }
+
                irq_lookup_table = (void *)__get_free_pages(
                                GFP_KERNEL | __GFP_ZERO,
                                get_order(rlookup_table_size));
-- 
2.27.0

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

Reply via email to