]> git.proxmox.com Git - mirror_iproute2.git/blobdiff - rdma/utils.c
rdma: Properly mark RDMAtool license
[mirror_iproute2.git] / rdma / utils.c
index 157699c078c8dd898a3401b1f021973c1f6da1ba..bce052d5b17b11163d0f691ef8722cd9d119ef9a 100644 (file)
@@ -1,16 +1,12 @@
+// SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB
 /*
  * utils.c     RDMA tool
- *
- *              This program is free software; you can redistribute it and/or
- *              modify it under the terms of the GNU General Public License
- *              as published by the Free Software Foundation; either version
- *              2 of the License, or (at your option) any later version.
- *
  * Authors:     Leon Romanovsky <leonro@mellanox.com>
  */
 
 #include "rdma.h"
 #include <ctype.h>
+#include <inttypes.h>
 
 int rd_argc(struct rd *rd)
 {
@@ -24,7 +20,7 @@ char *rd_argv(struct rd *rd)
        return *rd->argv;
 }
 
-static int strcmpx(const char *str1, const char *str2)
+int strcmpx(const char *str1, const char *str2)
 {
        if (strlen(str1) > strlen(str2))
                return -1;
@@ -98,6 +94,10 @@ static struct dev_map *dev_map_alloc(const char *dev_name)
        if (!dev_map)
                return NULL;
        dev_map->dev_name = strdup(dev_name);
+       if (!dev_map->dev_name) {
+               free(dev_map);
+               return NULL;
+       }
 
        return dev_map;
 }
