diff options
author | Thomas Graf <tgraf@suug.ch> | 2007-09-14 23:28:01 (GMT) |
---|---|---|
committer | Thomas Graf <tgraf@suug.ch> | 2007-09-14 23:28:01 (GMT) |
commit | 44d362409d5469aed47d19e7908d19bd194493a4 (patch) | |
tree | 5d1e739a4566f3af796273e5c3f78ca53d234df6 /lib/route | |
download | libnl-44d362409d5469aed47d19e7908d19bd194493a4.zip libnl-44d362409d5469aed47d19e7908d19bd194493a4.tar.gz libnl-44d362409d5469aed47d19e7908d19bd194493a4.tar.bz2 |
Initial import
Diffstat (limited to 'lib/route')
33 files changed, 14173 insertions, 0 deletions
diff --git a/lib/route/addr.c b/lib/route/addr.c new file mode 100644 index 0000000..1947b5a --- /dev/null +++ b/lib/route/addr.c @@ -0,0 +1,1130 @@ +/* + * lib/route/addr.c Addresses + * + * 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> + * Baruch Even <baruch@ev-en.org>, + * Mediatrix Telecom, inc. <ericb@mediatrix.com> + */ + +/** + * @ingroup rtnl + * @defgroup rtaddr Addresses + * @brief + * + * @note The maximum size of an address label is IFNAMSIZ. + * + * @note The address may not contain a prefix length if the peer address + * has been specified already. + * + * @par 1) Address Addition + * @code + * // Allocate an empty address object to be filled out with the attributes + * // of the new address. + * struct rtnl_addr *addr = rtnl_addr_alloc(); + * + * // Fill out the mandatory attributes of the new address. Setting the + * // local address will automatically set the address family and the + * // prefix length to the correct values. + * rtnl_addr_set_ifindex(addr, ifindex); + * rtnl_addr_set_local(addr, local_addr); + * + * // The label of the address can be specified, currently only supported + * // by IPv4 and DECnet. + * rtnl_addr_set_label(addr, "mylabel"); + * + * // The peer address can be specified if necessary, in either case a peer + * // address will be sent to the kernel in order to fullfil the interface + * // requirements. If none is set, it will equal the local address. + * // Note: Real peer addresses are only supported by IPv4 for now. + * rtnl_addr_set_peer(addr, peer_addr); + * + * // In case you want to have the address have a scope other than global + * // it may be overwritten using rtnl_addr_set_scope(). The scope currently + * // cannot be set for IPv6 addresses. + * rtnl_addr_set_scope(addr, rtnl_str2scope("site")); + * + * // Broadcast and anycast address may be specified using the relevant + * // functions, the address family will be verified if one of the other + * // addresses has been set already. Currently only works for IPv4. + * rtnl_addr_set_broadcast(addr, broadcast_addr); + * rtnl_addr_set_anycast(addr, anycast_addr); + * + * // Build the netlink message and send it to the kernel, the operation will + * // block until the operation has been completed. Alternatively the required + * // netlink message can be built using rtnl_addr_build_add_request() to be + * // sent out using nl_send_auto_complete(). + * rtnl_addr_add(handle, addr, 0); + * + * // Free the memory + * rtnl_addr_put(addr); + * @endcode + * + * @par 2) Address Deletion + * @code + * // Allocate an empty address object to be filled out with the attributes + * // matching the address to be deleted. Alternatively a fully equipped + * // address object out of a cache can be used instead. + * struct rtnl_addr *addr = rtnl_addr_alloc(); + * + * // The only mandatory parameter besides the address family is the interface + * // index the address is on, i.e. leaving out all other parameters will + * // result in all addresses of the specified address family interface tuple + * // to be deleted. + * rtnl_addr_set_ifindex(addr, ifindex); + * + * // Specyfing the address family manually is only required if neither the + * // local nor peer address have been specified. + * rtnl_addr_set_family(addr, AF_INET); + * + * // Specyfing the local address is optional but the best choice to delete + * // specific addresses. + * rtnl_addr_set_local(addr, local_addr); + * + * // The label of the address can be specified, currently only supported + * // by IPv4 and DECnet. + * rtnl_addr_set_label(addr, "mylabel"); + * + * // The peer address can be specified if necessary, in either case a peer + * // address will be sent to the kernel in order to fullfil the interface + * // requirements. If none is set, it will equal the local address. + * // Note: Real peer addresses are only supported by IPv4 for now. + * rtnl_addr_set_peer(addr, peer_addr); + * + * // Build the netlink message and send it to the kernel, the operation will + * // block until the operation has been completed. Alternatively the required + * // netlink message can be built using rtnl_addr_build_delete_request() + * // to be sent out using nl_send_auto_complete(). + * rtnl_addr_delete(handle, addr, 0); + * + * // Free the memory + * rtnl_addr_put(addr); + * @endcode + * @{ + */ + +#include <netlink-local.h> +#include <netlink/netlink.h> +#include <netlink/route/rtnl.h> +#include <netlink/route/addr.h> +#include <netlink/route/route.h> +#include <netlink/route/link.h> +#include <netlink/utils.h> + +/** @cond SKIP */ +#define ADDR_ATTR_FAMILY 0x0001 +#define ADDR_ATTR_PREFIXLEN 0x0002 +#define ADDR_ATTR_FLAGS 0x0004 +#define ADDR_ATTR_SCOPE 0x0008 +#define ADDR_ATTR_IFINDEX 0x0010 +#define ADDR_ATTR_LABEL 0x0020 +#define ADDR_ATTR_CACHEINFO 0x0040 +#define ADDR_ATTR_PEER 0x0080 +#define ADDR_ATTR_LOCAL 0x0100 +#define ADDR_ATTR_BROADCAST 0x0200 +#define ADDR_ATTR_ANYCAST 0x0400 +#define ADDR_ATTR_MULTICAST 0x0800 + +static struct nl_cache_ops rtnl_addr_ops; +static struct nl_object_ops addr_obj_ops; +/** @endcond */ + +static void addr_free_data(struct nl_object *obj) +{ + struct rtnl_addr *addr = nl_object_priv(obj); + + if (!addr) + return; + + nl_addr_put(addr->a_peer); + nl_addr_put(addr->a_local); + nl_addr_put(addr->a_bcast); + nl_addr_put(addr->a_anycast); + nl_addr_put(addr->a_multicast); +} + +static int addr_clone(struct nl_object *_dst, struct nl_object *_src) +{ + struct rtnl_addr *dst = nl_object_priv(_dst); + struct rtnl_addr *src = nl_object_priv(_src); + + if (src->a_peer) + if (!(dst->a_peer = nl_addr_clone(src->a_peer))) + goto errout; + + if (src->a_local) + if (!(dst->a_local = nl_addr_clone(src->a_local))) + goto errout; + + if (src->a_bcast) + if (!(dst->a_bcast = nl_addr_clone(src->a_bcast))) + goto errout; + + if (src->a_anycast) + if (!(dst->a_anycast = nl_addr_clone(src->a_anycast))) + goto errout; + + if (src->a_multicast) + if (!(dst->a_multicast = nl_addr_clone(src->a_multicast))) + goto errout; + + return 0; +errout: + return nl_get_errno(); +} + +static struct nla_policy addr_policy[IFA_MAX+1] = { + [IFA_LABEL] = { .type = NLA_STRING, + .maxlen = IFNAMSIZ }, + [IFA_CACHEINFO] = { .minlen = sizeof(struct ifa_cacheinfo) }, +}; + +static int addr_msg_parser(struct nl_cache_ops *ops, struct sockaddr_nl *who, + struct nlmsghdr *nlh, void *arg) +{ + struct rtnl_addr *addr; + struct nl_parser_param *pp = arg; + struct ifaddrmsg *ifa; + struct nlattr *tb[IFA_MAX+1]; + int err = -ENOMEM, peer_prefix = 0; + + addr = rtnl_addr_alloc(); + if (!addr) { + err = nl_errno(ENOMEM); + goto errout; + } + addr->ce_msgtype = nlh->nlmsg_type; + + err = nlmsg_parse(nlh, sizeof(*ifa), tb, IFA_MAX, addr_policy); + if (err < 0) + goto errout_free; + + ifa = nlmsg_data(nlh); + addr->a_family = ifa->ifa_family; + addr->a_prefixlen = ifa->ifa_prefixlen; + addr->a_flags = ifa->ifa_flags; + addr->a_scope = ifa->ifa_scope; + addr->a_ifindex = ifa->ifa_index; + + addr->ce_mask = (ADDR_ATTR_FAMILY | ADDR_ATTR_PREFIXLEN | + ADDR_ATTR_FLAGS | ADDR_ATTR_SCOPE | ADDR_ATTR_IFINDEX); + + if (tb[IFA_LABEL]) { + nla_strlcpy(addr->a_label, tb[IFA_LABEL], IFNAMSIZ); + addr->ce_mask |= ADDR_ATTR_LABEL; + } + + if (tb[IFA_CACHEINFO]) { + struct ifa_cacheinfo *ca; + + ca = nla_data(tb[IFA_CACHEINFO]); + addr->a_cacheinfo.aci_prefered = ca->ifa_prefered; + addr->a_cacheinfo.aci_valid = ca->ifa_valid; + addr->a_cacheinfo.aci_cstamp = ca->cstamp; + addr->a_cacheinfo.aci_tstamp = ca->tstamp; + addr->ce_mask |= ADDR_ATTR_CACHEINFO; + } + + if (tb[IFA_LOCAL]) { + addr->a_local = nla_get_addr(tb[IFA_LOCAL], addr->a_family); + if (!addr->a_local) + goto errout_free; + addr->ce_mask |= ADDR_ATTR_LOCAL; + } + + if (tb[IFA_ADDRESS]) { + struct nl_addr *a; + + a = nla_get_addr(tb[IFA_ADDRESS], addr->a_family); + if (!a) + goto errout_free; + + /* IPv6 sends the local address as IFA_ADDRESS with + * no IFA_LOCAL, IPv4 sends both IFA_LOCAL and IFA_ADDRESS + * with IFA_ADDRESS being the peer address if they differ */ + if (!tb[IFA_LOCAL] || !nl_addr_cmp(a, addr->a_local)) { + nl_addr_put(addr->a_local); + addr->a_local = a; + addr->ce_mask |= ADDR_ATTR_LOCAL; + } else { + addr->a_peer = a; + addr->ce_mask |= ADDR_ATTR_PEER; + peer_prefix = 1; + } + } + + nl_addr_set_prefixlen(peer_prefix ? addr->a_peer : addr->a_local, + addr->a_prefixlen); + + if (tb[IFA_BROADCAST]) { + addr->a_bcast = nla_get_addr(tb[IFA_BROADCAST], addr->a_family); + if (!addr->a_bcast) + goto errout_free; + + addr->ce_mask |= ADDR_ATTR_BROADCAST; + } + + if (tb[IFA_ANYCAST]) { + addr->a_anycast = nla_get_addr(tb[IFA_ANYCAST], addr->a_family); + if (!addr->a_anycast) + goto errout_free; + + addr->ce_mask |= ADDR_ATTR_ANYCAST; + } + + if (tb[IFA_MULTICAST]) { + addr->a_multicast = nla_get_addr(tb[IFA_MULTICAST], + addr->a_family); + if (!addr->a_multicast) + goto errout_free; + + addr->ce_mask |= ADDR_ATTR_MULTICAST; + } + + err = pp->pp_cb((struct nl_object *) addr, pp); + if (err < 0) + goto errout_free; + + return P_ACCEPT; + +errout_free: + rtnl_addr_put(addr); +errout: + return err; +} + +static int addr_request_update(struct nl_cache *cache, struct nl_handle *handle) +{ + return nl_rtgen_request(handle, RTM_GETADDR, AF_UNSPEC, NLM_F_DUMP); +} + +static int addr_dump_brief(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_addr *addr = (struct rtnl_addr *) obj; + struct nl_cache *link_cache; + char buf[128]; + + link_cache = nl_cache_mngt_require("route/link"); + + if (addr->ce_mask & ADDR_ATTR_LOCAL) + dp_dump(p, "%s", + nl_addr2str(addr->a_local, buf, sizeof(buf))); + else + dp_dump(p, "none"); + + if (addr->ce_mask & ADDR_ATTR_PEER) + dp_dump(p, " peer %s", + nl_addr2str(addr->a_peer, buf, sizeof(buf))); + + dp_dump(p, " %s ", nl_af2str(addr->a_family, buf, sizeof(buf))); + + if (link_cache) + dp_dump(p, "dev %s ", + rtnl_link_i2name(link_cache, addr->a_ifindex, + buf, sizeof(buf))); + else + dp_dump(p, "dev %d ", addr->a_ifindex); + + dp_dump(p, "scope %s", + rtnl_scope2str(addr->a_scope, buf, sizeof(buf))); + + rtnl_addr_flags2str(addr->a_flags, buf, sizeof(buf)); + if (buf[0]) + dp_dump(p, " <%s>", buf); + + dp_dump(p, "\n"); + + return 1; +} + +static int addr_dump_full(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_addr *addr = (struct rtnl_addr *) obj; + int line = addr_dump_brief(obj, p); + char buf[128]; + + if (addr->ce_mask & (ADDR_ATTR_LABEL | ADDR_ATTR_BROADCAST | + ADDR_ATTR_ANYCAST | ADDR_ATTR_MULTICAST)) { + dp_dump_line(p, line++, " "); + + if (addr->ce_mask & ADDR_ATTR_LABEL) + dp_dump(p, " label %s", addr->a_label); + + if (addr->ce_mask & ADDR_ATTR_BROADCAST) + dp_dump(p, " broadcast %s", + nl_addr2str(addr->a_bcast, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_ANYCAST) + dp_dump(p, " anycast %s", + nl_addr2str(addr->a_anycast, buf, + sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_MULTICAST) + dp_dump(p, " multicast %s", + nl_addr2str(addr->a_multicast, buf, + sizeof(buf))); + + dp_dump(p, "\n"); + } + + if (addr->ce_mask & ADDR_ATTR_CACHEINFO) { + struct rtnl_addr_cacheinfo *ci = &addr->a_cacheinfo; + + dp_dump_line(p, line++, " valid-lifetime %s", + ci->aci_valid == 0xFFFFFFFFU ? "forever" : + nl_msec2str(ci->aci_valid * 1000, + buf, sizeof(buf))); + + dp_dump(p, " preferred-lifetime %s\n", + ci->aci_prefered == 0xFFFFFFFFU ? "forever" : + nl_msec2str(ci->aci_prefered * 1000, + buf, sizeof(buf))); + + dp_dump_line(p, line++, " created boot-time+%s ", + nl_msec2str(addr->a_cacheinfo.aci_cstamp * 10, + buf, sizeof(buf))); + + dp_dump(p, "last-updated boot-time+%s\n", + nl_msec2str(addr->a_cacheinfo.aci_tstamp * 10, + buf, sizeof(buf))); + } + + return line; +} + +static int addr_dump_stats(struct nl_object *obj, struct nl_dump_params *p) +{ + return addr_dump_full(obj, p); +} + +static int addr_dump_xml(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_addr *addr = (struct rtnl_addr *) obj; + struct nl_cache *link_cache; + char buf[128]; + int line = 0; + + dp_dump_line(p, line++, "<address>\n"); + dp_dump_line(p, line++, " <family>%s</family>\n", + nl_af2str(addr->a_family, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_LOCAL) + dp_dump_line(p, line++, " <local>%s</local>\n", + nl_addr2str(addr->a_local, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_PEER) + dp_dump_line(p, line++, " <peer>%s</peer>\n", + nl_addr2str(addr->a_peer, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_BROADCAST) + dp_dump_line(p, line++, " <broadcast>%s</broadcast>\n", + nl_addr2str(addr->a_bcast, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_ANYCAST) + dp_dump_line(p, line++, " <anycast>%s</anycast>\n", + nl_addr2str(addr->a_anycast, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_MULTICAST) + dp_dump_line(p, line++, " <multicast>%s</multicast>\n", + nl_addr2str(addr->a_multicast, buf, + sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_PREFIXLEN) + dp_dump_line(p, line++, " <prefixlen>%u</prefixlen>\n", + addr->a_prefixlen); + link_cache = nl_cache_mngt_require("route/link"); + + if (link_cache) + dp_dump_line(p, line++, " <device>%s</device>\n", + rtnl_link_i2name(link_cache, addr->a_ifindex, + buf, sizeof(buf))); + else + dp_dump_line(p, line++, " <device>%u</device>\n", + addr->a_ifindex); + + if (addr->ce_mask & ADDR_ATTR_SCOPE) + dp_dump_line(p, line++, " <scope>%s</scope>\n", + rtnl_scope2str(addr->a_scope, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_LABEL) + dp_dump_line(p, line++, " <label>%s</label>\n", addr->a_label); + + rtnl_addr_flags2str(addr->a_flags, buf, sizeof(buf)); + if (buf[0]) + dp_dump_line(p, line++, " <flags>%s</flags>\n", buf); + + if (addr->ce_mask & ADDR_ATTR_CACHEINFO) { + struct rtnl_addr_cacheinfo *ci = &addr->a_cacheinfo; + + dp_dump_line(p, line++, " <cacheinfo>\n"); + + dp_dump_line(p, line++, " <valid>%s</valid>\n", + ci->aci_valid == 0xFFFFFFFFU ? "forever" : + nl_msec2str(ci->aci_valid * 1000, + buf, sizeof(buf))); + + dp_dump_line(p, line++, " <prefered>%s</prefered>\n", + ci->aci_prefered == 0xFFFFFFFFU ? "forever" : + nl_msec2str(ci->aci_prefered * 1000, + buf, sizeof(buf))); + + dp_dump_line(p, line++, " <created>%s</created>\n", + nl_msec2str(addr->a_cacheinfo.aci_cstamp * 10, + buf, sizeof(buf))); + + dp_dump_line(p, line++, " <last-update>%s</last-update>\n", + nl_msec2str(addr->a_cacheinfo.aci_tstamp * 10, + buf, sizeof(buf))); + + dp_dump_line(p, line++, " </cacheinfo>\n"); + } + + dp_dump_line(p, line++, "</address>\n"); + + return line; +} + +static int addr_dump_env(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_addr *addr = (struct rtnl_addr *) obj; + struct nl_cache *link_cache; + char buf[128]; + int line = 0; + + dp_dump_line(p, line++, "ADDR_FAMILY=%s\n", + nl_af2str(addr->a_family, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_LOCAL) + dp_dump_line(p, line++, "ADDR_LOCAL=%s\n", + nl_addr2str(addr->a_local, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_PEER) + dp_dump_line(p, line++, "ADDR_PEER=%s\n", + nl_addr2str(addr->a_peer, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_BROADCAST) + dp_dump_line(p, line++, "ADDR_BROADCAST=%s\n", + nl_addr2str(addr->a_bcast, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_ANYCAST) + dp_dump_line(p, line++, "ADDR_ANYCAST=%s\n", + nl_addr2str(addr->a_anycast, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_MULTICAST) + dp_dump_line(p, line++, "ADDR_MULTICAST=%s\n", + nl_addr2str(addr->a_multicast, buf, + sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_PREFIXLEN) + dp_dump_line(p, line++, "ADDR_PREFIXLEN=%u\n", + addr->a_prefixlen); + link_cache = nl_cache_mngt_require("route/link"); + + dp_dump_line(p, line++, "ADDR_IFINDEX=%u\n", addr->a_ifindex); + if (link_cache) + dp_dump_line(p, line++, "ADDR_IFNAME=%s\n", + rtnl_link_i2name(link_cache, addr->a_ifindex, + buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_SCOPE) + dp_dump_line(p, line++, "ADDR_SCOPE=%s\n", + rtnl_scope2str(addr->a_scope, buf, sizeof(buf))); + + if (addr->ce_mask & ADDR_ATTR_LABEL) + dp_dump_line(p, line++, "ADDR_LABEL=%s\n", addr->a_label); + + rtnl_addr_flags2str(addr->a_flags, buf, sizeof(buf)); + if (buf[0]) + dp_dump_line(p, line++, "ADDR_FLAGS=%s\n", buf); + + if (addr->ce_mask & ADDR_ATTR_CACHEINFO) { + struct rtnl_addr_cacheinfo *ci = &addr->a_cacheinfo; + + dp_dump_line(p, line++, "ADDR_CACHEINFO_VALID=%s\n", + ci->aci_valid == 0xFFFFFFFFU ? "forever" : + nl_msec2str(ci->aci_valid * 1000, + buf, sizeof(buf))); + + dp_dump_line(p, line++, "ADDR_CACHEINFO_PREFERED=%s\n", + ci->aci_prefered == 0xFFFFFFFFU ? "forever" : + nl_msec2str(ci->aci_prefered * 1000, + buf, sizeof(buf))); + + dp_dump_line(p, line++, "ADDR_CACHEINFO_CREATED=%s\n", + nl_msec2str(addr->a_cacheinfo.aci_cstamp * 10, + buf, sizeof(buf))); + + dp_dump_line(p, line++, "ADDR_CACHEINFO_LASTUPDATE=%s\n", + nl_msec2str(addr->a_cacheinfo.aci_tstamp * 10, + buf, sizeof(buf))); + } + + return line; +} + +static int addr_compare(struct nl_object *_a, struct nl_object *_b, + uint32_t attrs, int flags) +{ + struct rtnl_addr *a = (struct rtnl_addr *) _a; + struct rtnl_addr *b = (struct rtnl_addr *) _b; + int diff = 0; + +#define ADDR_DIFF(ATTR, EXPR) ATTR_DIFF(attrs, ADDR_ATTR_##ATTR, a, b, EXPR) + + diff |= ADDR_DIFF(IFINDEX, a->a_ifindex != b->a_ifindex); + diff |= ADDR_DIFF(FAMILY, a->a_family != b->a_family); + diff |= ADDR_DIFF(SCOPE, a->a_scope != b->a_scope); + diff |= ADDR_DIFF(LABEL, strcmp(a->a_label, b->a_label)); + diff |= ADDR_DIFF(PEER, nl_addr_cmp(a->a_peer, b->a_peer)); + diff |= ADDR_DIFF(LOCAL, nl_addr_cmp(a->a_local, b->a_local)); + diff |= ADDR_DIFF(ANYCAST, nl_addr_cmp(a->a_anycast,b->a_anycast)); + diff |= ADDR_DIFF(MULTICAST, nl_addr_cmp(a->a_multicast, + b->a_multicast)); + diff |= ADDR_DIFF(BROADCAST, nl_addr_cmp(a->a_bcast, b->a_bcast)); + + if (flags & LOOSE_FLAG_COMPARISON) + diff |= ADDR_DIFF(FLAGS, + (a->a_flags ^ b->a_flags) & b->a_flag_mask); + else + diff |= ADDR_DIFF(FLAGS, a->a_flags != b->a_flags); + +#undef ADDR_DIFF + + return diff; +} + +static struct trans_tbl addr_attrs[] = { + __ADD(ADDR_ATTR_FAMILY, family) + __ADD(ADDR_ATTR_PREFIXLEN, prefixlen) + __ADD(ADDR_ATTR_FLAGS, flags) + __ADD(ADDR_ATTR_SCOPE, scope) + __ADD(ADDR_ATTR_IFINDEX, ifindex) + __ADD(ADDR_ATTR_LABEL, label) + __ADD(ADDR_ATTR_CACHEINFO, cacheinfo) + __ADD(ADDR_ATTR_PEER, peer) + __ADD(ADDR_ATTR_LOCAL, local) + __ADD(ADDR_ATTR_BROADCAST, broadcast) + __ADD(ADDR_ATTR_ANYCAST, anycast) + __ADD(ADDR_ATTR_MULTICAST, multicast) +}; + +static char *addr_attrs2str(int attrs, char *buf, size_t len) +{ + return __flags2str(attrs, buf, len, addr_attrs, + ARRAY_SIZE(addr_attrs)); +} + +/** + * @name Allocation/Freeing + * @{ + */ + +struct rtnl_addr *rtnl_addr_alloc(void) +{ + return (struct rtnl_addr *) nl_object_alloc(&addr_obj_ops); +} + +void rtnl_addr_put(struct rtnl_addr *addr) +{ + nl_object_put((struct nl_object *) addr); +} + +/** @} */ + +/** + * @name Cache Management + * @{ + */ + +struct nl_cache *rtnl_addr_alloc_cache(struct nl_handle *handle) +{ + struct nl_cache *cache; + + cache = nl_cache_alloc(&rtnl_addr_ops); + if (!cache) + return NULL; + + if (handle && nl_cache_refill(handle, cache) < 0) { + nl_cache_free(cache); + return NULL; + } + + return cache; +} + +/** @} */ + +static struct nl_msg *build_addr_msg(struct rtnl_addr *tmpl, int cmd, int flags) +{ + struct nl_msg *msg; + struct ifaddrmsg am = { + .ifa_family = tmpl->a_family, + .ifa_index = tmpl->a_ifindex, + .ifa_prefixlen = tmpl->a_prefixlen, + }; + + if (tmpl->ce_mask & ADDR_ATTR_SCOPE) + am.ifa_scope = tmpl->a_scope; + else { + /* compatibility hack */ + if (tmpl->a_family == AF_INET && + tmpl->ce_mask & ADDR_ATTR_LOCAL && + *((char *) nl_addr_get_binary_addr(tmpl->a_local)) == 127) + am.ifa_scope = RT_SCOPE_HOST; + else + am.ifa_scope = RT_SCOPE_UNIVERSE; + } + + msg = nlmsg_alloc_simple(cmd, flags); + if (!msg) + goto nla_put_failure; + + if (nlmsg_append(msg, &am, sizeof(am), NLMSG_ALIGNTO) < 0) + goto nla_put_failure; + + if (tmpl->ce_mask & ADDR_ATTR_LOCAL) + NLA_PUT_ADDR(msg, IFA_LOCAL, tmpl->a_local); + + if (tmpl->ce_mask & ADDR_ATTR_PEER) + NLA_PUT_ADDR(msg, IFA_ADDRESS, tmpl->a_peer); + else + NLA_PUT_ADDR(msg, IFA_ADDRESS, tmpl->a_local); + + if (tmpl->ce_mask & ADDR_ATTR_LABEL) + NLA_PUT_STRING(msg, IFA_LABEL, tmpl->a_label); + + if (tmpl->ce_mask & ADDR_ATTR_BROADCAST) + NLA_PUT_ADDR(msg, IFA_BROADCAST, tmpl->a_bcast); + + if (tmpl->ce_mask & ADDR_ATTR_ANYCAST) + NLA_PUT_ADDR(msg, IFA_ANYCAST, tmpl->a_anycast); + + return msg; + +nla_put_failure: + nlmsg_free(msg); + return NULL; +} + +/** + * @name Addition + * @{ + */ + +/** + * Build netlink request message to request addition of new address + * @arg addr Address object representing the new address. + * @arg flags Additional netlink message flags. + * + * Builds a new netlink message requesting the addition of a new + * address. The netlink message header isn't fully equipped with + * all relevant fields and must thus be sent out via nl_send_auto_complete() + * or supplemented as needed. + * + * Minimal required attributes: + * - interface index (rtnl_addr_set_ifindex()) + * - local address (rtnl_addr_set_local()) + * + * The scope will default to universe except for loopback addresses in + * which case a host scope is used if not specified otherwise. + * + * @note Free the memory after usage using nlmsg_free(). + * @return Newly allocated netlink message or NULL if an error occured. + */ +struct nl_msg *rtnl_addr_build_add_request(struct rtnl_addr *addr, int flags) +{ + int required = ADDR_ATTR_IFINDEX | ADDR_ATTR_FAMILY | + ADDR_ATTR_PREFIXLEN | ADDR_ATTR_LOCAL; + + if ((addr->ce_mask & required) != required) { + nl_error(EINVAL, "Missing mandatory attributes, required are: " + "ifindex, family, prefixlen, local address."); + return NULL; + } + + return build_addr_msg(addr, RTM_NEWADDR, NLM_F_CREATE | flags); +} + +/** + * Request addition of new address + * @arg handle Netlink handle. + * @arg addr Address object representing the new address. + * @arg flags Additional netlink message flags. + * + * Builds a netlink message by calling rtnl_addr_build_add_request(), + * sends the request to the kernel and waits for the next ACK to be + * received and thus blocks until the request has been fullfilled. + * + * @see rtnl_addr_build_add_request() + * + * @return 0 on sucess or a negative error if an error occured. + */ +int rtnl_addr_add(struct nl_handle *handle, struct rtnl_addr *addr, int flags) +{ + struct nl_msg *msg; + int err; + + msg = rtnl_addr_build_add_request(addr, flags); + if (!msg) + return nl_get_errno(); + + err = nl_send_auto_complete(handle, msg); + nlmsg_free(msg); + if (err < 0) + return err; + + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name Deletion + * @{ + */ + +/** + * Build a netlink request message to request deletion of an address + * @arg addr Address object to be deleteted. + * @arg flags Additional netlink message flags. + * + * Builds a new netlink message requesting a deletion of an address. + * The netlink message header isn't fully equipped with all relevant + * fields and must thus be sent out via nl_send_auto_complete() + * or supplemented as needed. + * + * Minimal required attributes: + * - interface index (rtnl_addr_set_ifindex()) + * - address family (rtnl_addr_set_family()) + * + * Optional attributes: + * - local address (rtnl_addr_set_local()) + * - label (rtnl_addr_set_label(), IPv4/DECnet only) + * - peer address (rtnl_addr_set_peer(), IPv4 only) + * + * @note Free the memory after usage using nlmsg_free(). + * @return Newly allocated netlink message or NULL if an error occured. + */ +struct nl_msg *rtnl_addr_build_delete_request(struct rtnl_addr *addr, int flags) +{ + int required = ADDR_ATTR_IFINDEX | ADDR_ATTR_FAMILY; + + if ((addr->ce_mask & required) != required) { + nl_error(EINVAL, "Missing mandatory attributes, required are: " + "ifindex, family"); + return NULL; + } + + return build_addr_msg(addr, RTM_DELADDR, flags); +} + +/** + * Request deletion of an address + * @arg handle Netlink handle. + * @arg addr Address object to be deleted. + * @arg flags Additional netlink message flags. + * + * Builds a netlink message by calling rtnl_addr_build_delete_request(), + * sends the request to the kernel and waits for the next ACK to be + * received and thus blocks until the request has been fullfilled. + * + * @see rtnl_addr_build_delete_request(); + * + * @return 0 on sucess or a negative error if an error occured. + */ +int rtnl_addr_delete(struct nl_handle *handle, struct rtnl_addr *addr, + int flags) +{ + struct nl_msg *msg; + int err; + + msg = rtnl_addr_build_delete_request(addr, flags); + if (!msg) + return nl_get_errno(); + + err = nl_send_auto_complete(handle, msg); + nlmsg_free(msg); + if (err < 0) + return err; + + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name Attributes + * @{ + */ + +void rtnl_addr_set_label(struct rtnl_addr *addr, const char *label) +{ + strncpy(addr->a_label, label, sizeof(addr->a_label) - 1); + addr->ce_mask |= ADDR_ATTR_LABEL; +} + +char *rtnl_addr_get_label(struct rtnl_addr *addr) +{ + if (addr->ce_mask & ADDR_ATTR_LABEL) + return addr->a_label; + else + return NULL; +} + +void rtnl_addr_set_ifindex(struct rtnl_addr *addr, int ifindex) +{ + addr->a_ifindex = ifindex; + addr->ce_mask |= ADDR_ATTR_IFINDEX; +} + +int rtnl_addr_get_ifindex(struct rtnl_addr *addr) +{ + if (addr->ce_mask & ADDR_ATTR_IFINDEX) + return addr->a_ifindex; + else + return RTNL_LINK_NOT_FOUND; +} + +void rtnl_addr_set_family(struct rtnl_addr *addr, int family) +{ + addr->a_family = family; + addr->ce_mask |= ADDR_ATTR_FAMILY; +} + +int rtnl_addr_get_family(struct rtnl_addr *addr) +{ + if (addr->ce_mask & ADDR_ATTR_FAMILY) + return addr->a_family; + else + return AF_UNSPEC; +} + +void rtnl_addr_set_prefixlen(struct rtnl_addr *addr, int prefix) +{ + addr->a_prefixlen = prefix; + addr->ce_mask |= ADDR_ATTR_PREFIXLEN; +} + +int rtnl_addr_get_prefixlen(struct rtnl_addr *addr) +{ + if (addr->ce_mask & ADDR_ATTR_PREFIXLEN) + return addr->a_prefixlen; + else + return -1; +} + +void rtnl_addr_set_scope(struct rtnl_addr *addr, int scope) +{ + addr->a_scope = scope; + addr->ce_mask |= ADDR_ATTR_SCOPE; +} + +int rtnl_addr_get_scope(struct rtnl_addr *addr) +{ + if (addr->ce_mask & ADDR_ATTR_SCOPE) + return addr->a_scope; + else + return -1; +} + +void rtnl_addr_set_flags(struct rtnl_addr *addr, unsigned int flags) +{ + addr->a_flag_mask |= flags; + addr->a_flags |= flags; + addr->ce_mask |= ADDR_ATTR_FLAGS; +} + +void rtnl_addr_unset_flags(struct rtnl_addr *addr, unsigned int flags) +{ + addr->a_flag_mask |= flags; + addr->a_flags &= ~flags; + addr->ce_mask |= ADDR_ATTR_FLAGS; +} + +unsigned int rtnl_addr_get_flags(struct rtnl_addr *addr) +{ + return addr->a_flags; +} + +static inline int __assign_addr(struct rtnl_addr *addr, struct nl_addr **pos, + struct nl_addr *new, int flag) +{ + if (addr->ce_mask & ADDR_ATTR_FAMILY) { + if (new->a_family != addr->a_family) + return nl_error(EINVAL, "Address family mismatch"); + } else + addr->a_family = new->a_family; + + if (*pos) + nl_addr_put(*pos); + + *pos = nl_addr_get(new); + addr->ce_mask |= (flag | ADDR_ATTR_FAMILY); + + return 0; +} + +int rtnl_addr_set_local(struct rtnl_addr *addr, struct nl_addr *local) +{ + int err; + + err = __assign_addr(addr, &addr->a_local, local, ADDR_ATTR_LOCAL); + if (err < 0) + return err; + + if (!(addr->ce_mask & ADDR_ATTR_PEER)) { + addr->a_prefixlen = nl_addr_get_prefixlen(addr->a_local); + addr->ce_mask |= ADDR_ATTR_PREFIXLEN; + } + + return 0; +} + +struct nl_addr *rtnl_addr_get_local(struct rtnl_addr *addr) +{ + if (addr->ce_mask & ADDR_ATTR_LOCAL) + return addr->a_local; + else + return NULL; +} + +int rtnl_addr_set_peer(struct rtnl_addr *addr, struct nl_addr *peer) +{ + return __assign_addr(addr, &addr->a_peer, peer, ADDR_ATTR_PEER); + + addr->a_prefixlen = nl_addr_get_prefixlen(addr->a_peer); + addr->ce_mask |= ADDR_ATTR_PREFIXLEN; + + return 0; +} + +struct nl_addr *rtnl_addr_get_peer(struct rtnl_addr *addr) +{ + if (addr->ce_mask & ADDR_ATTR_PEER) + return addr->a_peer; + else + return NULL; +} + +int rtnl_addr_set_broadcast(struct rtnl_addr *addr, struct nl_addr *bcast) +{ + return __assign_addr(addr, &addr->a_bcast, bcast, ADDR_ATTR_BROADCAST); +} + +struct nl_addr *rtnl_addr_get_broadcast(struct rtnl_addr *addr) +{ + if (addr->ce_mask & ADDR_ATTR_BROADCAST) + return addr->a_bcast; + else + return NULL; +} + +int rtnl_addr_set_anycast(struct rtnl_addr *addr, struct nl_addr *anycast) +{ + return __assign_addr(addr, &addr->a_anycast, anycast, + ADDR_ATTR_ANYCAST); +} + +struct nl_addr *rtnl_addr_get_anycast(struct rtnl_addr *addr) +{ + if (addr->ce_mask & ADDR_ATTR_ANYCAST) + return addr->a_anycast; + else + return NULL; +} + +int rtnl_addr_set_multicast(struct rtnl_addr *addr, struct nl_addr *multicast) +{ + return __assign_addr(addr, &addr->a_multicast, multicast, + ADDR_ATTR_MULTICAST); +} + +struct nl_addr *rtnl_addr_get_multicast(struct rtnl_addr *addr) +{ + if (addr->ce_mask & ADDR_ATTR_MULTICAST) + return addr->a_multicast; + else + return NULL; +} + +/** @} */ + +/** + * @name Flags Translations + * @{ + */ + +static struct trans_tbl addr_flags[] = { + __ADD(IFA_F_SECONDARY, secondary) + __ADD(IFA_F_DEPRECATED, deprecated) + __ADD(IFA_F_TENTATIVE, tentative) + __ADD(IFA_F_PERMANENT, permanent) +}; + +char *rtnl_addr_flags2str(int flags, char *buf, size_t size) +{ + return __flags2str(flags, buf, size, addr_flags, + ARRAY_SIZE(addr_flags)); +} + +int rtnl_addr_str2flags(const char *name) +{ + return __str2flags(name, addr_flags, ARRAY_SIZE(addr_flags)); +} + +/** @} */ + +static struct nl_object_ops addr_obj_ops = { + .oo_name = "route/addr", + .oo_size = sizeof(struct rtnl_addr), + .oo_free_data = addr_free_data, + .oo_clone = addr_clone, + .oo_dump[NL_DUMP_BRIEF] = addr_dump_brief, + .oo_dump[NL_DUMP_FULL] = addr_dump_full, + .oo_dump[NL_DUMP_STATS] = addr_dump_stats, + .oo_dump[NL_DUMP_XML] = addr_dump_xml, + .oo_dump[NL_DUMP_ENV] = addr_dump_env, + .oo_compare = addr_compare, + .oo_attrs2str = addr_attrs2str, + .oo_id_attrs = (ADDR_ATTR_FAMILY | ADDR_ATTR_IFINDEX | + ADDR_ATTR_LOCAL | ADDR_ATTR_PREFIXLEN | + ADDR_ATTR_PEER), +}; + +static struct nl_af_group addr_groups[] = { + { AF_INET, RTNLGRP_IPV4_IFADDR }, + { AF_INET6, RTNLGRP_IPV6_IFADDR }, + { END_OF_GROUP_LIST }, +}; + +static struct nl_cache_ops rtnl_addr_ops = { + .co_name = "route/addr", + .co_hdrsize = sizeof(struct ifaddrmsg), + .co_msgtypes = { + { RTM_NEWADDR, NL_ACT_NEW, "new" }, + { RTM_DELADDR, NL_ACT_DEL, "del" }, + { RTM_GETADDR, NL_ACT_GET, "get" }, + END_OF_MSGTYPES_LIST, + }, + .co_protocol = NETLINK_ROUTE, + .co_groups = addr_groups, + .co_request_update = addr_request_update, + .co_msg_parser = addr_msg_parser, + .co_obj_ops = &addr_obj_ops, +}; + +static void __init addr_init(void) +{ + nl_cache_mngt_register(&rtnl_addr_ops); +} + +static void __exit addr_exit(void) +{ + nl_cache_mngt_unregister(&rtnl_addr_ops); +} + +/** @} */ diff --git a/lib/route/class.c b/lib/route/class.c new file mode 100644 index 0000000..7f2289e --- /dev/null +++ b/lib/route/class.c @@ -0,0 +1,230 @@ +/* + * lib/route/class.c Queueing Classes + * + * 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> + */ + +/** + * @ingroup tc + * @defgroup class Queueing Classes + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/route/tc.h> +#include <netlink/route/class.h> +#include <netlink/route/class-modules.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/classifier.h> +#include <netlink/utils.h> + +static struct nl_cache_ops rtnl_class_ops; + +static int class_msg_parser(struct nl_cache_ops *ops, struct sockaddr_nl *who, + struct nlmsghdr *n, void *arg) +{ + int err; + struct nl_parser_param *pp = arg; + struct rtnl_class *class; + struct rtnl_class_ops *cops; + + class = rtnl_class_alloc(); + if (!class) { + err = nl_errno(ENOMEM); + goto errout; + } + class->ce_msgtype = n->nlmsg_type; + + err = tca_msg_parser(n, (struct rtnl_tca *) class); + if (err < 0) + goto errout_free; + + cops = rtnl_class_lookup_ops(class); + if (cops && cops->co_msg_parser) { + err = cops->co_msg_parser(class); + if (err < 0) + goto errout_free; + } + + err = pp->pp_cb((struct nl_object *) class, pp); + if (err < 0) + goto errout_free; + + return P_ACCEPT; + +errout_free: + rtnl_class_put(class); +errout: + return err; +} + +static int class_request_update(struct nl_cache *cache, + struct nl_handle *handle) +{ + struct tcmsg tchdr = { + .tcm_family = AF_UNSPEC, + .tcm_ifindex = cache->c_iarg1, + }; + + return nl_send_simple(handle, RTM_GETTCLASS, NLM_F_DUMP, &tchdr, + sizeof(tchdr)); +} + +/** + * @name Addition/Modification + * @{ + */ + +static struct nl_msg *class_build(struct rtnl_class *class, int type, int flags) +{ + struct rtnl_class_ops *cops; + struct nl_msg *msg; + int err; + + msg = tca_build_msg((struct rtnl_tca *) class, type, flags); + if (!msg) + goto errout; + + cops = rtnl_class_lookup_ops(class); + if (cops && cops->co_get_opts) { + struct nl_msg *opts; + + opts = cops->co_get_opts(class); + if (opts) { + err = nla_put_nested(msg, TCA_OPTIONS, opts); + nlmsg_free(opts); + if (err < 0) + goto errout; + } + } + + return msg; +errout: + nlmsg_free(msg); + return NULL; +} + +/** + * Build a netlink message to add a new class + * @arg class class to add + * @arg flags additional netlink message flags + * + * Builds a new netlink message requesting an addition of a class. + * The netlink message header isn't fully equipped with all relevant + * fields and must be sent out via nl_send_auto_complete() or + * supplemented as needed. + * + * Common message flags + * - NLM_F_REPLACE - replace possibly existing classes + * + * @return New netlink message + */ +struct nl_msg *rtnl_class_build_add_request(struct rtnl_class *class, int flags) +{ + return class_build(class, RTM_NEWTCLASS, NLM_F_CREATE | flags); +} + +/** + * Add a new class + * @arg handle netlink handle + * @arg class class to delete + * @arg flags additional netlink message flags + * + * Builds a netlink message by calling rtnl_qdisc_build_add_request(), + * sends the request to the kernel and waits for the next ACK to be + * received and thus blocks until the request has been processed. + * + * Common message flags + * - NLM_F_REPLACE - replace possibly existing classes + * + * @return 0 on success or a negative error code + */ +int rtnl_class_add(struct nl_handle *handle, struct rtnl_class *class, + int flags) +{ + struct nl_msg *msg; + int err; + + msg = rtnl_class_build_add_request(class, flags); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name Cache Management + * @{ + */ + +/** + * Build a class cache including all classes attached to the specified interface + * @arg handle netlink handle + * @arg ifindex interface index of the link the classes are + * attached to. + * + * Allocates a new cache, initializes it properly and updates it to + * include all classes attached to the specified interface. + * + * @return The cache or NULL if an error has occured. + */ +struct nl_cache * rtnl_class_alloc_cache(struct nl_handle *handle, int ifindex) +{ + struct nl_cache * cache; + + cache = nl_cache_alloc(&rtnl_class_ops); + if (!cache) + return NULL; + + cache->c_iarg1 = ifindex; + + if (handle && nl_cache_refill(handle, cache) < 0) { + nl_cache_free(cache); + return NULL; + } + + return cache; +} + +/** @} */ + +static struct nl_cache_ops rtnl_class_ops = { + .co_name = "route/class", + .co_hdrsize = sizeof(struct tcmsg), + .co_msgtypes = { + { RTM_NEWTCLASS, NL_ACT_NEW, "new" }, + { RTM_DELTCLASS, NL_ACT_DEL, "del" }, + { RTM_GETTCLASS, NL_ACT_GET, "get" }, + END_OF_MSGTYPES_LIST, + }, + .co_protocol = NETLINK_ROUTE, + .co_request_update = &class_request_update, + .co_msg_parser = &class_msg_parser, + .co_obj_ops = &class_obj_ops, +}; + +static void __init class_init(void) +{ + nl_cache_mngt_register(&rtnl_class_ops); +} + +static void __exit class_exit(void) +{ + nl_cache_mngt_unregister(&rtnl_class_ops); +} + +/** @} */ diff --git a/lib/route/class_api.c b/lib/route/class_api.c new file mode 100644 index 0000000..c814486 --- /dev/null +++ b/lib/route/class_api.c @@ -0,0 +1,102 @@ +/* + * lib/route/class_api.c Queueing Classes Module API + * + * 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> + */ + +/** + * @ingroup class + * @defgroup class_api Class Modules + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/route/tc.h> +#include <netlink/route/class.h> +#include <netlink/route/class-modules.h> +#include <netlink/utils.h> + +static struct rtnl_class_ops *class_ops_list; + +/** + * @name Module API + * @{ + */ + +/** + * Register a class module + * @arg cops class module operations + */ +int rtnl_class_register(struct rtnl_class_ops *cops) +{ + struct rtnl_class_ops *o, **op; + + if (!cops->co_kind[0]) + BUG(); + + for (op = &class_ops_list; (o = *op) != NULL; op = &o->co_next) + if (!strcasecmp(cops->co_kind, o->co_kind)) + return nl_errno(EEXIST); + + cops->co_next = NULL; + *op = cops; + + return 0; +} + +/** + * Unregister a class module + * @arg cops class module operations + */ +int rtnl_class_unregister(struct rtnl_class_ops *cops) +{ + struct rtnl_class_ops *o, **op; + + for (op = &class_ops_list; (o = *op) != NULL; op = &o->co_next) + if (!strcasecmp(cops->co_kind, o->co_kind)) + break; + + if (!o) + return nl_errno(ENOENT); + + *op = cops->co_next; + + return 0; +} + +struct rtnl_class_ops *__rtnl_class_lookup_ops(const char *kind) +{ + struct rtnl_class_ops *cops; + + for (cops = class_ops_list; cops; cops = cops->co_next) + if (!strcmp(kind, cops->co_kind)) + return cops; + + return NULL; +} + +/** + * Lookup class operations for a class object + * @arg class Class object. + * + * @return Class operations or NULL if not found. + */ +struct rtnl_class_ops *rtnl_class_lookup_ops(struct rtnl_class *class) +{ + if (!class->c_ops) + class->c_ops = __rtnl_class_lookup_ops(class->c_kind); + + return class->c_ops; +} + + +/** @} */ + +/** @} */ diff --git a/lib/route/class_obj.c b/lib/route/class_obj.c new file mode 100644 index 0000000..0601bdf --- /dev/null +++ b/lib/route/class_obj.c @@ -0,0 +1,287 @@ +/* + * lib/route/class.c Queueing Classes + * + * 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> + */ + +/** + * @ingroup class + * @defgroup class_obj Class Object + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/route/tc.h> +#include <netlink/route/class.h> +#include <netlink/route/class-modules.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/classifier.h> +#include <netlink/utils.h> + +static void class_free_data(struct nl_object *obj) +{ + struct rtnl_class *class = (struct rtnl_class *) obj; + struct rtnl_class_ops *cops; + + tca_free_data((struct rtnl_tca *) class); + + cops = rtnl_class_lookup_ops(class); + if (cops && cops->co_free_data) + cops->co_free_data(class); +} + +static int class_clone(struct nl_object *_dst, struct nl_object *_src) +{ + struct rtnl_class *dst = nl_object_priv(_dst); + struct rtnl_class *src = nl_object_priv(_src); + struct rtnl_class_ops *cops; + int err; + + err = tca_clone((struct rtnl_tca *) dst, (struct rtnl_tca *) src); + if (err < 0) + goto errout; + + cops = rtnl_class_lookup_ops(src); + if (cops && cops->co_clone) + err = cops->co_clone(dst, src); +errout: + return err; +} + +static int class_dump_brief(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_class *class = (struct rtnl_class *) obj; + struct rtnl_class_ops *cops; + + int line = tca_dump_brief((struct rtnl_tca *) class, "class", p, 0); + + cops = rtnl_class_lookup_ops(class); + if (cops && cops->co_dump[NL_DUMP_BRIEF]) + line = cops->co_dump[NL_DUMP_BRIEF](class, p, line); + dp_dump(p, "\n"); + + return line; +} + +static int class_dump_full(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_class *class = (struct rtnl_class *) obj; + struct rtnl_class_ops *cops; + int line; + + line = class_dump_brief(obj, p); + line = tca_dump_full((struct rtnl_tca *) class, p, line); + + if (class->c_info) { + char buf[32]; + dp_dump(p, "child-qdisc %s ", + rtnl_tc_handle2str(class->c_info, buf, sizeof(buf))); + } + + cops = rtnl_class_lookup_ops(class); + if (cops && cops->co_dump[NL_DUMP_FULL]) + line = cops->co_dump[NL_DUMP_FULL](class, p, line); + else if (!class->c_info) + dp_dump(p, "noop (no leaf qdisc)"); + + dp_dump(p, "\n"); + + return line; +} + +static int class_dump_stats(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_class *class = (struct rtnl_class *) obj; + struct rtnl_class_ops *cops; + int line; + + line = class_dump_full(obj, p); + line = tca_dump_stats((struct rtnl_tca *) class, p, line); + dp_dump(p, "\n"); + + cops = rtnl_class_lookup_ops(class); + if (cops && cops->co_dump[NL_DUMP_STATS]) + line = cops->co_dump[NL_DUMP_STATS](class, p, line); + + return line; +} + +/** + * @name Allocation/Freeing + * @{ + */ + +struct rtnl_class *rtnl_class_alloc(void) +{ + return (struct rtnl_class *) nl_object_alloc(&class_obj_ops); +} + +void rtnl_class_put(struct rtnl_class *class) +{ + nl_object_put((struct nl_object *) class); +} + +/** @} */ + +/** + * @name Leaf Qdisc + * @{ + */ + +/** + * Lookup the leaf qdisc of a class + * @arg class the parent class + * @arg cache a qdisc cache including at laest all qdiscs of the + * interface the specified class is attached to + * @return The qdisc from the cache or NULL if the class has no leaf qdisc + */ +struct rtnl_qdisc *rtnl_class_leaf_qdisc(struct rtnl_class *class, + struct nl_cache *cache) +{ + struct rtnl_qdisc *leaf; + + if (!class->c_info) + return NULL; + + leaf = rtnl_qdisc_get_by_parent(cache, class->c_ifindex, + class->c_handle); + if (!leaf || leaf->q_handle != class->c_info) + return NULL; + + return leaf; +} + +/** @} */ + + +/** + * @name Iterators + * @{ + */ + +/** + * Call a callback for each child of a class + * @arg class the parent class + * @arg cache a class cache including all classes of the interface + * the specified class is attached to + * @arg cb callback function + * @arg arg argument to be passed to callback function + */ +void rtnl_class_foreach_child(struct rtnl_class *class, struct nl_cache *cache, + void (*cb)(struct nl_object *, void *), void *arg) +{ + struct rtnl_class *filter; + + filter = rtnl_class_alloc(); + if (!filter) + return; + + rtnl_class_set_parent(filter, class->c_handle); + rtnl_class_set_ifindex(filter, class->c_ifindex); + rtnl_class_set_kind(filter, class->c_kind); + + nl_cache_foreach_filter(cache, (struct nl_object *) filter, cb, arg); + rtnl_class_put(filter); +} + +/** + * Call a callback for each classifier attached to the class + * @arg class the parent class + * @arg cache a filter cache including at least all the filters + * attached to the specified class + * @arg cb callback function + * @arg arg argument to be passed to callback function + */ +void rtnl_class_foreach_cls(struct rtnl_class *class, struct nl_cache *cache, + void (*cb)(struct nl_object *, void *), void *arg) +{ + struct rtnl_cls *filter; + + filter = rtnl_cls_alloc(); + if (!filter) + return; + + rtnl_cls_set_ifindex(filter, class->c_ifindex); + rtnl_cls_set_parent(filter, class->c_parent); + + nl_cache_foreach_filter(cache, (struct nl_object *) filter, cb, arg); + rtnl_cls_put(filter); +} + +/** @} */ + + +/** + * @name Attributes + * @{ + */ + +void rtnl_class_set_ifindex(struct rtnl_class *class, int ifindex) +{ + tca_set_ifindex((struct rtnl_tca *) class, ifindex); +} + +int rtnl_class_get_ifindex(struct rtnl_class *class) +{ + return tca_get_ifindex((struct rtnl_tca *) class); +} + +void rtnl_class_set_handle(struct rtnl_class *class, uint32_t handle) +{ + tca_set_handle((struct rtnl_tca *) class, handle); +} + +uint32_t rtnl_class_get_handle(struct rtnl_class *class) +{ + return tca_get_handle((struct rtnl_tca *) class); +} + +void rtnl_class_set_parent(struct rtnl_class *class, uint32_t parent) +{ + tca_set_parent((struct rtnl_tca *) class, parent); +} + +uint32_t rtnl_class_get_parent(struct rtnl_class *class) +{ + return tca_get_parent((struct rtnl_tca *) class); +} + +void rtnl_class_set_kind(struct rtnl_class *class, const char *name) +{ + tca_set_kind((struct rtnl_tca *) class, name); + class->c_ops = __rtnl_class_lookup_ops(name); +} + +char *rtnl_class_get_kind(struct rtnl_class *class) +{ + return tca_get_kind((struct rtnl_tca *) class); +} + +uint64_t rtnl_class_get_stat(struct rtnl_class *class, + enum rtnl_tc_stats_id id) +{ + return tca_get_stat((struct rtnl_tca *) class, id); +} + +/** @} */ + +struct nl_object_ops class_obj_ops = { + .oo_name = "route/class", + .oo_size = sizeof(struct rtnl_class), + .oo_free_data = class_free_data, + .oo_clone = class_clone, + .oo_dump[NL_DUMP_BRIEF] = class_dump_brief, + .oo_dump[NL_DUMP_FULL] = class_dump_full, + .oo_dump[NL_DUMP_STATS] = class_dump_stats, + .oo_compare = tca_compare, + .oo_id_attrs = (TCA_ATTR_IFINDEX | TCA_ATTR_HANDLE), +}; + +/** @} */ diff --git a/lib/route/classifier.c b/lib/route/classifier.c new file mode 100644 index 0000000..7ad4722 --- /dev/null +++ b/lib/route/classifier.c @@ -0,0 +1,343 @@ +/* + * lib/route/classifier.c Classifier + * + * 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> + */ + +/** + * @ingroup tc + * @defgroup cls Classifiers + * + * @par Classifier Identification + * - protocol + * - priority + * - parent + * - interface + * - kind + * - handle + * + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/tc.h> +#include <netlink/route/classifier.h> +#include <netlink/route/classifier-modules.h> +#include <netlink/route/link.h> + +static struct nl_cache_ops rtnl_cls_ops; + +static int cls_msg_parser(struct nl_cache_ops *ops, struct sockaddr_nl *who, + struct nlmsghdr *nlh, void *arg) +{ + int err; + struct nl_parser_param *pp = arg; + struct rtnl_cls *cls; + struct rtnl_cls_ops *cops; + + cls = rtnl_cls_alloc(); + if (!cls) { + err = nl_errno(ENOMEM); + goto errout; + } + cls->ce_msgtype = nlh->nlmsg_type; + + err = tca_msg_parser(nlh, (struct rtnl_tca *) cls); + if (err < 0) + goto errout_free; + + cls->c_prio = TC_H_MAJ(cls->c_info) >> 16; + cls->c_protocol = ntohs(TC_H_MIN(cls->c_info)); + + cops = rtnl_cls_lookup_ops(cls); + if (cops && cops->co_msg_parser) { + err = cops->co_msg_parser(cls); + if (err < 0) + goto errout_free; + } + + err = pp->pp_cb((struct nl_object *) cls, pp); + if (err < 0) + goto errout_free; + + return P_ACCEPT; + +errout_free: + rtnl_cls_put(cls); +errout: + return err; +} + +static int cls_request_update(struct nl_cache *cache, struct nl_handle *handle) +{ + struct tcmsg tchdr = { + .tcm_family = AF_UNSPEC, + .tcm_ifindex = cache->c_iarg1, + .tcm_parent = cache->c_iarg2, + }; + + return nl_send_simple(handle, RTM_GETTFILTER, NLM_F_DUMP, &tchdr, + sizeof(tchdr)); +} + + +static struct nl_msg *cls_build(struct rtnl_cls *cls, int type, int flags) +{ + struct nl_msg *msg; + struct rtnl_cls_ops *cops; + int err, prio, proto; + struct tcmsg *tchdr; + + msg = tca_build_msg((struct rtnl_tca *) cls, type, flags); + if (!msg) + goto errout; + + tchdr = nlmsg_data(nlmsg_hdr(msg)); + prio = rtnl_cls_get_prio(cls); + proto = rtnl_cls_get_protocol(cls); + tchdr->tcm_info = TC_H_MAKE(prio << 16, htons(proto)), + + cops = rtnl_cls_lookup_ops(cls); + if (cops && cops->co_get_opts) { + struct nl_msg *opts; + + opts = cops->co_get_opts(cls); + if (opts) { + err = nla_put_nested(msg, TCA_OPTIONS, opts); + nlmsg_free(opts); + if (err < 0) + goto errout; + } + } + + return msg; +errout: + nlmsg_free(msg); + return NULL; +} + +/** + * @name Classifier Addition/Modification/Deletion + * @{ + */ + +/** + * Build a netlink message to add a new classifier + * @arg cls classifier to add + * @arg flags additional netlink message flags + * + * Builds a new netlink message requesting an addition of a classifier + * The netlink message header isn't fully equipped with all relevant + * fields and must be sent out via nl_send_auto_complete() or + * supplemented as needed. \a classifier must contain the attributes of + * the new classifier set via \c rtnl_cls_set_* functions. \a opts + * may point to the clsasifier specific options. + * + * @return New netlink message + */ +struct nl_msg * rtnl_cls_build_add_request(struct rtnl_cls *cls, int flags) +{ + return cls_build(cls, RTM_NEWTFILTER, NLM_F_CREATE | flags); +} + +/** + * Add a new classifier + * @arg handle netlink handle + * @arg cls classifier to add + * @arg flags additional netlink message flags + * + * Builds a netlink message by calling rtnl_cls_build_add_request(), + * sends the request to the kernel and waits for the next ACK to be + * received and thus blocks until the request has been processed. + * + * @return 0 on sucess or a negative error if an error occured. + */ +int rtnl_cls_add(struct nl_handle *handle, struct rtnl_cls *cls, int flags) +{ + int err; + struct nl_msg *msg; + + msg = rtnl_cls_build_add_request(cls, flags); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** + * Build a netlink message to change classifier attributes + * @arg cls classifier to change + * @arg flags additional netlink message flags + * + * Builds a new netlink message requesting a change of a neigh + * attributes. The netlink message header isn't fully equipped with + * all relevant fields and must thus be sent out via nl_send_auto_complete() + * or supplemented as needed. + * + * @return The netlink message + */ +struct nl_msg *rtnl_cls_build_change_request(struct rtnl_cls *cls, int flags) +{ + return cls_build(cls, RTM_NEWTFILTER, NLM_F_REPLACE | flags); +} + +/** + * Change a classifier + * @arg handle netlink handle + * @arg cls classifier to change + * @arg flags additional netlink message flags + * + * Builds a netlink message by calling rtnl_cls_build_change_request(), + * sends the request to the kernel and waits for the next ACK to be + * received and thus blocks until the request has been processed. + * + * @return 0 on sucess or a negative error if an error occured. + */ +int rtnl_cls_change(struct nl_handle *handle, struct rtnl_cls *cls, + int flags) +{ + int err; + struct nl_msg *msg; + + msg = rtnl_cls_build_change_request(cls, flags); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** + * Build a netlink request message to delete a classifier + * @arg cls classifier to delete + * @arg flags additional netlink message flags + * + * Builds a new netlink message requesting a deletion of a classifier. + * The netlink message header isn't fully equipped with all relevant + * fields and must thus be sent out via nl_send_auto_complete() + * or supplemented as needed. + * + * @return New netlink message + */ +struct nl_msg *rtnl_cls_build_delete_request(struct rtnl_cls *cls, int flags) +{ + return cls_build(cls, RTM_DELTFILTER, flags); +} + + +/** + * Delete a classifier + * @arg handle netlink handle + * @arg cls classifier to delete + * @arg flags additional netlink message flags + * + * Builds a netlink message by calling rtnl_cls_build_delete_request(), + * sends the request to the kernel and waits for the next ACK to be + * received and thus blocks until the request has been processed. + * + * @return 0 on sucess or a negative error if an error occured. + */ +int rtnl_cls_delete(struct nl_handle *handle, struct rtnl_cls *cls, int flags) +{ + int err; + struct nl_msg *msg; + + msg = rtnl_cls_build_delete_request(cls, flags); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name Cache Management + * @{ + */ + +/** + * Build a classifier cache including all classifiers attached to the + * specified class/qdisc on eht specified interface. + * @arg handle netlink handle + * @arg ifindex interface index of the link the classes are + * attached to. + * @arg parent parent qdisc/class + * + * Allocates a new cache, initializes it properly and updates it to + * include all classes attached to the specified interface. + * + * @note The caller is responsible for destroying and freeing the + * cache after using it. + * @return The cache or NULL if an error has occured. + */ +struct nl_cache *rtnl_cls_alloc_cache(struct nl_handle *handle, + int ifindex, uint32_t parent) +{ + struct nl_cache * cache; + + cache = nl_cache_alloc(&rtnl_cls_ops); + if (cache == NULL) + return NULL; + + cache->c_iarg1 = ifindex; + cache->c_iarg2 = parent; + + if (handle && nl_cache_refill(handle, cache) < 0) { + nl_cache_free(cache); + return NULL; + } + + return cache; +} + +/** @} */ + +static struct nl_cache_ops rtnl_cls_ops = { + .co_name = "route/cls", + .co_hdrsize = sizeof(struct tcmsg), + .co_msgtypes = { + { RTM_NEWTFILTER, NL_ACT_NEW, "new" }, + { RTM_DELTFILTER, NL_ACT_DEL, "del" }, + { RTM_GETTFILTER, NL_ACT_GET, "get" }, + END_OF_MSGTYPES_LIST, + }, + .co_protocol = NETLINK_ROUTE, + .co_request_update = cls_request_update, + .co_msg_parser = cls_msg_parser, + .co_obj_ops = &cls_obj_ops, +}; + +static void __init cls_init(void) +{ + nl_cache_mngt_register(&rtnl_cls_ops); +} + +static void __exit cls_exit(void) +{ + nl_cache_mngt_unregister(&rtnl_cls_ops); +} + +/** @} */ diff --git a/lib/route/cls/fw.c b/lib/route/cls/fw.c new file mode 100644 index 0000000..7ca7619 --- /dev/null +++ b/lib/route/cls/fw.c @@ -0,0 +1,251 @@ +/* + * lib/route/cls/fw.c fw classifier + * + * 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> + * Copyright (c) 2006 Petr Gotthard <petr.gotthard@siemens.com> + * Copyright (c) 2006 Siemens AG Oesterreich + */ + +/** + * @ingroup cls_api + * @defgroup fw Firewall Classifier + * + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/route/classifier.h> +#include <netlink/route/classifier-modules.h> +#include <netlink/route/cls/fw.h> + +/** @cond SKIP */ +#define FW_ATTR_CLASSID 0x001 +#define FW_ATTR_ACTION 0x002 +#define FW_ATTR_POLICE 0x004 +#define FW_ATTR_INDEV 0x008 +/** @endcond */ + +static inline struct rtnl_fw *fw_cls(struct rtnl_cls *cls) +{ + return (struct rtnl_fw *) cls->c_subdata; +} + +static inline struct rtnl_fw *fw_alloc(struct rtnl_cls *cls) +{ + if (!cls->c_subdata) + cls->c_subdata = calloc(1, sizeof(struct rtnl_fw)); + + return fw_cls(cls); +} + +static struct nla_policy fw_policy[TCA_FW_MAX+1] = { + [TCA_FW_CLASSID] = { .type = NLA_U32 }, + [TCA_FW_INDEV] = { .type = NLA_STRING, + .maxlen = IFNAMSIZ }, +}; + +static int fw_msg_parser(struct rtnl_cls *cls) +{ + int err; + struct nlattr *tb[TCA_FW_MAX + 1]; + struct rtnl_fw *f; + + err = tca_parse(tb, TCA_FW_MAX, (struct rtnl_tca *) cls, fw_policy); + if (err < 0) + return err; + + f = fw_alloc(cls); + if (!f) + goto errout_nomem; + + if (tb[TCA_FW_CLASSID]) { + f->cf_classid = nla_get_u32(tb[TCA_FW_CLASSID]); + f->cf_mask |= FW_ATTR_CLASSID; + } + + if (tb[TCA_FW_ACT]) { + f->cf_act = nla_get_data(tb[TCA_FW_ACT]); + if (!f->cf_act) + goto errout_nomem; + f->cf_mask |= FW_ATTR_ACTION; + } + + if (tb[TCA_FW_POLICE]) { + f->cf_police = nla_get_data(tb[TCA_FW_POLICE]); + if (!f->cf_police) + goto errout_nomem; + f->cf_mask |= FW_ATTR_POLICE; + } + + if (tb[TCA_FW_INDEV]) { + nla_strlcpy(f->cf_indev, tb[TCA_FW_INDEV], IFNAMSIZ); + f->cf_mask |= FW_ATTR_INDEV; + } + + return 0; + +errout_nomem: + err = nl_errno(ENOMEM); + + return err; +} + +static void fw_free_data(struct rtnl_cls *cls) +{ + struct rtnl_fw *f = fw_cls(cls); + + if (!f) + return; + + nl_data_free(f->cf_act); + nl_data_free(f->cf_police); + + free(cls->c_subdata); +} + +static int fw_clone(struct rtnl_cls *_dst, struct rtnl_cls *_src) +{ + struct rtnl_fw *dst, *src = fw_cls(_src); + + if (!src) + return 0; + + dst = fw_alloc(_dst); + if (!dst) + return nl_errno(ENOMEM); + + if (src->cf_act) + if (!(dst->cf_act = nl_data_clone(src->cf_act))) + goto errout; + + if (src->cf_police) + if (!(dst->cf_police = nl_data_clone(src->cf_police))) + goto errout; + + return 0; +errout: + return nl_get_errno(); +} + +static int fw_dump_brief(struct rtnl_cls *cls, struct nl_dump_params *p, + int line) +{ + struct rtnl_fw *f = fw_cls(cls); + char buf[32]; + + if (!f) + goto ignore; + + if (f->cf_mask & FW_ATTR_CLASSID) + dp_dump(p, " target %s", + rtnl_tc_handle2str(f->cf_classid, buf, sizeof(buf))); + +ignore: + return line; +} + +static int fw_dump_full(struct rtnl_cls *cls, struct nl_dump_params *p, + int line) +{ + struct rtnl_fw *f = fw_cls(cls); + + if (!f) + goto ignore; + + if (f->cf_mask & FW_ATTR_INDEV) + dp_dump(p, "indev %s ", f->cf_indev); + +ignore: + return line; +} + +static int fw_dump_stats(struct rtnl_cls *cls, struct nl_dump_params *p, + int line) +{ + struct rtnl_fw *f = fw_cls(cls); + + if (!f) + goto ignore; + +ignore: + return line; +} + +static struct nl_msg *fw_get_opts(struct rtnl_cls *cls) +{ + struct rtnl_fw *f; + struct nl_msg *msg; + + f = fw_cls(cls); + if (!f) + return NULL; + + msg = nlmsg_alloc(); + if (!msg) + return NULL; + + if (f->cf_mask & FW_ATTR_CLASSID) + nla_put_u32(msg, TCA_FW_CLASSID, f->cf_classid); + + if (f->cf_mask & FW_ATTR_ACTION) + nla_put_data(msg, TCA_FW_ACT, f->cf_act); + + if (f->cf_mask & FW_ATTR_POLICE) + nla_put_data(msg, TCA_FW_POLICE, f->cf_police); + + if (f->cf_mask & FW_ATTR_INDEV) + nla_put_string(msg, TCA_FW_INDEV, f->cf_indev); + + return msg; +} + +/** + * @name Attribute Modifications + * @{ + */ + +int rtnl_fw_set_classid(struct rtnl_cls *cls, uint32_t classid) +{ + struct rtnl_fw *f; + + f = fw_alloc(cls); + if (!f) + return nl_errno(ENOMEM); + + f->cf_classid = classid; + f->cf_mask |= FW_ATTR_CLASSID; + + return 0; +} + +/** @} */ + +static struct rtnl_cls_ops fw_ops = { + .co_kind = "fw", + .co_msg_parser = fw_msg_parser, + .co_free_data = fw_free_data, + .co_clone = fw_clone, + .co_get_opts = fw_get_opts, + .co_dump[NL_DUMP_BRIEF] = fw_dump_brief, + .co_dump[NL_DUMP_FULL] = fw_dump_full, + .co_dump[NL_DUMP_STATS] = fw_dump_stats, +}; + +static void __init fw_init(void) +{ + rtnl_cls_register(&fw_ops); +} + +static void __exit fw_exit(void) +{ + rtnl_cls_unregister(&fw_ops); +} + +/** @} */ diff --git a/lib/route/cls/police.c b/lib/route/cls/police.c new file mode 100644 index 0000000..051c6b2 --- /dev/null +++ b/lib/route/cls/police.c @@ -0,0 +1,67 @@ +/* + * lib/route/cls/police.c Policer + * + * 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 <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/tc.h> +#include <netlink/route/classifier.h> +#include <netlink/route/classifier-modules.h> +#include <netlink/route/cls/police.h> + +/** + * @name Policer Type + * @{ + */ + +static struct trans_tbl police_types[] = { + __ADD(TC_POLICE_UNSPEC,unspec) + __ADD(TC_POLICE_OK,ok) + __ADD(TC_POLICE_RECLASSIFY,reclassify) + __ADD(TC_POLICE_SHOT,shot) +#ifdef TC_POLICE_PIPE + __ADD(TC_POLICE_PIPE,pipe) +#endif +}; + +/** + * Transform a policer type number into a character string (Reentrant). + * @arg type policer type + * @arg buf destination buffer + * @arg len buffer length + * + * Transforms a policer type number into a character string and stores + * it in the provided buffer. + * + * @return The destination buffer or the type encoded in hex if no match was found. + */ +char * nl_police2str(int type, char *buf, size_t len) +{ + return __type2str(type, buf, len, police_types, + ARRAY_SIZE(police_types)); +} + +/** + * Transform a character string into a policer type number + * @arg name policer type name + * + * Transform the provided character string specifying a policer + * type into the corresponding numeric value + * + * @return Policer type number or a negative value. + */ +int nl_str2police(const char *name) +{ + return __str2type(name, police_types, ARRAY_SIZE(police_types)); +} + +/** @} */ diff --git a/lib/route/cls/u32.c b/lib/route/cls/u32.c new file mode 100644 index 0000000..596e63f --- /dev/null +++ b/lib/route/cls/u32.c @@ -0,0 +1,602 @@ +/* + * lib/route/cls/u32.c u32 classifier + * + * 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> + * Copyright (c) 2005-2006 Petr Gotthard <petr.gotthard@siemens.com> + * Copyright (c) 2005-2006 Siemens AG Oesterreich + */ + +/** + * @ingroup cls_api + * @defgroup u32 Universal 32-bit Classifier + * + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/attr.h> +#include <netlink/utils.h> +#include <netlink/route/tc.h> +#include <netlink/route/classifier.h> +#include <netlink/route/classifier-modules.h> +#include <netlink/route/cls/u32.h> + +/** @cond SKIP */ +#define U32_ATTR_DIVISOR 0x001 +#define U32_ATTR_HASH 0x002 +#define U32_ATTR_CLASSID 0x004 +#define U32_ATTR_LINK 0x008 +#define U32_ATTR_PCNT 0x010 +#define U32_ATTR_SELECTOR 0x020 +#define U32_ATTR_ACTION 0x040 +#define U32_ATTR_POLICE 0x080 +#define U32_ATTR_INDEV 0x100 +/** @endcond */ + +static inline struct rtnl_u32 *u32_cls(struct rtnl_cls *cls) +{ + return (struct rtnl_u32 *) cls->c_subdata; +} + +static inline struct rtnl_u32 *u32_alloc(struct rtnl_cls *cls) +{ + if (!cls->c_subdata) + cls->c_subdata = calloc(1, sizeof(struct rtnl_u32)); + + return u32_cls(cls); +} + +static inline struct tc_u32_sel *u32_selector(struct rtnl_u32 *u) +{ + return (struct tc_u32_sel *) u->cu_selector->d_data; +} + +static inline struct tc_u32_sel *u32_selector_alloc(struct rtnl_u32 *u) +{ + if (!u->cu_selector) + u->cu_selector = nl_data_alloc(NULL, sizeof(struct tc_u32_sel)); + + return u32_selector(u); +} + +static struct nla_policy u32_policy[TCA_U32_MAX+1] = { + [TCA_U32_DIVISOR] = { .type = NLA_U32 }, + [TCA_U32_HASH] = { .type = NLA_U32 }, + [TCA_U32_CLASSID] = { .type = NLA_U32 }, + [TCA_U32_LINK] = { .type = NLA_U32 }, + [TCA_U32_INDEV] = { .type = NLA_STRING, + .maxlen = IFNAMSIZ }, + [TCA_U32_SEL] = { .minlen = sizeof(struct tc_u32_sel) }, + [TCA_U32_PCNT] = { .minlen = sizeof(struct tc_u32_pcnt) }, +}; + +static int u32_msg_parser(struct rtnl_cls *cls) +{ + int err; + struct nlattr *tb[TCA_U32_MAX + 1]; + struct rtnl_u32 *u; + + err = tca_parse(tb, TCA_U32_MAX, (struct rtnl_tca *) cls, u32_policy); + if (err < 0) + return err; + + u = u32_alloc(cls); + if (!u) + goto errout_nomem; + + if (tb[TCA_U32_DIVISOR]) { + u->cu_divisor = nla_get_u32(tb[TCA_U32_DIVISOR]); + u->cu_mask |= U32_ATTR_DIVISOR; + } + + if (tb[TCA_U32_SEL]) { + u->cu_selector = nla_get_data(tb[TCA_U32_SEL]); + if (!u->cu_selector) + goto errout_nomem; + u->cu_mask |= U32_ATTR_SELECTOR; + } + + if (tb[TCA_U32_HASH]) { + u->cu_hash = nla_get_u32(tb[TCA_U32_HASH]); + u->cu_mask |= U32_ATTR_HASH; + } + + if (tb[TCA_U32_CLASSID]) { + u->cu_classid = nla_get_u32(tb[TCA_U32_CLASSID]); + u->cu_mask |= U32_ATTR_CLASSID; + } + + if (tb[TCA_U32_LINK]) { + u->cu_link = nla_get_u32(tb[TCA_U32_LINK]); + u->cu_mask |= U32_ATTR_LINK; + } + + if (tb[TCA_U32_ACT]) { + u->cu_act = nla_get_data(tb[TCA_U32_ACT]); + if (!u->cu_act) + goto errout_nomem; + u->cu_mask |= U32_ATTR_ACTION; + } + + if (tb[TCA_U32_POLICE]) { + u->cu_police = nla_get_data(tb[TCA_U32_POLICE]); + if (!u->cu_police) + goto errout_nomem; + u->cu_mask |= U32_ATTR_POLICE; + } + + if (tb[TCA_U32_PCNT]) { + struct tc_u32_sel *sel; + int pcnt_size; + + if (!tb[TCA_U32_SEL]) { + err = nl_error(EINVAL, "Missing TCA_U32_SEL required " + "for TCA_U32_PCNT"); + goto errout; + } + + sel = u->cu_selector->d_data; + pcnt_size = sizeof(struct tc_u32_pcnt) + + (sel->nkeys * sizeof(uint64_t)); + if (nla_len(tb[TCA_U32_PCNT]) < pcnt_size) { + err = nl_error(EINVAL, "Invalid size for TCA_U32_PCNT"); + goto errout; + } + + u->cu_pcnt = nla_get_data(tb[TCA_U32_PCNT]); + if (!u->cu_pcnt) + goto errout_nomem; + u->cu_mask |= U32_ATTR_PCNT; + } + + if (tb[TCA_U32_INDEV]) { + nla_strlcpy(u->cu_indev, tb[TCA_U32_INDEV], IFNAMSIZ); + u->cu_mask |= U32_ATTR_INDEV; + } + + return 0; + +errout_nomem: + err = nl_errno(ENOMEM); +errout: + return err; +} + +static void u32_free_data(struct rtnl_cls *cls) +{ + struct rtnl_u32 *u = u32_cls(cls); + + if (!u) + return; + + nl_data_free(u->cu_selector); + nl_data_free(u->cu_act); + nl_data_free(u->cu_police); + nl_data_free(u->cu_pcnt); + + free(cls->c_subdata); +} + +static int u32_clone(struct rtnl_cls *_dst, struct rtnl_cls *_src) +{ + struct rtnl_u32 *dst, *src = u32_cls(_src); + + if (!src) + return 0; + + dst = u32_alloc(_dst); + if (!dst) + return nl_errno(ENOMEM); + + if (src->cu_selector) + if (!(dst->cu_selector = nl_data_clone(src->cu_selector))) + goto errout; + + if (src->cu_act) + if (!(dst->cu_act = nl_data_clone(src->cu_act))) + goto errout; + + if (src->cu_police) + if (!(dst->cu_police = nl_data_clone(src->cu_police))) + goto errout; + + if (src->cu_pcnt) + if (!(dst->cu_pcnt = nl_data_clone(src->cu_pcnt))) + goto errout; + + return 0; +errout: + return nl_get_errno(); +} + +static int u32_dump_brief(struct rtnl_cls *cls, struct nl_dump_params *p, + int line) +{ + struct rtnl_u32 *u = u32_cls(cls); + char buf[32]; + + if (!u) + goto ignore; + + if (u->cu_mask & U32_ATTR_DIVISOR) + dp_dump(p, " divisor %u", u->cu_divisor); + else if (u->cu_mask & U32_ATTR_CLASSID) + dp_dump(p, " target %s", + rtnl_tc_handle2str(u->cu_classid, buf, sizeof(buf))); + +ignore: + return line; +} + +static int print_selector(struct nl_dump_params *p, struct tc_u32_sel *sel, + struct rtnl_cls *cls, struct rtnl_u32 *u, int line) +{ + int i; + struct tc_u32_key *key; + + if (sel->hmask || sel->hoff) { + /* I guess this will never be used since the kernel only + * exports the selector if no divisor is set but hash offset + * and hash mask make only sense in hash filters with divisor + * set */ + dp_dump(p, " hash at %u & 0x%x", sel->hoff, sel->hmask); + } + + if (sel->flags & (TC_U32_OFFSET | TC_U32_VAROFFSET)) { + dp_dump(p, " offset at %u", sel->off); + + if (sel->flags & TC_U32_VAROFFSET) + dp_dump(p, " variable (at %u & 0x%x) >> %u", + sel->offoff, ntohs(sel->offmask), sel->offshift); + } + + if (sel->flags) { + int flags = sel->flags; + dp_dump(p, " <"); + +#define PRINT_FLAG(f) if (flags & TC_U32_##f) { \ + flags &= ~TC_U32_##f; dp_dump(p, #f "%s", flags ? "," : ""); } + + PRINT_FLAG(TERMINAL); + PRINT_FLAG(OFFSET); + PRINT_FLAG(VAROFFSET); + PRINT_FLAG(EAT); +#undef PRINT_FLAG + + dp_dump(p, ">"); + } + + + for (i = 0; i < sel->nkeys; i++) { + key = (struct tc_u32_key *) ((char *) sel + sizeof(*sel)) + i; + + dp_dump(p, "\n"); + dp_dump_line(p, line++, " match key at %s%u ", + key->offmask ? "nexthdr+" : "", key->off); + + if (key->offmask) + dp_dump(p, "[0x%u] ", key->offmask); + + dp_dump(p, "& 0x%08x == 0x%08x", ntohl(key->mask), ntohl(key->val)); + + if (p->dp_type == NL_DUMP_STATS && + (u->cu_mask & U32_ATTR_PCNT)) { + struct tc_u32_pcnt *pcnt = u->cu_pcnt->d_data; + dp_dump(p, " successful %" PRIu64, pcnt->kcnts[i]); + } + } + + return line; +} + + +static int u32_dump_full(struct rtnl_cls *cls, struct nl_dump_params *p, + int line) +{ + struct rtnl_u32 *u = u32_cls(cls); + struct tc_u32_sel *s; + + if (!u) + goto ignore; + + if (!(u->cu_mask & U32_ATTR_SELECTOR)) { + dp_dump(p, "no-selector\n"); + return line; + } + + s = u->cu_selector->d_data; + + dp_dump(p, "nkeys %u ", s->nkeys); + + if (u->cu_mask & U32_ATTR_HASH) + dp_dump(p, "ht key 0x%x hash 0x%u", + TC_U32_USERHTID(u->cu_hash), TC_U32_HASH(u->cu_hash)); + + if (u->cu_mask & U32_ATTR_LINK) + dp_dump(p, "link %u ", u->cu_link); + + if (u->cu_mask & U32_ATTR_INDEV) + dp_dump(p, "indev %s ", u->cu_indev); + + line = print_selector(p, s, cls, u, line); + dp_dump(p, "\n"); + +ignore: + return line; + +#if 0 +#define U32_ATTR_ACTION 0x040 +#define U32_ATTR_POLICE 0x080 + + struct nl_data act; + struct nl_data police; +#endif +} + +static int u32_dump_stats(struct rtnl_cls *cls, struct nl_dump_params *p, + int line) +{ + struct rtnl_u32 *u = u32_cls(cls); + + if (!u) + goto ignore; + + if (u->cu_mask & U32_ATTR_PCNT) { + struct tc_u32_pcnt *pc = u->cu_pcnt->d_data; + dp_dump(p, "\n"); + dp_dump_line(p, line++, "%s successful hits\n"); + dp_dump_line(p, line++, "%s %8llu %8llu\n", + pc->rhit, pc->rcnt); + } + +ignore: + return line; +} + +static struct nl_msg *u32_get_opts(struct rtnl_cls *cls) +{ + struct rtnl_u32 *u; + struct nl_msg *msg; + + u = u32_cls(cls); + if (!u) + return NULL; + + msg = nlmsg_alloc(); + if (!msg) + return NULL; + + if (u->cu_mask & U32_ATTR_DIVISOR) + nla_put_u32(msg, TCA_U32_DIVISOR, u->cu_divisor); + + if (u->cu_mask & U32_ATTR_HASH) + nla_put_u32(msg, TCA_U32_HASH, u->cu_hash); + + if (u->cu_mask & U32_ATTR_CLASSID) + nla_put_u32(msg, TCA_U32_CLASSID, u->cu_classid); + + if (u->cu_mask & U32_ATTR_LINK) + nla_put_u32(msg, TCA_U32_LINK, u->cu_link); + + if (u->cu_mask & U32_ATTR_SELECTOR) + nla_put_data(msg, TCA_U32_SEL, u->cu_selector); + + if (u->cu_mask & U32_ATTR_ACTION) + nla_put_data(msg, TCA_U32_ACT, u->cu_act); + + if (u->cu_mask & U32_ATTR_POLICE) + nla_put_data(msg, TCA_U32_POLICE, u->cu_police); + + if (u->cu_mask & U32_ATTR_INDEV) + nla_put_string(msg, TCA_U32_INDEV, u->cu_indev); + + return msg; +} + +/** + * @name Attribute Modifications + * @{ + */ + +void rtnl_u32_set_handle(struct rtnl_cls *cls, int htid, int hash, + int nodeid) +{ + uint32_t handle = (htid << 20) | (hash << 12) | nodeid; + + tca_set_handle((struct rtnl_tca *) cls, handle ); +} + +int rtnl_u32_set_classid(struct rtnl_cls *cls, uint32_t classid) +{ + struct rtnl_u32 *u; + + u = u32_alloc(cls); + if (!u) + return nl_errno(ENOMEM); + + u->cu_classid = classid; + u->cu_mask |= U32_ATTR_CLASSID; + + return 0; +} + +/** @} */ + +/** + * @name Selector Modifications + * @{ + */ + +int rtnl_u32_set_flags(struct rtnl_cls *cls, int flags) +{ + struct tc_u32_sel *sel; + struct rtnl_u32 *u; + + u = u32_alloc(cls); + if (!u) + return nl_errno(ENOMEM); + + sel = u32_selector_alloc(u); + if (!sel) + return nl_errno(ENOMEM); + + sel->flags |= flags; + u->cu_mask |= U32_ATTR_SELECTOR; + + return 0; +} + +/** + * Append new 32-bit key to the selector + * + * @arg cls classifier to be modifier + * @arg val value to be matched (network byte-order) + * @arg mask mask to be applied before matching (network byte-order) + * @arg off offset, in bytes, to start matching + * @arg offmask offset mask + * + * General selectors define the pattern, mask and offset the pattern will be + * matched to the packet contents. Using the general selectors you can match + * virtually any single bit in the IP (or upper layer) header. + * +*/ +int rtnl_u32_add_key(struct rtnl_cls *cls, uint32_t val, uint32_t mask, + int off, int offmask) +{ + struct tc_u32_sel *sel; + struct rtnl_u32 *u; + int err; + + u = u32_alloc(cls); + if (!u) + return nl_errno(ENOMEM); + + sel = u32_selector_alloc(u); + if (!sel) + return nl_errno(ENOMEM); + + err = nl_data_append(u->cu_selector, NULL, sizeof(struct tc_u32_key)); + if (err < 0) + return err; + + /* the selector might have been moved by realloc */ + sel = u32_selector(u); + + sel->keys[sel->nkeys].mask = mask; + sel->keys[sel->nkeys].val = val & mask; + sel->keys[sel->nkeys].off = off; + sel->keys[sel->nkeys].offmask = offmask; + sel->nkeys++; + u->cu_mask |= U32_ATTR_SELECTOR; + + return 0; +} + +int rtnl_u32_add_key_uint8(struct rtnl_cls *cls, uint8_t val, uint8_t mask, + int off, int offmask) +{ + int shift = 24 - 8 * (off & 3); + + return rtnl_u32_add_key(cls, htonl((uint32_t)val << shift), + htonl((uint32_t)mask << shift), + off & ~3, offmask); +} + +/** + * Append new selector key to match a 16-bit number + * + * @arg cls classifier to be modified + * @arg val value to be matched (host byte-order) + * @arg mask mask to be applied before matching (host byte-order) + * @arg off offset, in bytes, to start matching + * @arg offmask offset mask +*/ +int rtnl_u32_add_key_uint16(struct rtnl_cls *cls, uint16_t val, uint16_t mask, + int off, int offmask) +{ + int shift = ((off & 3) == 0 ? 16 : 0); + if (off % 2) + return nl_error(EINVAL, "Invalid offset alignment"); + + return rtnl_u32_add_key(cls, htonl((uint32_t)val << shift), + htonl((uint32_t)mask << shift), + off & ~3, offmask); +} + +/** + * Append new selector key to match a 32-bit number + * + * @arg cls classifier to be modified + * @arg val value to be matched (host byte-order) + * @arg mask mask to be applied before matching (host byte-order) + * @arg off offset, in bytes, to start matching + * @arg offmask offset mask +*/ +int rtnl_u32_add_key_uint32(struct rtnl_cls *cls, uint32_t val, uint32_t mask, + int off, int offmask) +{ + return rtnl_u32_add_key(cls, htonl(val), htonl(mask), + off & ~3, offmask); +} + +int rtnl_u32_add_key_in_addr(struct rtnl_cls *cls, struct in_addr *addr, + uint8_t bitmask, int off, int offmask) +{ + uint32_t mask = 0xFFFFFFFF << (32 - bitmask); + return rtnl_u32_add_key(cls, addr->s_addr, htonl(mask), off, offmask); +} + +int rtnl_u32_add_key_in6_addr(struct rtnl_cls *cls, struct in6_addr *addr, + uint8_t bitmask, int off, int offmask) +{ + int i, err; + + for (i = 1; i <= 4; i++) { + if (32 * i - bitmask <= 0) { + if ((err = rtnl_u32_add_key(cls, addr->s6_addr32[i-1], + 0xFFFFFFFF, off+4*(i-1), offmask)) < 0) + return err; + } + else if (32 * i - bitmask < 32) { + uint32_t mask = 0xFFFFFFFF << (32 * i - bitmask); + if ((err = rtnl_u32_add_key(cls, addr->s6_addr32[i-1], + htonl(mask), off+4*(i-1), offmask)) < 0) + return err; + } + /* otherwise, if (32*i - bitmask >= 32) no key is generated */ + } + + return 0; +} + +/** @} */ + +static struct rtnl_cls_ops u32_ops = { + .co_kind = "u32", + .co_msg_parser = u32_msg_parser, + .co_free_data = u32_free_data, + .co_clone = u32_clone, + .co_get_opts = u32_get_opts, + .co_dump[NL_DUMP_BRIEF] = u32_dump_brief, + .co_dump[NL_DUMP_FULL] = u32_dump_full, + .co_dump[NL_DUMP_STATS] = u32_dump_stats, +}; + +static void __init u32_init(void) +{ + rtnl_cls_register(&u32_ops); +} + +static void __exit u32_exit(void) +{ + rtnl_cls_unregister(&u32_ops); +} + +/** @} */ diff --git a/lib/route/cls_api.c b/lib/route/cls_api.c new file mode 100644 index 0000000..f5a083a --- /dev/null +++ b/lib/route/cls_api.c @@ -0,0 +1,103 @@ +/* + * lib/route/cls_api.c Classifier Module API + * + * 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> + */ + +/** + * @ingroup cls + * @defgroup cls_api Classifier Modules + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/tc.h> +#include <netlink/route/classifier.h> +#include <netlink/route/classifier-modules.h> +#include <netlink/route/link.h> + +static struct rtnl_cls_ops *cls_ops_list; + +/** + * @name Classifier Module API + * @{ + */ + +/** + * Register a classifier module + * @arg cops classifier module operations + */ +int rtnl_cls_register(struct rtnl_cls_ops *cops) +{ + struct rtnl_cls_ops *o, **op; + + if (!cops->co_kind) + BUG(); + + for (op = &cls_ops_list; (o = *op) != NULL; op = &o->co_next) + if (!strcasecmp(cops->co_kind, o->co_kind)) + return nl_errno(EEXIST); + + cops->co_next = NULL; + *op = cops; + + return 0; +} + +/** + * Unregister a classifier module + * @arg cops classifier module operations + */ +int rtnl_cls_unregister(struct rtnl_cls_ops *cops) +{ + struct rtnl_cls_ops *o, **op; + + for (op = &cls_ops_list; (o = *op) != NULL; op = &o->co_next) + if (!strcasecmp(cops->co_kind, o->co_kind)) + break; + + if (!o) + return nl_errno(ENOENT); + + *op = cops->co_next; + + return 0; +} + +struct rtnl_cls_ops *__rtnl_cls_lookup_ops(const char *kind) +{ + struct rtnl_cls_ops *cops; + + for (cops = cls_ops_list; cops; cops = cops->co_next) + if (!strcmp(kind, cops->co_kind)) + return cops; + + return NULL; +} + +/** + * Lookup classifier operations for a classifier object + * @arg cls Classifier object. + * + * @return Classifier operations or NULL if not found. + */ +struct rtnl_cls_ops *rtnl_cls_lookup_ops(struct rtnl_cls *cls) +{ + if (!cls->c_ops) + cls->c_ops = __rtnl_cls_lookup_ops(cls->c_kind); + + return cls->c_ops; +} + + +/** @} */ + +/** @} */ diff --git a/lib/route/cls_obj.c b/lib/route/cls_obj.c new file mode 100644 index 0000000..e12bc95 --- /dev/null +++ b/lib/route/cls_obj.c @@ -0,0 +1,203 @@ +/* + * lib/route/cls_api.c Classifier Object + * + * 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> + */ + +/** + * @ingroup cls + * @defgroup cls_obj Classifier Object + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/tc.h> +#include <netlink/route/classifier.h> +#include <netlink/route/classifier-modules.h> +#include <netlink/route/link.h> + +/** @cond SKIP */ +#define CLS_ATTR_PRIO (TCA_ATTR_MAX << 1) +#define CLS_ATTR_PROTOCOL (TCA_ATTR_MAX << 2) +/** @endcond */ + +static void cls_free_data(struct nl_object *obj) +{ + struct rtnl_cls *cls = (struct rtnl_cls *) obj; + struct rtnl_cls_ops *cops; + + tca_free_data((struct rtnl_tca *) cls); + + cops = rtnl_cls_lookup_ops(cls); + if (cops && cops->co_free_data) + cops->co_free_data(cls); +} + +static int cls_clone(struct nl_object *_dst, struct nl_object *_src) +{ + struct rtnl_cls *dst = nl_object_priv(_dst); + struct rtnl_cls *src = nl_object_priv(_src); + struct rtnl_cls_ops *cops; + int err; + + err = tca_clone((struct rtnl_tca *) dst, (struct rtnl_tca *) src); + if (err < 0) + goto errout; + + cops = rtnl_cls_lookup_ops(src); + if (cops && cops->co_clone) + err = cops->co_clone(dst, src); +errout: + return err; +} + +static int cls_dump_brief(struct nl_object *obj, struct nl_dump_params *p) +{ + char buf[32]; + struct rtnl_cls *cls = (struct rtnl_cls *) obj; + struct rtnl_cls_ops *cops; + int line; + + line = tca_dump_brief((struct rtnl_tca *) cls, "cls", p, 0); + + dp_dump(p, " prio %u protocol %s", cls->c_prio, + nl_ether_proto2str(cls->c_protocol, buf, sizeof(buf))); + + cops = rtnl_cls_lookup_ops(cls); + if (cops && cops->co_dump[NL_DUMP_BRIEF]) + line = cops->co_dump[NL_DUMP_BRIEF](cls, p, line); + dp_dump(p, "\n"); + + return line; +} + +static int cls_dump_full(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_cls *cls = (struct rtnl_cls *) obj; + struct rtnl_cls_ops *cops; + int line; + + line = cls_dump_brief(obj, p); + line = tca_dump_full((struct rtnl_tca *) cls, p, line); + + cops = rtnl_cls_lookup_ops(cls); + if (cops && cops->co_dump[NL_DUMP_FULL]) + line = cops->co_dump[NL_DUMP_FULL](cls, p, line); + else + dp_dump(p, "no options\n"); + + return line; +} + +static int cls_dump_stats(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_cls *cls = (struct rtnl_cls *) obj; + struct rtnl_cls_ops *cops; + int line; + + line = cls_dump_full(obj, p); + line = tca_dump_stats((struct rtnl_tca *) cls, p, line); + dp_dump(p, "\n"); + + cops = rtnl_cls_lookup_ops(cls); + if (cops && cops->co_dump[NL_DUMP_STATS]) + line = cops->co_dump[NL_DUMP_STATS](cls, p, line); + + return line; +} + +/** + * @name Allocation/Freeing + * @{ + */ + +struct rtnl_cls *rtnl_cls_alloc(void) +{ + return (struct rtnl_cls *) nl_object_alloc(&cls_obj_ops); +} + +void rtnl_cls_put(struct rtnl_cls *cls) +{ + nl_object_put((struct nl_object *) cls); +} + +/** @} */ + + +/** + * @name Attributes + * @{ + */ + +void rtnl_cls_set_ifindex(struct rtnl_cls *f, int ifindex) +{ + tca_set_ifindex((struct rtnl_tca *) f, ifindex); +} + +void rtnl_cls_set_handle(struct rtnl_cls *f, uint32_t handle) +{ + tca_set_handle((struct rtnl_tca *) f, handle); +} + +void rtnl_cls_set_parent(struct rtnl_cls *f, uint32_t parent) +{ + tca_set_parent((struct rtnl_tca *) f, parent); +} + +void rtnl_cls_set_kind(struct rtnl_cls *f, const char *kind) +{ + tca_set_kind((struct rtnl_tca *) f, kind); + f->c_ops = __rtnl_cls_lookup_ops(kind); +} + +void rtnl_cls_set_prio(struct rtnl_cls *cls, int prio) +{ + cls->c_prio = prio; + cls->ce_mask |= CLS_ATTR_PRIO; +} + +int rtnl_cls_get_prio(struct rtnl_cls *cls) +{ + if (cls->ce_mask & CLS_ATTR_PRIO) + return cls->c_prio; + else + return 0; +} + +void rtnl_cls_set_protocol(struct rtnl_cls *cls, int protocol) +{ + cls->c_protocol = protocol; + cls->ce_mask |= CLS_ATTR_PROTOCOL; +} + +int rtnl_cls_get_protocol(struct rtnl_cls *cls) +{ + if (cls->ce_mask & CLS_ATTR_PROTOCOL) + return cls->c_protocol; + else + return ETH_P_ALL; +} + +/** @} */ + +struct nl_object_ops cls_obj_ops = { + .oo_name = "route/cls", + .oo_size = sizeof(struct rtnl_cls), + .oo_free_data = cls_free_data, + .oo_clone = cls_clone, + .oo_dump[NL_DUMP_BRIEF] = cls_dump_brief, + .oo_dump[NL_DUMP_FULL] = cls_dump_full, + .oo_dump[NL_DUMP_STATS] = cls_dump_stats, + .oo_compare = tca_compare, + .oo_id_attrs = (TCA_ATTR_IFINDEX | TCA_ATTR_HANDLE), +}; + +/** @} */ diff --git a/lib/route/link.c b/lib/route/link.c new file mode 100644 index 0000000..7ecae12 --- /dev/null +++ b/lib/route/link.c @@ -0,0 +1,1313 @@ +/* + * lib/route/link.c Links (Interfaces) + * + * 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> + */ + +/** + * @ingroup rtnl + * @defgroup link Links (Interfaces) + * @brief + * + * @par Link Identification + * A link can be identified by either its interface index or by its + * name. The kernel favours the interface index but falls back to the + * interface name if the interface index is lesser-than 0 for kernels + * >= 2.6.11. Therefore you can request changes without mapping a + * interface name to the corresponding index first. + * + * @par Changeable Attributes + * @anchor link_changeable + * - Link layer address + * - Link layer broadcast address + * - device mapping (ifmap) (>= 2.6.9) + * - MTU (>= 2.6.9) + * - Transmission queue length (>= 2.6.9) + * - Weight (>= 2.6.9) + * - Link name (only via access through interface index) (>= 2.6.9) + * - Flags (>= 2.6.9) + * - IFF_DEBUG + * - IFF_NOTRAILERS + * - IFF_NOARP + * - IFF_DYNAMIC + * - IFF_MULTICAST + * - IFF_PORTSEL + * - IFF_AUTOMEDIA + * - IFF_UP + * - IFF_PROMISC + * - IFF_ALLMULTI + * + * @par Link Flags (linux/if.h) + * @anchor link_flags + * @code + * IFF_UP Status of link (up|down) + * IFF_BROADCAST Indicates this link allows broadcasting + * IFF_MULTICAST Indicates this link allows multicasting + * IFF_ALLMULTI Indicates this link is doing multicast routing + * IFF_DEBUG Tell the driver to do debugging (currently unused) + * IFF_LOOPBACK This is the loopback link + * IFF_POINTOPOINT Point-to-point link + * IFF_NOARP Link is unable to perform ARP + * IFF_PROMISC Status of promiscious mode flag + * IFF_MASTER Used by teql + * IFF_SLAVE Used by teql + * IFF_PORTSEL Indicates this link allows port selection + * IFF_AUTOMEDIA Indicates this link selects port automatically + * IFF_DYNAMIC Indicates the address of this link is dynamic + * IFF_RUNNING Link is running and carrier is ok. + * IFF_NOTRAILERS Unused, BSD compat. + * @endcode + * + * @par Notes on IFF_PROMISC and IFF_ALLMULTI flags + * Although you can query the status of IFF_PROMISC and IFF_ALLMULTI + * they do not represent the actual state in the kernel but rather + * whether the flag has been enabled/disabled by userspace. The link + * may be in promiscious mode even if IFF_PROMISC is not set in a link + * dump request response because promiscity might be needed by the driver + * for a period of time. + * + * @note The unit of the transmission queue length depends on the + * link type, a common unit is \a packets. + * + * @par 1) Retrieving information about available links + * @code + * // The first step is to retrieve a list of all available interfaces within + * // the kernel and put them into a cache. + * struct nl_cache *cache = rtnl_link_alloc_cache(nl_handle); + * + * // In a second step, a specific link may be looked up by either interface + * // index or interface name. + * struct rtnl_link *link = rtnl_link_get_by_name(cache, "lo"); + * + * // rtnl_link_get_by_name() is the short version for translating the + * // interface name to an interface index first like this: + * int ifindex = rtnl_link_name2i(cache, "lo"); + * struct rtnl_link *link = rtnl_link_get(cache, ifindex); + * + * // After successful usage, the object must be given back to the cache + * rtnl_link_put(link); + * @endcode + * + * @par 2) Changing link attributes + * @code + * // In order to change any attributes of an existing link, we must allocate + * // a new link to hold the change requests: + * struct rtnl_link *request = rtnl_link_alloc(); + * + * // Now we can go on and specify the attributes we want to change: + * rtnl_link_set_weight(request, 300); + * rtnl_link_set_mtu(request, 1360); + * + * // We can also shut an interface down administratively + * rtnl_link_unset_flags(request, rtnl_link_str2flags("up")); + * + * // Actually, we should know which link to change, so let's look it up + * struct rtnl_link *old = rtnl_link_get(cache, "eth0"); + * + * // Two ways exist to commit this change request, the first one is to + * // build the required netlink message and send it out in one single + * // step: + * rtnl_link_change(nl_handle, old, request); + * + * // An alternative way is to build the netlink message and send it + * // out yourself using nl_send_auto_complete() + * struct nl_msg *msg = rtnl_link_build_change_request(old, request); + * nl_send_auto_complete(nl_handle, nlmsg_hdr(msg)); + * nlmsg_free(msg); + * + * // Don't forget to give back the link object ;-> + * rtnl_link_put(old); + * @endcode + * @{ + */ + +#include <netlink-local.h> +#include <netlink/netlink.h> +#include <netlink/attr.h> +#include <netlink/utils.h> +#include <netlink/object.h> +#include <netlink/route/rtnl.h> +#include <netlink/route/link.h> + +/** @cond SKIP */ +#define LINK_ATTR_MTU 0x0001 +#define LINK_ATTR_LINK 0x0002 +#define LINK_ATTR_TXQLEN 0x0004 +#define LINK_ATTR_WEIGHT 0x0008 +#define LINK_ATTR_MASTER 0x0010 +#define LINK_ATTR_QDISC 0x0020 +#define LINK_ATTR_MAP 0x0040 +#define LINK_ATTR_ADDR 0x0080 +#define LINK_ATTR_BRD 0x0100 +#define LINK_ATTR_FLAGS 0x0200 +#define LINK_ATTR_IFNAME 0x0400 +#define LINK_ATTR_IFINDEX 0x0800 +#define LINK_ATTR_FAMILY 0x1000 +#define LINK_ATTR_ARPTYPE 0x2000 +#define LINK_ATTR_STATS 0x4000 +#define LINK_ATTR_CHANGE 0x8000 + +static struct nl_cache_ops rtnl_link_ops; +static struct nl_object_ops link_obj_ops; +/** @endcond */ + +static void link_free_data(struct nl_object *c) +{ + struct rtnl_link *link = nl_object_priv(c); + + if (link) { + nl_addr_put(link->l_addr); + nl_addr_put(link->l_bcast); + } +} + +static int link_clone(struct nl_object *_dst, struct nl_object *_src) +{ + struct rtnl_link *dst = nl_object_priv(_dst); + struct rtnl_link *src = nl_object_priv(_src); + + if (src->l_addr) + if (!(dst->l_addr = nl_addr_clone(src->l_addr))) + goto errout; + + if (src->l_bcast) + if (!(dst->l_bcast = nl_addr_clone(src->l_bcast))) + goto errout; + + return 0; +errout: + return nl_get_errno(); +} + +static struct nla_policy link_policy[IFLA_MAX+1] = { + [IFLA_IFNAME] = { .type = NLA_STRING, + .maxlen = IFNAMSIZ }, + [IFLA_MTU] = { .type = NLA_U32 }, + [IFLA_TXQLEN] = { .type = NLA_U32 }, + [IFLA_LINK] = { .type = NLA_U32 }, + [IFLA_WEIGHT] = { .type = NLA_U32 }, + [IFLA_MASTER] = { .type = NLA_U32 }, + [IFLA_QDISC] = { .type = NLA_STRING, + .maxlen = IFQDISCSIZ }, + [IFLA_STATS] = { .minlen = sizeof(struct rtnl_link_stats) }, + [IFLA_MAP] = { .minlen = sizeof(struct rtnl_link_ifmap) }, +}; + +static int link_msg_parser(struct nl_cache_ops *ops, struct sockaddr_nl *who, + struct nlmsghdr *n, void *arg) +{ + struct rtnl_link *link; + struct ifinfomsg *ifi; + struct nlattr *tb[IFLA_MAX+1]; + struct nl_parser_param *pp = arg; + int err; + + link = rtnl_link_alloc(); + if (link == NULL) { + err = nl_errno(ENOMEM); + goto errout; + } + + link->ce_msgtype = n->nlmsg_type; + + err = nlmsg_parse(n, sizeof(*ifi), tb, IFLA_MAX, link_policy); + if (err < 0) + goto errout; + + if (tb[IFLA_IFNAME] == NULL) { + err = nl_error(EINVAL, "Missing link name TLV"); + goto errout; + } + + nla_strlcpy(link->l_name, tb[IFLA_IFNAME], IFNAMSIZ); + + ifi = nlmsg_data(n); + link->l_family = ifi->ifi_family; + link->l_arptype = ifi->ifi_type; + link->l_index = ifi->ifi_index; + link->l_flags = ifi->ifi_flags; + link->l_change = ifi->ifi_change; + link->ce_mask = (LINK_ATTR_IFNAME | LINK_ATTR_FAMILY | + LINK_ATTR_ARPTYPE| LINK_ATTR_IFINDEX | + LINK_ATTR_FLAGS | LINK_ATTR_CHANGE); + + if (tb[IFLA_STATS]) { + struct rtnl_link_stats *st = nla_data(tb[IFLA_STATS]); + + link->l_stats[RTNL_LINK_RX_PACKETS] = st->rx_packets; + link->l_stats[RTNL_LINK_RX_BYTES] = st->rx_bytes; + link->l_stats[RTNL_LINK_RX_ERRORS] = st->rx_errors; + link->l_stats[RTNL_LINK_RX_DROPPED] = st->rx_dropped; + link->l_stats[RTNL_LINK_RX_COMPRESSED] = st->rx_compressed; + link->l_stats[RTNL_LINK_RX_FIFO_ERR] = st->rx_fifo_errors; + link->l_stats[RTNL_LINK_TX_PACKETS] = st->tx_packets; + link->l_stats[RTNL_LINK_TX_BYTES] = st->tx_bytes; + link->l_stats[RTNL_LINK_TX_ERRORS] = st->tx_errors; + link->l_stats[RTNL_LINK_TX_DROPPED] = st->tx_dropped; + link->l_stats[RTNL_LINK_TX_COMPRESSED] = st->tx_compressed; + link->l_stats[RTNL_LINK_TX_FIFO_ERR] = st->tx_fifo_errors; + link->l_stats[RTNL_LINK_RX_LEN_ERR] = st->rx_length_errors; + link->l_stats[RTNL_LINK_RX_OVER_ERR] = st->rx_over_errors; + link->l_stats[RTNL_LINK_RX_CRC_ERR] = st->rx_crc_errors; + link->l_stats[RTNL_LINK_RX_FRAME_ERR] = st->rx_frame_errors; + link->l_stats[RTNL_LINK_RX_MISSED_ERR] = st->rx_missed_errors; + link->l_stats[RTNL_LINK_TX_ABORT_ERR] = st->tx_aborted_errors; + link->l_stats[RTNL_LINK_TX_CARRIER_ERR] = st->tx_carrier_errors; + link->l_stats[RTNL_LINK_TX_HBEAT_ERR] = st->tx_heartbeat_errors; + link->l_stats[RTNL_LINK_TX_WIN_ERR] = st->tx_window_errors; + link->l_stats[RTNL_LINK_MULTICAST] = st->multicast; + + link->ce_mask |= LINK_ATTR_STATS; + } + + if (tb[IFLA_TXQLEN]) { + link->l_txqlen = nla_get_u32(tb[IFLA_TXQLEN]); + link->ce_mask |= LINK_ATTR_TXQLEN; + } + + if (tb[IFLA_MTU]) { + link->l_mtu = nla_get_u32(tb[IFLA_MTU]); + link->ce_mask |= LINK_ATTR_MTU; + } + + if (tb[IFLA_ADDRESS]) { + link->l_addr = nla_get_addr(tb[IFLA_ADDRESS], AF_UNSPEC); + if (link->l_addr == NULL) + goto errout; + nl_addr_set_family(link->l_addr, + nl_addr_guess_family(link->l_addr)); + link->ce_mask |= LINK_ATTR_ADDR; + } + + if (tb[IFLA_BROADCAST]) { + link->l_bcast = nla_get_addr(tb[IFLA_BROADCAST], AF_UNSPEC); + if (link->l_bcast == NULL) + goto errout; + nl_addr_set_family(link->l_bcast, + nl_addr_guess_family(link->l_bcast)); + link->ce_mask |= LINK_ATTR_BRD; + } + + if (tb[IFLA_LINK]) { + link->l_link = nla_get_u32(tb[IFLA_LINK]); + link->ce_mask |= LINK_ATTR_LINK; + } + + if (tb[IFLA_WEIGHT]) { + link->l_weight = nla_get_u32(tb[IFLA_WEIGHT]); + link->ce_mask |= LINK_ATTR_WEIGHT; + } + + if (tb[IFLA_QDISC]) { + nla_strlcpy(link->l_qdisc, tb[IFLA_QDISC], IFQDISCSIZ); + link->ce_mask |= LINK_ATTR_QDISC; + } + + if (tb[IFLA_MAP]) { + struct rtnl_link_ifmap *map = nla_data(tb[IFLA_MAP]); + link->l_map.lm_mem_start = map->mem_start; + link->l_map.lm_mem_end = map->mem_end; + link->l_map.lm_base_addr = map->base_addr; + link->l_map.lm_irq = map->irq; + link->l_map.lm_dma = map->dma; + link->l_map.lm_port = map->port; + link->ce_mask |= LINK_ATTR_MAP; + } + + if (tb[IFLA_MASTER]) { + link->l_master = nla_get_u32(tb[IFLA_MASTER]); + link->ce_mask |= LINK_ATTR_MASTER; + } + + err = pp->pp_cb((struct nl_object *) link, pp); + if (err < 0) + goto errout; + + return P_ACCEPT; + +errout: + rtnl_link_put(link); + return err; +} + +static int link_request_update(struct nl_cache *c, struct nl_handle *h) +{ + return nl_rtgen_request(h, RTM_GETLINK, AF_UNSPEC, NLM_F_DUMP); +} + +static int link_dump_brief(struct nl_object *obj, struct nl_dump_params *p) +{ + char buf[128]; + struct nl_cache *cache = dp_cache(obj); + struct rtnl_link *link = (struct rtnl_link *) obj; + int line = 1; + + dp_dump(p, "%s ", link->l_name); + + if (link->ce_mask & LINK_ATTR_LINK) { + struct rtnl_link *ll = rtnl_link_get(cache, link->l_link); + dp_dump(p, "@%s", ll ? ll->l_name : "NONE"); + if (ll) + rtnl_link_put(ll); + } + + dp_dump(p, "%s ", nl_llproto2str(link->l_arptype, buf, sizeof(buf))); + dp_dump(p, "%s ", link->l_addr ? nl_addr2str(link->l_addr, buf, + sizeof(buf)) : "none"); + dp_dump(p, "mtu %u ", link->l_mtu); + + if (link->ce_mask & LINK_ATTR_MASTER) { + struct rtnl_link *master = rtnl_link_get(cache, link->l_master); + dp_dump(p, "master %s ", master ? master->l_name : "inv"); + if (master) + rtnl_link_put(master); + } + + rtnl_link_flags2str(link->l_flags, buf, sizeof(buf)); + if (buf[0]) + dp_dump(p, "<%s>", buf); + + dp_dump(p, "\n"); + + return line; +} + +static int link_dump_full(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_link *link = (struct rtnl_link *) obj; + char buf[64]; + int line; + + line = link_dump_brief(obj, p); + dp_new_line(p, line++); + + dp_dump(p, " txqlen %u weight %u ", link->l_txqlen, link->l_weight); + + if (link->ce_mask & LINK_ATTR_QDISC) + dp_dump(p, "qdisc %s ", link->l_qdisc); + + if (link->ce_mask & LINK_ATTR_MAP && link->l_map.lm_irq) + dp_dump(p, "irq %u ", link->l_map.lm_irq); + + if (link->ce_mask & LINK_ATTR_IFINDEX) + dp_dump(p, "index %u ", link->l_index); + + if (link->ce_mask & LINK_ATTR_BRD) + dp_dump(p, "brd %s", nl_addr2str(link->l_bcast, buf, + sizeof(buf))); + + dp_dump(p, "\n"); + + return line; +} + +static int link_dump_stats(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_link *link = (struct rtnl_link *) obj; + char *unit, fmt[64]; + float res; + int line; + + line = link_dump_full(obj, p); + + dp_dump_line(p, line++, " Stats: bytes packets errors " + " dropped fifo-err compressed\n"); + + res = nl_cancel_down_bytes(link->l_stats[RTNL_LINK_RX_BYTES], &unit); + + strcpy(fmt, " RX %X.2f %s %10llu %10llu %10llu %10llu %10llu\n"); + fmt[9] = *unit == 'B' ? '9' : '7'; + + dp_dump_line(p, line++, fmt, + res, unit, + link->l_stats[RTNL_LINK_RX_PACKETS], + link->l_stats[RTNL_LINK_RX_ERRORS], + link->l_stats[RTNL_LINK_RX_DROPPED], + link->l_stats[RTNL_LINK_RX_FIFO_ERR], + link->l_stats[RTNL_LINK_RX_COMPRESSED]); + + res = nl_cancel_down_bytes(link->l_stats[RTNL_LINK_TX_BYTES], &unit); + + strcpy(fmt, " TX %X.2f %s %10llu %10llu %10llu %10llu %10llu\n"); + fmt[9] = *unit == 'B' ? '9' : '7'; + + dp_dump_line(p, line++, fmt, + res, unit, + link->l_stats[RTNL_LINK_TX_PACKETS], + link->l_stats[RTNL_LINK_TX_ERRORS], + link->l_stats[RTNL_LINK_TX_DROPPED], + link->l_stats[RTNL_LINK_TX_FIFO_ERR], + link->l_stats[RTNL_LINK_TX_COMPRESSED]); + + dp_dump_line(p, line++, " Errors: length over crc " + " frame missed multicast\n"); + + dp_dump_line(p, line++, " RX %10" PRIu64 " %10" PRIu64 " %10" + PRIu64 " %10" PRIu64 " %10" PRIu64 " %10" + PRIu64 "\n", + link->l_stats[RTNL_LINK_RX_LEN_ERR], + link->l_stats[RTNL_LINK_RX_OVER_ERR], + link->l_stats[RTNL_LINK_RX_CRC_ERR], + link->l_stats[RTNL_LINK_RX_FRAME_ERR], + link->l_stats[RTNL_LINK_RX_MISSED_ERR], + link->l_stats[RTNL_LINK_MULTICAST]); + + dp_dump_line(p, line++, " Errors: aborted carrier heartbeat " + " window collision\n"); + + dp_dump_line(p, line++, " TX %10" PRIu64 " %10" PRIu64 " %10" + PRIu64 " %10" PRIu64 " %10" PRIu64 "\n", + link->l_stats[RTNL_LINK_TX_ABORT_ERR], + link->l_stats[RTNL_LINK_TX_CARRIER_ERR], + link->l_stats[RTNL_LINK_TX_HBEAT_ERR], + link->l_stats[RTNL_LINK_TX_WIN_ERR], + link->l_stats[RTNL_LINK_TX_COLLISIONS]); + + return line; +} + +static int link_dump_xml(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_link *link = (struct rtnl_link *) obj; + struct nl_cache *cache = dp_cache(obj); + char buf[128]; + int i, line = 0; + + dp_dump_line(p, line++, "<link name=\"%s\" index=\"%u\">\n", + link->l_name, link->l_index); + dp_dump_line(p, line++, " <family>%s</family>\n", + nl_af2str(link->l_family, buf, sizeof(buf))); + dp_dump_line(p, line++, " <arptype>%s</arptype>\n", + nl_llproto2str(link->l_arptype, buf, sizeof(buf))); + dp_dump_line(p, line++, " <address>%s</address>\n", + nl_addr2str(link->l_addr, buf, sizeof(buf))); + dp_dump_line(p, line++, " <mtu>%u</mtu>\n", link->l_mtu); + dp_dump_line(p, line++, " <txqlen>%u</txqlen>\n", link->l_txqlen); + dp_dump_line(p, line++, " <weight>%u</weight>\n", link->l_weight); + + rtnl_link_flags2str(link->l_flags, buf, sizeof(buf)); + if (buf[0]) + dp_dump_line(p, line++, " <flags>%s</flags>\n", buf); + + if (link->ce_mask & LINK_ATTR_QDISC) + dp_dump_line(p, line++, " <qdisc>%s</qdisc>\n", link->l_qdisc); + + if (link->ce_mask & LINK_ATTR_LINK) { + struct rtnl_link *ll = rtnl_link_get(cache, link->l_link); + dp_dump_line(p, line++, " <link>%s</link>\n", + ll ? ll->l_name : "none"); + if (ll) + rtnl_link_put(ll); + } + + if (link->ce_mask & LINK_ATTR_MASTER) { + struct rtnl_link *master = rtnl_link_get(cache, link->l_master); + dp_dump_line(p, line++, " <master>%s</master>\n", + master ? master->l_name : "none"); + if (master) + rtnl_link_put(master); + } + + if (link->ce_mask & LINK_ATTR_BRD) + dp_dump_line(p, line++, " <broadcast>%s</broadcast>\n", + nl_addr2str(link->l_bcast, buf, sizeof(buf))); + + if (link->ce_mask & LINK_ATTR_STATS) { + dp_dump_line(p, line++, " <stats>\n"); + for (i = 0; i <= RTNL_LINK_STATS_MAX; i++) { + rtnl_link_stat2str(i, buf, sizeof(buf)); + dp_dump_line(p, line++, + " <%s>%" PRIu64 "</%s>\n", + buf, link->l_stats[i], buf); + } + dp_dump_line(p, line++, " </stats>\n"); + } + + dp_dump_line(p, line++, "</link>\n"); + +#if 0 + uint32_t l_change; /**< Change mask */ + struct rtnl_lifmap l_map; /**< Interface device mapping */ +#endif + + return line; +} + +static int link_dump_env(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_link *link = (struct rtnl_link *) obj; + struct nl_cache *cache = dp_cache(obj); + char buf[128]; + int i, line = 0; + + dp_dump_line(p, line++, "LINK_NAME=%s\n", link->l_name); + dp_dump_line(p, line++, "LINK_IFINDEX=%u\n", link->l_index); + dp_dump_line(p, line++, "LINK_FAMILY=%s\n", + nl_af2str(link->l_family, buf, sizeof(buf))); + dp_dump_line(p, line++, "LINK_TYPE=%s\n", + nl_llproto2str(link->l_arptype, buf, sizeof(buf))); + if (link->ce_mask & LINK_ATTR_ADDR) + dp_dump_line(p, line++, "LINK_ADDRESS=%s\n", + nl_addr2str(link->l_addr, buf, sizeof(buf))); + dp_dump_line(p, line++, "LINK_MTU=%u\n", link->l_mtu); + dp_dump_line(p, line++, "LINK_TXQUEUELEN=%u\n", link->l_txqlen); + dp_dump_line(p, line++, "LINK_WEIGHT=%u\n", link->l_weight); + + rtnl_link_flags2str(link->l_flags & ~IFF_RUNNING, buf, sizeof(buf)); + if (buf[0]) + dp_dump_line(p, line++, "LINK_FLAGS=%s\n", buf); + + if (link->ce_mask & LINK_ATTR_QDISC) + dp_dump_line(p, line++, "LINK_QDISC=%s\n", link->l_qdisc); + + if (link->ce_mask & LINK_ATTR_LINK) { + struct rtnl_link *ll = rtnl_link_get(cache, link->l_link); + + dp_dump_line(p, line++, "LINK_LINK_IFINDEX=%d\n", link->l_link); + if (ll) { + dp_dump_line(p, line++, "LINK_LINK_IFNAME=%s\n", + ll->l_name); + rtnl_link_put(ll); + } + } + + if (link->ce_mask & LINK_ATTR_MASTER) { + struct rtnl_link *master = rtnl_link_get(cache, link->l_master); + dp_dump_line(p, line++, "LINK_MASTER=%s\n", + master ? master->l_name : "none"); + if (master) + rtnl_link_put(master); + } + + if (link->ce_mask & LINK_ATTR_BRD) + dp_dump_line(p, line++, "LINK_BROADCAST=%s\n", + nl_addr2str(link->l_bcast, buf, sizeof(buf))); + + if (link->ce_mask & LINK_ATTR_STATS) { + for (i = 0; i <= RTNL_LINK_STATS_MAX; i++) { + char *c = buf; + + sprintf(buf, "LINK_"); + rtnl_link_stat2str(i, buf + 5, sizeof(buf) - 5); + while (*c) { + *c = toupper(*c); + c++; + } + dp_dump_line(p, line++, + "%s=%" PRIu64 "\n", buf, link->l_stats[i]); + } + } + + return line; +} + +#if 0 +static int link_handle_event(struct nl_object *a, struct rtnl_link_event_cb *cb) +{ + struct rtnl_link *l = (struct rtnl_link *) a; + struct nl_cache *c = dp_cache(a); + int nevents = 0; + + if (l->l_change == ~0U) { + if (l->ce_msgtype == RTM_NEWLINK) + cb->le_register(l); + else + cb->le_unregister(l); + + return 1; + } + + if (l->l_change & IFF_SLAVE) { + if (l->l_flags & IFF_SLAVE) { + struct rtnl_link *m = rtnl_link_get(c, l->l_master); + cb->le_new_bonding(l, m); + if (m) + rtnl_link_put(m); + } else + cb->le_cancel_bonding(l); + } + +#if 0 + if (l->l_change & IFF_UP && l->l_change & IFF_RUNNING) + dp_dump_line(p, line++, "link %s changed state to %s.\n", + l->l_name, l->l_flags & IFF_UP ? "up" : "down"); + + if (l->l_change & IFF_PROMISC) { + dp_new_line(p, line++); + dp_dump(p, "link %s %s promiscuous mode.\n", + l->l_name, l->l_flags & IFF_PROMISC ? "entered" : "left"); + } + + if (line == 0) + dp_dump_line(p, line++, "link %s sent unknown event.\n", + l->l_name); +#endif + + return nevents; +} +#endif + +static int link_compare(struct nl_object *_a, struct nl_object *_b, + uint32_t attrs, int flags) +{ + struct rtnl_link *a = (struct rtnl_link *) _a; + struct rtnl_link *b = (struct rtnl_link *) _b; + int diff = 0; + +#define LINK_DIFF(ATTR, EXPR) ATTR_DIFF(attrs, LINK_ATTR_##ATTR, a, b, EXPR) + + diff |= LINK_DIFF(IFINDEX, a->l_index != b->l_index); + diff |= LINK_DIFF(MTU, a->l_mtu != b->l_mtu); + diff |= LINK_DIFF(LINK, a->l_link != b->l_link); + diff |= LINK_DIFF(TXQLEN, a->l_txqlen != b->l_txqlen); + diff |= LINK_DIFF(WEIGHT, a->l_weight != b->l_weight); + diff |= LINK_DIFF(MASTER, a->l_master != b->l_master); + diff |= LINK_DIFF(FAMILY, a->l_family != b->l_family); + diff |= LINK_DIFF(QDISC, strcmp(a->l_qdisc, b->l_qdisc)); + diff |= LINK_DIFF(IFNAME, strcmp(a->l_name, b->l_name)); + diff |= LINK_DIFF(ADDR, nl_addr_cmp(a->l_addr, b->l_addr)); + diff |= LINK_DIFF(BRD, nl_addr_cmp(a->l_bcast, b->l_bcast)); + + if (flags & LOOSE_FLAG_COMPARISON) + diff |= LINK_DIFF(FLAGS, + (a->l_flags ^ b->l_flags) & b->l_flag_mask); + else + diff |= LINK_DIFF(FLAGS, a->l_flags != b->l_flags); + +#undef LINK_DIFF + + return diff; +} + +static struct trans_tbl link_attrs[] = { + __ADD(LINK_ATTR_MTU, mtu) + __ADD(LINK_ATTR_LINK, link) + __ADD(LINK_ATTR_TXQLEN, txqlen) + __ADD(LINK_ATTR_WEIGHT, weight) + __ADD(LINK_ATTR_MASTER, master) + __ADD(LINK_ATTR_QDISC, qdisc) + __ADD(LINK_ATTR_MAP, map) + __ADD(LINK_ATTR_ADDR, address) + __ADD(LINK_ATTR_BRD, broadcast) + __ADD(LINK_ATTR_FLAGS, flags) + __ADD(LINK_ATTR_IFNAME, name) + __ADD(LINK_ATTR_IFINDEX, ifindex) + __ADD(LINK_ATTR_FAMILY, family) + __ADD(LINK_ATTR_ARPTYPE, arptype) + __ADD(LINK_ATTR_STATS, stats) + __ADD(LINK_ATTR_CHANGE, change) +}; + +static char *link_attrs2str(int attrs, char *buf, size_t len) +{ + return __flags2str(attrs, buf, len, link_attrs, + ARRAY_SIZE(link_attrs)); +} + +/** + * @name Allocation/Freeing + * @{ + */ + +struct rtnl_link *rtnl_link_alloc(void) +{ + return (struct rtnl_link *) nl_object_alloc(&link_obj_ops); +} + +void rtnl_link_put(struct rtnl_link *link) +{ + nl_object_put((struct nl_object *) link); +} + +/** @} */ + +/** + * @name Cache Management + * @{ + */ + + +/** + * Allocate link cache and fill in all configured links. + * @arg handle Netlink handle. + * + * Allocates a new link cache, initializes it properly and updates it + * to include all links currently configured in the kernel. + * + * @note Free the memory after usage. + * @return Newly allocated cache or NULL if an error occured. + */ +struct nl_cache *rtnl_link_alloc_cache(struct nl_handle *handle) +{ + struct nl_cache * cache; + + cache = nl_cache_alloc(&rtnl_link_ops); + if (cache == NULL) + return NULL; + + if (handle && nl_cache_refill(handle, cache) < 0) { + nl_cache_free(cache); + return NULL; + } + + return cache; +} + +/** + * Look up link by interface index in the provided cache + * @arg cache link cache + * @arg ifindex link interface index + * + * The caller owns a reference on the returned object and + * must give the object back via rtnl_link_put(). + * + * @return pointer to link inside the cache or NULL if no match was found. + */ +struct rtnl_link *rtnl_link_get(struct nl_cache *cache, int ifindex) +{ + struct rtnl_link *link; + + if (cache->c_ops != &rtnl_link_ops) + return NULL; + + nl_list_for_each_entry(link, &cache->c_items, ce_list) { + if (link->l_index == ifindex) { + nl_object_get((struct nl_object *) link); + return link; + } + } + + return NULL; +} + +/** + * Look up link by link name in the provided cache + * @arg cache link cache + * @arg name link name + * + * The caller owns a reference on the returned object and + * must give the object back via rtnl_link_put(). + * + * @return pointer to link inside the cache or NULL if no match was found. + */ +struct rtnl_link *rtnl_link_get_by_name(struct nl_cache *cache, + const char *name) +{ + struct rtnl_link *link; + + if (cache->c_ops != &rtnl_link_ops) + return NULL; + + nl_list_for_each_entry(link, &cache->c_items, ce_list) { + if (!strcmp(name, link->l_name)) { + nl_object_get((struct nl_object *) link); + return link; + } + } + + return NULL; +} + +/** @} */ + +/** + * @name Link Modifications + * @{ + */ + +/** + * Builds a netlink change request message to change link attributes + * @arg old link to be changed + * @arg tmpl template with requested changes + * @arg flags additional netlink message flags + * + * Builds a new netlink message requesting a change of link attributes. + * The netlink message header isn't fully equipped with all relevant + * fields and must be sent out via nl_send_auto_complete() or + * supplemented as needed. + * \a old must point to a link currently configured in the kernel + * and \a tmpl must contain the attributes to be changed set via + * \c rtnl_link_set_* functions. + * + * @return New netlink message + * @note Not all attributes can be changed, see + * \ref link_changeable "Changeable Attributes" for more details. + */ +struct nl_msg * rtnl_link_build_change_request(struct rtnl_link *old, + struct rtnl_link *tmpl, + int flags) +{ + struct nl_msg *msg; + struct ifinfomsg ifi = { + .ifi_family = old->l_family, + .ifi_index = old->l_index, + }; + + if (tmpl->ce_mask & LINK_ATTR_FLAGS) { + ifi.ifi_flags = old->l_flags & ~tmpl->l_flag_mask; + ifi.ifi_flags |= tmpl->l_flags; + } + + msg = nlmsg_alloc_simple(RTM_SETLINK, flags); + if (!msg) + goto nla_put_failure; + + if (nlmsg_append(msg, &ifi, sizeof(ifi), NLMSG_ALIGNTO) < 0) + goto nla_put_failure; + + if (tmpl->ce_mask & LINK_ATTR_ADDR) + NLA_PUT_ADDR(msg, IFLA_ADDRESS, tmpl->l_addr); + + if (tmpl->ce_mask & LINK_ATTR_BRD) + NLA_PUT_ADDR(msg, IFLA_BROADCAST, tmpl->l_bcast); + + if (tmpl->ce_mask & LINK_ATTR_MTU) + NLA_PUT_U32(msg, IFLA_MTU, tmpl->l_mtu); + + if (tmpl->ce_mask & LINK_ATTR_TXQLEN) + NLA_PUT_U32(msg, IFLA_TXQLEN, tmpl->l_txqlen); + + if (tmpl->ce_mask & LINK_ATTR_WEIGHT) + NLA_PUT_U32(msg, IFLA_WEIGHT, tmpl->l_weight); + + if (tmpl->ce_mask & LINK_ATTR_IFNAME) + NLA_PUT_STRING(msg, IFLA_IFNAME, tmpl->l_name); + + return msg; + +nla_put_failure: + nlmsg_free(msg); + return NULL; +} + +/** + * Change link attributes + * @arg handle netlink handle + * @arg old link to be changed + * @arg tmpl template with requested changes + * @arg flags additional netlink message flags + * + * Builds a new netlink message by calling rtnl_link_build_change_request(), + * sends the request to the kernel and waits for the next ACK to be + * received, i.e. blocks until the request has been processed. + * + * @return 0 on success or a negative error code + * @note Not all attributes can be changed, see + * \ref link_changeable "Changeable Attributes" for more details. + */ +int rtnl_link_change(struct nl_handle *handle, struct rtnl_link *old, + struct rtnl_link *tmpl, int flags) +{ + int err; + struct nl_msg *msg; + + msg = rtnl_link_build_change_request(old, tmpl, flags); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name Name <-> Index Translations + * @{ + */ + +/** + * Translate an interface index to the corresponding link name + * @arg cache link cache + * @arg ifindex link interface index + * @arg dst destination buffer + * @arg len length of destination buffer + * + * Translates the specified interface index to the corresponding + * link name and stores the name in the destination buffer. + * + * @return link name or NULL if no match was found. + */ +char * rtnl_link_i2name(struct nl_cache *cache, int ifindex, char *dst, + size_t len) +{ + struct rtnl_link *link = rtnl_link_get(cache, ifindex); + + if (link) { + strncpy(dst, link->l_name, len - 1); + rtnl_link_put(link); + return dst; + } + + return NULL; +} + +/** + * Translate a link name to the corresponding interface index + * @arg cache link cache + * @arg name link name + * + * @return interface index or RTNL_LINK_NOT_FOUND if no match was found. + */ +int rtnl_link_name2i(struct nl_cache *cache, const char *name) +{ + int ifindex = RTNL_LINK_NOT_FOUND; + struct rtnl_link *link; + + link = rtnl_link_get_by_name(cache, name); + if (link) { + ifindex = link->l_index; + rtnl_link_put(link); + } + + return ifindex; +} + +/** @} */ + +/** + * @name Link Flags Translations + * @{ + */ + +static struct trans_tbl link_flags[] = { + __ADD(IFF_LOOPBACK, loopback) + __ADD(IFF_BROADCAST, broadcast) + __ADD(IFF_POINTOPOINT, pointopoint) + __ADD(IFF_MULTICAST, multicast) + __ADD(IFF_NOARP, noarp) + __ADD(IFF_ALLMULTI, allmulti) + __ADD(IFF_PROMISC, promisc) + __ADD(IFF_MASTER, master) + __ADD(IFF_SLAVE, slave) + __ADD(IFF_DEBUG, debug) + __ADD(IFF_DYNAMIC, dynamic) + __ADD(IFF_AUTOMEDIA, automedia) + __ADD(IFF_PORTSEL, portsel) + __ADD(IFF_NOTRAILERS, notrailers) + __ADD(IFF_UP, up) + __ADD(IFF_RUNNING, running) + __ADD(IFF_LOWER_UP, lowerup) + __ADD(IFF_DORMANT, dormant) +}; + +char * rtnl_link_flags2str(int flags, char *buf, size_t len) +{ + return __flags2str(flags, buf, len, link_flags, + ARRAY_SIZE(link_flags)); +} + +int rtnl_link_str2flags(const char *name) +{ + return __str2flags(name, link_flags, ARRAY_SIZE(link_flags)); +} + +/** @} */ + +/** + * @name Link Statistics Translations + * @{ + */ + +static struct trans_tbl link_stats[] = { + __ADD(RTNL_LINK_RX_PACKETS, rx_packets) + __ADD(RTNL_LINK_TX_PACKETS, tx_packets) + __ADD(RTNL_LINK_RX_BYTES, rx_bytes) + __ADD(RTNL_LINK_TX_BYTES, tx_bytes) + __ADD(RTNL_LINK_RX_ERRORS, rx_errors) + __ADD(RTNL_LINK_TX_ERRORS, tx_errors) + __ADD(RTNL_LINK_RX_DROPPED, rx_dropped) + __ADD(RTNL_LINK_TX_DROPPED, tx_dropped) + __ADD(RTNL_LINK_RX_COMPRESSED, rx_compressed) + __ADD(RTNL_LINK_TX_COMPRESSED, tx_compressed) + __ADD(RTNL_LINK_RX_FIFO_ERR, rx_fifo_err) + __ADD(RTNL_LINK_TX_FIFO_ERR, tx_fifo_err) + __ADD(RTNL_LINK_RX_LEN_ERR, rx_len_err) + __ADD(RTNL_LINK_RX_OVER_ERR, rx_over_err) + __ADD(RTNL_LINK_RX_CRC_ERR, rx_crc_err) + __ADD(RTNL_LINK_RX_FRAME_ERR, rx_frame_err) + __ADD(RTNL_LINK_RX_MISSED_ERR, rx_missed_err) + __ADD(RTNL_LINK_TX_ABORT_ERR, tx_abort_err) + __ADD(RTNL_LINK_TX_CARRIER_ERR, tx_carrier_err) + __ADD(RTNL_LINK_TX_HBEAT_ERR, tx_hbeat_err) + __ADD(RTNL_LINK_TX_WIN_ERR, tx_win_err) + __ADD(RTNL_LINK_TX_COLLISIONS, tx_collision) + __ADD(RTNL_LINK_MULTICAST, multicast) +}; + +char *rtnl_link_stat2str(int st, char *buf, size_t len) +{ + return __type2str(st, buf, len, link_stats, ARRAY_SIZE(link_stats)); +} + +int rtnl_link_str2stat(const char *name) +{ + return __str2type(name, link_stats, ARRAY_SIZE(link_stats)); +} + +/** @} */ + +/** + * @name Attributes + * @{ + */ + +void rtnl_link_set_qdisc(struct rtnl_link *link, const char *qdisc) +{ + strncpy(link->l_qdisc, qdisc, sizeof(link->l_qdisc) - 1); + link->ce_mask |= LINK_ATTR_QDISC; +} + +char *rtnl_link_get_qdisc(struct rtnl_link *link) +{ + if (link->ce_mask & LINK_ATTR_QDISC) + return link->l_qdisc; + else + return NULL; +} + +void rtnl_link_set_name(struct rtnl_link *link, const char *name) +{ + strncpy(link->l_name, name, sizeof(link->l_name) - 1); + link->ce_mask |= LINK_ATTR_IFNAME; +} + +char *rtnl_link_get_name(struct rtnl_link *link) +{ + if (link->ce_mask & LINK_ATTR_IFNAME) + return link->l_name; + else + return NULL; +} + +static inline void __assign_addr(struct rtnl_link *link, struct nl_addr **pos, + struct nl_addr *new, int flag) +{ + if (*pos) + nl_addr_put(*pos); + + nl_addr_get(new); + *pos = new; + + link->ce_mask |= flag; +} + +void rtnl_link_set_addr(struct rtnl_link *link, struct nl_addr *addr) +{ + __assign_addr(link, &link->l_addr, addr, LINK_ATTR_ADDR); +} + +struct nl_addr *rtnl_link_get_addr(struct rtnl_link *link) +{ + if (link->ce_mask & LINK_ATTR_ADDR) + return link->l_addr; + else + return NULL; +} + +void rtnl_link_set_broadcast(struct rtnl_link *link, struct nl_addr *brd) +{ + __assign_addr(link, &link->l_bcast, brd, LINK_ATTR_BRD); +} + +struct nl_addr *rtnl_link_get_broadcast(struct rtnl_link *link) +{ + if (link->ce_mask & LINK_ATTR_BRD) + return link->l_bcast; + else + return NULL; +} + +void rtnl_link_set_flags(struct rtnl_link *link, unsigned int flags) +{ + link->l_flag_mask |= flags; + link->l_flags |= flags; + link->ce_mask |= LINK_ATTR_FLAGS; +} + +void rtnl_link_unset_flags(struct rtnl_link *link, unsigned int flags) +{ + link->l_flag_mask |= flags; + link->l_flags &= ~flags; + link->ce_mask |= LINK_ATTR_FLAGS; +} + +unsigned int rtnl_link_get_flags(struct rtnl_link *link) +{ + return link->l_flags; +} + +void rtnl_link_set_family(struct rtnl_link *link, int family) +{ + link->l_family = family; + link->ce_mask |= LINK_ATTR_FAMILY; +} + +int rtnl_link_get_family(struct rtnl_link *link) +{ + if (link->l_family & LINK_ATTR_FAMILY) + return link->l_family; + else + return AF_UNSPEC; +} + +void rtnl_link_set_arptype(struct rtnl_link *link, unsigned int arptype) +{ + link->l_arptype = arptype; +} + +unsigned int rtnl_link_get_arptype(struct rtnl_link *link) +{ + return link->l_arptype; +} + +void rtnl_link_set_ifindex(struct rtnl_link *link, int ifindex) +{ + link->l_index = ifindex; + link->ce_mask |= LINK_ATTR_IFINDEX; +} + +int rtnl_link_get_ifindex(struct rtnl_link *link) +{ + if (link->ce_mask & LINK_ATTR_IFINDEX) + return link->l_index; + else + return RTNL_LINK_NOT_FOUND; +} + +void rtnl_link_set_mtu(struct rtnl_link *link, unsigned int mtu) +{ + link->l_mtu = mtu; + link->ce_mask |= LINK_ATTR_MTU; +} + +unsigned int rtnl_link_get_mtu(struct rtnl_link *link) +{ + if (link->ce_mask & LINK_ATTR_MTU) + return link->l_mtu; + else + return 0; +} + +void rtnl_link_set_txqlen(struct rtnl_link *link, unsigned int txqlen) +{ + link->l_txqlen = txqlen; + link->ce_mask |= LINK_ATTR_TXQLEN; +} + +unsigned int rtnl_link_get_txqlen(struct rtnl_link *link) +{ + if (link->ce_mask & LINK_ATTR_TXQLEN) + return link->l_txqlen; + else + return UINT_MAX; +} + +void rtnl_link_set_weight(struct rtnl_link *link, unsigned int weight) +{ + link->l_weight = weight; + link->ce_mask |= LINK_ATTR_WEIGHT; +} + +unsigned int rtnl_link_get_weight(struct rtnl_link *link) +{ + if (link->ce_mask & LINK_ATTR_WEIGHT) + return link->l_weight; + else + return UINT_MAX; +} + +void rtnl_link_set_link(struct rtnl_link *link, int ifindex) +{ + link->l_link = ifindex; + link->ce_mask |= LINK_ATTR_LINK; +} + +int rtnl_link_get_link(struct rtnl_link *link) +{ + if (link->ce_mask & LINK_ATTR_LINK) + return link->l_link; + else + return RTNL_LINK_NOT_FOUND; +} + +void rtnl_link_set_master(struct rtnl_link *link, int ifindex) +{ + link->l_master = ifindex; + link->ce_mask |= LINK_ATTR_MASTER; +} + +int rtnl_link_get_master(struct rtnl_link *link) +{ + if (link->ce_mask & LINK_ATTR_MASTER) + return link->l_master; + else + return RTNL_LINK_NOT_FOUND; +} + +uint64_t rtnl_link_get_stat(struct rtnl_link *link, int id) +{ + if (id < 0 || id > RTNL_LINK_STATS_MAX) + return 0; + + return link->l_stats[id]; +} + +/** @} */ + +static struct nl_object_ops link_obj_ops = { + .oo_name = "route/link", + .oo_size = sizeof(struct rtnl_link), + .oo_free_data = link_free_data, + .oo_clone = link_clone, + .oo_dump[NL_DUMP_BRIEF] = link_dump_brief, + .oo_dump[NL_DUMP_FULL] = link_dump_full, + .oo_dump[NL_DUMP_STATS] = link_dump_stats, + .oo_dump[NL_DUMP_XML] = link_dump_xml, + .oo_dump[NL_DUMP_ENV] = link_dump_env, + .oo_compare = link_compare, + .oo_attrs2str = link_attrs2str, + .oo_id_attrs = LINK_ATTR_IFINDEX, +}; + +static struct nl_af_group link_groups[] = { + { AF_UNSPEC, RTNLGRP_LINK }, + { END_OF_GROUP_LIST }, +}; + +static struct nl_cache_ops rtnl_link_ops = { + .co_name = "route/link", + .co_hdrsize = sizeof(struct ifinfomsg), + .co_msgtypes = { + { RTM_NEWLINK, NL_ACT_NEW, "new" }, + { RTM_DELLINK, NL_ACT_DEL, "del" }, + { RTM_GETLINK, NL_ACT_GET, "get" }, + END_OF_MSGTYPES_LIST, + }, + .co_protocol = NETLINK_ROUTE, + .co_groups = link_groups, + .co_request_update = link_request_update, + .co_msg_parser = link_msg_parser, + .co_obj_ops = &link_obj_ops, +}; + +static void __init link_init(void) +{ + nl_cache_mngt_register(&rtnl_link_ops); +} + +static void __exit link_exit(void) +{ + nl_cache_mngt_unregister(&rtnl_link_ops); +} + +/** @} */ diff --git a/lib/route/neigh.c b/lib/route/neigh.c new file mode 100644 index 0000000..aac4718 --- /dev/null +++ b/lib/route/neigh.c @@ -0,0 +1,1023 @@ +/* + * lib/route/neigh.c Neighbours + * + * 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> + */ + +/** + * @ingroup rtnl + * @defgroup neigh Neighbours + * @brief + * + * The neighbour table establishes bindings between protocol addresses and + * link layer addresses for hosts sharing the same physical link. This + * module allows you to access and manipulate the content of these tables. + * + * @par Neighbour States + * @code + * NUD_INCOMPLETE + * NUD_REACHABLE + * NUD_STALE + * NUD_DELAY + * NUD_PROBE + * NUD_FAILED + * NUD_NOARP + * NUD_PERMANENT + * @endcode + * + * @par Neighbour Flags + * @code + * NTF_PROXY + * NTF_ROUTER + * @endcode + * + * @par Neighbour Identification + * A neighbour is uniquely identified by the attributes listed below, whenever + * you refer to an existing neighbour all of the attributes must be set. + * Neighbours from caches automatically have all required attributes set. + * - interface index (rtnl_neigh_set_ifindex()) + * - destination address (rtnl_neigh_set_dst()) + * + * @par Changeable Attributes + * \anchor neigh_changeable + * - state (rtnl_neigh_set_state()) + * - link layer address (rtnl_neigh_set_lladdr()) + * + * @par Required Caches for Dumping + * In order to dump neighbour attributes you must provide the following + * caches via nl_cache_provide() + * - link cache holding all links + * + * @par TODO + * - Document proxy settings + * - Document states and their influence + * + * @par 1) Retrieving information about configured neighbours + * @code + * // The first step is to retrieve a list of all available neighbour within + * // the kernel and put them into a cache. + * struct nl_cache *cache = rtnl_neigh_alloc_cache(handle); + * + * // Neighbours can then be looked up by the interface and destination + * // address: + * struct rtnl_neigh *neigh = rtnl_neigh_get(cache, ifindex, dst_addr); + * + * // After successful usage, the object must be given back to the cache + * rtnl_neigh_put(neigh); + * @endcode + * + * @par 2) Adding new neighbours + * @code + * // Allocate an empty neighbour handle to be filled out with the attributes + * // of the new neighbour. + * struct rtnl_neigh *neigh = rtnl_neigh_alloc(); + * + * // Fill out the attributes of the new neighbour + * rtnl_neigh_set_ifindex(neigh, ifindex); + * rtnl_neigh_set_dst(neigh, dst_addr); + * rtnl_neigh_set_state(neigh, rtnl_neigh_str2state("permanent")); + * + * // Build the netlink message and send it to the kernel, the operation will + * // block until the operation has been completed. Alternatively the required + * // netlink message can be built using rtnl_neigh_build_add_request() + * // to be sent out using nl_send_auto_complete(). + * rtnl_neigh_add(nl_handle, neigh, NLM_F_REPLACE); + * + * // Free the memory + * rtnl_neigh_put(neigh); + * @endcode + * + * @par 3) Deleting an existing neighbour + * @code + * // Allocate an empty neighbour object to be filled out with the attributes + * // matching the neighbour to be deleted. Alternatively a fully equipped + * // neighbour object out of a cache can be used instead. + * struct rtnl_neigh *neigh = rtnl_neigh_alloc(); + * + * // Neighbours are uniquely identified by their interface index and + * // destination address, you may fill out other attributes but they + * // will have no influence. + * rtnl_neigh_set_ifindex(neigh, ifindex); + * rtnl_neigh_set_dst(neigh, dst_addr); + * + * // Build the netlink message and send it to the kernel, the operation will + * // block until the operation has been completed. Alternatively the required + * // netlink message can be built using rtnl_neigh_build_delete_request() + * // to be sent out using nl_send_auto_complete(). + * rtnl_neigh_delete(handle, neigh, 0); + * + * // Free the memory + * rtnl_neigh_put(neigh); + * @endcode + * + * @par 4) Changing neighbour attributes + * @code + * // Allocate an empty neighbour object to be filled out with the attributes + * // matching the neighbour to be changed and the new parameters. Alternatively + * // a fully equipped modified neighbour object out of a cache can be used. + * struct rtnl_neigh *neigh = rtnl_neigh_alloc(); + * + * // Identify the neighbour to be changed by its interface index and + * // destination address + * rtnl_neigh_set_ifindex(neigh, ifindex); + * rtnl_neigh_set_dst(neigh, dst_addr); + * + * // The link layer address may be modified, if so it is wise to change + * // its state to "permanent" in order to avoid having it overwritten. + * rtnl_neigh_set_lladdr(neigh, lladdr); + * + * // Secondly the state can be modified allowing normal neighbours to be + * // converted into permanent entries or to manually confirm a neighbour. + * rtnl_neigh_set_state(neigh, state); + * + * // Build the netlink message and send it to the kernel, the operation will + * // block until the operation has been completed. Alternatively the required + * // netlink message can be built using rtnl_neigh_build_change_request() + * // to be sent out using nl_send_auto_complete(). + * rtnl_neigh_change(handle, neigh, 0); + * + * // Free the memory + * rtnl_neigh_put(neigh); + * @endcode + * @{ + */ + +#include <netlink-local.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/rtnl.h> +#include <netlink/route/neighbour.h> +#include <netlink/route/link.h> + +/** @cond SKIP */ +#define NEIGH_ATTR_FLAGS 0x01 +#define NEIGH_ATTR_STATE 0x02 +#define NEIGH_ATTR_LLADDR 0x04 +#define NEIGH_ATTR_DST 0x08 +#define NEIGH_ATTR_CACHEINFO 0x10 +#define NEIGH_ATTR_IFINDEX 0x20 +#define NEIGH_ATTR_FAMILY 0x40 +#define NEIGH_ATTR_TYPE 0x80 +#define NEIGH_ATTR_PROBES 0x100 + +static struct nl_cache_ops rtnl_neigh_ops; +static struct nl_object_ops neigh_obj_ops; +/** @endcond */ + +static void neigh_free_data(struct nl_object *c) +{ + struct rtnl_neigh *neigh = nl_object_priv(c); + + if (!neigh) + return; + + nl_addr_put(neigh->n_lladdr); + nl_addr_put(neigh->n_dst); +} + +static int neigh_clone(struct nl_object *_dst, struct nl_object *_src) +{ + struct rtnl_neigh *dst = nl_object_priv(_dst); + struct rtnl_neigh *src = nl_object_priv(_src); + + if (src->n_lladdr) + if (!(dst->n_lladdr = nl_addr_clone(src->n_lladdr))) + goto errout; + + if (src->n_dst) + if (!(dst->n_dst = nl_addr_clone(src->n_dst))) + goto errout; + + return 0; +errout: + return nl_get_errno(); +} + +static int neigh_compare(struct nl_object *_a, struct nl_object *_b, + uint32_t attrs, int flags) +{ + struct rtnl_neigh *a = (struct rtnl_neigh *) _a; + struct rtnl_neigh *b = (struct rtnl_neigh *) _b; + int diff = 0; + +#define NEIGH_DIFF(ATTR, EXPR) ATTR_DIFF(attrs, NEIGH_ATTR_##ATTR, a, b, EXPR) + + diff |= NEIGH_DIFF(IFINDEX, a->n_ifindex != b->n_ifindex); + diff |= NEIGH_DIFF(FAMILY, a->n_family != b->n_family); + diff |= NEIGH_DIFF(TYPE, a->n_type != b->n_type); + diff |= NEIGH_DIFF(LLADDR, nl_addr_cmp(a->n_lladdr, b->n_lladdr)); + diff |= NEIGH_DIFF(DST, nl_addr_cmp(a->n_dst, b->n_dst)); + + if (flags & LOOSE_FLAG_COMPARISON) { + diff |= NEIGH_DIFF(STATE, + (a->n_state ^ b->n_state) & b->n_state_mask); + diff |= NEIGH_DIFF(FLAGS, + (a->n_flags ^ b->n_flags) & b->n_flag_mask); + } else { + diff |= NEIGH_DIFF(STATE, a->n_state != b->n_state); + diff |= NEIGH_DIFF(FLAGS, a->n_flags != b->n_flags); + } + +#undef NEIGH_DIFF + + return diff; +} + +static struct trans_tbl neigh_attrs[] = { + __ADD(NEIGH_ATTR_FLAGS, flags) + __ADD(NEIGH_ATTR_STATE, state) + __ADD(NEIGH_ATTR_LLADDR, lladdr) + __ADD(NEIGH_ATTR_DST, dst) + __ADD(NEIGH_ATTR_CACHEINFO, cacheinfo) + __ADD(NEIGH_ATTR_IFINDEX, ifindex) + __ADD(NEIGH_ATTR_FAMILY, family) + __ADD(NEIGH_ATTR_TYPE, type) + __ADD(NEIGH_ATTR_PROBES, probes) +}; + +static char *neigh_attrs2str(int attrs, char *buf, size_t len) +{ + return __flags2str(attrs, buf, len, neigh_attrs, + ARRAY_SIZE(neigh_attrs)); +} + +static struct nla_policy neigh_policy[NDA_MAX+1] = { + [NDA_CACHEINFO] = { .minlen = sizeof(struct nda_cacheinfo) }, + [NDA_PROBES] = { .type = NLA_U32 }, +}; + +static int neigh_msg_parser(struct nl_cache_ops *ops, struct sockaddr_nl *who, + struct nlmsghdr *n, void *arg) +{ + struct rtnl_neigh *neigh; + struct nlattr *tb[NDA_MAX + 1]; + struct nl_parser_param *pp = arg; + struct ndmsg *nm; + int err; + + neigh = rtnl_neigh_alloc(); + if (!neigh) { + err = nl_errno(ENOMEM); + goto errout; + } + + neigh->ce_msgtype = n->nlmsg_type; + nm = nlmsg_data(n); + + err = nlmsg_parse(n, sizeof(*nm), tb, NDA_MAX, neigh_policy); + if (err < 0) + goto errout; + + neigh->n_family = nm->ndm_family; + neigh->n_ifindex = nm->ndm_ifindex; + neigh->n_state = nm->ndm_state; + neigh->n_flags = nm->ndm_flags; + neigh->n_type = nm->ndm_type; + + neigh->ce_mask |= (NEIGH_ATTR_FAMILY | NEIGH_ATTR_IFINDEX | + NEIGH_ATTR_STATE | NEIGH_ATTR_FLAGS | + NEIGH_ATTR_TYPE); + + if (tb[NDA_LLADDR]) { + neigh->n_lladdr = nla_get_addr(tb[NDA_LLADDR], AF_UNSPEC); + if (!neigh->n_lladdr) + goto errout; + nl_addr_set_family(neigh->n_lladdr, + nl_addr_guess_family(neigh->n_lladdr)); + neigh->ce_mask |= NEIGH_ATTR_LLADDR; + } + + if (tb[NDA_DST]) { + neigh->n_dst = nla_get_addr(tb[NDA_DST], neigh->n_family); + if (!neigh->n_dst) + goto errout; + neigh->ce_mask |= NEIGH_ATTR_DST; + } + + if (tb[NDA_CACHEINFO]) { + struct nda_cacheinfo *ci = nla_data(tb[NDA_CACHEINFO]); + + neigh->n_cacheinfo.nci_confirmed = ci->ndm_confirmed; + neigh->n_cacheinfo.nci_used = ci->ndm_used; + neigh->n_cacheinfo.nci_updated = ci->ndm_updated; + neigh->n_cacheinfo.nci_refcnt = ci->ndm_refcnt; + + neigh->ce_mask |= NEIGH_ATTR_CACHEINFO; + } + + if (tb[NDA_PROBES]) { + neigh->n_probes = nla_get_u32(tb[NDA_PROBES]); + neigh->ce_mask |= NEIGH_ATTR_PROBES; + } + + err = pp->pp_cb((struct nl_object *) neigh, pp); + if (err < 0) + goto errout; + + return P_ACCEPT; + +errout: + rtnl_neigh_put(neigh); + return err; +} + +static int neigh_request_update(struct nl_cache *c, struct nl_handle *h) +{ + return nl_rtgen_request(h, RTM_GETNEIGH, AF_UNSPEC, NLM_F_DUMP); +} + + +static int neigh_dump_brief(struct nl_object *a, struct nl_dump_params *p) +{ + char dst[INET6_ADDRSTRLEN+5], lladdr[INET6_ADDRSTRLEN+5]; + struct rtnl_neigh *n = (struct rtnl_neigh *) a; + struct nl_cache *link_cache; + char state[128], flags[64]; + + link_cache = nl_cache_mngt_require("route/link"); + + dp_dump(p, "%s ", nl_addr2str(n->n_dst, dst, sizeof(dst))); + + if (link_cache) + dp_dump(p, "dev %s ", + rtnl_link_i2name(link_cache, n->n_ifindex, + state, sizeof(state))); + else + dp_dump(p, "dev %d ", n->n_ifindex); + + if (n->ce_mask & NEIGH_ATTR_LLADDR) + dp_dump(p, "lladdr %s ", + nl_addr2str(n->n_lladdr, lladdr, sizeof(lladdr))); + + rtnl_neigh_state2str(n->n_state, state, sizeof(state)); + rtnl_neigh_flags2str(n->n_flags, flags, sizeof(flags)); + + if (state[0]) + dp_dump(p, "<%s", state); + if (flags[0]) + dp_dump(p, "%s%s", state[0] ? "," : "<", flags); + if (state[0] || flags[0]) + dp_dump(p, ">"); + dp_dump(p, "\n"); + + return 1; +} + +static int neigh_dump_full(struct nl_object *a, struct nl_dump_params *p) +{ + char rtn_type[32]; + struct rtnl_neigh *n = (struct rtnl_neigh *) a; + int hz = nl_get_hz(); + + int line = neigh_dump_brief(a, p); + + dp_dump_line(p, line++, " refcnt %u type %s confirmed %u used " + "%u updated %u\n", + n->n_cacheinfo.nci_refcnt, + nl_rtntype2str(n->n_type, rtn_type, sizeof(rtn_type)), + n->n_cacheinfo.nci_confirmed/hz, + n->n_cacheinfo.nci_used/hz, n->n_cacheinfo.nci_updated/hz); + + return line; +} + +static int neigh_dump_stats(struct nl_object *a, struct nl_dump_params *p) +{ + return neigh_dump_full(a, p); +} + +static int neigh_dump_xml(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_neigh *neigh = (struct rtnl_neigh *) obj; + char buf[128]; + int line = 0; + + dp_dump_line(p, line++, "<neighbour>\n"); + dp_dump_line(p, line++, " <family>%s</family>\n", + nl_af2str(neigh->n_family, buf, sizeof(buf))); + + if (neigh->ce_mask & NEIGH_ATTR_LLADDR) + dp_dump_line(p, line++, " <lladdr>%s</lladdr>\n", + nl_addr2str(neigh->n_lladdr, buf, sizeof(buf))); + + if (neigh->ce_mask & NEIGH_ATTR_DST) + dp_dump_line(p, line++, " <dst>%s</dst>\n", + nl_addr2str(neigh->n_dst, buf, sizeof(buf))); + + if (neigh->ce_mask & NEIGH_ATTR_IFINDEX) { + struct nl_cache *link_cache; + + link_cache = nl_cache_mngt_require("route/link"); + + if (link_cache) + dp_dump_line(p, line++, " <device>%s</device>\n", + rtnl_link_i2name(link_cache, + neigh->n_ifindex, + buf, sizeof(buf))); + else + dp_dump_line(p, line++, " <device>%u</device>\n", + neigh->n_ifindex); + } + + if (neigh->ce_mask & NEIGH_ATTR_PROBES) + dp_dump_line(p, line++, " <probes>%u</probes>\n", + neigh->n_probes); + + if (neigh->ce_mask & NEIGH_ATTR_TYPE) + dp_dump_line(p, line++, " <type>%s</type>\n", + nl_rtntype2str(neigh->n_type, buf, sizeof(buf))); + + rtnl_neigh_flags2str(neigh->n_flags, buf, sizeof(buf)); + if (buf[0]) + dp_dump_line(p, line++, " <flags>%s</flags>\n", buf); + + rtnl_neigh_state2str(neigh->n_state, buf, sizeof(buf)); + if (buf[0]) + dp_dump_line(p, line++, " <state>%s</state>\n", buf); + + dp_dump_line(p, line++, "</neighbour>\n"); + +#if 0 + struct rtnl_ncacheinfo n_cacheinfo; +#endif + + return line; +} + +static int neigh_dump_env(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_neigh *neigh = (struct rtnl_neigh *) obj; + char buf[128]; + int line = 0; + + dp_dump_line(p, line++, "NEIGH_FAMILY=%s\n", + nl_af2str(neigh->n_family, buf, sizeof(buf))); + + if (neigh->ce_mask & NEIGH_ATTR_LLADDR) + dp_dump_line(p, line++, "NEIGHT_LLADDR=%s\n", + nl_addr2str(neigh->n_lladdr, buf, sizeof(buf))); + + if (neigh->ce_mask & NEIGH_ATTR_DST) + dp_dump_line(p, line++, "NEIGH_DST=%s\n", + nl_addr2str(neigh->n_dst, buf, sizeof(buf))); + + if (neigh->ce_mask & NEIGH_ATTR_IFINDEX) { + struct nl_cache *link_cache; + + dp_dump_line(p, line++, "NEIGH_IFINDEX=%u\n", + neigh->n_ifindex); + + link_cache = nl_cache_mngt_require("route/link"); + if (link_cache) + dp_dump_line(p, line++, "NEIGH_IFNAME=%s\n", + rtnl_link_i2name(link_cache, + neigh->n_ifindex, + buf, sizeof(buf))); + } + + if (neigh->ce_mask & NEIGH_ATTR_PROBES) + dp_dump_line(p, line++, "NEIGH_PROBES=%u\n", + neigh->n_probes); + + if (neigh->ce_mask & NEIGH_ATTR_TYPE) + dp_dump_line(p, line++, "NEIGH_TYPE=%s\n", + nl_rtntype2str(neigh->n_type, buf, sizeof(buf))); + + rtnl_neigh_flags2str(neigh->n_flags, buf, sizeof(buf)); + if (buf[0]) + dp_dump_line(p, line++, "NEIGH_FLAGS=%s\n", buf); + + rtnl_neigh_state2str(neigh->n_state, buf, sizeof(buf)); + if (buf[0]) + dp_dump_line(p, line++, "NEIGH_STATE=%s\n", buf); + + return line; +} + +/** + * @name Neighbour Object Allocation/Freeage + * @{ + */ + +struct rtnl_neigh *rtnl_neigh_alloc(void) +{ + return (struct rtnl_neigh *) nl_object_alloc(&neigh_obj_ops); +} + +void rtnl_neigh_put(struct rtnl_neigh *neigh) +{ + nl_object_put((struct nl_object *) neigh); +} + +/** @} */ + +/** + * @name Neighbour Cache Managament + * @{ + */ + +/** + * Build a neighbour cache including all neighbours currently configured in the kernel. + * @arg handle netlink handle + * + * Allocates a new neighbour cache, initializes it properly and updates it + * to include all neighbours currently configured in the kernel. + * + * @note The caller is responsible for destroying and freeing the + * cache after using it. + * @return The new cache or NULL if an error occured. + */ +struct nl_cache *rtnl_neigh_alloc_cache(struct nl_handle *handle) +{ + struct nl_cache *cache; + + cache = nl_cache_alloc(&rtnl_neigh_ops); + if (cache == NULL) + return NULL; + + if (handle && nl_cache_refill(handle, cache) < 0) { + nl_cache_free(cache); + return NULL; + } + + NL_DBG(2, "Returning new cache %p\n", cache); + + return cache; +} + +/** + * Look up a neighbour by interface index and destination address + * @arg cache neighbour cache + * @arg ifindex interface index the neighbour is on + * @arg dst destination address of the neighbour + * @return neighbour handle or NULL if no match was found. + */ +struct rtnl_neigh * rtnl_neigh_get(struct nl_cache *cache, int ifindex, + struct nl_addr *dst) +{ + struct rtnl_neigh *neigh; + + nl_list_for_each_entry(neigh, &cache->c_items, ce_list) { + if (neigh->n_ifindex == ifindex && + !nl_addr_cmp(neigh->n_dst, dst)) { + nl_object_get((struct nl_object *) neigh); + return neigh; + } + } + + return NULL; +} + +/** @} */ + +/** + * @name Neighbour Addition + * @{ + */ + +static struct nl_msg * build_neigh_msg(struct rtnl_neigh *tmpl, int cmd, + int flags) +{ + struct nl_msg *msg; + struct ndmsg nhdr = { + .ndm_ifindex = tmpl->n_ifindex, + .ndm_family = nl_addr_get_family(tmpl->n_dst), + .ndm_state = NUD_PERMANENT, + }; + + if (tmpl->ce_mask & NEIGH_ATTR_STATE) + nhdr.ndm_state = tmpl->n_state; + + msg = nlmsg_alloc_simple(cmd, flags); + if (!msg) + return NULL; + + if (nlmsg_append(msg, &nhdr, sizeof(nhdr), NLMSG_ALIGNTO) < 0) + goto nla_put_failure; + + NLA_PUT_ADDR(msg, NDA_DST, tmpl->n_dst); + + if (tmpl->ce_mask & NEIGH_ATTR_LLADDR) + NLA_PUT_ADDR(msg, NDA_LLADDR, tmpl->n_lladdr); + + return msg; + +nla_put_failure: + nlmsg_free(msg); + return NULL; +} + +/** + * Build netlink request message to add a new neighbour + * @arg tmpl template with data of new neighbour + * @arg flags additional netlink message flags + * + * Builds a new netlink message requesting a addition of a new + * neighbour. The netlink message header isn't fully equipped with + * all relevant fields and must thus be sent out via nl_send_auto_complete() + * or supplemented as needed. \a tmpl must contain the attributes of the new + * neighbour set via \c rtnl_neigh_set_* functions. + * + * The following attributes must be set in the template: + * - Interface index (rtnl_neigh_set_ifindex()) + * - State (rtnl_neigh_set_state()) + * - Destination address (rtnl_neigh_set_dst()) + * - Link layer address (rtnl_neigh_set_lladdr()) + * + * @return The netlink message + */ +struct nl_msg * rtnl_neigh_build_add_request(struct rtnl_neigh *tmpl, int flags) +{ + return build_neigh_msg(tmpl, RTM_NEWNEIGH, NLM_F_CREATE | flags); +} + +/** + * Add a new neighbour + * @arg handle netlink handle + * @arg tmpl template with requested changes + * @arg flags additional netlink message flags + * + * Builds a netlink message by calling rtnl_neigh_build_add_request(), + * sends the request to the kernel and waits for the next ACK to be + * received and thus blocks until the request has been fullfilled. + * + * The following attributes must be set in the template: + * - Interface index (rtnl_neigh_set_ifindex()) + * - State (rtnl_neigh_set_state()) + * - Destination address (rtnl_neigh_set_dst()) + * - Link layer address (rtnl_neigh_set_lladdr()) + * + * @return 0 on sucess or a negative error if an error occured. + */ +int rtnl_neigh_add(struct nl_handle *handle, struct rtnl_neigh *tmpl, int flags) +{ + int err; + struct nl_msg *msg; + + msg = rtnl_neigh_build_add_request(tmpl, flags); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name Neighbour Deletion + * @{ + */ + +/** + * Build a netlink request message to delete a neighbour + * @arg neigh neighbour to delete + * @arg flags additional netlink message flags + * + * Builds a new netlink message requesting a deletion of a neighbour. + * The netlink message header isn't fully equipped with all relevant + * fields and must thus be sent out via nl_send_auto_complete() + * or supplemented as needed. \a neigh must point to an existing + * neighbour. + * + * @return The netlink message + */ +struct nl_msg *rtnl_neigh_build_delete_request(struct rtnl_neigh *neigh, + int flags) +{ + return build_neigh_msg(neigh, RTM_DELNEIGH, flags); +} + +/** + * Delete a neighbour + * @arg handle netlink handle + * @arg neigh neighbour to delete + * @arg flags additional netlink message flags + * + * Builds a netlink message by calling rtnl_neigh_build_delete_request(), + * sends the request to the kernel and waits for the next ACK to be + * received and thus blocks until the request has been fullfilled. + * + * @return 0 on sucess or a negative error if an error occured. + */ +int rtnl_neigh_delete(struct nl_handle *handle, struct rtnl_neigh *neigh, + int flags) +{ + int err; + struct nl_msg *msg; + + msg = rtnl_neigh_build_delete_request(neigh, flags); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name Neighbour Modification + * @{ + */ + +/** + * Build a netlink request message to change neighbour attributes + * @arg neigh the neighbour to change + * @arg flags additional netlink message flags + * + * Builds a new netlink message requesting a change of a neigh + * attributes. The netlink message header isn't fully equipped with + * all relevant fields and must thus be sent out via nl_send_auto_complete() + * or supplemented as needed. + * + * @return The netlink message + * @note Not all attributes can be changed, see + * \ref neigh_changeable "Changeable Attributes" for a list. + */ +struct nl_msg *rtnl_neigh_build_change_request(struct rtnl_neigh *neigh, + int flags) +{ + return build_neigh_msg(neigh, RTM_NEWNEIGH, NLM_F_REPLACE | flags); +} + +/** + * Change neighbour attributes + * @arg handle netlink handle + * @arg neigh neighbour to be changed + * @arg flags additional netlink message flags + * + * Builds a netlink message by calling rtnl_neigh_build_change_request(), + * sends the request to the kernel and waits for the next ACK to be + * received and thus blocks until the request has been fullfilled. + * + * @return 0 on sucess or a negative error if an error occured. + * @note Not all attributes can be changed, see + * \ref neigh_changeable "Changeable Attributes" for a list. + */ +int rtnl_neigh_change(struct nl_handle *handle, struct rtnl_neigh *neigh, + int flags) +{ + int err; + struct nl_msg *msg; + + msg = rtnl_neigh_build_change_request(neigh, flags); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name Neighbour States Translations + * @{ + */ + +static struct trans_tbl neigh_states[] = { + __ADD(NUD_INCOMPLETE, incomplete) + __ADD(NUD_REACHABLE, reachable) + __ADD(NUD_STALE, stale) + __ADD(NUD_DELAY, delay) + __ADD(NUD_PROBE, probe) + __ADD(NUD_FAILED, failed) + __ADD(NUD_NOARP, norarp) + __ADD(NUD_PERMANENT, permanent) +}; + +char * rtnl_neigh_state2str(int state, char *buf, size_t len) +{ + return __flags2str(state, buf, len, neigh_states, + ARRAY_SIZE(neigh_states)); +} + +int rtnl_neigh_str2state(const char *name) +{ + return __str2type(name, neigh_states, ARRAY_SIZE(neigh_states)); +} + +/** @} */ + +/** + * @name Neighbour Flags Translations + * @{ + */ + +static struct trans_tbl neigh_flags[] = { + __ADD(NTF_PROXY, proxy) + __ADD(NTF_ROUTER, router) +}; + +char * rtnl_neigh_flags2str(int flags, char *buf, size_t len) +{ + return __flags2str(flags, buf, len, neigh_flags, + ARRAY_SIZE(neigh_flags)); +} + +int rtnl_neigh_str2flag(const char *name) +{ + return __str2type(name, neigh_flags, ARRAY_SIZE(neigh_flags)); +} + +/** @} */ + +/** + * @name Attributes + * @{ + */ + +void rtnl_neigh_set_state(struct rtnl_neigh *neigh, int state) +{ + neigh->n_state_mask |= state; + neigh->n_state |= state; + neigh->ce_mask |= NEIGH_ATTR_STATE; +} + +int rtnl_neigh_get_state(struct rtnl_neigh *neigh) +{ + if (neigh->ce_mask & NEIGH_ATTR_STATE) + return neigh->n_state; + else + return -1; +} + +void rtnl_neigh_unset_state(struct rtnl_neigh *neigh, int state) +{ + neigh->n_state_mask |= state; + neigh->n_state &= ~state; + neigh->ce_mask |= NEIGH_ATTR_STATE; +} + +void rtnl_neigh_set_flags(struct rtnl_neigh *neigh, unsigned int flags) +{ + neigh->n_flag_mask |= flags; + neigh->n_flags |= flags; + neigh->ce_mask |= NEIGH_ATTR_FLAGS; +} + +unsigned int rtnl_neigh_get_flags(struct rtnl_neigh *neigh) +{ + return neigh->n_flags; +} + +void rtnl_neigh_unset_flags(struct rtnl_neigh *neigh, unsigned int flags) +{ + neigh->n_flag_mask |= flags; + neigh->n_flags &= ~flags; + neigh->ce_mask |= NEIGH_ATTR_FLAGS; +} + +void rtnl_neigh_set_ifindex(struct rtnl_neigh *neigh, int ifindex) +{ + neigh->n_ifindex = ifindex; + neigh->ce_mask |= NEIGH_ATTR_IFINDEX; +} + +int rtnl_neigh_get_ifindex(struct rtnl_neigh *neigh) +{ + if (neigh->ce_mask & NEIGH_ATTR_IFINDEX) + return neigh->n_ifindex; + else + return RTNL_LINK_NOT_FOUND; +} + +static inline int __assign_addr(struct rtnl_neigh *neigh, struct nl_addr **pos, + struct nl_addr *new, int flag, int nocheck) +{ + if (!nocheck) { + if (neigh->ce_mask & NEIGH_ATTR_FAMILY) { + if (new->a_family != neigh->n_family) + return nl_error(EINVAL, + "Address family mismatch"); + } else { + neigh->n_family = new->a_family; + neigh->ce_mask |= NEIGH_ATTR_FAMILY; + } + } + + if (*pos) + nl_addr_put(*pos); + + nl_addr_get(new); + *pos = new; + + neigh->ce_mask |= flag; + + return 0; +} + +void rtnl_neigh_set_lladdr(struct rtnl_neigh *neigh, struct nl_addr *addr) +{ + __assign_addr(neigh, &neigh->n_lladdr, addr, NEIGH_ATTR_LLADDR, 1); +} + +struct nl_addr *rtnl_neigh_get_lladdr(struct rtnl_neigh *neigh) +{ + if (neigh->ce_mask & NEIGH_ATTR_LLADDR) + return neigh->n_lladdr; + else + return NULL; +} + +int rtnl_neigh_set_dst(struct rtnl_neigh *neigh, struct nl_addr *addr) +{ + return __assign_addr(neigh, &neigh->n_dst, addr, + NEIGH_ATTR_DST, 0); +} + +struct nl_addr *rtnl_neigh_get_dst(struct rtnl_neigh *neigh) +{ + if (neigh->ce_mask & NEIGH_ATTR_DST) + return neigh->n_dst; + else + return NULL; +} + +void rtnl_neigh_set_family(struct rtnl_neigh *neigh, int family) +{ + neigh->n_family = family; + neigh->ce_mask |= NEIGH_ATTR_FAMILY; +} + +void rtnl_neigh_set_type(struct rtnl_neigh *neigh, int type) +{ + neigh->n_type = type; + neigh->ce_mask = NEIGH_ATTR_TYPE; +} + +int rtnl_neigh_get_type(struct rtnl_neigh *neigh) +{ + if (neigh->ce_mask & NEIGH_ATTR_TYPE) + return neigh->n_type; + else + return -1; +} + +/** @} */ + +static struct nl_object_ops neigh_obj_ops = { + .oo_name = "route/neigh", + .oo_size = sizeof(struct rtnl_neigh), + .oo_free_data = neigh_free_data, + .oo_clone = neigh_clone, + .oo_dump[NL_DUMP_BRIEF] = neigh_dump_brief, + .oo_dump[NL_DUMP_FULL] = neigh_dump_full, + .oo_dump[NL_DUMP_STATS] = neigh_dump_stats, + .oo_dump[NL_DUMP_XML] = neigh_dump_xml, + .oo_dump[NL_DUMP_ENV] = neigh_dump_env, + .oo_compare = neigh_compare, + .oo_attrs2str = neigh_attrs2str, + .oo_id_attrs = (NEIGH_ATTR_DST | NEIGH_ATTR_FAMILY), +}; + +static struct nl_af_group neigh_groups[] = { + { AF_UNSPEC, RTNLGRP_NEIGH }, + { END_OF_GROUP_LIST }, +}; + +static struct nl_cache_ops rtnl_neigh_ops = { + .co_name = "route/neigh", + .co_hdrsize = sizeof(struct ndmsg), + .co_msgtypes = { + { RTM_NEWNEIGH, NL_ACT_NEW, "new" }, + { RTM_DELNEIGH, NL_ACT_DEL, "del" }, + { RTM_GETNEIGH, NL_ACT_GET, "get" }, + END_OF_MSGTYPES_LIST, + }, + .co_protocol = NETLINK_ROUTE, + .co_groups = neigh_groups, + .co_request_update = neigh_request_update, + .co_msg_parser = neigh_msg_parser, + .co_obj_ops = &neigh_obj_ops, +}; + +static void __init neigh_init(void) +{ + nl_cache_mngt_register(&rtnl_neigh_ops); +} + +static void __exit neigh_exit(void) +{ + nl_cache_mngt_unregister(&rtnl_neigh_ops); +} + +/** @} */ diff --git a/lib/route/neightbl.c b/lib/route/neightbl.c new file mode 100644 index 0000000..afd64a6 --- /dev/null +++ b/lib/route/neightbl.c @@ -0,0 +1,825 @@ +/* + * lib/route/neightbl.c neighbour tables + * + * 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> + */ + +/** + * @ingroup rtnl + * @defgroup neightbl Neighbour Tables + * @brief + * @{ + */ + +#include <netlink-local.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/rtnl.h> +#include <netlink/route/neightbl.h> +#include <netlink/route/link.h> + +/** @cond SKIP */ +#define NEIGHTBL_ATTR_FAMILY 0x001 +#define NEIGHTBL_ATTR_STATS 0x002 +#define NEIGHTBL_ATTR_NAME 0x004 +#define NEIGHTBL_ATTR_THRESH1 0x008 +#define NEIGHTBL_ATTR_THRESH2 0x010 +#define NEIGHTBL_ATTR_THRESH3 0x020 +#define NEIGHTBL_ATTR_CONFIG 0x040 +#define NEIGHTBL_ATTR_PARMS 0x080 +#define NEIGHTBL_ATTR_GC_INTERVAL 0x100 + +#define NEIGHTBLPARM_ATTR_IFINDEX 0x0001 +#define NEIGHTBLPARM_ATTR_REFCNT 0x0002 +#define NEIGHTBLPARM_ATTR_QUEUE_LEN 0x0004 +#define NEIGHTBLPARM_ATTR_APP_PROBES 0x0008 +#define NEIGHTBLPARM_ATTR_UCAST_PROBES 0x0010 +#define NEIGHTBLPARM_ATTR_MCAST_PROBES 0x0020 +#define NEIGHTBLPARM_ATTR_PROXY_QLEN 0x0040 +#define NEIGHTBLPARM_ATTR_REACHABLE_TIME 0x0080 +#define NEIGHTBLPARM_ATTR_BASE_REACHABLE_TIME 0x0100 +#define NEIGHTBLPARM_ATTR_RETRANS_TIME 0x0200 +#define NEIGHTBLPARM_ATTR_GC_STALETIME 0x0400 +#define NEIGHTBLPARM_ATTR_DELAY_PROBE_TIME 0x0800 +#define NEIGHTBLPARM_ATTR_ANYCAST_DELAY 0x1000 +#define NEIGHTBLPARM_ATTR_PROXY_DELAY 0x2000 +#define NEIGHTBLPARM_ATTR_LOCKTIME 0x4000 + +static struct nl_cache_ops rtnl_neightbl_ops; +static struct nl_object_ops neightbl_obj_ops; +/** @endcond */ + +static int neightbl_compare(struct nl_object *_a, struct nl_object *_b, + uint32_t attrs, int flags) +{ + struct rtnl_neightbl *a = (struct rtnl_neightbl *) _a; + struct rtnl_neightbl *b = (struct rtnl_neightbl *) _b; + int diff = 0; + +#define NT_DIFF(ATTR, EXPR) ATTR_DIFF(attrs, NEIGHTBL_ATTR_##ATTR, a, b, EXPR) + + diff |= NT_DIFF(FAMILY, a->nt_family != b->nt_family); + diff |= NT_DIFF(NAME, strcmp(a->nt_name, b->nt_name)); + diff |= NT_DIFF(THRESH1, a->nt_gc_thresh1 != b->nt_gc_thresh1); + diff |= NT_DIFF(THRESH2, a->nt_gc_thresh2 != b->nt_gc_thresh2); + diff |= NT_DIFF(THRESH3, a->nt_gc_thresh3 != b->nt_gc_thresh3); + diff |= NT_DIFF(GC_INTERVAL, a->nt_gc_interval != b->nt_gc_interval); + +#undef NT_DIFF + + if (!(a->ce_mask & NEIGHTBL_ATTR_PARMS) && + !(b->ce_mask & NEIGHTBL_ATTR_PARMS)) + return diff; + + /* XXX: FIXME: Compare parameter table */ + + +#if 0 +#define REQ(F) (fp->ntp_mask & NEIGHTBLPARM_ATTR_##F) +#define AVAIL(F) (op->ntp_mask & NEIGHTBLPARM_ATTR_##F) +#define _C(F, N) (REQ(F) && (!AVAIL(F) || (op->N != fp->N))) + if (_C(IFINDEX, ntp_ifindex) || + _C(QUEUE_LEN, ntp_queue_len) || + _C(APP_PROBES, ntp_app_probes) || + _C(UCAST_PROBES, ntp_ucast_probes) || + _C(MCAST_PROBES, ntp_mcast_probes) || + _C(PROXY_QLEN, ntp_proxy_qlen) || + _C(LOCKTIME, ntp_locktime) || + _C(RETRANS_TIME, ntp_retrans_time) || + _C(BASE_REACHABLE_TIME, ntp_base_reachable_time) || + _C(GC_STALETIME, ntp_gc_stale_time) || + _C(DELAY_PROBE_TIME, ntp_probe_delay) || + _C(ANYCAST_DELAY, ntp_anycast_delay) || + _C(PROXY_DELAY, ntp_proxy_delay)) + return 0; +#undef REQ +#undef AVAIL +#undef _C +#endif + + return diff; +} + + +static struct nla_policy neightbl_policy[NDTA_MAX+1] = { + [NDTA_NAME] = { .type = NLA_STRING, + .maxlen = NTBLNAMSIZ }, + [NDTA_THRESH1] = { .type = NLA_U32 }, + [NDTA_THRESH2] = { .type = NLA_U32 }, + [NDTA_THRESH3] = { .type = NLA_U32 }, + [NDTA_GC_INTERVAL] = { .type = NLA_U32 }, + [NDTA_CONFIG] = { .minlen = sizeof(struct ndt_config) }, + [NDTA_STATS] = { .minlen = sizeof(struct ndt_stats) }, + [NDTA_PARMS] = { .type = NLA_NESTED }, +}; + +static int neightbl_msg_parser(struct nl_cache_ops *ops, + struct sockaddr_nl *who, struct nlmsghdr *n, + void *arg) +{ + struct rtnl_neightbl *ntbl; + struct nlattr *tb[NDTA_MAX + 1]; + struct nl_parser_param *pp = arg; + struct rtgenmsg *rtmsg; + int err; + + ntbl = rtnl_neightbl_alloc(); + if (!ntbl) { + err = nl_errno(ENOMEM); + goto errout; + } + + ntbl->ce_msgtype = n->nlmsg_type; + rtmsg = nlmsg_data(n); + + err = nlmsg_parse(n, sizeof(*rtmsg), tb, NDTA_MAX, neightbl_policy); + if (err < 0) + goto errout; + + ntbl->nt_family = rtmsg->rtgen_family; + + if (tb[NDTA_NAME] == NULL) { + err = nl_error(EINVAL, "NDTA_NAME is missing"); + goto errout; + } + + nla_strlcpy(ntbl->nt_name, tb[NDTA_NAME], NTBLNAMSIZ); + ntbl->ce_mask |= NEIGHTBL_ATTR_NAME; + + if (tb[NDTA_THRESH1]) { + ntbl->nt_gc_thresh1 = nla_get_u32(tb[NDTA_THRESH1]); + ntbl->ce_mask |= NEIGHTBL_ATTR_THRESH1; + } + + if (tb[NDTA_THRESH2]) { + ntbl->nt_gc_thresh2 = nla_get_u32(tb[NDTA_THRESH2]); + ntbl->ce_mask |= NEIGHTBL_ATTR_THRESH2; + } + + if (tb[NDTA_THRESH3]) { + ntbl->nt_gc_thresh3 = nla_get_u32(tb[NDTA_THRESH3]); + ntbl->ce_mask |= NEIGHTBL_ATTR_THRESH3; + } + + if (tb[NDTA_GC_INTERVAL]) { + ntbl->nt_gc_interval = nla_get_u32(tb[NDTA_GC_INTERVAL]); + ntbl->ce_mask |= NEIGHTBL_ATTR_GC_INTERVAL; + } + + if (tb[NDTA_CONFIG]) { + nla_memcpy(&ntbl->nt_config, tb[NDTA_CONFIG], + sizeof(ntbl->nt_config)); + ntbl->ce_mask |= NEIGHTBL_ATTR_CONFIG; + } + + if (tb[NDTA_STATS]) { + nla_memcpy(&ntbl->nt_stats, tb[NDTA_STATS], + sizeof(ntbl->nt_stats)); + ntbl->ce_mask |= NEIGHTBL_ATTR_STATS; + } + + if (tb[NDTA_PARMS]) { + struct nlattr *tbp[NDTPA_MAX + 1]; + struct rtnl_neightbl_parms *p = &ntbl->nt_parms; + + err = nla_parse_nested(tbp, NDTPA_MAX, tb[NDTA_PARMS], NULL); + if (err < 0) + goto errout; + +#define COPY_ENTRY(name, var) \ + if (tbp[NDTPA_ ##name]) { \ + p->ntp_ ##var = nla_get_u32(tbp[NDTPA_ ##name]); \ + p->ntp_mask |= NEIGHTBLPARM_ATTR_ ##name; \ + } + + COPY_ENTRY(IFINDEX, ifindex); + COPY_ENTRY(REFCNT, refcnt); + COPY_ENTRY(QUEUE_LEN, queue_len); + COPY_ENTRY(APP_PROBES, app_probes); + COPY_ENTRY(UCAST_PROBES, ucast_probes); + COPY_ENTRY(MCAST_PROBES, mcast_probes); + COPY_ENTRY(PROXY_QLEN, proxy_qlen); + COPY_ENTRY(PROXY_DELAY, proxy_delay); + COPY_ENTRY(ANYCAST_DELAY, anycast_delay); + COPY_ENTRY(LOCKTIME, locktime); + COPY_ENTRY(REACHABLE_TIME, reachable_time); + COPY_ENTRY(BASE_REACHABLE_TIME, base_reachable_time); + COPY_ENTRY(RETRANS_TIME, retrans_time); + COPY_ENTRY(GC_STALETIME, gc_stale_time); + COPY_ENTRY(DELAY_PROBE_TIME, probe_delay); +#undef COPY_ENTRY + + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; + } + + err = pp->pp_cb((struct nl_object *) ntbl, pp); + if (err < 0) + goto errout; + + return P_ACCEPT; +errout: + rtnl_neightbl_put(ntbl); + return err; +} + +static int neightbl_request_update(struct nl_cache *c, struct nl_handle *h) +{ + return nl_rtgen_request(h, RTM_GETNEIGHTBL, AF_UNSPEC, NLM_F_DUMP); +} + + +static int neightbl_dump_brief(struct nl_object *arg, struct nl_dump_params *p) +{ + int line = 1; + struct rtnl_neightbl *ntbl = (struct rtnl_neightbl *) arg; + + dp_dump(p, "%s", ntbl->nt_name); + + if (ntbl->nt_parms.ntp_mask & NEIGHTBLPARM_ATTR_IFINDEX) { + struct nl_cache *link_cache; + + link_cache = nl_cache_mngt_require("route/link"); + + if (link_cache) { + char buf[32]; + dp_dump(p, "<%s> ", + rtnl_link_i2name(link_cache, + ntbl->nt_parms.ntp_ifindex, + buf, sizeof(buf))); + } else + dp_dump(p, "<%u> ", ntbl->nt_parms.ntp_ifindex); + } else + dp_dump(p, " "); + + if (ntbl->ce_mask & NEIGHTBL_ATTR_CONFIG) + dp_dump(p, "entries %u ", ntbl->nt_config.ndtc_entries); + + if (ntbl->ce_mask & NEIGHTBL_ATTR_PARMS) { + char rt[32], rt2[32]; + struct rtnl_neightbl_parms *pa = &ntbl->nt_parms; + + dp_dump(p, "reachable-time %s retransmit-time %s", + nl_msec2str(pa->ntp_reachable_time, rt, sizeof(rt)), + nl_msec2str(pa->ntp_retrans_time, rt2, sizeof(rt2))); + } + + dp_dump(p, "\n"); + + return line; +} + +static int neightbl_dump_full(struct nl_object *arg, struct nl_dump_params *p) +{ + char x[32], y[32], z[32]; + struct rtnl_neightbl *ntbl = (struct rtnl_neightbl *) arg; + + int line = neightbl_dump_brief(arg, p); + + if (ntbl->ce_mask & NEIGHTBL_ATTR_CONFIG) { + dp_new_line(p, line++); + dp_dump(p, " key-len %u entry-size %u last-flush %s\n", + ntbl->nt_config.ndtc_key_len, + ntbl->nt_config.ndtc_entry_size, + nl_msec2str(ntbl->nt_config.ndtc_last_flush, + x, sizeof(x))); + + dp_new_line(p, line++); + dp_dump(p, " gc threshold %u/%u/%u interval %s " \ + "chain-position %u\n", + ntbl->nt_gc_thresh1, ntbl->nt_gc_thresh2, + ntbl->nt_gc_thresh3, + nl_msec2str(ntbl->nt_gc_interval, x, sizeof(x)), + ntbl->nt_config.ndtc_hash_chain_gc); + + dp_new_line(p, line++); + dp_dump(p, " hash-rand 0x%08X/0x%08X last-rand %s\n", + ntbl->nt_config.ndtc_hash_rnd, + ntbl->nt_config.ndtc_hash_mask, + nl_msec2str(ntbl->nt_config.ndtc_last_rand, + x, sizeof(x))); + } + + if (ntbl->ce_mask & NEIGHTBL_ATTR_PARMS) { + struct rtnl_neightbl_parms *pa = &ntbl->nt_parms; + + dp_new_line(p, line++); + dp_dump(p, " refcnt %u pending-queue-limit %u " \ + "proxy-delayed-queue-limit %u\n", + pa->ntp_refcnt, + pa->ntp_queue_len, + pa->ntp_proxy_qlen); + + dp_new_line(p, line++); + dp_dump(p, " num-userspace-probes %u num-unicast-probes " \ + "%u num-multicast-probes %u\n", + pa->ntp_app_probes, + pa->ntp_ucast_probes, + pa->ntp_mcast_probes); + + dp_new_line(p, line++); + dp_dump(p, " min-age %s base-reachable-time %s " \ + "stale-check-interval %s\n", + nl_msec2str(pa->ntp_locktime, x, sizeof(x)), + nl_msec2str(pa->ntp_base_reachable_time, + y, sizeof(y)), + nl_msec2str(pa->ntp_gc_stale_time, z, sizeof(z))); + + dp_new_line(p, line++); + dp_dump(p, " initial-probe-delay %s answer-delay %s " \ + "proxy-answer-delay %s\n", + nl_msec2str(pa->ntp_probe_delay, x, sizeof(x)), + nl_msec2str(pa->ntp_anycast_delay, y, sizeof(y)), + nl_msec2str(pa->ntp_proxy_delay, z, sizeof(z))); + } + + return line; +} + +static int neightbl_dump_stats(struct nl_object *arg, struct nl_dump_params *p) +{ + struct rtnl_neightbl *ntbl = (struct rtnl_neightbl *) arg; + int line = neightbl_dump_full(arg, p); + + if (!(ntbl->ce_mask & NEIGHTBL_ATTR_STATS)) + return line; + + dp_new_line(p, line++); + dp_dump(p, " lookups %lld hits %lld failed %lld " \ + "allocations %lld destroys %lld\n", + ntbl->nt_stats.ndts_lookups, + ntbl->nt_stats.ndts_hits, + ntbl->nt_stats.ndts_res_failed, + ntbl->nt_stats.ndts_allocs, + ntbl->nt_stats.ndts_destroys); + + dp_new_line(p, line++); + dp_dump(p, " hash-grows %lld forced-gc-runs %lld " \ + "periodic-gc-runs %lld\n", + ntbl->nt_stats.ndts_hash_grows, + ntbl->nt_stats.ndts_forced_gc_runs, + ntbl->nt_stats.ndts_periodic_gc_runs); + + dp_dump(p, " rcv-unicast-probes %lld rcv-multicast-probes %lld\n", + ntbl->nt_stats.ndts_rcv_probes_ucast, + ntbl->nt_stats.ndts_rcv_probes_mcast); + + return line; +} + +/** + * @name Allocation/Freeing + * @{ + */ + +struct rtnl_neightbl *rtnl_neightbl_alloc(void) +{ + return (struct rtnl_neightbl *) nl_object_alloc(&neightbl_obj_ops); +} + +void rtnl_neightbl_put(struct rtnl_neightbl *neightbl) +{ + nl_object_put((struct nl_object *) neightbl); +} + +/** @} */ + +/** + * @name Neighbour Table Cache Management + * @{ + */ + +/** + * Build a neighbour table cache including all neighbour tables currently configured in the kernel. + * @arg handle netlink handle + * + * Allocates a new neighbour table cache, initializes it properly and + * updates it to include all neighbour tables currently configured in + * the kernel. + * + * @note The caller is responsible for destroying and freeing the + * cache after using it. + * @return The new cache or NULL if an error occured. + */ +struct nl_cache * rtnl_neightbl_alloc_cache(struct nl_handle *handle) +{ + struct nl_cache * cache; + + cache = nl_cache_alloc(&rtnl_neightbl_ops); + if (cache == NULL) + return NULL; + + if (handle && nl_cache_refill(handle, cache) < 0) { + nl_cache_free(cache); + return NULL; + } + + return cache; +} + +/** + * Lookup neighbour table by name and optional interface index + * @arg cache neighbour table cache + * @arg name name of table + * @arg ifindex optional interface index + * + * Looks up the neighbour table matching the specified name and + * optionally the specified ifindex to retrieve device specific + * parameter sets. + * + * @return ptr to neighbour table inside the cache or NULL if no + * match was found. + */ +struct rtnl_neightbl *rtnl_neightbl_get(struct nl_cache *cache, + const char *name, int ifindex) +{ + struct rtnl_neightbl *nt; + + if (cache->c_ops != &rtnl_neightbl_ops) + return NULL; + + nl_list_for_each_entry(nt, &cache->c_items, ce_list) { + if (!strcasecmp(nt->nt_name, name) && + ((!ifindex && !nt->nt_parms.ntp_ifindex) || + (ifindex && ifindex == nt->nt_parms.ntp_ifindex))) { + nl_object_get((struct nl_object *) nt); + return nt; + } + } + + return NULL; +} + +/** @} */ + +/** + * @name Neighbour Table Modifications + * @{ + */ + +/** + * Builds a netlink change request message to change neighbour table attributes + * @arg old neighbour table to change + * @arg tmpl template with requested changes + * + * Builds a new netlink message requesting a change of neighbour table + * attributes. The netlink message header isn't fully equipped with all + * relevant fields and must be sent out via nl_send_auto_complete() or + * supplemented as needed. + * \a old must point to a neighbour table currently configured in the + * kernel and \a tmpl must contain the attributes to be changed set via + * \c rtnl_neightbl_set_* functions. + * + * @return New netlink message + */ +struct nl_msg * rtnl_neightbl_build_change_request(struct rtnl_neightbl *old, + struct rtnl_neightbl *tmpl) +{ + struct nl_msg *m; + struct ndtmsg ndt = { + .ndtm_family = old->nt_family, + }; + + m = nlmsg_alloc_simple(RTM_SETNEIGHTBL, 0); + nlmsg_append(m, &ndt, sizeof(ndt), NLMSG_ALIGNTO); + + nla_put_string(m, NDTA_NAME, old->nt_name); + + if (tmpl->ce_mask & NEIGHTBL_ATTR_THRESH1) + nla_put_u32(m, NDTA_THRESH1, tmpl->nt_gc_thresh1); + + if (tmpl->ce_mask & NEIGHTBL_ATTR_THRESH2) + nla_put_u32(m, NDTA_THRESH2, tmpl->nt_gc_thresh2); + + if (tmpl->ce_mask & NEIGHTBL_ATTR_THRESH2) + nla_put_u32(m, NDTA_THRESH2, tmpl->nt_gc_thresh2); + + if (tmpl->ce_mask & NEIGHTBL_ATTR_GC_INTERVAL) + nla_put_u64(m, NDTA_GC_INTERVAL, + tmpl->nt_gc_interval); + + if (tmpl->ce_mask & NEIGHTBL_ATTR_PARMS) { + struct rtnl_neightbl_parms *p = &tmpl->nt_parms; + struct nl_msg *parms = nlmsg_alloc(); + + if (old->nt_parms.ntp_mask & NEIGHTBLPARM_ATTR_IFINDEX) + nla_put_u32(parms, NDTPA_IFINDEX, + old->nt_parms.ntp_ifindex); + + + if (p->ntp_mask & NEIGHTBLPARM_ATTR_QUEUE_LEN) + nla_put_u32(parms, NDTPA_QUEUE_LEN, p->ntp_queue_len); + + if (p->ntp_mask & NEIGHTBLPARM_ATTR_APP_PROBES) + nla_put_u32(parms, NDTPA_APP_PROBES, p->ntp_app_probes); + + if (p->ntp_mask & NEIGHTBLPARM_ATTR_UCAST_PROBES) + nla_put_u32(parms, NDTPA_UCAST_PROBES, + p->ntp_ucast_probes); + + if (p->ntp_mask & NEIGHTBLPARM_ATTR_MCAST_PROBES) + nla_put_u32(parms, NDTPA_MCAST_PROBES, + p->ntp_mcast_probes); + + if (p->ntp_mask & NEIGHTBLPARM_ATTR_PROXY_QLEN) + nla_put_u32(parms, NDTPA_PROXY_QLEN, + p->ntp_proxy_qlen); + + if (p->ntp_mask & NEIGHTBLPARM_ATTR_BASE_REACHABLE_TIME) + nla_put_u64(parms, NDTPA_BASE_REACHABLE_TIME, + p->ntp_base_reachable_time); + + if (p->ntp_mask & NEIGHTBLPARM_ATTR_RETRANS_TIME) + nla_put_u64(parms, NDTPA_RETRANS_TIME, + p->ntp_retrans_time); + + if (p->ntp_mask & NEIGHTBLPARM_ATTR_GC_STALETIME) + nla_put_u64(parms, NDTPA_GC_STALETIME, + p->ntp_gc_stale_time); + + if (p->ntp_mask & NEIGHTBLPARM_ATTR_DELAY_PROBE_TIME) + nla_put_u64(parms, NDTPA_DELAY_PROBE_TIME, + p->ntp_proxy_delay); + + if (p->ntp_mask & NEIGHTBLPARM_ATTR_ANYCAST_DELAY) + nla_put_u64(parms, NDTPA_ANYCAST_DELAY, + p->ntp_anycast_delay); + + if (p->ntp_mask & NEIGHTBLPARM_ATTR_PROXY_DELAY) + nla_put_u64(parms, NDTPA_PROXY_DELAY, + p->ntp_proxy_delay); + + if (p->ntp_mask & NEIGHTBLPARM_ATTR_LOCKTIME) + nla_put_u64(parms, NDTPA_LOCKTIME, p->ntp_locktime); + + nla_put_nested(m, NDTA_PARMS, parms); + nlmsg_free(parms); + } + + return m; +} + +/** + * Change neighbour table attributes + * @arg handle netlink handle + * @arg old neighbour table to be changed + * @arg tmpl template with requested changes + * + * Builds a new netlink message by calling + * rtnl_neightbl_build_change_request(), sends the request to the + * kernel and waits for the next ACK to be received, i.e. blocks + * until the request has been processed. + * + * @return 0 on success or a negative error code + */ +int rtnl_neightbl_change(struct nl_handle *handle, struct rtnl_neightbl *old, + struct rtnl_neightbl *tmpl) +{ + int err; + struct nl_msg *msg; + + msg = rtnl_neightbl_build_change_request(old, tmpl); + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name Attribute Modification + * @{ + */ + +void rtnl_neightbl_set_family(struct rtnl_neightbl *ntbl, int family) +{ + ntbl->nt_family = family; + ntbl->ce_mask |= NEIGHTBL_ATTR_FAMILY; +} + +void rtnl_neightbl_set_gc_interval(struct rtnl_neightbl *ntbl, uint64_t ms) +{ + ntbl->nt_gc_interval = ms; + ntbl->ce_mask |= NEIGHTBL_ATTR_GC_INTERVAL; +} + +void rtnl_neightbl_set_gc_tresh1(struct rtnl_neightbl *ntbl, int thresh) +{ + ntbl->nt_gc_thresh1 = thresh; + ntbl->ce_mask |= NEIGHTBL_ATTR_THRESH1; +} + +void rtnl_neightbl_set_gc_tresh2(struct rtnl_neightbl *ntbl, int thresh) +{ + ntbl->nt_gc_thresh2 = thresh; + ntbl->ce_mask |= NEIGHTBL_ATTR_THRESH2; +} + +void rtnl_neightbl_set_gc_tresh3(struct rtnl_neightbl *ntbl, int thresh) +{ + ntbl->nt_gc_thresh3 = thresh; + ntbl->ce_mask |= NEIGHTBL_ATTR_THRESH3; +} + +void rtnl_neightbl_set_name(struct rtnl_neightbl *ntbl, const char *name) +{ + strncpy(ntbl->nt_name, name, sizeof(ntbl->nt_name) - 1); + ntbl->ce_mask |= NEIGHTBL_ATTR_NAME; +} + +void rtnl_neightbl_set_dev(struct rtnl_neightbl *ntbl, int ifindex) +{ + ntbl->nt_parms.ntp_ifindex = ifindex; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_IFINDEX; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** + * Set the queue length for pending requests of a neighbour table to the specified value + * @arg ntbl neighbour table to change + * @arg len new queue len + */ +void rtnl_neightbl_set_queue_len(struct rtnl_neightbl *ntbl, int len) +{ + ntbl->nt_parms.ntp_queue_len = len; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_QUEUE_LEN; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** + * Set the queue length for delay proxy arp requests of a neighbour table to the specified value + * @arg ntbl neighbour table to change + * @arg len new queue len + */ +void rtnl_neightbl_set_proxy_queue_len(struct rtnl_neightbl *ntbl, int len) +{ + ntbl->nt_parms.ntp_proxy_qlen = len; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_PROXY_QLEN; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** + * Set the number of application probes of a neighbour table to the specified value + * @arg ntbl neighbour table to change + * @arg probes new probes value + */ +void rtnl_neightbl_set_app_probes(struct rtnl_neightbl *ntbl, int probes) +{ + ntbl->nt_parms.ntp_app_probes = probes; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_APP_PROBES; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** + * Set the number of unicast probes of a neighbour table to the specified value + * @arg ntbl neighbour table to change + * @arg probes new probes value + */ +void rtnl_neightbl_set_ucast_probes(struct rtnl_neightbl *ntbl, int probes) +{ + ntbl->nt_parms.ntp_ucast_probes = probes; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_UCAST_PROBES; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** + * Set the number of multicast probes of a neighbour table to the specified value + * @arg ntbl neighbour table to change + * @arg probes new probes value + */ +void rtnl_neightbl_set_mcast_probes(struct rtnl_neightbl *ntbl, int probes) +{ + ntbl->nt_parms.ntp_mcast_probes = probes; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_MCAST_PROBES; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** + * Set the base reachable time of a neighbour table to the specified value + * @arg ntbl neighbour table to change + * @arg ms new base reachable time in milliseconds + */ +void rtnl_neightbl_set_base_reachable_time(struct rtnl_neightbl *ntbl, + uint64_t ms) +{ + ntbl->nt_parms.ntp_base_reachable_time = ms; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_BASE_REACHABLE_TIME; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** + * Set the retransmit time of a neighbour table to the specified value + * @arg ntbl neighbour table to change + * @arg ms new retransmit time + */ +void rtnl_neightbl_set_retrans_time(struct rtnl_neightbl *ntbl, uint64_t ms) +{ + ntbl->nt_parms.ntp_retrans_time = ms; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_RETRANS_TIME; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** + * Set the gc stale time of a neighbour table to the specified value + * @arg ntbl neighbour table to change + * @arg ms new gc stale time in milliseconds + */ +void rtnl_neightbl_set_gc_stale_time(struct rtnl_neightbl *ntbl, uint64_t ms) +{ + ntbl->nt_parms.ntp_gc_stale_time = ms; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_GC_STALETIME; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** + * Set the first probe delay time of a neighbour table to the specified value + * @arg ntbl neighbour table to change + * @arg ms new first probe delay time in milliseconds + */ +void rtnl_neightbl_set_delay_probe_time(struct rtnl_neightbl *ntbl, uint64_t ms) +{ + ntbl->nt_parms.ntp_probe_delay = ms; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_DELAY_PROBE_TIME; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** + * Set the anycast delay of a neighbour table to the specified value + * @arg ntbl neighbour table to change + * @arg ms new anycast delay in milliseconds + */ +void rtnl_neightbl_set_anycast_delay(struct rtnl_neightbl *ntbl, uint64_t ms) +{ + ntbl->nt_parms.ntp_anycast_delay = ms; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_ANYCAST_DELAY; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** + * Set the proxy delay of a neighbour table to the specified value + * @arg ntbl neighbour table to change + * @arg ms new proxy delay in milliseconds + */ +void rtnl_neightbl_set_proxy_delay(struct rtnl_neightbl *ntbl, uint64_t ms) +{ + ntbl->nt_parms.ntp_proxy_delay = ms; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_PROXY_DELAY; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** + * Set the locktime of a neighbour table to the specified value + * @arg ntbl neighbour table to change + * @arg ms new locktime in milliseconds + */ +void rtnl_neightbl_set_locktime(struct rtnl_neightbl *ntbl, uint64_t ms) +{ + ntbl->nt_parms.ntp_locktime = ms; + ntbl->nt_parms.ntp_mask |= NEIGHTBLPARM_ATTR_LOCKTIME; + ntbl->ce_mask |= NEIGHTBL_ATTR_PARMS; +} + +/** @} */ + +static struct nl_object_ops neightbl_obj_ops = { + .oo_name = "route/neightbl", + .oo_size = sizeof(struct rtnl_neightbl), + .oo_dump[NL_DUMP_BRIEF] = neightbl_dump_brief, + .oo_dump[NL_DUMP_FULL] = neightbl_dump_full, + .oo_dump[NL_DUMP_STATS] = neightbl_dump_stats, + .oo_compare = neightbl_compare, +}; + +static struct nl_cache_ops rtnl_neightbl_ops = { + .co_name = "route/neightbl", + .co_hdrsize = sizeof(struct rtgenmsg), + .co_msgtypes = { + { RTM_NEWNEIGHTBL, NL_ACT_NEW, "new" }, + { RTM_SETNEIGHTBL, NL_ACT_SET, "set" }, + { RTM_GETNEIGHTBL, NL_ACT_GET, "get" }, + END_OF_MSGTYPES_LIST, + }, + .co_protocol = NETLINK_ROUTE, + .co_request_update = neightbl_request_update, + .co_msg_parser = neightbl_msg_parser, + .co_obj_ops = &neightbl_obj_ops, +}; + +static void __init neightbl_init(void) +{ + nl_cache_mngt_register(&rtnl_neightbl_ops); +} + +static void __exit neightbl_exit(void) +{ + nl_cache_mngt_unregister(&rtnl_neightbl_ops); +} + +/** @} */ diff --git a/lib/route/nexthop.c b/lib/route/nexthop.c new file mode 100644 index 0000000..7486769 --- /dev/null +++ b/lib/route/nexthop.c @@ -0,0 +1,151 @@ +/* + * lib/route/nexthop.c Routing Nexthop + * + * 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> + */ + +/** + * @ingroup route_obj + * @defgroup nexthop Nexthop + * @{ + */ + +#include <netlink-local.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/rtnl.h> +#include <netlink/route/route.h> + +/** + * @name Allocation/Freeing + * @{ + */ + +struct rtnl_nexthop *rtnl_route_nh_alloc(void) +{ + struct rtnl_nexthop *nh; + + nh = calloc(1, sizeof(*nh)); + if (!nh) { + nl_errno(ENOMEM); + return NULL; + } + + nl_init_list_head(&nh->rtnh_list); + + return nh; +} + +struct rtnl_nexthop *rtnl_route_nh_clone(struct rtnl_nexthop *src) +{ + struct rtnl_nexthop *nh; + + nh = rtnl_route_nh_alloc(); + if (!nh) + return NULL; + + nh->rtnh_flags = src->rtnh_flags; + nh->rtnh_flag_mask = src->rtnh_flag_mask; + nh->rtnh_weight = src->rtnh_weight; + nh->rtnh_ifindex = src->rtnh_ifindex; + nh->rtnh_mask = src->rtnh_mask; + + if (src->rtnh_gateway) { + nh->rtnh_gateway = nl_addr_clone(src->rtnh_gateway); + if (!nh->rtnh_gateway) { + free(nh); + return NULL; + } + } + + return nh; +} + +void rtnl_route_nh_free(struct rtnl_nexthop *nh) +{ + nl_addr_put(nh->rtnh_gateway); + free(nh); +} + +/** @} */ + +/** + * @name Attributes + */ + +void rtnl_route_nh_set_weight(struct rtnl_nexthop *nh, int weight) +{ + nh->rtnh_weight = weight; + nh->rtnh_mask |= NEXTHOP_HAS_WEIGHT; +} + +int rtnl_route_nh_get_weight(struct rtnl_nexthop *nh) +{ + if (nh->rtnh_mask & NEXTHOP_HAS_WEIGHT) + return nh->rtnh_weight; + else + return 0; +} + +void rtnl_route_nh_set_ifindex(struct rtnl_nexthop *nh, int ifindex) +{ + nh->rtnh_ifindex = ifindex; + nh->rtnh_mask |= NEXTHOP_HAS_IFINDEX; +} + +int rtnl_route_nh_get_ifindex(struct rtnl_nexthop *nh) +{ + if (nh->rtnh_mask & NEXTHOP_HAS_IFINDEX) + return nh->rtnh_ifindex; + else + return -1; +} + +void rtnl_route_nh_set_gateway(struct rtnl_nexthop *nh, struct nl_addr *addr) +{ + struct nl_addr *old = nh->rtnh_gateway; + + nh->rtnh_gateway = nl_addr_get(addr); + if (old) + nl_addr_put(old); + + nh->rtnh_mask |= NEXTHOP_HAS_GATEWAY; +} + +struct nl_addr *rtnl_route_nh_get_gateway(struct rtnl_nexthop *nh) +{ + if (nh->rtnh_mask & NEXTHOP_HAS_GATEWAY) + return nh->rtnh_gateway; + else + return NULL; +} + +void rtnl_route_nh_set_flags(struct rtnl_nexthop *nh, unsigned int flags) +{ + nh->rtnh_flag_mask |= flags; + nh->rtnh_flags |= flags; + nh->rtnh_mask |= NEXTHOP_HAS_FLAGS; +} + +void rtnl_route_nh_unset_flags(struct rtnl_nexthop *nh, unsigned int flags) +{ + nh->rtnh_flag_mask |= flags; + nh->rtnh_flags &= ~flags; + nh->rtnh_mask |= NEXTHOP_HAS_FLAGS; +} + +unsigned int rtnl_route_nh_get_flags(struct rtnl_nexthop *nh) +{ + if (nh->rtnh_mask & NEXTHOP_HAS_FLAGS) + return nh->rtnh_flags; + else + return 0; +} + +/** @} */ +/** @} */ diff --git a/lib/route/qdisc.c b/lib/route/qdisc.c new file mode 100644 index 0000000..818ca7b --- /dev/null +++ b/lib/route/qdisc.c @@ -0,0 +1,478 @@ +/* + * lib/route/qdisc.c Queueing Disciplines + * + * 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> + */ + +/** + * @ingroup tc + * @defgroup qdisc Queueing Disciplines + * + * @par Qdisc Handles + * In general, qdiscs are identified by the major part of a traffic control + * handle (the upper 16 bits). A few special values exist though: + * - \c TC_H_ROOT: root qdisc (directly attached to the device) + * - \c TC_H_INGRESS: ingress qdisc (directly attached to the device) + * - \c TC_H_UNSPEC: unspecified qdisc (no reference) + * + * @par 1) Adding a Qdisc + * @code + * // Allocate a new empty qdisc to be filled out + * struct rtnl_qdisc *qdisc = rtnl_qdisc_alloc(); + * + * // ... specify the kind of the Qdisc + * rtnl_qdisc_set_kind(qdisc, "pfifo"); + * + * // Specify the device the qdisc should be attached to + * rtnl_qdisc_set_ifindex(qdisc, ifindex); + * + * // ... specify the parent qdisc + * rtnl_qdisc_set_parent(qdisc, TC_H_ROOT); + * + * // Specifying the handle is not required but makes reidentifying easier + * // and may help to avoid adding a qdisc twice. + * rtnl_qdisc_set_handle(qdisc, 0x000A0000); + * + * // Now on to specify the qdisc specific options, see the relevant qdisc + * // modules for documentation, in this example we set the upper limit of + * // the packet fifo qdisc to 64 + * rtnl_qdisc_fifo_set_limit(qdisc, 64); + * + * rtnl_qdisc_add(handle, qdisc, NLM_R_REPLACE); + * + * // Free up the memory + * rtnl_qdisc_put(qdisc); + * @endcode + * + * @par 2) Deleting a Qdisc + * @code + * // Allocate a new empty qdisc to be filled out with the parameters + * // specifying the qdisc to be deleted. Alternatively a fully equiped + * // Qdisc object from a cache can be used. + * struct rtnl_qdisc *qdisc = rtnl_qdisc_alloc(); + * + * // The interface index of the device the qdisc is on and the parent handle + * // are the least required fields to be filled out. + * // Note: Specify TC_H_ROOT or TC_H_INGRESS as parent handle to delete the + * // root respectively root ingress qdisc. + * rtnl_qdisc_set_ifindex(qdisc, ifindex); + * rtnl_qdisc_set_parent(qdisc, parent_handle); + * + * // If required for identification, the handle can be specified as well. + * rtnl_qdisc_set_handle(qdisc, qdisc_handle); + * + * // Not required but maybe helpful as sanity check, the kind of the qdisc + * // can be specified to avoid mistakes. + * rtnl_qdisc_set_kind(qdisc, "pfifo"); + * + * // Finally delete the qdisc with rtnl_qdisc_delete(), alternatively + * // rtnl_qdisc_build_delete_request() can be invoked to generate an + * // appropritate netlink message to send out. + * rtnl_qdisc_delete(handle, qdisc); + * + * // Free up the memory + * rtnl_qdisc_put(qdisc); + * @endcode + * + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/link.h> +#include <netlink/route/tc.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/class.h> +#include <netlink/route/classifier.h> +#include <netlink/route/qdisc-modules.h> + +static struct nl_cache_ops rtnl_qdisc_ops; + +static int qdisc_msg_parser(struct nl_cache_ops *ops, struct sockaddr_nl *who, + struct nlmsghdr *n, void *arg) +{ + int err = -ENOMEM; + struct nl_parser_param *pp = arg; + struct rtnl_qdisc *qdisc; + struct rtnl_qdisc_ops *qops; + + qdisc = rtnl_qdisc_alloc(); + if (!qdisc) { + err = nl_errno(ENOMEM); + goto errout; + } + + qdisc->ce_msgtype = n->nlmsg_type; + + err = tca_msg_parser(n, (struct rtnl_tca *) qdisc); + if (err < 0) + goto errout_free; + + qops = rtnl_qdisc_lookup_ops(qdisc); + if (qops && qops->qo_msg_parser) { + err = qops->qo_msg_parser(qdisc); + if (err < 0) + goto errout_free; + } + + err = pp->pp_cb((struct nl_object *) qdisc, pp); + if (err < 0) + goto errout_free; + + return P_ACCEPT; + +errout_free: + rtnl_qdisc_put(qdisc); +errout: + return err; +} + +static int qdisc_request_update(struct nl_cache *c, struct nl_handle *h) +{ + struct tcmsg tchdr = { + .tcm_family = AF_UNSPEC, + .tcm_ifindex = c->c_iarg1, + }; + + return nl_send_simple(h, RTM_GETQDISC, NLM_F_DUMP, &tchdr, + sizeof(tchdr)); +} + +/** + * @name QDisc Addition + * @{ + */ + +static struct nl_msg *qdisc_build(struct rtnl_qdisc *qdisc, int type, int flags) +{ + struct rtnl_qdisc_ops *qops; + struct nl_msg *msg; + int err; + + msg = tca_build_msg((struct rtnl_tca *) qdisc, type, flags); + if (!msg) + goto errout; + + qops = rtnl_qdisc_lookup_ops(qdisc); + if (qops && qops->qo_get_opts) { + struct nl_msg *opts; + + opts = qops->qo_get_opts(qdisc); + if (opts) { + err = nla_put_nested(msg, TCA_OPTIONS, opts); + nlmsg_free(opts); + if (err < 0) + goto errout; + } + } + + return msg; +errout: + nlmsg_free(msg); + + return NULL; +} + +/** + * Build a netlink message to add a new qdisc + * @arg qdisc qdisc to add + * @arg flags additional netlink message flags + * + * Builds a new netlink message requesting an addition of a qdisc. + * The netlink message header isn't fully equipped with all relevant + * fields and must be sent out via nl_send_auto_complete() or + * supplemented as needed. + * + * Common message flags used: + * - NLM_F_REPLACE - replace a potential existing qdisc + * + * @return New netlink message + */ +struct nl_msg *rtnl_qdisc_build_add_request(struct rtnl_qdisc *qdisc, + int flags) +{ + struct nl_msg *msg; + + msg = qdisc_build(qdisc, RTM_NEWQDISC, NLM_F_CREATE | flags); + if (!msg) + nl_errno(ENOMEM); + + return msg; +} + +/** + * Add a new qdisc + * @arg handle netlink handle + * @arg qdisc qdisc to delete + * @arg flags additional netlink message flags + * + * Builds a netlink message by calling rtnl_qdisc_build_add_request(), + * sends the request to the kernel and waits for the ACK to be + * received and thus blocks until the request has been processed. + * + * Common message flags used: + * - NLM_F_REPLACE - replace a potential existing qdisc + * + * @return 0 on success or a negative error code + */ +int rtnl_qdisc_add(struct nl_handle *handle, struct rtnl_qdisc *qdisc, + int flags) +{ + struct nl_msg *msg; + int err; + + msg = rtnl_qdisc_build_add_request(qdisc, flags); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name QDisc Modification + * @{ + */ + +/** + * Build a netlink message to change attributes of a existing qdisc + * @arg qdisc qdisc to change + * @arg new new qdisc attributes + * + * Builds a new netlink message requesting an change of qdisc + * attributes. The netlink message header isn't fully equipped + * with all relevant fields and must be sent out via + * nl_send_auto_complete() or supplemented as needed. + * + * @return New netlink message + */ +struct nl_msg *rtnl_qdisc_build_change_request(struct rtnl_qdisc *qdisc, + struct rtnl_qdisc *new) +{ + return qdisc_build(qdisc, RTM_NEWQDISC, NLM_F_REPLACE); +} + +/** + * Change attributes of a qdisc + * @arg handle netlink handle + * @arg qdisc qdisc to change + * @arg new new qdisc attributes + * + * Builds a netlink message by calling rtnl_qdisc_build_change_request(), + * sends the request to the kernel and waits for the ACK to be + * received and thus blocks until the request has been processed. + * + * @return 0 on success or a negative error code + */ +int rtnl_qdisc_change(struct nl_handle *handle, struct rtnl_qdisc *qdisc, + struct rtnl_qdisc *new) +{ + struct nl_msg *msg; + int err; + + msg = rtnl_qdisc_build_change_request(qdisc, new); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name QDisc Deletion + * @{ + */ + +/** + * Build a netlink request message to delete a qdisc + * @arg qdisc qdisc to delete + * + * Builds a new netlink message requesting a deletion of a qdisc. + * The netlink message header isn't fully equipped with all relevant + * fields and must thus be sent out via nl_send_auto_complete() + * or supplemented as needed. + * + * @return New netlink message + */ +struct nl_msg *rtnl_qdisc_build_delete_request(struct rtnl_qdisc *qdisc) +{ + struct nl_msg *msg; + struct tcmsg tchdr; + int required = TCA_ATTR_IFINDEX | TCA_ATTR_PARENT; + + if ((qdisc->ce_mask & required) != required) + BUG(); + + msg = nlmsg_alloc_simple(RTM_DELQDISC, 0); + if (!msg) + return NULL; + + tchdr.tcm_family = AF_UNSPEC, + tchdr.tcm_handle = qdisc->q_handle, + tchdr.tcm_parent = qdisc->q_parent, + tchdr.tcm_ifindex = qdisc->q_ifindex, + nlmsg_append(msg, &tchdr, sizeof(tchdr), NLMSG_ALIGNTO); + + return msg; +} + +/** + * Delete a qdisc + * @arg handle netlink handle + * @arg qdisc qdisc to delete + * + * Builds a netlink message by calling rtnl_qdisc_build_delete_request(), + * sends the request to the kernel and waits for the ACK to be + * received and thus blocks until the request has been processed. + * + * @return 0 on success or a negative error code + */ +int rtnl_qdisc_delete(struct nl_handle *handle, struct rtnl_qdisc *qdisc) +{ + struct nl_msg *msg; + int err; + + msg = rtnl_qdisc_build_delete_request(qdisc); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name Qdisc Cache Management + * @{ + */ + +/** + * Build a qdisc cache including all qdiscs currently configured in + * the kernel + * @arg handle netlink handle + * + * Allocates a new cache, initializes it properly and updates it to + * include all qdiscs currently configured in the kernel. + * + * @note The caller is responsible for destroying and freeing the + * cache after using it. + * @return The cache or NULL if an error has occured. + */ +struct nl_cache * rtnl_qdisc_alloc_cache(struct nl_handle *handle) +{ + struct nl_cache * cache; + + cache = nl_cache_alloc(&rtnl_qdisc_ops); + if (cache == NULL) + return NULL; + + if (handle && nl_cache_refill(handle, cache) < 0) { + nl_cache_free(cache); + return NULL; + } + + return cache; +} + +/** + * Look up qdisc by its parent in the provided cache + * @arg cache qdisc cache + * @arg ifindex interface the qdisc is attached to + * @arg parent parent handle + * @return pointer to qdisc inside the cache or NULL if no match was found. + */ +struct rtnl_qdisc * rtnl_qdisc_get_by_parent(struct nl_cache *cache, + int ifindex, uint32_t parent) +{ + struct rtnl_qdisc *q; + + if (cache->c_ops != &rtnl_qdisc_ops) + return NULL; + + nl_list_for_each_entry(q, &cache->c_items, ce_list) { + if (q->q_parent == parent && q->q_ifindex == ifindex) { + nl_object_get((struct nl_object *) q); + return q; + } + } + + return NULL; +} + +/** + * Look up qdisc by its handle in the provided cache + * @arg cache qdisc cache + * @arg ifindex interface the qdisc is attached to + * @arg handle qdisc handle + * @return pointer to qdisc inside the cache or NULL if no match was found. + */ +struct rtnl_qdisc * rtnl_qdisc_get(struct nl_cache *cache, + int ifindex, uint32_t handle) +{ + struct rtnl_qdisc *q; + + if (cache->c_ops != &rtnl_qdisc_ops) + return NULL; + + nl_list_for_each_entry(q, &cache->c_items, ce_list) { + if (q->q_handle == handle && q->q_ifindex == ifindex) { + nl_object_get((struct nl_object *) q); + return q; + } + } + + return NULL; +} + +/** @} */ + +static struct nl_cache_ops rtnl_qdisc_ops = { + .co_name = "route/qdisc", + .co_hdrsize = sizeof(struct tcmsg), + .co_msgtypes = { + { RTM_NEWQDISC, NL_ACT_NEW, "new" }, + { RTM_DELQDISC, NL_ACT_DEL, "del" }, + { RTM_GETQDISC, NL_ACT_GET, "get" }, + END_OF_MSGTYPES_LIST, + }, + .co_protocol = NETLINK_ROUTE, + .co_request_update = qdisc_request_update, + .co_msg_parser = qdisc_msg_parser, + .co_obj_ops = &qdisc_obj_ops, +}; + +static void __init qdisc_init(void) +{ + nl_cache_mngt_register(&rtnl_qdisc_ops); +} + +static void __exit qdisc_exit(void) +{ + nl_cache_mngt_unregister(&rtnl_qdisc_ops); +} + +/** @} */ diff --git a/lib/route/qdisc_api.c b/lib/route/qdisc_api.c new file mode 100644 index 0000000..ef4d07a --- /dev/null +++ b/lib/route/qdisc_api.c @@ -0,0 +1,98 @@ +/* + * lib/route/qdisc_api.c Queueing Discipline Module API + * + * 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> + */ + +/** + * @ingroup qdisc + * @defgroup qdisc_api Queueing Discipline Modules + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/link.h> +#include <netlink/route/tc.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/class.h> +#include <netlink/route/classifier.h> +#include <netlink/route/qdisc-modules.h> + +static struct rtnl_qdisc_ops *qdisc_ops_list; + +/** + * @name Module API + * @{ + */ + +/** + * Register a qdisc module + * @arg qops qdisc module operations + */ +int rtnl_qdisc_register(struct rtnl_qdisc_ops *qops) +{ + struct rtnl_qdisc_ops *o, **op; + + if (!qops->qo_kind[0]) + BUG(); + + for (op = &qdisc_ops_list; (o = *op) != NULL; op = &o->qo_next) + if (!strcasecmp(qops->qo_kind, o->qo_kind)) + return nl_errno(EEXIST); + + qops->qo_next = NULL; + *op = qops; + + return 0; +} + +/** + * Unregister a qdisc module + * @arg qops qdisc module operations + */ +int rtnl_qdisc_unregister(struct rtnl_qdisc_ops *qops) +{ + struct rtnl_qdisc_ops *o, **op; + + for (op = &qdisc_ops_list; (o = *op) != NULL; op = &o->qo_next) + if (!strcasecmp(qops->qo_kind, o->qo_kind)) + break; + + if (!o) + return nl_errno(ENOENT); + + *op = qops->qo_next; + + return 0; +} + +struct rtnl_qdisc_ops *__rtnl_qdisc_lookup_ops(const char *kind) +{ + struct rtnl_qdisc_ops *qops; + + for (qops = qdisc_ops_list; qops; qops = qops->qo_next) + if (!strcmp(kind, qops->qo_kind)) + return qops; + + return NULL; +} + +struct rtnl_qdisc_ops *rtnl_qdisc_lookup_ops(struct rtnl_qdisc *qdisc) +{ + if (!qdisc->q_ops) + qdisc->q_ops = __rtnl_qdisc_lookup_ops(qdisc->q_kind); + + return qdisc->q_ops; +} + +/** @} */ + +/** @} */ diff --git a/lib/route/qdisc_obj.c b/lib/route/qdisc_obj.c new file mode 100644 index 0000000..bbbb954 --- /dev/null +++ b/lib/route/qdisc_obj.c @@ -0,0 +1,273 @@ +/* + * lib/route/qdisc_obj.c Queueing Discipline Object + * + * 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> + */ + +/** + * @ingroup qdisc + * @defgroup qdisc_obj Queueing Discipline Object + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/link.h> +#include <netlink/route/tc.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/class.h> +#include <netlink/route/classifier.h> +#include <netlink/route/qdisc-modules.h> + +static void qdisc_free_data(struct nl_object *obj) +{ + struct rtnl_qdisc *qdisc = (struct rtnl_qdisc *) obj; + struct rtnl_qdisc_ops *qops; + + tca_free_data((struct rtnl_tca *) qdisc); + + qops = rtnl_qdisc_lookup_ops(qdisc); + if (qops && qops->qo_free_data) + qops->qo_free_data(qdisc); +} + +static int qdisc_clone(struct nl_object *_dst, struct nl_object *_src) +{ + struct rtnl_qdisc *dst = (struct rtnl_qdisc *) _dst; + struct rtnl_qdisc *src = (struct rtnl_qdisc *) _src; + struct rtnl_qdisc_ops *qops; + int err; + + err = tca_clone((struct rtnl_tca *) dst, (struct rtnl_tca *) src); + if (err < 0) + goto errout; + + qops = rtnl_qdisc_lookup_ops(src); + if (qops && qops->qo_clone) + err = qops->qo_clone(dst, src); +errout: + return err; +} + +static int qdisc_dump_brief(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_qdisc *qdisc = (struct rtnl_qdisc *) obj; + struct rtnl_qdisc_ops *qops; + + int line = tca_dump_brief((struct rtnl_tca *) qdisc, "qdisc", p, 0); + + qops = rtnl_qdisc_lookup_ops(qdisc); + if (qops && qops->qo_dump[NL_DUMP_BRIEF]) + line = qops->qo_dump[NL_DUMP_BRIEF](qdisc, p, line); + + dp_dump(p, "\n"); + + return line; +} + +static int qdisc_dump_full(struct nl_object *arg, struct nl_dump_params *p) +{ + struct rtnl_qdisc *qdisc = (struct rtnl_qdisc *) arg; + struct rtnl_qdisc_ops *qops; + + int line = qdisc_dump_brief(arg, p); + + line = tca_dump_full((struct rtnl_tca *) qdisc, p, line); + dp_dump(p, "refcnt %u ", qdisc->q_info); + + qops = rtnl_qdisc_lookup_ops(qdisc); + if (qops && qops->qo_dump[NL_DUMP_FULL]) + line = qops->qo_dump[NL_DUMP_FULL](qdisc, p, line); + + dp_dump(p, "\n"); + return line; +} + +static int qdisc_dump_stats(struct nl_object *arg, struct nl_dump_params *p) +{ + struct rtnl_qdisc *qdisc = (struct rtnl_qdisc *) arg; + struct rtnl_qdisc_ops *qops; + + int line = qdisc_dump_full(arg, p); + line = tca_dump_stats((struct rtnl_tca *) qdisc, p, line ); + dp_dump(p, "\n"); + + qops = rtnl_qdisc_lookup_ops(qdisc); + if (qops && qops->qo_dump[NL_DUMP_STATS]) + line = qops->qo_dump[NL_DUMP_STATS](qdisc, p, line); + + return line; +} + +/** + * @name Allocation/Freeing + * @{ + */ + +struct rtnl_qdisc *rtnl_qdisc_alloc(void) +{ + return (struct rtnl_qdisc *) nl_object_alloc(&qdisc_obj_ops); +} + +void rtnl_qdisc_put(struct rtnl_qdisc *qdisc) +{ + nl_object_put((struct nl_object *) qdisc); +} + +/** @} */ + +/** + * @name Iterators + * @{ + */ + +/** + * Call a callback for each child class of a qdisc + * @arg qdisc the parent qdisc + * @arg cache a class cache including all classes of the interface + * the specified qdisc is attached to + * @arg cb callback function + * @arg arg argument to be passed to callback function + */ +void rtnl_qdisc_foreach_child(struct rtnl_qdisc *qdisc, struct nl_cache *cache, + void (*cb)(struct nl_object *, void *), void *arg) +{ + struct rtnl_class *filter; + + filter = rtnl_class_alloc(); + if (!filter) + return; + + rtnl_class_set_parent(filter, qdisc->q_handle); + rtnl_class_set_ifindex(filter, qdisc->q_ifindex); + rtnl_class_set_kind(filter, qdisc->q_kind); + + nl_cache_foreach_filter(cache, (struct nl_object *) filter, cb, arg); + + rtnl_class_put(filter); +} + +/** + * Call a callback for each filter attached to the qdisc + * @arg qdisc the parent qdisc + * @arg cache a filter cache including at least all the filters + * attached to the specified qdisc + * @arg cb callback function + * @arg arg argument to be passed to callback function + */ +void rtnl_qdisc_foreach_cls(struct rtnl_qdisc *qdisc, struct nl_cache *cache, + void (*cb)(struct nl_object *, void *), void *arg) +{ + struct rtnl_cls *filter; + + filter = rtnl_cls_alloc(); + if (!filter) + return; + + rtnl_cls_set_ifindex(filter, qdisc->q_ifindex); + rtnl_cls_set_parent(filter, qdisc->q_parent); + + nl_cache_foreach_filter(cache, (struct nl_object *) filter, cb, arg); + rtnl_cls_put(filter); +} + +/** @} */ + +/** + * @name Attributes + * @{ + */ + +void rtnl_qdisc_set_ifindex(struct rtnl_qdisc *qdisc, int ifindex) +{ + tca_set_ifindex((struct rtnl_tca *) qdisc, ifindex); +} + +int rtnl_qdisc_get_ifindex(struct rtnl_qdisc *qdisc) +{ + return tca_get_ifindex((struct rtnl_tca *) qdisc); +} + +void rtnl_qdisc_set_handle(struct rtnl_qdisc *qdisc, uint32_t handle) +{ + tca_set_handle((struct rtnl_tca *) qdisc, handle); +} + +uint32_t rtnl_qdisc_get_handle(struct rtnl_qdisc *qdisc) +{ + return tca_get_handle((struct rtnl_tca *) qdisc); +} + +void rtnl_qdisc_set_parent(struct rtnl_qdisc *qdisc, uint32_t parent) +{ + tca_set_parent((struct rtnl_tca *) qdisc, parent); +} + +uint32_t rtnl_qdisc_get_parent(struct rtnl_qdisc *qdisc) +{ + return tca_get_parent((struct rtnl_tca *) qdisc); +} + +void rtnl_qdisc_set_kind(struct rtnl_qdisc *qdisc, const char *name) +{ + tca_set_kind((struct rtnl_tca *) qdisc, name); + qdisc->q_ops = __rtnl_qdisc_lookup_ops(name); +} + +char *rtnl_qdisc_get_kind(struct rtnl_qdisc *qdisc) +{ + return tca_get_kind((struct rtnl_tca *) qdisc); +} + +uint64_t rtnl_qdisc_get_stat(struct rtnl_qdisc *qdisc, + enum rtnl_tc_stats_id id) +{ + return tca_get_stat((struct rtnl_tca *) qdisc, id); +} + +/** @} */ + +/** + * @name Qdisc Specific Options + * @{ + */ + +/** + * Return qdisc specific options for use in TCA_OPTIONS + * @arg qdisc qdisc carrying the optiosn + * + * @return new headerless netlink message carrying the options as payload + */ +struct nl_msg *rtnl_qdisc_get_opts(struct rtnl_qdisc *qdisc) +{ + struct rtnl_qdisc_ops *ops; + + ops = rtnl_qdisc_lookup_ops(qdisc); + if (ops && ops->qo_get_opts) + return ops->qo_get_opts(qdisc); + + return NULL; +} + +/** @} */ + +struct nl_object_ops qdisc_obj_ops = { + .oo_name = "route/qdisc", + .oo_size = sizeof(struct rtnl_qdisc), + .oo_free_data = qdisc_free_data, + .oo_clone = qdisc_clone, + .oo_dump[NL_DUMP_BRIEF] = qdisc_dump_brief, + .oo_dump[NL_DUMP_FULL] = qdisc_dump_full, + .oo_dump[NL_DUMP_STATS] = qdisc_dump_stats, + .oo_compare = tca_compare, + .oo_id_attrs = (TCA_ATTR_IFINDEX | TCA_ATTR_HANDLE), +}; + +/** @} */ diff --git a/lib/route/route.c b/lib/route/route.c new file mode 100644 index 0000000..4382e19 --- /dev/null +++ b/lib/route/route.c @@ -0,0 +1,444 @@ +/* + * lib/route/route.c Routes + * + * 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> + */ + +/** + * @ingroup rtnl + * @defgroup route Routing + * @brief + * @{ + */ + +#include <netlink-local.h> +#include <netlink/netlink.h> +#include <netlink/cache.h> +#include <netlink/utils.h> +#include <netlink/data.h> +#include <netlink/route/rtnl.h> +#include <netlink/route/route.h> +#include <netlink/route/link.h> + +static struct nl_cache_ops rtnl_route_ops; + +static struct nla_policy route_policy[RTA_MAX+1] = { + [RTA_IIF] = { .type = NLA_STRING, + .maxlen = IFNAMSIZ, }, + [RTA_OIF] = { .type = NLA_U32 }, + [RTA_PRIORITY] = { .type = NLA_U32 }, + [RTA_FLOW] = { .type = NLA_U32 }, + [RTA_MP_ALGO] = { .type = NLA_U32 }, + [RTA_CACHEINFO] = { .minlen = sizeof(struct rta_cacheinfo) }, + [RTA_METRICS] = { .type = NLA_NESTED }, + [RTA_MULTIPATH] = { .type = NLA_NESTED }, +}; + +static void copy_cacheinfo_into_route(struct rta_cacheinfo *ci, + struct rtnl_route *route) +{ + struct rtnl_rtcacheinfo nci = { + .rtci_clntref = ci->rta_clntref, + .rtci_last_use = ci->rta_lastuse, + .rtci_expires = ci->rta_expires, + .rtci_error = ci->rta_error, + .rtci_used = ci->rta_used, + .rtci_id = ci->rta_id, + .rtci_ts = ci->rta_ts, + .rtci_tsage = ci->rta_tsage, + }; + + rtnl_route_set_cacheinfo(route, &nci); +} + +static int route_msg_parser(struct nl_cache_ops *ops, struct sockaddr_nl *who, + struct nlmsghdr *nlh, void *arg) +{ + struct nl_parser_param *pp = arg; + struct rtmsg *rtm; + struct rtnl_route *route; + struct nlattr *tb[RTA_MAX + 1]; + struct nl_addr *src = NULL, *dst = NULL, *addr; + int err; + + route = rtnl_route_alloc(); + if (!route) { + err = nl_errno(ENOMEM); + goto errout; + } + + route->ce_msgtype = nlh->nlmsg_type; + + err = nlmsg_parse(nlh, sizeof(struct rtmsg), tb, RTA_MAX, + route_policy); + if (err < 0) + goto errout; + + rtm = nlmsg_data(nlh); + rtnl_route_set_family(route, rtm->rtm_family); + rtnl_route_set_tos(route, rtm->rtm_tos); + rtnl_route_set_table(route, rtm->rtm_table); + rtnl_route_set_type(route, rtm->rtm_type); + rtnl_route_set_scope(route, rtm->rtm_scope); + rtnl_route_set_protocol(route, rtm->rtm_protocol); + rtnl_route_set_flags(route, rtm->rtm_flags); + + if (tb[RTA_DST]) { + dst = nla_get_addr(tb[RTA_DST], rtm->rtm_family); + if (dst == NULL) + goto errout_errno; + } else if (rtm->rtm_dst_len) + dst = nl_addr_alloc(0); + + if (dst) { + nl_addr_set_prefixlen(dst, rtm->rtm_dst_len); + rtnl_route_set_dst(route, dst); + nl_addr_put(dst); + } + + if (tb[RTA_SRC]) { + src = nla_get_addr(tb[RTA_SRC], rtm->rtm_family); + if (src == NULL) + goto errout_errno; + } else if (rtm->rtm_src_len) + src = nl_addr_alloc(0); + + if (src) { + nl_addr_set_prefixlen(src, rtm->rtm_src_len); + rtnl_route_set_src(route, src); + nl_addr_put(src); + } + + if (tb[RTA_IIF]) + rtnl_route_set_iif(route, nla_get_string(tb[RTA_IIF])); + + if (tb[RTA_OIF]) + rtnl_route_set_oif(route, nla_get_u32(tb[RTA_OIF])); + + if (tb[RTA_GATEWAY]) { + addr = nla_get_addr(tb[RTA_GATEWAY], route->rt_family); + if (addr == NULL) + goto errout_errno; + rtnl_route_set_gateway(route, addr); + nl_addr_put(addr); + } + + if (tb[RTA_PRIORITY]) + rtnl_route_set_prio(route, nla_get_u32(tb[RTA_PRIORITY])); + + if (tb[RTA_PREFSRC]) { + addr = nla_get_addr(tb[RTA_PREFSRC], route->rt_family); + if (addr == NULL) + goto errout_errno; + rtnl_route_set_pref_src(route, addr); + nl_addr_put(addr); + } + + if (tb[RTA_METRICS]) { + struct nlattr *mtb[RTAX_MAX + 1]; + int i; + + err = nla_parse_nested(mtb, RTAX_MAX, tb[RTA_METRICS], NULL); + if (err < 0) + goto errout; + + for (i = 1; i <= RTAX_MAX; i++) { + if (mtb[i] && nla_len(mtb[i]) >= sizeof(uint32_t)) { + uint32_t m = nla_get_u32(mtb[i]); + if (rtnl_route_set_metric(route, i, m) < 0) + goto errout_errno; + } + } + } + + if (tb[RTA_MULTIPATH]) { + struct rtnl_nexthop *nh; + struct rtnexthop *rtnh = nla_data(tb[RTA_MULTIPATH]); + size_t tlen = nla_len(tb[RTA_MULTIPATH]); + + while (tlen >= sizeof(*rtnh) && tlen >= rtnh->rtnh_len) { + nh = rtnl_route_nh_alloc(); + if (!nh) + goto errout; + + rtnl_route_nh_set_weight(nh, rtnh->rtnh_hops); + rtnl_route_nh_set_ifindex(nh, rtnh->rtnh_ifindex); + rtnl_route_nh_set_flags(nh, rtnh->rtnh_flags); + + if (rtnh->rtnh_len > sizeof(*rtnh)) { + struct nlattr *ntb[RTA_MAX + 1]; + nla_parse(ntb, RTA_MAX, (struct nlattr *) + RTNH_DATA(rtnh), + rtnh->rtnh_len - sizeof(*rtnh), + route_policy); + + if (ntb[RTA_GATEWAY]) { + nh->rtnh_gateway = nla_get_addr( + ntb[RTA_GATEWAY], + route->rt_family); + nh->rtnh_mask = NEXTHOP_HAS_GATEWAY; + } + } + + rtnl_route_add_nexthop(route, nh); + tlen -= RTNH_ALIGN(rtnh->rtnh_len); + rtnh = RTNH_NEXT(rtnh); + } + } + + if (tb[RTA_FLOW]) + rtnl_route_set_realms(route, nla_get_u32(tb[RTA_FLOW])); + + if (tb[RTA_CACHEINFO]) + copy_cacheinfo_into_route(nla_data(tb[RTA_CACHEINFO]), route); + + if (tb[RTA_MP_ALGO]) + rtnl_route_set_mp_algo(route, nla_get_u32(tb[RTA_MP_ALGO])); + + err = pp->pp_cb((struct nl_object *) route, pp); + if (err < 0) + goto errout; + + return P_ACCEPT; + +errout_errno: + err = nl_get_errno(); +errout: + rtnl_route_put(route); + return err; + +} + +static int route_request_update(struct nl_cache *c, struct nl_handle *h) +{ + return nl_rtgen_request(h, RTM_GETROUTE, AF_UNSPEC, NLM_F_DUMP); +} + +/** + * @name Cache Management + * @{ + */ + +/** + * Build a route cache holding all routes currently configured in the kernel + * @arg handle netlink handle + * + * Allocates a new cache, initializes it properly and updates it to + * contain all routes currently configured in the kernel. + * + * @note The caller is responsible for destroying and freeing the + * cache after using it. + * @return The cache or NULL if an error has occured. + */ +struct nl_cache *rtnl_route_alloc_cache(struct nl_handle *handle) +{ + struct nl_cache *cache; + + cache = nl_cache_alloc(&rtnl_route_ops); + if (!cache) + return NULL; + + if (handle && nl_cache_refill(handle, cache) < 0) { + free(cache); + return NULL; + } + + return cache; +} + +/** @} */ + +/** + * @name Route Addition + * @{ + */ + +static struct nl_msg *build_route_msg(struct rtnl_route *tmpl, int cmd, + int flags) +{ + struct nl_msg *msg; + struct nl_addr *addr; + int scope, i, oif, nmetrics = 0; + struct nlattr *metrics; + struct rtmsg rtmsg = { + .rtm_family = rtnl_route_get_family(tmpl), + .rtm_dst_len = rtnl_route_get_dst_len(tmpl), + .rtm_src_len = rtnl_route_get_src_len(tmpl), + .rtm_tos = rtnl_route_get_tos(tmpl), + .rtm_table = rtnl_route_get_table(tmpl), + .rtm_type = rtnl_route_get_type(tmpl), + .rtm_protocol = rtnl_route_get_protocol(tmpl), + .rtm_flags = rtnl_route_get_flags(tmpl), + }; + + if (rtmsg.rtm_family == AF_UNSPEC) { + nl_error(EINVAL, "Cannot build route message, address " \ + "family is unknown."); + return NULL; + } + + scope = rtnl_route_get_scope(tmpl); + if (scope == RT_SCOPE_NOWHERE) { + if (rtmsg.rtm_type == RTN_LOCAL) + scope = RT_SCOPE_HOST; + else { + /* XXX Change to UNIVERSE if gw || nexthops */ + scope = RT_SCOPE_LINK; + } + } + + rtmsg.rtm_scope = scope; + + msg = nlmsg_alloc_simple(cmd, flags); + if (msg == NULL) + return NULL; + + if (nlmsg_append(msg, &rtmsg, sizeof(rtmsg), NLMSG_ALIGNTO) < 0) + goto nla_put_failure; + + addr = rtnl_route_get_dst(tmpl); + if (addr) + NLA_PUT_ADDR(msg, RTA_DST, addr); + + addr = rtnl_route_get_src(tmpl); + if (addr) + NLA_PUT_ADDR(msg, RTA_SRC, addr); + + addr = rtnl_route_get_gateway(tmpl); + if (addr) + NLA_PUT_ADDR(msg, RTA_GATEWAY, addr); + + addr = rtnl_route_get_pref_src(tmpl); + if (addr) + NLA_PUT_ADDR(msg, RTA_PREFSRC, addr); + + NLA_PUT_U32(msg, RTA_PRIORITY, rtnl_route_get_prio(tmpl)); + + oif = rtnl_route_get_oif(tmpl); + if (oif != RTNL_LINK_NOT_FOUND) + NLA_PUT_U32(msg, RTA_OIF, oif); + + for (i = 1; i <= RTAX_MAX; i++) + if (rtnl_route_get_metric(tmpl, i) != UINT_MAX) + nmetrics++; + + if (nmetrics > 0) { + unsigned int val; + + metrics = nla_nest_start(msg, RTA_METRICS); + if (metrics == NULL) + goto nla_put_failure; + + for (i = 1; i <= RTAX_MAX; i++) { + val = rtnl_route_get_metric(tmpl, i); + if (val != UINT_MAX) + NLA_PUT_U32(msg, i, val); + } + + nla_nest_end(msg, metrics); + } + +#if 0 + RTA_IIF, + RTA_MULTIPATH, + RTA_PROTOINFO, + RTA_FLOW, + RTA_CACHEINFO, + RTA_SESSION, + RTA_MP_ALGO, +#endif + + return msg; + +nla_put_failure: + nlmsg_free(msg); + return NULL; +} + +struct nl_msg *rtnl_route_build_add_request(struct rtnl_route *tmpl, int flags) +{ + return build_route_msg(tmpl, RTM_NEWROUTE, NLM_F_CREATE | flags); +} + +int rtnl_route_add(struct nl_handle *handle, struct rtnl_route *route, + int flags) +{ + struct nl_msg *msg; + int err; + + msg = rtnl_route_build_add_request(route, flags); + if (!msg) + return nl_get_errno(); + + err = nl_send_auto_complete(handle, msg); + nlmsg_free(msg); + if (err < 0) + return err; + + return nl_wait_for_ack(handle); +} + +struct nl_msg *rtnl_route_build_del_request(struct rtnl_route *tmpl, int flags) +{ + return build_route_msg(tmpl, RTM_DELROUTE, flags); +} + +int rtnl_route_del(struct nl_handle *handle, struct rtnl_route *route, + int flags) +{ + struct nl_msg *msg; + int err; + + msg = rtnl_route_build_del_request(route, flags); + if (!msg) + return nl_get_errno(); + + err = nl_send_auto_complete(handle, msg); + nlmsg_free(msg); + if (err < 0) + return err; + + return nl_wait_for_ack(handle); +} + +/** @} */ + +static struct nl_af_group route_groups[] = { + { AF_INET, RTNLGRP_IPV4_ROUTE }, + { AF_INET6, RTNLGRP_IPV6_ROUTE }, + { AF_DECnet, RTNLGRP_DECnet_ROUTE }, + { END_OF_GROUP_LIST }, +}; + +static struct nl_cache_ops rtnl_route_ops = { + .co_name = "route/route", + .co_hdrsize = sizeof(struct rtmsg), + .co_msgtypes = { + { RTM_NEWROUTE, NL_ACT_NEW, "new" }, + { RTM_DELROUTE, NL_ACT_DEL, "del" }, + { RTM_GETROUTE, NL_ACT_GET, "get" }, + END_OF_MSGTYPES_LIST, + }, + .co_protocol = NETLINK_ROUTE, + .co_groups = route_groups, + .co_request_update = route_request_update, + .co_msg_parser = route_msg_parser, + .co_obj_ops = &route_obj_ops, +}; + +static void __init route_init(void) +{ + nl_cache_mngt_register(&rtnl_route_ops); +} + +static void __exit route_exit(void) +{ + nl_cache_mngt_unregister(&rtnl_route_ops); +} + +/** @} */ diff --git a/lib/route/route_obj.c b/lib/route/route_obj.c new file mode 100644 index 0000000..7d954fc --- /dev/null +++ b/lib/route/route_obj.c @@ -0,0 +1,908 @@ +/* + * lib/route/route_obj.c Route Object + * + * 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> + */ + +/** + * @ingroup route + * @defgroup route_obj Route Object + * + * @par Attributes + * @code + * Name Default + * ------------------------------------------------------------- + * routing table RT_TABLE_MAIN + * scope RT_SCOPE_NOWHERE + * tos 0 + * realms 0 + * protocol RTPROT_STATIC + * prio 0 + * family AF_UNSPEC + * type RTN_UNICAST + * oif RTNL_LINK_NOT_FOUND + * iif NULL + * mpalgo IP_MP_ALG_NONE + * @endcode + * + * @{ + */ + +#include <netlink-local.h> +#include <netlink/netlink.h> +#include <netlink/cache.h> +#include <netlink/utils.h> +#include <netlink/data.h> +#include <netlink/route/rtnl.h> +#include <netlink/route/route.h> +#include <netlink/route/link.h> + +/** @cond SKIP */ +#define ROUTE_ATTR_FAMILY 0x000001 +#define ROUTE_ATTR_TOS 0x000002 +#define ROUTE_ATTR_TABLE 0x000004 +#define ROUTE_ATTR_PROTOCOL 0x000008 +#define ROUTE_ATTR_SCOPE 0x000010 +#define ROUTE_ATTR_TYPE 0x000020 +#define ROUTE_ATTR_FLAGS 0x000040 +#define ROUTE_ATTR_DST 0x000080 +#define ROUTE_ATTR_SRC 0x000100 +#define ROUTE_ATTR_IIF 0x000200 +#define ROUTE_ATTR_OIF 0x000400 +#define ROUTE_ATTR_GATEWAY 0x000800 +#define ROUTE_ATTR_PRIO 0x001000 +#define ROUTE_ATTR_PREF_SRC 0x002000 +#define ROUTE_ATTR_METRICS 0x004000 +#define ROUTE_ATTR_MULTIPATH 0x008000 +#define ROUTE_ATTR_REALMS 0x010000 +#define ROUTE_ATTR_CACHEINFO 0x020000 +#define ROUTE_ATTR_MP_ALGO 0x040000 +/** @endcond */ + +static int route_dump_brief(struct nl_object *a, struct nl_dump_params *p); + +static void route_constructor(struct nl_object *c) +{ + struct rtnl_route *r = (struct rtnl_route *) c; + + nl_init_list_head(&r->rt_nexthops); +} + +static void route_free_data(struct nl_object *c) +{ + struct rtnl_route *r = (struct rtnl_route *) c; + struct rtnl_nexthop *nh, *tmp; + + if (r == NULL) + return; + + nl_addr_put(r->rt_dst); + nl_addr_put(r->rt_src); + nl_addr_put(r->rt_gateway); + nl_addr_put(r->rt_pref_src); + + nl_list_for_each_entry_safe(nh, tmp, &r->rt_nexthops, rtnh_list) { + rtnl_route_remove_nexthop(nh); + rtnl_route_nh_free(nh); + } +} + +static int route_clone(struct nl_object *_dst, struct nl_object *_src) +{ + struct rtnl_route *dst = (struct rtnl_route *) _dst; + struct rtnl_route *src = (struct rtnl_route *) _src; + struct rtnl_nexthop *nh, *new; + + if (src->rt_dst) + if (!(dst->rt_dst = nl_addr_clone(src->rt_dst))) + goto errout; + + if (src->rt_src) + if (!(dst->rt_src = nl_addr_clone(src->rt_src))) + goto errout; + + if (src->rt_gateway) + if (!(dst->rt_gateway = nl_addr_clone(src->rt_gateway))) + goto errout; + + if (src->rt_pref_src) + if (!(dst->rt_pref_src = nl_addr_clone(src->rt_pref_src))) + goto errout; + + nl_init_list_head(&dst->rt_nexthops); + nl_list_for_each_entry(nh, &src->rt_nexthops, rtnh_list) { + new = rtnl_route_nh_clone(nh); + if (!new) + goto errout; + + rtnl_route_add_nexthop(dst, new); + } + + return 0; +errout: + return nl_get_errno(); +} + +static int route_dump_brief(struct nl_object *a, struct nl_dump_params *p) +{ + struct rtnl_route *r = (struct rtnl_route *) a; + struct nl_cache *link_cache; + char buf[64]; + + link_cache = nl_cache_mngt_require("route/link"); + + if (r->ce_mask & ROUTE_ATTR_DST) + dp_dump(p, "%s ", nl_addr2str(r->rt_dst, buf, sizeof(buf))); + else + dp_dump(p, "default "); + + if (r->ce_mask & ROUTE_ATTR_OIF) { + if (link_cache) + dp_dump(p, "dev %s ", + rtnl_link_i2name(link_cache, r->rt_oif, + buf, sizeof(buf))); + else + dp_dump(p, "dev %d ", r->rt_oif); + } + + if (r->ce_mask & ROUTE_ATTR_GATEWAY) + dp_dump(p, "via %s ", nl_addr2str(r->rt_gateway, buf, + sizeof(buf))); + else if (r->ce_mask & ROUTE_ATTR_MULTIPATH) + dp_dump(p, "via nexthops "); + + if (r->ce_mask & ROUTE_ATTR_SCOPE) + dp_dump(p, "scope %s ", + rtnl_scope2str(r->rt_scope, buf, sizeof(buf))); + + if (r->ce_mask & ROUTE_ATTR_FLAGS && r->rt_flags) { + int flags = r->rt_flags; + + dp_dump(p, "<"); + +#define PRINT_FLAG(f) if (flags & RTNH_F_##f) { \ + flags &= ~RTNH_F_##f; dp_dump(p, #f "%s", flags ? "," : ""); } + PRINT_FLAG(DEAD); + PRINT_FLAG(ONLINK); + PRINT_FLAG(PERVASIVE); +#undef PRINT_FLAG + +#define PRINT_FLAG(f) if (flags & RTM_F_##f) { \ + flags &= ~RTM_F_##f; dp_dump(p, #f "%s", flags ? "," : ""); } + PRINT_FLAG(NOTIFY); + PRINT_FLAG(CLONED); + PRINT_FLAG(EQUALIZE); + PRINT_FLAG(PREFIX); +#undef PRINT_FLAG + + dp_dump(p, ">"); + } + + dp_dump(p, "\n"); + + return 1; +} + +static int route_dump_full(struct nl_object *a, struct nl_dump_params *p) +{ + struct rtnl_route *r = (struct rtnl_route *) a; + struct nl_cache *link_cache; + char buf[128]; + int i, line; + + link_cache = nl_cache_mngt_require("route/link"); + line = route_dump_brief(a, p); + + if (r->ce_mask & ROUTE_ATTR_MULTIPATH) { + struct rtnl_nexthop *nh; + + nl_list_for_each_entry(nh, &r->rt_nexthops, rtnh_list) { + dp_dump_line(p, line++, " via "); + + if (nh->rtnh_mask & NEXTHOP_HAS_GATEWAY) + dp_dump(p, "%s ", + nl_addr2str(nh->rtnh_gateway, + buf, sizeof(buf))); + if (link_cache) { + dp_dump(p, "dev %s ", + rtnl_link_i2name(link_cache, + nh->rtnh_ifindex, + buf, sizeof(buf))); + } else + dp_dump(p, "dev %d ", nh->rtnh_ifindex); + + dp_dump(p, "weight %u <%s>\n", nh->rtnh_weight, + rtnl_route_nh_flags2str(nh->rtnh_flags, + buf, sizeof(buf))); + } + } + + dp_dump_line(p, line++, " "); + + if (r->ce_mask & ROUTE_ATTR_PREF_SRC) + dp_dump(p, "preferred-src %s ", + nl_addr2str(r->rt_pref_src, buf, sizeof(buf))); + + if (r->ce_mask & ROUTE_ATTR_TABLE) + dp_dump(p, "table %s ", + rtnl_route_table2str(r->rt_table, buf, sizeof(buf))); + + if (r->ce_mask & ROUTE_ATTR_TYPE) + dp_dump(p, "type %s ", + nl_rtntype2str(r->rt_type, buf, sizeof(buf))); + + if (r->ce_mask & ROUTE_ATTR_PRIO) + dp_dump(p, "metric %#x ", r->rt_prio); + + if (r->ce_mask & ROUTE_ATTR_FAMILY) + dp_dump(p, "family %s ", + nl_af2str(r->rt_family, buf, sizeof(buf))); + + if (r->ce_mask & ROUTE_ATTR_PROTOCOL) + dp_dump(p, "protocol %s ", + rtnl_route_proto2str(r->rt_protocol, buf, sizeof(buf))); + + dp_dump(p, "\n"); + + if ((r->ce_mask & (ROUTE_ATTR_IIF | ROUTE_ATTR_SRC | ROUTE_ATTR_TOS | + ROUTE_ATTR_REALMS)) || + ((r->ce_mask & ROUTE_ATTR_CACHEINFO) && + r->rt_cacheinfo.rtci_error)) { + dp_dump_line(p, line++, " "); + + if (r->ce_mask & ROUTE_ATTR_IIF) + dp_dump(p, "iif %s ", r->rt_iif); + + if (r->ce_mask & ROUTE_ATTR_SRC) + dp_dump(p, "src %s ", + nl_addr2str(r->rt_src, buf, sizeof(buf))); + + if (r->ce_mask & ROUTE_ATTR_TOS) + dp_dump(p, "tos %#x ", r->rt_tos); + + if (r->ce_mask & ROUTE_ATTR_REALMS) + dp_dump(p, "realm %04x:%04x ", + RTNL_REALM_FROM(r->rt_realms), + RTNL_REALM_TO(r->rt_realms)); + + if ((r->ce_mask & ROUTE_ATTR_CACHEINFO) && + r->rt_cacheinfo.rtci_error) + dp_dump(p, "error %d (%s) ", r->rt_cacheinfo.rtci_error, + strerror(-r->rt_cacheinfo.rtci_error)); + + dp_dump(p, "\n"); + } + + if (r->ce_mask & ROUTE_ATTR_METRICS) { + dp_dump_line(p, line++, " "); + for (i = 0; i < RTAX_MAX; i++) + if (r->rt_metrics_mask & (1 << i)) + dp_dump(p, "%s %u ", + rtnl_route_metric2str(i+1, + buf, sizeof(buf)), + r->rt_metrics[i]); + dp_dump(p, "\n"); + } + + return line; +} + +static int route_dump_stats(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_route *route = (struct rtnl_route *) obj; + int line; + + line = route_dump_full(obj, p); + + if (route->ce_mask & ROUTE_ATTR_CACHEINFO) { + struct rtnl_rtcacheinfo *ci = &route->rt_cacheinfo; + dp_dump_line(p, line++, " used %u refcnt %u ", + ci->rtci_used, ci->rtci_clntref); + dp_dump_line(p, line++, "last-use %us expires %us\n", + ci->rtci_last_use / nl_get_hz(), + ci->rtci_expires / nl_get_hz()); + } + + return line; +} + +static int route_dump_xml(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_route *route = (struct rtnl_route *) obj; + char buf[128]; + int line = 0; + + dp_dump_line(p, line++, "<route>\n"); + dp_dump_line(p, line++, " <family>%s</family>\n", + nl_af2str(route->rt_family, buf, sizeof(buf))); + + if (route->ce_mask & ROUTE_ATTR_DST) + dp_dump_line(p, line++, " <dst>%s</dst>\n", + nl_addr2str(route->rt_dst, buf, sizeof(buf))); + + if (route->ce_mask & ROUTE_ATTR_SRC) + dp_dump_line(p, line++, " <src>%s</src>\n", + nl_addr2str(route->rt_src, buf, sizeof(buf))); + + if (route->ce_mask & ROUTE_ATTR_GATEWAY) + dp_dump_line(p, line++, " <gateway>%s</gateway>\n", + nl_addr2str(route->rt_gateway, buf, sizeof(buf))); + + if (route->ce_mask & ROUTE_ATTR_PREF_SRC) + dp_dump_line(p, line++, " <prefsrc>%s</prefsrc>\n", + nl_addr2str(route->rt_pref_src, buf, sizeof(buf))); + + if (route->ce_mask & ROUTE_ATTR_IIF) + dp_dump_line(p, line++, " <iif>%s</iif>\n", route->rt_iif); + + if (route->ce_mask & ROUTE_ATTR_REALMS) + dp_dump_line(p, line++, " <realms>%u</realms>\n", + route->rt_realms); + + if (route->ce_mask & ROUTE_ATTR_TOS) + dp_dump_line(p, line++, " <tos>%u</tos>\n", route->rt_tos); + + if (route->ce_mask & ROUTE_ATTR_TABLE) + dp_dump_line(p, line++, " <table>%u</table>\n", + route->rt_table); + + if (route->ce_mask & ROUTE_ATTR_SCOPE) + dp_dump_line(p, line++, " <scope>%s</scope>\n", + rtnl_scope2str(route->rt_scope, buf, sizeof(buf))); + + if (route->ce_mask & ROUTE_ATTR_PRIO) + dp_dump_line(p, line++, " <metric>%u</metric>\n", + route->rt_prio); + + if (route->ce_mask & ROUTE_ATTR_OIF) { + struct nl_cache *link_cache; + + link_cache = nl_cache_mngt_require("route/link"); + if (link_cache) + dp_dump_line(p, line++, " <oif>%s</oif>\n", + rtnl_link_i2name(link_cache, + route->rt_oif, + buf, sizeof(buf))); + else + dp_dump_line(p, line++, " <oif>%u</oif>\n", + route->rt_oif); + } + + if (route->ce_mask & ROUTE_ATTR_TYPE) + dp_dump_line(p, line++, " <type>%s</type>\n", + nl_rtntype2str(route->rt_type, buf, sizeof(buf))); + + dp_dump_line(p, line++, "</route>\n"); + +#if 0 + uint8_t rt_protocol; + uint32_t rt_flags; + uint32_t rt_metrics[RTAX_MAX]; + uint32_t rt_metrics_mask; + struct rtnl_nexthop * rt_nexthops; + struct rtnl_rtcacheinfo rt_cacheinfo; + uint32_t rt_mp_algo; + +#endif + + return line; +} + +static int route_dump_env(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_route *route = (struct rtnl_route *) obj; + char buf[128]; + int line = 0; + + dp_dump_line(p, line++, "ROUTE_FAMILY=%s\n", + nl_af2str(route->rt_family, buf, sizeof(buf))); + + if (route->ce_mask & ROUTE_ATTR_DST) + dp_dump_line(p, line++, "ROUTE_DST=%s\n", + nl_addr2str(route->rt_dst, buf, sizeof(buf))); + + if (route->ce_mask & ROUTE_ATTR_SRC) + dp_dump_line(p, line++, "ROUTE_SRC=%s\n", + nl_addr2str(route->rt_src, buf, sizeof(buf))); + + if (route->ce_mask & ROUTE_ATTR_GATEWAY) + dp_dump_line(p, line++, "ROUTE_GATEWAY=%s\n", + nl_addr2str(route->rt_gateway, buf, sizeof(buf))); + + if (route->ce_mask & ROUTE_ATTR_PREF_SRC) + dp_dump_line(p, line++, "ROUTE_PREFSRC=%s\n", + nl_addr2str(route->rt_pref_src, buf, sizeof(buf))); + + if (route->ce_mask & ROUTE_ATTR_IIF) + dp_dump_line(p, line++, "ROUTE_IIF=%s\n", route->rt_iif); + + if (route->ce_mask & ROUTE_ATTR_REALMS) + dp_dump_line(p, line++, "ROUTE_REALM=%u\n", + route->rt_realms); + + if (route->ce_mask & ROUTE_ATTR_TOS) + dp_dump_line(p, line++, "ROUTE_TOS=%u\n", route->rt_tos); + + if (route->ce_mask & ROUTE_ATTR_TABLE) + dp_dump_line(p, line++, "ROUTE_TABLE=%u\n", + route->rt_table); + + if (route->ce_mask & ROUTE_ATTR_SCOPE) + dp_dump_line(p, line++, "ROUTE_SCOPE=%s\n", + rtnl_scope2str(route->rt_scope, buf, sizeof(buf))); + + if (route->ce_mask & ROUTE_ATTR_PRIO) + dp_dump_line(p, line++, "ROUTE_METRIC=%u\n", + route->rt_prio); + + if (route->ce_mask & ROUTE_ATTR_OIF) { + struct nl_cache *link_cache; + + dp_dump_line(p, line++, "ROUTE_OIF_IFINDEX=%u\n", + route->rt_oif); + + link_cache = nl_cache_mngt_require("route/link"); + if (link_cache) + dp_dump_line(p, line++, "ROUTE_OIF_IFNAME=%s\n", + rtnl_link_i2name(link_cache, + route->rt_oif, + buf, sizeof(buf))); + } + + if (route->ce_mask & ROUTE_ATTR_TYPE) + dp_dump_line(p, line++, "ROUTE_TYPE=%s\n", + nl_rtntype2str(route->rt_type, buf, sizeof(buf))); + + return line; +} + +static int route_compare(struct nl_object *_a, struct nl_object *_b, + uint32_t attrs, int flags) +{ + struct rtnl_route *a = (struct rtnl_route *) _a; + struct rtnl_route *b = (struct rtnl_route *) _b; + int diff = 0; + +#define ROUTE_DIFF(ATTR, EXPR) ATTR_DIFF(attrs, ROUTE_ATTR_##ATTR, a, b, EXPR) + + diff |= ROUTE_DIFF(FAMILY, a->rt_family != b->rt_family); + diff |= ROUTE_DIFF(TOS, a->rt_tos != b->rt_tos); + diff |= ROUTE_DIFF(TABLE, a->rt_table != b->rt_table); + diff |= ROUTE_DIFF(PROTOCOL, a->rt_protocol != b->rt_protocol); + diff |= ROUTE_DIFF(SCOPE, a->rt_scope != b->rt_scope); + diff |= ROUTE_DIFF(TYPE, a->rt_type != b->rt_type); + diff |= ROUTE_DIFF(OIF, a->rt_oif != b->rt_oif); + diff |= ROUTE_DIFF(PRIO, a->rt_prio != b->rt_prio); + diff |= ROUTE_DIFF(REALMS, a->rt_realms != b->rt_realms); + diff |= ROUTE_DIFF(MP_ALGO, a->rt_mp_algo != b->rt_mp_algo); + diff |= ROUTE_DIFF(DST, nl_addr_cmp(a->rt_dst, b->rt_dst)); + diff |= ROUTE_DIFF(SRC, nl_addr_cmp(a->rt_src, b->rt_src)); + diff |= ROUTE_DIFF(IIF, strcmp(a->rt_iif, b->rt_iif)); + diff |= ROUTE_DIFF(PREF_SRC, nl_addr_cmp(a->rt_pref_src, + b->rt_pref_src)); + diff |= ROUTE_DIFF(GATEWAY, nl_addr_cmp(a->rt_gateway, + b->rt_gateway)); + + /* FIXME: Compare metrics, multipath config */ + + if (flags & LOOSE_FLAG_COMPARISON) + diff |= ROUTE_DIFF(FLAGS, + (a->rt_flags ^ b->rt_flags) & b->rt_flag_mask); + else + diff |= ROUTE_DIFF(FLAGS, a->rt_flags != b->rt_flags); + +#undef ROUTE_DIFF + + return diff; +} + +static struct trans_tbl route_attrs[] = { + __ADD(ROUTE_ATTR_FAMILY, family) + __ADD(ROUTE_ATTR_TOS, tos) + __ADD(ROUTE_ATTR_TABLE, table) + __ADD(ROUTE_ATTR_PROTOCOL, protocol) + __ADD(ROUTE_ATTR_SCOPE, scope) + __ADD(ROUTE_ATTR_TYPE, type) + __ADD(ROUTE_ATTR_FLAGS, flags) + __ADD(ROUTE_ATTR_DST, dst) + __ADD(ROUTE_ATTR_SRC, src) + __ADD(ROUTE_ATTR_IIF, iif) + __ADD(ROUTE_ATTR_OIF, oif) + __ADD(ROUTE_ATTR_GATEWAY, gateway) + __ADD(ROUTE_ATTR_PRIO, prio) + __ADD(ROUTE_ATTR_PREF_SRC, pref_src) + __ADD(ROUTE_ATTR_METRICS, metrics) + __ADD(ROUTE_ATTR_MULTIPATH, multipath) + __ADD(ROUTE_ATTR_REALMS, realms) + __ADD(ROUTE_ATTR_CACHEINFO, cacheinfo) + __ADD(ROUTE_ATTR_MP_ALGO, mp_algo) +}; + +static char *route_attrs2str(int attrs, char *buf, size_t len) +{ + return __flags2str(attrs, buf, len, route_attrs, + ARRAY_SIZE(route_attrs)); +} + +/** + * @name Allocation/Freeing + * @{ + */ + +struct rtnl_route *rtnl_route_alloc(void) +{ + return (struct rtnl_route *) nl_object_alloc(&route_obj_ops); +} + +void rtnl_route_get(struct rtnl_route *route) +{ + nl_object_get((struct nl_object *) route); +} + +void rtnl_route_put(struct rtnl_route *route) +{ + nl_object_put((struct nl_object *) route); +} + +/** @} */ + +/** + * @name Attributes + * @{ + */ + +void rtnl_route_set_table(struct rtnl_route *route, int table) +{ + route->rt_table = table; + route->ce_mask |= ROUTE_ATTR_TABLE; +} + +int rtnl_route_get_table(struct rtnl_route *route) +{ + if (route->ce_mask & ROUTE_ATTR_TABLE) + return route->rt_table; + else + return RT_TABLE_MAIN; +} + +void rtnl_route_set_scope(struct rtnl_route *route, int scope) +{ + route->rt_scope = scope; + route->ce_mask |= ROUTE_ATTR_SCOPE; +} + +int rtnl_route_get_scope(struct rtnl_route *route) +{ + if (route->ce_mask & ROUTE_ATTR_SCOPE) + return route->rt_scope; + else + return RT_SCOPE_NOWHERE; +} + +void rtnl_route_set_tos(struct rtnl_route *route, int tos) +{ + route->rt_tos = tos; + route->ce_mask |= ROUTE_ATTR_TOS; +} + +int rtnl_route_get_tos(struct rtnl_route *route) +{ + return route->rt_tos; +} + +void rtnl_route_set_realms(struct rtnl_route *route, realm_t realms) +{ + route->rt_realms = realms; + route->ce_mask |= ROUTE_ATTR_REALMS; +} + +realm_t rtnl_route_get_realms(struct rtnl_route *route) +{ + return route->rt_realms; +} + +void rtnl_route_set_protocol(struct rtnl_route *route, int proto) +{ + route->rt_protocol = proto; + route->ce_mask |= ROUTE_ATTR_PROTOCOL; +} + +int rtnl_route_get_protocol(struct rtnl_route *route) +{ + if (route->ce_mask & ROUTE_ATTR_PROTOCOL) + return route->rt_protocol; + else + return RTPROT_STATIC; +} + +void rtnl_route_set_prio(struct rtnl_route *route, int prio) +{ + route->rt_prio = prio; + route->ce_mask |= ROUTE_ATTR_PRIO; +} + +int rtnl_route_get_prio(struct rtnl_route *route) +{ + return route->rt_prio; +} + +void rtnl_route_set_family(struct rtnl_route *route, int family) +{ + route->rt_family = family; + route->ce_mask |= ROUTE_ATTR_FAMILY; +} + +int rtnl_route_get_family(struct rtnl_route *route) +{ + if (route->ce_mask & ROUTE_ATTR_FAMILY) + return route->rt_family; + else + return AF_UNSPEC; +} + +int rtnl_route_set_dst(struct rtnl_route *route, struct nl_addr *addr) +{ + if (route->ce_mask & ROUTE_ATTR_FAMILY) { + if (addr->a_family != route->rt_family) + return nl_error(EINVAL, "Address family mismatch"); + } else + route->rt_family = addr->a_family; + + if (route->rt_dst) + nl_addr_put(route->rt_dst); + + nl_addr_get(addr); + route->rt_dst = addr; + + route->ce_mask |= (ROUTE_ATTR_DST | ROUTE_ATTR_FAMILY); + + return 0; +} + +struct nl_addr *rtnl_route_get_dst(struct rtnl_route *route) +{ + return route->rt_dst; +} + +int rtnl_route_get_dst_len(struct rtnl_route *route) +{ + if (route->ce_mask & ROUTE_ATTR_DST) + return nl_addr_get_prefixlen(route->rt_dst); + else + return 0; +} + +int rtnl_route_set_src(struct rtnl_route *route, struct nl_addr *addr) +{ + if (route->ce_mask & ROUTE_ATTR_FAMILY) { + if (addr->a_family != route->rt_family) + return nl_error(EINVAL, "Address family mismatch"); + } else + route->rt_family = addr->a_family; + + if (route->rt_src) + nl_addr_put(route->rt_src); + + nl_addr_get(addr); + route->rt_src = addr; + route->ce_mask |= (ROUTE_ATTR_SRC | ROUTE_ATTR_FAMILY); + + return 0; +} + +struct nl_addr *rtnl_route_get_src(struct rtnl_route *route) +{ + return route->rt_src; +} + +int rtnl_route_get_src_len(struct rtnl_route *route) +{ + if (route->ce_mask & ROUTE_ATTR_SRC) + return nl_addr_get_prefixlen(route->rt_src); + else + return 0; +} + +int rtnl_route_set_gateway(struct rtnl_route *route, struct nl_addr *addr) +{ + if (route->ce_mask & ROUTE_ATTR_FAMILY) { + if (addr->a_family != route->rt_family) + return nl_error(EINVAL, "Address family mismatch"); + } else + route->rt_family = addr->a_family; + + if (route->rt_gateway) + nl_addr_put(route->rt_gateway); + + nl_addr_get(addr); + route->rt_gateway = addr; + route->ce_mask |= (ROUTE_ATTR_GATEWAY | ROUTE_ATTR_FAMILY); + + return 0; +} + +struct nl_addr *rtnl_route_get_gateway(struct rtnl_route *route) +{ + return route->rt_gateway; +} + +void rtnl_route_set_type(struct rtnl_route *route, int type) +{ + route->rt_type = type; + route->ce_mask |= ROUTE_ATTR_TYPE; +} + +int rtnl_route_get_type(struct rtnl_route *route) +{ + if (route->ce_mask & ROUTE_ATTR_TYPE) + return route->rt_type; + else + return RTN_UNICAST; +} + +void rtnl_route_set_flags(struct rtnl_route *route, unsigned int flags) +{ + route->rt_flag_mask |= flags; + route->rt_flags |= flags; + route->ce_mask |= ROUTE_ATTR_FLAGS; +} + +void rtnl_route_unset_flags(struct rtnl_route *route, unsigned int flags) +{ + route->rt_flag_mask |= flags; + route->rt_flags &= ~flags; + route->ce_mask |= ROUTE_ATTR_FLAGS; +} + +unsigned int rtnl_route_get_flags(struct rtnl_route *route) +{ + return route->rt_flags; +} + +int rtnl_route_set_metric(struct rtnl_route *route, int metric, uint32_t value) +{ + if (metric > RTAX_MAX || metric < 1) + return nl_error(EINVAL, "Metric out of range (1..%d)", + RTAX_MAX); + + route->rt_metrics[metric - 1] = value; + route->rt_metrics_mask |= (1 << (metric - 1)); + + return 0; +} + +int rtnl_route_unset_metric(struct rtnl_route *route, int metric) +{ + if (metric > RTAX_MAX || metric < 1) + return nl_error(EINVAL, "Metric out of range (1..%d)", + RTAX_MAX); + + route->rt_metrics_mask &= ~(1 << (metric - 1)); + + return 0; +} + +unsigned int rtnl_route_get_metric(struct rtnl_route *route, int metric) +{ + if (metric > RTAX_MAX || metric < 1) + return UINT_MAX; + + if (!(route->rt_metrics_mask & (1 << (metric - 1)))) + return UINT_MAX; + + return route->rt_metrics[metric - 1]; +} + +int rtnl_route_set_pref_src(struct rtnl_route *route, struct nl_addr *addr) +{ + if (route->ce_mask & ROUTE_ATTR_FAMILY) { + if (addr->a_family != route->rt_family) + return nl_error(EINVAL, "Address family mismatch"); + } else + route->rt_family = addr->a_family; + + if (route->rt_pref_src) + nl_addr_put(route->rt_pref_src); + + nl_addr_get(addr); + route->rt_pref_src = addr; + route->ce_mask |= (ROUTE_ATTR_PREF_SRC | ROUTE_ATTR_FAMILY); + + return 0; +} + +struct nl_addr *rtnl_route_get_pref_src(struct rtnl_route *route) +{ + return route->rt_pref_src; +} + +void rtnl_route_set_oif(struct rtnl_route *route, int ifindex) +{ + route->rt_oif = ifindex; + route->ce_mask |= ROUTE_ATTR_OIF; +} + +int rtnl_route_get_oif(struct rtnl_route *route) +{ + if (route->ce_mask & ROUTE_ATTR_OIF) + return route->rt_oif; + else + return RTNL_LINK_NOT_FOUND; +} + +void rtnl_route_set_iif(struct rtnl_route *route, const char *name) +{ + strncpy(route->rt_iif, name, sizeof(route->rt_iif) - 1); + route->ce_mask |= ROUTE_ATTR_IIF; +} + +char *rtnl_route_get_iif(struct rtnl_route *route) +{ + if (route->ce_mask & ROUTE_ATTR_IIF) + return route->rt_iif; + else + return NULL; +} + +void rtnl_route_add_nexthop(struct rtnl_route *route, struct rtnl_nexthop *nh) +{ + nl_list_add_tail(&nh->rtnh_list, &route->rt_nexthops); + route->ce_mask |= ROUTE_ATTR_MULTIPATH; +} + +void rtnl_route_remove_nexthop(struct rtnl_nexthop *nh) +{ + nl_list_del(&nh->rtnh_list); +} + +struct nl_list_head *rtnl_route_get_nexthops(struct rtnl_route *route) +{ + return &route->rt_nexthops; +} + +void rtnl_route_set_cacheinfo(struct rtnl_route *route, + struct rtnl_rtcacheinfo *ci) +{ + memcpy(&route->rt_cacheinfo, ci, sizeof(*ci)); + route->ce_mask |= ROUTE_ATTR_CACHEINFO; +} + +uint32_t rtnl_route_get_mp_algo(struct rtnl_route *route) +{ + if (route->ce_mask & ROUTE_ATTR_MP_ALGO) + return route->rt_mp_algo; + else + return IP_MP_ALG_NONE; +} + +void rtnl_route_set_mp_algo(struct rtnl_route *route, uint32_t algo) +{ + route->rt_mp_algo = algo; + route->ce_mask |= ROUTE_ATTR_MP_ALGO; +} + +/** @} */ + +struct nl_object_ops route_obj_ops = { + .oo_name = "route/route", + .oo_size = sizeof(struct rtnl_route), + .oo_constructor = route_constructor, + .oo_free_data = route_free_data, + .oo_clone = route_clone, + .oo_dump[NL_DUMP_BRIEF] = route_dump_brief, + .oo_dump[NL_DUMP_FULL] = route_dump_full, + .oo_dump[NL_DUMP_STATS] = route_dump_stats, + .oo_dump[NL_DUMP_XML] = route_dump_xml, + .oo_dump[NL_DUMP_ENV] = route_dump_env, + .oo_compare = route_compare, + .oo_attrs2str = route_attrs2str, + .oo_id_attrs = (ROUTE_ATTR_FAMILY | ROUTE_ATTR_TOS | + ROUTE_ATTR_TABLE | ROUTE_ATTR_DST), +}; + +/** @} */ diff --git a/lib/route/route_utils.c b/lib/route/route_utils.c new file mode 100644 index 0000000..a12d169 --- /dev/null +++ b/lib/route/route_utils.c @@ -0,0 +1,183 @@ +/* + * lib/route/route_utils.c Routing Utilities + * + * 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> + */ + +/** + * @ingroup route + * @defgroup route_utils Utilities + * @brief Routing Utility Functions + * + * + * @par 1) Translating Routing Table Names + * @code + * // libnl is only aware of the de facto standard routing table names. + * // Additional name <-> identifier associations have to be read in via + * // a configuration file, f.e. /etc/iproute2/rt_tables + * err = rtnl_route_read_table_names("/etc/iproute2/rt_tables"); + * + * // Translating a table name to its idenfier + * int table = rtnl_route_str2table("main"); + * + * // ... and the other way around. + * char buf[32]; + * printf("Name: %s\n", + * rtnl_route_table2str(table, buf, sizeof(buf))); + * @endcode + * + * + * + * + * @{ + */ + +#include <netlink-local.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/rtnl.h> +#include <netlink/route/route.h> + +/** + * @name Routing Table Identifier Translations + * @{ + */ + +static NL_LIST_HEAD(table_names); + +static int add_routing_table_name(long id, const char *name) +{ + return __trans_list_add(id, name, &table_names); +} + +static void __init init_routing_table_names(void) +{ + add_routing_table_name(RT_TABLE_UNSPEC, "unspec"); + add_routing_table_name(RT_TABLE_DEFAULT, "default"); + add_routing_table_name(RT_TABLE_MAIN, "main"); + add_routing_table_name(RT_TABLE_LOCAL, "local"); +}; + +int rtnl_route_read_table_names(const char *path) +{ + __trans_list_clear(&table_names); + + return __nl_read_num_str_file(path, &add_routing_table_name); +} + +char *rtnl_route_table2str(int table, char *buf, size_t size) +{ + return __list_type2str(table, buf, size, &table_names); +} + +int rtnl_route_str2table(const char *name) +{ + return __list_str2type(name, &table_names); +} + + +/** @} */ + +/** + * @name Routing Protocol Translations + * @{ + */ + +static NL_LIST_HEAD(proto_names); + +static int add_proto_name(long id, const char *name) +{ + return __trans_list_add(id, name, &proto_names); +} + +static void __init init_proto_names(void) +{ + add_proto_name(RTPROT_UNSPEC, "unspec"); + add_proto_name(RTPROT_REDIRECT, "redirect"); + add_proto_name(RTPROT_KERNEL, "kernel"); + add_proto_name(RTPROT_BOOT, "boot"); + add_proto_name(RTPROT_STATIC, "static"); +}; + +int rtnl_route_read_protocol_names(const char *path) +{ + __trans_list_clear(&proto_names); + + return __nl_read_num_str_file(path, &add_proto_name); +} + +char *rtnl_route_proto2str(int proto, char *buf, size_t size) +{ + return __list_type2str(proto, buf, size, &proto_names); +} + +int rtnl_route_str2proto(const char *name) +{ + return __list_str2type(name, &proto_names); +} + +/** @} */ + +/** + * @name Routing Metrices Translations + * @{ + */ + +static struct trans_tbl route_metrices[] = { + __ADD(RTAX_UNSPEC, unspec) + __ADD(RTAX_LOCK, lock) + __ADD(RTAX_MTU, mtu) + __ADD(RTAX_WINDOW, window) + __ADD(RTAX_RTT, rtt) + __ADD(RTAX_RTTVAR, rttvar) + __ADD(RTAX_SSTHRESH, ssthresh) + __ADD(RTAX_CWND, cwnd) + __ADD(RTAX_ADVMSS, advmss) + __ADD(RTAX_REORDERING, reordering) + __ADD(RTAX_HOPLIMIT, hoplimit) + __ADD(RTAX_INITCWND, initcwnd) + __ADD(RTAX_FEATURES, features) +}; + +char *rtnl_route_metric2str(int metric, char *buf, size_t size) +{ + return __type2str(metric, buf, size, route_metrices, + ARRAY_SIZE(route_metrices)); +} + +int rtnl_route_str2metric(const char *name) +{ + return __str2type(name, route_metrices, ARRAY_SIZE(route_metrices)); +} + +/** @} */ + +/** + * @name Nexthop Flags Translations + * @{ + */ + +static struct trans_tbl nh_flags[] = { + __ADD(RTNH_F_DEAD, dead) + __ADD(RTNH_F_PERVASIVE, pervasive) + __ADD(RTNH_F_ONLINK, onlink) +}; + +char * rtnl_route_nh_flags2str(int flags, char *buf, size_t len) +{ + return __flags2str(flags, buf, len, nh_flags, ARRAY_SIZE(nh_flags)); +} + +int rtnl_route_nh_str2flags(const char *name) +{ + return __str2flags(name, nh_flags, ARRAY_SIZE(nh_flags)); +} + +/** @} */ + +/** @} */ diff --git a/lib/route/rtnl.c b/lib/route/rtnl.c new file mode 100644 index 0000000..81ddf94 --- /dev/null +++ b/lib/route/rtnl.c @@ -0,0 +1,125 @@ +/* + * lib/route/rtnl.c Routing Netlink + * + * 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> + */ + +/** + * @ingroup nlfam + * @defgroup rtnl Routing Netlink + * @{ + */ + +#include <netlink-local.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/rtnl.h> + +/** + * @name Sending + * @{ + */ + +/** + * Send routing netlink request message + * @arg handle Netlink handle. + * @arg type Netlink message type. + * @arg family Address family. + * @arg flags Additional netlink message flags. + * + * Fills out a routing netlink request message and sends it out + * using nl_send_simple(). + * + * @return 0 on success or a negative error code. + */ +int nl_rtgen_request(struct nl_handle *handle, int type, int family, int flags) +{ + struct rtgenmsg gmsg = { + .rtgen_family = family, + }; + + return nl_send_simple(handle, type, flags, &gmsg, sizeof(gmsg)); +} + +/** @} */ + +/** + * @name Routing Type Translations + * @{ + */ + +static struct trans_tbl rtntypes[] = { + __ADD(RTN_UNSPEC,unspec) + __ADD(RTN_UNICAST,unicast) + __ADD(RTN_LOCAL,local) + __ADD(RTN_BROADCAST,broadcast) + __ADD(RTN_ANYCAST,anycast) + __ADD(RTN_MULTICAST,multicast) + __ADD(RTN_BLACKHOLE,blackhole) + __ADD(RTN_UNREACHABLE,unreachable) + __ADD(RTN_PROHIBIT,prohibit) + __ADD(RTN_THROW,throw) + __ADD(RTN_NAT,nat) + __ADD(RTN_XRESOLVE,xresolve) +}; + +char *nl_rtntype2str(int type, char *buf, size_t size) +{ + return __type2str(type, buf, size, rtntypes, ARRAY_SIZE(rtntypes)); +} + +int nl_str2rtntype(const char *name) +{ + return __str2type(name, rtntypes, ARRAY_SIZE(rtntypes)); +} + +/** @} */ + +/** + * @name Scope Translations + * @{ + */ + +static struct trans_tbl scopes[] = { + __ADD(255,nowhere) + __ADD(254,host) + __ADD(253,link) + __ADD(200,site) + __ADD(0,universe) +}; + +char *rtnl_scope2str(int scope, char *buf, size_t size) +{ + return __type2str(scope, buf, size, scopes, ARRAY_SIZE(scopes)); +} + +int rtnl_str2scope(const char *name) +{ + return __str2type(name, scopes, ARRAY_SIZE(scopes)); +} + +/** @} */ + +/** + * @name Realms Translations + * @{ + */ + +char * rtnl_realms2str(uint32_t realms, char *buf, size_t len) +{ + int from = RTNL_REALM_FROM(realms); + int to = RTNL_REALM_TO(realms); + + snprintf(buf, len, "%d/%d", from, to); + + return buf; +} + +/** @} */ + +/** @} */ diff --git a/lib/route/rule.c b/lib/route/rule.c new file mode 100644 index 0000000..76e0060 --- /dev/null +++ b/lib/route/rule.c @@ -0,0 +1,900 @@ +/* + * lib/route/rule.c Routing Rules + * + * 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> + */ + +/** + * @ingroup rtnl + * @defgroup rule Routing Rules + * @brief + * @{ + */ + +#include <netlink-local.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/rtnl.h> +#include <netlink/route/rule.h> +#include <inttypes.h> + +/** @cond SKIP */ +#define RULE_ATTR_FAMILY 0x0001 +#define RULE_ATTR_PRIO 0x0002 +#define RULE_ATTR_MARK 0x0004 +#define RULE_ATTR_IIF 0x0008 +#define RULE_ATTR_REALMS 0x0010 +#define RULE_ATTR_SRC 0x0020 +#define RULE_ATTR_DST 0x0040 +#define RULE_ATTR_DSFIELD 0x0080 +#define RULE_ATTR_TABLE 0x0100 +#define RULE_ATTR_TYPE 0x0200 +#define RULE_ATTR_SRC_LEN 0x0400 +#define RULE_ATTR_DST_LEN 0x0800 +#define RULE_ATTR_SRCMAP 0x1000 + +static struct nl_cache_ops rtnl_rule_ops; +static struct nl_object_ops rule_obj_ops; +/** @endcond */ + +static void rule_free_data(struct nl_object *c) +{ + struct rtnl_rule *rule = nl_object_priv(c); + + if (!rule) + return; + + nl_addr_put(rule->r_src); + nl_addr_put(rule->r_dst); +} + +static int rule_clone(struct nl_object *_dst, struct nl_object *_src) +{ + struct rtnl_rule *dst = nl_object_priv(_dst); + struct rtnl_rule *src = nl_object_priv(_src); + + if (src->r_src) + if (!(dst->r_src = nl_addr_clone(src->r_src))) + goto errout; + + if (src->r_dst) + if (!(dst->r_dst = nl_addr_clone(src->r_dst))) + goto errout; + + return 0; +errout: + return nl_get_errno(); +} + +static struct nla_policy rule_policy[RTA_MAX+1] = { + [RTA_PRIORITY] = { .type = NLA_U32 }, + [RTA_FLOW] = { .type = NLA_U32 }, + [RTA_PROTOINFO] = { .type = NLA_U32 }, + [RTA_IIF] = { .type = NLA_STRING, + .maxlen = IFNAMSIZ, }, +}; + +static int rule_msg_parser(struct nl_cache_ops *ops, struct sockaddr_nl *who, + struct nlmsghdr *n, void *arg) +{ + struct rtnl_rule *rule; + struct rtmsg *r; + struct nlattr *tb[RTA_MAX+1]; + struct nl_parser_param *pp = arg; + int err = 1; + + rule = rtnl_rule_alloc(); + if (!rule) { + err = nl_errno(ENOMEM); + goto errout; + } + + rule->ce_msgtype = n->nlmsg_type; + r = nlmsg_data(n); + + err = nlmsg_parse(n, sizeof(*r), tb, RTA_MAX, rule_policy); + if (err < 0) + goto errout; + + rule->r_family = r->rtm_family; + rule->r_type = r->rtm_type; + rule->r_dsfield = r->rtm_tos; + rule->r_src_len = r->rtm_src_len; + rule->r_dst_len = r->rtm_dst_len; + rule->r_table = r->rtm_table; + rule->ce_mask = (RULE_ATTR_FAMILY | RULE_ATTR_TYPE | RULE_ATTR_DSFIELD | + RULE_ATTR_SRC_LEN | RULE_ATTR_DST_LEN |RULE_ATTR_TYPE); + + if (tb[RTA_PRIORITY]) { + rule->r_prio = nla_get_u32(tb[RTA_PRIORITY]); + rule->ce_mask |= RULE_ATTR_PRIO; + } + + if (tb[RTA_SRC]) { + rule->r_src = nla_get_addr(tb[RTA_SRC], r->rtm_family); + if (!rule->r_src) { + err = nl_errno(ENOMEM); + goto errout; + } + nl_addr_set_prefixlen(rule->r_src, r->rtm_src_len); + rule->ce_mask |= RULE_ATTR_SRC; + } + + if (tb[RTA_DST]) { + rule->r_dst = nla_get_addr(tb[RTA_DST], r->rtm_family); + if (!rule->r_dst) { + err = nl_errno(ENOMEM); + goto errout; + } + nl_addr_set_prefixlen(rule->r_dst, r->rtm_dst_len); + rule->ce_mask |= RULE_ATTR_DST; + } + + if (tb[RTA_PROTOINFO]) { + rule->r_mark = nla_get_u32(tb[RTA_PROTOINFO]); + rule->ce_mask |= RULE_ATTR_MARK; + } + + if (tb[RTA_IIF]) { + nla_strlcpy(rule->r_iif, tb[RTA_IIF], IFNAMSIZ); + rule->ce_mask |= RULE_ATTR_IIF; + } + + if (tb[RTA_FLOW]) { + rule->r_realms = nla_get_u32(tb[RTA_FLOW]); + rule->ce_mask |= RULE_ATTR_REALMS; + } + + if (tb[RTA_GATEWAY]) { + rule->r_srcmap = nla_get_addr(tb[RTA_GATEWAY], r->rtm_family); + if (!rule->r_srcmap) { + err = nl_errno(ENOMEM); + goto errout; + } + rule->ce_mask |= RULE_ATTR_SRCMAP; + } + + err = pp->pp_cb((struct nl_object *) rule, pp); + if (err < 0) + goto errout; + + return P_ACCEPT; + +errout: + rtnl_rule_put(rule); + return err; +} + +static int rule_request_update(struct nl_cache *c, struct nl_handle *h) +{ + return nl_rtgen_request(h, RTM_GETRULE, AF_UNSPEC, NLM_F_DUMP); +} + +static int rule_dump_brief(struct nl_object *o, struct nl_dump_params *p) +{ + struct rtnl_rule *r = (struct rtnl_rule *) o; + char buf[128]; + + if (r->ce_mask & RULE_ATTR_PRIO) + dp_dump(p, "%d:\t", r->r_prio); + else + dp_dump(p, "0:\t"); + + if (r->ce_mask & RULE_ATTR_SRC) + dp_dump(p, "from %s ", + nl_addr2str(r->r_src, buf, sizeof(buf))); + else if (r->ce_mask & RULE_ATTR_SRC_LEN && r->r_src_len) + dp_dump(p, "from 0/%d ", r->r_src_len); + + if (r->ce_mask & RULE_ATTR_DST) + dp_dump(p, "to %s ", + nl_addr2str(r->r_dst, buf, sizeof(buf))); + else if (r->ce_mask & RULE_ATTR_DST_LEN && r->r_dst_len) + dp_dump(p, "to 0/%d ", r->r_dst_len); + + if (r->ce_mask & RULE_ATTR_DSFIELD && r->r_dsfield) + dp_dump(p, "tos %d ", r->r_dsfield); + + if (r->ce_mask & RULE_ATTR_MARK) + dp_dump(p, "mark %" PRIx64 , r->r_mark); + + if (r->ce_mask & RULE_ATTR_IIF) + dp_dump(p, "iif %s ", r->r_iif); + + if (r->ce_mask & RULE_ATTR_TABLE) + dp_dump(p, "lookup %s ", + rtnl_route_table2str(r->r_table, buf, sizeof(buf))); + + if (r->ce_mask & RULE_ATTR_REALMS) + dp_dump(p, "realms %s ", + rtnl_realms2str(r->r_realms, buf, sizeof(buf))); + + dp_dump(p, "action %s\n", + nl_rtntype2str(r->r_type, buf, sizeof(buf))); + + return 1; +} + +static int rule_dump_full(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_rule *rule = (struct rtnl_rule *) obj; + char buf[128]; + int line; + + line = rule_dump_brief(obj, p); + + dp_dump_line(p, line++, " family %s", + nl_af2str(rule->r_family, buf, sizeof(buf))); + + if (rule->ce_mask & RULE_ATTR_SRCMAP) + dp_dump(p, " srcmap %s", + nl_addr2str(rule->r_srcmap, buf, sizeof(buf))); + + dp_dump(p, "\n"); + + return line; +} + +static int rule_dump_stats(struct nl_object *obj, struct nl_dump_params *p) +{ + return rule_dump_full(obj, p); +} + +static int rule_dump_xml(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_rule *rule = (struct rtnl_rule *) obj; + char buf[128]; + int line = 0; + + dp_dump_line(p, line++, "<rule>\n"); + + dp_dump_line(p, line++, " <priority>%u</priority>\n", + rule->r_prio); + dp_dump_line(p, line++, " <family>%s</family>\n", + nl_af2str(rule->r_family, buf, sizeof(buf))); + + if (rule->ce_mask & RULE_ATTR_DST) + dp_dump_line(p, line++, " <dst>%s</dst>\n", + nl_addr2str(rule->r_dst, buf, sizeof(buf))); + + if (rule->ce_mask & RULE_ATTR_DST_LEN) + dp_dump_line(p, line++, " <dstlen>%u</dstlen>\n", + rule->r_dst_len); + + if (rule->ce_mask & RULE_ATTR_SRC) + dp_dump_line(p, line++, " <src>%s</src>\n", + nl_addr2str(rule->r_src, buf, sizeof(buf))); + + if (rule->ce_mask & RULE_ATTR_SRC_LEN) + dp_dump_line(p, line++, " <srclen>%u</srclen>\n", + rule->r_src_len); + + if (rule->ce_mask & RULE_ATTR_IIF) + dp_dump_line(p, line++, " <iif>%s</iif>\n", rule->r_iif); + + if (rule->ce_mask & RULE_ATTR_TABLE) + dp_dump_line(p, line++, " <table>%u</table>\n", + rule->r_table); + + if (rule->ce_mask & RULE_ATTR_REALMS) + dp_dump_line(p, line++, " <realms>%u</realms>\n", + rule->r_realms); + + if (rule->ce_mask & RULE_ATTR_MARK) + dp_dump_line(p, line++, " <mark>%" PRIx64 "</mark>\n", + rule->r_mark); + + if (rule->ce_mask & RULE_ATTR_DSFIELD) + dp_dump_line(p, line++, " <dsfield>%u</dsfield>\n", + rule->r_dsfield); + + if (rule->ce_mask & RULE_ATTR_TYPE) + dp_dump_line(p, line++, "<type>%s</type>\n", + nl_rtntype2str(rule->r_type, buf, sizeof(buf))); + + if (rule->ce_mask & RULE_ATTR_SRCMAP) + dp_dump_line(p, line++, "<srcmap>%s</srcmap>\n", + nl_addr2str(rule->r_srcmap, buf, sizeof(buf))); + + dp_dump_line(p, line++, "</rule>\n"); + + return line; +} + +static int rule_dump_env(struct nl_object *obj, struct nl_dump_params *p) +{ + struct rtnl_rule *rule = (struct rtnl_rule *) obj; + char buf[128]; + int line = 0; + + dp_dump_line(p, line++, "RULE_PRIORITY=%u\n", + rule->r_prio); + dp_dump_line(p, line++, "RULE_FAMILY=%s\n", + nl_af2str(rule->r_family, buf, sizeof(buf))); + + if (rule->ce_mask & RULE_ATTR_DST) + dp_dump_line(p, line++, "RULE_DST=%s\n", + nl_addr2str(rule->r_dst, buf, sizeof(buf))); + + if (rule->ce_mask & RULE_ATTR_DST_LEN) + dp_dump_line(p, line++, "RULE_DSTLEN=%u\n", + rule->r_dst_len); + + if (rule->ce_mask & RULE_ATTR_SRC) + dp_dump_line(p, line++, "RULE_SRC=%s\n", + nl_addr2str(rule->r_src, buf, sizeof(buf))); + + if (rule->ce_mask & RULE_ATTR_SRC_LEN) + dp_dump_line(p, line++, "RULE_SRCLEN=%u\n", + rule->r_src_len); + + if (rule->ce_mask & RULE_ATTR_IIF) + dp_dump_line(p, line++, "RULE_IIF=%s\n", rule->r_iif); + + if (rule->ce_mask & RULE_ATTR_TABLE) + dp_dump_line(p, line++, "RULE_TABLE=%u\n", + rule->r_table); + + if (rule->ce_mask & RULE_ATTR_REALMS) + dp_dump_line(p, line++, "RULE_REALM=%u\n", + rule->r_realms); + + if (rule->ce_mask & RULE_ATTR_MARK) + dp_dump_line(p, line++, "RULE_MARK=0x%" PRIx64 "\n", + rule->r_mark); + + if (rule->ce_mask & RULE_ATTR_DSFIELD) + dp_dump_line(p, line++, "RULE_DSFIELD=%u\n", + rule->r_dsfield); + + if (rule->ce_mask & RULE_ATTR_TYPE) + dp_dump_line(p, line++, "RULE_TYPE=%s\n", + nl_rtntype2str(rule->r_type, buf, sizeof(buf))); + + if (rule->ce_mask & RULE_ATTR_SRCMAP) + dp_dump_line(p, line++, "RULE_SRCMAP=%s\n", + nl_addr2str(rule->r_srcmap, buf, sizeof(buf))); + + return line; +} + +static int rule_compare(struct nl_object *_a, struct nl_object *_b, + uint32_t attrs, int flags) +{ + struct rtnl_rule *a = (struct rtnl_rule *) _a; + struct rtnl_rule *b = (struct rtnl_rule *) _b; + int diff = 0; + +#define RULE_DIFF(ATTR, EXPR) ATTR_DIFF(attrs, RULE_ATTR_##ATTR, a, b, EXPR) + + diff |= RULE_DIFF(FAMILY, a->r_family != b->r_family); + diff |= RULE_DIFF(TABLE, a->r_table != b->r_table); + diff |= RULE_DIFF(REALMS, a->r_realms != b->r_realms); + diff |= RULE_DIFF(DSFIELD, a->r_dsfield != b->r_dsfield); + diff |= RULE_DIFF(TYPE, a->r_type != b->r_type); + diff |= RULE_DIFF(PRIO, a->r_prio != b->r_prio); + diff |= RULE_DIFF(MARK, a->r_mark != b->r_mark); + diff |= RULE_DIFF(SRC_LEN, a->r_src_len != b->r_src_len); + diff |= RULE_DIFF(DST_LEN, a->r_dst_len != b->r_dst_len); + diff |= RULE_DIFF(SRC, nl_addr_cmp(a->r_src, b->r_src)); + diff |= RULE_DIFF(DST, nl_addr_cmp(a->r_dst, b->r_dst)); + diff |= RULE_DIFF(IIF, strcmp(a->r_iif, b->r_iif)); + +#undef RULE_DIFF + + return diff; +} + +static struct trans_tbl rule_attrs[] = { + __ADD(RULE_ATTR_FAMILY, family) + __ADD(RULE_ATTR_PRIO, prio) + __ADD(RULE_ATTR_MARK, mark) + __ADD(RULE_ATTR_IIF, iif) + __ADD(RULE_ATTR_REALMS, realms) + __ADD(RULE_ATTR_SRC, src) + __ADD(RULE_ATTR_DST, dst) + __ADD(RULE_ATTR_DSFIELD, dsfield) + __ADD(RULE_ATTR_TABLE, table) + __ADD(RULE_ATTR_TYPE, type) + __ADD(RULE_ATTR_SRC_LEN, src_len) + __ADD(RULE_ATTR_DST_LEN, dst_len) + __ADD(RULE_ATTR_SRCMAP, srcmap) +}; + +static char *rule_attrs2str(int attrs, char *buf, size_t len) +{ + return __flags2str(attrs, buf, len, rule_attrs, + ARRAY_SIZE(rule_attrs)); +} + +/** + * @name Allocation/Freeing + * @{ + */ + +struct rtnl_rule *rtnl_rule_alloc(void) +{ + return (struct rtnl_rule *) nl_object_alloc(&rule_obj_ops); +} + +void rtnl_rule_put(struct rtnl_rule *rule) +{ + nl_object_put((struct nl_object *) rule); +} + +/** @} */ + +/** + * @name Cache Management + * @{ + */ + +/** + * Build a rule cache including all rules of the specified family currently configured in the kernel. + * @arg handle netlink handle + * @arg family address family + * + * Allocates a new rule cache, initializes it properly and updates it + * to include all rules of the specified address family currently + * configured in the kernel. + * + * @note The caller is responsible for destroying and freeing the + * cache after using it. (nl_cache_destroy_and_free()) + * @return The new cache or NULL if an error occured. + */ +struct nl_cache * rtnl_rule_alloc_cache_by_family(struct nl_handle *handle, + int family) +{ + struct nl_cache * cache; + + cache = nl_cache_alloc(&rtnl_rule_ops); + if (cache == NULL) + return NULL; + + /* XXX RULE_CACHE_FAMILY(cache) = family; */ + + if (handle && nl_cache_refill(handle, cache) < 0) { + free(cache); + return NULL; + } + + return cache; +} + +/** + * Build a rule cache including all rules currently configured in the kernel. + * @arg handle netlink handle + * + * Allocates a new rule cache, initializes it properly and updates it + * to include all rules currently configured in the kernel. + * + * @note The caller is responsible for destroying and freeing the + * cache after using it. (nl_cache_destroy_and_free()) + * @return The new cache or NULL if an error occured. + */ +struct nl_cache * rtnl_rule_alloc_cache(struct nl_handle *handle) +{ + return rtnl_rule_alloc_cache_by_family(handle, AF_UNSPEC); +} + +/** @} */ + +/** + * @name Rule Addition + * @{ + */ + +static struct nl_msg *build_rule_msg(struct rtnl_rule *tmpl, int cmd, int flags) +{ + struct nl_msg *msg; + struct rtmsg rtm = { + .rtm_type = RTN_UNSPEC + }; + + if (cmd == RTM_NEWRULE) + rtm.rtm_type = RTN_UNICAST; + + if (tmpl->ce_mask & RULE_ATTR_FAMILY) + rtm.rtm_family = tmpl->r_family; + + if (tmpl->ce_mask & RULE_ATTR_TABLE) + rtm.rtm_table = tmpl->r_table; + + if (tmpl->ce_mask & RULE_ATTR_DSFIELD) + rtm.rtm_tos = tmpl->r_dsfield; + + if (tmpl->ce_mask & RULE_ATTR_TYPE) + rtm.rtm_type = tmpl->r_type; + + if (tmpl->ce_mask & RULE_ATTR_SRC_LEN) + rtm.rtm_src_len = tmpl->r_src_len; + + if (tmpl->ce_mask & RULE_ATTR_DST_LEN) + rtm.rtm_dst_len = tmpl->r_dst_len; + + msg = nlmsg_alloc_simple(cmd, flags); + if (!msg) + goto nla_put_failure; + + if (nlmsg_append(msg, &rtm, sizeof(rtm), NLMSG_ALIGNTO) < 0) + goto nla_put_failure; + + if (tmpl->ce_mask & RULE_ATTR_SRC) + NLA_PUT_ADDR(msg, RTA_SRC, tmpl->r_src); + + if (tmpl->ce_mask & RULE_ATTR_DST) + NLA_PUT_ADDR(msg, RTA_DST, tmpl->r_dst); + + if (tmpl->ce_mask & RULE_ATTR_PRIO) + NLA_PUT_U32(msg, RTA_PRIORITY, tmpl->r_prio); + + if (tmpl->ce_mask & RULE_ATTR_MARK) + NLA_PUT_U32(msg, RTA_PROTOINFO, tmpl->r_mark); + + if (tmpl->ce_mask & RULE_ATTR_REALMS) + NLA_PUT_U32(msg, RTA_FLOW, tmpl->r_realms); + + if (tmpl->ce_mask & RULE_ATTR_IIF) + NLA_PUT_STRING(msg, RTA_IIF, tmpl->r_iif); + + return msg; + +nla_put_failure: + nlmsg_free(msg); + return NULL; +} + +/** + * Build netlink request message to add a new rule + * @arg tmpl template with data of new rule + * @arg flags additional netlink message flags + * + * Builds a new netlink message requesting a addition of a new + * rule. The netlink message header isn't fully equipped with + * all relevant fields and must thus be sent out via nl_send_auto_complete() + * or supplemented as needed. \a tmpl must contain the attributes of the new + * address set via \c rtnl_rule_set_* functions. + * + * @return The netlink message + */ +struct nl_msg *rtnl_rule_build_add_request(struct rtnl_rule *tmpl, int flags) +{ + return build_rule_msg(tmpl, RTM_NEWRULE, NLM_F_CREATE | flags); +} + +/** + * Add a new rule + * @arg handle netlink handle + * @arg tmpl template with requested changes + * @arg flags additional netlink message flags + * + * Builds a netlink message by calling rtnl_rule_build_add_request(), + * sends the request to the kernel and waits for the next ACK to be + * received and thus blocks until the request has been fullfilled. + * + * @return 0 on sucess or a negative error if an error occured. + */ +int rtnl_rule_add(struct nl_handle *handle, struct rtnl_rule *tmpl, int flags) +{ + int err; + struct nl_msg *msg; + + msg = rtnl_rule_build_add_request(tmpl, flags); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name Rule Deletion + * @{ + */ + +/** + * Build a netlink request message to delete a rule + * @arg rule rule to delete + * @arg flags additional netlink message flags + * + * Builds a new netlink message requesting a deletion of a rule. + * The netlink message header isn't fully equipped with all relevant + * fields and must thus be sent out via nl_send_auto_complete() + * or supplemented as needed. \a rule must point to an existing + * address. + * + * @return The netlink message + */ +struct nl_msg *rtnl_rule_build_delete_request(struct rtnl_rule *rule, int flags) +{ + return build_rule_msg(rule, RTM_DELRULE, flags); +} + +/** + * Delete a rule + * @arg handle netlink handle + * @arg rule rule to delete + * @arg flags additional netlink message flags + * + * Builds a netlink message by calling rtnl_rule_build_delete_request(), + * sends the request to the kernel and waits for the next ACK to be + * received and thus blocks until the request has been fullfilled. + * + * @return 0 on sucess or a negative error if an error occured. + */ +int rtnl_rule_delete(struct nl_handle *handle, struct rtnl_rule *rule, + int flags) +{ + int err; + struct nl_msg *msg; + + msg = rtnl_rule_build_delete_request(rule, flags); + if (!msg) + return nl_errno(ENOMEM); + + err = nl_send_auto_complete(handle, msg); + if (err < 0) + return err; + + nlmsg_free(msg); + return nl_wait_for_ack(handle); +} + +/** @} */ + +/** + * @name Attribute Modification + * @{ + */ + +void rtnl_rule_set_family(struct rtnl_rule *rule, int family) +{ + rule->r_family = family; + rule->ce_mask |= RULE_ATTR_FAMILY; +} + +int rtnl_rule_get_family(struct rtnl_rule *rule) +{ + if (rule->ce_mask & RULE_ATTR_FAMILY) + return rule->r_family; + else + return AF_UNSPEC; +} + +void rtnl_rule_set_prio(struct rtnl_rule *rule, int prio) +{ + rule->r_prio = prio; + rule->ce_mask |= RULE_ATTR_PRIO; +} + +int rtnl_rule_get_prio(struct rtnl_rule *rule) +{ + if (rule->ce_mask & RULE_ATTR_PRIO) + return rule->r_prio; + else + return -1; +} + +void rtnl_rule_set_mark(struct rtnl_rule *rule, uint64_t mark) +{ + rule->r_mark = mark; + rule->ce_mask |= RULE_ATTR_MARK; +} + +uint64_t rtnl_rule_get_mark(struct rtnl_rule *rule) +{ + if (rule->ce_mask & RULE_ATTR_MARK) + return rule->r_mark; + else + return UINT_LEAST64_MAX; +} + +void rtnl_rule_set_table(struct rtnl_rule *rule, int table) +{ + rule->r_table = table; + rule->ce_mask |= RULE_ATTR_TABLE; +} + +int rtnl_rule_get_table(struct rtnl_rule *rule) +{ + if (rule->ce_mask & RULE_ATTR_TABLE) + return rule->r_table; + else + return -1; +} + +void rtnl_rule_set_dsfield(struct rtnl_rule *rule, int dsfield) +{ + rule->r_dsfield = dsfield; + rule->ce_mask |= RULE_ATTR_DSFIELD; +} + +int rtnl_rule_get_dsfield(struct rtnl_rule *rule) +{ + if (rule->ce_mask & RULE_ATTR_DSFIELD) + return rule->r_dsfield; + else + return -1; +} + +void rtnl_rule_set_src_len(struct rtnl_rule *rule, int len) +{ + rule->r_src_len = len; + if (rule->ce_mask & RULE_ATTR_SRC) + nl_addr_set_prefixlen(rule->r_src, len); + rule->ce_mask |= RULE_ATTR_SRC_LEN; +} + +int rtnl_rule_get_src_len(struct rtnl_rule *rule) +{ + if (rule->ce_mask & RULE_ATTR_SRC_LEN) + return rule->r_src_len; + else + return -1; +} + +void rtnl_rule_set_dst_len(struct rtnl_rule *rule, int len) +{ + rule->r_dst_len = len; + if (rule->ce_mask & RULE_ATTR_DST) + nl_addr_set_prefixlen(rule->r_dst, len); + rule->ce_mask |= RULE_ATTR_DST_LEN; +} + +int rtnl_rule_get_dst_len(struct rtnl_rule *rule) +{ + if (rule->ce_mask & RULE_ATTR_DST_LEN) + return rule->r_dst_len; + else + return -1; +} + +static inline int __assign_addr(struct rtnl_rule *rule, struct nl_addr **pos, + struct nl_addr *new, uint8_t *len, int flag) +{ + if (rule->ce_mask & RULE_ATTR_FAMILY) { + if (new->a_family != rule->r_family) + return nl_error(EINVAL, "Address family mismatch"); + } else + rule->r_family = new->a_family; + + if (*pos) + nl_addr_put(*pos); + + nl_addr_get(new); + *pos = new; + *len = nl_addr_get_prefixlen(new); + + rule->ce_mask |= (flag | RULE_ATTR_FAMILY); + + return 0; +} + +int rtnl_rule_set_src(struct rtnl_rule *rule, struct nl_addr *src) +{ + return __assign_addr(rule, &rule->r_src, src, &rule->r_src_len, + RULE_ATTR_SRC | RULE_ATTR_SRC_LEN); +} + +struct nl_addr *rtnl_rule_get_src(struct rtnl_rule *rule) +{ + if (rule->ce_mask & RULE_ATTR_SRC) + return rule->r_src; + else + return NULL; +} + +int rtnl_rule_set_dst(struct rtnl_rule *rule, struct nl_addr *dst) +{ + return __assign_addr(rule, &rule->r_dst, dst, &rule->r_dst_len, + RULE_ATTR_DST | RULE_ATTR_DST_LEN); +} + +struct nl_addr *rtnl_rule_get_dst(struct rtnl_rule *rule) +{ + if (rule->ce_mask & RULE_ATTR_DST) + return rule->r_dst; + else + return NULL; +} + +int rtnl_rule_set_iif(struct rtnl_rule *rule, const char *dev) +{ + if (strlen(dev) > IFNAMSIZ-1) + return nl_errno(ERANGE); + + strcpy(rule->r_iif, dev); + rule->ce_mask |= RULE_ATTR_IIF; + return 0; +} + +char *rtnl_rule_get_iif(struct rtnl_rule *rule) +{ + if (rule->ce_mask & RULE_ATTR_IIF) + return rule->r_iif; + else + return NULL; +} + +void rtnl_rule_set_action(struct rtnl_rule *rule, int type) +{ + rule->r_type = type; + rule->ce_mask |= RULE_ATTR_TYPE; +} + +int rtnl_rule_get_action(struct rtnl_rule *rule) +{ + if (rule->ce_mask & RULE_ATTR_TYPE) + return rule->r_type; + else + return nl_errno(ENOENT); +} + +void rtnl_rule_set_realms(struct rtnl_rule *rule, realm_t realms) +{ + rule->r_realms = realms; + rule->ce_mask |= RULE_ATTR_REALMS; +} + +realm_t rtnl_rule_get_realms(struct rtnl_rule *rule) +{ + if (rule->ce_mask & RULE_ATTR_REALMS) + return rule->r_realms; + else + return 0; +} + +/** @} */ + +static struct nl_object_ops rule_obj_ops = { + .oo_name = "route/rule", + .oo_size = sizeof(struct rtnl_rule), + .oo_free_data = rule_free_data, + .oo_clone = rule_clone, + .oo_dump[NL_DUMP_BRIEF] = rule_dump_brief, + .oo_dump[NL_DUMP_FULL] = rule_dump_full, + .oo_dump[NL_DUMP_STATS] = rule_dump_stats, + .oo_dump[NL_DUMP_XML] = rule_dump_xml, + .oo_dump[NL_DUMP_ENV] = rule_dump_env, + .oo_compare = rule_compare, + .oo_attrs2str = rule_attrs2str, + .oo_id_attrs = ~0, +}; + +static struct nl_cache_ops rtnl_rule_ops = { + .co_name = "route/rule", + .co_hdrsize = sizeof(struct rtmsg), + .co_msgtypes = { + { RTM_NEWRULE, NL_ACT_NEW, "new" }, + { RTM_DELRULE, NL_ACT_DEL, "del" }, + { RTM_GETRULE, NL_ACT_GET, "get" }, + END_OF_MSGTYPES_LIST, + }, + .co_protocol = NETLINK_ROUTE, + .co_request_update = rule_request_update, + .co_msg_parser = rule_msg_parser, + .co_obj_ops = &rule_obj_ops, +}; + +static void __init rule_init(void) +{ + nl_cache_mngt_register(&rtnl_rule_ops); +} + +static void __exit rule_exit(void) +{ + nl_cache_mngt_unregister(&rtnl_rule_ops); +} + +/** @} */ diff --git a/lib/route/sch/blackhole.c b/lib/route/sch/blackhole.c new file mode 100644 index 0000000..a30b693 --- /dev/null +++ b/lib/route/sch/blackhole.c @@ -0,0 +1,38 @@ +/* + * lib/route/sch/blackhole.c Blackhole Qdisc + * + * 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> + */ + +/** + * @ingroup qdisc_api + * @defgroup blackhole Blackhole + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/qdisc-modules.h> + +static struct rtnl_qdisc_ops blackhole_ops = { + .qo_kind = "blackhole", +}; + +static void __init blackhole_init(void) +{ + rtnl_qdisc_register(&blackhole_ops); +} + +static void __exit blackhole_exit(void) +{ + rtnl_qdisc_unregister(&blackhole_ops); +} + +/** @} */ diff --git a/lib/route/sch/cbq.c b/lib/route/sch/cbq.c new file mode 100644 index 0000000..9808509 --- /dev/null +++ b/lib/route/sch/cbq.c @@ -0,0 +1,306 @@ +/* + * lib/route/sch/cbq.c Class Based Queueing + * + * 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 <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/qdisc-modules.h> +#include <netlink/route/class.h> +#include <netlink/route/class-modules.h> +#include <netlink/route/link.h> +#include <netlink/route/sch/cbq.h> +#include <netlink/route/cls/police.h> + +/** + * @ingroup qdisc_api + * @ingroup class_api + * @defgroup cbq Class Based Queueing (CBQ) + * @{ + */ + +static struct trans_tbl ovl_strategies[] = { + __ADD(TC_CBQ_OVL_CLASSIC,classic) + __ADD(TC_CBQ_OVL_DELAY,delay) + __ADD(TC_CBQ_OVL_LOWPRIO,lowprio) + __ADD(TC_CBQ_OVL_DROP,drop) + __ADD(TC_CBQ_OVL_RCLASSIC,rclassic) +}; + +/** + * Convert a CBQ OVL strategy to a character string + * @arg type CBQ OVL strategy + * @arg buf destination buffer + * @arg len length of destination buffer + * + * Converts a CBQ OVL strategy to a character string and stores in the + * provided buffer. Returns the destination buffer or the type + * encoded in hex if no match was found. + */ +char *nl_ovl_strategy2str(int type, char *buf, size_t len) +{ + return __type2str(type, buf, len, ovl_strategies, + ARRAY_SIZE(ovl_strategies)); +} + +/** + * Convert a string to a CBQ OVL strategy + * @arg name CBQ OVL stragegy name + * + * Converts a CBQ OVL stragegy name to it's corresponding CBQ OVL strategy + * type. Returns the type or -1 if none was found. + */ +int nl_str2ovl_strategy(const char *name) +{ + return __str2type(name, ovl_strategies, ARRAY_SIZE(ovl_strategies)); +} + +static struct nla_policy cbq_policy[TCA_CBQ_MAX+1] = { + [TCA_CBQ_LSSOPT] = { .minlen = sizeof(struct tc_cbq_lssopt) }, + [TCA_CBQ_RATE] = { .minlen = sizeof(struct tc_ratespec) }, + [TCA_CBQ_WRROPT] = { .minlen = sizeof(struct tc_cbq_wrropt) }, + [TCA_CBQ_OVL_STRATEGY] = { .minlen = sizeof(struct tc_cbq_ovl) }, + [TCA_CBQ_FOPT] = { .minlen = sizeof(struct tc_cbq_fopt) }, + [TCA_CBQ_POLICE] = { .minlen = sizeof(struct tc_cbq_police) }, +}; + +static inline struct rtnl_cbq *cbq_qdisc(struct rtnl_tca *tca) +{ + return (struct rtnl_cbq *) tca->tc_subdata; +} + +static inline struct rtnl_cbq *cbq_alloc(struct rtnl_tca *tca) +{ + if (!tca->tc_subdata) + tca->tc_subdata = calloc(1, sizeof(struct rtnl_qdisc)); + + return cbq_qdisc(tca); +} + + +static int cbq_msg_parser(struct rtnl_tca *tca) +{ + struct nlattr *tb[TCA_CBQ_MAX + 1]; + struct rtnl_cbq *cbq; + int err; + + err = tca_parse(tb, TCA_CBQ_MAX, tca, cbq_policy); + if (err < 0) + return err; + + cbq = cbq_alloc(tca); + if (!cbq) + return nl_errno(ENOMEM); + + nla_memcpy(&cbq->cbq_lss, tb[TCA_CBQ_LSSOPT], sizeof(cbq->cbq_lss)); + nla_memcpy(&cbq->cbq_rate, tb[TCA_CBQ_RATE], sizeof(cbq->cbq_rate)); + nla_memcpy(&cbq->cbq_wrr, tb[TCA_CBQ_WRROPT], sizeof(cbq->cbq_wrr)); + nla_memcpy(&cbq->cbq_fopt, tb[TCA_CBQ_FOPT], sizeof(cbq->cbq_fopt)); + nla_memcpy(&cbq->cbq_ovl, tb[TCA_CBQ_OVL_STRATEGY], + sizeof(cbq->cbq_ovl)); + nla_memcpy(&cbq->cbq_police, tb[TCA_CBQ_POLICE], + sizeof(cbq->cbq_police)); + + return 0; +} + +static int cbq_qdisc_msg_parser(struct rtnl_qdisc *qdisc) +{ + return cbq_msg_parser((struct rtnl_tca *) qdisc); +} + +static int cbq_class_msg_parser(struct rtnl_class *class) +{ + return cbq_msg_parser((struct rtnl_tca *) class); +} + +static void cbq_qdisc_free_data(struct rtnl_qdisc *qdisc) +{ + free(qdisc->q_subdata); +} + +static int cbq_clone(struct rtnl_tca *_dst, struct rtnl_tca *_src) +{ + struct rtnl_cbq *src = cbq_qdisc(_src); + + if (src && !cbq_alloc(_dst)) + return nl_errno(ENOMEM); + else + return 0; +} + +static int cbq_qdisc_clone(struct rtnl_qdisc *dst, struct rtnl_qdisc *src) +{ + return cbq_clone((struct rtnl_tca *) dst, (struct rtnl_tca *) src); +} + +static void cbq_class_free_data(struct rtnl_class *class) +{ + free(class->c_subdata); +} + +static int cbq_class_clone(struct rtnl_class *dst, struct rtnl_class *src) +{ + return cbq_clone((struct rtnl_tca *) dst, (struct rtnl_tca *) src); +} + +static int cbq_dump_brief(struct rtnl_tca *tca, struct nl_dump_params *p, + int line) +{ + struct rtnl_cbq *cbq; + double r, rbit; + char *ru, *rubit; + + cbq = cbq_qdisc(tca); + if (!cbq) + goto ignore; + + r = nl_cancel_down_bytes(cbq->cbq_rate.rate, &ru); + rbit = nl_cancel_down_bits(cbq->cbq_rate.rate * 8, &rubit); + + dp_dump(p, " rate %.2f%s/s (%.0f%s) prio %u", + r, ru, rbit, rubit, cbq->cbq_wrr.priority); + +ignore: + return line; +} + +static int cbq_qdisc_dump_brief(struct rtnl_qdisc *qdisc, + struct nl_dump_params *p, int line) +{ + return cbq_dump_brief((struct rtnl_tca *) qdisc, p, line); +} + +static int cbq_class_dump_brief(struct rtnl_class *class, + struct nl_dump_params *p, int line) +{ + return cbq_dump_brief((struct rtnl_tca *) class, p, line); +} + +static int cbq_dump_full(struct rtnl_tca *tca, struct nl_dump_params *p, + int line) +{ + struct rtnl_cbq *cbq; + char *unit, buf[32]; + double w; + uint32_t el; + + cbq = cbq_qdisc(tca); + if (!cbq) + goto ignore; + + w = nl_cancel_down_bits(cbq->cbq_wrr.weight * 8, &unit); + + dp_dump(p, "avgpkt %u mpu %u cell %u allot %u weight %.0f%s\n", + cbq->cbq_lss.avpkt, + cbq->cbq_rate.mpu, + 1 << cbq->cbq_rate.cell_log, + cbq->cbq_wrr.allot, w, unit); + + el = cbq->cbq_lss.ewma_log; + dp_dump_line(p, line++, " minidle %uus maxidle %uus offtime " + "%uus level %u ewma_log %u\n", + nl_ticks2us(cbq->cbq_lss.minidle >> el), + nl_ticks2us(cbq->cbq_lss.maxidle >> el), + nl_ticks2us(cbq->cbq_lss.offtime >> el), + cbq->cbq_lss.level, + cbq->cbq_lss.ewma_log); + + dp_dump_line(p, line++, " penalty %uus strategy %s ", + nl_ticks2us(cbq->cbq_ovl.penalty), + nl_ovl_strategy2str(cbq->cbq_ovl.strategy, buf, sizeof(buf))); + + dp_dump(p, "split %s defmap 0x%08x ", + rtnl_tc_handle2str(cbq->cbq_fopt.split, buf, sizeof(buf)), + cbq->cbq_fopt.defmap); + + dp_dump(p, "police %s", + nl_police2str(cbq->cbq_police.police, buf, sizeof(buf))); + +ignore: + return line; +} + +static int cbq_qdisc_dump_full(struct rtnl_qdisc *qdisc, + struct nl_dump_params *p, int line) +{ + return cbq_dump_full((struct rtnl_tca *) qdisc, p, line); +} + +static int cbq_class_dump_full(struct rtnl_class *class, + struct nl_dump_params *p, int line) +{ + return cbq_dump_full((struct rtnl_tca *) class, p, line); +} + +static int cbq_dump_with_stats(struct rtnl_tca *tca, struct nl_dump_params *p, + int line) +{ + struct tc_cbq_xstats *x = tca_xstats(tca); + + if (!x) + goto ignore; + + dp_dump_line(p, line++, " borrows overact " + " avgidle undertime\n"); + dp_dump_line(p, line++, " %10u %10u %10u %10u\n", + x->borrows, x->overactions, x->avgidle, x->undertime); + +ignore: + return line; +} + +static int cbq_qdisc_dump_with_stats(struct rtnl_qdisc *qdisc, + struct nl_dump_params *p, int line) +{ + return cbq_dump_with_stats((struct rtnl_tca *) qdisc, p, line); +} + +static int cbq_class_dump_with_stats(struct rtnl_class *class, + struct nl_dump_params *p, int line) +{ + return cbq_dump_with_stats((struct rtnl_tca *) class, p, line); +} + +static struct rtnl_qdisc_ops cbq_qdisc_ops = { + .qo_kind = "cbq", + .qo_msg_parser = cbq_qdisc_msg_parser, + .qo_free_data = cbq_qdisc_free_data, + .qo_clone = cbq_qdisc_clone, + .qo_dump[NL_DUMP_BRIEF] = cbq_qdisc_dump_brief, + .qo_dump[NL_DUMP_FULL] = cbq_qdisc_dump_full, + .qo_dump[NL_DUMP_STATS] = cbq_qdisc_dump_with_stats, +}; + +static struct rtnl_class_ops cbq_class_ops = { + .co_kind = "cbq", + .co_msg_parser = cbq_class_msg_parser, + .co_free_data = cbq_class_free_data, + .co_clone = cbq_class_clone, + .co_dump[NL_DUMP_BRIEF] = cbq_class_dump_brief, + .co_dump[NL_DUMP_FULL] = cbq_class_dump_full, + .co_dump[NL_DUMP_STATS] = cbq_class_dump_with_stats, +}; + +static void __init cbq_init(void) +{ + rtnl_qdisc_register(&cbq_qdisc_ops); + rtnl_class_register(&cbq_class_ops); +} + +static void __exit cbq_exit(void) +{ + rtnl_qdisc_unregister(&cbq_qdisc_ops); + rtnl_class_unregister(&cbq_class_ops); +} + +/** @} */ diff --git a/lib/route/sch/dsmark.c b/lib/route/sch/dsmark.c new file mode 100644 index 0000000..5ba6b92 --- /dev/null +++ b/lib/route/sch/dsmark.c @@ -0,0 +1,459 @@ +/* + * lib/route/sch/dsmark.c DSMARK + * + * 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> + */ + +/** + * @ingroup qdisc_api + * @ingroup class_api + * @defgroup dsmark Differentiated Services Marker (DSMARK) + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/qdisc-modules.h> +#include <netlink/route/class.h> +#include <netlink/route/class-modules.h> +#include <netlink/route/sch/dsmark.h> + +/** @cond SKIP */ +#define SCH_DSMARK_ATTR_INDICES 0x1 +#define SCH_DSMARK_ATTR_DEFAULT_INDEX 0x2 +#define SCH_DSMARK_ATTR_SET_TC_INDEX 0x4 + +#define SCH_DSMARK_ATTR_MASK 0x1 +#define SCH_DSMARK_ATTR_VALUE 0x2 +/** @endcond */ + +static inline struct rtnl_dsmark_qdisc *dsmark_qdisc(struct rtnl_qdisc *qdisc) +{ + return (struct rtnl_dsmark_qdisc *) qdisc->q_subdata; +} + +static inline struct rtnl_dsmark_qdisc * +dsmark_qdisc_alloc(struct rtnl_qdisc *qdisc) +{ + if (!qdisc->q_subdata) + qdisc->q_subdata = calloc(1, sizeof(struct rtnl_dsmark_qdisc)); + + return dsmark_qdisc(qdisc); +} + +static struct nla_policy dsmark_policy[TCA_DSMARK_MAX+1] = { + [TCA_DSMARK_INDICES] = { .type = NLA_U16 }, + [TCA_DSMARK_DEFAULT_INDEX] = { .type = NLA_U16 }, + [TCA_DSMARK_SET_TC_INDEX] = { .type = NLA_FLAG }, + [TCA_DSMARK_VALUE] = { .type = NLA_U8 }, + [TCA_DSMARK_MASK] = { .type = NLA_U8 }, +}; + +static int dsmark_qdisc_msg_parser(struct rtnl_qdisc *qdisc) +{ + int err; + struct nlattr *tb[TCA_DSMARK_MAX + 1]; + struct rtnl_dsmark_qdisc *dsmark; + + err = tca_parse(tb, TCA_DSMARK_MAX, (struct rtnl_tca *) qdisc, + dsmark_policy); + if (err < 0) + return err; + + dsmark = dsmark_qdisc_alloc(qdisc); + if (!dsmark) + return nl_errno(ENOMEM); + + if (tb[TCA_DSMARK_INDICES]) { + dsmark->qdm_indices = nla_get_u16(tb[TCA_DSMARK_INDICES]); + dsmark->qdm_mask |= SCH_DSMARK_ATTR_INDICES; + } + + if (tb[TCA_DSMARK_DEFAULT_INDEX]) { + dsmark->qdm_default_index = + nla_get_u16(tb[TCA_DSMARK_DEFAULT_INDEX]); + dsmark->qdm_mask |= SCH_DSMARK_ATTR_DEFAULT_INDEX; + } + + if (tb[TCA_DSMARK_SET_TC_INDEX]) { + dsmark->qdm_set_tc_index = 1; + dsmark->qdm_mask |= SCH_DSMARK_ATTR_SET_TC_INDEX; + } + + return 0; +} + +static inline struct rtnl_dsmark_class *dsmark_class(struct rtnl_class *class) +{ + return (struct rtnl_dsmark_class *) class->c_subdata; +} + +static inline struct rtnl_dsmark_class * +dsmark_class_alloc(struct rtnl_class *class) +{ + if (!class->c_subdata) + class->c_subdata = calloc(1, sizeof(struct rtnl_dsmark_class)); + + return dsmark_class(class); +} + +static int dsmark_class_msg_parser(struct rtnl_class *class) +{ + int err; + struct nlattr *tb[TCA_DSMARK_MAX + 1]; + struct rtnl_dsmark_class *dsmark; + + err = tca_parse(tb, TCA_DSMARK_MAX, (struct rtnl_tca *) class, + dsmark_policy); + if (err < 0) + return err; + + dsmark = dsmark_class_alloc(class); + if (!dsmark) + return nl_errno(ENOMEM); + + if (tb[TCA_DSMARK_MASK]) { + dsmark->cdm_bmask = nla_get_u8(tb[TCA_DSMARK_MASK]); + dsmark->cdm_mask |= SCH_DSMARK_ATTR_MASK; + } + + if (tb[TCA_DSMARK_VALUE]) { + dsmark->cdm_value = nla_get_u8(tb[TCA_DSMARK_VALUE]); + dsmark->cdm_mask |= SCH_DSMARK_ATTR_VALUE; + } + + return 0; +} + +static int dsmark_qdisc_dump_brief(struct rtnl_qdisc *qdisc, + struct nl_dump_params *p, int line) +{ + struct rtnl_dsmark_qdisc *dsmark = dsmark_qdisc(qdisc); + + if (dsmark && (dsmark->qdm_mask & SCH_DSMARK_ATTR_INDICES)) + dp_dump(p, " indices 0x%04x", dsmark->qdm_indices); + + return line; +} + +static int dsmark_qdisc_dump_full(struct rtnl_qdisc *qdisc, + struct nl_dump_params *p, int line) +{ + struct rtnl_dsmark_qdisc *dsmark = dsmark_qdisc(qdisc); + + if (!dsmark) + goto ignore; + + if (dsmark->qdm_mask & SCH_DSMARK_ATTR_DEFAULT_INDEX) + dp_dump(p, " default index 0x%04x", dsmark->qdm_default_index); + + if (dsmark->qdm_mask & SCH_DSMARK_ATTR_SET_TC_INDEX) + dp_dump(p, " set-tc-index"); + +ignore: + return line; +} + +static int dsmark_class_dump_brief(struct rtnl_class *class, + struct nl_dump_params *p, int line) +{ + struct rtnl_dsmark_class *dsmark = dsmark_class(class); + + if (!dsmark) + goto ignore; + + if (dsmark->cdm_mask & SCH_DSMARK_ATTR_VALUE) + dp_dump(p, " value 0x%02x", dsmark->cdm_value); + + if (dsmark->cdm_mask & SCH_DSMARK_ATTR_MASK) + dp_dump(p, " mask 0x%02x", dsmark->cdm_bmask); + +ignore: + return line; +} + +static struct nl_msg *dsmark_qdisc_get_opts(struct rtnl_qdisc *qdisc) +{ + struct rtnl_dsmark_qdisc *dsmark = dsmark_qdisc(qdisc); + struct nl_msg *msg; + + if (!dsmark) + return NULL; + + msg = nlmsg_alloc(); + if (!msg) + goto nla_put_failure; + + if (dsmark->qdm_mask & SCH_DSMARK_ATTR_INDICES) + NLA_PUT_U16(msg, TCA_DSMARK_INDICES, dsmark->qdm_indices); + + if (dsmark->qdm_mask & SCH_DSMARK_ATTR_DEFAULT_INDEX) + NLA_PUT_U16(msg, TCA_DSMARK_DEFAULT_INDEX, + dsmark->qdm_default_index); + + if (dsmark->qdm_mask & SCH_DSMARK_ATTR_SET_TC_INDEX) + NLA_PUT_FLAG(msg, TCA_DSMARK_SET_TC_INDEX); + + return msg; + +nla_put_failure: + nlmsg_free(msg); + return NULL; +} + +static struct nl_msg *dsmark_class_get_opts(struct rtnl_class *class) +{ + struct rtnl_dsmark_class *dsmark = dsmark_class(class); + struct nl_msg *msg; + + if (!dsmark) + return NULL; + + msg = nlmsg_alloc(); + if (!msg) + goto nla_put_failure; + + if (dsmark->cdm_mask & SCH_DSMARK_ATTR_MASK) + NLA_PUT_U8(msg, TCA_DSMARK_MASK, dsmark->cdm_bmask); + + if (dsmark->cdm_mask & SCH_DSMARK_ATTR_VALUE) + NLA_PUT_U8(msg, TCA_DSMARK_VALUE, dsmark->cdm_value); + + return msg; + +nla_put_failure: + nlmsg_free(msg); + return NULL; +} + +/** + * @name Class Attribute Access + * @{ + */ + +/** + * Set bitmask of DSMARK class. + * @arg class DSMARK class to be modified. + * @arg mask New bitmask. + * @return 0 on success or a negative error code. + */ +int rtnl_class_dsmark_set_bitmask(struct rtnl_class *class, uint8_t mask) +{ + struct rtnl_dsmark_class *dsmark; + + dsmark = dsmark_class(class); + if (!dsmark) + return nl_errno(ENOMEM); + + dsmark->cdm_bmask = mask; + dsmark->cdm_mask |= SCH_DSMARK_ATTR_MASK; + + return 0; +} + +/** + * Get bitmask of DSMARK class. + * @arg class DSMARK class. + * @return Bitmask or a negative error code. + */ +int rtnl_class_dsmark_get_bitmask(struct rtnl_class *class) +{ + struct rtnl_dsmark_class *dsmark; + + dsmark = dsmark_class(class); + if (dsmark && dsmark->cdm_mask & SCH_DSMARK_ATTR_MASK) + return dsmark->cdm_bmask; + else + return nl_errno(ENOENT); +} + +/** + * Set value of DSMARK class. + * @arg class DSMARK class to be modified. + * @arg value New value. + * @return 0 on success or a negative errror code. + */ +int rtnl_class_dsmark_set_value(struct rtnl_class *class, uint8_t value) +{ + struct rtnl_dsmark_class *dsmark; + + dsmark = dsmark_class(class); + if (!dsmark) + return nl_errno(ENOMEM); + + dsmark->cdm_value = value; + dsmark->cdm_mask |= SCH_DSMARK_ATTR_VALUE; + + return 0; +} + +/** + * Get value of DSMARK class. + * @arg class DSMARK class. + * @return Value or a negative error code. + */ +int rtnl_class_dsmark_get_value(struct rtnl_class *class) +{ + struct rtnl_dsmark_class *dsmark; + + dsmark = dsmark_class(class); + if (dsmark && dsmark->cdm_mask & SCH_DSMARK_ATTR_VALUE) + return dsmark->cdm_value; + else + return nl_errno(ENOENT); +} + +/** @} */ + +/** + * @name Qdisc Attribute Access + * @{ + */ + +/** + * Set indices of DSMARK qdisc. + * @arg qdisc DSMARK qdisc to be modified. + * @arg indices New indices. + */ +int rtnl_qdisc_dsmark_set_indices(struct rtnl_qdisc *qdisc, uint16_t indices) +{ + struct rtnl_dsmark_qdisc *dsmark; + + dsmark = dsmark_qdisc(qdisc); + if (!dsmark) + return nl_errno(ENOMEM); + + dsmark->qdm_indices = indices; + dsmark->qdm_mask |= SCH_DSMARK_ATTR_INDICES; + + return 0; +} + +/** + * Get indices of DSMARK qdisc. + * @arg qdisc DSMARK qdisc. + * @return Indices or a negative error code. + */ +int rtnl_qdisc_dsmark_get_indices(struct rtnl_qdisc *qdisc) +{ + struct rtnl_dsmark_qdisc *dsmark; + + dsmark = dsmark_qdisc(qdisc); + if (dsmark && dsmark->qdm_mask & SCH_DSMARK_ATTR_INDICES) + return dsmark->qdm_indices; + else + return nl_errno(ENOENT); +} + +/** + * Set default index of DSMARK qdisc. + * @arg qdisc DSMARK qdisc to be modified. + * @arg default_index New default index. + * @return 0 on success or a negative error code. + */ +int rtnl_qdisc_dsmark_set_default_index(struct rtnl_qdisc *qdisc, + uint16_t default_index) +{ + struct rtnl_dsmark_qdisc *dsmark; + + dsmark = dsmark_qdisc(qdisc); + if (!dsmark) + return nl_errno(ENOMEM); + + dsmark->qdm_default_index = default_index; + dsmark->qdm_mask |= SCH_DSMARK_ATTR_DEFAULT_INDEX; + + return 0; +} + +/** + * Get default index of DSMARK qdisc. + * @arg qdisc DSMARK qdisc. + * @return Default index or a negative error code. + */ +int rtnl_qdisc_dsmark_get_default_index(struct rtnl_qdisc *qdisc) +{ + struct rtnl_dsmark_qdisc *dsmark; + + dsmark = dsmark_qdisc(qdisc); + if (dsmark && dsmark->qdm_mask & SCH_DSMARK_ATTR_DEFAULT_INDEX) + return dsmark->qdm_default_index; + else + return nl_errno(ENOENT); +} + +/** + * Set set-tc-index flag of DSMARK qdisc. + * @arg qdisc DSMARK qdisc to be modified. + * @arg flag Flag indicating whether to enable or disable. + * @return 0 on success or a negative error code. + */ +int rtnl_qdisc_dsmark_set_set_tc_index(struct rtnl_qdisc *qdisc, int flag) +{ + struct rtnl_dsmark_qdisc *dsmark; + + dsmark = dsmark_qdisc(qdisc); + if (!dsmark) + return nl_errno(ENOMEM); + + dsmark->qdm_set_tc_index = !!flag; + dsmark->qdm_mask |= SCH_DSMARK_ATTR_SET_TC_INDEX; + + return 0; +} + +/** + * Get set-tc-index flag of DSMARK qdisc. + * @arg qdisc DSMARK qdisc to be modified. + * @return 1 or 0 to indicate wehther the flag is enabled or a negative + * error code. + */ +int rtnl_qdisc_dsmark_get_set_tc_index(struct rtnl_qdisc *qdisc) +{ + struct rtnl_dsmark_qdisc *dsmark; + + dsmark = dsmark_qdisc(qdisc); + if (dsmark && dsmark->qdm_mask & SCH_DSMARK_ATTR_SET_TC_INDEX) + return dsmark->qdm_set_tc_index; + else + return nl_errno(ENOENT); +} + +/** @} */ + +static struct rtnl_qdisc_ops dsmark_qdisc_ops = { + .qo_kind = "dsmark", + .qo_msg_parser = dsmark_qdisc_msg_parser, + .qo_dump[NL_DUMP_BRIEF] = dsmark_qdisc_dump_brief, + .qo_dump[NL_DUMP_FULL] = dsmark_qdisc_dump_full, + .qo_get_opts = dsmark_qdisc_get_opts, +}; + +static struct rtnl_class_ops dsmark_class_ops = { + .co_kind = "dsmark", + .co_msg_parser = dsmark_class_msg_parser, + .co_dump[NL_DUMP_BRIEF] = dsmark_class_dump_brief, + .co_get_opts = dsmark_class_get_opts, +}; + +static void __init dsmark_init(void) +{ + rtnl_qdisc_register(&dsmark_qdisc_ops); + rtnl_class_register(&dsmark_class_ops); +} + +static void __exit dsmark_exit(void) +{ + rtnl_qdisc_unregister(&dsmark_qdisc_ops); + rtnl_class_unregister(&dsmark_class_ops); +} + +/** @} */ diff --git a/lib/route/sch/fifo.c b/lib/route/sch/fifo.c new file mode 100644 index 0000000..4f8d202 --- /dev/null +++ b/lib/route/sch/fifo.c @@ -0,0 +1,205 @@ +/* + * lib/route/sch/fifo.c (p|b)fifo + * + * 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> + */ + +/** + * @ingroup qdisc_api + * @defgroup fifo Packet/Bytes FIFO (pfifo/bfifo) + * @brief + * + * The FIFO qdisc comes in two flavours: + * @par bfifo (Byte FIFO) + * Allows enqueuing until the currently queued volume in bytes exceeds + * the configured limit.backlog contains currently enqueued volume in bytes. + * + * @par pfifo (Packet FIFO) + * Allows enquueing until the currently queued number of packets + * exceeds the configured limit. + * + * The configuration is exactly the same, the decision which of + * the two variations is going to be used is made based on the + * kind of the qdisc (rtnl_qdisc_set_kind()). + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/qdisc-modules.h> +#include <netlink/route/sch/fifo.h> +#include <netlink/utils.h> + +/** @cond SKIP */ +#define SCH_FIFO_ATTR_LIMIT 1 +/** @endcond */ + +static inline struct rtnl_fifo *fifo_qdisc(struct rtnl_qdisc *qdisc) +{ + return (struct rtnl_fifo *) qdisc->q_subdata; +} + +static inline struct rtnl_fifo *fifo_alloc(struct rtnl_qdisc *qdisc) +{ + if (!qdisc->q_subdata) + qdisc->q_subdata = calloc(1, sizeof(struct rtnl_fifo)); + + return fifo_qdisc(qdisc); +} + +static int fifo_msg_parser(struct rtnl_qdisc *qdisc) +{ + struct rtnl_fifo *fifo; + struct tc_fifo_qopt *opt; + + if (qdisc->q_opts->d_size < sizeof(struct tc_fifo_qopt)) + return nl_error(EINVAL, "FIFO options size mismatch"); + + fifo = fifo_alloc(qdisc); + if (!fifo) + return nl_errno(ENOMEM); + + opt = (struct tc_fifo_qopt *) qdisc->q_opts->d_data; + fifo->qf_limit = opt->limit; + fifo->qf_mask = SCH_FIFO_ATTR_LIMIT; + + return 0; +} + +static void fifo_free_data(struct rtnl_qdisc *qdisc) +{ + free(qdisc->q_subdata); +} + +static int pfifo_dump_brief(struct rtnl_qdisc *qdisc, + struct nl_dump_params *p, int line) +{ + struct rtnl_fifo *fifo = fifo_qdisc(qdisc); + + if (fifo) + dp_dump(p, " limit %u packets", fifo->qf_limit); + + return line; +} + +static int bfifo_dump_brief(struct rtnl_qdisc *qdisc, + struct nl_dump_params *p, int line) +{ + struct rtnl_fifo *fifo = fifo_qdisc(qdisc); + + if (fifo) { + char *unit; + double r; + + r = nl_cancel_down_bytes(fifo->qf_limit, &unit); + dp_dump(p, " limit %.1f%s", r, unit); + } + + return line; +} + +static struct nl_msg *fifo_get_opts(struct rtnl_qdisc *qdisc) +{ + struct rtnl_fifo *fifo; + struct tc_fifo_qopt opts; + struct nl_msg *msg; + + fifo = fifo_qdisc(qdisc); + if (!fifo || !(fifo->qf_mask & SCH_FIFO_ATTR_LIMIT)) + return NULL; + + msg = nlmsg_alloc(); + if (!msg) + goto errout; + + memset(&opts, 0, sizeof(opts)); + opts.limit = fifo->qf_limit; + + if (nlmsg_append(msg, &opts, sizeof(opts), NL_DONTPAD) < 0) + goto errout; + + return msg; +errout: + nlmsg_free(msg); + return NULL; +} + +/** + * @name Attribute Modification + * @{ + */ + +/** + * Set limit of FIFO qdisc. + * @arg qdisc FIFO qdisc to be modified. + * @arg limit New limit. + * @return 0 on success or a negative error code. + */ +int rtnl_qdisc_fifo_set_limit(struct rtnl_qdisc *qdisc, int limit) +{ + struct rtnl_fifo *fifo; + + fifo = fifo_alloc(qdisc); + if (!fifo) + return nl_errno(ENOMEM); + + fifo->qf_limit = limit; + fifo->qf_mask |= SCH_FIFO_ATTR_LIMIT; + + return 0; +} + +/** + * Get limit of a FIFO qdisc. + * @arg qdisc FIFO qdisc. + * @return Numeric limit or a negative error code. + */ +int rtnl_qdisc_fifo_get_limit(struct rtnl_qdisc *qdisc) +{ + struct rtnl_fifo *fifo; + + fifo = fifo_qdisc(qdisc); + if (fifo && fifo->qf_mask & SCH_FIFO_ATTR_LIMIT) + return fifo->qf_limit; + else + return nl_errno(ENOMEM); +} + +/** @} */ + +static struct rtnl_qdisc_ops pfifo_ops = { + .qo_kind = "pfifo", + .qo_msg_parser = fifo_msg_parser, + .qo_free_data = fifo_free_data, + .qo_dump[NL_DUMP_BRIEF] = pfifo_dump_brief, + .qo_get_opts = fifo_get_opts, +}; + +static struct rtnl_qdisc_ops bfifo_ops = { + .qo_kind = "bfifo", + .qo_msg_parser = fifo_msg_parser, + .qo_free_data = fifo_free_data, + .qo_dump[NL_DUMP_BRIEF] = bfifo_dump_brief, + .qo_get_opts = fifo_get_opts, +}; + +static void __init fifo_init(void) +{ + rtnl_qdisc_register(&pfifo_ops); + rtnl_qdisc_register(&bfifo_ops); +} + +static void __exit fifo_exit(void) +{ + rtnl_qdisc_unregister(&pfifo_ops); + rtnl_qdisc_unregister(&bfifo_ops); +} + +/** @} */ diff --git a/lib/route/sch/htb.c b/lib/route/sch/htb.c new file mode 100644 index 0000000..6de87b3 --- /dev/null +++ b/lib/route/sch/htb.c @@ -0,0 +1,553 @@ +/* + * lib/route/sch/htb.c HTB Qdisc + * + * 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> + * Copyright (c) 2005-2006 Petr Gotthard <petr.gotthard@siemens.com> + * Copyright (c) 2005-2006 Siemens AG Oesterreich + */ + +/** + * @ingroup qdisc_api + * @ingroup class_api + * @defgroup htb Hierachical Token Bucket (HTB) + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/cache.h> +#include <netlink/utils.h> +#include <netlink/route/tc.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/qdisc-modules.h> +#include <netlink/route/class.h> +#include <netlink/route/class-modules.h> +#include <netlink/route/link.h> +#include <netlink/route/sch/htb.h> + +/** @cond SKIP */ +#define SCH_HTB_HAS_RATE2QUANTUM 0x01 +#define SCH_HTB_HAS_DEFCLS 0x02 + +#define SCH_HTB_HAS_PRIO 0x001 +#define SCH_HTB_HAS_MTU 0x002 +#define SCH_HTB_HAS_RATE 0x004 +#define SCH_HTB_HAS_CEIL 0x008 +#define SCH_HTB_HAS_RBUFFER 0x010 +#define SCH_HTB_HAS_CBUFFER 0x020 +#define SCH_HTB_HAS_QUANTUM 0x040 +#define SCH_HTB_HAS_OVERHEAD 0x080 +#define SCH_HTB_HAS_MPU 0x100 +/** @endcond */ + +static inline struct rtnl_htb_qdisc *htb_qdisc(struct rtnl_qdisc *qdisc) +{ + if (qdisc->q_subdata == NULL) + qdisc->q_subdata = calloc(1, sizeof(struct rtnl_htb_qdisc)); + + return (struct rtnl_htb_qdisc *) qdisc->q_subdata; +} + +static struct nla_policy htb_policy[TCA_HTB_MAX+1] = { + [TCA_HTB_INIT] = { .minlen = sizeof(struct tc_htb_glob) }, + [TCA_HTB_PARMS] = { .minlen = sizeof(struct tc_htb_opt) }, +}; + +static int htb_qdisc_msg_parser(struct rtnl_qdisc *qdisc) +{ + int err; + struct nlattr *tb[TCA_HTB_MAX + 1]; + struct rtnl_htb_qdisc *d; + + err = tca_parse(tb, TCA_HTB_MAX, (struct rtnl_tca *) qdisc, htb_policy); + if (err < 0) + return err; + + d = htb_qdisc(qdisc); + + if (tb[TCA_HTB_INIT]) { + struct tc_htb_glob opts; + + nla_memcpy(&opts, tb[TCA_HTB_INIT], sizeof(opts)); + d->qh_rate2quantum = opts.rate2quantum; + d->qh_defcls = opts.defcls; + + d->qh_mask = (SCH_HTB_HAS_RATE2QUANTUM | SCH_HTB_HAS_DEFCLS); + } + + return 0; +} + +static void htb_qdisc_free_data(struct rtnl_qdisc *qdisc) +{ + free(qdisc->q_subdata); +} + +static inline struct rtnl_htb_class *htb_class(struct rtnl_class *class) +{ + if (class->c_subdata == NULL) + class->c_subdata = calloc(1, sizeof(struct rtnl_htb_class)); + + return (struct rtnl_htb_class *) class->c_subdata; +} + +static int htb_class_msg_parser(struct rtnl_class *class) +{ + int err; + struct nlattr *tb[TCA_HTB_MAX + 1]; + struct rtnl_htb_class *d; + + err = tca_parse(tb, TCA_HTB_MAX, (struct rtnl_tca *) class, htb_policy); + if (err < 0) + return err; + + d = htb_class(class); + + if (tb[TCA_HTB_PARMS]) { + struct tc_htb_opt opts; + + nla_memcpy(&opts, tb[TCA_HTB_PARMS], sizeof(opts)); + d->ch_prio = opts.prio; + rtnl_copy_ratespec(&d->ch_rate, &opts.rate); + rtnl_copy_ratespec(&d->ch_ceil, &opts.ceil); + d->ch_rbuffer = rtnl_tc_calc_bufsize(opts.buffer, opts.rate.rate); + d->ch_cbuffer = rtnl_tc_calc_bufsize(opts.cbuffer, opts.ceil.rate); + d->ch_quantum = opts.quantum; + d->ch_overhead = (opts.rate.mpu >> 8) & 0xff; + d->ch_mpu = opts.rate.mpu & 0xff; + + d->ch_mask = (SCH_HTB_HAS_PRIO | SCH_HTB_HAS_RATE | + SCH_HTB_HAS_CEIL | SCH_HTB_HAS_RBUFFER | + SCH_HTB_HAS_CBUFFER | SCH_HTB_HAS_QUANTUM | + SCH_HTB_HAS_OVERHEAD | SCH_HTB_HAS_MPU); + } + + return 0; +} + +static void htb_class_free_data(struct rtnl_class *class) +{ + free(class->c_subdata); +} + +static int htb_qdisc_dump_brief(struct rtnl_qdisc *qdisc, + struct nl_dump_params *p, int line) +{ + struct rtnl_htb_qdisc *d = (struct rtnl_htb_qdisc *) qdisc->q_subdata; + + if (d == NULL) + goto ignore; + + if (d->qh_mask & SCH_HTB_HAS_RATE2QUANTUM) + dp_dump(p, " r2q %u", d->qh_rate2quantum); + + if (d->qh_mask & SCH_HTB_HAS_DEFCLS) { + char buf[32]; + dp_dump(p, " default %s", + rtnl_tc_handle2str(d->qh_defcls, buf, sizeof(buf))); + } + +ignore: + return line; +} + +static int htb_class_dump_brief(struct rtnl_class *class, + struct nl_dump_params *p, int line) +{ + struct rtnl_htb_class *d = (struct rtnl_htb_class *) class->c_subdata; + + if (d == NULL) + goto ignore; + + if (d->ch_mask & SCH_HTB_HAS_RATE) { + double r, rbit; + char *ru, *rubit; + + r = nl_cancel_down_bytes(d->ch_rate.rs_rate, &ru); + rbit = nl_cancel_down_bits(d->ch_rate.rs_rate*8, &rubit); + + dp_dump(p, " rate %.2f%s/s (%.0f%s) log %u", + r, ru, rbit, rubit, 1<<d->ch_rate.rs_cell_log); + } + +ignore: + return line; +} + +static int htb_class_dump_full(struct rtnl_class *class, + struct nl_dump_params *p, int line) +{ + struct rtnl_htb_class *d = (struct rtnl_htb_class *) class->c_subdata; + + if (d == NULL) + goto ignore; + + /* line 1 */ + if (d->ch_mask & SCH_HTB_HAS_CEIL) { + double r, rbit; + char *ru, *rubit; + + r = nl_cancel_down_bytes(d->ch_ceil.rs_rate, &ru); + rbit = nl_cancel_down_bits(d->ch_ceil.rs_rate*8, &rubit); + + dp_dump(p, " ceil %.2f%s/s (%.0f%s) log %u", + r, ru, rbit, rubit, 1<<d->ch_ceil.rs_cell_log); + } + + if (d->ch_mask & SCH_HTB_HAS_PRIO) + dp_dump(p, " prio %u", d->ch_prio); + + if (d->ch_mask & SCH_HTB_HAS_MTU) + dp_dump(p, " mtu %u", d->ch_mtu); + + if (d->ch_mask & SCH_HTB_HAS_RBUFFER) { + double b; + char *bu; + + b = nl_cancel_down_bytes(d->ch_rbuffer, &bu); + dp_dump(p, " rbuffer %.2f%s", b, bu); + } + + if (d->ch_mask & SCH_HTB_HAS_CBUFFER) { + double b; + char *bu; + + b = nl_cancel_down_bytes(d->ch_cbuffer, &bu); + dp_dump(p, " cbuffer %.2f%s", b, bu); + } + + if (d->ch_mask & SCH_HTB_HAS_QUANTUM) + dp_dump(p, " quantum %u", d->ch_quantum); + + if (d->ch_mask & SCH_HTB_HAS_OVERHEAD) + dp_dump(p, " overhead %u", d->ch_overhead); + + if (d->ch_mask & SCH_HTB_HAS_MPU) + dp_dump(p, " mpu %u", d->ch_mpu); + +ignore: + return line; +} + +static struct nl_msg *htb_qdisc_get_opts(struct rtnl_qdisc *qdisc) +{ + struct rtnl_htb_qdisc *d = (struct rtnl_htb_qdisc *) qdisc->q_subdata; + struct tc_htb_glob opts; + struct nl_msg *msg; + + if (d == NULL) + return NULL; + + msg = nlmsg_alloc(); + if (msg == NULL) + return NULL; + + memset(&opts, 0, sizeof(opts)); + opts.version = TC_HTB_PROTOVER; + + if (d->qh_mask & SCH_HTB_HAS_RATE2QUANTUM) + opts.rate2quantum = d->qh_rate2quantum; + if (d->qh_mask & SCH_HTB_HAS_DEFCLS) + opts.defcls = d->qh_defcls; + + nla_put(msg, TCA_HTB_INIT, sizeof(opts), &opts); + + return msg; +} + +static uint8_t compute_cell(uint32_t rate, uint32_t mtu) +{ + uint8_t cell_log = 0; + while (mtu > 255) { + mtu >>= 1; + cell_log++; + } + + return cell_log; +} + +static struct nl_msg *htb_class_get_opts(struct rtnl_class *class) +{ + struct rtnl_htb_class *d = (struct rtnl_htb_class *) class->c_subdata; + uint32_t mtu, rtable[RTNL_TC_RTABLE_SIZE], ctable[RTNL_TC_RTABLE_SIZE]; + struct tc_htb_opt opts; + struct nl_msg *msg; + int buffer, cbuffer; + uint8_t overhead = 0, mpu = 0; + + if (d == NULL) + return NULL; + + msg = nlmsg_alloc(); + memset(&opts, 0, sizeof(opts)); + + /* if not set, zero (0) is used as priority */ + if (d->ch_mask & SCH_HTB_HAS_PRIO) + opts.prio = d->ch_prio; + + if (d->ch_mask & SCH_HTB_HAS_MTU) + mtu = d->ch_mtu; + else + mtu = 1600; /* eth packet len */ + + if (!(d->ch_mask & SCH_HTB_HAS_RATE)) + BUG(); + + rtnl_rcopy_ratespec(&opts.rate, &d->ch_rate); + /* if cell_log not set, compute default value */ + if (opts.rate.cell_log == UINT8_MAX) + opts.rate.cell_log = compute_cell(opts.rate.rate, mtu); + + /* if not set, configured rate is used as ceil, which implies no borrowing */ + if (d->ch_mask & SCH_HTB_HAS_CEIL) + rtnl_rcopy_ratespec(&opts.ceil, &d->ch_ceil); + else + memcpy(&opts.ceil, &opts.rate, sizeof(struct tc_ratespec)); + /* if cell_log not set, compute default value */ + if (opts.ceil.cell_log == UINT8_MAX) + opts.ceil.cell_log = compute_cell(opts.ceil.rate, mtu); + + if (d->ch_mask & SCH_HTB_HAS_RBUFFER) + buffer = d->ch_rbuffer; + else + buffer = opts.rate.rate / nl_get_hz() + mtu; + + opts.buffer = rtnl_tc_calc_txtime(buffer, opts.rate.rate); + + if (d->ch_mask & SCH_HTB_HAS_CBUFFER) + cbuffer = d->ch_cbuffer; + else + cbuffer = opts.ceil.rate / nl_get_hz() + mtu; + + opts.cbuffer = rtnl_tc_calc_txtime(cbuffer, opts.ceil.rate); + + if (d->ch_mask & SCH_HTB_HAS_QUANTUM) + opts.quantum = d->ch_quantum; + + if (d->ch_mask & SCH_HTB_HAS_OVERHEAD) + overhead = d->ch_overhead; + + if (d->ch_mask & SCH_HTB_HAS_MPU) + mpu = d->ch_mpu; + + opts.rate.mpu = mpu | (overhead << 8); + opts.ceil.mpu = mpu | (overhead << 8); + + nla_put(msg, TCA_HTB_PARMS, sizeof(opts), &opts); + + rtnl_tc_build_rate_table(rtable, mpu, overhead, + 1 << opts.rate.cell_log, + opts.rate.rate); + nla_put(msg, TCA_HTB_RTAB, sizeof(rtable), &rtable); + + rtnl_tc_build_rate_table(ctable, mpu, overhead, + 1 << opts.ceil.cell_log, + opts.ceil.rate); + nla_put(msg, TCA_HTB_CTAB, sizeof(ctable), &ctable); + + return msg; +} + +/** + * @name Attribute Modifications + * @{ + */ + +void rtnl_htb_set_rate2quantum(struct rtnl_qdisc *qdisc, uint32_t rate2quantum) +{ + struct rtnl_htb_qdisc *d = htb_qdisc(qdisc); + if (d == NULL) + return; + + d->qh_rate2quantum = rate2quantum; + d->qh_mask |= SCH_HTB_HAS_RATE2QUANTUM; +} + +/** + * Set default class of the htb qdisc to the specified value + * @arg qdisc qdisc to change + * @arg defcls new default class + */ +void rtnl_htb_set_defcls(struct rtnl_qdisc *qdisc, uint32_t defcls) +{ + struct rtnl_htb_qdisc *d = htb_qdisc(qdisc); + if (d == NULL) + return; + + d->qh_defcls = defcls; + d->qh_mask |= SCH_HTB_HAS_DEFCLS; +} + +void rtnl_htb_set_prio(struct rtnl_class *class, uint32_t prio) +{ + struct rtnl_htb_class *d = htb_class(class); + if (d == NULL) + return; + + d->ch_prio = prio; + d->ch_mask |= SCH_HTB_HAS_PRIO; +} + +/** + * Set MTU of the data link. + * @arg class HTB class to be modified. + * @arg mtu New MTU in bytes. + * + * Sets MTU of the data link controlled by the HTB class. + * If not set, the Ethernet MTU (1600) is used. + */ +void rtnl_htb_set_mtu(struct rtnl_class *class, uint32_t mtu) +{ + struct rtnl_htb_class *d = htb_class(class); + if (d == NULL) + return; + + d->ch_mtu = mtu; + d->ch_mask |= SCH_HTB_HAS_MTU; +} + +/** + * Set rate of HTB class. + * @arg class HTB class to be modified. + * @arg rate New rate in bytes per second. + */ +void rtnl_htb_set_rate(struct rtnl_class *class, uint32_t rate) +{ + struct rtnl_htb_class *d = htb_class(class); + if (d == NULL) + return; + + d->ch_rate.rs_cell_log = UINT8_MAX; /* use default value */ + d->ch_rate.rs_rate = rate; + d->ch_mask |= SCH_HTB_HAS_RATE; +} + +/** + * Set ceil of HTB class. + * @arg class HTB class to be modified. + * @arg ceil New ceil in bytes per second. + */ +void rtnl_htb_set_ceil(struct rtnl_class *class, uint32_t ceil) +{ + struct rtnl_htb_class *d = htb_class(class); + if (d == NULL) + return; + + d->ch_ceil.rs_cell_log = UINT8_MAX; /* use default value */ + d->ch_ceil.rs_rate = ceil; + d->ch_mask |= SCH_HTB_HAS_CEIL; +} + +/** + * Set size of the rate bucket of HTB class. + * @arg class HTB class to be modified. + * @arg rbuffer New size in bytes. + */ +void rtnl_htb_set_rbuffer(struct rtnl_class *class, uint32_t rbuffer) +{ + struct rtnl_htb_class *d = htb_class(class); + if (d == NULL) + return; + + d->ch_rbuffer = rbuffer; + d->ch_mask |= SCH_HTB_HAS_RBUFFER; +} + +/** + * Set size of the ceil bucket of HTB class. + * @arg class HTB class to be modified. + * @arg cbuffer New size in bytes. + */ +void rtnl_htb_set_cbuffer(struct rtnl_class *class, uint32_t cbuffer) +{ + struct rtnl_htb_class *d = htb_class(class); + if (d == NULL) + return; + + d->ch_cbuffer = cbuffer; + d->ch_mask |= SCH_HTB_HAS_CBUFFER; +} + +/** + * Set how much bytes to serve from leaf at once of HTB class {use r2q}. + * @arg class HTB class to be modified. + * @arg quantum New size in bytes. + */ +void rtnl_htb_set_quantum(struct rtnl_class *class, uint32_t quantum) +{ + struct rtnl_htb_class *d = htb_class(class); + if (d == NULL) + return; + + d->ch_quantum = quantum; + d->ch_mask |= SCH_HTB_HAS_QUANTUM; +} + +/** + * Set per-packet size overhead used in rate computations of HTB class. + * @arg class HTB class to be modified. + * @arg overhead Size in bytes. + */ +void rtnl_htb_set_overhead(struct rtnl_class *class, uint8_t overhead) +{ + struct rtnl_htb_class *d = htb_class(class); + if (d == NULL) + return; + + d->ch_overhead = overhead; + d->ch_mask |= SCH_HTB_HAS_OVERHEAD; +} + +/** + * Set the minimum packet size used in rate computations of HTB class. + * @arg class HTB class to be modified. + * @arg mpu Size in bytes. + */ +void rtnl_htb_set_mpu(struct rtnl_class *class, uint8_t mpu) +{ + struct rtnl_htb_class *d = htb_class(class); + if (d == NULL) + return; + + d->ch_mpu = mpu; + d->ch_mask |= SCH_HTB_HAS_MPU; +} + +/** @} */ + +static struct rtnl_qdisc_ops htb_qdisc_ops = { + .qo_kind = "htb", + .qo_msg_parser = htb_qdisc_msg_parser, + .qo_free_data = htb_qdisc_free_data, + .qo_dump[NL_DUMP_BRIEF] = htb_qdisc_dump_brief, + .qo_get_opts = htb_qdisc_get_opts, +}; + +static struct rtnl_class_ops htb_class_ops = { + .co_kind = "htb", + .co_msg_parser = htb_class_msg_parser, + .co_free_data = htb_class_free_data, + .co_dump[NL_DUMP_BRIEF] = htb_class_dump_brief, + .co_dump[NL_DUMP_FULL] = htb_class_dump_full, + .co_get_opts = htb_class_get_opts, +}; + +static void __init htb_init(void) +{ + rtnl_qdisc_register(&htb_qdisc_ops); + rtnl_class_register(&htb_class_ops); +} + +static void __exit htb_exit(void) +{ + rtnl_qdisc_unregister(&htb_qdisc_ops); + rtnl_class_unregister(&htb_class_ops); +} + +/** @} */ diff --git a/lib/route/sch/netem.c b/lib/route/sch/netem.c new file mode 100644 index 0000000..e8b8913 --- /dev/null +++ b/lib/route/sch/netem.c @@ -0,0 +1,603 @@ +/* + * lib/route/sch/netem.c Network Emulator Qdisc + * + * 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> + */ + +/** + * @ingroup qdisc_api + * @defgroup netem Network Emulator + * @brief + * + * For further documentation see http://linux-net.osdl.org/index.php/Netem + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/qdisc-modules.h> +#include <netlink/route/sch/netem.h> + +/** @cond SKIP */ +#define SCH_NETEM_ATTR_LATENCY 0x001 +#define SCH_NETEM_ATTR_LIMIT 0x002 +#define SCH_NETEM_ATTR_LOSS 0x004 +#define SCH_NETEM_ATTR_GAP 0x008 +#define SCH_NETEM_ATTR_DUPLICATE 0x010 +#define SCH_NETEM_ATTR_JITTER 0x020 +#define SCH_NETEM_ATTR_DELAY_CORR 0x040 +#define SCH_NETEM_ATTR_LOSS_CORR 0x080 +#define SCH_NETEM_ATTR_DUP_CORR 0x100 +#define SCH_NETEM_ATTR_RO_PROB 0x200 +#define SCH_NETEM_ATTR_RO_CORR 0x400 +/** @endcond */ + +static inline struct rtnl_netem *netem_qdisc(struct rtnl_qdisc *qdisc) +{ + return (struct rtnl_netem *) qdisc->q_subdata; +} + +static inline struct rtnl_netem *netem_alloc(struct rtnl_qdisc *qdisc) +{ + if (!qdisc->q_subdata) + qdisc->q_subdata = calloc(1, sizeof(struct rtnl_netem)); + + return netem_qdisc(qdisc); +} + +static struct nla_policy netem_policy[TCA_NETEM_MAX+1] = { + [TCA_NETEM_CORR] = { .minlen = sizeof(struct tc_netem_corr) }, + [TCA_NETEM_REORDER] = { .minlen = sizeof(struct tc_netem_reorder) }, +}; + +static int netem_msg_parser(struct rtnl_qdisc *qdisc) +{ + int len, err = 0; + struct rtnl_netem *netem; + struct tc_netem_qopt *opts; + + if (qdisc->q_opts->d_size < sizeof(*opts)) + return nl_error(EINVAL, "Netem specific options size mismatch"); + + netem = netem_alloc(qdisc); + if (!netem) + return nl_errno(ENOMEM); + + opts = (struct tc_netem_qopt *) qdisc->q_opts->d_data; + netem->qnm_latency = opts->latency; + netem->qnm_limit = opts->limit; + netem->qnm_loss = opts->loss; + netem->qnm_gap = opts->gap; + netem->qnm_duplicate = opts->duplicate; + netem->qnm_jitter = opts->jitter; + + netem->qnm_mask = (SCH_NETEM_ATTR_LATENCY | SCH_NETEM_ATTR_LIMIT | + SCH_NETEM_ATTR_LOSS | SCH_NETEM_ATTR_GAP | + SCH_NETEM_ATTR_DUPLICATE | SCH_NETEM_ATTR_JITTER); + + len = qdisc->q_opts->d_size - sizeof(*opts); + + if (len > 0) { + struct nlattr *tb[TCA_NETEM_MAX+1]; + + err = nla_parse(tb, TCA_NETEM_MAX, (struct nlattr *) + qdisc->q_opts->d_data + sizeof(*opts), + len, netem_policy); + if (err < 0) { + free(netem); + return err; + } + + if (tb[TCA_NETEM_CORR]) { + struct tc_netem_corr cor; + + nla_memcpy(&cor, tb[TCA_NETEM_CORR], sizeof(cor)); + netem->qnm_corr.nmc_delay = cor.delay_corr; + netem->qnm_corr.nmc_loss = cor.loss_corr; + netem->qnm_corr.nmc_duplicate = cor.dup_corr; + + netem->qnm_mask |= (SCH_NETEM_ATTR_DELAY_CORR | + SCH_NETEM_ATTR_LOSS_CORR | + SCH_NETEM_ATTR_DELAY_CORR); + } + + if (tb[TCA_NETEM_REORDER]) { + struct tc_netem_reorder ro; + + nla_memcpy(&ro, tb[TCA_NETEM_REORDER], sizeof(ro)); + netem->qnm_ro.nmro_probability = ro.probability; + netem->qnm_ro.nmro_correlation = ro.correlation; + + netem->qnm_mask |= (SCH_NETEM_ATTR_RO_PROB | + SCH_NETEM_ATTR_RO_CORR); + } + } + + return 0; +} + +static void netem_free_data(struct rtnl_qdisc *qdisc) +{ + free(qdisc->q_subdata); +} + +static int netem_dump_brief(struct rtnl_qdisc *qdisc, struct nl_dump_params *p, + int line) +{ + struct rtnl_netem *netem = netem_qdisc(qdisc); + + if (netem) + dp_dump(p, "limit %d", netem->qnm_limit); + + return line; +} + +static int netem_dump_full(struct rtnl_qdisc *qdisc, struct nl_dump_params *p, + int line) +{ + return line; +} + +static struct nl_msg *netem_get_opts(struct rtnl_qdisc *qdisc) +{ + return NULL; +} + +/** + * @name Queue Limit + * @{ + */ + +/** + * Set limit of netem qdisc. + * @arg qdisc Netem qdisc to be modified. + * @arg limit New limit in bytes. + * @return 0 on success or a negative error code. + */ +int rtnl_netem_set_limit(struct rtnl_qdisc *qdisc, int limit) +{ + struct rtnl_netem *netem; + + netem = netem_alloc(qdisc); + if (!netem) + return nl_errno(ENOMEM); + + netem->qnm_limit = limit; + netem->qnm_mask |= SCH_NETEM_ATTR_LIMIT; + + return 0; +} + +/** + * Get limit of netem qdisc. + * @arg qdisc Netem qdisc. + * @return Limit in bytes or a negative error code. + */ +int rtnl_netem_get_limit(struct rtnl_qdisc *qdisc) +{ + struct rtnl_netem *netem; + + netem = netem_qdisc(qdisc); + if (netem && (netem->qnm_mask & SCH_NETEM_ATTR_LIMIT)) + return netem->qnm_limit; + else + return nl_errno(ENOENT); +} + +/** @} */ + +/** + * @name Packet Re-ordering + * @{ + */ + +/** + * Set re-ordering gap of netem qdisc. + * @arg qdisc Netem qdisc to be modified. + * @arg gap New gap in number of packets. + * @return 0 on success or a negative error code. + */ +int rtnl_netem_set_gap(struct rtnl_qdisc *qdisc, int gap) +{ + struct rtnl_netem *netem; + + netem = netem_alloc(qdisc); + if (!netem) + return nl_errno(ENOMEM); + + netem->qnm_gap = gap; + netem->qnm_mask |= SCH_NETEM_ATTR_GAP; + + return 0; +} + +/** + * Get re-ordering gap of netem qdisc. + * @arg qdisc Netem qdisc. + * @return Re-ordering gap in packets or a negative error code. + */ +int rtnl_netem_get_gap(struct rtnl_qdisc *qdisc) +{ + struct rtnl_netem *netem; + + netem = netem_qdisc(qdisc); + if (netem && (netem->qnm_mask & SCH_NETEM_ATTR_GAP)) + return netem->qnm_gap; + else + return nl_errno(ENOENT); +} + +/** + * Set re-ordering probability of netem qdisc. + * @arg qdisc Netem qdisc to be modified. + * @arg prob New re-ordering probability. + * @return 0 on success or a negative error code. + */ +int rtnl_netem_set_reorder_probability(struct rtnl_qdisc *qdisc, int prob) +{ + struct rtnl_netem *netem; + + netem = netem_alloc(qdisc); + if (!netem) + return nl_errno(ENOMEM); + + netem->qnm_ro.nmro_probability = prob; + netem->qnm_mask |= SCH_NETEM_ATTR_RO_PROB; + + return 0; +} + +/** + * Get re-ordering probability of netem qdisc. + * @arg qdisc Netem qdisc. + * @return Re-ordering probability or a negative error code. + */ +int rtnl_netem_get_reorder_probability(struct rtnl_qdisc *qdisc) +{ + struct rtnl_netem *netem; + + netem = netem_qdisc(qdisc); + if (netem && (netem->qnm_mask & SCH_NETEM_ATTR_RO_PROB)) + return netem->qnm_ro.nmro_probability; + else + return nl_errno(ENOENT); +} + +/** + * Set re-order correlation probability of netem qdisc. + * @arg qdisc Netem qdisc to be modified. + * @arg prob New re-ordering correlation probability. + * @return 0 on success or a negative error code. + */ +int rtnl_netem_set_reorder_correlation(struct rtnl_qdisc *qdisc, int prob) +{ + struct rtnl_netem *netem; + + netem = netem_alloc(qdisc); + if (!netem) + return nl_errno(ENOMEM); + + netem->qnm_ro.nmro_correlation = prob; + netem->qnm_mask |= SCH_NETEM_ATTR_RO_CORR; + + return 0; +} + +/** + * Get re-ordering correlation probability of netem qdisc. + * @arg qdisc Netem qdisc. + * @return Re-ordering correlation probability or a negative error code. + */ +int rtnl_netem_get_reorder_correlation(struct rtnl_qdisc *qdisc) +{ + struct rtnl_netem *netem; + + netem = netem_qdisc(qdisc); + if (netem && (netem->qnm_mask & SCH_NETEM_ATTR_RO_CORR)) + return netem->qnm_ro.nmro_correlation; + else + return nl_errno(ENOENT); +} + +/** @} */ + +/** + * @name Packet Loss + * @{ + */ + +/** + * Set packet loss probability of netem qdisc. + * @arg qdisc Netem qdisc to be modified. + * @arg prob New packet loss probability. + * @return 0 on success or a negative error code. + */ +int rtnl_netem_set_loss(struct rtnl_qdisc *qdisc, int prob) +{ + struct rtnl_netem *netem; + + netem = netem_alloc(qdisc); + if (!netem) + return nl_errno(ENOMEM); + + netem->qnm_loss = prob; + netem->qnm_mask |= SCH_NETEM_ATTR_LOSS; + + return 0; +} + +/** + * Get packet loss probability of netem qdisc. + * @arg qdisc Netem qdisc. + * @return Packet loss probability or a negative error code. + */ +int rtnl_netem_get_loss(struct rtnl_qdisc *qdisc) +{ + struct rtnl_netem *netem; + + netem = netem_qdisc(qdisc); + if (netem && (netem->qnm_mask & SCH_NETEM_ATTR_LOSS)) + return netem->qnm_loss; + else + return nl_errno(ENOENT); +} + +/** + * Set packet loss correlation probability of netem qdisc. + * @arg qdisc Netem qdisc to be modified. + * @arg prob New packet loss correlation. + * @return 0 on success or a negative error code. + */ +int rtnl_netem_set_loss_correlation(struct rtnl_qdisc *qdisc, int prob) +{ + struct rtnl_netem *netem; + + netem = netem_alloc(qdisc); + if (!netem) + return nl_errno(ENOMEM); + + netem->qnm_corr.nmc_loss = prob; + netem->qnm_mask |= SCH_NETEM_ATTR_LOSS_CORR; + + return 0; +} + +/** + * Get packet loss correlation probability of netem qdisc. + * @arg qdisc Netem qdisc. + * @return Packet loss correlation probability or a negative error code. + */ +int rtnl_netem_get_loss_correlation(struct rtnl_qdisc *qdisc) +{ + struct rtnl_netem *netem; + + netem = netem_qdisc(qdisc); + if (netem && (netem->qnm_mask & SCH_NETEM_ATTR_LOSS_CORR)) + return netem->qnm_corr.nmc_loss; + else + return nl_errno(ENOENT); +} + +/** @} */ + +/** + * @name Packet Duplication + * @{ + */ + +/** + * Set packet duplication probability of netem qdisc. + * @arg qdisc Netem qdisc to be modified. + * @arg prob New packet duplication probability. + * @return 0 on success or a negative error code. + */ +int rtnl_netem_set_duplicate(struct rtnl_qdisc *qdisc, int prob) +{ + struct rtnl_netem *netem; + + netem = netem_alloc(qdisc); + if (!netem) + return nl_errno(ENOMEM); + + netem->qnm_duplicate = prob; + netem->qnm_mask |= SCH_NETEM_ATTR_DUPLICATE; + + return 0; +} + +/** + * Get packet duplication probability of netem qdisc. + * @arg qdisc Netem qdisc. + * @return Packet duplication probability or a negative error code. + */ +int rtnl_netem_get_duplicate(struct rtnl_qdisc *qdisc) +{ + struct rtnl_netem *netem; + + netem = netem_qdisc(qdisc); + if (netem && (netem->qnm_mask & SCH_NETEM_ATTR_DUPLICATE)) + return netem->qnm_duplicate; + else + return nl_errno(ENOENT); +} + +/** + * Set packet duplication correlation probability of netem qdisc. + * @arg qdisc Netem qdisc to be modified. + * @arg prob New packet duplication correlation probability. + * @return 0 on sucess or a negative error code. + */ +int rtnl_netem_set_duplicate_correlation(struct rtnl_qdisc *qdisc, int prob) +{ + struct rtnl_netem *netem; + + netem = netem_alloc(qdisc); + if (!netem) + return nl_errno(ENOMEM); + + netem->qnm_corr.nmc_duplicate = prob; + netem->qnm_mask |= SCH_NETEM_ATTR_DUP_CORR; + + return 0; +} + +/** + * Get packet duplication correlation probability of netem qdisc. + * @arg qdisc Netem qdisc. + * @return Packet duplication correlation probability or a negative error code. + */ +int rtnl_netem_get_duplicate_correlation(struct rtnl_qdisc *qdisc) +{ + struct rtnl_netem *netem; + + netem = netem_qdisc(qdisc); + if (netem && (netem->qnm_mask & SCH_NETEM_ATTR_DUP_CORR)) + return netem->qnm_corr.nmc_duplicate; + else + return nl_errno(ENOENT); +} + +/** @} */ + +/** + * @name Packet Delay + * @{ + */ + +/** + * Set packet delay of netem qdisc. + * @arg qdisc Netem qdisc to be modified. + * @arg delay New packet delay in micro seconds. + * @return 0 on success or a negative error code. + */ +int rtnl_netem_set_delay(struct rtnl_qdisc *qdisc, int delay) +{ + struct rtnl_netem *netem; + + netem = netem_alloc(qdisc); + if (!netem) + return nl_errno(ENOMEM); + + netem->qnm_latency = nl_us2ticks(delay); + netem->qnm_mask |= SCH_NETEM_ATTR_LATENCY; + + return 0; +} + +/** + * Get packet delay of netem qdisc. + * @arg qdisc Netem qdisc. + * @return Packet delay in micro seconds or a negative error code. + */ +int rtnl_netem_get_delay(struct rtnl_qdisc *qdisc) +{ + struct rtnl_netem *netem; + + netem = netem_qdisc(qdisc); + if (netem && (netem->qnm_mask & SCH_NETEM_ATTR_LATENCY)) + return nl_ticks2us(netem->qnm_latency); + else + return nl_errno(ENOENT); +} + +/** + * Set packet delay jitter of netem qdisc. + * @arg qdisc Netem qdisc to be modified. + * @arg jitter New packet delay jitter in micro seconds. + * @return 0 on success or a negative error code. + */ +int rtnl_netem_set_jitter(struct rtnl_qdisc *qdisc, int jitter) +{ + struct rtnl_netem *netem; + + netem = netem_alloc(qdisc); + if (!netem) + return nl_errno(ENOMEM); + + netem->qnm_jitter = nl_us2ticks(jitter); + netem->qnm_mask |= SCH_NETEM_ATTR_JITTER; + + return 0; +} + +/** + * Get packet delay jitter of netem qdisc. + * @arg qdisc Netem qdisc. + * @return Packet delay jitter in micro seconds or a negative error code. + */ +int rtnl_netem_get_jitter(struct rtnl_qdisc *qdisc) +{ + struct rtnl_netem *netem; + + netem = netem_qdisc(qdisc); + if (netem && (netem->qnm_mask & SCH_NETEM_ATTR_JITTER)) + return nl_ticks2us(netem->qnm_jitter); + else + return nl_errno(ENOENT); +} + +/** + * Set packet delay correlation probability of netem qdisc. + * @arg qdisc Netem qdisc to be modified. + * @arg prob New packet delay correlation probability. + */ +int rtnl_netem_set_delay_correlation(struct rtnl_qdisc *qdisc, int prob) +{ + struct rtnl_netem *netem; + + netem = netem_alloc(qdisc); + if (!netem) + return nl_errno(ENOMEM); + + netem->qnm_corr.nmc_delay = prob; + netem->qnm_mask |= SCH_NETEM_ATTR_DELAY_CORR; + + return 0; +} + +/** + * Get packet delay correlation probability of netem qdisc. + * @arg qdisc Netem qdisc. + * @return Packet delay correlation probability or a negative error code. + */ +int rtnl_netem_get_delay_correlation(struct rtnl_qdisc *qdisc) +{ + struct rtnl_netem *netem; + + netem = netem_qdisc(qdisc); + if (netem && (netem->qnm_mask & SCH_NETEM_ATTR_DELAY_CORR)) + return netem->qnm_corr.nmc_delay; + else + return nl_errno(ENOENT); +} + +/** @} */ + +static struct rtnl_qdisc_ops netem_ops = { + .qo_kind = "netem", + .qo_msg_parser = netem_msg_parser, + .qo_free_data = netem_free_data, + .qo_dump[NL_DUMP_BRIEF] = netem_dump_brief, + .qo_dump[NL_DUMP_FULL] = netem_dump_full, + .qo_get_opts = netem_get_opts, +}; + +static void __init netem_init(void) +{ + rtnl_qdisc_register(&netem_ops); +} + +static void __exit netem_exit(void) +{ + rtnl_qdisc_unregister(&netem_ops); +} + +/** @} */ diff --git a/lib/route/sch/prio.c b/lib/route/sch/prio.c new file mode 100644 index 0000000..4e3d624 --- /dev/null +++ b/lib/route/sch/prio.c @@ -0,0 +1,332 @@ +/* + * lib/route/sch/prio.c PRIO Qdisc/Class + * + * 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> + */ + +/** + * @ingroup qdisc_api + * @defgroup prio (Fast) Prio + * @brief + * + * @par 1) Typical PRIO configuration + * @code + * // Specify the maximal number of bands to be used for this PRIO qdisc. + * rtnl_qdisc_prio_set_bands(qdisc, QDISC_PRIO_DEFAULT_BANDS); + * + * // Provide a map assigning each priority to a band number. + * uint8_t map[] = QDISC_PRIO_DEFAULT_PRIOMAP; + * rtnl_qdisc_prio_set_priomap(qdisc, map, sizeof(map)); + * @endcode + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/qdisc-modules.h> +#include <netlink/route/sch/prio.h> + +/** @cond SKIP */ +#define SCH_PRIO_ATTR_BANDS 1 +#define SCH_PRIO_ATTR_PRIOMAP 2 +/** @endcond */ + +static inline struct rtnl_prio *prio_qdisc(struct rtnl_qdisc *qdisc) +{ + return (struct rtnl_prio *) qdisc->q_subdata; +} + +static inline struct rtnl_prio *prio_alloc(struct rtnl_qdisc *qdisc) +{ + if (!qdisc->q_subdata) + qdisc->q_subdata = calloc(1, sizeof(struct rtnl_prio)); + + return prio_qdisc(qdisc); +} + +static int prio_msg_parser(struct rtnl_qdisc *qdisc) +{ + struct rtnl_prio *prio; + struct tc_prio_qopt *opt; + + if (qdisc->q_opts->d_size < sizeof(*opt)) + return nl_error(EINVAL, "prio specific option size mismatch"); + + prio = prio_alloc(qdisc); + if (!prio) + return nl_errno(ENOMEM); + + opt = (struct tc_prio_qopt *) qdisc->q_opts->d_data; + prio->qp_bands = opt->bands; + memcpy(prio->qp_priomap, opt->priomap, sizeof(prio->qp_priomap)); + prio->qp_mask = (SCH_PRIO_ATTR_BANDS | SCH_PRIO_ATTR_PRIOMAP); + + return 0; +} + +static void prio_free_data(struct rtnl_qdisc *qdisc) +{ + free(qdisc->q_subdata); +} + +static int prio_dump_brief(struct rtnl_qdisc *qdisc, + struct nl_dump_params *p, int line) +{ + struct rtnl_prio *prio = prio_qdisc(qdisc); + + if (prio) + dp_dump(p, " bands %u", prio->qp_bands); + + return line; +} + +static int prio_dump_full(struct rtnl_qdisc *qdisc, + struct nl_dump_params *p, int line) +{ + struct rtnl_prio *prio = prio_qdisc(qdisc); + int i, hp; + + if (!prio) + goto ignore; + + dp_dump(p, "priomap ["); + + for (i = 0; i <= TC_PRIO_MAX; i++) + dp_dump(p, "%u%s", prio->qp_priomap[i], + i < TC_PRIO_MAX ? " " : ""); + + dp_dump(p, "]\n"); + dp_new_line(p, line++); + + hp = (((TC_PRIO_MAX/2) + 1) & ~1); + + for (i = 0; i < hp; i++) { + char a[32]; + dp_dump(p, " %18s => %u", + rtnl_prio2str(i, a, sizeof(a)), + prio->qp_priomap[i]); + if (hp+i <= TC_PRIO_MAX) { + dp_dump(p, " %18s => %u", + rtnl_prio2str(hp+i, a, sizeof(a)), + prio->qp_priomap[hp+i]); + if (i < (hp - 1)) { + dp_dump(p, "\n"); + dp_new_line(p, line++); + } + } + } + +ignore: + return line; +} + +static struct nl_msg *prio_get_opts(struct rtnl_qdisc *qdisc) +{ + struct rtnl_prio *prio; + struct tc_prio_qopt opts; + struct nl_msg *msg; + + prio = prio_qdisc(qdisc); + if (!prio || + !(prio->qp_mask & SCH_PRIO_ATTR_PRIOMAP)) + goto errout; + + opts.bands = prio->qp_bands; + memcpy(opts.priomap, prio->qp_priomap, sizeof(opts.priomap)); + + msg = nlmsg_alloc(); + if (!msg) + goto errout; + + if (nlmsg_append(msg, &opts, sizeof(opts), NL_DONTPAD) < 0) { + nlmsg_free(msg); + goto errout; + } + + return msg; +errout: + return NULL; +} + +/** + * @name Attribute Modification + * @{ + */ + +/** + * Set number of bands of PRIO qdisc. + * @arg qdisc PRIO qdisc to be modified. + * @arg bands New number of bands. + * @return 0 on success or a negative error code. + */ +int rtnl_qdisc_prio_set_bands(struct rtnl_qdisc *qdisc, int bands) +{ + struct rtnl_prio *prio; + + prio = prio_alloc(qdisc); + if (!prio) + return nl_errno(ENOMEM); + + prio->qp_bands = bands; + prio->qp_mask |= SCH_PRIO_ATTR_BANDS; + + return 0; +} + +/** + * Get number of bands of PRIO qdisc. + * @arg qdisc PRIO qdisc. + * @return Number of bands or a negative error code. + */ +int rtnl_qdisc_prio_get_bands(struct rtnl_qdisc *qdisc) +{ + struct rtnl_prio *prio; + + prio = prio_qdisc(qdisc); + if (prio && prio->qp_mask & SCH_PRIO_ATTR_BANDS) + return prio->qp_bands; + else + return nl_errno(ENOMEM); +} + +/** + * Set priomap of the PRIO qdisc. + * @arg qdisc PRIO qdisc to be modified. + * @arg priomap New priority mapping. + * @arg len Length of priomap (# of elements). + * @return 0 on success or a negative error code. + */ +int rtnl_qdisc_prio_set_priomap(struct rtnl_qdisc *qdisc, uint8_t priomap[], + int len) +{ + struct rtnl_prio *prio; + int i; + + prio = prio_alloc(qdisc); + if (!prio) + return nl_errno(ENOMEM); + + if (!(prio->qp_mask & SCH_PRIO_ATTR_BANDS)) + return nl_error(EINVAL, "Set number of bands first"); + + if ((len / sizeof(uint8_t)) > (TC_PRIO_MAX+1)) + return nl_error(ERANGE, "priomap length out of bounds"); + + for (i = 0; i <= TC_PRIO_MAX; i++) { + if (priomap[i] > prio->qp_bands) + return nl_error(ERANGE, "priomap element %d " \ + "out of bounds, increase bands number"); + } + + memcpy(prio->qp_priomap, priomap, len); + prio->qp_mask |= SCH_PRIO_ATTR_PRIOMAP; + + return 0; +} + +/** + * Get priomap of a PRIO qdisc. + * @arg qdisc PRIO qdisc. + * @return Priority mapping as array of size TC_PRIO_MAX+1 + * or NULL if an error occured. + */ +uint8_t *rtnl_qdisc_prio_get_priomap(struct rtnl_qdisc *qdisc) +{ + struct rtnl_prio *prio; + + prio = prio_qdisc(qdisc); + if (prio && prio->qp_mask & SCH_PRIO_ATTR_PRIOMAP) + return prio->qp_priomap; + else { + nl_errno(ENOENT); + return NULL; + } +} + +/** @} */ + +/** + * @name Priority Band Translations + * @{ + */ + +static struct trans_tbl prios[] = { + __ADD(TC_PRIO_BESTEFFORT,besteffort) + __ADD(TC_PRIO_FILLER,filler) + __ADD(TC_PRIO_BULK,bulk) + __ADD(TC_PRIO_INTERACTIVE_BULK,interactive_bulk) + __ADD(TC_PRIO_INTERACTIVE,interactive) + __ADD(TC_PRIO_CONTROL,control) +}; + +/** + * Convert priority to character string. + * @arg prio Priority. + * @arg buf Destination buffer + * @arg size Size of destination buffer. + * + * Converts a priority to a character string and stores the result in + * the specified destination buffer. + * + * @return Name of priority as character string. + */ +char * rtnl_prio2str(int prio, char *buf, size_t size) +{ + return __type2str(prio, buf, size, prios, ARRAY_SIZE(prios)); +} + +/** + * Convert character string to priority. + * @arg name Name of priority. + * + * Converts the provided character string specifying a priority + * to the corresponding numeric value. + * + * @return Numeric priority or a negative value if no match was found. + */ +int rtnl_str2prio(const char *name) +{ + return __str2type(name, prios, ARRAY_SIZE(prios)); +} + +/** @} */ + +static struct rtnl_qdisc_ops prio_ops = { + .qo_kind = "prio", + .qo_msg_parser = prio_msg_parser, + .qo_free_data = prio_free_data, + .qo_dump[NL_DUMP_BRIEF] = prio_dump_brief, + .qo_dump[NL_DUMP_FULL] = prio_dump_full, + .qo_get_opts = prio_get_opts, +}; + +static struct rtnl_qdisc_ops pfifo_fast_ops = { + .qo_kind = "pfifo_fast", + .qo_msg_parser = prio_msg_parser, + .qo_free_data = prio_free_data, + .qo_dump[NL_DUMP_BRIEF] = prio_dump_brief, + .qo_dump[NL_DUMP_FULL] = prio_dump_full, + .qo_get_opts = prio_get_opts, +}; + +static void __init prio_init(void) +{ + rtnl_qdisc_register(&prio_ops); + rtnl_qdisc_register(&pfifo_fast_ops); +} + +static void __exit prio_exit(void) +{ + rtnl_qdisc_unregister(&prio_ops); + rtnl_qdisc_unregister(&pfifo_fast_ops); +} + +/** @} */ diff --git a/lib/route/sch/red.c b/lib/route/sch/red.c new file mode 100644 index 0000000..a31c358 --- /dev/null +++ b/lib/route/sch/red.c @@ -0,0 +1,219 @@ +/* + * lib/route/sch/red.c RED Qdisc + * + * 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> + */ + +/** + * @ingroup qdisc_api + * @defgroup red Random Early Detection (RED) + * @brief + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/qdisc-modules.h> +#include <netlink/route/sch/red.h> + +/** @cond SKIP */ +#define RED_ATTR_LIMIT 0x01 +#define RED_ATTR_QTH_MIN 0x02 +#define RED_ATTR_QTH_MAX 0x04 +#define RED_ATTR_FLAGS 0x08 +#define RED_ATTR_WLOG 0x10 +#define RED_ATTR_PLOG 0x20 +#define RED_ATTR_SCELL_LOG 0x40 +/** @endcond */ + +static inline struct rtnl_red *red_qdisc(struct rtnl_qdisc *qdisc) +{ + return (struct rtnl_red *) qdisc->q_subdata; +} + +static inline struct rtnl_red *red_alloc(struct rtnl_qdisc *qdisc) +{ + if (!qdisc->q_subdata) + qdisc->q_subdata = calloc(1, sizeof(struct rtnl_red)); + + return red_qdisc(qdisc); +} + +static struct nla_policy red_policy[TCA_RED_MAX+1] = { + [TCA_RED_PARMS] = { .minlen = sizeof(struct tc_red_qopt) }, +}; + +static int red_msg_parser(struct rtnl_qdisc *qdisc) +{ + struct nlattr *tb[TCA_RED_MAX+1]; + struct rtnl_red *red; + struct tc_red_qopt *opts; + int err; + + if (!(qdisc->ce_mask & TCA_ATTR_OPTS)) + return 0; + + err = tca_parse(tb, TCA_RED_MAX, (struct rtnl_tca *) qdisc, red_policy); + if (err < 0) + return err; + + if (!tb[TCA_RED_PARMS]) + return nl_error(EINVAL, "Missing TCA_RED_PARMS"); + + red = red_alloc(qdisc); + if (!red) + return nl_errno(ENOMEM); + + opts = nla_data(tb[TCA_RED_PARMS]); + + red->qr_limit = opts->limit; + red->qr_qth_min = opts->qth_min; + red->qr_qth_max = opts->qth_max; + red->qr_flags = opts->flags; + red->qr_wlog = opts->Wlog; + red->qr_plog = opts->Plog; + red->qr_scell_log = opts->Scell_log; + + red->qr_mask = (RED_ATTR_LIMIT | RED_ATTR_QTH_MIN | RED_ATTR_QTH_MAX | + RED_ATTR_FLAGS | RED_ATTR_WLOG | RED_ATTR_PLOG | + RED_ATTR_SCELL_LOG); + + return 0; +} + +static int red_dump_brief(struct rtnl_qdisc *qdisc, struct nl_dump_params *p, + int line) +{ + struct rtnl_red *red = red_qdisc(qdisc); + + if (red) { + /* XXX: limit, min, max, flags */ + } + + return line; +} + +static int red_dump_full(struct rtnl_qdisc *qdisc, struct nl_dump_params *p, + int line) +{ + struct rtnl_red *red = red_qdisc(qdisc); + + if (red) { + /* XXX: wlog, plog, scell_log */ + } + + return line; +} + +static int red_dump_stats(struct rtnl_qdisc *qdisc, struct nl_dump_params *p, + int line) +{ + struct rtnl_red *red = red_qdisc(qdisc); + + if (red) { + /* XXX: xstats */ + } + + return line; +} + +static struct nl_msg *red_get_opts(struct rtnl_qdisc *qdisc) +{ + struct rtnl_red *red; + struct nl_msg *msg; + + red = red_qdisc(qdisc); + if (!red) + return NULL; + + msg = nlmsg_alloc(); + if (!msg) + goto errout; + +#if 0 + memset(&opts, 0, sizeof(opts)); + opts.quantum = sfq->qs_quantum; + opts.perturb_period = sfq->qs_perturb; + opts.limit = sfq->qs_limit; + + if (nlmsg_append(msg, &opts, sizeof(opts), NL_DONTPAD) < 0) + goto errout; +#endif + + return msg; +errout: + nlmsg_free(msg); + return NULL; +} + +/** + * @name Attribute Access + * @{ + */ + +/** + * Set limit of RED qdisc. + * @arg qdisc RED qdisc to be modified. + * @arg limit New limit in number of packets. + * @return 0 on success or a negative error code. + */ +int rtnl_red_set_limit(struct rtnl_qdisc *qdisc, int limit) +{ + struct rtnl_red *red; + + red = red_alloc(qdisc); + if (!red) + return nl_errno(ENOMEM); + + red->qr_limit = limit; + red->qr_mask |= RED_ATTR_LIMIT; + + return 0; +} + +/** + * Get limit of RED qdisc. + * @arg qdisc RED qdisc. + * @return Limit or a negative error code. + */ +int rtnl_red_get_limit(struct rtnl_qdisc *qdisc) +{ + struct rtnl_red *red; + + red = red_qdisc(qdisc); + if (red && (red->qr_mask & RED_ATTR_LIMIT)) + return red->qr_limit; + else + return nl_errno(ENOENT); +} + +/** @} */ + +static struct rtnl_qdisc_ops red_ops = { + .qo_kind = "red", + .qo_msg_parser = red_msg_parser, + .qo_dump[NL_DUMP_BRIEF] = red_dump_brief, + .qo_dump[NL_DUMP_FULL] = red_dump_full, + .qo_dump[NL_DUMP_STATS] = red_dump_stats, + .qo_get_opts = red_get_opts, +}; + +static void __init red_init(void) +{ + rtnl_qdisc_register(&red_ops); +} + +static void __exit red_exit(void) +{ + rtnl_qdisc_unregister(&red_ops); +} + +/** @} */ diff --git a/lib/route/sch/sfq.c b/lib/route/sch/sfq.c new file mode 100644 index 0000000..d530c0f --- /dev/null +++ b/lib/route/sch/sfq.c @@ -0,0 +1,294 @@ +/* + * lib/route/sch/sfq.c SFQ Qdisc + * + * 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> + */ + +/** + * @ingroup qdisc_api + * @defgroup sfq Stochastic Fairness Queueing (SFQ) + * @brief + * + * @par Parameter Description + * - \b Quantum: Number of bytes to send out per slot and round. + * - \b Perturbation: Timer period between changing the hash function. + * - \b Limit: Upper limit of queue in number of packets before SFQ starts + * dropping packets. + * - \b Divisor: Hash table divisor, i.e. size of hash table. + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/qdisc-modules.h> +#include <netlink/route/sch/sfq.h> + +/** @cond SKIP */ +#define SCH_SFQ_ATTR_QUANTUM 0x01 +#define SCH_SFQ_ATTR_PERTURB 0x02 +#define SCH_SFQ_ATTR_LIMIT 0x04 +#define SCH_SFQ_ATTR_DIVISOR 0x08 +#define SCH_SFQ_ATTR_FLOWS 0x10 +/** @endcond */ + +static inline struct rtnl_sfq *sfq_qdisc(struct rtnl_qdisc *qdisc) +{ + return (struct rtnl_sfq *) qdisc->q_subdata; +} + +static inline struct rtnl_sfq *sfq_alloc(struct rtnl_qdisc *qdisc) +{ + if (!qdisc->q_subdata) + qdisc->q_subdata = calloc(1, sizeof(struct rtnl_sfq)); + + return sfq_qdisc(qdisc); +} + +static int sfq_msg_parser(struct rtnl_qdisc *qdisc) +{ + struct rtnl_sfq *sfq; + struct tc_sfq_qopt *opts; + + if (!(qdisc->ce_mask & TCA_ATTR_OPTS)) + return 0; + + if (qdisc->q_opts->d_size < sizeof(*opts)) + return nl_error(EINVAL, "SFQ specific options size mismatch"); + + sfq = sfq_alloc(qdisc); + if (!sfq) + return nl_errno(ENOMEM); + + opts = (struct tc_sfq_qopt *) qdisc->q_opts->d_data; + + sfq->qs_quantum = opts->quantum; + sfq->qs_perturb = opts->perturb_period; + sfq->qs_limit = opts->limit; + sfq->qs_divisor = opts->divisor; + sfq->qs_flows = opts->flows; + + sfq->qs_mask = (SCH_SFQ_ATTR_QUANTUM | SCH_SFQ_ATTR_PERTURB | + SCH_SFQ_ATTR_LIMIT | SCH_SFQ_ATTR_DIVISOR | + SCH_SFQ_ATTR_FLOWS); + + return 0; +} + +static void sfq_free_data(struct rtnl_qdisc *qdisc) +{ + free(qdisc->q_subdata); +} + +static int sfq_dump_brief(struct rtnl_qdisc *qdisc, struct nl_dump_params *p, + int line) +{ + struct rtnl_sfq *sfq = sfq_qdisc(qdisc); + + if (sfq) + dp_dump(p, " quantum %u perturb %us", + sfq->qs_quantum, + nl_ticks2us(sfq->qs_perturb * nl_get_hz())); + + return line; +} + +static int sfq_dump_full(struct rtnl_qdisc *qdisc, struct nl_dump_params *p, + int line) +{ + struct rtnl_sfq *sfq = sfq_qdisc(qdisc); + + if (sfq) + dp_dump(p, "limit %u divisor %u", + sfq->qs_limit, sfq->qs_divisor); + + return line; +} + +static struct nl_msg *sfq_get_opts(struct rtnl_qdisc *qdisc) +{ + struct rtnl_sfq *sfq; + struct tc_sfq_qopt opts; + struct nl_msg *msg; + + sfq = sfq_qdisc(qdisc); + if (!sfq) + return NULL; + + msg = nlmsg_alloc(); + if (!msg) + goto errout; + + memset(&opts, 0, sizeof(opts)); + opts.quantum = sfq->qs_quantum; + opts.perturb_period = sfq->qs_perturb; + opts.limit = sfq->qs_limit; + + if (nlmsg_append(msg, &opts, sizeof(opts), NL_DONTPAD) < 0) + goto errout; + + return msg; +errout: + nlmsg_free(msg); + return NULL; +} + +/** + * @name Attribute Access + * @{ + */ + +/** + * Set quantum of SFQ qdisc. + * @arg qdisc SFQ qdisc to be modified. + * @arg quantum New quantum in bytes. + * @return 0 on success or a negative error code. + */ +int rtnl_sfq_set_quantum(struct rtnl_qdisc *qdisc, int quantum) +{ + struct rtnl_sfq *sfq; + + sfq = sfq_alloc(qdisc); + if (!sfq) + return nl_errno(ENOMEM); + + sfq->qs_quantum = quantum; + sfq->qs_mask |= SCH_SFQ_ATTR_QUANTUM; + + return 0; +} + +/** + * Get quantum of SFQ qdisc. + * @arg qdisc SFQ qdisc. + * @return Quantum in bytes or a negative error code. + */ +int rtnl_sfq_get_quantum(struct rtnl_qdisc *qdisc) +{ + struct rtnl_sfq *sfq; + + sfq = sfq_qdisc(qdisc); + if (sfq && sfq->qs_mask & SCH_SFQ_ATTR_QUANTUM) + return sfq->qs_quantum; + else + return nl_errno(ENOENT); +} + +/** + * Set limit of SFQ qdisc. + * @arg qdisc SFQ qdisc to be modified. + * @arg limit New limit in number of packets. + * @return 0 on success or a negative error code. + */ +int rtnl_sfq_set_limit(struct rtnl_qdisc *qdisc, int limit) +{ + struct rtnl_sfq *sfq; + + sfq = sfq_alloc(qdisc); + if (!sfq) + return nl_errno(ENOMEM); + + sfq->qs_limit = limit; + sfq->qs_mask |= SCH_SFQ_ATTR_LIMIT; + + return 0; +} + +/** + * Get limit of SFQ qdisc. + * @arg qdisc SFQ qdisc. + * @return Limit or a negative error code. + */ +int rtnl_sfq_get_limit(struct rtnl_qdisc *qdisc) +{ + struct rtnl_sfq *sfq; + + sfq = sfq_qdisc(qdisc); + if (sfq && sfq->qs_mask & SCH_SFQ_ATTR_LIMIT) + return sfq->qs_limit; + else + return nl_errno(ENOENT); +} + +/** + * Set perturbation interval of SFQ qdisc. + * @arg qdisc SFQ qdisc to be modified. + * @arg perturb New perturbation interval in seconds. + * @note A value of 0 disables perturbation altogether. + * @return 0 on success or a negative error code. + */ +int rtnl_sfq_set_perturb(struct rtnl_qdisc *qdisc, int perturb) +{ + struct rtnl_sfq *sfq; + + sfq = sfq_alloc(qdisc); + if (!sfq) + return nl_errno(ENOMEM); + + sfq->qs_perturb = perturb; + sfq->qs_mask |= SCH_SFQ_ATTR_PERTURB; + + return 0; +} + +/** + * Get perturbation interval of SFQ qdisc. + * @arg qdisc SFQ qdisc. + * @return Perturbation interval in seconds or a negative error code. + */ +int rtnl_sfq_get_perturb(struct rtnl_qdisc *qdisc) +{ + struct rtnl_sfq *sfq; + + sfq = sfq_qdisc(qdisc); + if (sfq && sfq->qs_mask & SCH_SFQ_ATTR_PERTURB) + return sfq->qs_perturb; + else + return nl_errno(ENOENT); +} + +/** + * Get divisor of SFQ qdisc. + * @arg qdisc SFQ qdisc. + * @return Divisor in number of entries or a negative error code. + */ +int rtnl_sfq_get_divisor(struct rtnl_qdisc *qdisc) +{ + struct rtnl_sfq *sfq; + + sfq = sfq_qdisc(qdisc); + if (sfq && sfq->qs_mask & SCH_SFQ_ATTR_DIVISOR) + return sfq->qs_divisor; + else + return nl_errno(ENOENT); +} + +/** @} */ + +static struct rtnl_qdisc_ops sfq_ops = { + .qo_kind = "sfq", + .qo_msg_parser = sfq_msg_parser, + .qo_free_data = sfq_free_data, + .qo_dump[NL_DUMP_BRIEF] = sfq_dump_brief, + .qo_dump[NL_DUMP_FULL] = sfq_dump_full, + .qo_get_opts = sfq_get_opts, +}; + +static void __init sfq_init(void) +{ + rtnl_qdisc_register(&sfq_ops); +} + +static void __exit sfq_exit(void) +{ + rtnl_qdisc_unregister(&sfq_ops); +} + +/** @} */ diff --git a/lib/route/sch/tbf.c b/lib/route/sch/tbf.c new file mode 100644 index 0000000..04d1689 --- /dev/null +++ b/lib/route/sch/tbf.c @@ -0,0 +1,540 @@ +/* + * lib/route/sch/tbf.c TBF Qdisc + * + * 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> + */ + +/** + * @ingroup qdisc_api + * @defgroup tbf Token Bucket Filter (TBF) + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/cache.h> +#include <netlink/utils.h> +#include <netlink/route/tc.h> +#include <netlink/route/qdisc.h> +#include <netlink/route/qdisc-modules.h> +#include <netlink/route/class.h> +#include <netlink/route/class-modules.h> +#include <netlink/route/link.h> +#include <netlink/route/sch/tbf.h> + +/** @cond SKIP */ +#define TBF_ATTR_LIMIT 0x01 +#define TBF_ATTR_RATE 0x02 +#define TBF_ATTR_PEAKRATE 0x10 +#define TBF_ATTR_MPU 0x80 +/** @endcond */ + +static inline struct rtnl_tbf *tbf_qdisc(struct rtnl_qdisc *qdisc) +{ + return (struct rtnl_tbf *) qdisc->q_subdata; +} + +static inline struct rtnl_tbf *tbf_alloc(struct rtnl_qdisc *qdisc) +{ + if (!qdisc->q_subdata) + qdisc->q_subdata = calloc(1, sizeof(struct rtnl_tbf)); + + return tbf_qdisc(qdisc); +} + +static struct nla_policy tbf_policy[TCA_TBF_MAX+1] = { + [TCA_TBF_PARMS] = { .minlen = sizeof(struct tc_tbf_qopt) }, +}; + +static int tbf_msg_parser(struct rtnl_qdisc *q) +{ + int err; + struct nlattr *tb[TCA_TBF_MAX + 1]; + struct rtnl_tbf *tbf; + + err = tca_parse(tb, TCA_TBF_MAX, (struct rtnl_tca *) q, tbf_policy); + if (err < 0) + return err; + + tbf = tbf_qdisc(q); + if (!tbf) + return nl_errno(ENOMEM); + + if (tb[TCA_TBF_PARMS]) { + struct tc_tbf_qopt opts; + int bufsize; + + nla_memcpy(&opts, tb[TCA_TBF_PARMS], sizeof(opts)); + tbf->qt_limit = opts.limit; + tbf->qt_mpu = opts.rate.mpu; + + rtnl_copy_ratespec(&tbf->qt_rate, &opts.rate); + tbf->qt_rate_txtime = opts.buffer; + bufsize = rtnl_tc_calc_bufsize(nl_ticks2us(opts.buffer), + opts.rate.rate); + tbf->qt_rate_bucket = bufsize; + + rtnl_copy_ratespec(&tbf->qt_peakrate, &opts.peakrate); + tbf->qt_peakrate_txtime = opts.mtu; + bufsize = rtnl_tc_calc_bufsize(nl_ticks2us(opts.mtu), + opts.peakrate.rate); + tbf->qt_peakrate_bucket = bufsize; + + tbf->qt_mask = (TBF_ATTR_LIMIT | TBF_ATTR_MPU | TBF_ATTR_RATE | + TBF_ATTR_PEAKRATE); + } + + return 0; +} + +static int tbf_dump_brief(struct rtnl_qdisc *qdisc, struct nl_dump_params *p, + int line) +{ + double r, rbit, lim; + char *ru, *rubit, *limu; + struct rtnl_tbf *tbf = tbf_qdisc(qdisc); + + if (!tbf) + goto ignore; + + r = nl_cancel_down_bytes(tbf->qt_rate.rs_rate, &ru); + rbit = nl_cancel_down_bits(tbf->qt_rate.rs_rate*8, &rubit); + lim = nl_cancel_down_bytes(tbf->qt_limit, &limu); + + dp_dump(p, " rate %.2f%s/s (%.0f%s) limit %.2f%s", + r, ru, rbit, rubit, lim, limu); + +ignore: + return line; +} + +static int tbf_dump_full(struct rtnl_qdisc *qdisc, struct nl_dump_params *p, + int line) +{ + struct rtnl_tbf *tbf = tbf_qdisc(qdisc); + + if (!tbf) + goto ignore; + + if (1) { + char *bu, *cu; + double bs = nl_cancel_down_bytes(tbf->qt_rate_bucket, &bu); + double cl = nl_cancel_down_bytes(1 << tbf->qt_rate.rs_cell_log, + &cu); + + dp_dump(p, "mpu %u rate-bucket-size %1.f%s " + "rate-cell-size %.1f%s\n", + tbf->qt_mpu, bs, bu, cl, cu); + + } + + if (tbf->qt_mask & TBF_ATTR_PEAKRATE) { + char *pru, *prbu, *bsu, *clu; + double pr, prb, bs, cl; + + pr = nl_cancel_down_bytes(tbf->qt_peakrate.rs_rate, &pru); + prb = nl_cancel_down_bits(tbf->qt_peakrate.rs_rate * 8, &prbu); + bs = nl_cancel_down_bits(tbf->qt_peakrate_bucket, &bsu); + cl = nl_cancel_down_bits(1 << tbf->qt_peakrate.rs_cell_log, + &clu); + + dp_dump_line(p, line++, " peak-rate %.2f%s/s (%.0f%s) " + "bucket-size %.1f%s cell-size %.1f%s", + "latency %.1f%s", + pr, pru, prb, prbu, bs, bsu, cl, clu); + } + +ignore: + return line; +} + +static struct nl_msg *tbf_get_opts(struct rtnl_qdisc *qdisc) +{ + struct tc_tbf_qopt opts; + struct rtnl_tbf *tbf; + struct nl_msg *msg; + uint32_t rtab[RTNL_TC_RTABLE_SIZE]; + uint32_t ptab[RTNL_TC_RTABLE_SIZE]; + int required = TBF_ATTR_RATE | TBF_ATTR_LIMIT; + + memset(&opts, 0, sizeof(opts)); + + tbf = tbf_qdisc(qdisc); + if (!tbf) + return NULL; + + if (!(tbf->qt_mask & required) != required) + return NULL; + + opts.limit = tbf->qt_limit; + opts.buffer = tbf->qt_rate_txtime; + tbf->qt_rate.rs_mpu = tbf->qt_mpu; + rtnl_rcopy_ratespec(&opts.rate, &tbf->qt_rate); + + rtnl_tc_build_rate_table(rtab, tbf->qt_mpu & 0xff, tbf->qt_mpu >> 8, + 1 << tbf->qt_rate.rs_cell_log, + tbf->qt_rate.rs_rate); + + if (tbf->qt_mask & TBF_ATTR_PEAKRATE) { + opts.mtu = tbf->qt_peakrate_txtime; + tbf->qt_peakrate.rs_mpu = tbf->qt_mpu; + rtnl_rcopy_ratespec(&opts.peakrate, &tbf->qt_peakrate); + + rtnl_tc_build_rate_table(ptab, tbf->qt_mpu & 0xff, + tbf->qt_mpu >> 8, + 1 << tbf->qt_peakrate.rs_cell_log, + tbf->qt_peakrate.rs_rate); + } + + msg = nlmsg_alloc(); + if (!msg) + goto nla_put_failure; + + NLA_PUT(msg, TCA_TBF_PARMS, sizeof(opts), &opts); + NLA_PUT(msg, TCA_TBF_RTAB, sizeof(rtab), rtab); + + if (tbf->qt_mask & TBF_ATTR_PEAKRATE) + NLA_PUT(msg, TCA_TBF_PTAB, sizeof(ptab), ptab); + + return msg; + +nla_put_failure: + nlmsg_free(msg); + return NULL; +} + +/** + * @name Attribute Access + * @{ + */ + +/** + * Set limit of TBF qdisc. + * @arg qdisc TBF qdisc to be modified. + * @arg limit New limit in bytes. + * @return 0 on success or a negative error code. + */ +int rtnl_qdisc_tbf_set_limit(struct rtnl_qdisc *qdisc, int limit) +{ + struct rtnl_tbf *tbf; + + tbf = tbf_alloc(qdisc); + if (!tbf) + return nl_errno(ENOMEM); + + tbf->qt_limit = limit; + tbf->qt_mask |= TBF_ATTR_LIMIT; + + return 0; +} + +static inline double calc_limit(struct rtnl_ratespec *spec, int latency, + int bucket) +{ + double limit; + + limit = (double) spec->rs_rate * ((double) latency / 1000000.); + limit += bucket; + + return limit; +} + +/** + * Set limit of TBF qdisc by latency. + * @arg qdisc TBF qdisc to be modified. + * @arg latency Latency in micro seconds. + * + * Calculates and sets the limit based on the desired latency and the + * configured rate and peak rate. In order for this operation to succeed, + * the rate and if required the peak rate must have been set in advance. + * + * @f[ + * limit_n = \frac{{rate_n} \times {latency}}{10^6}+{bucketsize}_n + * @f] + * @f[ + * limit = min(limit_{rate},limit_{peak}) + * @f] + * + * @return 0 on success or a negative error code. + */ +int rtnl_qdisc_tbf_set_limit_by_latency(struct rtnl_qdisc *qdisc, int latency) +{ + struct rtnl_tbf *tbf; + double limit, limit2; + + tbf = tbf_alloc(qdisc); + if (!tbf) + return nl_errno(ENOMEM); + + if (!(tbf->qt_mask & TBF_ATTR_RATE)) + return nl_error(EINVAL, "The rate must be specified before " + "limit can be calculated based on latency."); + + limit = calc_limit(&tbf->qt_rate, latency, tbf->qt_rate_bucket); + + if (tbf->qt_mask & TBF_ATTR_PEAKRATE) { + limit2 = calc_limit(&tbf->qt_peakrate, latency, + tbf->qt_peakrate_bucket); + + if (limit2 < limit) + limit = limit2; + } + + return rtnl_qdisc_tbf_set_limit(qdisc, (int) limit); +} + +/** + * Get limit of TBF qdisc. + * @arg qdisc TBF qdisc. + * @return Limit in bytes or a negative error code. + */ +int rtnl_qdisc_tbf_get_limit(struct rtnl_qdisc *qdisc) +{ + struct rtnl_tbf *tbf; + + tbf = tbf_qdisc(qdisc); + if (tbf && (tbf->qt_mask & TBF_ATTR_LIMIT)) + return tbf->qt_limit; + return + nl_errno(ENOENT); +} + +/** + * Set MPU of TBF qdisc. + * @arg qdisc TBF qdisc to be modified. + * @arg mpu New MPU in bytes. + * @return 0 on success or a negative error code. + */ +int rtnl_qdisc_tbf_set_mpu(struct rtnl_qdisc *qdisc, int mpu) +{ + struct rtnl_tbf *tbf; + + tbf = tbf_alloc(qdisc); + if (!tbf) + return nl_errno(ENOMEM); + + tbf->qt_mpu = mpu; + tbf->qt_mask |= TBF_ATTR_MPU; + + return 0; +} + +/** + * Get MPU of TBF qdisc. + * @arg qdisc TBF qdisc. + * @return MPU in bytes or a negative error code. + */ +int rtnl_qdisc_tbf_get_mpu(struct rtnl_qdisc *qdisc) +{ + struct rtnl_tbf *tbf; + + tbf = tbf_qdisc(qdisc); + if (tbf && (tbf->qt_mask & TBF_ATTR_MPU)) + return tbf->qt_mpu; + return + nl_errno(ENOENT); +} + +static inline int calc_cell_log(int cell, int bucket) +{ + if (cell > 0) + cell = rtnl_tc_calc_cell_log(cell); + else { + cell = 0; + + if (!bucket) + bucket = 2047; /* defaults to cell_log=3 */ + + while ((bucket >> cell) > 255) + cell++; + } + + return cell; +} + +/** + * Set rate of TBF qdisc. + * @arg qdisc TBF qdisc to be modified. + * @arg rate New rate in bytes per second. + * @arg bucket Size of bucket in bytes. + * @arg cell Size of a rate cell or 0 to get default value. + * @return 0 on success or a negative error code. + */ +int rtnl_qdisc_tbf_set_rate(struct rtnl_qdisc *qdisc, int rate, int bucket, + int cell) +{ + struct rtnl_tbf *tbf; + int cell_log; + + tbf = tbf_alloc(qdisc); + if (!tbf) + return nl_errno(ENOMEM); + + cell_log = calc_cell_log(cell, bucket); + if (cell_log < 0) + return cell_log; + + tbf->qt_rate.rs_rate = rate; + tbf->qt_rate_bucket = bucket; + tbf->qt_rate.rs_cell_log = cell_log; + tbf->qt_rate_txtime = rtnl_tc_calc_txtime(bucket, rate); + tbf->qt_mask |= TBF_ATTR_RATE; + + return 0; +} + +/** + * Get rate of TBF qdisc. + * @arg qdisc TBF qdisc. + * @return Rate in bytes per seconds or a negative error code. + */ +int rtnl_qdisc_tbf_get_rate(struct rtnl_qdisc *qdisc) +{ + struct rtnl_tbf *tbf; + + tbf = tbf_qdisc(qdisc); + if (tbf && (tbf->qt_mask & TBF_ATTR_RATE)) + return tbf->qt_rate.rs_rate; + else + return -1; +} + +/** + * Get rate bucket size of TBF qdisc. + * @arg qdisc TBF qdisc. + * @return Size of rate bucket or a negative error code. + */ +int rtnl_qdisc_tbf_get_rate_bucket(struct rtnl_qdisc *qdisc) +{ + struct rtnl_tbf *tbf; + + tbf = tbf_qdisc(qdisc); + if (tbf && (tbf->qt_mask & TBF_ATTR_RATE)) + return tbf->qt_rate_bucket; + else + return -1; +} + +/** + * Get rate cell size of TBF qdisc. + * @arg qdisc TBF qdisc. + * @return Size of rate cell in bytes or a negative error code. + */ +int rtnl_qdisc_tbf_get_rate_cell(struct rtnl_qdisc *qdisc) +{ + struct rtnl_tbf *tbf; + + tbf = tbf_qdisc(qdisc); + if (tbf && (tbf->qt_mask & TBF_ATTR_RATE)) + return (1 << tbf->qt_rate.rs_cell_log); + else + return -1; +} + +/** + * Set peak rate of TBF qdisc. + * @arg qdisc TBF qdisc to be modified. + * @arg rate New peak rate in bytes per second. + * @arg bucket Size of peakrate bucket. + * @arg cell Size of a peakrate cell or 0 to get default value. + * @return 0 on success or a negative error code. + */ +int rtnl_qdisc_tbf_set_peakrate(struct rtnl_qdisc *qdisc, int rate, int bucket, + int cell) +{ + struct rtnl_tbf *tbf; + int cell_log; + + tbf = tbf_alloc(qdisc); + if (!tbf) + return nl_errno(ENOMEM); + + cell_log = calc_cell_log(cell, bucket); + if (cell_log < 0) + return cell_log; + + tbf->qt_peakrate.rs_rate = rate; + tbf->qt_peakrate_bucket = bucket; + tbf->qt_peakrate.rs_cell_log = cell_log; + tbf->qt_peakrate_txtime = rtnl_tc_calc_txtime(bucket, rate); + + tbf->qt_mask |= TBF_ATTR_PEAKRATE; + + return 0; +} + +/** + * Get peak rate of TBF qdisc. + * @arg qdisc TBF qdisc. + * @return Peak rate in bytes per seconds or a negative error code. + */ +int rtnl_qdisc_tbf_get_peakrate(struct rtnl_qdisc *qdisc) +{ + struct rtnl_tbf *tbf; + + tbf = tbf_qdisc(qdisc); + if (tbf && (tbf->qt_mask & TBF_ATTR_PEAKRATE)) + return tbf->qt_peakrate.rs_rate; + else + return -1; +} + +/** + * Get peak rate bucket size of TBF qdisc. + * @arg qdisc TBF qdisc. + * @return Size of peak rate bucket or a negative error code. + */ +int rtnl_qdisc_tbf_get_peakrate_bucket(struct rtnl_qdisc *qdisc) +{ + struct rtnl_tbf *tbf; + + tbf = tbf_qdisc(qdisc); + if (tbf && (tbf->qt_mask & TBF_ATTR_PEAKRATE)) + return tbf->qt_peakrate_bucket; + else + return -1; +} + +/** + * Get peak rate cell size of TBF qdisc. + * @arg qdisc TBF qdisc. + * @return Size of peak rate cell in bytes or a negative error code. + */ +int rtnl_qdisc_tbf_get_peakrate_cell(struct rtnl_qdisc *qdisc) +{ + struct rtnl_tbf *tbf; + + tbf = tbf_qdisc(qdisc); + if (tbf && (tbf->qt_mask & TBF_ATTR_PEAKRATE)) + return (1 << tbf->qt_peakrate.rs_cell_log); + else + return -1; +} + +/** @} */ + +static struct rtnl_qdisc_ops tbf_qdisc_ops = { + .qo_kind = "tbf", + .qo_msg_parser = tbf_msg_parser, + .qo_dump[NL_DUMP_BRIEF] = tbf_dump_brief, + .qo_dump[NL_DUMP_FULL] = tbf_dump_full, + .qo_get_opts = tbf_get_opts, +}; + +static void __init tbf_init(void) +{ + rtnl_qdisc_register(&tbf_qdisc_ops); +} + +static void __exit tbf_exit(void) +{ + rtnl_qdisc_unregister(&tbf_qdisc_ops); +} + +/** @} */ diff --git a/lib/route/tc.c b/lib/route/tc.c new file mode 100644 index 0000000..1351fa2 --- /dev/null +++ b/lib/route/tc.c @@ -0,0 +1,585 @@ +/* + * lib/route/tc.c Traffic Control + * + * 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> + */ + +/** + * @ingroup rtnl + * @defgroup tc Traffic Control + * @brief + * @{ + */ + +#include <netlink-local.h> +#include <netlink-tc.h> +#include <netlink/netlink.h> +#include <netlink/utils.h> +#include <netlink/route/rtnl.h> +#include <netlink/route/link.h> +#include <netlink/route/tc.h> + +/** @cond SKIP */ + +static struct nla_policy tc_policy[TCA_MAX+1] = { + [TCA_KIND] = { .type = NLA_STRING, + .maxlen = TCKINDSIZ }, + [TCA_STATS] = { .minlen = sizeof(struct tc_stats) }, + [TCA_STATS2] = { .type = NLA_NESTED }, +}; + +int tca_parse(struct nlattr **tb, int maxattr, struct rtnl_tca *g, + struct nla_policy *policy) +{ + + if (g->ce_mask & TCA_ATTR_OPTS) + return nla_parse(tb, maxattr, + (struct nlattr *) g->tc_opts->d_data, + g->tc_opts->d_size, policy); + else { + /* Ugly but tb[] must be in a defined state even if no + * attributes can be found. */ + memset(tb, 0, sizeof(struct nlattr *) * (maxattr + 1)); + return 0; + } +} + +static struct nla_policy tc_stats2_policy[TCA_STATS_MAX+1] = { + [TCA_STATS_BASIC] = { .minlen = sizeof(struct gnet_stats_basic) }, + [TCA_STATS_RATE_EST] = { .minlen = sizeof(struct gnet_stats_rate_est) }, + [TCA_STATS_QUEUE] = { .minlen = sizeof(struct gnet_stats_queue) }, +}; + +int tca_msg_parser(struct nlmsghdr *n, struct rtnl_tca *g) +{ + struct nlattr *tb[TCA_MAX + 1]; + struct tcmsg *tm; + int err; + + err = nlmsg_parse(n, sizeof(*tm), tb, TCA_MAX, tc_policy); + if (err < 0) + return err; + + if (tb[TCA_KIND] == NULL) + return nl_error(EINVAL, "Missing tca kind TLV"); + + nla_strlcpy(g->tc_kind, tb[TCA_KIND], TCKINDSIZ); + + tm = nlmsg_data(n); + g->tc_family = tm->tcm_family; + g->tc_ifindex = tm->tcm_ifindex; + g->tc_handle = tm->tcm_handle; + g->tc_parent = tm->tcm_parent; + g->tc_info = tm->tcm_info; + + g->ce_mask = (TCA_ATTR_FAMILY | TCA_ATTR_IFINDEX | TCA_ATTR_HANDLE | + TCA_ATTR_PARENT | TCA_ATTR_INFO | TCA_ATTR_KIND); + + if (tb[TCA_OPTIONS]) { + g->tc_opts = nla_get_data(tb[TCA_OPTIONS]); + if (!g->tc_opts) + return nl_errno(ENOMEM); + g->ce_mask |= TCA_ATTR_OPTS; + } + + + if (tb[TCA_STATS2]) { + struct nlattr *tbs[TCA_STATS_MAX + 1]; + + err = nla_parse_nested(tbs, TCA_STATS_MAX, tb[TCA_STATS2], + tc_stats2_policy); + if (err < 0) + return err; + + if (tbs[TCA_STATS_BASIC]) { + struct gnet_stats_basic *bs; + + bs = nla_data(tbs[TCA_STATS_BASIC]); + g->tc_stats[RTNL_TC_BYTES] = bs->bytes; + g->tc_stats[RTNL_TC_PACKETS] = bs->packets; + } + + if (tbs[TCA_STATS_RATE_EST]) { + struct gnet_stats_rate_est *re; + + re = nla_data(tbs[TCA_STATS_RATE_EST]); + g->tc_stats[RTNL_TC_RATE_BPS] = re->bps; + g->tc_stats[RTNL_TC_RATE_PPS] = re->pps; + } + + if (tbs[TCA_STATS_QUEUE]) { + struct gnet_stats_queue *q; + + q = nla_data(tbs[TCA_STATS_QUEUE]); + g->tc_stats[RTNL_TC_QLEN] = q->qlen; + g->tc_stats[RTNL_TC_BACKLOG] = q->backlog; + g->tc_stats[RTNL_TC_DROPS] = q->drops; + g->tc_stats[RTNL_TC_REQUEUES] = q->requeues; + g->tc_stats[RTNL_TC_OVERLIMITS] = q->overlimits; + } + + g->ce_mask |= TCA_ATTR_STATS; + + if (tbs[TCA_STATS_APP]) { + g->tc_xstats = nla_get_data(tbs[TCA_STATS_APP]); + if (g->tc_xstats == NULL) + return -ENOMEM; + } else + goto compat_xstats; + } else { + if (tb[TCA_STATS]) { + struct tc_stats *st = nla_data(tb[TCA_STATS]); + + g->tc_stats[RTNL_TC_BYTES] = st->bytes; + g->tc_stats[RTNL_TC_PACKETS] = st->packets; + g->tc_stats[RTNL_TC_RATE_BPS] = st->bps; + g->tc_stats[RTNL_TC_RATE_PPS] = st->pps; + g->tc_stats[RTNL_TC_QLEN] = st->qlen; + g->tc_stats[RTNL_TC_BACKLOG] = st->backlog; + g->tc_stats[RTNL_TC_DROPS] = st->drops; + g->tc_stats[RTNL_TC_OVERLIMITS] = st->overlimits; + + g->ce_mask |= TCA_ATTR_STATS; + } + +compat_xstats: + if (tb[TCA_XSTATS]) { + g->tc_xstats = nla_get_data(tb[TCA_XSTATS]); + if (g->tc_xstats == NULL) + return -ENOMEM; + g->ce_mask |= TCA_ATTR_XSTATS; + } + } + + + return 0; +} + +void tca_free_data(struct rtnl_tca *tca) +{ + nl_data_free(tca->tc_opts); + nl_data_free(tca->tc_xstats); +} + +int tca_clone(struct rtnl_tca *dst, struct rtnl_tca *src) +{ + if (src->tc_opts) { + dst->tc_opts = nl_data_clone(src->tc_opts); + if (!dst->tc_opts) + goto errout; + } + + if (src->tc_xstats) { + dst->tc_xstats = nl_data_clone(src->tc_xstats); + if (!dst->tc_xstats) + goto errout; + } + + return 0; +errout: + return nl_get_errno(); +} + +int tca_dump_brief(struct rtnl_tca *g, const char *type, + struct nl_dump_params *p, int line) +{ + char handle[32], parent[32]; + struct nl_cache *link_cache; + + link_cache = nl_cache_mngt_require("route/link"); + + dp_dump(p, "%s %s ", g->tc_kind, type); + + if (link_cache) { + char buf[32]; + dp_dump(p, "dev %s ", + rtnl_link_i2name(link_cache, g->tc_ifindex, + buf, sizeof(buf))); + } else + dp_dump(p, "dev %u ", g->tc_ifindex); + + dp_dump(p, "handle %s parent %s", + rtnl_tc_handle2str(g->tc_handle, handle, sizeof(handle)), + rtnl_tc_handle2str(g->tc_parent, parent, sizeof(parent))); + + return 1; +} + +int tca_dump_full(struct rtnl_tca *g, struct nl_dump_params *p, int line) +{ + dp_dump_line(p, line++, " "); + return line; +} + +int tca_dump_stats(struct rtnl_tca *g, struct nl_dump_params *p, int line) +{ + char *unit, fmt[64]; + float res; + strcpy(fmt, " %7.2f %s %10u %10u %10u %10u %10u\n"); + + dp_dump_line(p, line++, + " Stats: bytes packets drops overlimits" \ + " qlen backlog\n"); + + res = nl_cancel_down_bytes(g->tc_stats[RTNL_TC_BYTES], &unit); + if (*unit == 'B') + fmt[11] = '9'; + + dp_dump_line(p, line++, fmt, res, unit, + g->tc_stats[RTNL_TC_PACKETS], + g->tc_stats[RTNL_TC_DROPS], + g->tc_stats[RTNL_TC_OVERLIMITS], + g->tc_stats[RTNL_TC_QLEN], + g->tc_stats[RTNL_TC_BACKLOG]); + + res = nl_cancel_down_bytes(g->tc_stats[RTNL_TC_RATE_BPS], &unit); + + strcpy(fmt, " %7.2f %s/s%9u pps"); + + if (*unit == 'B') + fmt[11] = '9'; + + dp_dump_line(p, line++, fmt, res, unit, g->tc_stats[RTNL_TC_RATE_PPS]); + + return line; +} + +int tca_compare(struct nl_object *_a, struct nl_object *_b, + uint32_t attrs, int flags) +{ + struct rtnl_tca *a = (struct rtnl_tca *) _a; + struct rtnl_tca *b = (struct rtnl_tca *) _b; + int diff = 0; + +#define TC_DIFF(ATTR, EXPR) ATTR_DIFF(attrs, TCA_ATTR_##ATTR, a, b, EXPR) + + diff |= TC_DIFF(HANDLE, a->tc_handle != b->tc_handle); + diff |= TC_DIFF(PARENT, a->tc_parent != b->tc_parent); + diff |= TC_DIFF(IFINDEX, a->tc_ifindex != b->tc_ifindex); + diff |= TC_DIFF(KIND, strcmp(a->tc_kind, b->tc_kind)); + +#undef TC_DIFF + + return diff; +} + +void tca_set_ifindex(struct rtnl_tca *t, int ifindex) +{ + t->tc_ifindex = ifindex; + t->ce_mask |= TCA_ATTR_IFINDEX; +} + +int tca_get_ifindex(struct rtnl_tca *t) +{ + if (t->ce_mask & TCA_ATTR_IFINDEX) + return t->tc_ifindex; + else + return RTNL_LINK_NOT_FOUND; +} + +void tca_set_handle(struct rtnl_tca *t, uint32_t handle) +{ + t->tc_handle = handle; + t->ce_mask |= TCA_ATTR_HANDLE; +} + +uint32_t tca_get_handle(struct rtnl_tca *t) +{ + if (t->ce_mask & TCA_ATTR_HANDLE) + return t->tc_handle; + else + return 0; +} + +void tca_set_parent(struct rtnl_tca *t, uint32_t parent) +{ + t->tc_parent = parent; + t->ce_mask |= TCA_ATTR_PARENT; +} + +uint32_t tca_get_parent(struct rtnl_tca *t) +{ + if (t->ce_mask & TCA_ATTR_PARENT) + return t->tc_parent; + else + return 0; +} + +void tca_set_kind(struct rtnl_tca *t, const char *kind) +{ + strncpy(t->tc_kind, kind, sizeof(t->tc_kind) - 1); + t->ce_mask |= TCA_ATTR_KIND; +} + +char *tca_get_kind(struct rtnl_tca *t) +{ + if (t->ce_mask & TCA_ATTR_KIND) + return t->tc_kind; + else + return NULL; +} + +uint64_t tca_get_stat(struct rtnl_tca *t, int id) +{ + if (id < 0 || id > RTNL_TC_STATS_MAX) + return 0; + + return t->tc_stats[id]; +} + +struct nl_msg *tca_build_msg(struct rtnl_tca *tca, int type, int flags) +{ + struct nl_msg *msg; + struct tcmsg tchdr = { + .tcm_family = AF_UNSPEC, + .tcm_ifindex = tca->tc_ifindex, + .tcm_handle = tca->tc_handle, + .tcm_parent = tca->tc_parent, + }; + + msg = nlmsg_alloc_simple(type, flags); + if (!msg) + goto nla_put_failure; + + if (nlmsg_append(msg, &tchdr, sizeof(tchdr), NLMSG_ALIGNTO) < 0) + goto nla_put_failure; + + if (tca->ce_mask & TCA_ATTR_KIND) + NLA_PUT_STRING(msg, TCA_KIND, tca->tc_kind); + + return msg; + +nla_put_failure: + nlmsg_free(msg); + return NULL; +} + +/** @endcond */ + +/** + * @name Utilities + * @{ + */ + +/** + * Calculate time required to transmit buffer at a specific rate + * @arg bufsize Size of buffer to be transmited in bytes. + * @arg rate Transmit rate in bytes per second. + * + * Calculates the number of micro seconds required to transmit a + * specific buffer at a specific transmit rate. + * + * @f[ + * txtime=\frac{bufsize}{rate}10^6 + * @f] + * + * @return Required transmit time in micro seconds. + */ +int rtnl_tc_calc_txtime(int bufsize, int rate) +{ + double tx_time_secs; + + tx_time_secs = (double) bufsize / (double) rate; + + return tx_time_secs * 1000000.; +} + +/** + * Calculate buffer size able to transmit in a specific time and rate. + * @arg txtime Available transmit time in micro seconds. + * @arg rate Transmit rate in bytes per second. + * + * Calculates the size of the buffer that can be transmitted in a + * specific time period at a specific transmit rate. + * + * @f[ + * bufsize=\frac{{txtime} \times {rate}}{10^6} + * @f] + * + * @return Size of buffer in bytes. + */ +int rtnl_tc_calc_bufsize(int txtime, int rate) +{ + double bufsize; + + bufsize = (double) txtime * (double) rate; + + return bufsize / 1000000.; +} + +/** + * Calculate the binary logarithm for a specific cell size + * @arg cell_size Size of cell, must be a power of two. + * @return Binary logirhtm of cell size or a negative error code. + */ +int rtnl_tc_calc_cell_log(int cell_size) +{ + int i; + + for (i = 0; i < 32; i++) + if ((1 << i) == cell_size) + return i; + + return nl_errno(EINVAL); +} + + +/** @} */ + +/** + * @name Rate Tables + * @{ + */ + +/** + * Compute a transmission time lookup table + * @arg dst Destination buffer of RTNL_TC_RTABLE_SIZE uint32_t[]. + * @arg mpu Minimal size of a packet at all times. + * @arg overhead Overhead to be added to each packet. + * @arg cell Size of cell, i.e. size of step between entries in bytes. + * @arg rate Rate in bytes per second. + * + * Computes a table of RTNL_TC_RTABLE_SIZE entries specyfing the + * transmission times for various packet sizes, e.g. the transmission + * time for a packet of size \c pktsize could be looked up: + * @code + * txtime = table[pktsize >> log2(cell)]; + * @endcode + */ +int rtnl_tc_build_rate_table(uint32_t *dst, uint8_t mpu, uint8_t overhead, + int cell, int rate) +{ + int i, size, cell_log; + + cell_log = rtnl_tc_calc_cell_log(cell); + if (cell_log < 0) + return cell_log; + + for (i = 0; i < RTNL_TC_RTABLE_SIZE; i++) { + size = (i << cell_log) + overhead; + if (size < mpu) + size = mpu; + + dst[i] = rtnl_tc_calc_txtime(size, rate); + } + + return 0; +} + +/** @} */ + +/** + * @name Traffic Control Handle Translations + * @{ + */ + +/** + * Convert a traffic control handle to a character string (Reentrant). + * @arg handle traffic control handle + * @arg buf destination buffer + * @arg len buffer length + * + * Converts a tarffic control handle to a character string in the + * form of \c MAJ:MIN and stores it in the specified destination buffer. + * + * @return The destination buffer or the type encoded in hexidecimal + * form if no match was found. + */ +char * rtnl_tc_handle2str(uint32_t handle, char *buf, size_t len) +{ + if (TC_H_ROOT == handle) + snprintf(buf, len, "root"); + else if (TC_H_UNSPEC == handle) + snprintf(buf, len, "none"); + else if (0 == TC_H_MAJ(handle)) + snprintf(buf, len, ":%02x", TC_H_MIN(handle)); + else if (0 == TC_H_MIN(handle)) + snprintf(buf, len, "%02x:", TC_H_MAJ(handle) >> 16); + else + snprintf(buf, len, "%02x:%02x", + TC_H_MAJ(handle) >> 16, TC_H_MIN(handle)); + + return buf; +} + +/** + * Convert a charactering strint to a traffic control handle + * @arg name traffic control handle as character string + * @arg res destination buffer + * + * Converts the provided character string specifying a traffic + * control handle to the corresponding numeric value. + * + * The handle must be provided in one of the following formats: + * - root + * - none + * - XXXX: + * - :YYYY + * - XXXX:YYYY + * - XXXXYYYY + * + * @return 0 on success or a negative error code + */ +int rtnl_tc_str2handle(const char *name, uint32_t *res) +{ + char *colon, *end; + uint32_t h; + + if (!strcasecmp(name, "root")) { + *res = TC_H_ROOT; + return 0; + } + + if (!strcasecmp(name, "none")) { + *res = TC_H_UNSPEC; + return 0; + } + + h = strtoul(name, &colon, 16); + + if (colon == name) { + /* :YYYY */ + h = 0; + if (':' != *colon) + return -EINVAL; + } + + if (':' == *colon) { + /* check if we would lose bits */ + if (TC_H_MAJ(h)) + return -ERANGE; + h <<= 16; + + if ('\0' == colon[1]) { + /* XXXX: */ + *res = h; + } else { + /* XXXX:YYYY */ + uint32_t l = strtoul(colon+1, &end, 16); + + /* check if we overlap with major part */ + if (TC_H_MAJ(l)) + return -ERANGE; + + if ('\0' != *end) + return -EINVAL; + + *res = (h | l); + } + } else if ('\0' == *colon) { + /* XXXXYYYY */ + *res = h; + } else + return -EINVAL; + + return 0; +} + +/** @} */ + +/** @} */ |