--- /dev/null
+/*
+ * LGPL HEADER START
+ *
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of the
+ * License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library. If not, see <http://www.gnu.org/licenses/>.
+ *
+ * LGPL HEADER END
+ *
+ * Copyright (c) 2013, Intel Corporation.
+ *
+ * Author:
+ * Amir Shehata <amir.shehata@intel.com>
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <lnet/lnetctl.h>
+#include <libcfs/libcfsutil.h>
+#include "cyaml/cyaml.h"
+#include "lnetconfig/liblnetconfig.h"
+
+#define LNET_CONFIGURE true
+#define LNET_UNCONFIGURE false
+
+static int jt_config_lnet(int argc, char **argv);
+static int jt_unconfig_lnet(int argc, char **argv);
+static int jt_add_route(int argc, char **argv);
+static int jt_add_net(int argc, char **argv);
+static int jt_set_routing(int argc, char **argv);
+static int jt_del_route(int argc, char **argv);
+static int jt_del_net(int argc, char **argv);
+static int jt_show_route(int argc, char **argv);
+static int jt_show_net(int argc, char **argv);
+static int jt_show_routing(int argc, char **argv);
+static int jt_show_stats(int argc, char **argv);
+static int jt_show_peer_credits(int argc, char **argv);
+static int jt_set_tiny(int argc, char **argv);
+static int jt_set_small(int argc, char **argv);
+static int jt_set_large(int argc, char **argv);
+
+command_t lnet_cmds[] = {
+ {"configure", jt_config_lnet, 0, "configure lnet\n"
+ "\t--all: load NI configuration from module parameters\n"},
+ {"unconfigure", jt_unconfig_lnet, 0, "unconfigure lnet\n"},
+ { 0, 0, 0, NULL }
+};
+
+command_t route_cmds[] = {
+ {"add", jt_add_route, 0, "add a route\n"
+ "\t--net: net name (e.g. tcp0)\n"
+ "\t--gateway: gateway nid (e.g. 10.1.1.2@tcp)\n"
+ "\t--hop: number to final destination (1 < hops < 255)\n"
+ "\t--priority: priority of route (0 - highest prio\n"},
+ {"del", jt_del_route, 0, "delete a route\n"
+ "\t--net: net name (e.g. tcp0)\n"
+ "\t--gateway: gateway nid (e.g. 10.1.1.2@tcp)\n"},
+ {"show", jt_show_route, 0, "show routes\n"
+ "\t--net: net name (e.g. tcp0) to filter on\n"
+ "\t--gateway: gateway nid (e.g. 10.1.1.2@tcp) to filter on\n"
+ "\t--hop: number to final destination (1 < hops < 255) to filter on\n"
+ "\t--priority: priority of route (0 - highest prio to filter on\n"
+ "\t--vebose: display detailed output per route\n"},
+ { 0, 0, 0, NULL }
+};
+
+command_t net_cmds[] = {
+ {"add", jt_add_net, 0, "add a network\n"
+ "\t--net: net name (e.g. tcp0)\n"
+ "\t--if: physical interface (e.g. eth0)\n"
+ "\t--ip2net: specify networks based on IP address patterns\n"
+ "\t--peer-timeout: time to wait before declaring a peer dead\n"
+ "\t--peer-credits: define the max number of inflight messages\n"
+ "\t--peer-buffer-credits: the number of buffer credits per peer\n"
+ "\t--credits: Network Interface credits\n"
+ "\t--cpt: CPU Partitions configured net uses (e.g. [0,1]\n"},
+ {"del", jt_del_net, 0, "delete a network\n"
+ "\t--net: net name (e.g. tcp0)\n"},
+ {"show", jt_show_net, 0, "show networks\n"
+ "\t--net: net name (e.g. tcp0) to filter on\n"
+ "\t--verbose: display detailed output per network\n"},
+ { 0, 0, 0, NULL }
+};
+
+command_t routing_cmds[] = {
+ {"show", jt_show_routing, 0, "show routing information\n"},
+ { 0, 0, 0, NULL }
+};
+
+command_t stats_cmds[] = {
+ {"show", jt_show_stats, 0, "show LNET statistics\n"},
+ { 0, 0, 0, NULL }
+};
+
+command_t credits_cmds[] = {
+ {"show", jt_show_peer_credits, 0, "show peer credits\n"},
+ { 0, 0, 0, NULL }
+};
+
+command_t set_cmds[] = {
+ {"tiny_buffers", jt_set_tiny, 0, "set tiny routing buffers\n"
+ "\tVALUE must be greater than 0\n"},
+ {"small_buffers", jt_set_small, 0, "set small routing buffers\n"
+ "\tVALUE must be greater than 0\n"},
+ {"large_buffers", jt_set_large, 0, "set large routing buffers\n"
+ "\tVALUE must be greater than 0\n"},
+ {"routing", jt_set_routing, 0, "enable/disable routing\n"
+ "\t0 - disable routing\n"
+ "\t1 - enable routing\n"},
+ { 0, 0, 0, NULL }
+};
+
+static inline void print_help(const command_t cmds[], const char *cmd_type,
+ const char *pc_name)
+{
+ const command_t *cmd;
+
+ for (cmd = cmds; cmd->pc_name; cmd++) {
+ if (pc_name != NULL &&
+ strcmp(cmd->pc_name, pc_name) == 0) {
+ printf("%s %s: %s\n", cmd_type, cmd->pc_name,
+ cmd->pc_help);
+ return;
+ } else if (pc_name != NULL) {
+ continue;
+ }
+ printf("%s %s: %s\n", cmd_type, cmd->pc_name, cmd->pc_help);
+ }
+}
+
+static int parse_long(const char *number, long int *value)
+{
+ char *end;
+
+ *value = strtol(number, &end, 0);
+ if (end != NULL && *end != 0)
+ return -1;
+
+ return 0;
+}
+
+static int handle_help(const command_t *cmd_list, const char *cmd,
+ const char *sub_cmd, int argc, char **argv)
+{
+ int opt;
+ int rc = -1;
+ optind = 0;
+ opterr = 0;
+
+ const char *const short_options = "h";
+ const struct option long_options[] = {
+ { "help", 0, NULL, 'h' },
+ { NULL, 0, NULL, 0 },
+ };
+
+ while ((opt = getopt_long(argc, argv, short_options,
+ long_options, NULL)) != -1) {
+ switch (opt) {
+ case 'h':
+ print_help(cmd_list, cmd, sub_cmd);
+ rc = 0;
+ break;
+ default:
+ rc = -1;
+ break;
+ }
+ }
+
+ opterr = 1;
+ optind = 0;
+ return rc;
+}
+
+static int jt_set_tiny(int argc, char **argv)
+{
+ long int value;
+ int rc;
+ struct cYAML *err_rc = NULL;
+
+ if (handle_help(set_cmds, "set", "tiny_buffers", argc, argv) == 0)
+ return 0;
+
+ rc = parse_long(argv[1], &value);
+ if (rc != 0) {
+ cYAML_build_error(-1, -1, "parser", "set",
+ "cannot parse tiny_buffers value", &err_rc);
+ cYAML_print_tree2file(stderr, err_rc);
+ cYAML_free_tree(err_rc);
+ return -1;
+ }
+
+ rc = lustre_lnet_config_buffers(value, -1, -1, -1, &err_rc);
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+
+ cYAML_free_tree(err_rc);
+
+ return rc;
+}
+
+static int jt_set_small(int argc, char **argv)
+{
+ long int value;
+ int rc;
+ struct cYAML *err_rc = NULL;
+
+ if (handle_help(set_cmds, "set", "small_buffers", argc, argv) == 0)
+ return 0;
+
+ rc = parse_long(argv[1], &value);
+ if (rc != 0) {
+ cYAML_build_error(-1, -1, "parser", "set",
+ "cannot parse small_buffers value", &err_rc);
+ cYAML_print_tree2file(stderr, err_rc);
+ cYAML_free_tree(err_rc);
+ return -1;
+ }
+
+ rc = lustre_lnet_config_buffers(-1, value, -1, -1, &err_rc);
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+
+ cYAML_free_tree(err_rc);
+
+ return rc;
+}
+
+static int jt_set_large(int argc, char **argv)
+{
+ long int value;
+ int rc;
+ struct cYAML *err_rc = NULL;
+
+ if (handle_help(set_cmds, "set", "large_buffers", argc, argv) == 0)
+ return 0;
+
+ rc = parse_long(argv[1], &value);
+ if (rc != 0) {
+ cYAML_build_error(-1, -1, "parser", "set",
+ "cannot parse large_buffers value", &err_rc);
+ cYAML_print_tree2file(stderr, err_rc);
+ cYAML_free_tree(err_rc);
+ return -1;
+ }
+
+ rc = lustre_lnet_config_buffers(-1, -1, value, -1, &err_rc);
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+
+ cYAML_free_tree(err_rc);
+
+ return rc;
+}
+
+static int jt_set_routing(int argc, char **argv)
+{
+ long int value;
+ struct cYAML *err_rc = NULL;
+ int rc;
+
+ if (handle_help(set_cmds, "set", "routing", argc, argv) == 0)
+ return 0;
+
+ rc = parse_long(argv[1], &value);
+ if (rc != 0 || (value != 0 && value != 1)) {
+ cYAML_build_error(-1, -1, "parser", "set",
+ "cannot parse routing value.\n"
+ "must be 0 for disable or 1 for enable",
+ &err_rc);
+ cYAML_print_tree2file(stderr, err_rc);
+ cYAML_free_tree(err_rc);
+ return -1;
+ }
+
+ rc = lustre_lnet_enable_routing(value, -1, &err_rc);
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+
+ cYAML_free_tree(err_rc);
+
+ return rc;
+}
+
+static int jt_config_lnet(int argc, char **argv)
+{
+ struct cYAML *err_rc = NULL;
+ bool load_mod_params = false;
+ int rc, opt;
+ optind = 0;
+
+ const char *const short_options = "ah";
+ const struct option long_options[] = {
+ { "all", 0, NULL, 'a' },
+ { "help", 0, NULL, 'h' },
+ { NULL, 0, NULL, 0 },
+ };
+
+ while ((opt = getopt_long(argc, argv, short_options,
+ long_options, NULL)) != -1) {
+ switch (opt) {
+ case 'a':
+ load_mod_params = true;
+ break;
+ case 'h':
+ print_help(lnet_cmds, "lnet", "configure");
+ return 0;
+ default:
+ return 0;
+ }
+ }
+
+ rc = lustre_lnet_config_ni_system(LNET_CONFIGURE, load_mod_params,
+ -1, &err_rc);
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+
+ cYAML_free_tree(err_rc);
+
+ return rc;
+}
+
+static int jt_unconfig_lnet(int argc, char **argv)
+{
+ struct cYAML *err_rc = NULL;
+ int rc;
+ optind = 0;
+
+ if (handle_help(lnet_cmds, "lnet", "unconfigure", argc, argv) == 0)
+ return 0;
+
+ rc = lustre_lnet_config_ni_system(LNET_UNCONFIGURE, 0, -1, &err_rc);
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+
+ cYAML_free_tree(err_rc);
+
+ return rc;
+}
+static int jt_add_route(int argc, char **argv)
+{
+ char *network = NULL, *gateway = NULL;
+ long int hop = -1, prio = -1;
+ struct cYAML *err_rc = NULL;
+ int rc, opt;
+ optind = 0;
+
+ const char *const short_options = "n:g:c:p:h";
+ const struct option long_options[] = {
+ { "net", 1, NULL, 'n' },
+ { "gateway", 1, NULL, 'g' },
+ { "hop-count", 1, NULL, 'c' },
+ { "priority", 1, NULL, 'p' },
+ { "help", 0, NULL, 'h' },
+ { NULL, 0, NULL, 0 },
+ };
+
+ while ((opt = getopt_long(argc, argv, short_options,
+ long_options, NULL)) != -1) {
+ switch (opt) {
+ case 'n':
+ network = optarg;
+ break;
+ case 'g':
+ gateway = optarg;
+ break;
+ case 'c':
+ rc = parse_long(optarg, &hop);
+ if (rc != 0) {
+ /* ignore option */
+ hop = -1;
+ continue;
+ }
+ break;
+ case 'p':
+ rc = parse_long(optarg, &prio);
+ if (rc != 0) {
+ /* ingore option */
+ prio = -1;
+ continue;
+ }
+ break;
+ case 'h':
+ print_help(route_cmds, "route", "add");
+ return 0;
+ default:
+ return 0;
+ }
+ }
+
+ rc = lustre_lnet_config_route(network, gateway, hop, prio, -1, &err_rc);
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+
+ cYAML_free_tree(err_rc);
+
+ return rc;
+}
+
+static int jt_add_net(int argc, char **argv)
+{
+ char *network = NULL, *intf = NULL, *ip2net = NULL, *cpt = NULL;
+ long int pto = -1, pc = -1, pbc = -1, cre = -1;
+ struct cYAML *err_rc = NULL;
+ int rc, opt;
+ optind = 0;
+
+ const char *const short_options = "n:i:p:t:c:b:r:s:h";
+ const struct option long_options[] = {
+ { "net", 1, NULL, 'n' },
+ { "if", 1, NULL, 'i' },
+ { "ip2net", 1, NULL, 'p' },
+ { "peer-timeout", 1, NULL, 't' },
+ { "peer-credits", 1, NULL, 'c' },
+ { "peer-buffer-credits", 1, NULL, 'b' },
+ { "credits", 1, NULL, 'r' },
+ { "cpt", 1, NULL, 's' },
+ { "help", 0, NULL, 'h' },
+ { NULL, 0, NULL, 0 },
+ };
+
+ while ((opt = getopt_long(argc, argv, short_options,
+ long_options, NULL)) != -1) {
+ switch (opt) {
+ case 'n':
+ network = optarg;
+ break;
+ case 'i':
+ intf = optarg;
+ break;
+ case 'p':
+ ip2net = optarg;
+ break;
+ case 't':
+ rc = parse_long(optarg, &pto);
+ if (rc != 0) {
+ /* ignore option */
+ pto = -1;
+ continue;
+ }
+ break;
+ case 'c':
+ rc = parse_long(optarg, &pc);
+ if (rc != 0) {
+ /* ignore option */
+ pc = -1;
+ continue;
+ }
+ break;
+ case 'b':
+ rc = parse_long(optarg, &pbc);
+ if (rc != 0) {
+ /* ignore option */
+ pbc = -1;
+ continue;
+ }
+ break;
+ case 'r':
+ rc = parse_long(optarg, &cre);
+ if (rc != 0) {
+ /* ignore option */
+ cre = -1;
+ continue;
+ }
+ break;
+ case 's':
+ cpt = optarg;
+ break;
+ case 'h':
+ print_help(net_cmds, "net", "add");
+ return 0;
+ default:
+ return 0;
+ }
+ }
+
+ rc = lustre_lnet_config_net(network, intf, ip2net, pto, pc, pbc,
+ cre, cpt, -1, &err_rc);
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+
+ cYAML_free_tree(err_rc);
+
+ return rc;
+}
+
+static int jt_del_route(int argc, char **argv)
+{
+ char *network = NULL, *gateway = NULL;
+ struct cYAML *err_rc = NULL;
+ int rc, opt;
+ optind = 0;
+
+ const char *const short_options = "n:g:h";
+ const struct option long_options[] = {
+ { "net", 1, NULL, 'n' },
+ { "gateway", 1, NULL, 'g' },
+ { "help", 0, NULL, 'h' },
+ { NULL, 0, NULL, 0 },
+ };
+
+ while ((opt = getopt_long(argc, argv, short_options,
+ long_options, NULL)) != -1) {
+ switch (opt) {
+ case 'n':
+ network = optarg;
+ break;
+ case 'g':
+ gateway = optarg;
+ break;
+ case 'h':
+ print_help(route_cmds, "route", "del");
+ return 0;
+ default:
+ return 0;
+ }
+ }
+
+ rc = lustre_lnet_del_route(network, gateway, -1, &err_rc);
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+
+ cYAML_free_tree(err_rc);
+
+ return rc;
+}
+
+static int jt_del_net(int argc, char **argv)
+{
+ char *network = NULL;
+ struct cYAML *err_rc = NULL;
+ int rc, opt;
+ optind = 0;
+
+ const char *const short_options = "n:h";
+ const struct option long_options[] = {
+ { "net", 1, NULL, 'n' },
+ { "help", 0, NULL, 'h' },
+ { NULL, 0, NULL, 0 },
+ };
+
+ while ((opt = getopt_long(argc, argv, short_options,
+ long_options, NULL)) != -1) {
+ switch (opt) {
+ case 'n':
+ network = optarg;
+ break;
+ case 'h':
+ print_help(net_cmds, "net", "del");
+ return 0;
+ default:
+ return 0;
+ }
+ }
+
+ rc = lustre_lnet_del_net(network, -1, &err_rc);
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+
+ cYAML_free_tree(err_rc);
+
+ return rc;
+}
+
+static int jt_show_route(int argc, char **argv)
+{
+ char *network = NULL, *gateway = NULL;
+ long int hop = -1, prio = -1;
+ int detail = 0, rc, opt;
+ struct cYAML *err_rc = NULL, *show_rc = NULL;
+ optind = 0;
+
+ const char *const short_options = "n:g:h:p:vh";
+ const struct option long_options[] = {
+ { "net", 1, NULL, 'n' },
+ { "gateway", 1, NULL, 'g' },
+ { "hop-count", 1, NULL, 'c' },
+ { "priority", 1, NULL, 'p' },
+ { "verbose", 0, NULL, 'v' },
+ { "help", 0, NULL, 'h' },
+ { NULL, 0, NULL, 0 },
+ };
+
+ while ((opt = getopt_long(argc, argv, short_options,
+ long_options, NULL)) != -1) {
+ switch (opt) {
+ case 'n':
+ network = optarg;
+ break;
+ case 'g':
+ gateway = optarg;
+ break;
+ case 'c':
+ rc = parse_long(optarg, &hop);
+ if (rc != 0) {
+ /* ignore option */
+ hop = -1;
+ continue;
+ }
+ break;
+ case 'p':
+ rc = parse_long(optarg, &prio);
+ if (rc != 0) {
+ /* ignore option */
+ prio = -1;
+ continue;
+ }
+ break;
+ case 'v':
+ detail = 1;
+ break;
+ case 'h':
+ print_help(route_cmds, "route", "show");
+ return 0;
+ default:
+ return 0;
+ }
+ }
+
+ rc = lustre_lnet_show_route(network, gateway, hop, prio, detail, -1,
+ &show_rc, &err_rc);
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+ else if (show_rc)
+ cYAML_print_tree(show_rc);
+
+ cYAML_free_tree(err_rc);
+ cYAML_free_tree(show_rc);
+
+ return rc;
+}
+
+static int jt_show_net(int argc, char **argv)
+{
+ char *network = NULL;
+ int detail = 0, rc, opt;
+ struct cYAML *err_rc = NULL, *show_rc = NULL;
+ optind = 0;
+
+ const char *const short_options = "n:vh";
+ const struct option long_options[] = {
+ { "net", 1, NULL, 'n' },
+ { "verbose", 0, NULL, 'v' },
+ { "help", 0, NULL, 'h' },
+ { NULL, 0, NULL, 0 },
+ };
+
+ while ((opt = getopt_long(argc, argv, short_options,
+ long_options, NULL)) != -1) {
+ switch (opt) {
+ case 'n':
+ network = optarg;
+ break;
+ case 'v':
+ detail = 1;
+ break;
+ case 'h':
+ print_help(net_cmds, "net", "show");
+ return 0;
+ default:
+ return 0;
+ }
+ }
+
+ rc = lustre_lnet_show_net(network, detail, -1, &show_rc, &err_rc);
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+ else if (show_rc)
+ cYAML_print_tree(show_rc);
+
+ cYAML_free_tree(err_rc);
+ cYAML_free_tree(show_rc);
+
+ return rc;
+}
+
+static int jt_show_routing(int argc, char **argv)
+{
+ struct cYAML *err_rc = NULL, *show_rc = NULL;
+ int rc;
+
+ if (handle_help(routing_cmds, "routing", "show", argc, argv) == 0)
+ return 0;
+
+ rc = lustre_lnet_show_routing(-1, &show_rc, &err_rc);
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+ else if (show_rc)
+ cYAML_print_tree(show_rc);
+
+ cYAML_free_tree(err_rc);
+ cYAML_free_tree(show_rc);
+
+ return rc;
+}
+
+static int jt_show_stats(int argc, char **argv)
+{
+ int rc;
+ struct cYAML *show_rc = NULL, *err_rc = NULL;
+
+ if (handle_help(stats_cmds, "stats", "show", argc, argv) == 0)
+ return 0;
+
+ rc = lustre_lnet_show_stats(-1, &show_rc, &err_rc);
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+ else if (show_rc)
+ cYAML_print_tree(show_rc);
+
+ cYAML_free_tree(err_rc);
+ cYAML_free_tree(show_rc);
+
+ return rc;
+}
+
+static int jt_show_peer_credits(int argc, char **argv)
+{
+ int rc;
+ struct cYAML *show_rc = NULL, *err_rc = NULL;
+ optind = 0;
+
+ if (handle_help(credits_cmds, "peer_credits", "show", argc, argv) == 0)
+ return 0;
+
+ rc = lustre_lnet_show_peer_credits(-1, &show_rc, &err_rc);
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+ else if (show_rc)
+ cYAML_print_tree(show_rc);
+
+ cYAML_free_tree(err_rc);
+ cYAML_free_tree(show_rc);
+
+ return rc;
+}
+
+static inline int jt_lnet(int argc, char **argv)
+{
+ if (argc < 2)
+ return CMD_HELP;
+
+ if (argc == 2 &&
+ handle_help(lnet_cmds, "lnet", NULL, argc, argv) == 0)
+ return 0;
+
+ return Parser_execarg(argc - 1, &argv[1], lnet_cmds);
+}
+
+static inline int jt_route(int argc, char **argv)
+{
+ if (argc < 2)
+ return CMD_HELP;
+
+ if (argc == 2 &&
+ handle_help(route_cmds, "route", NULL, argc, argv) == 0)
+ return 0;
+
+ return Parser_execarg(argc - 1, &argv[1], route_cmds);
+}
+
+static inline int jt_net(int argc, char **argv)
+{
+ if (argc < 2)
+ return CMD_HELP;
+
+ if (argc == 2 &&
+ handle_help(net_cmds, "net", NULL, argc, argv) == 0)
+ return 0;
+
+ return Parser_execarg(argc - 1, &argv[1], net_cmds);
+}
+
+static inline int jt_routing(int argc, char **argv)
+{
+ if (argc < 2)
+ return CMD_HELP;
+
+ if (argc == 2 &&
+ handle_help(routing_cmds, "routing", NULL, argc, argv) == 0)
+ return 0;
+
+ return Parser_execarg(argc - 1, &argv[1], routing_cmds);
+}
+
+static inline int jt_stats(int argc, char **argv)
+{
+ if (argc < 2)
+ return CMD_HELP;
+
+ if (argc == 2 &&
+ handle_help(stats_cmds, "stats", NULL, argc, argv) == 0)
+ return 0;
+
+ return Parser_execarg(argc - 1, &argv[1], stats_cmds);
+}
+
+static inline int jt_peer_credits(int argc, char **argv)
+{
+ if (argc < 2)
+ return CMD_HELP;
+
+ if (argc == 2 &&
+ handle_help(credits_cmds, "peer_credits", NULL, argc, argv) == 0)
+ return 0;
+
+ return Parser_execarg(argc - 1, &argv[1], credits_cmds);
+}
+
+static inline int jt_set(int argc, char **argv)
+{
+ if (argc < 2)
+ return CMD_HELP;
+
+ if (argc == 2 &&
+ handle_help(set_cmds, "set", NULL, argc, argv) == 0)
+ return 0;
+
+ return Parser_execarg(argc - 1, &argv[1], set_cmds);
+}
+
+static int jt_import(int argc, char **argv)
+{
+ char *file = NULL;
+ struct cYAML *err_rc = NULL;
+ struct cYAML *show_rc = NULL;
+ int rc = 0, opt, opt_found = 0;
+ optind = 0;
+ char cmd = 'a';
+
+ const char *const short_options = "adsh";
+ const struct option long_options[] = {
+ { "add", 0, NULL, 'a' },
+ { "del", 0, NULL, 'd' },
+ { "show", 0, NULL, 's' },
+ { "help", 0, NULL, 'h' },
+ { NULL, 0, NULL, 0 },
+ };
+
+ while ((opt = getopt_long(argc, argv, short_options,
+ long_options, NULL)) != -1) {
+ opt_found = 1;
+ switch (opt) {
+ case 'a':
+ case 'd':
+ case 's':
+ cmd = opt;
+ break;
+ case 'h':
+ printf("import FILE\n"
+ "import < FILE : import a file\n"
+ "\t--add: add configuration\n"
+ "\t--del: delete configuration\n"
+ "\t--show: show configuration\n"
+ "\t--help: display this help\n"
+ "If no command option is given then --add"
+ " is assumed by default\n");
+ return 0;
+ default:
+ return 0;
+ }
+ }
+
+ /* grab the file name if one exists */
+ if (opt_found && argc == 3)
+ file = argv[2];
+ else if (!opt_found && argc == 2)
+ file = argv[1];
+
+ switch (cmd) {
+ case 'a':
+ rc = lustre_yaml_config(file, &err_rc);
+ break;
+ case 'd':
+ rc = lustre_yaml_del(file, &err_rc);
+ break;
+ case 's':
+ rc = lustre_yaml_show(file, &show_rc, &err_rc);
+ cYAML_print_tree(show_rc);
+ cYAML_free_tree(show_rc);
+ break;
+ }
+
+ if (rc != LUSTRE_CFG_RC_NO_ERR)
+ cYAML_print_tree2file(stderr, err_rc);
+
+ cYAML_free_tree(err_rc);
+
+ return rc;
+}
+
+static int jt_export(int argc, char **argv)
+{
+ struct cYAML *show_rc = NULL;
+ struct cYAML *err_rc = NULL;
+ int rc, opt;
+ FILE *f = NULL;
+ optind = 0;
+
+ const char *const short_options = "h";
+ const struct option long_options[] = {
+ { "help", 0, NULL, 'h' },
+ { NULL, 0, NULL, 0 },
+ };
+
+ while ((opt = getopt_long(argc, argv, short_options,
+ long_options, NULL)) != -1) {
+ switch (opt) {
+ case 'h':
+ printf("export FILE\n"
+ "export > FILE : export configuration\n"
+ "\t--help: display this help\n");
+ return 0;
+ default:
+ return 0;
+ }
+ }
+
+ if (argc >= 2) {
+ f = fopen(argv[1], "w");
+ if (f == NULL)
+ return -1;
+ } else
+ f = stdout;
+
+ rc = lustre_lnet_show_net(NULL, 1, -1, &show_rc, &err_rc);
+ if (rc != LUSTRE_CFG_RC_NO_ERR) {
+ cYAML_print_tree2file(stderr, err_rc);
+ cYAML_free_tree(err_rc);
+ }
+
+ rc = lustre_lnet_show_route(NULL, NULL, -1, -1, 1, -1, &show_rc,
+ &err_rc);
+ if (rc != LUSTRE_CFG_RC_NO_ERR) {
+ cYAML_print_tree2file(stderr, err_rc);
+ cYAML_free_tree(err_rc);
+ }
+
+ rc = lustre_lnet_show_routing(-1, &show_rc, &err_rc);
+ if (rc != LUSTRE_CFG_RC_NO_ERR) {
+ cYAML_print_tree2file(stderr, err_rc);
+ cYAML_free_tree(err_rc);
+ }
+
+ if (show_rc != NULL) {
+ cYAML_print_tree2file(f, show_rc);
+ cYAML_free_tree(show_rc);
+ }
+
+ if (argc >= 2)
+ fclose(f);
+
+ return 0;
+}
+
+command_t list[] = {
+ {"lnet", jt_lnet, 0, "lnet {configure | unconfigure} [--all]"},
+ {"route", jt_route, 0, "route {add | del | show | help}"},
+ {"net", jt_net, 0, "net {add | del | show | help}"},
+ {"routing", jt_routing, 0, "routing {show | help}"},
+ {"set", jt_set, 0, "set {tiny_buffers | small_buffers | large_buffers"
+ " | routing}"},
+ {"import", jt_import, 0, "import {--add | --del | --show | "
+ "--help} FILE.yaml"},
+ {"export", jt_export, 0, "export {--help} FILE.yaml"},
+ {"stats", jt_stats, 0, "stats {show | help}"},
+ {"peer_credits", jt_peer_credits, 0, "stats {show | help}"},
+ {"help", Parser_help, 0, "help"},
+ {"exit", Parser_quit, 0, "quit"},
+ {"quit", Parser_quit, 0, "quit"},
+ { 0, 0, 0, NULL }
+};
+
+int main(int argc, char **argv)
+{
+ int rc = 0;
+ struct cYAML *err_rc = NULL;
+
+ rc = lustre_lnet_config_lib_init();
+ if (rc < 0) {
+ cYAML_build_error(-1, -1, "lnetctl", "startup",
+ "cannot register LNet device", &err_rc);
+ cYAML_print_tree2file(stderr, err_rc);
+ return rc;
+ }
+
+ rc = libcfs_arch_init();
+ if (rc < 0) {
+ cYAML_build_error(-1, -1, "lnetctl", "startup",
+ "cannot initialize libcfs", &err_rc);
+ cYAML_print_tree2file(stderr, err_rc);
+ return rc;
+ }
+
+ Parser_init("lnetctl > ", list);
+ if (argc > 1) {
+ rc = Parser_execarg(argc - 1, &argv[1], list);
+ goto errorout;
+ }
+
+ Parser_commands();
+
+errorout:
+ libcfs_arch_cleanup();
+ return rc;
+}