[PATCH v3] IB/umem: Release pid in error and ODP flow

From: Kenneth Lee
Date: Fri Dec 30 2016 - 05:52:41 EST

There are two bugfixes in this patch:

Fixes: 87773dd56d54 ("IB: ib_umem_release() should decrement mm->pinned_vm from ib_umem_get")
This patch introduce the get_task_pid but not put it back on all error

Fixes: 8ada2c1c0c1d ("IB/core: Add support for on demand paging regions")
This patch introduce a ODP flow without release pid before enter it

Signed-off-by: Kenneth Lee <liguozhu@xxxxxxxxxxxxx>
Reviewed-by: Haggai Eran <haggaie@xxxxxxxxxxxx>
Change from v1 to v2:
Correcting the patch title and description
Change from v2 to v3:
Update the title and add "Fixes" fields in the description

drivers/infiniband/core/umem.c | 2 ++
1 file changed, 2 insertions(+)

diff --git a/drivers/infiniband/core/umem.c b/drivers/infiniband/core/umem.c
index 1e62a5f..4609b92 100644
--- a/drivers/infiniband/core/umem.c
+++ b/drivers/infiniband/core/umem.c
@@ -134,6 +134,7 @@ struct ib_umem *ib_umem_get(struct ib_ucontext *context, unsigned long addr,

if (access & IB_ACCESS_ON_DEMAND) {
+ put_pid(umem->pid);
ret = ib_umem_odp_get(context, umem);
if (ret) {
@@ -149,6 +150,7 @@ struct ib_umem *ib_umem_get(struct ib_ucontext *context, unsigned long addr,

page_list = (struct page **) __get_free_page(GFP_KERNEL);
if (!page_list) {
+ put_pid(umem->pid);
return ERR_PTR(-ENOMEM);