Implement IOMMU_INV_TYPE_TLB invalidations. When
nr_pages is null we interpret this as a context
invalidation.

Signed-off-by: Eric Auger <[email protected]>

---

The user API needs to be refined to discriminate context
invalidations from NH_VA invalidations. Also the leaf attribute
is not yet properly handled.
---
 drivers/iommu/arm-smmu-v3.c | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/drivers/iommu/arm-smmu-v3.c b/drivers/iommu/arm-smmu-v3.c
index 26afafa..e4c7ae8 100644
--- a/drivers/iommu/arm-smmu-v3.c
+++ b/drivers/iommu/arm-smmu-v3.c
@@ -2101,6 +2101,39 @@ static void arm_smmu_unbind_guest_stage(struct 
iommu_domain *domain,
        spin_unlock_irqrestore(&smmu_domain->devices_lock, flags);
 }
 
+static int arm_smmu_tlb_invalidate(struct iommu_domain *domain,
+                                  struct device *dev,
+                                  struct iommu_tlb_invalidate_info *inv_info)
+{
+       struct arm_smmu_domain *smmu_domain = to_smmu_domain(domain);
+       struct arm_smmu_device *smmu = smmu_domain->smmu;
+       size_t granule = 1 << (inv_info->size + 12);
+
+       if (smmu_domain->stage != ARM_SMMU_DOMAIN_NESTED)
+               return -EINVAL;
+
+       if (!smmu)
+               return -EINVAL;
+
+       if (inv_info->hdr.type != IOMMU_INV_TYPE_TLB)
+               return -EINVAL;
+
+       /*
+        * TODO: On context invalidation, the userspace sets nr_pages to 0.
+        * Refine the API to add a dedicated flags and also properly handle
+        * the leaf parameter.
+        */
+       if (!inv_info->nr_pages) {
+               arm_smmu_tlb_inv_context(smmu_domain);
+       } else {
+               arm_smmu_tlb_inv_range_nosync(inv_info->addr,
+                                             inv_info->nr_pages * granule,
+                                             granule, false, smmu_domain);
+               __arm_smmu_tlb_sync(smmu);
+       }
+       return 0;
+}
+
 static struct iommu_ops arm_smmu_ops = {
        .capable                = arm_smmu_capable,
        .domain_alloc           = arm_smmu_domain_alloc,
@@ -2122,6 +2155,7 @@ static struct iommu_ops arm_smmu_ops = {
        .put_resv_regions       = arm_smmu_put_resv_regions,
        .bind_guest_stage       = arm_smmu_bind_guest_stage,
        .unbind_guest_stage     = arm_smmu_unbind_guest_stage,
+       .tlb_invalidate         = arm_smmu_tlb_invalidate,
        .pgsize_bitmap          = -1UL, /* Restricted during device attach */
 };
 
-- 
2.5.5

_______________________________________________
iommu mailing list
[email protected]
https://lists.linuxfoundation.org/mailman/listinfo/iommu

Reply via email to