Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 79bb5b7e authored by Leon Romanovsky's avatar Leon Romanovsky Committed by Doug Ledford
Browse files

RDMA/umem: Fix missing mmap_sem in get umem ODP call 



Add mmap_sem lock around VMA inspection in ib_umem_odp_get().

Fixes: 0008b84e ('IB/umem: Add support to huge ODP')
Signed-off-by: default avatarArtemy Kovalyov <artemyko@mellanox.com>
Reviewed-by: default avatarEli Cohen <eli@mellanox.com>
Signed-off-by: default avatarLeon Romanovsky <leonro@mellanox.com>
Signed-off-by: default avatarDoug Ledford <dledford@redhat.com>
parent 53376fed

drivers/infiniband/core/umem_odp.c

+5 −1
Original line number Diff line number Diff line
@@ -321,11 +321,15 @@ int ib_umem_odp_get(struct ib_ucontext *context, struct ib_umem *umem, 
		struct vm_area_struct *vma;

		struct hstate *h;



		down_read(&mm->mmap_sem);

		vma = find_vma(mm, ib_umem_start(umem));

		if (!vma || !is_vm_hugetlb_page(vma))

		if (!vma || !is_vm_hugetlb_page(vma)) {

			up_read(&mm->mmap_sem);

			return -EINVAL;

		}

		h = hstate_vma(vma);

		umem->page_shift = huge_page_shift(h);

		up_read(&mm->mmap_sem);

		umem->hugetlb = 1;

	} else {

		umem->hugetlb = 0;