Replace it with per PCI segment last_bdf variable.

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 |  3 ---
 drivers/iommu/amd/init.c            | 35 ++++++++++-------------------
 drivers/iommu/amd/iommu.c           | 10 ++++++---
 3 files changed, 19 insertions(+), 29 deletions(-)

diff --git a/drivers/iommu/amd/amd_iommu_types.h 
b/drivers/iommu/amd/amd_iommu_types.h
index 0aa170014b85..1109961e1042 100644
--- a/drivers/iommu/amd/amd_iommu_types.h
+++ b/drivers/iommu/amd/amd_iommu_types.h
@@ -829,9 +829,6 @@ struct unity_map_entry {
 /* size of the dma_ops aperture as power of 2 */
 extern unsigned amd_iommu_aperture_order;
 
-/* largest PCI device id we expect translation requests for */
-extern u16 amd_iommu_last_bdf;
-
 /* allocation bitmap for domain ids */
 extern unsigned long *amd_iommu_pd_alloc_bitmap;
 
diff --git a/drivers/iommu/amd/init.c b/drivers/iommu/amd/init.c
index b3905b1c4bc9..093304d16c85 100644
--- a/drivers/iommu/amd/init.c
+++ b/drivers/iommu/amd/init.c
@@ -161,9 +161,6 @@ static bool amd_iommu_disabled __initdata;
 static bool amd_iommu_force_enable __initdata;
 static int amd_iommu_target_ivhd_type;
 
-u16 amd_iommu_last_bdf;                        /* largest PCI device id we have
-                                          to handle */
-
 LIST_HEAD(amd_iommu_pci_seg_list);     /* list of all PCI segments */
 LIST_HEAD(amd_iommu_list);             /* list of all AMD IOMMUs in the
                                           system */
@@ -245,16 +242,10 @@ static void init_translation_status(struct amd_iommu 
*iommu)
                iommu->flags |= AMD_IOMMU_FLAG_TRANS_PRE_ENABLED;
 }
 
-static inline void update_last_devid(u16 devid)
-{
-       if (devid > amd_iommu_last_bdf)
-               amd_iommu_last_bdf = devid;
-}
-
-static inline unsigned long tbl_size(int entry_size)
+static inline unsigned long tbl_size(int entry_size, int last_bdf)
 {
        unsigned shift = PAGE_SHIFT +
-                        get_order(((int)amd_iommu_last_bdf + 1) * entry_size);
+                        get_order((last_bdf + 1) * entry_size);
 
        return 1UL << shift;
 }
@@ -538,7 +529,6 @@ static int __init find_last_devid_from_ivhd(struct 
ivhd_header *h)
                switch (dev->type) {
                case IVHD_DEV_ALL:
                        /* Use maximum BDF value for DEV_ALL */
-                       update_last_devid(0xffff);
                        return 0xffff;
                        break;
                case IVHD_DEV_SELECT:
@@ -546,7 +536,6 @@ static int __init find_last_devid_from_ivhd(struct 
ivhd_header *h)
                case IVHD_DEV_ALIAS:
                case IVHD_DEV_EXT_SELECT:
                        /* all the above subfield types refer to device ids */
-                       update_last_devid(dev->devid);
                        if (dev->devid > last_devid)
                                last_devid = dev->devid;
                        break;
@@ -688,7 +677,7 @@ static int __init alloc_alias_table(struct 
amd_iommu_pci_seg *pci_seg)
        /*
         * let all alias entries point to itself
         */
-       for (i = 0; i <= amd_iommu_last_bdf; ++i)
+       for (i = 0; i <= pci_seg->last_bdf; ++i)
                pci_seg->alias_table[i] = i;
 
        return 0;
@@ -1054,7 +1043,7 @@ static bool __copy_device_table(struct amd_iommu *iommu)
                return false;
        }
 
