From: Suravee Suthikulpanit <suravee.suthikulpa...@amd.com>
Current get_device_id() only provide 16-bit PCI device ID (i.e. BDF).
With multiple PCI segment support, we need to extend the helper function
to include PCI segment ID.
So, introduce a new helper function get_device_sbdf_id() to replace
the current get_pci_device_id().
Signed-off-by: Suravee Suthikulpanit <suravee.suthikulpa...@amd.com>
Signed-off-by: Vasant Hegde <vasant.he...@amd.com>
---
drivers/iommu/amd/amd_iommu.h | 7 ++++++
drivers/iommu/amd/iommu.c | 40 +++++++++++++++++------------------
2 files changed, 27 insertions(+), 20 deletions(-)
diff --git a/drivers/iommu/amd/amd_iommu.h b/drivers/iommu/amd/amd_iommu.h
index 64c954e168d7..4dad1b442409 100644
--- a/drivers/iommu/amd/amd_iommu.h
+++ b/drivers/iommu/amd/amd_iommu.h
@@ -115,6 +115,13 @@ void amd_iommu_domain_clr_pt_root(struct protection_domain
*domain)
amd_iommu_domain_set_pt_root(domain, 0);
}
+static inline int get_pci_sbdf_id(struct pci_dev *pdev)
+{
+ int seg = pci_domain_nr(pdev->bus);
+ u16 devid = pci_dev_id(pdev);
+
+ return ((seg << 16) | (devid & 0xffff));
+}
extern bool translation_pre_enabled(struct amd_iommu *iommu);
extern bool amd_iommu_is_attach_deferred(struct device *dev);
diff --git a/drivers/iommu/amd/iommu.c b/drivers/iommu/amd/iommu.c
index 60fbb1abb15f..4fe77f77dfa1 100644
--- a/drivers/iommu/amd/iommu.c
+++ b/drivers/iommu/amd/iommu.c
@@ -92,13 +92,6 @@ static void detach_device(struct device *dev);
*
****************************************************************************/
-static inline u16 get_pci_device_id(struct device *dev)
-{
- struct pci_dev *pdev = to_pci_dev(dev);
-
- return pci_dev_id(pdev);
-}
-
static inline int get_acpihid_device_id(struct device *dev,
struct acpihid_map_entry **entry)
{
@@ -119,16 +112,16 @@ static inline int get_acpihid_device_id(struct device
*dev,
return -EINVAL;
}
-static inline int get_device_id(struct device *dev)
+static inline int get_device_sbdf_id(struct device *dev)
{
- int devid;
+ int sbdf;
if (dev_is_pci(dev))
- devid = get_pci_device_id(dev);
+ sbdf = get_pci_sbdf_id(to_pci_dev(dev));
else
- devid = get_acpihid_device_id(dev, NULL);
+ sbdf = get_acpihid_device_id(dev, NULL);
- return devid;
+ return sbdf;
}
struct dev_table_entry *get_dev_table(struct amd_iommu *iommu)
@@ -182,9 +175,11 @@ static struct amd_iommu *__rlookup_amd_iommu(u16 seg, u16
devid)
static struct amd_iommu *rlookup_amd_iommu(struct device *dev)
{
u16 seg = get_device_segment(dev);
- u16 devid = get_device_id(dev);
+ int devid = get_device_sbdf_id(dev);
- return __rlookup_amd_iommu(seg, devid);
+ if (devid < 0)
+ return NULL;
+ return __rlookup_amd_iommu(seg, (devid & 0xffff));
}
static struct protection_domain *to_pdomain(struct iommu_domain *dom)
@@ -364,9 +359,10 @@ static bool check_device(struct device *dev)
if (!dev)
return false;
- devid = get_device_id(dev);
+ devid = get_device_sbdf_id(dev);
if (devid < 0)
return false;
+ devid &= 0xffff;
iommu = rlookup_amd_iommu(dev);
if (!iommu)
@@ -374,7 +370,7 @@ static bool check_device(struct device *dev)
/* Out of our scope? */
pci_seg = iommu->pci_seg;
- if ((devid & 0xffff) > pci_seg->last_bdf)
+ if (devid > pci_seg->last_bdf)
return false;
return true;
@@ -388,10 +384,11 @@ static int iommu_init_device(struct amd_iommu *iommu,
struct device *dev)
if (dev_iommu_priv_get(dev))
return 0;
- devid = get_device_id(dev);
+ devid = get_device_sbdf_id(dev);
if (devid < 0)
return devid;
+ devid &= 0xffff;
dev_data = find_dev_data(iommu, devid);
if (!dev_data)
return -ENOMEM;
@@ -421,10 +418,11 @@ static void iommu_ignore_device(struct amd_iommu *iommu,
struct device *dev)
struct dev_table_entry *dev_table = get_dev_table(iommu);
int devid;
- devid = (get_device_id(dev)) & 0xffff;
+ devid = get_device_sbdf_id(dev);
if (devid < 0)
return;
+ devid &= 0xffff;
pci_seg->rlookup_table[devid] = NULL;
memset(&dev_table[devid], 0, sizeof(struct dev_table_entry));
@@ -2262,9 +2260,11 @@ static void amd_iommu_get_resv_regions(struct device
*dev,
struct amd_iommu_pci_seg *pci_seg;
int devid;
- devid = get_device_id(dev);
+ devid = get_device_sbdf_id(dev);
if (devid < 0)
return;
+ devid &= 0xffff;
+
iommu = rlookup_amd_iommu(dev);
if (!iommu)
return;
@@ -3151,7 +3151,7 @@ static int get_devid(struct irq_alloc_info *info)
return get_hpet_devid(info->devid);
case X86_IRQ_ALLOC_TYPE_PCI_MSI:
case X86_IRQ_ALLOC_TYPE_PCI_MSIX:
- return get_device_id(msi_desc_to_dev(info->desc));
+ return get_device_sbdf_id(msi_desc_to_dev(info->desc));
default:
WARN_ON_ONCE(1);
return -1;
--
2.27.0
_______________________________________________
iommu mailing list
iommu@lists.linux-foundation.org
https://lists.linuxfoundation.org/mailman/listinfo/iommu