@@ -356,11 +356,51 @@ static const enum mnl_attr_data_type nldev_policy[RDMA_NLDEV_ATTR_MAX] = {
        [RDMA_NLDEV_ATTR_PORT_STATE] = MNL_TYPE_U8,
        [RDMA_NLDEV_ATTR_PORT_PHYS_STATE] = MNL_TYPE_U8,
        [RDMA_NLDEV_ATTR_DEV_NODE_TYPE] = MNL_TYPE_U8,
+       [RDMA_NLDEV_ATTR_RES_SUMMARY]   = MNL_TYPE_NESTED,
+       [RDMA_NLDEV_ATTR_RES_SUMMARY_ENTRY]     = MNL_TYPE_NESTED,
+       [RDMA_NLDEV_ATTR_RES_SUMMARY_ENTRY_NAME] = MNL_TYPE_NUL_STRING,
+       [RDMA_NLDEV_ATTR_RES_SUMMARY_ENTRY_CURR] = MNL_TYPE_U64,
+       [RDMA_NLDEV_ATTR_RES_QP]                = MNL_TYPE_NESTED,
+       [RDMA_NLDEV_ATTR_RES_QP_ENTRY]          = MNL_TYPE_NESTED,
+       [RDMA_NLDEV_ATTR_RES_LQPN]      = MNL_TYPE_U32,
+       [RDMA_NLDEV_ATTR_RES_RQPN]      = MNL_TYPE_U32,
+       [RDMA_NLDEV_ATTR_RES_RQ_PSN]            = MNL_TYPE_U32,
+       [RDMA_NLDEV_ATTR_RES_SQ_PSN]            = MNL_TYPE_U32,
+       [RDMA_NLDEV_ATTR_RES_PATH_MIG_STATE]    = MNL_TYPE_U8,
+       [RDMA_NLDEV_ATTR_RES_TYPE]              = MNL_TYPE_U8,
+       [RDMA_NLDEV_ATTR_RES_STATE]             = MNL_TYPE_U8,
+       [RDMA_NLDEV_ATTR_RES_PID]               = MNL_TYPE_U32,
+       [RDMA_NLDEV_ATTR_RES_KERN_NAME] = MNL_TYPE_NUL_STRING,
+       [RDMA_NLDEV_ATTR_RES_CM_ID]             = MNL_TYPE_NESTED,
+       [RDMA_NLDEV_ATTR_RES_CM_ID_ENTRY]       = MNL_TYPE_NESTED,
+       [RDMA_NLDEV_ATTR_RES_PS]                = MNL_TYPE_U32,
+       [RDMA_NLDEV_ATTR_RES_SRC_ADDR]          = MNL_TYPE_UNSPEC,
+       [RDMA_NLDEV_ATTR_RES_DST_ADDR]          = MNL_TYPE_UNSPEC,
+       [RDMA_NLDEV_ATTR_RES_CQ] = MNL_TYPE_NESTED,
+       [RDMA_NLDEV_ATTR_RES_CQ_ENTRY] = MNL_TYPE_NESTED,
+       [RDMA_NLDEV_ATTR_RES_CQE] = MNL_TYPE_U32,
+       [RDMA_NLDEV_ATTR_RES_USECNT] = MNL_TYPE_U64,
+       [RDMA_NLDEV_ATTR_RES_POLL_CTX] = MNL_TYPE_U8,
+       [RDMA_NLDEV_ATTR_RES_MR] = MNL_TYPE_NESTED,
+       [RDMA_NLDEV_ATTR_RES_MR_ENTRY] = MNL_TYPE_NESTED,
+       [RDMA_NLDEV_ATTR_RES_RKEY] = MNL_TYPE_U32,
+       [RDMA_NLDEV_ATTR_RES_LKEY] = MNL_TYPE_U32,
+       [RDMA_NLDEV_ATTR_RES_IOVA] = MNL_TYPE_U64,
+       [RDMA_NLDEV_ATTR_RES_MRLEN] = MNL_TYPE_U64,
+       [RDMA_NLDEV_ATTR_NDEV_INDEX]            = MNL_TYPE_U32,
+       [RDMA_NLDEV_ATTR_NDEV_NAME]             = MNL_TYPE_NUL_STRING,
+       [RDMA_NLDEV_ATTR_DRIVER] = MNL_TYPE_NESTED,
+       [RDMA_NLDEV_ATTR_DRIVER_ENTRY] = MNL_TYPE_NESTED,
+       [RDMA_NLDEV_ATTR_DRIVER_STRING] = MNL_TYPE_NUL_STRING,
+       [RDMA_NLDEV_ATTR_DRIVER_PRINT_TYPE] = MNL_TYPE_U8,
+       [RDMA_NLDEV_ATTR_DRIVER_S32] = MNL_TYPE_U32,
+       [RDMA_NLDEV_ATTR_DRIVER_U32] = MNL_TYPE_U32,
+       [RDMA_NLDEV_ATTR_DRIVER_S64] = MNL_TYPE_U64,
+       [RDMA_NLDEV_ATTR_DRIVER_U64] = MNL_TYPE_U64,
 };
 
-int rd_attr_cb(const struct nlattr *attr, void *data)
+int rd_attr_check(const struct nlattr *attr, int *typep)
 {
-       const struct nlattr **tb = data;
        int type;
 
        if (mnl_attr_type_valid(attr, RDMA_NLDEV_ATTR_MAX) < 0)
@@ -368,6 +408,24 @@ int rd_attr_cb(const struct nlattr *attr, void *data)
 
        type = mnl_attr_get_type(attr);
 
+       if (mnl_attr_validate(attr, nldev_policy[type]) < 0)
+               return MNL_CB_ERROR;
+
+       *typep = nldev_policy[type];
+       return MNL_CB_OK;
+}
+
+int rd_attr_cb(const struct nlattr *attr, void *data)
+{
+       const struct nlattr **tb = data;
+       int type;
+
+       if (mnl_attr_type_valid(attr, RDMA_NLDEV_ATTR_MAX - 1) < 0)
+               /* We received unknown attribute */
+               return MNL_CB_OK;
+
+       type = mnl_attr_get_type(attr);
+
        if (mnl_attr_validate(attr, nldev_policy[type]) < 0)
                return MNL_CB_ERROR;
 
@@ -412,6 +470,25 @@ void rd_free(struct rd *rd)
        filters_cleanup(rd);
 }
 
