blob: e39672394c7b64dc28617878bc7b26ef1ef8abb9 [file] [log] [blame]
Patrick McHardye5dfb812008-01-31 18:37:42 -08001/*
2 * net/sched/cls_flow.c Generic flow classifier
3 *
4 * Copyright (c) 2007, 2008 Patrick McHardy <kaber@trash.net>
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version 2
9 * of the License, or (at your option) any later version.
10 */
11
12#include <linux/kernel.h>
13#include <linux/init.h>
14#include <linux/list.h>
15#include <linux/jhash.h>
16#include <linux/random.h>
17#include <linux/pkt_cls.h>
18#include <linux/skbuff.h>
19#include <linux/in.h>
20#include <linux/ip.h>
21#include <linux/ipv6.h>
Patrick McHardy9ec13812008-02-05 16:21:04 -080022#include <linux/if_vlan.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +090023#include <linux/slab.h>
Paul Gortmaker3a9a2312011-05-27 09:12:25 -040024#include <linux/module.h>
Eric Dumazet743b2a62015-11-08 10:54:10 -080025#include <net/inet_sock.h>
Patrick McHardye5dfb812008-01-31 18:37:42 -080026
27#include <net/pkt_cls.h>
28#include <net/ip.h>
29#include <net/route.h>
Jiri Pirko1bd758e2015-05-12 14:56:07 +020030#include <net/flow_dissector.h>
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +000031
Javier Martinez Canillas0013de32016-09-09 08:43:18 -040032#if IS_ENABLED(CONFIG_NF_CONNTRACK)
Patrick McHardye5dfb812008-01-31 18:37:42 -080033#include <net/netfilter/nf_conntrack.h>
34#endif
35
36struct flow_head {
37 struct list_head filters;
John Fastabend70da9f02014-09-12 20:06:55 -070038 struct rcu_head rcu;
Patrick McHardye5dfb812008-01-31 18:37:42 -080039};
40
41struct flow_filter {
42 struct list_head list;
43 struct tcf_exts exts;
44 struct tcf_ematch_tree ematches;
John Fastabend70da9f02014-09-12 20:06:55 -070045 struct tcf_proto *tp;
Patrick McHardy72d97942008-07-14 20:36:32 -070046 struct timer_list perturb_timer;
47 u32 perturb_period;
Patrick McHardye5dfb812008-01-31 18:37:42 -080048 u32 handle;
49
50 u32 nkeys;
51 u32 keymask;
52 u32 mode;
53 u32 mask;
54 u32 xor;
55 u32 rshift;
56 u32 addend;
57 u32 divisor;
58 u32 baseclass;
Patrick McHardy72d97942008-07-14 20:36:32 -070059 u32 hashrnd;
John Fastabend70da9f02014-09-12 20:06:55 -070060 struct rcu_head rcu;
Patrick McHardye5dfb812008-01-31 18:37:42 -080061};
62
Patrick McHardye5dfb812008-01-31 18:37:42 -080063static inline u32 addr_fold(void *addr)
64{
65 unsigned long a = (unsigned long)addr;
66
67 return (a & 0xFFFFFFFF) ^ (BITS_PER_LONG > 32 ? a >> 32 : 0);
68}
69
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +000070static u32 flow_get_src(const struct sk_buff *skb, const struct flow_keys *flow)
Patrick McHardye5dfb812008-01-31 18:37:42 -080071{
Tom Herbertc3f83242015-06-04 09:16:40 -070072 __be32 src = flow_get_u32_src(flow);
73
74 if (src)
75 return ntohl(src);
76
Changli Gao4b95c3d2010-08-04 04:48:12 +000077 return addr_fold(skb->sk);
Patrick McHardye5dfb812008-01-31 18:37:42 -080078}
79
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +000080static u32 flow_get_dst(const struct sk_buff *skb, const struct flow_keys *flow)
Patrick McHardye5dfb812008-01-31 18:37:42 -080081{
Tom Herbertc3f83242015-06-04 09:16:40 -070082 __be32 dst = flow_get_u32_dst(flow);
83
84 if (dst)
85 return ntohl(dst);
86
Jiri Pirkod8b96052015-01-13 17:13:43 +010087 return addr_fold(skb_dst(skb)) ^ (__force u16) tc_skb_protocol(skb);
Patrick McHardye5dfb812008-01-31 18:37:42 -080088}
89
Jamal Hadi Salim5a7a5552016-09-18 08:45:33 -040090static u32 flow_get_proto(const struct sk_buff *skb,
91 const struct flow_keys *flow)
Patrick McHardye5dfb812008-01-31 18:37:42 -080092{
Jiri Pirko06635a32015-05-12 14:56:16 +020093 return flow->basic.ip_proto;
Patrick McHardye5dfb812008-01-31 18:37:42 -080094}
95
Jamal Hadi Salim5a7a5552016-09-18 08:45:33 -040096static u32 flow_get_proto_src(const struct sk_buff *skb,
97 const struct flow_keys *flow)
Patrick McHardye5dfb812008-01-31 18:37:42 -080098{
Jiri Pirko06635a32015-05-12 14:56:16 +020099 if (flow->ports.ports)
Jiri Pirko59346af2015-05-12 14:56:20 +0200100 return ntohs(flow->ports.src);
Eric Dumazet859c2012011-10-23 17:59:41 +0000101
Changli Gao4b95c3d2010-08-04 04:48:12 +0000102 return addr_fold(skb->sk);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800103}
104
Jamal Hadi Salim5a7a5552016-09-18 08:45:33 -0400105static u32 flow_get_proto_dst(const struct sk_buff *skb,
106 const struct flow_keys *flow)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800107{
Jiri Pirko06635a32015-05-12 14:56:16 +0200108 if (flow->ports.ports)
Jiri Pirko59346af2015-05-12 14:56:20 +0200109 return ntohs(flow->ports.dst);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800110
Jiri Pirkod8b96052015-01-13 17:13:43 +0100111 return addr_fold(skb_dst(skb)) ^ (__force u16) tc_skb_protocol(skb);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800112}
113
114static u32 flow_get_iif(const struct sk_buff *skb)
115{
Eric Dumazet8964be42009-11-20 15:35:04 -0800116 return skb->skb_iif;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800117}
118
119static u32 flow_get_priority(const struct sk_buff *skb)
120{
121 return skb->priority;
122}
123
124static u32 flow_get_mark(const struct sk_buff *skb)
125{
126 return skb->mark;
127}
128
129static u32 flow_get_nfct(const struct sk_buff *skb)
130{
Javier Martinez Canillas0013de32016-09-09 08:43:18 -0400131#if IS_ENABLED(CONFIG_NF_CONNTRACK)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800132 return addr_fold(skb->nfct);
133#else
134 return 0;
135#endif
136}
137
Javier Martinez Canillas0013de32016-09-09 08:43:18 -0400138#if IS_ENABLED(CONFIG_NF_CONNTRACK)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800139#define CTTUPLE(skb, member) \
140({ \
141 enum ip_conntrack_info ctinfo; \
Eric Dumazet859c2012011-10-23 17:59:41 +0000142 const struct nf_conn *ct = nf_ct_get(skb, &ctinfo); \
Patrick McHardye5dfb812008-01-31 18:37:42 -0800143 if (ct == NULL) \
144 goto fallback; \
145 ct->tuplehash[CTINFO2DIR(ctinfo)].tuple.member; \
146})
147#else
148#define CTTUPLE(skb, member) \
149({ \
150 goto fallback; \
151 0; \
152})
153#endif
154
Jamal Hadi Salim5a7a5552016-09-18 08:45:33 -0400155static u32 flow_get_nfct_src(const struct sk_buff *skb,
156 const struct flow_keys *flow)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800157{
Jiri Pirkod8b96052015-01-13 17:13:43 +0100158 switch (tc_skb_protocol(skb)) {
Arnaldo Carvalho de Melo60678042008-09-20 22:20:49 -0700159 case htons(ETH_P_IP):
Patrick McHardye5dfb812008-01-31 18:37:42 -0800160 return ntohl(CTTUPLE(skb, src.u3.ip));
Arnaldo Carvalho de Melo60678042008-09-20 22:20:49 -0700161 case htons(ETH_P_IPV6):
Patrick McHardye5dfb812008-01-31 18:37:42 -0800162 return ntohl(CTTUPLE(skb, src.u3.ip6[3]));
163 }
164fallback:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000165 return flow_get_src(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800166}
167
Jamal Hadi Salim5a7a5552016-09-18 08:45:33 -0400168static u32 flow_get_nfct_dst(const struct sk_buff *skb,
169 const struct flow_keys *flow)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800170{
Jiri Pirkod8b96052015-01-13 17:13:43 +0100171 switch (tc_skb_protocol(skb)) {
Arnaldo Carvalho de Melo60678042008-09-20 22:20:49 -0700172 case htons(ETH_P_IP):
Patrick McHardye5dfb812008-01-31 18:37:42 -0800173 return ntohl(CTTUPLE(skb, dst.u3.ip));
Arnaldo Carvalho de Melo60678042008-09-20 22:20:49 -0700174 case htons(ETH_P_IPV6):
Patrick McHardye5dfb812008-01-31 18:37:42 -0800175 return ntohl(CTTUPLE(skb, dst.u3.ip6[3]));
176 }
177fallback:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000178 return flow_get_dst(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800179}
180
Jamal Hadi Salim5a7a5552016-09-18 08:45:33 -0400181static u32 flow_get_nfct_proto_src(const struct sk_buff *skb,
182 const struct flow_keys *flow)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800183{
184 return ntohs(CTTUPLE(skb, src.u.all));
185fallback:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000186 return flow_get_proto_src(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800187}
188
Jamal Hadi Salim5a7a5552016-09-18 08:45:33 -0400189static u32 flow_get_nfct_proto_dst(const struct sk_buff *skb,
190 const struct flow_keys *flow)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800191{
192 return ntohs(CTTUPLE(skb, dst.u.all));
193fallback:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000194 return flow_get_proto_dst(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800195}
196
197static u32 flow_get_rtclassid(const struct sk_buff *skb)
198{
Patrick McHardyc7066f72011-01-14 13:36:42 +0100199#ifdef CONFIG_IP_ROUTE_CLASSID
Eric Dumazetadf30902009-06-02 05:19:30 +0000200 if (skb_dst(skb))
201 return skb_dst(skb)->tclassid;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800202#endif
203 return 0;
204}
205
206static u32 flow_get_skuid(const struct sk_buff *skb)
207{
Eric Dumazet743b2a62015-11-08 10:54:10 -0800208 struct sock *sk = skb_to_full_sk(skb);
209
210 if (sk && sk->sk_socket && sk->sk_socket->file) {
211 kuid_t skuid = sk->sk_socket->file->f_cred->fsuid;
212
Eric W. Biedermana6c67962012-05-25 13:49:36 -0600213 return from_kuid(&init_user_ns, skuid);
214 }
Patrick McHardye5dfb812008-01-31 18:37:42 -0800215 return 0;
216}
217
218static u32 flow_get_skgid(const struct sk_buff *skb)
219{
Eric Dumazet743b2a62015-11-08 10:54:10 -0800220 struct sock *sk = skb_to_full_sk(skb);
221
222 if (sk && sk->sk_socket && sk->sk_socket->file) {
223 kgid_t skgid = sk->sk_socket->file->f_cred->fsgid;
224
Eric W. Biedermana6c67962012-05-25 13:49:36 -0600225 return from_kgid(&init_user_ns, skgid);
226 }
Patrick McHardye5dfb812008-01-31 18:37:42 -0800227 return 0;
228}
229
Patrick McHardy9ec13812008-02-05 16:21:04 -0800230static u32 flow_get_vlan_tag(const struct sk_buff *skb)
231{
232 u16 uninitialized_var(tag);
233
234 if (vlan_get_tag(skb, &tag) < 0)
235 return 0;
236 return tag & VLAN_VID_MASK;
237}
238
Changli Gao739a91e2010-08-21 06:23:15 +0000239static u32 flow_get_rxhash(struct sk_buff *skb)
240{
Tom Herbert3958afa1b2013-12-15 22:12:06 -0800241 return skb_get_hash(skb);
Changli Gao739a91e2010-08-21 06:23:15 +0000242}
243
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000244static u32 flow_key_get(struct sk_buff *skb, int key, struct flow_keys *flow)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800245{
246 switch (key) {
247 case FLOW_KEY_SRC:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000248 return flow_get_src(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800249 case FLOW_KEY_DST:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000250 return flow_get_dst(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800251 case FLOW_KEY_PROTO:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000252 return flow_get_proto(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800253 case FLOW_KEY_PROTO_SRC:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000254 return flow_get_proto_src(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800255 case FLOW_KEY_PROTO_DST:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000256 return flow_get_proto_dst(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800257 case FLOW_KEY_IIF:
258 return flow_get_iif(skb);
259 case FLOW_KEY_PRIORITY:
260 return flow_get_priority(skb);
261 case FLOW_KEY_MARK:
262 return flow_get_mark(skb);
263 case FLOW_KEY_NFCT:
264 return flow_get_nfct(skb);
265 case FLOW_KEY_NFCT_SRC:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000266 return flow_get_nfct_src(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800267 case FLOW_KEY_NFCT_DST:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000268 return flow_get_nfct_dst(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800269 case FLOW_KEY_NFCT_PROTO_SRC:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000270 return flow_get_nfct_proto_src(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800271 case FLOW_KEY_NFCT_PROTO_DST:
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000272 return flow_get_nfct_proto_dst(skb, flow);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800273 case FLOW_KEY_RTCLASSID:
274 return flow_get_rtclassid(skb);
275 case FLOW_KEY_SKUID:
276 return flow_get_skuid(skb);
277 case FLOW_KEY_SKGID:
278 return flow_get_skgid(skb);
Patrick McHardy9ec13812008-02-05 16:21:04 -0800279 case FLOW_KEY_VLAN_TAG:
280 return flow_get_vlan_tag(skb);
Changli Gao739a91e2010-08-21 06:23:15 +0000281 case FLOW_KEY_RXHASH:
282 return flow_get_rxhash(skb);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800283 default:
284 WARN_ON(1);
285 return 0;
286 }
287}
288
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000289#define FLOW_KEYS_NEEDED ((1 << FLOW_KEY_SRC) | \
290 (1 << FLOW_KEY_DST) | \
291 (1 << FLOW_KEY_PROTO) | \
292 (1 << FLOW_KEY_PROTO_SRC) | \
293 (1 << FLOW_KEY_PROTO_DST) | \
294 (1 << FLOW_KEY_NFCT_SRC) | \
295 (1 << FLOW_KEY_NFCT_DST) | \
296 (1 << FLOW_KEY_NFCT_PROTO_SRC) | \
297 (1 << FLOW_KEY_NFCT_PROTO_DST))
298
Eric Dumazetdc7f9f62011-07-05 23:25:42 +0000299static int flow_classify(struct sk_buff *skb, const struct tcf_proto *tp,
Patrick McHardye5dfb812008-01-31 18:37:42 -0800300 struct tcf_result *res)
301{
John Fastabend70da9f02014-09-12 20:06:55 -0700302 struct flow_head *head = rcu_dereference_bh(tp->root);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800303 struct flow_filter *f;
304 u32 keymask;
305 u32 classid;
306 unsigned int n, key;
307 int r;
308
John Fastabend70da9f02014-09-12 20:06:55 -0700309 list_for_each_entry_rcu(f, &head->filters, list) {
Eric Dumazet3a539432011-12-14 02:30:00 +0000310 u32 keys[FLOW_KEY_MAX + 1];
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000311 struct flow_keys flow_keys;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800312
313 if (!tcf_em_tree_match(skb, &f->ematches, NULL))
314 continue;
315
316 keymask = f->keymask;
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000317 if (keymask & FLOW_KEYS_NEEDED)
Tom Herbertcd79a232015-09-01 09:24:27 -0700318 skb_flow_dissect_flow_keys(skb, &flow_keys, 0);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800319
320 for (n = 0; n < f->nkeys; n++) {
321 key = ffs(keymask) - 1;
322 keymask &= ~(1 << key);
Eric Dumazet6bd2a9a2011-11-28 05:24:18 +0000323 keys[n] = flow_key_get(skb, key, &flow_keys);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800324 }
325
326 if (f->mode == FLOW_MODE_HASH)
Patrick McHardy72d97942008-07-14 20:36:32 -0700327 classid = jhash2(keys, f->nkeys, f->hashrnd);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800328 else {
329 classid = keys[0];
330 classid = (classid & f->mask) ^ f->xor;
331 classid = (classid >> f->rshift) + f->addend;
332 }
333
334 if (f->divisor)
335 classid %= f->divisor;
336
337 res->class = 0;
338 res->classid = TC_H_MAKE(f->baseclass, f->baseclass + classid);
339
340 r = tcf_exts_exec(skb, &f->exts, res);
341 if (r < 0)
342 continue;
343 return r;
344 }
345 return -1;
346}
347
Patrick McHardy72d97942008-07-14 20:36:32 -0700348static void flow_perturbation(unsigned long arg)
349{
350 struct flow_filter *f = (struct flow_filter *)arg;
351
352 get_random_bytes(&f->hashrnd, 4);
353 if (f->perturb_period)
354 mod_timer(&f->perturb_timer, jiffies + f->perturb_period);
355}
356
Patrick McHardye5dfb812008-01-31 18:37:42 -0800357static const struct nla_policy flow_policy[TCA_FLOW_MAX + 1] = {
358 [TCA_FLOW_KEYS] = { .type = NLA_U32 },
359 [TCA_FLOW_MODE] = { .type = NLA_U32 },
360 [TCA_FLOW_BASECLASS] = { .type = NLA_U32 },
361 [TCA_FLOW_RSHIFT] = { .type = NLA_U32 },
362 [TCA_FLOW_ADDEND] = { .type = NLA_U32 },
363 [TCA_FLOW_MASK] = { .type = NLA_U32 },
364 [TCA_FLOW_XOR] = { .type = NLA_U32 },
365 [TCA_FLOW_DIVISOR] = { .type = NLA_U32 },
366 [TCA_FLOW_ACT] = { .type = NLA_NESTED },
367 [TCA_FLOW_POLICE] = { .type = NLA_NESTED },
368 [TCA_FLOW_EMATCHES] = { .type = NLA_NESTED },
Patrick McHardy72d97942008-07-14 20:36:32 -0700369 [TCA_FLOW_PERTURB] = { .type = NLA_U32 },
Patrick McHardye5dfb812008-01-31 18:37:42 -0800370};
371
John Fastabend70da9f02014-09-12 20:06:55 -0700372static void flow_destroy_filter(struct rcu_head *head)
373{
374 struct flow_filter *f = container_of(head, struct flow_filter, rcu);
375
376 del_timer_sync(&f->perturb_timer);
WANG Cong18d02642014-09-25 10:26:37 -0700377 tcf_exts_destroy(&f->exts);
John Fastabend82a470f2014-10-05 21:27:53 -0700378 tcf_em_tree_destroy(&f->ematches);
John Fastabend70da9f02014-09-12 20:06:55 -0700379 kfree(f);
380}
381
Benjamin LaHaisec1b52732013-01-14 05:15:39 +0000382static int flow_change(struct net *net, struct sk_buff *in_skb,
Eric W. Biedermanaf4c6642012-05-25 13:42:45 -0600383 struct tcf_proto *tp, unsigned long base,
Patrick McHardye5dfb812008-01-31 18:37:42 -0800384 u32 handle, struct nlattr **tca,
Cong Wang2f7ef2f2014-04-25 13:54:06 -0700385 unsigned long *arg, bool ovr)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800386{
John Fastabend70da9f02014-09-12 20:06:55 -0700387 struct flow_head *head = rtnl_dereference(tp->root);
388 struct flow_filter *fold, *fnew;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800389 struct nlattr *opt = tca[TCA_OPTIONS];
390 struct nlattr *tb[TCA_FLOW_MAX + 1];
391 struct tcf_exts e;
392 struct tcf_ematch_tree t;
393 unsigned int nkeys = 0;
Patrick McHardy72d97942008-07-14 20:36:32 -0700394 unsigned int perturb_period = 0;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800395 u32 baseclass = 0;
396 u32 keymask = 0;
397 u32 mode;
398 int err;
399
400 if (opt == NULL)
401 return -EINVAL;
402
403 err = nla_parse_nested(tb, TCA_FLOW_MAX, opt, flow_policy);
404 if (err < 0)
405 return err;
406
407 if (tb[TCA_FLOW_BASECLASS]) {
408 baseclass = nla_get_u32(tb[TCA_FLOW_BASECLASS]);
409 if (TC_H_MIN(baseclass) == 0)
410 return -EINVAL;
411 }
412
413 if (tb[TCA_FLOW_KEYS]) {
414 keymask = nla_get_u32(tb[TCA_FLOW_KEYS]);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800415
416 nkeys = hweight32(keymask);
417 if (nkeys == 0)
418 return -EINVAL;
Patrick McHardy4f250492008-02-05 16:19:59 -0800419
420 if (fls(keymask) - 1 > FLOW_KEY_MAX)
421 return -EOPNOTSUPP;
Eric W. Biedermana6c67962012-05-25 13:49:36 -0600422
423 if ((keymask & (FLOW_KEY_SKUID|FLOW_KEY_SKGID)) &&
Patrick McHardye32123e2013-04-17 06:46:57 +0000424 sk_user_ns(NETLINK_CB(in_skb).sk) != &init_user_ns)
Eric W. Biedermana6c67962012-05-25 13:49:36 -0600425 return -EOPNOTSUPP;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800426 }
427
WANG Congb9a24bb2016-08-19 12:36:54 -0700428 err = tcf_exts_init(&e, TCA_FLOW_ACT, TCA_FLOW_POLICE);
429 if (err < 0)
430 goto err1;
Cong Wang2f7ef2f2014-04-25 13:54:06 -0700431 err = tcf_exts_validate(net, tp, tb, tca[TCA_RATE], &e, ovr);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800432 if (err < 0)
WANG Congb9a24bb2016-08-19 12:36:54 -0700433 goto err1;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800434
435 err = tcf_em_tree_validate(tp, tb[TCA_FLOW_EMATCHES], &t);
436 if (err < 0)
437 goto err1;
438
John Fastabend70da9f02014-09-12 20:06:55 -0700439 err = -ENOBUFS;
440 fnew = kzalloc(sizeof(*fnew), GFP_KERNEL);
441 if (!fnew)
442 goto err2;
443
WANG Congb9a24bb2016-08-19 12:36:54 -0700444 err = tcf_exts_init(&fnew->exts, TCA_FLOW_ACT, TCA_FLOW_POLICE);
445 if (err < 0)
446 goto err3;
Daniel Borkmann32b2f4b2015-07-17 22:38:45 +0200447
John Fastabend70da9f02014-09-12 20:06:55 -0700448 fold = (struct flow_filter *)*arg;
449 if (fold) {
Patrick McHardye5dfb812008-01-31 18:37:42 -0800450 err = -EINVAL;
John Fastabend70da9f02014-09-12 20:06:55 -0700451 if (fold->handle != handle && handle)
WANG Congb9a24bb2016-08-19 12:36:54 -0700452 goto err3;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800453
John Fastabend70da9f02014-09-12 20:06:55 -0700454 /* Copy fold into fnew */
John Fastabend70da9f02014-09-12 20:06:55 -0700455 fnew->tp = fold->tp;
John Fastabend70da9f02014-09-12 20:06:55 -0700456 fnew->handle = fold->handle;
457 fnew->nkeys = fold->nkeys;
458 fnew->keymask = fold->keymask;
459 fnew->mode = fold->mode;
460 fnew->mask = fold->mask;
461 fnew->xor = fold->xor;
462 fnew->rshift = fold->rshift;
463 fnew->addend = fold->addend;
464 fnew->divisor = fold->divisor;
465 fnew->baseclass = fold->baseclass;
466 fnew->hashrnd = fold->hashrnd;
467
468 mode = fold->mode;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800469 if (tb[TCA_FLOW_MODE])
470 mode = nla_get_u32(tb[TCA_FLOW_MODE]);
471 if (mode != FLOW_MODE_HASH && nkeys > 1)
WANG Congb9a24bb2016-08-19 12:36:54 -0700472 goto err3;
Patrick McHardy72d97942008-07-14 20:36:32 -0700473
474 if (mode == FLOW_MODE_HASH)
John Fastabend70da9f02014-09-12 20:06:55 -0700475 perturb_period = fold->perturb_period;
Patrick McHardy72d97942008-07-14 20:36:32 -0700476 if (tb[TCA_FLOW_PERTURB]) {
477 if (mode != FLOW_MODE_HASH)
WANG Congb9a24bb2016-08-19 12:36:54 -0700478 goto err3;
Patrick McHardy72d97942008-07-14 20:36:32 -0700479 perturb_period = nla_get_u32(tb[TCA_FLOW_PERTURB]) * HZ;
480 }
Patrick McHardye5dfb812008-01-31 18:37:42 -0800481 } else {
482 err = -EINVAL;
483 if (!handle)
WANG Congb9a24bb2016-08-19 12:36:54 -0700484 goto err3;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800485 if (!tb[TCA_FLOW_KEYS])
WANG Congb9a24bb2016-08-19 12:36:54 -0700486 goto err3;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800487
488 mode = FLOW_MODE_MAP;
489 if (tb[TCA_FLOW_MODE])
490 mode = nla_get_u32(tb[TCA_FLOW_MODE]);
491 if (mode != FLOW_MODE_HASH && nkeys > 1)
WANG Congb9a24bb2016-08-19 12:36:54 -0700492 goto err3;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800493
Patrick McHardy72d97942008-07-14 20:36:32 -0700494 if (tb[TCA_FLOW_PERTURB]) {
495 if (mode != FLOW_MODE_HASH)
WANG Congb9a24bb2016-08-19 12:36:54 -0700496 goto err3;
Patrick McHardy72d97942008-07-14 20:36:32 -0700497 perturb_period = nla_get_u32(tb[TCA_FLOW_PERTURB]) * HZ;
498 }
499
Patrick McHardye5dfb812008-01-31 18:37:42 -0800500 if (TC_H_MAJ(baseclass) == 0)
501 baseclass = TC_H_MAKE(tp->q->handle, baseclass);
502 if (TC_H_MIN(baseclass) == 0)
503 baseclass = TC_H_MAKE(baseclass, 1);
504
John Fastabend70da9f02014-09-12 20:06:55 -0700505 fnew->handle = handle;
506 fnew->mask = ~0U;
507 fnew->tp = tp;
508 get_random_bytes(&fnew->hashrnd, 4);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800509 }
510
John Fastabend70da9f02014-09-12 20:06:55 -0700511 fnew->perturb_timer.function = flow_perturbation;
512 fnew->perturb_timer.data = (unsigned long)fnew;
513 init_timer_deferrable(&fnew->perturb_timer);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800514
John Fastabend70da9f02014-09-12 20:06:55 -0700515 tcf_exts_change(tp, &fnew->exts, &e);
516 tcf_em_tree_change(tp, &fnew->ematches, &t);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800517
Eric Dumazet02875872014-10-05 18:38:35 -0700518 netif_keep_dst(qdisc_dev(tp->q));
519
Patrick McHardye5dfb812008-01-31 18:37:42 -0800520 if (tb[TCA_FLOW_KEYS]) {
John Fastabend70da9f02014-09-12 20:06:55 -0700521 fnew->keymask = keymask;
522 fnew->nkeys = nkeys;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800523 }
524
John Fastabend70da9f02014-09-12 20:06:55 -0700525 fnew->mode = mode;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800526
527 if (tb[TCA_FLOW_MASK])
John Fastabend70da9f02014-09-12 20:06:55 -0700528 fnew->mask = nla_get_u32(tb[TCA_FLOW_MASK]);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800529 if (tb[TCA_FLOW_XOR])
John Fastabend70da9f02014-09-12 20:06:55 -0700530 fnew->xor = nla_get_u32(tb[TCA_FLOW_XOR]);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800531 if (tb[TCA_FLOW_RSHIFT])
John Fastabend70da9f02014-09-12 20:06:55 -0700532 fnew->rshift = nla_get_u32(tb[TCA_FLOW_RSHIFT]);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800533 if (tb[TCA_FLOW_ADDEND])
John Fastabend70da9f02014-09-12 20:06:55 -0700534 fnew->addend = nla_get_u32(tb[TCA_FLOW_ADDEND]);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800535
536 if (tb[TCA_FLOW_DIVISOR])
John Fastabend70da9f02014-09-12 20:06:55 -0700537 fnew->divisor = nla_get_u32(tb[TCA_FLOW_DIVISOR]);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800538 if (baseclass)
John Fastabend70da9f02014-09-12 20:06:55 -0700539 fnew->baseclass = baseclass;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800540
John Fastabend70da9f02014-09-12 20:06:55 -0700541 fnew->perturb_period = perturb_period;
Patrick McHardy72d97942008-07-14 20:36:32 -0700542 if (perturb_period)
John Fastabend70da9f02014-09-12 20:06:55 -0700543 mod_timer(&fnew->perturb_timer, jiffies + perturb_period);
Patrick McHardy72d97942008-07-14 20:36:32 -0700544
Patrick McHardye5dfb812008-01-31 18:37:42 -0800545 if (*arg == 0)
John Fastabend70da9f02014-09-12 20:06:55 -0700546 list_add_tail_rcu(&fnew->list, &head->filters);
547 else
Daniel Borkmann32b2f4b2015-07-17 22:38:45 +0200548 list_replace_rcu(&fold->list, &fnew->list);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800549
John Fastabend70da9f02014-09-12 20:06:55 -0700550 *arg = (unsigned long)fnew;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800551
John Fastabend70da9f02014-09-12 20:06:55 -0700552 if (fold)
553 call_rcu(&fold->rcu, flow_destroy_filter);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800554 return 0;
555
WANG Congb9a24bb2016-08-19 12:36:54 -0700556err3:
557 tcf_exts_destroy(&fnew->exts);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800558err2:
John Fastabend82a470f2014-10-05 21:27:53 -0700559 tcf_em_tree_destroy(&t);
John Fastabend70da9f02014-09-12 20:06:55 -0700560 kfree(fnew);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800561err1:
WANG Cong18d02642014-09-25 10:26:37 -0700562 tcf_exts_destroy(&e);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800563 return err;
564}
565
Patrick McHardye5dfb812008-01-31 18:37:42 -0800566static int flow_delete(struct tcf_proto *tp, unsigned long arg)
567{
568 struct flow_filter *f = (struct flow_filter *)arg;
569
John Fastabend70da9f02014-09-12 20:06:55 -0700570 list_del_rcu(&f->list);
571 call_rcu(&f->rcu, flow_destroy_filter);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800572 return 0;
573}
574
575static int flow_init(struct tcf_proto *tp)
576{
577 struct flow_head *head;
578
Patrick McHardye5dfb812008-01-31 18:37:42 -0800579 head = kzalloc(sizeof(*head), GFP_KERNEL);
580 if (head == NULL)
581 return -ENOBUFS;
582 INIT_LIST_HEAD(&head->filters);
John Fastabend70da9f02014-09-12 20:06:55 -0700583 rcu_assign_pointer(tp->root, head);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800584 return 0;
585}
586
Cong Wang1e052be2015-03-06 11:47:59 -0800587static bool flow_destroy(struct tcf_proto *tp, bool force)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800588{
John Fastabend70da9f02014-09-12 20:06:55 -0700589 struct flow_head *head = rtnl_dereference(tp->root);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800590 struct flow_filter *f, *next;
591
Cong Wang1e052be2015-03-06 11:47:59 -0800592 if (!force && !list_empty(&head->filters))
593 return false;
594
Patrick McHardye5dfb812008-01-31 18:37:42 -0800595 list_for_each_entry_safe(f, next, &head->filters, list) {
John Fastabend70da9f02014-09-12 20:06:55 -0700596 list_del_rcu(&f->list);
597 call_rcu(&f->rcu, flow_destroy_filter);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800598 }
John Fastabend70da9f02014-09-12 20:06:55 -0700599 RCU_INIT_POINTER(tp->root, NULL);
600 kfree_rcu(head, rcu);
Cong Wang1e052be2015-03-06 11:47:59 -0800601 return true;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800602}
603
604static unsigned long flow_get(struct tcf_proto *tp, u32 handle)
605{
John Fastabend70da9f02014-09-12 20:06:55 -0700606 struct flow_head *head = rtnl_dereference(tp->root);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800607 struct flow_filter *f;
608
Jiri Pirko6a659cd2014-12-02 18:00:34 +0100609 list_for_each_entry(f, &head->filters, list)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800610 if (f->handle == handle)
611 return (unsigned long)f;
612 return 0;
613}
614
WANG Cong832d1d52014-01-09 16:14:01 -0800615static int flow_dump(struct net *net, struct tcf_proto *tp, unsigned long fh,
Patrick McHardye5dfb812008-01-31 18:37:42 -0800616 struct sk_buff *skb, struct tcmsg *t)
617{
618 struct flow_filter *f = (struct flow_filter *)fh;
619 struct nlattr *nest;
620
621 if (f == NULL)
622 return skb->len;
623
624 t->tcm_handle = f->handle;
625
626 nest = nla_nest_start(skb, TCA_OPTIONS);
627 if (nest == NULL)
628 goto nla_put_failure;
629
David S. Miller1b34ec42012-03-29 05:11:39 -0400630 if (nla_put_u32(skb, TCA_FLOW_KEYS, f->keymask) ||
631 nla_put_u32(skb, TCA_FLOW_MODE, f->mode))
632 goto nla_put_failure;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800633
634 if (f->mask != ~0 || f->xor != 0) {
David S. Miller1b34ec42012-03-29 05:11:39 -0400635 if (nla_put_u32(skb, TCA_FLOW_MASK, f->mask) ||
636 nla_put_u32(skb, TCA_FLOW_XOR, f->xor))
637 goto nla_put_failure;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800638 }
David S. Miller1b34ec42012-03-29 05:11:39 -0400639 if (f->rshift &&
640 nla_put_u32(skb, TCA_FLOW_RSHIFT, f->rshift))
641 goto nla_put_failure;
642 if (f->addend &&
643 nla_put_u32(skb, TCA_FLOW_ADDEND, f->addend))
644 goto nla_put_failure;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800645
David S. Miller1b34ec42012-03-29 05:11:39 -0400646 if (f->divisor &&
647 nla_put_u32(skb, TCA_FLOW_DIVISOR, f->divisor))
648 goto nla_put_failure;
649 if (f->baseclass &&
650 nla_put_u32(skb, TCA_FLOW_BASECLASS, f->baseclass))
651 goto nla_put_failure;
Patrick McHardye5dfb812008-01-31 18:37:42 -0800652
David S. Miller1b34ec42012-03-29 05:11:39 -0400653 if (f->perturb_period &&
654 nla_put_u32(skb, TCA_FLOW_PERTURB, f->perturb_period / HZ))
655 goto nla_put_failure;
Patrick McHardy72d97942008-07-14 20:36:32 -0700656
WANG Cong5da57f42013-12-15 20:15:07 -0800657 if (tcf_exts_dump(skb, &f->exts) < 0)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800658 goto nla_put_failure;
Rami Rosen0aead542008-02-05 02:56:48 -0800659#ifdef CONFIG_NET_EMATCH
Patrick McHardye5dfb812008-01-31 18:37:42 -0800660 if (f->ematches.hdr.nmatches &&
661 tcf_em_tree_dump(skb, &f->ematches, TCA_FLOW_EMATCHES) < 0)
662 goto nla_put_failure;
Rami Rosen0aead542008-02-05 02:56:48 -0800663#endif
Patrick McHardye5dfb812008-01-31 18:37:42 -0800664 nla_nest_end(skb, nest);
665
WANG Cong5da57f42013-12-15 20:15:07 -0800666 if (tcf_exts_dump_stats(skb, &f->exts) < 0)
Patrick McHardye5dfb812008-01-31 18:37:42 -0800667 goto nla_put_failure;
668
669 return skb->len;
670
671nla_put_failure:
Jiri Pirko6ea3b442014-12-09 22:23:29 +0100672 nla_nest_cancel(skb, nest);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800673 return -1;
674}
675
676static void flow_walk(struct tcf_proto *tp, struct tcf_walker *arg)
677{
John Fastabend70da9f02014-09-12 20:06:55 -0700678 struct flow_head *head = rtnl_dereference(tp->root);
Patrick McHardye5dfb812008-01-31 18:37:42 -0800679 struct flow_filter *f;
680
Jiri Pirko6a659cd2014-12-02 18:00:34 +0100681 list_for_each_entry(f, &head->filters, list) {
Patrick McHardye5dfb812008-01-31 18:37:42 -0800682 if (arg->count < arg->skip)
683 goto skip;
684 if (arg->fn(tp, (unsigned long)f, arg) < 0) {
685 arg->stop = 1;
686 break;
687 }
688skip:
689 arg->count++;
690 }
691}
692
693static struct tcf_proto_ops cls_flow_ops __read_mostly = {
694 .kind = "flow",
695 .classify = flow_classify,
696 .init = flow_init,
697 .destroy = flow_destroy,
698 .change = flow_change,
699 .delete = flow_delete,
700 .get = flow_get,
Patrick McHardye5dfb812008-01-31 18:37:42 -0800701 .dump = flow_dump,
702 .walk = flow_walk,
703 .owner = THIS_MODULE,
704};
705
706static int __init cls_flow_init(void)
707{
708 return register_tcf_proto_ops(&cls_flow_ops);
709}
710
711static void __exit cls_flow_exit(void)
712{
713 unregister_tcf_proto_ops(&cls_flow_ops);
714}
715
716module_init(cls_flow_init);
717module_exit(cls_flow_exit);
718
719MODULE_LICENSE("GPL");
720MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
721MODULE_DESCRIPTION("TC flow classifier");