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

RDMA/restrack: Track ucontext 



Add ability to track allocated ib_ucontext, which are limited
resource and worth to be visible by users.

Signed-off-by: default avatarLeon Romanovsky <leonro@mellanox.com>
Signed-off-by: default avatarDoug Ledford <dledford@redhat.com>
parent 61d69528

drivers/infiniband/core/restrack.c

+5 −0
Original line number Diff line number Diff line
@@ -32,6 +32,7 @@ static const char *type2str(enum rdma_restrack_type type) 
		[RDMA_RESTRACK_QP] = "QP",

		[RDMA_RESTRACK_CM_ID] = "CM_ID",

		[RDMA_RESTRACK_MR] = "MR",

		[RDMA_RESTRACK_CTX] = "CTX",

	};



	return names[type];
@@ -130,6 +131,8 @@ static struct ib_device *res_to_dev(struct rdma_restrack_entry *res) 
				    res)->id.device;

	case RDMA_RESTRACK_MR:

		return container_of(res, struct ib_mr, res)->device;

	case RDMA_RESTRACK_CTX:

		return container_of(res, struct ib_ucontext, res)->device;

	default:

		WARN_ONCE(true, "Wrong resource tracking type %u\n", res->type);

		return NULL;
@@ -149,6 +152,8 @@ static bool res_is_user(struct rdma_restrack_entry *res) 
		return !res->kern_name;

	case RDMA_RESTRACK_MR:

		return container_of(res, struct ib_mr, res)->pd->uobject;

	case RDMA_RESTRACK_CTX:

		return true;

	default:

		WARN_ONCE(true, "Wrong resource tracking type %u\n", res->type);

		return false;

include/rdma/ib_verbs.h

+4 −0
Original line number Diff line number Diff line
@@ -1505,6 +1505,10 @@ struct ib_ucontext { 
#endif



	struct ib_rdmacg_object	cg_obj;

	/*

	 * Implementation details of the RDMA core, don't use in drivers:

	 */

	struct rdma_restrack_entry res;

};



struct ib_uobject {

include/rdma/restrack.h

+4 −0
Original line number Diff line number Diff line
@@ -38,6 +38,10 @@ enum rdma_restrack_type { 
	 * @RDMA_RESTRACK_MR: Memory Region (MR)

	 */

	RDMA_RESTRACK_MR,

	/**

	 * @RDMA_RESTRACK_CTX: Verbs contexts (CTX)

	 */

	RDMA_RESTRACK_CTX,

	/**

	 * @RDMA_RESTRACK_MAX: Last entry, used for array dclarations

	 */