iommu/amd: Implement select() method on remapping irqdomain

Signed-off-by: David Woodhouse <dwmw@amazon.co.uk>

diff --git a/drivers/iommu/amd/iommu.c b/drivers/iommu/amd/iommu.c
index 13d0a8f42d56ffb38a914787ded3eafdbb92f8ec..7ecebc5d255f8011e1c031ea13b6d255aa18debf 100644 (file)
--- a/drivers/iommu/amd/iommu.c
+++ b/drivers/iommu/amd/iommu.c
@@ -3590,6 +3590,24 @@ struct irq_remap_ops amd_iommu_irq_ops = {
        .get_irq_domain         = get_irq_domain,
 };
 
+static int irq_remapping_select(struct irq_domain *d, struct irq_fwspec *fwspec,
+                               enum irq_domain_bus_token bus_token)
+{
+       struct amd_iommu *iommu;
+       int devid = -1;
+
+       if (x86_fwspec_is_ioapic(fwspec))
+               devid = get_ioapic_devid(fwspec->param[0]);
+       else if (x86_fwspec_is_ioapic(fwspec))
+               devid = get_hpet_devid(fwspec->param[0]);
+
+       if (devid < 0)
+               return 0;
+
+       iommu = amd_iommu_rlookup_table[devid];
+       return iommu && iommu->ir_domain == d;
+}
+
 static void irq_remapping_prepare_irte(struct amd_ir_data *data,
                                       struct irq_cfg *irq_cfg,
                                       struct irq_alloc_info *info,
@@ -3813,6 +3831,7 @@ static void irq_remapping_deactivate(struct irq_domain *domain,
 }
 
 static const struct irq_domain_ops amd_ir_domain_ops = {
+       .select = irq_remapping_select,
        .alloc = irq_remapping_alloc,
        .free = irq_remapping_free,
        .activate = irq_remapping_activate,