Re: [PATCH v2 6/6] iommu/mediatek: Convert tlb_flush_walk to gather_add_page

Robin Murphy
2020-11-25T16:38:56Z

On 2020-11-19 06:18, Yong Wu wrote:
> MediaTek TLB flush don't care about granule. when unmap, it could gather
> whole the iova range then do tlb flush once.
>
> In current v7s, If unmap the lvl2 pagetable, the steps are:
> step1: set this current pdg to 0.
> step2: tlb flush for this lvl2 block iova(1M).
> step3: free the lvl2 pagetable.
>
> This patch means we delay the step2 after unmap whole the iova.
> the iommu consumer HW should have stopped before it call dma_free_xx,
> thus, this delay looks ok.

If you can guarantee that no kind of speculative table walks can happen
(i.e. the hardware has no kind of translation prefetching that could be
triggered by, say, an unrelated device accessing an adjacent page) then
I guess it's probably OK to reason that this can be safe.

However at that point I wonder whether you really need to do anything
for flush_walk at all, if you don't need to differentiate between leaf
and non-leaf invalidations either. In DMA API usage you should never be
actually unmapping a table at block granularity, since the matching map
request would have used a block mapping in the first place, so I guess
you're hitting this in the case of *mapping* a block over an empty
table. For that, the subsequent iotlb_sync_map() will overlap the whole
region anyway. You'd still want to handle the general unmap case
properly for the sake of correctness, but AFAICS you could just
manipulate the gather data directly in mtk_iommu_unmap() far more easily
than threading a special case all the way through io-pgtable.

> Since tlb_flush_walk doesn't have the "gather" parameter, so we have to
> add this "gather" in ourself private data.
>
> Meanswhile, After this patch, the gather_add_pages will always be called,
> then "gather->start == ULONG_MAX" is impossible. remove this checking.
>
> Signed-off-by: Yong Wu <yong.wu@mediatek.com>
> ---
> tlb_flush_walk is designed for tlb flush range, I'm not sure whether it's ok
> if adding "gather" as a parameter in tlb_flush_walk. in this version, I put
> it into our private data.
> ---
> drivers/iommu/mtk_iommu.c | 21 ++++++++++++++++-----
> 1 file changed, 16 insertions(+), 5 deletions(-)
>
> diff --git a/drivers/iommu/mtk_iommu.c b/drivers/iommu/mtk_iommu.c
> index 94786860bd84..4c8200f4403a 100644
> --- a/drivers/iommu/mtk_iommu.c
> +++ b/drivers/iommu/mtk_iommu.c
> @@ -128,6 +128,8 @@ struct mtk_iommu_domain {
> struct io_pgtable_ops *iop;
>
> struct iommu_domain domain;
> +
> + struct iommu_iotlb_gather *gather;
> };
>
> static const struct iommu_ops mtk_iommu_ops;
> @@ -227,6 +229,17 @@ static void mtk_iommu_tlb_flush_range_sync(unsigned long iova, size_t size,
> }
> }
>
> +static void mtk_iommu_tlb_flush_walk(unsigned long iova, size_t size,
> + size_t granule, void *cookie)
> +{
> + struct mtk_iommu_data *data = cookie;
> + struct mtk_iommu_domain *m4u_dom = data->m4u_dom;
> + struct iommu_domain *domain = &m4u_dom->domain;
> +
> + /* Gather all the iova and tlb flush once after unmap. */
> + iommu_iotlb_gather_add_page(domain, m4u_dom->gather, iova, size);
> +}
> +
> static void mtk_iommu_tlb_flush_page_nosync(struct iommu_iotlb_gather *gather,
> unsigned long iova, size_t granule,
> void *cookie)
> @@ -239,8 +252,8 @@ static void mtk_iommu_tlb_flush_page_nosync(struct iommu_iotlb_gather *gather,
>
> static const struct iommu_flush_ops mtk_iommu_flush_ops = {
> .tlb_flush_all = mtk_iommu_tlb_flush_all,
> - .tlb_flush_walk = mtk_iommu_tlb_flush_range_sync,
> - .tlb_flush_leaf = mtk_iommu_tlb_flush_range_sync,
> + .tlb_flush_walk = mtk_iommu_tlb_flush_walk,
> + .tlb_flush_leaf = mtk_iommu_tlb_flush_walk,
> .tlb_add_page = mtk_iommu_tlb_flush_page_nosync,
> };
>
> @@ -432,6 +445,7 @@ static size_t mtk_iommu_unmap(struct iommu_domain *domain,
> {
> struct mtk_iommu_domain *dom = to_mtk_domain(domain);
>
> + dom->gather = gather;

Either way, this definitely doesn't work - multiple threads could be
unmapping different regions in the same domain at the same time. That's
why the gather structure has to be on each caller's stack in the first
place.

Robin.

> gather->granule_ignore = true;
> return dom->iop->unmap(dom->iop, iova, size, gather);
> }
> @@ -447,9 +461,6 @@ static void mtk_iommu_iotlb_sync(struct iommu_domain *domain,
> struct mtk_iommu_data *data = mtk_iommu_get_m4u_data();
> size_t length = gather->end - gather->start;
>
> - if (gather->start == ULONG_MAX)
> - return;
> -
> mtk_iommu_tlb_flush_range_sync(gather->start, length, gather->pgsize,
> data);
> }
>