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
help(char * name)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
cmd_help(struct rd * rd)22 static int cmd_help(struct rd *rd)
23 {
24 help(rd->filename);
25 return 0;
26 }
27
rd_cmd(struct rd * rd)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
rd_init(struct rd * rd,int argc,char ** argv,char * filename)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
rd_free(struct rd * rd)73 static void rd_free(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
main(int argc,char ** argv)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_free(&rd);
142 return err ? EXIT_FAILURE : EXIT_SUCCESS;
143 }
144