rdma/rdma.c

   1 /*
   2  * rdma.c       RDMA tool
   3  *
   4  *              This program is free software; you can redistribute it and/or
   5  *              modify it under the terms of the GNU General Public License
   6  *              as published by the Free Software Foundation; either version
   7  *              2 of the License, or (at your option) any later version.
   8  *
   9  * Authors:     Leon Romanovsky <leonro@mellanox.com>
  10  */
  11
  12 #include "rdma.h"
  13 #include "SNAPSHOT.h"
  14
  15 static void help(char *name)
  16 {
  17         pr_out("Usage: %s [ OPTIONS ] OBJECT { COMMAND | help }\n"
  18                "where  OBJECT := { dev | link | help }\n"
  19                "       OPTIONS := { -V[ersion] | -d[etails] | -j[son] | -p[retty]}\n", name);
  20 }
  21
  22 static int cmd_help(struct rd *rd)
  23 {
  24         help(rd->filename);
  25         return 0;
  26 }
  27
  28 static int rd_cmd(struct rd *rd)
  29 {
  30         const struct rd_cmd cmds[] = {
  31                 { NULL,         cmd_help },
  32                 { "help",       cmd_help },
  33                 { "dev",        cmd_dev },
  34                 { "link",       cmd_link },
  35                 { 0 }
  36         };
  37
  38         return rd_exec_cmd(rd, cmds, "object");
  39 }
  40
  41 static int rd_init(struct rd *rd, int argc, char **argv, char *filename)
  42 {
  43         uint32_t seq;
  44         int ret;
  45
  46         rd->filename = filename;
  47         rd->argc = argc;
  48         rd->argv = argv;
  49         INIT_LIST_HEAD(&rd->dev_map_list);
  50
  51         if (rd->json_output) {
  52                 rd->jw = jsonw_new(stdout);
  53                 if (!rd->jw) {
  54                         pr_err("Failed to create JSON writer\n");
  55                         return -ENOMEM;
  56                 }
  57                 jsonw_pretty(rd->jw, rd->pretty_output);
  58         }
  59
  60         rd->buff = malloc(MNL_SOCKET_BUFFER_SIZE);
  61         if (!rd->buff)
  62                 return -ENOMEM;
  63
  64         rd_prepare_msg(rd, RDMA_NLDEV_CMD_GET,
  65                        &seq, (NLM_F_REQUEST | NLM_F_ACK | NLM_F_DUMP));
  66         ret = rd_send_msg(rd);
  67         if (ret)
  68                 return ret;
  69
  70         return rd_recv_msg(rd, rd_dev_init_cb, rd, seq);
  71 }
  72
  73 static void rd_cleanup(struct rd *rd)
  74 {
  75         if (rd->json_output)
  76                 jsonw_destroy(&rd->jw);
  77         free(rd->buff);
  78         rd_free_devmap(rd);
  79 }
  80
  81 int main(int argc, char **argv)
  82 {
  83         static const struct option long_options[] = {
  84                 { "version",            no_argument,            NULL, 'V' },
  85                 { "help",               no_argument,            NULL, 'h' },
  86                 { "json",               no_argument,            NULL, 'j' },
  87                 { "pretty",             no_argument,            NULL, 'p' },
  88                 { "details",            no_argument,            NULL, 'd' },
  89                 { NULL, 0, NULL, 0 }
  90         };
  91         bool pretty_output = false;
  92         bool show_details = false;
  93         bool json_output = false;
  94         char *filename;
  95         struct rd rd;
  96         int opt;
  97         int err;
  98
  99         filename = basename(argv[0]);
 100
 101         while ((opt = getopt_long(argc, argv, "Vhdpj",
 102                                   long_options, NULL)) >= 0) {
 103                 switch (opt) {
 104                 case 'V':
 105                         printf("%s utility, iproute2-ss%s\n",
 106                                filename, SNAPSHOT);
 107                         return EXIT_SUCCESS;
 108                 case 'p':
 109                         pretty_output = true;
 110                         break;
 111                 case 'd':
 112                         show_details = true;
 113                         break;
 114                 case 'j':
 115                         json_output = true;
 116                         break;
 117                 case 'h':
 118                         help(filename);
 119                         return EXIT_SUCCESS;
 120                 default:
 121                         pr_err("Unknown option.\n");
 122                         help(filename);
 123                         return EXIT_FAILURE;
 124                 }
 125         }
 126
 127         argc -= optind;
 128         argv += optind;
 129
 130         rd.show_details = show_details;
 131         rd.json_output = json_output;
 132         rd.pretty_output = pretty_output;
 133
 134         err = rd_init(&rd, argc, argv, filename);
 135         if (err)
 136                 goto out;
 137
 138         err = rd_cmd(&rd);
 139 out:
 140         /* Always cleanup */
 141         rd_cleanup(&rd);
 142         return err ? EXIT_FAILURE : EXIT_SUCCESS;
 143 }