Whamcloud - gitweb
LU-7549 mdt: cancel the atomic lock in the same thread
[fs/lustre-release.git] / lnet / utils / lnetctl.c
1 /*
2  * LGPL HEADER START
3  *
4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU Lesser General Public License as
8  * published by the Free Software Foundation; either version 2.1 of the
9  * License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful, but
12  * WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library. If not, see <http://www.gnu.org/licenses/>.
18  *
19  * LGPL HEADER END
20  *
21  * Copyright (c) 2014, Intel Corporation.
22  *
23  * Author:
24  *   Amir Shehata <amir.shehata@intel.com>
25  */
26 #include <getopt.h>
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <libcfs/util/parser.h>
31 #include <lnet/lnetctl.h>
32 #include "cyaml/cyaml.h"
33 #include "lnetconfig/liblnetconfig.h"
34
35 #define LNET_CONFIGURE          true
36 #define LNET_UNCONFIGURE        false
37
38 static int jt_config_lnet(int argc, char **argv);
39 static int jt_unconfig_lnet(int argc, char **argv);
40 static int jt_add_route(int argc, char **argv);
41 static int jt_add_net(int argc, char **argv);
42 static int jt_set_routing(int argc, char **argv);
43 static int jt_del_route(int argc, char **argv);
44 static int jt_del_net(int argc, char **argv);
45 static int jt_show_route(int argc, char **argv);
46 static int jt_show_net(int argc, char **argv);
47 static int jt_show_routing(int argc, char **argv);
48 static int jt_show_stats(int argc, char **argv);
49 static int jt_show_peer_credits(int argc, char **argv);
50 static int jt_set_tiny(int argc, char **argv);
51 static int jt_set_small(int argc, char **argv);
52 static int jt_set_large(int argc, char **argv);
53
54 command_t lnet_cmds[] = {
55         {"configure", jt_config_lnet, 0, "configure lnet\n"
56          "\t--all: load NI configuration from module parameters\n"},
57         {"unconfigure", jt_unconfig_lnet, 0, "unconfigure lnet\n"},
58         { 0, 0, 0, NULL }
59 };
60
61 command_t route_cmds[] = {
62         {"add", jt_add_route, 0, "add a route\n"
63          "\t--net: net name (e.g. tcp0)\n"
64          "\t--gateway: gateway nid (e.g. 10.1.1.2@tcp)\n"
65          "\t--hop: number to final destination (1 < hops < 255)\n"
66          "\t--priority: priority of route (0 - highest prio\n"},
67         {"del", jt_del_route, 0, "delete a route\n"
68          "\t--net: net name (e.g. tcp0)\n"
69          "\t--gateway: gateway nid (e.g. 10.1.1.2@tcp)\n"},
70         {"show", jt_show_route, 0, "show routes\n"
71          "\t--net: net name (e.g. tcp0) to filter on\n"
72          "\t--gateway: gateway nid (e.g. 10.1.1.2@tcp) to filter on\n"
73          "\t--hop: number to final destination (1 < hops < 255) to filter on\n"
74          "\t--priority: priority of route (0 - highest prio to filter on\n"
75          "\t--verbose: display detailed output per route\n"},
76         { 0, 0, 0, NULL }
77 };
78
79 command_t net_cmds[] = {
80         {"add", jt_add_net, 0, "add a network\n"
81          "\t--net: net name (e.g. tcp0)\n"
82          "\t--if: physical interface (e.g. eth0)\n"
83          "\t--ip2net: specify networks based on IP address patterns\n"
84          "\t--peer-timeout: time to wait before declaring a peer dead\n"
85          "\t--peer-credits: define the max number of inflight messages\n"
86          "\t--peer-buffer-credits: the number of buffer credits per peer\n"
87          "\t--credits: Network Interface credits\n"
88          "\t--cpt: CPU Partitions configured net uses (e.g. [0,1]\n"},
89         {"del", jt_del_net, 0, "delete a network\n"
90          "\t--net: net name (e.g. tcp0)\n"},
91         {"show", jt_show_net, 0, "show networks\n"
92          "\t--net: net name (e.g. tcp0) to filter on\n"
93          "\t--verbose: display detailed output per network\n"},
94         { 0, 0, 0, NULL }
95 };
96
97 command_t routing_cmds[] = {
98         {"show", jt_show_routing, 0, "show routing information\n"},
99         { 0, 0, 0, NULL }
100 };
101
102 command_t stats_cmds[] = {
103         {"show", jt_show_stats, 0, "show LNET statistics\n"},
104         { 0, 0, 0, NULL }
105 };
106
107 command_t credits_cmds[] = {
108         {"show", jt_show_peer_credits, 0, "show peer credits\n"},
109         { 0, 0, 0, NULL }
110 };
111
112 command_t set_cmds[] = {
113         {"tiny_buffers", jt_set_tiny, 0, "set tiny routing buffers\n"
114          "\tVALUE must be greater than 0\n"},
115         {"small_buffers", jt_set_small, 0, "set small routing buffers\n"
116          "\tVALUE must be greater than 0\n"},
117         {"large_buffers", jt_set_large, 0, "set large routing buffers\n"
118          "\tVALUE must be greater than 0\n"},
119         {"routing", jt_set_routing, 0, "enable/disable routing\n"
120          "\t0 - disable routing\n"
121          "\t1 - enable routing\n"},
122         { 0, 0, 0, NULL }
123 };
124
125 static inline void print_help(const command_t cmds[], const char *cmd_type,
126                               const char *pc_name)
127 {
128         const command_t *cmd;
129
130         for (cmd = cmds; cmd->pc_name; cmd++) {
131                 if (pc_name != NULL &&
132                     strcmp(cmd->pc_name, pc_name) == 0) {
133                         printf("%s %s: %s\n", cmd_type, cmd->pc_name,
134                                cmd->pc_help);
135                         return;
136                 } else if (pc_name != NULL) {
137                         continue;
138                 }
139                 printf("%s %s: %s\n", cmd_type, cmd->pc_name, cmd->pc_help);
140         }
141 }
142
143 static int parse_long(const char *number, long int *value)
144 {
145         char *end;
146
147         *value = strtol(number,  &end, 0);
148         if (end != NULL && *end != 0)
149                 return -1;
150
151         return 0;
152 }
153
154 static int handle_help(const command_t *cmd_list, const char *cmd,
155                        const char *sub_cmd, int argc, char **argv)
156 {
157         int opt;
158         int rc = -1;
159         optind = 0;
160         opterr = 0;
161
162         const char *const short_options = "h";
163         const struct option long_options[] = {
164                 { "help", 0, NULL, 'h' },
165                 { NULL, 0, NULL, 0 },
166         };
167
168         while ((opt = getopt_long(argc, argv, short_options,
169                                   long_options, NULL)) != -1) {
170                 switch (opt) {
171                 case 'h':
172                         print_help(cmd_list, cmd, sub_cmd);
173                         rc = 0;
174                         break;
175                 default:
176                         rc = -1;
177                         break;
178                 }
179         }
180
181         opterr = 1;
182         optind = 0;
183         return rc;
184 }
185
186 static int jt_set_tiny(int argc, char **argv)
187 {
188         long int value;
189         int rc;
190         struct cYAML *err_rc = NULL;
191
192         if (handle_help(set_cmds, "set", "tiny_buffers", argc, argv) == 0)
193                 return 0;
194
195         rc = parse_long(argv[1], &value);
196         if (rc != 0) {
197                 cYAML_build_error(-1, -1, "parser", "set",
198                                   "cannot parse tiny_buffers value", &err_rc);
199                 cYAML_print_tree2file(stderr, err_rc);
200                 cYAML_free_tree(err_rc);
201                 return -1;
202         }
203
204         rc = lustre_lnet_config_buffers(value, -1, -1, -1, &err_rc);
205         if (rc != LUSTRE_CFG_RC_NO_ERR)
206                 cYAML_print_tree2file(stderr, err_rc);
207
208         cYAML_free_tree(err_rc);
209
210         return rc;
211 }
212
213 static int jt_set_small(int argc, char **argv)
214 {
215         long int value;
216         int rc;
217         struct cYAML *err_rc = NULL;
218
219         if (handle_help(set_cmds, "set", "small_buffers", argc, argv) == 0)
220                 return 0;
221
222         rc = parse_long(argv[1], &value);
223         if (rc != 0) {
224                 cYAML_build_error(-1, -1, "parser", "set",
225                                   "cannot parse small_buffers value", &err_rc);
226                 cYAML_print_tree2file(stderr, err_rc);
227                 cYAML_free_tree(err_rc);
228                 return -1;
229         }
230
231         rc = lustre_lnet_config_buffers(-1, value, -1, -1, &err_rc);
232         if (rc != LUSTRE_CFG_RC_NO_ERR)
233                 cYAML_print_tree2file(stderr, err_rc);
234
235         cYAML_free_tree(err_rc);
236
237         return rc;
238 }
239
240 static int jt_set_large(int argc, char **argv)
241 {
242         long int value;
243         int rc;
244         struct cYAML *err_rc = NULL;
245
246         if (handle_help(set_cmds, "set", "large_buffers", argc, argv) == 0)
247                 return 0;
248
249         rc = parse_long(argv[1], &value);
250         if (rc != 0) {
251                 cYAML_build_error(-1, -1, "parser", "set",
252                                   "cannot parse large_buffers value", &err_rc);
253                 cYAML_print_tree2file(stderr, err_rc);
254                 cYAML_free_tree(err_rc);
255                 return -1;
256         }
257
258         rc = lustre_lnet_config_buffers(-1, -1, value, -1, &err_rc);
259         if (rc != LUSTRE_CFG_RC_NO_ERR)
260                 cYAML_print_tree2file(stderr, err_rc);
261
262         cYAML_free_tree(err_rc);
263
264         return rc;
265 }
266
267 static int jt_set_routing(int argc, char **argv)
268 {
269         long int value;
270         struct cYAML *err_rc = NULL;
271         int rc;
272
273         if (handle_help(set_cmds, "set", "routing", argc, argv) == 0)
274                 return 0;
275
276         rc = parse_long(argv[1], &value);
277         if (rc != 0 || (value != 0 && value != 1)) {
278                 cYAML_build_error(-1, -1, "parser", "set",
279                                   "cannot parse routing value.\n"
280                                   "must be 0 for disable or 1 for enable",
281                                   &err_rc);
282                 cYAML_print_tree2file(stderr, err_rc);
283                 cYAML_free_tree(err_rc);
284                 return -1;
285         }
286
287         rc = lustre_lnet_enable_routing(value, -1, &err_rc);
288
289         if (rc != LUSTRE_CFG_RC_NO_ERR)
290                 cYAML_print_tree2file(stderr, err_rc);
291
292         cYAML_free_tree(err_rc);
293
294         return rc;
295 }
296
297 static int jt_config_lnet(int argc, char **argv)
298 {
299         struct cYAML *err_rc = NULL;
300         bool load_mod_params = false;
301         int rc, opt;
302
303         const char *const short_options = "ah";
304         const struct option long_options[] = {
305                 { "all", 0, NULL, 'a' },
306                 { "help", 0, NULL, 'h' },
307                 { NULL, 0, NULL, 0 },
308         };
309
310         while ((opt = getopt_long(argc, argv, short_options,
311                                    long_options, NULL)) != -1) {
312                 switch (opt) {
313                 case 'a':
314                         load_mod_params = true;
315                         break;
316                 case 'h':
317                         print_help(lnet_cmds, "lnet", "configure");
318                         return 0;
319                 default:
320                         return 0;
321                 }
322         }
323
324         rc = lustre_lnet_config_ni_system(LNET_CONFIGURE, load_mod_params,
325                                           -1, &err_rc);
326
327         if (rc != LUSTRE_CFG_RC_NO_ERR)
328                 cYAML_print_tree2file(stderr, err_rc);
329
330         cYAML_free_tree(err_rc);
331
332         return rc;
333 }
334
335 static int jt_unconfig_lnet(int argc, char **argv)
336 {
337         struct cYAML *err_rc = NULL;
338         int rc;
339
340         if (handle_help(lnet_cmds, "lnet", "unconfigure", argc, argv) == 0)
341                 return 0;
342
343         rc = lustre_lnet_config_ni_system(LNET_UNCONFIGURE, 0, -1, &err_rc);
344
345         if (rc != LUSTRE_CFG_RC_NO_ERR)
346                 cYAML_print_tree2file(stderr, err_rc);
347
348         cYAML_free_tree(err_rc);
349
350         return rc;
351 }
352 static int jt_add_route(int argc, char **argv)
353 {
354         char *network = NULL, *gateway = NULL;
355         long int hop = -1, prio = -1;
356         struct cYAML *err_rc = NULL;
357         int rc, opt;
358
359         const char *const short_options = "n:g:c:p:h";
360         const struct option long_options[] = {
361                 { "net", 1, NULL, 'n' },
362                 { "gateway", 1, NULL, 'g' },
363                 { "hop-count", 1, NULL, 'c' },
364                 { "priority", 1, NULL, 'p' },
365                 { "help", 0, NULL, 'h' },
366                 { NULL, 0, NULL, 0 },
367         };
368
369         while ((opt = getopt_long(argc, argv, short_options,
370                                    long_options, NULL)) != -1) {
371                 switch (opt) {
372                 case 'n':
373                         network = optarg;
374                         break;
375                 case 'g':
376                         gateway = optarg;
377                         break;
378                 case 'c':
379                         rc = parse_long(optarg, &hop);
380                         if (rc != 0) {
381                                 /* ignore option */
382                                 hop = -1;
383                                 continue;
384                         }
385                         break;
386                 case 'p':
387                         rc = parse_long(optarg, &prio);
388                         if (rc != 0) {
389                                 /* ingore option */
390                                 prio = -1;
391                                 continue;
392                         }
393                         break;
394                 case 'h':
395                         print_help(route_cmds, "route", "add");
396                         return 0;
397                 default:
398                         return 0;
399                 }
400         }
401
402         rc = lustre_lnet_config_route(network, gateway, hop, prio, -1, &err_rc);
403
404         if (rc != LUSTRE_CFG_RC_NO_ERR)
405                 cYAML_print_tree2file(stderr, err_rc);
406
407         cYAML_free_tree(err_rc);
408
409         return rc;
410 }
411
412 static int jt_add_net(int argc, char **argv)
413 {
414         char *network = NULL, *intf = NULL, *ip2net = NULL, *cpt = NULL;
415         long int pto = -1, pc = -1, pbc = -1, cre = -1;
416         struct cYAML *err_rc = NULL;
417         int rc, opt;
418
419         const char *const short_options = "n:i:p:t:c:b:r:s:h";
420         const struct option long_options[] = {
421                 { "net", 1, NULL, 'n' },
422                 { "if", 1, NULL, 'i' },
423                 { "ip2net", 1, NULL, 'p' },
424                 { "peer-timeout", 1, NULL, 't' },
425                 { "peer-credits", 1, NULL, 'c' },
426                 { "peer-buffer-credits", 1, NULL, 'b' },
427                 { "credits", 1, NULL, 'r' },
428                 { "cpt", 1, NULL, 's' },
429                 { "help", 0, NULL, 'h' },
430                 { NULL, 0, NULL, 0 },
431         };
432
433         while ((opt = getopt_long(argc, argv, short_options,
434                                    long_options, NULL)) != -1) {
435                 switch (opt) {
436                 case 'n':
437                         network = optarg;
438                         break;
439                 case 'i':
440                         intf = optarg;
441                         break;
442                 case 'p':
443                         ip2net = optarg;
444                         break;
445                 case 't':
446                         rc = parse_long(optarg, &pto);
447                         if (rc != 0) {
448                                 /* ignore option */
449                                 pto = -1;
450                                 continue;
451                         }
452                         break;
453                 case 'c':
454                         rc = parse_long(optarg, &pc);
455                         if (rc != 0) {
456                                 /* ignore option */
457                                 pc = -1;
458                                 continue;
459                         }
460                         break;
461                 case 'b':
462                         rc = parse_long(optarg, &pbc);
463                         if (rc != 0) {
464                                 /* ignore option */
465                                 pbc = -1;
466                                 continue;
467                         }
468                         break;
469                 case 'r':
470                         rc = parse_long(optarg, &cre);
471                         if (rc != 0) {
472                                 /* ignore option */
473                                 cre = -1;
474                                 continue;
475                         }
476                         break;
477                 case 's':
478                         cpt = optarg;
479                         break;
480                 case 'h':
481                         print_help(net_cmds, "net", "add");
482                         return 0;
483                 default:
484                         return 0;
485                 }
486         }
487
488         rc = lustre_lnet_config_net(network, intf, ip2net, pto, pc, pbc,
489                                     cre, cpt, -1, &err_rc);
490
491         if (rc != LUSTRE_CFG_RC_NO_ERR)
492                 cYAML_print_tree2file(stderr, err_rc);
493
494         cYAML_free_tree(err_rc);
495
496         return rc;
497 }
498
499 static int jt_del_route(int argc, char **argv)
500 {
501         char *network = NULL, *gateway = NULL;
502         struct cYAML *err_rc = NULL;
503         int rc, opt;
504
505         const char *const short_options = "n:g:h";
506         const struct option long_options[] = {
507                 { "net", 1, NULL, 'n' },
508                 { "gateway", 1, NULL, 'g' },
509                 { "help", 0, NULL, 'h' },
510                 { NULL, 0, NULL, 0 },
511         };
512
513         while ((opt = getopt_long(argc, argv, short_options,
514                                    long_options, NULL)) != -1) {
515                 switch (opt) {
516                 case 'n':
517                         network = optarg;
518                         break;
519                 case 'g':
520                         gateway = optarg;
521                         break;
522                 case 'h':
523                         print_help(route_cmds, "route", "del");
524                         return 0;
525                 default:
526                         return 0;
527                 }
528         }
529
530         rc = lustre_lnet_del_route(network, gateway, -1, &err_rc);
531
532         if (rc != LUSTRE_CFG_RC_NO_ERR)
533                 cYAML_print_tree2file(stderr, err_rc);
534
535         cYAML_free_tree(err_rc);
536
537         return rc;
538 }
539
540 static int jt_del_net(int argc, char **argv)
541 {
542         char *network = NULL;
543         struct cYAML *err_rc = NULL;
544         int rc, opt;
545
546         const char *const short_options = "n:h";
547         const struct option long_options[] = {
548                 { "net", 1, NULL, 'n' },
549                 { "help", 0, NULL, 'h' },
550                 { NULL, 0, NULL, 0 },
551         };
552
553         while ((opt = getopt_long(argc, argv, short_options,
554                                    long_options, NULL)) != -1) {
555                 switch (opt) {
556                 case 'n':
557                         network = optarg;
558                         break;
559                 case 'h':
560                         print_help(net_cmds, "net", "del");
561                         return 0;
562                 default:
563                         return 0;
564                 }
565         }
566
567         rc = lustre_lnet_del_net(network, -1, &err_rc);
568
569         if (rc != LUSTRE_CFG_RC_NO_ERR)
570                 cYAML_print_tree2file(stderr, err_rc);
571
572         cYAML_free_tree(err_rc);
573
574         return rc;
575 }
576
577 static int jt_show_route(int argc, char **argv)
578 {
579         char *network = NULL, *gateway = NULL;
580         long int hop = -1, prio = -1;
581         int detail = 0, rc, opt;
582         struct cYAML *err_rc = NULL, *show_rc = NULL;
583
584         const char *const short_options = "n:g:h:p:vh";
585         const struct option long_options[] = {
586                 { "net", 1, NULL, 'n' },
587                 { "gateway", 1, NULL, 'g' },
588                 { "hop-count", 1, NULL, 'c' },
589                 { "priority", 1, NULL, 'p' },
590                 { "verbose", 0, NULL, 'v' },
591                 { "help", 0, NULL, 'h' },
592                 { NULL, 0, NULL, 0 },
593         };
594
595         while ((opt = getopt_long(argc, argv, short_options,
596                                    long_options, NULL)) != -1) {
597                 switch (opt) {
598                 case 'n':
599                         network = optarg;
600                         break;
601                 case 'g':
602                         gateway = optarg;
603                         break;
604                 case 'c':
605                         rc = parse_long(optarg, &hop);
606                         if (rc != 0) {
607                                 /* ignore option */
608                                 hop = -1;
609                                 continue;
610                         }
611                         break;
612                 case 'p':
613                         rc = parse_long(optarg, &prio);
614                         if (rc != 0) {
615                                 /* ignore option */
616                                 prio = -1;
617                                 continue;
618                         }
619                         break;
620                 case 'v':
621                         detail = 1;
622                         break;
623                 case 'h':
624                         print_help(route_cmds, "route", "show");
625                         return 0;
626                 default:
627                         return 0;
628                 }
629         }
630
631         rc = lustre_lnet_show_route(network, gateway, hop, prio, detail, -1,
632                                     &show_rc, &err_rc);
633
634         if (rc != LUSTRE_CFG_RC_NO_ERR)
635                 cYAML_print_tree2file(stderr, err_rc);
636         else if (show_rc)
637                 cYAML_print_tree(show_rc);
638
639         cYAML_free_tree(err_rc);
640         cYAML_free_tree(show_rc);
641
642         return rc;
643 }
644
645 static int jt_show_net(int argc, char **argv)
646 {
647         char *network = NULL;
648         int detail = 0, rc, opt;
649         struct cYAML *err_rc = NULL, *show_rc = NULL;
650
651         const char *const short_options = "n:vh";
652         const struct option long_options[] = {
653                 { "net", 1, NULL, 'n' },
654                 { "verbose", 0, NULL, 'v' },
655                 { "help", 0, NULL, 'h' },
656                 { NULL, 0, NULL, 0 },
657         };
658
659         while ((opt = getopt_long(argc, argv, short_options,
660                                    long_options, NULL)) != -1) {
661                 switch (opt) {
662                 case 'n':
663                         network = optarg;
664                         break;
665                 case 'v':
666                         detail = 1;
667                         break;
668                 case 'h':
669                         print_help(net_cmds, "net", "show");
670                         return 0;
671                 default:
672                         return 0;
673                 }
674         }
675
676         rc = lustre_lnet_show_net(network, detail, -1, &show_rc, &err_rc);
677
678         if (rc != LUSTRE_CFG_RC_NO_ERR)
679                 cYAML_print_tree2file(stderr, err_rc);
680         else if (show_rc)
681                 cYAML_print_tree(show_rc);
682
683         cYAML_free_tree(err_rc);
684         cYAML_free_tree(show_rc);
685
686         return rc;
687 }
688
689 static int jt_show_routing(int argc, char **argv)
690 {
691         struct cYAML *err_rc = NULL, *show_rc = NULL;
692         int rc;
693
694         if (handle_help(routing_cmds, "routing", "show", argc, argv) == 0)
695                 return 0;
696
697         rc = lustre_lnet_show_routing(-1, &show_rc, &err_rc);
698
699         if (rc != LUSTRE_CFG_RC_NO_ERR)
700                 cYAML_print_tree2file(stderr, err_rc);
701         else if (show_rc)
702                 cYAML_print_tree(show_rc);
703
704         cYAML_free_tree(err_rc);
705         cYAML_free_tree(show_rc);
706
707         return rc;
708 }
709
710 static int jt_show_stats(int argc, char **argv)
711 {
712         int rc;
713         struct cYAML *show_rc = NULL, *err_rc = NULL;
714
715         if (handle_help(stats_cmds, "stats", "show", argc, argv) == 0)
716                 return 0;
717
718         rc = lustre_lnet_show_stats(-1, &show_rc, &err_rc);
719
720         if (rc != LUSTRE_CFG_RC_NO_ERR)
721                 cYAML_print_tree2file(stderr, err_rc);
722         else if (show_rc)
723                 cYAML_print_tree(show_rc);
724
725         cYAML_free_tree(err_rc);
726         cYAML_free_tree(show_rc);
727
728         return rc;
729 }
730
731 static int jt_show_peer_credits(int argc, char **argv)
732 {
733         int rc;
734         struct cYAML *show_rc = NULL, *err_rc = NULL;
735
736         if (handle_help(credits_cmds, "peer_credits", "show", argc, argv) == 0)
737                 return 0;
738
739         rc = lustre_lnet_show_peer_credits(-1, &show_rc, &err_rc);
740
741         if (rc != LUSTRE_CFG_RC_NO_ERR)
742                 cYAML_print_tree2file(stderr, err_rc);
743         else if (show_rc)
744                 cYAML_print_tree(show_rc);
745
746         cYAML_free_tree(err_rc);
747         cYAML_free_tree(show_rc);
748
749         return rc;
750 }
751
752 static inline int jt_lnet(int argc, char **argv)
753 {
754         if (argc < 2)
755                 return CMD_HELP;
756
757         if (argc == 2 &&
758             handle_help(lnet_cmds, "lnet", NULL, argc, argv) == 0)
759                 return 0;
760
761         return Parser_execarg(argc - 1, &argv[1], lnet_cmds);
762 }
763
764 static inline int jt_route(int argc, char **argv)
765 {
766         if (argc < 2)
767                 return CMD_HELP;
768
769         if (argc == 2 &&
770             handle_help(route_cmds, "route", NULL, argc, argv) == 0)
771                 return 0;
772
773         return Parser_execarg(argc - 1, &argv[1], route_cmds);
774 }
775
776 static inline int jt_net(int argc, char **argv)
777 {
778         if (argc < 2)
779                 return CMD_HELP;
780
781         if (argc == 2 &&
782             handle_help(net_cmds, "net", NULL, argc, argv) == 0)
783                 return 0;
784
785         return Parser_execarg(argc - 1, &argv[1], net_cmds);
786 }
787
788 static inline int jt_routing(int argc, char **argv)
789 {
790         if (argc < 2)
791                 return CMD_HELP;
792
793         if (argc == 2 &&
794             handle_help(routing_cmds, "routing", NULL, argc, argv) == 0)
795                 return 0;
796
797         return Parser_execarg(argc - 1, &argv[1], routing_cmds);
798 }
799
800 static inline int jt_stats(int argc, char **argv)
801 {
802         if (argc < 2)
803                 return CMD_HELP;
804
805         if (argc == 2 &&
806             handle_help(stats_cmds, "stats", NULL, argc, argv) == 0)
807                 return 0;
808
809         return Parser_execarg(argc - 1, &argv[1], stats_cmds);
810 }
811
812 static inline int jt_peer_credits(int argc, char **argv)
813 {
814         if (argc < 2)
815                 return CMD_HELP;
816
817         if (argc == 2 &&
818             handle_help(credits_cmds, "peer_credits", NULL, argc, argv) == 0)
819                 return 0;
820
821         return Parser_execarg(argc - 1, &argv[1], credits_cmds);
822 }
823
824 static inline int jt_set(int argc, char **argv)
825 {
826         if (argc < 2)
827                 return CMD_HELP;
828
829         if (argc == 2  &&
830             handle_help(set_cmds, "set", NULL, argc, argv) == 0)
831                 return 0;
832
833         return Parser_execarg(argc - 1, &argv[1], set_cmds);
834 }
835
836 static int jt_import(int argc, char **argv)
837 {
838         char *file = NULL;
839         struct cYAML *err_rc = NULL;
840         struct cYAML *show_rc = NULL;
841         int rc = 0, opt, opt_found = 0;
842         char cmd = 'a';
843
844         const char *const short_options = "adsh";
845         const struct option long_options[] = {
846                 { "add", 0, NULL, 'a' },
847                 { "del", 0, NULL, 'd' },
848                 { "show", 0, NULL, 's' },
849                 { "help", 0, NULL, 'h' },
850                 { NULL, 0, NULL, 0 },
851         };
852
853         while ((opt = getopt_long(argc, argv, short_options,
854                                    long_options, NULL)) != -1) {
855                 opt_found = 1;
856                 switch (opt) {
857                 case 'a':
858                 case 'd':
859                 case 's':
860                         cmd = opt;
861                         break;
862                 case 'h':
863                         printf("import FILE\n"
864                                "import < FILE : import a file\n"
865                                "\t--add: add configuration\n"
866                                "\t--del: delete configuration\n"
867                                "\t--show: show configuration\n"
868                                "\t--help: display this help\n"
869                                "If no command option is given then --add"
870                                " is assumed by default\n");
871                         return 0;
872                 default:
873                         return 0;
874                 }
875         }
876
877         /* grab the file name if one exists */
878         if (opt_found && argc == 3)
879                 file = argv[2];
880         else if (!opt_found && argc == 2)
881                 file = argv[1];
882
883         switch (cmd) {
884         case 'a':
885                 rc = lustre_yaml_config(file, &err_rc);
886                 break;
887         case 'd':
888                 rc = lustre_yaml_del(file, &err_rc);
889                 break;
890         case 's':
891                 rc = lustre_yaml_show(file, &show_rc, &err_rc);
892                 cYAML_print_tree(show_rc);
893                 cYAML_free_tree(show_rc);
894                 break;
895         }
896
897         if (rc != LUSTRE_CFG_RC_NO_ERR)
898                 cYAML_print_tree2file(stderr, err_rc);
899
900         cYAML_free_tree(err_rc);
901
902         return rc;
903 }
904
905 static int jt_export(int argc, char **argv)
906 {
907         struct cYAML *show_rc = NULL;
908         struct cYAML *err_rc = NULL;
909         int rc, opt;
910         FILE *f = NULL;
911
912         const char *const short_options = "h";
913         const struct option long_options[] = {
914                 { "help", 0, NULL, 'h' },
915                 { NULL, 0, NULL, 0 },
916         };
917
918         while ((opt = getopt_long(argc, argv, short_options,
919                                    long_options, NULL)) != -1) {
920                 switch (opt) {
921                 case 'h':
922                         printf("export FILE\n"
923                                "export > FILE : export configuration\n"
924                                "\t--help: display this help\n");
925                         return 0;
926                 default:
927                         return 0;
928                 }
929         }
930
931         if (argc >= 2) {
932                 f = fopen(argv[1], "w");
933                 if (f == NULL)
934                         return -1;
935         } else
936                 f = stdout;
937
938         rc = lustre_lnet_show_net(NULL, 1, -1, &show_rc, &err_rc);
939         if (rc != LUSTRE_CFG_RC_NO_ERR) {
940                 cYAML_print_tree2file(stderr, err_rc);
941                 cYAML_free_tree(err_rc);
942         }
943
944         rc = lustre_lnet_show_route(NULL, NULL, -1, -1, 1, -1, &show_rc,
945                                     &err_rc);
946         if (rc != LUSTRE_CFG_RC_NO_ERR) {
947                 cYAML_print_tree2file(stderr, err_rc);
948                 cYAML_free_tree(err_rc);
949         }
950
951         rc = lustre_lnet_show_routing(-1, &show_rc, &err_rc);
952         if (rc != LUSTRE_CFG_RC_NO_ERR) {
953                 cYAML_print_tree2file(stderr, err_rc);
954                 cYAML_free_tree(err_rc);
955         }
956
957         if (show_rc != NULL) {
958                 cYAML_print_tree2file(f, show_rc);
959                 cYAML_free_tree(show_rc);
960         }
961
962         if (argc >= 2)
963                 fclose(f);
964
965         return 0;
966 }
967
968 command_t list[] = {
969         {"lnet", jt_lnet, 0, "lnet {configure | unconfigure} [--all]"},
970         {"route", jt_route, 0, "route {add | del | show | help}"},
971         {"net", jt_net, 0, "net {add | del | show | help}"},
972         {"routing", jt_routing, 0, "routing {show | help}"},
973         {"set", jt_set, 0, "set {tiny_buffers | small_buffers | large_buffers"
974                            " | routing}"},
975         {"import", jt_import, 0, "import {--add | --del | --show | "
976                                  "--help} FILE.yaml"},
977         {"export", jt_export, 0, "export {--help} FILE.yaml"},
978         {"stats", jt_stats, 0, "stats {show | help}"},
979         {"peer_credits", jt_peer_credits, 0, "peer_credits {show | help}"},
980         {"help", Parser_help, 0, "help"},
981         {"exit", Parser_quit, 0, "quit"},
982         {"quit", Parser_quit, 0, "quit"},
983         { 0, 0, 0, NULL }
984 };
985
986 int main(int argc, char **argv)
987 {
988         int rc = 0;
989         struct cYAML *err_rc = NULL;
990
991         rc = lustre_lnet_config_lib_init();
992         if (rc < 0) {
993                 cYAML_build_error(-1, -1, "lnetctl", "startup",
994                                   "cannot register LNet device", &err_rc);
995                 cYAML_print_tree2file(stderr, err_rc);
996                 return rc;
997         }
998
999         Parser_init("lnetctl > ", list);
1000         if (argc > 1) {
1001                 rc = Parser_execarg(argc - 1, &argv[1], list);
1002                 goto errorout;
1003         }
1004
1005         Parser_commands();
1006
1007 errorout:
1008         return rc;
1009 }