-       for (devid = 0; devid <= amd_iommu_last_bdf; ++devid) {
+       for (devid = 0; devid <= pci_seg->last_bdf; ++devid) {
                pci_seg->old_dev_tbl_cpy[devid] = old_devtb[devid];
                dom_id = old_devtb[devid].data[1] & DEV_DOMID_MASK;
                dte_v = old_devtb[devid].data[0] & DTE_FLAG_V;
@@ -1315,7 +1304,7 @@ static int __init init_iommu_from_acpi(struct amd_iommu 
*iommu,
 
                        DUMP_printk("  DEV_ALL\t\t\tflags: %02x\n", e->flags);
 
-                       for (dev_i = 0; dev_i <= amd_iommu_last_bdf; ++dev_i)
+                       for (dev_i = 0; dev_i <= pci_seg->last_bdf; ++dev_i)
                                set_dev_entry_from_acpi(iommu, dev_i, e->flags, 
0);
                        break;
                case IVHD_DEV_SELECT:
@@ -1560,9 +1549,9 @@ static struct amd_iommu_pci_seg *__init 
alloc_pci_segment(u16 id,
 
        pci_seg->last_bdf = last_bdf;
        DUMP_printk("PCI segment : 0x%0x, last bdf : 0x%04x\n", id, last_bdf);
-       pci_seg->dev_table_size     = tbl_size(DEV_TABLE_ENTRY_SIZE);
-       pci_seg->alias_table_size   = tbl_size(ALIAS_TABLE_ENTRY_SIZE);
-       pci_seg->rlookup_table_size = tbl_size(RLOOKUP_TABLE_ENTRY_SIZE);
+       pci_seg->dev_table_size     = tbl_size(DEV_TABLE_ENTRY_SIZE, last_bdf);
+       pci_seg->alias_table_size   = tbl_size(ALIAS_TABLE_ENTRY_SIZE, 
last_bdf);
+       pci_seg->rlookup_table_size = tbl_size(RLOOKUP_TABLE_ENTRY_SIZE, 
last_bdf);
 
        pci_seg->id = id;
        init_llist_head(&pci_seg->dev_data_list);
@@ -2443,7 +2432,7 @@ static int __init init_unity_map_range(struct ivmd_header 
*m,
        case ACPI_IVMD_TYPE_ALL:
                s = "IVMD_TYPE_ALL\t\t";
                e->devid_start = 0;
-               e->devid_end = amd_iommu_last_bdf;
+               e->devid_end = pci_seg->last_bdf;
                break;
        case ACPI_IVMD_TYPE_RANGE:
                s = "IVMD_TYPE_RANGE\t\t";
@@ -2510,7 +2499,7 @@ static void init_device_table_dma(struct 
amd_iommu_pci_seg *pci_seg)
        if (dev_table == NULL)
                return;
 
-       for (devid = 0; devid <= amd_iommu_last_bdf; ++devid) {
+       for (devid = 0; devid <= pci_seg->last_bdf; ++devid) {
                __set_dev_entry_bit(dev_table, devid, DEV_ENTRY_VALID);
                __set_dev_entry_bit(dev_table, devid, DEV_ENTRY_TRANSLATION);
        }
@@ -2524,7 +2513,7 @@ static void __init uninit_device_table_dma(struct 
amd_iommu_pci_seg *pci_seg)
        if (dev_table == NULL)
                return;
 
-       for (devid = 0; devid <= amd_iommu_last_bdf; ++devid) {
+       for (devid = 0; devid <= pci_seg->last_bdf; ++devid) {
                dev_table[devid].data[0] = 0ULL;
                dev_table[devid].data[1] = 0ULL;
        }
@@ -2539,7 +2528,7 @@ static void init_device_table(void)
                return;
 
        for_each_pci_segment(pci_seg) {
-               for (devid = 0; devid <= amd_iommu_last_bdf; ++devid)
+               for (devid = 0; devid <= pci_seg->last_bdf; ++devid)
                        __set_dev_entry_bit(pci_seg->dev_table,
                                            devid, DEV_ENTRY_IRQ_TBL_EN);
        }
diff --git a/drivers/iommu/amd/iommu.c b/drivers/iommu/amd/iommu.c
index 476217f2890d..5976038d48a3 100644
--- a/drivers/iommu/amd/iommu.c
+++ b/drivers/iommu/amd/iommu.c
@@ -358,6 +358,8 @@ static bool pci_iommuv2_capable(struct pci_dev *pdev)
  */
 static bool check_device(struct device *dev)
 {
+       struct amd_iommu_pci_seg *pci_seg;
+       struct amd_iommu *iommu;
        int devid;
 
        if (!dev)
@@ -367,11 +369,13 @@ static bool check_device(struct device *dev)
        if (devid < 0)
                return false;
 
-       /* Out of our scope? */
-       if (devid > amd_iommu_last_bdf)
+       iommu = rlookup_amd_iommu(dev);
+       if (!iommu)
                return false;
 
-       if (rlookup_amd_iommu(dev) == NULL)
+       /* Out of our scope? */
+       pci_seg = iommu->pci_seg;
+       if ((devid & 0xffff) > pci_seg->last_bdf)
                return false;
 
        return true;
-- 
2.27.0

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

Reply via email to