From: Leon Romanovsky <[email protected]>

Signed-off-by: Leon Romanovsky <[email protected]>
---
 rdma/link.c  | 46 +---------------------------------------------
 rdma/rdma.h  |  1 +
 rdma/utils.c | 49 +++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 51 insertions(+), 45 deletions(-)

diff --git a/rdma/link.c b/rdma/link.c
index d9392289..676cb21d 100644
--- a/rdma/link.c
+++ b/rdma/link.c
@@ -285,51 +285,7 @@ static int link_one_show(struct rd *rd)
 
 static int link_show(struct rd *rd)
 {
-       struct dev_map *dev_map;
-       uint32_t port;
-       int ret = 0;
-
-       if (rd->json_output)
-               jsonw_start_array(rd->jw);
-       if (rd_no_arg(rd)) {
-               list_for_each_entry(dev_map, &rd->dev_map_list, list) {
-                       rd->dev_idx = dev_map->idx;
-                       for (port = 1; port < dev_map->num_ports + 1; port++) {
-                               rd->port_idx = port;
-                               ret = link_one_show(rd);
-                               if (ret)
-                                       goto out;
-                       }
-               }
-
-       } else {
-               dev_map = dev_map_lookup(rd, true);
-               port = get_port_from_argv(rd);
-               if (!dev_map || port > dev_map->num_ports) {
-                       pr_err("Wrong device name\n");
-                       ret = -ENOENT;
-                       goto out;
-               }
-               rd_arg_inc(rd);
-               rd->dev_idx = dev_map->idx;
-               rd->port_idx = port ? : 1;
-               for (; rd->port_idx < dev_map->num_ports + 1; rd->port_idx++) {
-                       ret = link_one_show(rd);
-                       if (ret)
-                               goto out;
-                       if (port)
-                               /*
-                                * We got request to show link for devname
-                                * with port index.
-                                */
-                               break;
-               }
-       }
-
-out:
-       if (rd->json_output)
-               jsonw_end_array(rd->jw);
-       return ret;
+       return rd_exec_link(rd, link_one_show);
 }
 
 int cmd_link(struct rd *rd)
diff --git a/rdma/rdma.h b/rdma/rdma.h
index afc0e413..8d53d3a0 100644
--- a/rdma/rdma.h
+++ b/rdma/rdma.h
@@ -73,6 +73,7 @@ int cmd_dev(struct rd *rd);
 int cmd_link(struct rd *rd);
 int rd_exec_cmd(struct rd *rd, const struct rd_cmd *c, const char *str);
 int rd_exec_dev(struct rd *rd, int (*cb)(struct rd *rd));
+int rd_exec_link(struct rd *rd, int (*cb)(struct rd *rd));
 void rd_free(struct rd *rd);
 
 /*
diff --git a/rdma/utils.c b/rdma/utils.c
index d7284801..7b2001e2 100644
--- a/rdma/utils.c
+++ b/rdma/utils.c
@@ -152,6 +152,55 @@ void rd_free(struct rd *rd)
        dev_map_cleanup(rd);
 }
 
+int rd_exec_link(struct rd *rd, int (*cb)(struct rd *rd))
+{
+       struct dev_map *dev_map;
+       uint32_t port;
+       int ret = 0;
+
+       if (rd->json_output)
+               jsonw_start_array(rd->jw);
+       if (rd_no_arg(rd)) {
+               list_for_each_entry(dev_map, &rd->dev_map_list, list) {
+                       rd->dev_idx = dev_map->idx;
+                       for (port = 1; port < dev_map->num_ports + 1; port++) {
+                               rd->port_idx = port;
+                               ret = cb(rd);
+                               if (ret)
+                                       goto out;
+                       }
+               }
+
+       } else {
+               dev_map = dev_map_lookup(rd, true);
+               port = get_port_from_argv(rd);
+               if (!dev_map || port > dev_map->num_ports) {
+                       pr_err("Wrong device name\n");
+                       ret = -ENOENT;
+                       goto out;
+               }
+               rd_arg_inc(rd);
+               rd->dev_idx = dev_map->idx;
+               rd->port_idx = port ? : 1;
+               for (; rd->port_idx < dev_map->num_ports + 1; rd->port_idx++) {
+                       ret = cb(rd);
+                       if (ret)
+                               goto out;
+                       if (port)
+                               /*
+                                * We got request to show link for devname
+                                * with port index.
+                                */
+                               break;
+               }
+       }
+
+out:
+       if (rd->json_output)
+               jsonw_end_array(rd->jw);
+       return ret;
+}
+
 int rd_exec_dev(struct rd *rd, int (*cb)(struct rd *rd))
 {
        struct dev_map *dev_map;
-- 
2.15.1

Reply via email to