21 #include <netlink-private/netlink.h>
22 #include <netlink-private/tc.h>
23 #include <netlink/netlink.h>
24 #include <netlink/attr.h>
25 #include <netlink/utils.h>
26 #include <netlink-private/route/tc-api.h>
27 #include <netlink/route/classifier.h>
28 #include <netlink/route/cls/u32.h>
29 #include <netlink/route/action.h>
32 #define U32_ATTR_DIVISOR 0x001
33 #define U32_ATTR_HASH 0x002
34 #define U32_ATTR_CLASSID 0x004
35 #define U32_ATTR_LINK 0x008
36 #define U32_ATTR_PCNT 0x010
37 #define U32_ATTR_SELECTOR 0x020
38 #define U32_ATTR_ACTION 0x040
39 #define U32_ATTR_POLICE 0x080
40 #define U32_ATTR_INDEV 0x100
43 static inline struct tc_u32_sel *u32_selector(
struct rtnl_u32 *u)
45 return (
struct tc_u32_sel *) u->cu_selector->d_data;
48 static inline struct tc_u32_sel *u32_selector_alloc(
struct rtnl_u32 *u)
51 u->cu_selector =
nl_data_alloc(NULL,
sizeof(
struct tc_u32_sel));
53 return u32_selector(u);
56 static struct nla_policy u32_policy[TCA_U32_MAX+1] = {
58 [TCA_U32_HASH] = { .type =
NLA_U32 },
59 [TCA_U32_CLASSID] = { .type =
NLA_U32 },
60 [TCA_U32_LINK] = { .type =
NLA_U32 },
63 [TCA_U32_SEL] = { .minlen =
sizeof(
struct tc_u32_sel) },
64 [TCA_U32_PCNT] = { .minlen =
sizeof(
struct tc_u32_pcnt) },
67 static int u32_msg_parser(
struct rtnl_tc *tc,
void *data)
69 struct rtnl_u32 *u = data;
70 struct nlattr *tb[TCA_U32_MAX + 1];
73 err = tca_parse(tb, TCA_U32_MAX, tc, u32_policy);
77 if (tb[TCA_U32_DIVISOR]) {
79 u->cu_mask |= U32_ATTR_DIVISOR;
82 if (tb[TCA_U32_SEL]) {
86 u->cu_mask |= U32_ATTR_SELECTOR;
89 if (tb[TCA_U32_HASH]) {
91 u->cu_mask |= U32_ATTR_HASH;
94 if (tb[TCA_U32_CLASSID]) {
96 u->cu_mask |= U32_ATTR_CLASSID;
99 if (tb[TCA_U32_LINK]) {
101 u->cu_mask |= U32_ATTR_LINK;
104 if (tb[TCA_U32_ACT]) {
105 u->cu_mask |= U32_ATTR_ACTION;
106 err = rtnl_act_parse(&u->cu_act, tb[TCA_U32_ACT]);
111 if (tb[TCA_U32_POLICE]) {
115 u->cu_mask |= U32_ATTR_POLICE;
118 if (tb[TCA_U32_PCNT]) {
119 struct tc_u32_sel *sel;
122 if (!tb[TCA_U32_SEL]) {
123 err = -NLE_MISSING_ATTR;
127 sel = u->cu_selector->d_data;
128 pcnt_size =
sizeof(
struct tc_u32_pcnt) +
129 (sel->nkeys *
sizeof(uint64_t));
130 if (
nla_len(tb[TCA_U32_PCNT]) < pcnt_size) {
138 u->cu_mask |= U32_ATTR_PCNT;
141 if (tb[TCA_U32_INDEV]) {
142 nla_strlcpy(u->cu_indev, tb[TCA_U32_INDEV], IFNAMSIZ);
143 u->cu_mask |= U32_ATTR_INDEV;
154 static void u32_free_data(
struct rtnl_tc *tc,
void *data)
156 struct rtnl_u32 *u = data;
159 rtnl_act_put_all(&u->cu_act);
165 static int u32_clone(
void *_dst,
void *_src)
167 struct rtnl_u32 *dst = _dst, *src = _src;
169 if (src->cu_selector &&
174 if (!(dst->cu_act = rtnl_act_alloc()))
177 memcpy(dst->cu_act, src->cu_act,
sizeof(
struct rtnl_act));
180 if (src->cu_police && !(dst->cu_police =
nl_data_clone(src->cu_police)))
183 if (src->cu_pcnt && !(dst->cu_pcnt =
nl_data_clone(src->cu_pcnt)))
189 static void u32_dump_line(
struct rtnl_tc *tc,
void *data,
192 struct rtnl_u32 *u = data;
198 if (u->cu_mask & U32_ATTR_DIVISOR)
199 nl_dump(p,
" divisor %u", u->cu_divisor);
200 else if (u->cu_mask & U32_ATTR_CLASSID)
205 static void print_selector(
struct nl_dump_params *p,
struct tc_u32_sel *sel,
209 struct tc_u32_key *key;
211 if (sel->hmask || sel->hoff) {
216 nl_dump(p,
" hash at %u & 0x%x", sel->hoff, sel->hmask);
219 if (sel->flags & (TC_U32_OFFSET | TC_U32_VAROFFSET)) {
220 nl_dump(p,
" offset at %u", sel->off);
222 if (sel->flags & TC_U32_VAROFFSET)
223 nl_dump(p,
" variable (at %u & 0x%x) >> %u",
224 sel->offoff, ntohs(sel->offmask), sel->offshift);
228 int flags = sel->flags;
231 #define PRINT_FLAG(f) if (flags & TC_U32_##f) { \
232 flags &= ~TC_U32_##f; nl_dump(p, #f "%s", flags ? "," : ""); }
234 PRINT_FLAG(TERMINAL);
236 PRINT_FLAG(VAROFFSET);
244 for (i = 0; i < sel->nkeys; i++) {
245 key = (
struct tc_u32_key *) ((
char *) sel +
sizeof(*sel)) + i;
248 nl_dump_line(p,
" match key at %s%u ",
249 key->offmask ?
"nexthdr+" :
"", key->off);
252 nl_dump(p,
"[0x%u] ", key->offmask);
254 nl_dump(p,
"& 0x%08x == 0x%08x", ntohl(key->mask), ntohl(key->val));
257 (u->cu_mask & U32_ATTR_PCNT)) {
258 struct tc_u32_pcnt *pcnt = u->cu_pcnt->d_data;
259 nl_dump(p,
" successful %" PRIu64, pcnt->kcnts[i]);
264 static void u32_dump_details(
struct rtnl_tc *tc,
void *data,
267 struct rtnl_u32 *u = data;
268 struct tc_u32_sel *s;
273 if (!(u->cu_mask & U32_ATTR_SELECTOR)) {
278 s = u->cu_selector->d_data;
280 nl_dump(p,
"nkeys %u ", s->nkeys);
282 if (u->cu_mask & U32_ATTR_HASH)
283 nl_dump(p,
"ht key 0x%x hash 0x%u",
284 TC_U32_USERHTID(u->cu_hash), TC_U32_HASH(u->cu_hash));
286 if (u->cu_mask & U32_ATTR_LINK)
287 nl_dump(p,
"link %u ", u->cu_link);
289 if (u->cu_mask & U32_ATTR_INDEV)
290 nl_dump(p,
"indev %s ", u->cu_indev);
292 print_selector(p, s, u);
296 static void u32_dump_stats(
struct rtnl_tc *tc,
void *data,
299 struct rtnl_u32 *u = data;
304 if (u->cu_mask & U32_ATTR_PCNT) {
305 struct tc_u32_pcnt *pc = u->cu_pcnt->d_data;
307 nl_dump_line(p,
" hit %8" PRIu64
" count %8" PRIu64
"\n",
312 static int u32_msg_fill(
struct rtnl_tc *tc,
void *data,
struct nl_msg *msg)
314 struct rtnl_u32 *u = data;
319 if (u->cu_mask & U32_ATTR_DIVISOR)
322 if (u->cu_mask & U32_ATTR_HASH)
325 if (u->cu_mask & U32_ATTR_CLASSID)
328 if (u->cu_mask & U32_ATTR_LINK)
331 if (u->cu_mask & U32_ATTR_SELECTOR)
334 if (u->cu_mask & U32_ATTR_ACTION) {
337 err = rtnl_act_fill(msg, TCA_U32_ACT, u->cu_act);
342 if (u->cu_mask & U32_ATTR_POLICE)
345 if (u->cu_mask & U32_ATTR_INDEV)
359 void rtnl_u32_set_handle(
struct rtnl_cls *cls,
int htid,
int hash,
362 uint32_t handle = (htid << 20) | (hash << 12) | nodeid;
367 int rtnl_u32_set_classid(
struct rtnl_cls *cls, uint32_t classid)
374 u->cu_classid = classid;
375 u->cu_mask |= U32_ATTR_CLASSID;
380 int rtnl_u32_set_divisor(
struct rtnl_cls *cls, uint32_t divisor)
387 u->cu_divisor = divisor;
388 u->cu_mask |= U32_ATTR_DIVISOR;
392 int rtnl_u32_set_link(
struct rtnl_cls *cls, uint32_t link)
400 u->cu_mask |= U32_ATTR_LINK;
404 int rtnl_u32_set_hashtable(
struct rtnl_cls *cls, uint32_t ht)
412 u->cu_mask |= U32_ATTR_HASH;
416 int rtnl_u32_set_hashmask(
struct rtnl_cls *cls, uint32_t hashmask, uint32_t offset)
419 struct tc_u32_sel *sel;
422 hashmask = htonl(hashmask);
427 sel = u32_selector_alloc(u);
431 err =
nl_data_append(u->cu_selector, NULL,
sizeof(
struct tc_u32_key));
435 sel = u32_selector(u);
437 sel->hmask = hashmask;
442 int rtnl_u32_set_cls_terminal(
struct rtnl_cls *cls)
445 struct tc_u32_sel *sel;
451 sel = u32_selector_alloc(u);
455 err =
nl_data_append(u->cu_selector, NULL,
sizeof(
struct tc_u32_key));
459 sel = u32_selector(u);
461 sel->flags |= TC_U32_TERMINAL;
465 int rtnl_u32_add_action(
struct rtnl_cls *cls,
struct rtnl_act *act)
475 u->cu_mask |= U32_ATTR_ACTION;
478 return rtnl_act_append(&u->cu_act, act);
481 int rtnl_u32_del_action(
struct rtnl_cls *cls,
struct rtnl_act *act)
492 if (!(u->cu_mask & U32_ATTR_ACTION))
495 ret = rtnl_act_remove(&u->cu_act, act);
500 u->cu_mask &= ~U32_ATTR_ACTION;
511 int rtnl_u32_set_flags(
struct rtnl_cls *cls,
int flags)
513 struct tc_u32_sel *sel;
519 sel = u32_selector_alloc(u);
524 u->cu_mask |= U32_ATTR_SELECTOR;
544 int off,
int offmask)
546 struct tc_u32_sel *sel;
553 sel = u32_selector_alloc(u);
557 err =
nl_data_append(u->cu_selector, NULL,
sizeof(
struct tc_u32_key));
562 sel = u32_selector(u);
564 sel->keys[sel->nkeys].mask = mask;
565 sel->keys[sel->nkeys].val = val & mask;
566 sel->keys[sel->nkeys].off = off;
567 sel->keys[sel->nkeys].offmask = offmask;
569 u->cu_mask |= U32_ATTR_SELECTOR;
586 uint32_t *val, uint32_t *mask,
int *off,
int *offmask)
588 struct tc_u32_sel *sel;
594 if (!(u->cu_mask & U32_ATTR_SELECTOR))
598 sel = u32_selector(u);
599 if (index >= sel->nkeys)
602 *mask = sel->keys[index].mask;
603 *val = sel->keys[index].val;
604 *off = sel->keys[index].off;
605 *offmask = sel->keys[index].offmask;
610 int rtnl_u32_add_key_uint8(
struct rtnl_cls *cls, uint8_t val, uint8_t mask,
611 int off,
int offmask)
613 int shift = 24 - 8 * (off & 3);
616 htonl((uint32_t)mask << shift),
630 int off,
int offmask)
632 int shift = ((off & 3) == 0 ? 16 : 0);
637 htonl((uint32_t)mask << shift),
651 int off,
int offmask)
657 int rtnl_u32_add_key_in_addr(
struct rtnl_cls *cls,
const struct in_addr *addr,
658 uint8_t bitmask,
int off,
int offmask)
660 uint32_t mask = 0xFFFFFFFF << (32 - bitmask);
664 int rtnl_u32_add_key_in6_addr(
struct rtnl_cls *cls,
const struct in6_addr *addr,
665 uint8_t bitmask,
int off,
int offmask)
669 for (i = 1; i <= 4; i++) {
670 if (32 * i - bitmask <= 0) {
672 0xFFFFFFFF, off+4*(i-1), offmask)) < 0)
675 else if (32 * i - bitmask < 32) {
676 uint32_t mask = 0xFFFFFFFF << (32 * i - bitmask);
678 htonl(mask), off+4*(i-1), offmask)) < 0)
689 static struct rtnl_tc_ops u32_ops = {
691 .to_type = RTNL_TC_TYPE_CLS,
692 .to_size =
sizeof(
struct rtnl_u32),
693 .to_msg_parser = u32_msg_parser,
694 .to_free_data = u32_free_data,
695 .to_clone = u32_clone,
696 .to_msg_fill = u32_msg_fill,
704 static void __init u32_init(
void)
709 static void __exit u32_exit(
void)
Dump object briefly on one line.
int rtnl_tc_register(struct rtnl_tc_ops *ops)
Register a traffic control module.
struct nl_data * nl_data_alloc_attr(struct nlattr *nla)
Allocate abstract data object based on netlink attribute.
struct nl_data * nl_data_clone(struct nl_data *src)
Clone an abstract data object.
Attribute validation policy.
#define NLA_PUT_DATA(msg, attrtype, data)
Add abstract data attribute to netlink message.
enum nl_dump_type dp_type
Specifies the type of dump that is requested.
int rtnl_u32_add_key_uint32(struct rtnl_cls *cls, uint32_t val, uint32_t mask, int off, int offmask)
Append new selector key to match a 32-bit number.
int rtnl_u32_add_key(struct rtnl_cls *cls, uint32_t val, uint32_t mask, int off, int offmask)
Append new 32-bit key to the selector.
int nl_data_append(struct nl_data *data, void *buf, size_t size)
Append data to an abstract data object.
NUL terminated character string.
Dump all attributes but no statistics.
void rtnl_tc_unregister(struct rtnl_tc_ops *ops)
Unregister a traffic control module.
int rtnl_u32_get_key(struct rtnl_cls *cls, uint8_t index, uint32_t *val, uint32_t *mask, int *off, int *offmask)
Get the 32-bit key from the selector.
#define TC_CAST(ptr)
Macro to cast qdisc/class/classifier to tc object.
int rtnl_u32_add_key_uint16(struct rtnl_cls *cls, uint16_t val, uint16_t mask, int off, int offmask)
Append new selector key to match a 16-bit number.
#define NLA_PUT_U32(msg, attrtype, value)
Add 32 bit integer attribute to netlink message.
int nla_len(const struct nlattr *nla)
Return length of the payload .
void rtnl_tc_set_handle(struct rtnl_tc *tc, uint32_t id)
Set identifier of traffic control object.
struct nl_data * nl_data_alloc(void *buf, size_t size)
Allocate a new abstract data object.
void * rtnl_tc_data(struct rtnl_tc *tc)
Return pointer to private data of traffic control object.
#define NLA_PUT_STRING(msg, attrtype, value)
Add string attribute to netlink message.
uint16_t type
Type of attribute or NLA_UNSPEC.
char * rtnl_tc_handle2str(uint32_t handle, char *buf, size_t len)
Convert a traffic control handle to a character string (Reentrant).
void nl_dump(struct nl_dump_params *params, const char *fmt,...)
Dump a formatted character string.
uint32_t nla_get_u32(struct nlattr *nla)
Return payload of 32 bit integer attribute.
Dump all attributes including statistics.
size_t nla_strlcpy(char *dst, const struct nlattr *nla, size_t dstsize)
Copy string attribute payload to a buffer.
void nl_data_free(struct nl_data *data)
Free an abstract data object.