This commit was generated by cvs2svn to compensate for changes in r786,

[libnl.git] / src / nl-qdisc-add.c

diff --git a/src/nl-qdisc-add.c b/src/nl-qdisc-add.c
new file mode 100644 (file)
index 0000000..0bf2939
--- /dev/null
+++ b/src/nl-qdisc-add.c
@@ -0,0 +1,196 @@
+/*
+ * src/nl-qdisc-dump.c     Dump qdisc attributes
+ *
+ *     This library 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 version 2.1
+ *     of the License.
+ *
+ * Copyright (c) 2003-2006 Thomas Graf <tgraf@suug.ch>
+ */
+
+#include "utils.h"
+#include <netlink/route/sch/fifo.h>
+#include <netlink/route/sch/prio.h>
+
+static void print_usage(void)
+{
+       printf(
+"Usage: nl-qdisc-add <ifindex> <handle> <parent> <kind>\n");
+       exit(1);
+}
+
+static int parse_blackhole_opts(struct rtnl_qdisc *qdisc, char *argv[],
+                               int argc)
+{
+       return 0;
+}
+
+static int parse_pfifo_opts(struct rtnl_qdisc *qdisc, char *argv[], int argc)
+{
+       int err, limit;
+
+       if (argc > 0) {
+               if (argc != 2 || strcasecmp(argv[0], "limit")) {
+                       fprintf(stderr, "Usage: ... pfifo limit <limit>\n");
+                       return -1;
+               }
+
+               limit = strtoul(argv[1], NULL, 0);
+               err = rtnl_qdisc_fifo_set_limit(qdisc, limit);
+               if (err < 0) {
+                       fprintf(stderr, "%s\n", nl_geterror());
+                       return -1;
+               }
+       }
+
+       return 0;
+}
+
+static int parse_bfifo_opts(struct rtnl_qdisc *qdisc, char *argv[], int argc)
+{
+       int err, limit;
+
+       if (argc > 0) {
+               if (argc != 2 || strcasecmp(argv[0], "limit")) {
+                       fprintf(stderr, "Usage: ... bfifo limit <limit>\n");
+                       return -1;
+               }
+
+               limit = nl_size2int(argv[1]);
+               if (limit < 0) {
+                       fprintf(stderr, "Invalid value for limit.\n");
+                       return -1;
+               }
+
+               err = rtnl_qdisc_fifo_set_limit(qdisc, limit);
+               if (err < 0) {
+                       fprintf(stderr, "%s\n", nl_geterror());
+                       return -1;
+               }
+       }
+
+       return 0;
+}
+
+static int parse_prio_opts(struct rtnl_qdisc *qdisc, char *argv[], int argc)
+{
+       int i, err, bands;
+       uint8_t map[] = QDISC_PRIO_DEFAULT_PRIOMAP;
+
+       if (argc > 0) {
+               if (argc < 2 || strcasecmp(argv[0], "bands"))
+                       goto usage;
+
+               bands = strtoul(argv[1], NULL, 0);
+               err = rtnl_qdisc_prio_set_bands(qdisc, bands);
+               if (err < 0) {
+                       fprintf(stderr, "%s\n", nl_geterror());
+                       return -1;
+               }
+       }
+
+       if (argc > 2) {
+               if (argc < 5 || strcasecmp(argv[2], "map"))
+                       goto usage;
+
+               for (i = 3; i < (argc & ~1U); i += 2) {
+                       int prio, band;
+
+                       prio = rtnl_str2prio(argv[i]);
+                       if (prio < 0 || prio > sizeof(map)/sizeof(map[0])) {
+                               fprintf(stderr, "Invalid priority \"%s\"\n",
+                                       argv[i]);
+                               return -1;
+                       }
+
+                       band = strtoul(argv[i+1], NULL, 0);
+
+                       map[prio] = band;
+               }
+       }
+
+       err = rtnl_qdisc_prio_set_priomap(qdisc, map, sizeof(map));
+       if (err < 0) {
+               fprintf(stderr, "%s\n", nl_geterror());
+               return -1;
+       }
+
+       return 0;
+usage:
+       fprintf(stderr, "Usage: ... prio bands <nbands> map MAP\n"
+                       "MAP := <prio> <band>\n");
+       return -1;
+}
+
+int main(int argc, char *argv[])
+{
+       struct nl_handle *nlh;
+       struct rtnl_qdisc *qdisc;
+       uint32_t handle, parent;
+       int err = 1;
+
+       if (nltool_init(argc, argv) < 0)
+               return -1;
+
+       if (argc < 5 || !strcmp(argv[1], "-h"))
+               print_usage();
+
+       nlh = nl_handle_alloc_nondefault(nltool_cbset);
+       if (!nlh)
+               goto errout;
+
+       qdisc = rtnl_qdisc_alloc();
+       if (!qdisc)
+               goto errout_free_handle;
+
+       rtnl_qdisc_set_ifindex(qdisc, strtoul(argv[1], NULL, 0));
+
+       if (rtnl_tc_str2handle(argv[2], &handle) < 0) {
+               fprintf(stderr, "%s\n", nl_geterror());
+               goto errout_free_qdisc;
+       }
+
+       if (rtnl_tc_str2handle(argv[3], &parent) < 0) {
+               fprintf(stderr, "%s\n", nl_geterror());
+               goto errout_free_qdisc;
+       }
+
+       rtnl_qdisc_set_handle(qdisc, handle);
+       rtnl_qdisc_set_parent(qdisc, parent);
+       rtnl_qdisc_set_kind(qdisc, argv[4]);
+
+       if (!strcasecmp(argv[4], "blackhole"))
+               err = parse_blackhole_opts(qdisc, &argv[5], argc-5);
+       else if (!strcasecmp(argv[4], "pfifo"))
+               err = parse_pfifo_opts(qdisc, &argv[5], argc-5);
+       else if (!strcasecmp(argv[4], "bfifo"))
+               err = parse_bfifo_opts(qdisc, &argv[5], argc-5);
+       else if (!strcasecmp(argv[4], "prio"))
+               err = parse_prio_opts(qdisc, &argv[5], argc-5);
+       else {
+               fprintf(stderr, "Unknown qdisc \"%s\"\n", argv[4]);
+               goto errout_free_qdisc;
+       }
+
+       if (err < 0)
+               goto errout_free_qdisc;
+
+       if (nltool_connect(nlh, NETLINK_ROUTE) < 0)
+               goto errout_free_qdisc;
+
+       if (rtnl_qdisc_add(nlh, qdisc, NLM_F_REPLACE) < 0) {
+               fprintf(stderr, "Unable to add Qdisc: %s\n", nl_geterror());
+               goto errout_close;
+       }
+
+       err = 0;
+errout_close:
+       nl_close(nlh);
+errout_free_qdisc:
+       rtnl_qdisc_put(qdisc);
+errout_free_handle:
+       nl_handle_destroy(nlh);
+errout:
+       return err;
+}