The invalidate_range() notifier is called for any change to the address
space. Perform the required ATC invalidations.

Signed-off-by: Jean-Philippe Brucker <[email protected]>
---
 drivers/iommu/arm-smmu-v3.c | 36 ++++++++++++++++++++++++++++++++----
 1 file changed, 32 insertions(+), 4 deletions(-)

diff --git a/drivers/iommu/arm-smmu-v3.c b/drivers/iommu/arm-smmu-v3.c
index af551f3c78a78..972c061399fc9 100644
--- a/drivers/iommu/arm-smmu-v3.c
+++ b/drivers/iommu/arm-smmu-v3.c
@@ -2378,6 +2378,20 @@ arm_smmu_atc_inv_to_cmd(int ssid, unsigned long iova, 
size_t size,
        size_t inval_grain_shift = 12;
        unsigned long page_start, page_end;
 
+       /*
+        * ATS and PASID:
+        *
+        * If substream_valid is clear, the PCIe TLP is sent without a PASID
+        * prefix. In that case all ATC entries within the address range are
+        * invalidated, including those that were requested with a PASID! There
+        * is no way to invalidate only entries without PASID.
+        *
+        * When using STRTAB_STE_1_S1DSS_SSID0 (reserving CD 0 for non-PASID
+        * traffic), translation requests without PASID create ATC entries
+        * without PASID, which must be invalidated with substream_valid clear.
+        * This has the unpleasant side-effect of invalidating all PASID-tagged
+        * ATC entries within the address range.
+        */
        *cmd = (struct arm_smmu_cmdq_ent) {
                .opcode                 = CMDQ_OP_ATC_INV,
                .substream_valid        = !!ssid,
@@ -2421,12 +2435,12 @@ arm_smmu_atc_inv_to_cmd(int ssid, unsigned long iova, 
size_t size,
        cmd->atc.size   = log2_span;
 }
 
-static int arm_smmu_atc_inv_master(struct arm_smmu_master *master)
+static int arm_smmu_atc_inv_master(struct arm_smmu_master *master, int ssid)
 {
        int i;
        struct arm_smmu_cmdq_ent cmd;
 
-       arm_smmu_atc_inv_to_cmd(0, 0, 0, &cmd);
+       arm_smmu_atc_inv_to_cmd(ssid, 0, 0, &cmd);
 
        for (i = 0; i < master->num_sids; i++) {
                cmd.atc.sid = master->sids[i];
@@ -2955,7 +2969,7 @@ static void arm_smmu_disable_ats(struct arm_smmu_master 
*master)
         * ATC invalidation via the SMMU.
         */
        wmb();
-       arm_smmu_atc_inv_master(master);
+       arm_smmu_atc_inv_master(master, 0);
        atomic_dec(&smmu_domain->nr_ats_masters);
 }
 
@@ -3152,6 +3166,16 @@ arm_smmu_iova_to_phys(struct iommu_domain *domain, 
dma_addr_t iova)
        return ops->iova_to_phys(ops, iova);
 }
 
+static void arm_smmu_mm_invalidate_range(struct mmu_notifier *mn,
+                                        struct mm_struct *mm,
+                                        unsigned long start, unsigned long end)
+{
+       struct arm_smmu_mmu_notifier *smmu_mn = mn_to_smmu(mn);
+
+       arm_smmu_atc_inv_domain(smmu_mn->domain, mm->pasid, start,
+                               end - start + 1);
+}
+
 static void arm_smmu_mm_release(struct mmu_notifier *mn, struct mm_struct *mm)
 {
        struct arm_smmu_mmu_notifier *smmu_mn = mn_to_smmu(mn);
@@ -3170,6 +3194,7 @@ static void arm_smmu_mm_release(struct mmu_notifier *mn, 
struct mm_struct *mm)
        arm_smmu_write_ctx_desc(smmu_domain, mm->pasid, &invalid_cd);
 
        arm_smmu_tlb_inv_asid(smmu_domain->smmu, smmu_mn->cd->asid);
+       arm_smmu_atc_inv_domain(smmu_domain, mm->pasid, 0, 0);
 
        smmu_mn->cleared = true;
        mutex_unlock(&sva_lock);
@@ -3181,6 +3206,7 @@ static void arm_smmu_mmu_notifier_free(struct 
mmu_notifier *mn)
 }
 
 static struct mmu_notifier_ops arm_smmu_mmu_notifier_ops = {
+       .invalidate_range       = arm_smmu_mm_invalidate_range,
        .release                = arm_smmu_mm_release,
        .free_notifier          = arm_smmu_mmu_notifier_free,
 };
@@ -3257,8 +3283,10 @@ static void arm_smmu_mmu_notifier_put(struct 
arm_smmu_mmu_notifier *smmu_mn)
         * If we went through clear(), we've already invalidated, and no
         * new TLB entry can have been formed.
         */
-       if (!smmu_mn->cleared)
+       if (!smmu_mn->cleared) {
                arm_smmu_tlb_inv_asid(smmu_domain->smmu, cd->asid);
+               arm_smmu_atc_inv_domain(smmu_domain, mm->pasid, 0, 0);
+       }
 
        /* Frees smmu_mn */
        mmu_notifier_put(&smmu_mn->mn);
-- 
2.27.0

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

Reply via email to