nexmon – Rev 1

?pathlinks?

Blame – Last modification – View Log – Download

/*
 * 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_sock *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 = nltool_alloc_handle();
        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;
}