utils.c

/*
 * src/cls-utils.c     Classifier Helpers
 *
 *      This library is free software; you can redistribute it and/or
 *      modify it under the terms of the GNU General Public License as
 *      published by the Free Software Foundation version 2 of the License.
 *
 * Copyright (c) 2008-2009 Thomas Graf <tgraf@suug.ch>
 */

#include "utils.h"

struct rtnl_cls *nlt_alloc_cls(void)
{
        struct rtnl_cls *cls;

        cls = rtnl_cls_alloc();
        if (!cls)
                fatal(ENOMEM, "Unable to allocate classifier object");

        return cls;
}

void parse_dev(struct rtnl_cls *cls, struct nl_cache *link_cache, char *arg)
{
        int ival;

        if (!(ival = rtnl_link_name2i(link_cache, arg)))
                fatal(ENOENT, "Link \"%s\" does not exist", arg);

        rtnl_cls_set_ifindex(cls, ival);
}
  
void parse_prio(struct rtnl_cls *cls, char *arg)
{
        uint32_t prio = parse_u32(arg);
        rtnl_cls_set_prio(cls, prio);
}

void parse_parent(struct rtnl_cls *cls, char *arg)
{
        uint32_t parent;
        int err;

        if ((err = rtnl_tc_str2handle(arg, &parent)) < 0)
                fatal(err, "Unable to parse handle \"%s\": %s",
                      arg, nl_geterror(err));

        rtnl_cls_set_parent(cls, parent);
}

void parse_handle(struct rtnl_cls *cls, char *arg)
{
        uint32_t handle;
        int err;

        if ((err = rtnl_tc_str2handle(arg, &handle)) < 0)
                fatal(err, "Unable to parse handle \"%s\": %s",
                      arg, nl_geterror(err));

        rtnl_cls_set_handle(cls, handle);
}

void parse_proto(struct rtnl_cls *cls, char *arg)
{
        int proto = nl_str2ether_proto(arg);
        if (proto < 0)
                fatal(proto, "Unable to parse protocol \"%s\": %s",
                      arg, nl_geterror(proto));
        rtnl_cls_set_protocol(cls, proto);
}

static NL_LIST_HEAD(cls_modules);

struct cls_module *lookup_cls_mod(struct rtnl_cls_ops *ops)
{
        struct cls_module *mod;

        nl_list_for_each_entry(mod, &cls_modules, list) {
                if (mod->ops == ops)
                        return mod;
        }

        return NULL;
}

void register_cls_module(struct cls_module *mod)
{
        struct rtnl_cls_ops *ops;

        if (!(ops = __rtnl_cls_lookup_ops(mod->name)))
                fatal(ENOENT, "Could not locate classifier module \"%s\"",
                        mod->name);

        if (lookup_cls_mod(ops) != NULL)
                fatal(EEXIST, "Duplicate classifier module registration.");

        mod->ops = ops;
        nl_list_add_tail(&mod->list, &cls_modules);
}

void unregister_cls_module(struct cls_module *mod)
{
        nl_list_del(&mod->list);
}