drivers/misc/cxl: Use maple tree iterators for cxl_prefault_vma()

Signed-off-by: Liam R. Howlett <Liam.Howlett@Oracle.com>

diff --git a/drivers/misc/cxl/fault.c b/drivers/misc/cxl/fault.c
index 01153b74334a1b394a1a9e5dba7fffed5921807d..b1e5ded5ee9037724738ee952045f3508f01d616 100644 (file)
--- a/drivers/misc/cxl/fault.c
+++ b/drivers/misc/cxl/fault.c
@@ -313,6 +313,7 @@ static void cxl_prefault_vma(struct cxl_context *ctx)
        struct vm_area_struct *vma;
        int rc;
        struct mm_struct *mm;
+       MA_STATE(mas, NULL, 0, 0);
 
        mm = get_mem_context(ctx);
        if (mm == NULL) {
@@ -321,8 +322,9 @@ static void cxl_prefault_vma(struct cxl_context *ctx)
                return;
        }
 
+       mas.tree = &mm->mm_mt;
        mmap_read_lock(mm);
-       for (vma = mm->mmap; vma; vma = vma->vm_next) {
+       mas_for_each(&mas, vma, ULONG_MAX) {
                for (ea = vma->vm_start; ea < vma->vm_end;
                                ea = next_segment(ea, slb.vsid)) {
                        rc = copro_calculate_slb(mm, ea, &slb);