src/lib/qdisc.c

/*
 * src/lib/qdisc.c     CLI QDisc Helpers
 *
 *	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) 2008-2010 Thomas Graf <tgraf@suug.ch>
 */

/**
 * @ingroup cli
 * @defgroup cli_qdisc Queueing Disciplines
 * @{
 */

#include <netlink/cli/utils.h>
#include <netlink/cli/qdisc.h>
#include <netlink/route/class.h>
#include <netlink/route/class-modules.h>

struct rtnl_qdisc *nl_cli_qdisc_alloc(void)
{
	struct rtnl_qdisc *qdisc;

	qdisc = rtnl_qdisc_alloc();
	if (!qdisc)
		nl_cli_fatal(ENOMEM, "Unable to allocate qdisc object");

	return qdisc;
}

void nl_cli_qdisc_parse_dev(struct rtnl_qdisc *qdisc, struct nl_cache *link_cache, char *arg)
{
	int ival;

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

	rtnl_qdisc_set_ifindex(qdisc, ival);
}

void nl_cli_qdisc_parse_parent(struct rtnl_qdisc *qdisc, char *arg)
{
	uint32_t parent;
	int err;

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

	rtnl_qdisc_set_parent(qdisc, parent);
}

void nl_cli_qdisc_parse_handle(struct rtnl_qdisc *qdisc, char *arg)
{
	uint32_t handle;
	int err;

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

	rtnl_qdisc_set_handle(qdisc, handle);
}

void nl_cli_qdisc_parse_kind(struct rtnl_qdisc *qdisc, char *arg)
{
	rtnl_qdisc_set_kind(qdisc, arg);
}

static NL_LIST_HEAD(qdisc_modules);

struct nl_cli_qdisc_module *__nl_cli_qdisc_lookup(struct rtnl_qdisc_ops *ops)
{
	struct nl_cli_qdisc_module *qm;

	nl_list_for_each_entry(qm, &qdisc_modules, qm_list)
		if (qm->qm_ops == ops)
			return qm;

	return NULL;
}

struct nl_cli_qdisc_module *__nl_cli_class_lookup(struct rtnl_class_ops *ops)
{
	struct nl_cli_qdisc_module *qm;

	nl_list_for_each_entry(qm, &qdisc_modules, qm_list)
		if (qm->qm_class_ops == ops)
			return qm;

	return NULL;
}

struct nl_cli_qdisc_module *nl_cli_qdisc_lookup(struct rtnl_qdisc_ops *ops)
{
	struct nl_cli_qdisc_module *qm;

	if ((qm = __nl_cli_qdisc_lookup(ops)))
		return qm;

	nl_cli_load_module("cli/qdisc", ops->qo_kind);

	if (!(qm = __nl_cli_qdisc_lookup(ops)))  {
		nl_cli_fatal(EINVAL, "Application bug: The shared library for "
			"the qdisc \"%s\" was successfully loaded but it "
			"seems that module did not register itself");
	}

	return qm;
}

struct nl_cli_qdisc_module *nl_cli_qdisc_lookup_by_class(struct rtnl_class_ops *ops)
{
	struct nl_cli_qdisc_module *qm;

	if ((qm = __nl_cli_class_lookup(ops)))
		return qm;

	nl_cli_load_module("cli/qdisc", ops->co_kind);

	if (!(qm = __nl_cli_class_lookup(ops)))  {
		nl_cli_fatal(EINVAL, "Application bug: The shared library for "
			"the class \"%s\" was successfully loaded but it "
			"seems that module did not register itself");
	}

	return qm;
}

void nl_cli_qdisc_register(struct nl_cli_qdisc_module *qm)
{
	struct rtnl_qdisc_ops *ops;

	if (!(ops = __rtnl_qdisc_lookup_ops(qm->qm_name))) {
		nl_cli_fatal(ENOENT, "Unable to register CLI qdisc module "
		"\"%s\": No matching libnl qdisc module found.", qm->qm_name);
	}

	if (__nl_cli_qdisc_lookup(ops)) {
		nl_cli_fatal(EEXIST, "Unable to register CLI qdisc module "
		"\"%s\": Module already registered.", qm->qm_name);
	}

	qm->qm_ops = ops;
	qm->qm_class_ops = __rtnl_class_lookup_ops(qm->qm_name);

	nl_list_add_tail(&qm->qm_list, &qdisc_modules);
}

void nl_cli_qdisc_unregister(struct nl_cli_qdisc_module *qm)
{
	nl_list_del(&qm->qm_list);
}

/** @} */