1 files changed, 7 insertions, 5 deletions

diff --git a/drivers/media/platform/omap/omap_vout.c b/drivers/media/platform/omap/omap_vout.c
index dade3ceab092..96c4a17e4280 100644
--- a/drivers/media/platform/omap/omap_vout.c
+++ b/drivers/media/platform/omap/omap_vout.c
@@ -205,19 +205,21 @@ static u32 omap_vout_uservirt_to_phys(u32 virtp)
         struct vm_area_struct *vma;
         struct mm_struct *mm = current->mm;
 
-        vma = find_vma(mm, virtp);
         /* For kernel direct-mapped memory, take the easy way */
-        if (virtp >= PAGE_OFFSET) {
-                physp = virt_to_phys((void *) virtp);
-        } else if (vma && (vma->vm_flags & VM_IO) && vma->vm_pgoff) {
+        if (virtp >= PAGE_OFFSET)
+                return virt_to_phys((void *) virtp);
+
+        down_read(&current->mm->mmap_sem);
+        vma = find_vma(mm, virtp);
+        if (vma && (vma->vm_flags & VM_IO) && vma->vm_pgoff) {
                 /* this will catch, kernel-allocated, mmaped-to-usermode
                    addresses */
                 physp = (vma->vm_pgoff << PAGE_SHIFT) + (virtp - vma->vm_start);
+                up_read(&current->mm->mmap_sem);
         } else {
                 /* otherwise, use get_user_pages() for general userland pages */
                 int res, nr_pages = 1;
                 struct page *pages;
-                down_read(&current->mm->mmap_sem);
 
                 res = get_user_pages(current, current->mm, virtp, nr_pages, 1,
                                 0, &pages, NULL);