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

Commit 29cf1351 authored by Steve Wise's avatar Steve Wise Committed by Doug Ledford
Browse files

RDMA/nldev: provide detailed PD information 



Implement the RDMA nldev netlink interface for dumping detailed PD
information.

Reviewed-by: default avatarLeon Romanovsky <leonro@mellanox.com>
Signed-off-by: default avatarSteve Wise <swise@opengridcomputing.com>
Signed-off-by: default avatarDoug Ledford <dledford@redhat.com>
parent 52924434

drivers/infiniband/core/nldev.c

+57 −0
Original line number Diff line number Diff line
@@ -91,6 +91,10 @@ static const struct nla_policy nldev_policy[RDMA_NLDEV_ATTR_MAX] = { 
	[RDMA_NLDEV_ATTR_RES_LKEY]		= { .type = NLA_U32 },

	[RDMA_NLDEV_ATTR_RES_IOVA]		= { .type = NLA_U64 },

	[RDMA_NLDEV_ATTR_RES_MRLEN]		= { .type = NLA_U64 },

	[RDMA_NLDEV_ATTR_RES_PD]		= { .type = NLA_NESTED },

	[RDMA_NLDEV_ATTR_RES_PD_ENTRY]		= { .type = NLA_NESTED },

	[RDMA_NLDEV_ATTR_RES_LOCAL_DMA_LKEY]	= { .type = NLA_U32 },

	[RDMA_NLDEV_ATTR_RES_UNSAFE_GLOBAL_RKEY] = { .type = NLA_U32 },

};



static int fill_nldev_handle(struct sk_buff *msg, struct ib_device *device)
@@ -439,6 +443,45 @@ static int fill_res_mr_entry(struct sk_buff *msg, struct netlink_callback *cb, 
	return -EMSGSIZE;

}



static int fill_res_pd_entry(struct sk_buff *msg, struct netlink_callback *cb,

			     struct rdma_restrack_entry *res, uint32_t port)

{

	struct ib_pd *pd = container_of(res, struct ib_pd, res);

	struct nlattr *entry_attr;



	entry_attr = nla_nest_start(msg, RDMA_NLDEV_ATTR_RES_PD_ENTRY);

	if (!entry_attr)

		goto out;



	if (netlink_capable(cb->skb, CAP_NET_ADMIN)) {

		if (nla_put_u32(msg, RDMA_NLDEV_ATTR_RES_LOCAL_DMA_LKEY,

				pd->local_dma_lkey))

			goto err;

		if ((pd->flags & IB_PD_UNSAFE_GLOBAL_RKEY) &&

		    nla_put_u32(msg, RDMA_NLDEV_ATTR_RES_UNSAFE_GLOBAL_RKEY,

				pd->unsafe_global_rkey))

			goto err;

	}

	if (nla_put_u64_64bit(msg, RDMA_NLDEV_ATTR_RES_USECNT,

			      atomic_read(&pd->usecnt), 0))

		goto err;

	if ((pd->flags & IB_PD_UNSAFE_GLOBAL_RKEY) &&

	    nla_put_u32(msg, RDMA_NLDEV_ATTR_RES_UNSAFE_GLOBAL_RKEY,

			pd->unsafe_global_rkey))

		goto err;



	if (fill_res_name_pid(msg, res))

		goto err;



	nla_nest_end(msg, entry_attr);

	return 0;



err:

	nla_nest_cancel(msg, entry_attr);

out:

	return -EMSGSIZE;

}



static int nldev_get_doit(struct sk_buff *skb, struct nlmsghdr *nlh,

			  struct netlink_ext_ack *extack)

{
@@ -741,6 +784,11 @@ static const struct nldev_fill_res_entry fill_entries[RDMA_RESTRACK_MAX] = { 
		.nldev_cmd = RDMA_NLDEV_CMD_RES_MR_GET,

		.nldev_attr = RDMA_NLDEV_ATTR_RES_MR,

	},

	[RDMA_RESTRACK_PD] = {

		.fill_res_func = fill_res_pd_entry,

		.nldev_cmd = RDMA_NLDEV_CMD_RES_PD_GET,

		.nldev_attr = RDMA_NLDEV_ATTR_RES_PD,

	},

};



static int res_get_common_dumpit(struct sk_buff *skb,
@@ -901,6 +949,12 @@ static int nldev_res_get_mr_dumpit(struct sk_buff *skb, 
	return res_get_common_dumpit(skb, cb, RDMA_RESTRACK_MR);

}



static int nldev_res_get_pd_dumpit(struct sk_buff *skb,

				   struct netlink_callback *cb)

{

	return res_get_common_dumpit(skb, cb, RDMA_RESTRACK_PD);

}



static const struct rdma_nl_cbs nldev_cb_table[RDMA_NLDEV_NUM_OPS] = {

	[RDMA_NLDEV_CMD_GET] = {

		.doit = nldev_get_doit,
@@ -936,6 +990,9 @@ static const struct rdma_nl_cbs nldev_cb_table[RDMA_NLDEV_NUM_OPS] = { 
	[RDMA_NLDEV_CMD_RES_MR_GET] = {

		.dump = nldev_res_get_mr_dumpit,

	},

	[RDMA_NLDEV_CMD_RES_PD_GET] = {

		.dump = nldev_res_get_pd_dumpit,

	},

};



void __init nldev_init(void)

include/uapi/rdma/rdma_netlink.h

+7 −0
Original line number Diff line number Diff line
@@ -244,6 +244,8 @@ enum rdma_nldev_command { 


	RDMA_NLDEV_CMD_RES_MR_GET, /* can dump */



	RDMA_NLDEV_CMD_RES_PD_GET, /* can dump */



	RDMA_NLDEV_NUM_OPS

};
@@ -381,6 +383,11 @@ enum rdma_nldev_attr { 
	RDMA_NLDEV_ATTR_RES_IOVA,		/* u64 */

	RDMA_NLDEV_ATTR_RES_MRLEN,		/* u64 */



	RDMA_NLDEV_ATTR_RES_PD,			/* nested table */

	RDMA_NLDEV_ATTR_RES_PD_ENTRY,		/* nested table */

	RDMA_NLDEV_ATTR_RES_LOCAL_DMA_LKEY,	/* u32 */

	RDMA_NLDEV_ATTR_RES_UNSAFE_GLOBAL_RKEY,	/* u32 */



	RDMA_NLDEV_ATTR_MAX

};

#endif /* _UAPI_RDMA_NETLINK_H */