mm/damon/vaddr-test.h: stop using vma_mas_store() for maple tree store

[platform/kernel/linux-rpi.git] / mm / damon / vaddr-test.h

diff --git a/mm/damon/vaddr-test.h b/mm/damon/vaddr-test.h
index bce37c4..c4b455b 100644 (file)
--- a/mm/damon/vaddr-test.h
+++ b/mm/damon/vaddr-test.h
@@ -14,19 +14,26 @@
 
 #include <kunit/test.h>
 
-static void __link_vmas(struct maple_tree *mt, struct vm_area_struct *vmas,
+static int __link_vmas(struct maple_tree *mt, struct vm_area_struct *vmas,
                        ssize_t nr_vmas)
 {
-       int i;
+       int i, ret = -ENOMEM;
        MA_STATE(mas, mt, 0, 0);
 
        if (!nr_vmas)
-               return;
+               return 0;
 
        mas_lock(&mas);
-       for (i = 0; i < nr_vmas; i++)
-               vma_mas_store(&vmas[i], &mas);
+       for (i = 0; i < nr_vmas; i++) {
+               mas_set_range(&mas, vmas[i].vm_start, vmas[i].vm_end - 1);
+               if (mas_store_gfp(&mas, &vmas[i], GFP_KERNEL))
+                       goto failed;
+       }
+
+       ret = 0;
+failed:
        mas_unlock(&mas);
+       return ret;
 }
 
 /*
@@ -71,7 +78,8 @@ static void damon_test_three_regions_in_vmas(struct kunit *test)
        };
 
        mt_init_flags(&mm.mm_mt, MM_MT_FLAGS);
-       __link_vmas(&mm.mm_mt, vmas, ARRAY_SIZE(vmas));
+       if (__link_vmas(&mm.mm_mt, vmas, ARRAY_SIZE(vmas)))
+               kunit_skip(test, "Failed to create VMA tree");
 
        __damon_va_three_regions(&mm, regions);