mm/mmap: Use the maple tree in find_vma() instead of the rbtree.
authorLiam R. Howlett <Liam.Howlett@Oracle.com>
Fri, 24 Jul 2020 15:38:39 +0000 (11:38 -0400)
committerLiam R. Howlett <Liam.Howlett@Oracle.com>
Wed, 19 May 2021 20:39:06 +0000 (16:39 -0400)
Using the maple tree interface mt_find() will handle the RCU locking and
will start searching at the address up to the limit, ULONG_MAX in this
case.

Add kernel documentation to this API.

Signed-off-by: Liam R. Howlett <Liam.Howlett@Oracle.com>
mm/mmap.c

index 2b7270e7133ce8e19d4a4c08c6bdfca9b64f4504..fbf51fb56cbf0dde873d7a5aaabfd768d11197ac 100644 (file)
--- a/mm/mmap.c
+++ b/mm/mmap.c
@@ -2482,10 +2482,16 @@ get_unmapped_area(struct file *file, unsigned long addr, unsigned long len,
 
 EXPORT_SYMBOL(get_unmapped_area);
 
-/* Look up the first VMA which satisfies  addr < vm_end,  NULL if none. */
+/**
+ * find_vma() - Find the VMA for a given address, or the next vma.
+ * @mm: The mm_struct to check
+ * @addr: The address
+ *
+ * Returns: The VMA associated with addr, or the next vma.
+ * May return %NULL in the case of no vma at addr or above.
+ */
 struct vm_area_struct *find_vma(struct mm_struct *mm, unsigned long addr)
 {
-       struct rb_node *rb_node;
        struct vm_area_struct *vma;
 
        /* Check the cache first. */
@@ -2493,24 +2499,10 @@ struct vm_area_struct *find_vma(struct mm_struct *mm, unsigned long addr)
        if (likely(vma))
                return vma;
 
-       rb_node = mm->mm_rb.rb_node;
-
-       while (rb_node) {
-               struct vm_area_struct *tmp;
-
-               tmp = rb_entry(rb_node, struct vm_area_struct, vm_rb);
-
-               if (tmp->vm_end > addr) {
-                       vma = tmp;
-                       if (tmp->vm_start <= addr)
-                               break;
-                       rb_node = rb_node->rb_left;
-               } else
-                       rb_node = rb_node->rb_right;
-       }
-
+       vma = mt_find(&mm->mm_mt, &addr, ULONG_MAX);
        if (vma)
                vmacache_update(addr, vma);
+
        return vma;
 }
 EXPORT_SYMBOL(find_vma);