drivers/tee/optee: Use maple tree iterators for __check_mem_type()

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

diff --git a/drivers/tee/optee/call.c b/drivers/tee/optee/call.c
index 6132cc8d014c08512b487a31041844f65dbd6830..cfe55b9e92b6478a1bc15b1c1f494f6710c3e74a 100644 (file)
--- a/drivers/tee/optee/call.c
+++ b/drivers/tee/optee/call.c
@@ -550,11 +550,18 @@ static bool is_normal_memory(pgprot_t p)
 
 static int __check_mem_type(struct vm_area_struct *vma, unsigned long end)
 {
-       while (vma && is_normal_memory(vma->vm_page_prot)) {
-               if (vma->vm_end >= end)
-                       return 0;
-               vma = vma->vm_next;
+       MA_STATE(mas, &vma->vm_mm->mm_mt, vma->vm_start, vma->vm_start);
+
+
+       rcu_read_lock();
+       mas_for_each(&mas, vma, end) {
+               if (!is_normal_memory(vma->vm_page_prot))
+                   break;
        }
+       rcu_read_unlock();
+
+       if (!vma)
+               return 0;
 
        return -EINVAL;
 }