+int rd_set_arg_to_devname(struct rd *rd)
+{
+       int ret = 0;
+
+       while (!rd_no_arg(rd)) {
+               if (rd_argv_match(rd, "dev") || rd_argv_match(rd, "link")) {
+                       rd_arg_inc(rd);
+                       if (rd_no_arg(rd)) {
+                               pr_err("No device name was supplied\n");
+                               ret = -EINVAL;
+                       }
+                       goto out;
+               }
+               rd_arg_inc(rd);
+       }
+out:
+       return ret;
+}
+
 int rd_exec_link(struct rd *rd, int (*cb)(struct rd *rd), bool strict_port)
 {
        struct dev_map *dev_map;
@@ -495,6 +572,16 @@ out:
        return ret;
 }
 
+int rd_exec_require_dev(struct rd *rd, int (*cb)(struct rd *rd))
+{
+       if (rd_no_arg(rd)) {
+               pr_err("Please provide device name.\n");
+               return -EINVAL;
+       }
+
+       return rd_exec_dev(rd, cb);
+}
+
 int rd_exec_cmd(struct rd *rd, const struct rd_cmd *cmds, const char *str)
 {
        const struct rd_cmd *c;
@@ -603,3 +690,174 @@ struct dev_map *dev_map_lookup(struct rd *rd, bool allow_port_index)
        free(dev_name);
        return dev_map;
 }
