src/nl-qdisc-add.c

   1 /*
   2  * src/nl-qdisc-dump.c     Dump qdisc attributes
   3  *
   4  *      This library is free software; you can redistribute it and/or
   5  *      modify it under the terms of the GNU Lesser General Public
   6  *      License as published by the Free Software Foundation version 2.1
   7  *      of the License.
   8  *
   9  * Copyright (c) 2003-2006 Thomas Graf <tgraf@suug.ch>
  10  */
  11
  12 #include "utils.h"
  13 #include <netlink/route/sch/fifo.h>
  14 #include <netlink/route/sch/prio.h>
  15
  16 static void print_usage(void)
  17 {
  18         printf(
  19 "Usage: nl-qdisc-add <ifindex> <handle> <parent> <kind>\n");
  20         exit(1);
  21 }
  22
  23 static int parse_blackhole_opts(struct rtnl_qdisc *qdisc, char *argv[],
  24                                 int argc)
  25 {
  26         return 0;
  27 }
  28
  29 static int parse_pfifo_opts(struct rtnl_qdisc *qdisc, char *argv[], int argc)
  30 {
  31         int err, limit;
  32
  33         if (argc > 0) {
  34                 if (argc != 2 || strcasecmp(argv[0], "limit")) {
  35                         fprintf(stderr, "Usage: ... pfifo limit <limit>\n");
  36                         return -1;
  37                 }
  38
  39                 limit = strtoul(argv[1], NULL, 0);
  40                 err = rtnl_qdisc_fifo_set_limit(qdisc, limit);
  41                 if (err < 0) {
  42                         fprintf(stderr, "%s\n", nl_geterror());
  43                         return -1;
  44                 }
  45         }
  46
  47         return 0;
  48 }
  49
  50 static int parse_bfifo_opts(struct rtnl_qdisc *qdisc, char *argv[], int argc)
  51 {
  52         int err, limit;
  53
  54         if (argc > 0) {
  55                 if (argc != 2 || strcasecmp(argv[0], "limit")) {
  56                         fprintf(stderr, "Usage: ... bfifo limit <limit>\n");
  57                         return -1;
  58                 }
  59
  60                 limit = nl_size2int(argv[1]);
  61                 if (limit < 0) {
  62                         fprintf(stderr, "Invalid value for limit.\n");
  63                         return -1;
  64                 }
  65
  66                 err = rtnl_qdisc_fifo_set_limit(qdisc, limit);
  67                 if (err < 0) {
  68                         fprintf(stderr, "%s\n", nl_geterror());
  69                         return -1;
  70                 }
  71         }
  72
  73         return 0;
  74 }
  75
  76 static int parse_prio_opts(struct rtnl_qdisc *qdisc, char *argv[], int argc)
  77 {
  78         int i, err, bands;
  79         uint8_t map[] = QDISC_PRIO_DEFAULT_PRIOMAP;
  80
  81         if (argc > 0) {
  82                 if (argc < 2 || strcasecmp(argv[0], "bands"))
  83                         goto usage;
  84
  85                 bands = strtoul(argv[1], NULL, 0);
  86                 err = rtnl_qdisc_prio_set_bands(qdisc, bands);
  87                 if (err < 0) {
  88                         fprintf(stderr, "%s\n", nl_geterror());
  89                         return -1;
  90                 }
  91         }
  92
  93         if (argc > 2) {
  94                 if (argc < 5 || strcasecmp(argv[2], "map"))
  95                         goto usage;
  96
  97                 for (i = 3; i < (argc & ~1U); i += 2) {
  98                         int prio, band;
  99
 100                         prio = rtnl_str2prio(argv[i]);
 101                         if (prio < 0 || prio > sizeof(map)/sizeof(map[0])) {
 102                                 fprintf(stderr, "Invalid priority \"%s\"\n",
 103                                         argv[i]);
 104                                 return -1;
 105                         }
 106
 107                         band = strtoul(argv[i+1], NULL, 0);
 108
 109                         map[prio] = band;
 110                 }
 111         }
 112
 113         err = rtnl_qdisc_prio_set_priomap(qdisc, map, sizeof(map));
 114         if (err < 0) {
 115                 fprintf(stderr, "%s\n", nl_geterror());
 116                 return -1;
 117         }
 118
 119         return 0;
 120 usage:
 121         fprintf(stderr, "Usage: ... prio bands <nbands> map MAP\n"
 122                         "MAP := <prio> <band>\n");
 123         return -1;
 124 }
 125
 126 int main(int argc, char *argv[])
 127 {
 128         struct nl_handle *nlh;
 129         struct rtnl_qdisc *qdisc;
 130         uint32_t handle, parent;
 131         int err = 1;
 132
 133         if (nltool_init(argc, argv) < 0)
 134                 return -1;
 135
 136         if (argc < 5 || !strcmp(argv[1], "-h"))
 137                 print_usage();
 138
 139         nlh = nl_handle_alloc_nondefault(nltool_cbset);
 140         if (!nlh)
 141                 goto errout;
 142
 143         qdisc = rtnl_qdisc_alloc();
 144         if (!qdisc)
 145                 goto errout_free_handle;
 146
 147         rtnl_qdisc_set_ifindex(qdisc, strtoul(argv[1], NULL, 0));
 148
 149         if (rtnl_tc_str2handle(argv[2], &handle) < 0) {
 150                 fprintf(stderr, "%s\n", nl_geterror());
 151                 goto errout_free_qdisc;
 152         }
 153
 154         if (rtnl_tc_str2handle(argv[3], &parent) < 0) {
 155                 fprintf(stderr, "%s\n", nl_geterror());
 156                 goto errout_free_qdisc;
 157         }
 158
 159         rtnl_qdisc_set_handle(qdisc, handle);
 160         rtnl_qdisc_set_parent(qdisc, parent);
 161         rtnl_qdisc_set_kind(qdisc, argv[4]);
 162
 163         if (!strcasecmp(argv[4], "blackhole"))
 164                 err = parse_blackhole_opts(qdisc, &argv[5], argc-5);
 165         else if (!strcasecmp(argv[4], "pfifo"))
 166                 err = parse_pfifo_opts(qdisc, &argv[5], argc-5);
 167         else if (!strcasecmp(argv[4], "bfifo"))
 168                 err = parse_bfifo_opts(qdisc, &argv[5], argc-5);
 169         else if (!strcasecmp(argv[4], "prio"))
 170                 err = parse_prio_opts(qdisc, &argv[5], argc-5);
 171         else {
 172                 fprintf(stderr, "Unknown qdisc \"%s\"\n", argv[4]);
 173                 goto errout_free_qdisc;
 174         }
 175
 176         if (err < 0)
 177                 goto errout_free_qdisc;
 178
 179         if (nltool_connect(nlh, NETLINK_ROUTE) < 0)
 180                 goto errout_free_qdisc;
 181
 182         if (rtnl_qdisc_add(nlh, qdisc, NLM_F_REPLACE) < 0) {
 183                 fprintf(stderr, "Unable to add Qdisc: %s\n", nl_geterror());
 184                 goto errout_close;
 185         }
 186
 187         err = 0;
 188 errout_close:
 189         nl_close(nlh);
 190 errout_free_qdisc:
 191         rtnl_qdisc_put(qdisc);
 192 errout_free_handle:
 193         nl_handle_destroy(nlh);
 194 errout:
 195         return err;
 196 }