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

Commit 9d5f8c20 authored by Leon Romanovsky's avatar Leon Romanovsky Committed by Jason Gunthorpe
Browse files

RDMA/core: Add resource tracking for create and destroy PDs 



Track create and destroy operations of PD objects.

Reviewed-by: default avatarMark Bloch <markb@mellanox.com>
Signed-off-by: default avatarLeon Romanovsky <leonro@mellanox.com>
Reviewed-by: default avatarSteve Wise <swise@opengridcomputing.com>
Signed-off-by: default avatarJason Gunthorpe <jgg@mellanox.com>
parent 08f294a1

drivers/infiniband/core/uverbs_cmd.c

+2 −0
Original line number Diff line number Diff line
@@ -340,6 +340,8 @@ ssize_t ib_uverbs_alloc_pd(struct ib_uverbs_file *file, 
	uobj->object = pd;

	memset(&resp, 0, sizeof resp);

	resp.pd_handle = uobj->id;

	pd->res.type = RDMA_RESTRACK_PD;

	rdma_restrack_add(&pd->res);



	if (copy_to_user(u64_to_user_ptr(cmd.response), &resp, sizeof resp)) {

		ret = -EFAULT;

drivers/infiniband/core/verbs.c

+5 −0
Original line number Diff line number Diff line
@@ -263,6 +263,10 @@ struct ib_pd *__ib_alloc_pd(struct ib_device *device, unsigned int flags, 
		mr_access_flags |= IB_ACCESS_REMOTE_READ | IB_ACCESS_REMOTE_WRITE;

	}



	pd->res.type = RDMA_RESTRACK_PD;

	pd->res.kern_name = caller;

	rdma_restrack_add(&pd->res);



	if (mr_access_flags) {

		struct ib_mr *mr;
@@ -312,6 +316,7 @@ void ib_dealloc_pd(struct ib_pd *pd) 
	   requires the caller to guarantee we can't race here. */

	WARN_ON(atomic_read(&pd->usecnt));



	rdma_restrack_del(&pd->res);

	/* Making delalloc_pd a void return is a WIP, no driver should return

	   an error here. */

	ret = pd->device->dealloc_pd(pd);