[PATCH v2 rdma-next 4/5] RDMA/NLDEV: helper functions to add provider attributes

[Steve Wise <swise@xxxxxxxxxxxxxxxxxxxxx>]

These help rdma providers to fill out the provider entries.

Signed-off-by: Steve Wise <swise@xxxxxxxxxxxxxxxxxxxxx>
---
 drivers/infiniband/core/nldev.c | 79 +++++++++++++++++++++++++++++++++++++++++
 include/rdma/restrack.h         | 19 ++++++++++
 2 files changed, 98 insertions(+)

diff --git a/drivers/infiniband/core/nldev.c b/drivers/infiniband/core/nldev.c
index 902517f..301646b 100644
--- a/drivers/infiniband/core/nldev.c
+++ b/drivers/infiniband/core/nldev.c
@@ -131,6 +131,85 @@ static int provider_fill_port_info(struct sk_buff *msg,
 		device->res.fill_port_info(msg, cb, device, port) : 0;
 }
 
+static int put_provider_name_print_type(struct sk_buff *msg, const char *name,
+					enum rdma_nldev_print_type print_type)
+{
+	if (nla_put_string(msg, RDMA_NLDEV_ATTR_PROVIDER_STRING, name))
+		return -EMSGSIZE;
+	if (print_type != RDMA_NLDEV_PRINT_TYPE_UNSPEC &&
+	    nla_put_u8(msg, RDMA_NLDEV_ATTR_PROVIDER_PRINT_TYPE, print_type))
+		return -EMSGSIZE;
+
+	return 0;
+}
+
+int rdma_nl_put_provider_u32(struct sk_buff *msg, const char *name,
+			     enum rdma_nldev_print_type print_type,
+			     u32 value)
+{
+	if (put_provider_name_print_type(msg, name, print_type))
+		return -EMSGSIZE;
+	if (nla_put_u32(msg, RDMA_NLDEV_ATTR_PROVIDER_U32, value))
+		return -EMSGSIZE;
+
+	return 0;
+}
+EXPORT_SYMBOL(rdma_nl_put_provider_u32);
+
+int rdma_nl_put_provider_s32(struct sk_buff *msg, const char *name,
+			     enum rdma_nldev_print_type print_type,
+			     s32 value)
+{
+	if (put_provider_name_print_type(msg, name, print_type))
+		return -EMSGSIZE;
+	if (nla_put_s32(msg, RDMA_NLDEV_ATTR_PROVIDER_S32, value))
+		return -EMSGSIZE;
+
+	return 0;
+}
+EXPORT_SYMBOL(rdma_nl_put_provider_s32);
+
+int rdma_nl_put_provider_u64(struct sk_buff *msg, const char *name,
+			     enum rdma_nldev_print_type print_type,
+			     u64 value)
+{
+	if (put_provider_name_print_type(msg, name, print_type))
+		return -EMSGSIZE;
+	if (nla_put_u64_64bit(msg, RDMA_NLDEV_ATTR_PROVIDER_U64, value,
+			      RDMA_NLDEV_ATTR_PAD))
+		return -EMSGSIZE;
+
+	return 0;
+}
+EXPORT_SYMBOL(rdma_nl_put_provider_u64);
+
+int rdma_nl_put_provider_s64(struct sk_buff *msg, const char *name,
+			     enum rdma_nldev_print_type print_type,
+			     s64 value)
+{
+	if (put_provider_name_print_type(msg, name, print_type))
+		return -EMSGSIZE;
+	if (nla_put_s64(msg, RDMA_NLDEV_ATTR_PROVIDER_S64, value,
+			RDMA_NLDEV_ATTR_PAD))
+		return -EMSGSIZE;
+
+	return 0;
+}
+EXPORT_SYMBOL(rdma_nl_put_provider_s64);
+
+int rdma_nl_put_provider_string(struct sk_buff *msg, const char *name,
+				const char *value)
+{
+	if (put_provider_name_print_type(msg, name,
+					 RDMA_NLDEV_PRINT_TYPE_UNSPEC))
+		return -EMSGSIZE;
+	if (nla_put_string(msg, RDMA_NLDEV_ATTR_PROVIDER_STRING, value))
+		return -EMSGSIZE;
+
+	return 0;
+}
+EXPORT_SYMBOL(rdma_nl_put_provider_string);
+
 static int fill_nldev_handle(struct sk_buff *msg, struct ib_device *device)
 {
 	if (nla_put_u32(msg, RDMA_NLDEV_ATTR_DEV_INDEX, device->index))
diff --git a/include/rdma/restrack.h b/include/rdma/restrack.h
index 936b56f..094a209 100644
--- a/include/rdma/restrack.h
+++ b/include/rdma/restrack.h
@@ -12,6 +12,7 @@
 #include <linux/kref.h>
 #include <linux/completion.h>
 #include <linux/sched/task.h>
+#include <uapi/rdma/rdma_netlink.h>
 
 /**
  * enum rdma_restrack_type - HW objects to track
@@ -199,4 +200,22 @@ static inline void rdma_restrack_set_task(struct rdma_restrack_entry *res,
 	res->task = task;
 }
 
+/*
+ * Helper functions for rdma providers when filling out
+ * nldev provider attributes.
+ */
+int rdma_nl_put_provider_u32(struct sk_buff *msg, const char *name,
+			     enum rdma_nldev_print_type print_type,
+			     u32 value);
+int rdma_nl_put_provider_s32(struct sk_buff *msg, const char *name,
+			     enum rdma_nldev_print_type print_type,
+			     s32 value);
+int rdma_nl_put_provider_u64(struct sk_buff *msg, const char *name,
+			     enum rdma_nldev_print_type print_type,
+			     u64 value);
+int rdma_nl_put_provider_s64(struct sk_buff *msg, const char *name,
+			     enum rdma_nldev_print_type print_type,
+			     s64 value);
+int rdma_nl_put_provider_string(struct sk_buff *msg, const char *name,
+				const char *value);
 #endif /* _RDMA_RESTRACK_H_ */
-- 
1.8.3.1

--
To unsubscribe from this list: send the line "unsubscribe linux-rdma" in
the body of a message to majordomo@xxxxxxxxxxxxxxx
More majordomo info at  http://vger.kernel.org/majordomo-info.html