+
+#define nla_type(attr) ((attr)->nla_type & NLA_TYPE_MASK)
+
+void newline(struct rd *rd)
+{
+       if (rd->json_output)
+               jsonw_end_array(rd->jw);
+       else
+               pr_out("\n");
+}
+
+void newline_indent(struct rd *rd)
+{
+       newline(rd);
+       if (!rd->json_output)
+               pr_out("    ");
+}
+
+static int print_driver_string(struct rd *rd, const char *key_str,
+                                const char *val_str)
+{
+       if (rd->json_output) {
+               jsonw_string_field(rd->jw, key_str, val_str);
+               return 0;
+       } else {
+               return pr_out("%s %s ", key_str, val_str);
+       }
+}
+
+static int print_driver_s32(struct rd *rd, const char *key_str, int32_t val,
+                             enum rdma_nldev_print_type print_type)
+{
+       if (rd->json_output) {
+               jsonw_int_field(rd->jw, key_str, val);
+               return 0;
+       }
+       switch (print_type) {
+       case RDMA_NLDEV_PRINT_TYPE_UNSPEC:
+               return pr_out("%s %d ", key_str, val);
+       case RDMA_NLDEV_PRINT_TYPE_HEX:
+               return pr_out("%s 0x%x ", key_str, val);
+       default:
+               return -EINVAL;
+       }
+}
+
+static int print_driver_u32(struct rd *rd, const char *key_str, uint32_t val,
+                             enum rdma_nldev_print_type print_type)
+{
+       if (rd->json_output) {
+               jsonw_int_field(rd->jw, key_str, val);
+               return 0;
+       }
+       switch (print_type) {
+       case RDMA_NLDEV_PRINT_TYPE_UNSPEC:
+               return pr_out("%s %u ", key_str, val);
+       case RDMA_NLDEV_PRINT_TYPE_HEX:
+               return pr_out("%s 0x%x ", key_str, val);
+       default:
+               return -EINVAL;
+       }
+}
+
+static int print_driver_s64(struct rd *rd, const char *key_str, int64_t val,
+                             enum rdma_nldev_print_type print_type)
+{
+       if (rd->json_output) {
+               jsonw_int_field(rd->jw, key_str, val);
+               return 0;
+       }
+       switch (print_type) {
+       case RDMA_NLDEV_PRINT_TYPE_UNSPEC:
+               return pr_out("%s %" PRId64 " ", key_str, val);
+       case RDMA_NLDEV_PRINT_TYPE_HEX:
+               return pr_out("%s 0x%" PRIx64 " ", key_str, val);
+       default:
+               return -EINVAL;
+       }
+}
+
+static int print_driver_u64(struct rd *rd, const char *key_str, uint64_t val,
+                             enum rdma_nldev_print_type print_type)
+{
+       if (rd->json_output) {
+               jsonw_int_field(rd->jw, key_str, val);
+               return 0;
+       }
+       switch (print_type) {
+       case RDMA_NLDEV_PRINT_TYPE_UNSPEC:
+               return pr_out("%s %" PRIu64 " ", key_str, val);
+       case RDMA_NLDEV_PRINT_TYPE_HEX:
+               return pr_out("%s 0x%" PRIx64 " ", key_str, val);
+       default:
+               return -EINVAL;
+       }
+}
+
+static int print_driver_entry(struct rd *rd, struct nlattr *key_attr,
+                               struct nlattr *val_attr,
+                               enum rdma_nldev_print_type print_type)
+{
+       const char *key_str = mnl_attr_get_str(key_attr);
+       int attr_type = nla_type(val_attr);
+
+       switch (attr_type) {
+       case RDMA_NLDEV_ATTR_DRIVER_STRING:
+               return print_driver_string(rd, key_str,
+                               mnl_attr_get_str(val_attr));
+       case RDMA_NLDEV_ATTR_DRIVER_S32:
+               return print_driver_s32(rd, key_str,
+                               mnl_attr_get_u32(val_attr), print_type);
+       case RDMA_NLDEV_ATTR_DRIVER_U32:
+               return print_driver_u32(rd, key_str,
+                               mnl_attr_get_u32(val_attr), print_type);
+       case RDMA_NLDEV_ATTR_DRIVER_S64:
+               return print_driver_s64(rd, key_str,
+                               mnl_attr_get_u64(val_attr), print_type);
+       case RDMA_NLDEV_ATTR_DRIVER_U64:
+               return print_driver_u64(rd, key_str,
+                               mnl_attr_get_u64(val_attr), print_type);
+       }
+       return -EINVAL;
+}
+
+void print_driver_table(struct rd *rd, struct nlattr *tb)
+{
+       int print_type = RDMA_NLDEV_PRINT_TYPE_UNSPEC;
+       struct nlattr *tb_entry, *key = NULL, *val;
+       int type, cc = 0;
+       int ret;
+
+       if (!rd->show_driver_details || !tb)
+               return;
+
+       if (rd->pretty_output)
+               newline_indent(rd);
+
+       /*
+        * Driver attrs are tuples of {key, [print-type], value}.
+        * The key must be a string.  If print-type is present, it
+        * defines an alternate printf format type vs the native format
+        * for the attribute.  And the value can be any available
+        * driver type.
+        */
+       mnl_attr_for_each_nested(tb_entry, tb) {
+
+               if (cc > MAX_LINE_LENGTH) {
+                       if (rd->pretty_output)
+                               newline_indent(rd);
+                       cc = 0;
+               }
+               if (rd_attr_check(tb_entry, &type) != MNL_CB_OK)
+                       return;
+               if (!key) {
+                       if (type != MNL_TYPE_NUL_STRING)
+                               return;
+                       key = tb_entry;
+               } else if (type == MNL_TYPE_U8) {
+                       print_type = mnl_attr_get_u8(tb_entry);
+               } else {
+                       val = tb_entry;
+                       ret = print_driver_entry(rd, key, val, print_type);
+                       if (ret < 0)
+                               return;
+                       cc += ret;
+                       print_type = RDMA_NLDEV_PRINT_TYPE_UNSPEC;
+                       key = NULL;
+               }
+       }
+       return;
+}