mm/pagewalk: add walk_page_range_vma()

[ Upstream commit e07cda5f232fac4de0925d8a4c92e51e41fa2f6e ]

Let's add walk_page_range_vma(), which is similar to walk_page_vma(),
however, is only interested in a subset of the VMA range.

To be used in KSM code to stop using follow_page() next.

Link: https://lkml.kernel.org/r/20221021101141.84170-8-david@redhat.com
Signed-off-by: David Hildenbrand <david@redhat.com>
Cc: Andrea Arcangeli <aarcange@redhat.com>
Cc: Hugh Dickins <hughd@google.com>
Cc: Jason Gunthorpe <jgg@nvidia.com>
Cc: John Hubbard <jhubbard@nvidia.com>
Cc: Matthew Wilcox (Oracle) <willy@infradead.org>
Cc: Peter Xu <peterx@redhat.com>
Cc: Shuah Khan <shuah@kernel.org>
Cc: Vlastimil Babka <vbabka@suse.cz>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
Stable-dep-of: f5548c318d6 ("ksm: use range-walk function to jump over holes in scan_get_next_rmap_item")
Signed-off-by: Pedro Demarchi Gomes <pedrodemargomes@gmail.com>
Signed-off-by: Greg Kroah-Hartman <gregkh@linuxfoundation.org>

diff --git a/include/linux/pagewalk.h b/include/linux/pagewalk.h
index f3fafb731ffd87b0e1e1a4c8dae65764bd05f6f5..2f8f6cc980b4b5680a6901e2a83a5baeec6bea1a 100644 (file)
--- a/include/linux/pagewalk.h
+++ b/include/linux/pagewalk.h
@@ -99,6 +99,9 @@ int walk_page_range_novma(struct mm_struct *mm, unsigned long start,
                          unsigned long end, const struct mm_walk_ops *ops,
                          pgd_t *pgd,
                          void *private);
+int walk_page_range_vma(struct vm_area_struct *vma, unsigned long start,
+                       unsigned long end, const struct mm_walk_ops *ops,
+                       void *private);
 int walk_page_vma(struct vm_area_struct *vma, const struct mm_walk_ops *ops,
                void *private);
 int walk_page_mapping(struct address_space *mapping, pgoff_t first_index,

diff --git a/mm/pagewalk.c b/mm/pagewalk.c
index 2ff3a5bebcebb88778259a8814bb6d17cc83c878..5295f6d1a4faaed27abd4fe044846947c2aa28ee 100644 (file)
--- a/mm/pagewalk.c
+++ b/mm/pagewalk.c
@@ -517,6 +517,26 @@ int walk_page_range_novma(struct mm_struct *mm, unsigned long start,
        return walk_pgd_range(start, end, &walk);
 }
 
+int walk_page_range_vma(struct vm_area_struct *vma, unsigned long start,
+                       unsigned long end, const struct mm_walk_ops *ops,
+                       void *private)
+{
+       struct mm_walk walk = {
+               .ops            = ops,
+               .mm             = vma->vm_mm,
+               .vma            = vma,
+               .private        = private,
+       };
+
+       if (start >= end || !walk.mm)
+               return -EINVAL;
+       if (start < vma->vm_start || end > vma->vm_end)
+               return -EINVAL;
+
+       mmap_assert_locked(walk.mm);
+       return __walk_page_range(start, end, &walk);
+}
+
 int walk_page_vma(struct vm_area_struct *vma, const struct mm_walk_ops *ops,
                void *private)
 {