xref: /openbmc/linux/net/sched/cls_flower.c (revision 8730046c)
1 /*
2  * net/sched/cls_flower.c		Flower classifier
3  *
4  * Copyright (c) 2015 Jiri Pirko <jiri@resnulli.us>
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  */
11 
12 #include <linux/kernel.h>
13 #include <linux/init.h>
14 #include <linux/module.h>
15 #include <linux/rhashtable.h>
16 #include <linux/workqueue.h>
17 
18 #include <linux/if_ether.h>
19 #include <linux/in6.h>
20 #include <linux/ip.h>
21 
22 #include <net/sch_generic.h>
23 #include <net/pkt_cls.h>
24 #include <net/ip.h>
25 #include <net/flow_dissector.h>
26 
27 #include <net/dst.h>
28 #include <net/dst_metadata.h>
29 
30 struct fl_flow_key {
31 	int	indev_ifindex;
32 	struct flow_dissector_key_control control;
33 	struct flow_dissector_key_control enc_control;
34 	struct flow_dissector_key_basic basic;
35 	struct flow_dissector_key_eth_addrs eth;
36 	struct flow_dissector_key_vlan vlan;
37 	union {
38 		struct flow_dissector_key_ipv4_addrs ipv4;
39 		struct flow_dissector_key_ipv6_addrs ipv6;
40 	};
41 	struct flow_dissector_key_ports tp;
42 	struct flow_dissector_key_icmp icmp;
43 	struct flow_dissector_key_keyid enc_key_id;
44 	union {
45 		struct flow_dissector_key_ipv4_addrs enc_ipv4;
46 		struct flow_dissector_key_ipv6_addrs enc_ipv6;
47 	};
48 	struct flow_dissector_key_ports enc_tp;
49 } __aligned(BITS_PER_LONG / 8); /* Ensure that we can do comparisons as longs. */
50 
51 struct fl_flow_mask_range {
52 	unsigned short int start;
53 	unsigned short int end;
54 };
55 
56 struct fl_flow_mask {
57 	struct fl_flow_key key;
58 	struct fl_flow_mask_range range;
59 	struct rcu_head	rcu;
60 };
61 
62 struct cls_fl_head {
63 	struct rhashtable ht;
64 	struct fl_flow_mask mask;
65 	struct flow_dissector dissector;
66 	u32 hgen;
67 	bool mask_assigned;
68 	struct list_head filters;
69 	struct rhashtable_params ht_params;
70 	union {
71 		struct work_struct work;
72 		struct rcu_head	rcu;
73 	};
74 };
75 
76 struct cls_fl_filter {
77 	struct rhash_head ht_node;
78 	struct fl_flow_key mkey;
79 	struct tcf_exts exts;
80 	struct tcf_result res;
81 	struct fl_flow_key key;
82 	struct list_head list;
83 	u32 handle;
84 	u32 flags;
85 	struct rcu_head	rcu;
86 	struct tc_to_netdev tc;
87 	struct net_device *hw_dev;
88 };
89 
90 static unsigned short int fl_mask_range(const struct fl_flow_mask *mask)
91 {
92 	return mask->range.end - mask->range.start;
93 }
94 
95 static void fl_mask_update_range(struct fl_flow_mask *mask)
96 {
97 	const u8 *bytes = (const u8 *) &mask->key;
98 	size_t size = sizeof(mask->key);
99 	size_t i, first = 0, last = size - 1;
100 
101 	for (i = 0; i < sizeof(mask->key); i++) {
102 		if (bytes[i]) {
103 			if (!first && i)
104 				first = i;
105 			last = i;
106 		}
107 	}
108 	mask->range.start = rounddown(first, sizeof(long));
109 	mask->range.end = roundup(last + 1, sizeof(long));
110 }
111 
112 static void *fl_key_get_start(struct fl_flow_key *key,
113 			      const struct fl_flow_mask *mask)
114 {
115 	return (u8 *) key + mask->range.start;
116 }
117 
118 static void fl_set_masked_key(struct fl_flow_key *mkey, struct fl_flow_key *key,
119 			      struct fl_flow_mask *mask)
120 {
121 	const long *lkey = fl_key_get_start(key, mask);
122 	const long *lmask = fl_key_get_start(&mask->key, mask);
123 	long *lmkey = fl_key_get_start(mkey, mask);
124 	int i;
125 
126 	for (i = 0; i < fl_mask_range(mask); i += sizeof(long))
127 		*lmkey++ = *lkey++ & *lmask++;
128 }
129 
130 static void fl_clear_masked_range(struct fl_flow_key *key,
131 				  struct fl_flow_mask *mask)
132 {
133 	memset(fl_key_get_start(key, mask), 0, fl_mask_range(mask));
134 }
135 
136 static int fl_classify(struct sk_buff *skb, const struct tcf_proto *tp,
137 		       struct tcf_result *res)
138 {
139 	struct cls_fl_head *head = rcu_dereference_bh(tp->root);
140 	struct cls_fl_filter *f;
141 	struct fl_flow_key skb_key;
142 	struct fl_flow_key skb_mkey;
143 	struct ip_tunnel_info *info;
144 
145 	if (!atomic_read(&head->ht.nelems))
146 		return -1;
147 
148 	fl_clear_masked_range(&skb_key, &head->mask);
149 
150 	info = skb_tunnel_info(skb);
151 	if (info) {
152 		struct ip_tunnel_key *key = &info->key;
153 
154 		switch (ip_tunnel_info_af(info)) {
155 		case AF_INET:
156 			skb_key.enc_control.addr_type =
157 				FLOW_DISSECTOR_KEY_IPV4_ADDRS;
158 			skb_key.enc_ipv4.src = key->u.ipv4.src;
159 			skb_key.enc_ipv4.dst = key->u.ipv4.dst;
160 			break;
161 		case AF_INET6:
162 			skb_key.enc_control.addr_type =
163 				FLOW_DISSECTOR_KEY_IPV6_ADDRS;
164 			skb_key.enc_ipv6.src = key->u.ipv6.src;
165 			skb_key.enc_ipv6.dst = key->u.ipv6.dst;
166 			break;
167 		}
168 
169 		skb_key.enc_key_id.keyid = tunnel_id_to_key32(key->tun_id);
170 		skb_key.enc_tp.src = key->tp_src;
171 		skb_key.enc_tp.dst = key->tp_dst;
172 	}
173 
174 	skb_key.indev_ifindex = skb->skb_iif;
175 	/* skb_flow_dissect() does not set n_proto in case an unknown protocol,
176 	 * so do it rather here.
177 	 */
178 	skb_key.basic.n_proto = skb->protocol;
179 	skb_flow_dissect(skb, &head->dissector, &skb_key, 0);
180 
181 	fl_set_masked_key(&skb_mkey, &skb_key, &head->mask);
182 
183 	f = rhashtable_lookup_fast(&head->ht,
184 				   fl_key_get_start(&skb_mkey, &head->mask),
185 				   head->ht_params);
186 	if (f && !tc_skip_sw(f->flags)) {
187 		*res = f->res;
188 		return tcf_exts_exec(skb, &f->exts, res);
189 	}
190 	return -1;
191 }
192 
193 static int fl_init(struct tcf_proto *tp)
194 {
195 	struct cls_fl_head *head;
196 
197 	head = kzalloc(sizeof(*head), GFP_KERNEL);
198 	if (!head)
199 		return -ENOBUFS;
200 
201 	INIT_LIST_HEAD_RCU(&head->filters);
202 	rcu_assign_pointer(tp->root, head);
203 
204 	return 0;
205 }
206 
207 static void fl_destroy_filter(struct rcu_head *head)
208 {
209 	struct cls_fl_filter *f = container_of(head, struct cls_fl_filter, rcu);
210 
211 	tcf_exts_destroy(&f->exts);
212 	kfree(f);
213 }
214 
215 static void fl_hw_destroy_filter(struct tcf_proto *tp, struct cls_fl_filter *f)
216 {
217 	struct tc_cls_flower_offload offload = {0};
218 	struct net_device *dev = f->hw_dev;
219 	struct tc_to_netdev *tc = &f->tc;
220 
221 	if (!tc_can_offload(dev, tp))
222 		return;
223 
224 	offload.command = TC_CLSFLOWER_DESTROY;
225 	offload.cookie = (unsigned long)f;
226 
227 	tc->type = TC_SETUP_CLSFLOWER;
228 	tc->cls_flower = &offload;
229 
230 	dev->netdev_ops->ndo_setup_tc(dev, tp->q->handle, tp->protocol, tc);
231 }
232 
233 static int fl_hw_replace_filter(struct tcf_proto *tp,
234 				struct flow_dissector *dissector,
235 				struct fl_flow_key *mask,
236 				struct cls_fl_filter *f)
237 {
238 	struct net_device *dev = tp->q->dev_queue->dev;
239 	struct tc_cls_flower_offload offload = {0};
240 	struct tc_to_netdev *tc = &f->tc;
241 	int err;
242 
243 	if (!tc_can_offload(dev, tp)) {
244 		if (tcf_exts_get_dev(dev, &f->exts, &f->hw_dev) ||
245 		    (f->hw_dev && !tc_can_offload(f->hw_dev, tp))) {
246 			f->hw_dev = dev;
247 			return tc_skip_sw(f->flags) ? -EINVAL : 0;
248 		}
249 		dev = f->hw_dev;
250 		tc->egress_dev = true;
251 	} else {
252 		f->hw_dev = dev;
253 	}
254 
255 	offload.command = TC_CLSFLOWER_REPLACE;
256 	offload.cookie = (unsigned long)f;
257 	offload.dissector = dissector;
258 	offload.mask = mask;
259 	offload.key = &f->mkey;
260 	offload.exts = &f->exts;
261 
262 	tc->type = TC_SETUP_CLSFLOWER;
263 	tc->cls_flower = &offload;
264 
265 	err = dev->netdev_ops->ndo_setup_tc(dev, tp->q->handle, tp->protocol,
266 					    tc);
267 
268 	if (tc_skip_sw(f->flags))
269 		return err;
270 	return 0;
271 }
272 
273 static void fl_hw_update_stats(struct tcf_proto *tp, struct cls_fl_filter *f)
274 {
275 	struct tc_cls_flower_offload offload = {0};
276 	struct net_device *dev = f->hw_dev;
277 	struct tc_to_netdev *tc = &f->tc;
278 
279 	if (!tc_can_offload(dev, tp))
280 		return;
281 
282 	offload.command = TC_CLSFLOWER_STATS;
283 	offload.cookie = (unsigned long)f;
284 	offload.exts = &f->exts;
285 
286 	tc->type = TC_SETUP_CLSFLOWER;
287 	tc->cls_flower = &offload;
288 
289 	dev->netdev_ops->ndo_setup_tc(dev, tp->q->handle, tp->protocol, tc);
290 }
291 
292 static void __fl_delete(struct tcf_proto *tp, struct cls_fl_filter *f)
293 {
294 	list_del_rcu(&f->list);
295 	if (!tc_skip_hw(f->flags))
296 		fl_hw_destroy_filter(tp, f);
297 	tcf_unbind_filter(tp, &f->res);
298 	call_rcu(&f->rcu, fl_destroy_filter);
299 }
300 
301 static void fl_destroy_sleepable(struct work_struct *work)
302 {
303 	struct cls_fl_head *head = container_of(work, struct cls_fl_head,
304 						work);
305 	if (head->mask_assigned)
306 		rhashtable_destroy(&head->ht);
307 	kfree(head);
308 	module_put(THIS_MODULE);
309 }
310 
311 static void fl_destroy_rcu(struct rcu_head *rcu)
312 {
313 	struct cls_fl_head *head = container_of(rcu, struct cls_fl_head, rcu);
314 
315 	INIT_WORK(&head->work, fl_destroy_sleepable);
316 	schedule_work(&head->work);
317 }
318 
319 static bool fl_destroy(struct tcf_proto *tp, bool force)
320 {
321 	struct cls_fl_head *head = rtnl_dereference(tp->root);
322 	struct cls_fl_filter *f, *next;
323 
324 	if (!force && !list_empty(&head->filters))
325 		return false;
326 
327 	list_for_each_entry_safe(f, next, &head->filters, list)
328 		__fl_delete(tp, f);
329 
330 	__module_get(THIS_MODULE);
331 	call_rcu(&head->rcu, fl_destroy_rcu);
332 
333 	return true;
334 }
335 
336 static unsigned long fl_get(struct tcf_proto *tp, u32 handle)
337 {
338 	struct cls_fl_head *head = rtnl_dereference(tp->root);
339 	struct cls_fl_filter *f;
340 
341 	list_for_each_entry(f, &head->filters, list)
342 		if (f->handle == handle)
343 			return (unsigned long) f;
344 	return 0;
345 }
346 
347 static const struct nla_policy fl_policy[TCA_FLOWER_MAX + 1] = {
348 	[TCA_FLOWER_UNSPEC]		= { .type = NLA_UNSPEC },
349 	[TCA_FLOWER_CLASSID]		= { .type = NLA_U32 },
350 	[TCA_FLOWER_INDEV]		= { .type = NLA_STRING,
351 					    .len = IFNAMSIZ },
352 	[TCA_FLOWER_KEY_ETH_DST]	= { .len = ETH_ALEN },
353 	[TCA_FLOWER_KEY_ETH_DST_MASK]	= { .len = ETH_ALEN },
354 	[TCA_FLOWER_KEY_ETH_SRC]	= { .len = ETH_ALEN },
355 	[TCA_FLOWER_KEY_ETH_SRC_MASK]	= { .len = ETH_ALEN },
356 	[TCA_FLOWER_KEY_ETH_TYPE]	= { .type = NLA_U16 },
357 	[TCA_FLOWER_KEY_IP_PROTO]	= { .type = NLA_U8 },
358 	[TCA_FLOWER_KEY_IPV4_SRC]	= { .type = NLA_U32 },
359 	[TCA_FLOWER_KEY_IPV4_SRC_MASK]	= { .type = NLA_U32 },
360 	[TCA_FLOWER_KEY_IPV4_DST]	= { .type = NLA_U32 },
361 	[TCA_FLOWER_KEY_IPV4_DST_MASK]	= { .type = NLA_U32 },
362 	[TCA_FLOWER_KEY_IPV6_SRC]	= { .len = sizeof(struct in6_addr) },
363 	[TCA_FLOWER_KEY_IPV6_SRC_MASK]	= { .len = sizeof(struct in6_addr) },
364 	[TCA_FLOWER_KEY_IPV6_DST]	= { .len = sizeof(struct in6_addr) },
365 	[TCA_FLOWER_KEY_IPV6_DST_MASK]	= { .len = sizeof(struct in6_addr) },
366 	[TCA_FLOWER_KEY_TCP_SRC]	= { .type = NLA_U16 },
367 	[TCA_FLOWER_KEY_TCP_DST]	= { .type = NLA_U16 },
368 	[TCA_FLOWER_KEY_UDP_SRC]	= { .type = NLA_U16 },
369 	[TCA_FLOWER_KEY_UDP_DST]	= { .type = NLA_U16 },
370 	[TCA_FLOWER_KEY_VLAN_ID]	= { .type = NLA_U16 },
371 	[TCA_FLOWER_KEY_VLAN_PRIO]	= { .type = NLA_U8 },
372 	[TCA_FLOWER_KEY_VLAN_ETH_TYPE]	= { .type = NLA_U16 },
373 	[TCA_FLOWER_KEY_ENC_KEY_ID]	= { .type = NLA_U32 },
374 	[TCA_FLOWER_KEY_ENC_IPV4_SRC]	= { .type = NLA_U32 },
375 	[TCA_FLOWER_KEY_ENC_IPV4_SRC_MASK] = { .type = NLA_U32 },
376 	[TCA_FLOWER_KEY_ENC_IPV4_DST]	= { .type = NLA_U32 },
377 	[TCA_FLOWER_KEY_ENC_IPV4_DST_MASK] = { .type = NLA_U32 },
378 	[TCA_FLOWER_KEY_ENC_IPV6_SRC]	= { .len = sizeof(struct in6_addr) },
379 	[TCA_FLOWER_KEY_ENC_IPV6_SRC_MASK] = { .len = sizeof(struct in6_addr) },
380 	[TCA_FLOWER_KEY_ENC_IPV6_DST]	= { .len = sizeof(struct in6_addr) },
381 	[TCA_FLOWER_KEY_ENC_IPV6_DST_MASK] = { .len = sizeof(struct in6_addr) },
382 	[TCA_FLOWER_KEY_TCP_SRC_MASK]	= { .type = NLA_U16 },
383 	[TCA_FLOWER_KEY_TCP_DST_MASK]	= { .type = NLA_U16 },
384 	[TCA_FLOWER_KEY_UDP_SRC_MASK]	= { .type = NLA_U16 },
385 	[TCA_FLOWER_KEY_UDP_DST_MASK]	= { .type = NLA_U16 },
386 	[TCA_FLOWER_KEY_SCTP_SRC_MASK]	= { .type = NLA_U16 },
387 	[TCA_FLOWER_KEY_SCTP_DST_MASK]	= { .type = NLA_U16 },
388 	[TCA_FLOWER_KEY_SCTP_SRC]	= { .type = NLA_U16 },
389 	[TCA_FLOWER_KEY_SCTP_DST]	= { .type = NLA_U16 },
390 	[TCA_FLOWER_KEY_ENC_UDP_SRC_PORT]	= { .type = NLA_U16 },
391 	[TCA_FLOWER_KEY_ENC_UDP_SRC_PORT_MASK]	= { .type = NLA_U16 },
392 	[TCA_FLOWER_KEY_ENC_UDP_DST_PORT]	= { .type = NLA_U16 },
393 	[TCA_FLOWER_KEY_ENC_UDP_DST_PORT_MASK]	= { .type = NLA_U16 },
394 	[TCA_FLOWER_KEY_FLAGS]		= { .type = NLA_U32 },
395 	[TCA_FLOWER_KEY_FLAGS_MASK]	= { .type = NLA_U32 },
396 	[TCA_FLOWER_KEY_ICMPV4_TYPE]	= { .type = NLA_U8 },
397 	[TCA_FLOWER_KEY_ICMPV4_TYPE_MASK] = { .type = NLA_U8 },
398 	[TCA_FLOWER_KEY_ICMPV4_CODE]	= { .type = NLA_U8 },
399 	[TCA_FLOWER_KEY_ICMPV4_CODE_MASK] = { .type = NLA_U8 },
400 	[TCA_FLOWER_KEY_ICMPV6_TYPE]	= { .type = NLA_U8 },
401 	[TCA_FLOWER_KEY_ICMPV6_TYPE_MASK] = { .type = NLA_U8 },
402 	[TCA_FLOWER_KEY_ICMPV6_CODE]	= { .type = NLA_U8 },
403 	[TCA_FLOWER_KEY_ICMPV6_CODE_MASK] = { .type = NLA_U8 },
404 };
405 
406 static void fl_set_key_val(struct nlattr **tb,
407 			   void *val, int val_type,
408 			   void *mask, int mask_type, int len)
409 {
410 	if (!tb[val_type])
411 		return;
412 	memcpy(val, nla_data(tb[val_type]), len);
413 	if (mask_type == TCA_FLOWER_UNSPEC || !tb[mask_type])
414 		memset(mask, 0xff, len);
415 	else
416 		memcpy(mask, nla_data(tb[mask_type]), len);
417 }
418 
419 static void fl_set_key_vlan(struct nlattr **tb,
420 			    struct flow_dissector_key_vlan *key_val,
421 			    struct flow_dissector_key_vlan *key_mask)
422 {
423 #define VLAN_PRIORITY_MASK	0x7
424 
425 	if (tb[TCA_FLOWER_KEY_VLAN_ID]) {
426 		key_val->vlan_id =
427 			nla_get_u16(tb[TCA_FLOWER_KEY_VLAN_ID]) & VLAN_VID_MASK;
428 		key_mask->vlan_id = VLAN_VID_MASK;
429 	}
430 	if (tb[TCA_FLOWER_KEY_VLAN_PRIO]) {
431 		key_val->vlan_priority =
432 			nla_get_u8(tb[TCA_FLOWER_KEY_VLAN_PRIO]) &
433 			VLAN_PRIORITY_MASK;
434 		key_mask->vlan_priority = VLAN_PRIORITY_MASK;
435 	}
436 }
437 
438 static void fl_set_key_flag(u32 flower_key, u32 flower_mask,
439 			    u32 *dissector_key, u32 *dissector_mask,
440 			    u32 flower_flag_bit, u32 dissector_flag_bit)
441 {
442 	if (flower_mask & flower_flag_bit) {
443 		*dissector_mask |= dissector_flag_bit;
444 		if (flower_key & flower_flag_bit)
445 			*dissector_key |= dissector_flag_bit;
446 	}
447 }
448 
449 static int fl_set_key_flags(struct nlattr **tb,
450 			    u32 *flags_key, u32 *flags_mask)
451 {
452 	u32 key, mask;
453 
454 	/* mask is mandatory for flags */
455 	if (!tb[TCA_FLOWER_KEY_FLAGS_MASK])
456 		return -EINVAL;
457 
458 	key = be32_to_cpu(nla_get_u32(tb[TCA_FLOWER_KEY_FLAGS]));
459 	mask = be32_to_cpu(nla_get_u32(tb[TCA_FLOWER_KEY_FLAGS_MASK]));
460 
461 	*flags_key  = 0;
462 	*flags_mask = 0;
463 
464 	fl_set_key_flag(key, mask, flags_key, flags_mask,
465 			TCA_FLOWER_KEY_FLAGS_IS_FRAGMENT, FLOW_DIS_IS_FRAGMENT);
466 
467 	return 0;
468 }
469 
470 static int fl_set_key(struct net *net, struct nlattr **tb,
471 		      struct fl_flow_key *key, struct fl_flow_key *mask)
472 {
473 	__be16 ethertype;
474 	int ret = 0;
475 #ifdef CONFIG_NET_CLS_IND
476 	if (tb[TCA_FLOWER_INDEV]) {
477 		int err = tcf_change_indev(net, tb[TCA_FLOWER_INDEV]);
478 		if (err < 0)
479 			return err;
480 		key->indev_ifindex = err;
481 		mask->indev_ifindex = 0xffffffff;
482 	}
483 #endif
484 
485 	fl_set_key_val(tb, key->eth.dst, TCA_FLOWER_KEY_ETH_DST,
486 		       mask->eth.dst, TCA_FLOWER_KEY_ETH_DST_MASK,
487 		       sizeof(key->eth.dst));
488 	fl_set_key_val(tb, key->eth.src, TCA_FLOWER_KEY_ETH_SRC,
489 		       mask->eth.src, TCA_FLOWER_KEY_ETH_SRC_MASK,
490 		       sizeof(key->eth.src));
491 
492 	if (tb[TCA_FLOWER_KEY_ETH_TYPE]) {
493 		ethertype = nla_get_be16(tb[TCA_FLOWER_KEY_ETH_TYPE]);
494 
495 		if (ethertype == htons(ETH_P_8021Q)) {
496 			fl_set_key_vlan(tb, &key->vlan, &mask->vlan);
497 			fl_set_key_val(tb, &key->basic.n_proto,
498 				       TCA_FLOWER_KEY_VLAN_ETH_TYPE,
499 				       &mask->basic.n_proto, TCA_FLOWER_UNSPEC,
500 				       sizeof(key->basic.n_proto));
501 		} else {
502 			key->basic.n_proto = ethertype;
503 			mask->basic.n_proto = cpu_to_be16(~0);
504 		}
505 	}
506 
507 	if (key->basic.n_proto == htons(ETH_P_IP) ||
508 	    key->basic.n_proto == htons(ETH_P_IPV6)) {
509 		fl_set_key_val(tb, &key->basic.ip_proto, TCA_FLOWER_KEY_IP_PROTO,
510 			       &mask->basic.ip_proto, TCA_FLOWER_UNSPEC,
511 			       sizeof(key->basic.ip_proto));
512 	}
513 
514 	if (tb[TCA_FLOWER_KEY_IPV4_SRC] || tb[TCA_FLOWER_KEY_IPV4_DST]) {
515 		key->control.addr_type = FLOW_DISSECTOR_KEY_IPV4_ADDRS;
516 		mask->control.addr_type = ~0;
517 		fl_set_key_val(tb, &key->ipv4.src, TCA_FLOWER_KEY_IPV4_SRC,
518 			       &mask->ipv4.src, TCA_FLOWER_KEY_IPV4_SRC_MASK,
519 			       sizeof(key->ipv4.src));
520 		fl_set_key_val(tb, &key->ipv4.dst, TCA_FLOWER_KEY_IPV4_DST,
521 			       &mask->ipv4.dst, TCA_FLOWER_KEY_IPV4_DST_MASK,
522 			       sizeof(key->ipv4.dst));
523 	} else if (tb[TCA_FLOWER_KEY_IPV6_SRC] || tb[TCA_FLOWER_KEY_IPV6_DST]) {
524 		key->control.addr_type = FLOW_DISSECTOR_KEY_IPV6_ADDRS;
525 		mask->control.addr_type = ~0;
526 		fl_set_key_val(tb, &key->ipv6.src, TCA_FLOWER_KEY_IPV6_SRC,
527 			       &mask->ipv6.src, TCA_FLOWER_KEY_IPV6_SRC_MASK,
528 			       sizeof(key->ipv6.src));
529 		fl_set_key_val(tb, &key->ipv6.dst, TCA_FLOWER_KEY_IPV6_DST,
530 			       &mask->ipv6.dst, TCA_FLOWER_KEY_IPV6_DST_MASK,
531 			       sizeof(key->ipv6.dst));
532 	}
533 
534 	if (key->basic.ip_proto == IPPROTO_TCP) {
535 		fl_set_key_val(tb, &key->tp.src, TCA_FLOWER_KEY_TCP_SRC,
536 			       &mask->tp.src, TCA_FLOWER_KEY_TCP_SRC_MASK,
537 			       sizeof(key->tp.src));
538 		fl_set_key_val(tb, &key->tp.dst, TCA_FLOWER_KEY_TCP_DST,
539 			       &mask->tp.dst, TCA_FLOWER_KEY_TCP_DST_MASK,
540 			       sizeof(key->tp.dst));
541 	} else if (key->basic.ip_proto == IPPROTO_UDP) {
542 		fl_set_key_val(tb, &key->tp.src, TCA_FLOWER_KEY_UDP_SRC,
543 			       &mask->tp.src, TCA_FLOWER_KEY_UDP_SRC_MASK,
544 			       sizeof(key->tp.src));
545 		fl_set_key_val(tb, &key->tp.dst, TCA_FLOWER_KEY_UDP_DST,
546 			       &mask->tp.dst, TCA_FLOWER_KEY_UDP_DST_MASK,
547 			       sizeof(key->tp.dst));
548 	} else if (key->basic.ip_proto == IPPROTO_SCTP) {
549 		fl_set_key_val(tb, &key->tp.src, TCA_FLOWER_KEY_SCTP_SRC,
550 			       &mask->tp.src, TCA_FLOWER_KEY_SCTP_SRC_MASK,
551 			       sizeof(key->tp.src));
552 		fl_set_key_val(tb, &key->tp.dst, TCA_FLOWER_KEY_SCTP_DST,
553 			       &mask->tp.dst, TCA_FLOWER_KEY_SCTP_DST_MASK,
554 			       sizeof(key->tp.dst));
555 	} else if (key->basic.n_proto == htons(ETH_P_IP) &&
556 		   key->basic.ip_proto == IPPROTO_ICMP) {
557 		fl_set_key_val(tb, &key->icmp.type, TCA_FLOWER_KEY_ICMPV4_TYPE,
558 			       &mask->icmp.type,
559 			       TCA_FLOWER_KEY_ICMPV4_TYPE_MASK,
560 			       sizeof(key->icmp.type));
561 		fl_set_key_val(tb, &key->icmp.code, TCA_FLOWER_KEY_ICMPV4_CODE,
562 			       &mask->icmp.code,
563 			       TCA_FLOWER_KEY_ICMPV4_CODE_MASK,
564 			       sizeof(key->icmp.code));
565 	} else if (key->basic.n_proto == htons(ETH_P_IPV6) &&
566 		   key->basic.ip_proto == IPPROTO_ICMPV6) {
567 		fl_set_key_val(tb, &key->icmp.type, TCA_FLOWER_KEY_ICMPV6_TYPE,
568 			       &mask->icmp.type,
569 			       TCA_FLOWER_KEY_ICMPV6_TYPE_MASK,
570 			       sizeof(key->icmp.type));
571 		fl_set_key_val(tb, &key->icmp.code, TCA_FLOWER_KEY_ICMPV4_CODE,
572 			       &mask->icmp.code,
573 			       TCA_FLOWER_KEY_ICMPV4_CODE_MASK,
574 			       sizeof(key->icmp.code));
575 	}
576 
577 	if (tb[TCA_FLOWER_KEY_ENC_IPV4_SRC] ||
578 	    tb[TCA_FLOWER_KEY_ENC_IPV4_DST]) {
579 		key->enc_control.addr_type = FLOW_DISSECTOR_KEY_IPV4_ADDRS;
580 		mask->enc_control.addr_type = ~0;
581 		fl_set_key_val(tb, &key->enc_ipv4.src,
582 			       TCA_FLOWER_KEY_ENC_IPV4_SRC,
583 			       &mask->enc_ipv4.src,
584 			       TCA_FLOWER_KEY_ENC_IPV4_SRC_MASK,
585 			       sizeof(key->enc_ipv4.src));
586 		fl_set_key_val(tb, &key->enc_ipv4.dst,
587 			       TCA_FLOWER_KEY_ENC_IPV4_DST,
588 			       &mask->enc_ipv4.dst,
589 			       TCA_FLOWER_KEY_ENC_IPV4_DST_MASK,
590 			       sizeof(key->enc_ipv4.dst));
591 	}
592 
593 	if (tb[TCA_FLOWER_KEY_ENC_IPV6_SRC] ||
594 	    tb[TCA_FLOWER_KEY_ENC_IPV6_DST]) {
595 		key->enc_control.addr_type = FLOW_DISSECTOR_KEY_IPV6_ADDRS;
596 		mask->enc_control.addr_type = ~0;
597 		fl_set_key_val(tb, &key->enc_ipv6.src,
598 			       TCA_FLOWER_KEY_ENC_IPV6_SRC,
599 			       &mask->enc_ipv6.src,
600 			       TCA_FLOWER_KEY_ENC_IPV6_SRC_MASK,
601 			       sizeof(key->enc_ipv6.src));
602 		fl_set_key_val(tb, &key->enc_ipv6.dst,
603 			       TCA_FLOWER_KEY_ENC_IPV6_DST,
604 			       &mask->enc_ipv6.dst,
605 			       TCA_FLOWER_KEY_ENC_IPV6_DST_MASK,
606 			       sizeof(key->enc_ipv6.dst));
607 	}
608 
609 	fl_set_key_val(tb, &key->enc_key_id.keyid, TCA_FLOWER_KEY_ENC_KEY_ID,
610 		       &mask->enc_key_id.keyid, TCA_FLOWER_UNSPEC,
611 		       sizeof(key->enc_key_id.keyid));
612 
613 	fl_set_key_val(tb, &key->enc_tp.src, TCA_FLOWER_KEY_ENC_UDP_SRC_PORT,
614 		       &mask->enc_tp.src, TCA_FLOWER_KEY_ENC_UDP_SRC_PORT_MASK,
615 		       sizeof(key->enc_tp.src));
616 
617 	fl_set_key_val(tb, &key->enc_tp.dst, TCA_FLOWER_KEY_ENC_UDP_DST_PORT,
618 		       &mask->enc_tp.dst, TCA_FLOWER_KEY_ENC_UDP_DST_PORT_MASK,
619 		       sizeof(key->enc_tp.dst));
620 
621 	if (tb[TCA_FLOWER_KEY_FLAGS])
622 		ret = fl_set_key_flags(tb, &key->control.flags, &mask->control.flags);
623 
624 	return ret;
625 }
626 
627 static bool fl_mask_eq(struct fl_flow_mask *mask1,
628 		       struct fl_flow_mask *mask2)
629 {
630 	const long *lmask1 = fl_key_get_start(&mask1->key, mask1);
631 	const long *lmask2 = fl_key_get_start(&mask2->key, mask2);
632 
633 	return !memcmp(&mask1->range, &mask2->range, sizeof(mask1->range)) &&
634 	       !memcmp(lmask1, lmask2, fl_mask_range(mask1));
635 }
636 
637 static const struct rhashtable_params fl_ht_params = {
638 	.key_offset = offsetof(struct cls_fl_filter, mkey), /* base offset */
639 	.head_offset = offsetof(struct cls_fl_filter, ht_node),
640 	.automatic_shrinking = true,
641 };
642 
643 static int fl_init_hashtable(struct cls_fl_head *head,
644 			     struct fl_flow_mask *mask)
645 {
646 	head->ht_params = fl_ht_params;
647 	head->ht_params.key_len = fl_mask_range(mask);
648 	head->ht_params.key_offset += mask->range.start;
649 
650 	return rhashtable_init(&head->ht, &head->ht_params);
651 }
652 
653 #define FL_KEY_MEMBER_OFFSET(member) offsetof(struct fl_flow_key, member)
654 #define FL_KEY_MEMBER_SIZE(member) (sizeof(((struct fl_flow_key *) 0)->member))
655 
656 #define FL_KEY_IS_MASKED(mask, member)						\
657 	memchr_inv(((char *)mask) + FL_KEY_MEMBER_OFFSET(member),		\
658 		   0, FL_KEY_MEMBER_SIZE(member))				\
659 
660 #define FL_KEY_SET(keys, cnt, id, member)					\
661 	do {									\
662 		keys[cnt].key_id = id;						\
663 		keys[cnt].offset = FL_KEY_MEMBER_OFFSET(member);		\
664 		cnt++;								\
665 	} while(0);
666 
667 #define FL_KEY_SET_IF_MASKED(mask, keys, cnt, id, member)			\
668 	do {									\
669 		if (FL_KEY_IS_MASKED(mask, member))				\
670 			FL_KEY_SET(keys, cnt, id, member);			\
671 	} while(0);
672 
673 static void fl_init_dissector(struct cls_fl_head *head,
674 			      struct fl_flow_mask *mask)
675 {
676 	struct flow_dissector_key keys[FLOW_DISSECTOR_KEY_MAX];
677 	size_t cnt = 0;
678 
679 	FL_KEY_SET(keys, cnt, FLOW_DISSECTOR_KEY_CONTROL, control);
680 	FL_KEY_SET(keys, cnt, FLOW_DISSECTOR_KEY_BASIC, basic);
681 	FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
682 			     FLOW_DISSECTOR_KEY_ETH_ADDRS, eth);
683 	FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
684 			     FLOW_DISSECTOR_KEY_IPV4_ADDRS, ipv4);
685 	FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
686 			     FLOW_DISSECTOR_KEY_IPV6_ADDRS, ipv6);
687 	FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
688 			     FLOW_DISSECTOR_KEY_PORTS, tp);
689 	FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
690 			     FLOW_DISSECTOR_KEY_ICMP, icmp);
691 	FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
692 			     FLOW_DISSECTOR_KEY_VLAN, vlan);
693 	FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
694 			     FLOW_DISSECTOR_KEY_ENC_KEYID, enc_key_id);
695 	FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
696 			     FLOW_DISSECTOR_KEY_ENC_IPV4_ADDRS, enc_ipv4);
697 	FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
698 			     FLOW_DISSECTOR_KEY_ENC_IPV6_ADDRS, enc_ipv6);
699 	if (FL_KEY_IS_MASKED(&mask->key, enc_ipv4) ||
700 	    FL_KEY_IS_MASKED(&mask->key, enc_ipv6))
701 		FL_KEY_SET(keys, cnt, FLOW_DISSECTOR_KEY_ENC_CONTROL,
702 			   enc_control);
703 	FL_KEY_SET_IF_MASKED(&mask->key, keys, cnt,
704 			     FLOW_DISSECTOR_KEY_ENC_PORTS, enc_tp);
705 
706 	skb_flow_dissector_init(&head->dissector, keys, cnt);
707 }
708 
709 static int fl_check_assign_mask(struct cls_fl_head *head,
710 				struct fl_flow_mask *mask)
711 {
712 	int err;
713 
714 	if (head->mask_assigned) {
715 		if (!fl_mask_eq(&head->mask, mask))
716 			return -EINVAL;
717 		else
718 			return 0;
719 	}
720 
721 	/* Mask is not assigned yet. So assign it and init hashtable
722 	 * according to that.
723 	 */
724 	err = fl_init_hashtable(head, mask);
725 	if (err)
726 		return err;
727 	memcpy(&head->mask, mask, sizeof(head->mask));
728 	head->mask_assigned = true;
729 
730 	fl_init_dissector(head, mask);
731 
732 	return 0;
733 }
734 
735 static int fl_set_parms(struct net *net, struct tcf_proto *tp,
736 			struct cls_fl_filter *f, struct fl_flow_mask *mask,
737 			unsigned long base, struct nlattr **tb,
738 			struct nlattr *est, bool ovr)
739 {
740 	struct tcf_exts e;
741 	int err;
742 
743 	err = tcf_exts_init(&e, TCA_FLOWER_ACT, 0);
744 	if (err < 0)
745 		return err;
746 	err = tcf_exts_validate(net, tp, tb, est, &e, ovr);
747 	if (err < 0)
748 		goto errout;
749 
750 	if (tb[TCA_FLOWER_CLASSID]) {
751 		f->res.classid = nla_get_u32(tb[TCA_FLOWER_CLASSID]);
752 		tcf_bind_filter(tp, &f->res, base);
753 	}
754 
755 	err = fl_set_key(net, tb, &f->key, &mask->key);
756 	if (err)
757 		goto errout;
758 
759 	fl_mask_update_range(mask);
760 	fl_set_masked_key(&f->mkey, &f->key, mask);
761 
762 	tcf_exts_change(tp, &f->exts, &e);
763 
764 	return 0;
765 errout:
766 	tcf_exts_destroy(&e);
767 	return err;
768 }
769 
770 static u32 fl_grab_new_handle(struct tcf_proto *tp,
771 			      struct cls_fl_head *head)
772 {
773 	unsigned int i = 0x80000000;
774 	u32 handle;
775 
776 	do {
777 		if (++head->hgen == 0x7FFFFFFF)
778 			head->hgen = 1;
779 	} while (--i > 0 && fl_get(tp, head->hgen));
780 
781 	if (unlikely(i == 0)) {
782 		pr_err("Insufficient number of handles\n");
783 		handle = 0;
784 	} else {
785 		handle = head->hgen;
786 	}
787 
788 	return handle;
789 }
790 
791 static int fl_change(struct net *net, struct sk_buff *in_skb,
792 		     struct tcf_proto *tp, unsigned long base,
793 		     u32 handle, struct nlattr **tca,
794 		     unsigned long *arg, bool ovr)
795 {
796 	struct cls_fl_head *head = rtnl_dereference(tp->root);
797 	struct cls_fl_filter *fold = (struct cls_fl_filter *) *arg;
798 	struct cls_fl_filter *fnew;
799 	struct nlattr *tb[TCA_FLOWER_MAX + 1];
800 	struct fl_flow_mask mask = {};
801 	int err;
802 
803 	if (!tca[TCA_OPTIONS])
804 		return -EINVAL;
805 
806 	err = nla_parse_nested(tb, TCA_FLOWER_MAX, tca[TCA_OPTIONS], fl_policy);
807 	if (err < 0)
808 		return err;
809 
810 	if (fold && handle && fold->handle != handle)
811 		return -EINVAL;
812 
813 	fnew = kzalloc(sizeof(*fnew), GFP_KERNEL);
814 	if (!fnew)
815 		return -ENOBUFS;
816 
817 	err = tcf_exts_init(&fnew->exts, TCA_FLOWER_ACT, 0);
818 	if (err < 0)
819 		goto errout;
820 
821 	if (!handle) {
822 		handle = fl_grab_new_handle(tp, head);
823 		if (!handle) {
824 			err = -EINVAL;
825 			goto errout;
826 		}
827 	}
828 	fnew->handle = handle;
829 
830 	if (tb[TCA_FLOWER_FLAGS]) {
831 		fnew->flags = nla_get_u32(tb[TCA_FLOWER_FLAGS]);
832 
833 		if (!tc_flags_valid(fnew->flags)) {
834 			err = -EINVAL;
835 			goto errout;
836 		}
837 	}
838 
839 	err = fl_set_parms(net, tp, fnew, &mask, base, tb, tca[TCA_RATE], ovr);
840 	if (err)
841 		goto errout;
842 
843 	err = fl_check_assign_mask(head, &mask);
844 	if (err)
845 		goto errout;
846 
847 	if (!tc_skip_sw(fnew->flags)) {
848 		err = rhashtable_insert_fast(&head->ht, &fnew->ht_node,
849 					     head->ht_params);
850 		if (err)
851 			goto errout;
852 	}
853 
854 	if (!tc_skip_hw(fnew->flags)) {
855 		err = fl_hw_replace_filter(tp,
856 					   &head->dissector,
857 					   &mask.key,
858 					   fnew);
859 		if (err)
860 			goto errout;
861 	}
862 
863 	if (fold) {
864 		if (!tc_skip_sw(fold->flags))
865 			rhashtable_remove_fast(&head->ht, &fold->ht_node,
866 					       head->ht_params);
867 		if (!tc_skip_hw(fold->flags))
868 			fl_hw_destroy_filter(tp, fold);
869 	}
870 
871 	*arg = (unsigned long) fnew;
872 
873 	if (fold) {
874 		list_replace_rcu(&fold->list, &fnew->list);
875 		tcf_unbind_filter(tp, &fold->res);
876 		call_rcu(&fold->rcu, fl_destroy_filter);
877 	} else {
878 		list_add_tail_rcu(&fnew->list, &head->filters);
879 	}
880 
881 	return 0;
882 
883 errout:
884 	tcf_exts_destroy(&fnew->exts);
885 	kfree(fnew);
886 	return err;
887 }
888 
889 static int fl_delete(struct tcf_proto *tp, unsigned long arg)
890 {
891 	struct cls_fl_head *head = rtnl_dereference(tp->root);
892 	struct cls_fl_filter *f = (struct cls_fl_filter *) arg;
893 
894 	if (!tc_skip_sw(f->flags))
895 		rhashtable_remove_fast(&head->ht, &f->ht_node,
896 				       head->ht_params);
897 	__fl_delete(tp, f);
898 	return 0;
899 }
900 
901 static void fl_walk(struct tcf_proto *tp, struct tcf_walker *arg)
902 {
903 	struct cls_fl_head *head = rtnl_dereference(tp->root);
904 	struct cls_fl_filter *f;
905 
906 	list_for_each_entry_rcu(f, &head->filters, list) {
907 		if (arg->count < arg->skip)
908 			goto skip;
909 		if (arg->fn(tp, (unsigned long) f, arg) < 0) {
910 			arg->stop = 1;
911 			break;
912 		}
913 skip:
914 		arg->count++;
915 	}
916 }
917 
918 static int fl_dump_key_val(struct sk_buff *skb,
919 			   void *val, int val_type,
920 			   void *mask, int mask_type, int len)
921 {
922 	int err;
923 
924 	if (!memchr_inv(mask, 0, len))
925 		return 0;
926 	err = nla_put(skb, val_type, len, val);
927 	if (err)
928 		return err;
929 	if (mask_type != TCA_FLOWER_UNSPEC) {
930 		err = nla_put(skb, mask_type, len, mask);
931 		if (err)
932 			return err;
933 	}
934 	return 0;
935 }
936 
937 static int fl_dump_key_vlan(struct sk_buff *skb,
938 			    struct flow_dissector_key_vlan *vlan_key,
939 			    struct flow_dissector_key_vlan *vlan_mask)
940 {
941 	int err;
942 
943 	if (!memchr_inv(vlan_mask, 0, sizeof(*vlan_mask)))
944 		return 0;
945 	if (vlan_mask->vlan_id) {
946 		err = nla_put_u16(skb, TCA_FLOWER_KEY_VLAN_ID,
947 				  vlan_key->vlan_id);
948 		if (err)
949 			return err;
950 	}
951 	if (vlan_mask->vlan_priority) {
952 		err = nla_put_u8(skb, TCA_FLOWER_KEY_VLAN_PRIO,
953 				 vlan_key->vlan_priority);
954 		if (err)
955 			return err;
956 	}
957 	return 0;
958 }
959 
960 static void fl_get_key_flag(u32 dissector_key, u32 dissector_mask,
961 			    u32 *flower_key, u32 *flower_mask,
962 			    u32 flower_flag_bit, u32 dissector_flag_bit)
963 {
964 	if (dissector_mask & dissector_flag_bit) {
965 		*flower_mask |= flower_flag_bit;
966 		if (dissector_key & dissector_flag_bit)
967 			*flower_key |= flower_flag_bit;
968 	}
969 }
970 
971 static int fl_dump_key_flags(struct sk_buff *skb, u32 flags_key, u32 flags_mask)
972 {
973 	u32 key, mask;
974 	__be32 _key, _mask;
975 	int err;
976 
977 	if (!memchr_inv(&flags_mask, 0, sizeof(flags_mask)))
978 		return 0;
979 
980 	key = 0;
981 	mask = 0;
982 
983 	fl_get_key_flag(flags_key, flags_mask, &key, &mask,
984 			TCA_FLOWER_KEY_FLAGS_IS_FRAGMENT, FLOW_DIS_IS_FRAGMENT);
985 
986 	_key = cpu_to_be32(key);
987 	_mask = cpu_to_be32(mask);
988 
989 	err = nla_put(skb, TCA_FLOWER_KEY_FLAGS, 4, &_key);
990 	if (err)
991 		return err;
992 
993 	return nla_put(skb, TCA_FLOWER_KEY_FLAGS_MASK, 4, &_mask);
994 }
995 
996 static int fl_dump(struct net *net, struct tcf_proto *tp, unsigned long fh,
997 		   struct sk_buff *skb, struct tcmsg *t)
998 {
999 	struct cls_fl_head *head = rtnl_dereference(tp->root);
1000 	struct cls_fl_filter *f = (struct cls_fl_filter *) fh;
1001 	struct nlattr *nest;
1002 	struct fl_flow_key *key, *mask;
1003 
1004 	if (!f)
1005 		return skb->len;
1006 
1007 	t->tcm_handle = f->handle;
1008 
1009 	nest = nla_nest_start(skb, TCA_OPTIONS);
1010 	if (!nest)
1011 		goto nla_put_failure;
1012 
1013 	if (f->res.classid &&
1014 	    nla_put_u32(skb, TCA_FLOWER_CLASSID, f->res.classid))
1015 		goto nla_put_failure;
1016 
1017 	key = &f->key;
1018 	mask = &head->mask.key;
1019 
1020 	if (mask->indev_ifindex) {
1021 		struct net_device *dev;
1022 
1023 		dev = __dev_get_by_index(net, key->indev_ifindex);
1024 		if (dev && nla_put_string(skb, TCA_FLOWER_INDEV, dev->name))
1025 			goto nla_put_failure;
1026 	}
1027 
1028 	if (!tc_skip_hw(f->flags))
1029 		fl_hw_update_stats(tp, f);
1030 
1031 	if (fl_dump_key_val(skb, key->eth.dst, TCA_FLOWER_KEY_ETH_DST,
1032 			    mask->eth.dst, TCA_FLOWER_KEY_ETH_DST_MASK,
1033 			    sizeof(key->eth.dst)) ||
1034 	    fl_dump_key_val(skb, key->eth.src, TCA_FLOWER_KEY_ETH_SRC,
1035 			    mask->eth.src, TCA_FLOWER_KEY_ETH_SRC_MASK,
1036 			    sizeof(key->eth.src)) ||
1037 	    fl_dump_key_val(skb, &key->basic.n_proto, TCA_FLOWER_KEY_ETH_TYPE,
1038 			    &mask->basic.n_proto, TCA_FLOWER_UNSPEC,
1039 			    sizeof(key->basic.n_proto)))
1040 		goto nla_put_failure;
1041 
1042 	if (fl_dump_key_vlan(skb, &key->vlan, &mask->vlan))
1043 		goto nla_put_failure;
1044 
1045 	if ((key->basic.n_proto == htons(ETH_P_IP) ||
1046 	     key->basic.n_proto == htons(ETH_P_IPV6)) &&
1047 	    fl_dump_key_val(skb, &key->basic.ip_proto, TCA_FLOWER_KEY_IP_PROTO,
1048 			    &mask->basic.ip_proto, TCA_FLOWER_UNSPEC,
1049 			    sizeof(key->basic.ip_proto)))
1050 		goto nla_put_failure;
1051 
1052 	if (key->control.addr_type == FLOW_DISSECTOR_KEY_IPV4_ADDRS &&
1053 	    (fl_dump_key_val(skb, &key->ipv4.src, TCA_FLOWER_KEY_IPV4_SRC,
1054 			     &mask->ipv4.src, TCA_FLOWER_KEY_IPV4_SRC_MASK,
1055 			     sizeof(key->ipv4.src)) ||
1056 	     fl_dump_key_val(skb, &key->ipv4.dst, TCA_FLOWER_KEY_IPV4_DST,
1057 			     &mask->ipv4.dst, TCA_FLOWER_KEY_IPV4_DST_MASK,
1058 			     sizeof(key->ipv4.dst))))
1059 		goto nla_put_failure;
1060 	else if (key->control.addr_type == FLOW_DISSECTOR_KEY_IPV6_ADDRS &&
1061 		 (fl_dump_key_val(skb, &key->ipv6.src, TCA_FLOWER_KEY_IPV6_SRC,
1062 				  &mask->ipv6.src, TCA_FLOWER_KEY_IPV6_SRC_MASK,
1063 				  sizeof(key->ipv6.src)) ||
1064 		  fl_dump_key_val(skb, &key->ipv6.dst, TCA_FLOWER_KEY_IPV6_DST,
1065 				  &mask->ipv6.dst, TCA_FLOWER_KEY_IPV6_DST_MASK,
1066 				  sizeof(key->ipv6.dst))))
1067 		goto nla_put_failure;
1068 
1069 	if (key->basic.ip_proto == IPPROTO_TCP &&
1070 	    (fl_dump_key_val(skb, &key->tp.src, TCA_FLOWER_KEY_TCP_SRC,
1071 			     &mask->tp.src, TCA_FLOWER_KEY_TCP_SRC_MASK,
1072 			     sizeof(key->tp.src)) ||
1073 	     fl_dump_key_val(skb, &key->tp.dst, TCA_FLOWER_KEY_TCP_DST,
1074 			     &mask->tp.dst, TCA_FLOWER_KEY_TCP_DST_MASK,
1075 			     sizeof(key->tp.dst))))
1076 		goto nla_put_failure;
1077 	else if (key->basic.ip_proto == IPPROTO_UDP &&
1078 		 (fl_dump_key_val(skb, &key->tp.src, TCA_FLOWER_KEY_UDP_SRC,
1079 				  &mask->tp.src, TCA_FLOWER_KEY_UDP_SRC_MASK,
1080 				  sizeof(key->tp.src)) ||
1081 		  fl_dump_key_val(skb, &key->tp.dst, TCA_FLOWER_KEY_UDP_DST,
1082 				  &mask->tp.dst, TCA_FLOWER_KEY_UDP_DST_MASK,
1083 				  sizeof(key->tp.dst))))
1084 		goto nla_put_failure;
1085 	else if (key->basic.ip_proto == IPPROTO_SCTP &&
1086 		 (fl_dump_key_val(skb, &key->tp.src, TCA_FLOWER_KEY_SCTP_SRC,
1087 				  &mask->tp.src, TCA_FLOWER_KEY_SCTP_SRC_MASK,
1088 				  sizeof(key->tp.src)) ||
1089 		  fl_dump_key_val(skb, &key->tp.dst, TCA_FLOWER_KEY_SCTP_DST,
1090 				  &mask->tp.dst, TCA_FLOWER_KEY_SCTP_DST_MASK,
1091 				  sizeof(key->tp.dst))))
1092 		goto nla_put_failure;
1093 	else if (key->basic.n_proto == htons(ETH_P_IP) &&
1094 		 key->basic.ip_proto == IPPROTO_ICMP &&
1095 		 (fl_dump_key_val(skb, &key->icmp.type,
1096 				  TCA_FLOWER_KEY_ICMPV4_TYPE, &mask->icmp.type,
1097 				  TCA_FLOWER_KEY_ICMPV4_TYPE_MASK,
1098 				  sizeof(key->icmp.type)) ||
1099 		  fl_dump_key_val(skb, &key->icmp.code,
1100 				  TCA_FLOWER_KEY_ICMPV4_CODE, &mask->icmp.code,
1101 				  TCA_FLOWER_KEY_ICMPV4_CODE_MASK,
1102 				  sizeof(key->icmp.code))))
1103 		goto nla_put_failure;
1104 	else if (key->basic.n_proto == htons(ETH_P_IPV6) &&
1105 		 key->basic.ip_proto == IPPROTO_ICMPV6 &&
1106 		 (fl_dump_key_val(skb, &key->icmp.type,
1107 				  TCA_FLOWER_KEY_ICMPV6_TYPE, &mask->icmp.type,
1108 				  TCA_FLOWER_KEY_ICMPV6_TYPE_MASK,
1109 				  sizeof(key->icmp.type)) ||
1110 		  fl_dump_key_val(skb, &key->icmp.code,
1111 				  TCA_FLOWER_KEY_ICMPV6_CODE, &mask->icmp.code,
1112 				  TCA_FLOWER_KEY_ICMPV6_CODE_MASK,
1113 				  sizeof(key->icmp.code))))
1114 		goto nla_put_failure;
1115 
1116 	if (key->enc_control.addr_type == FLOW_DISSECTOR_KEY_IPV4_ADDRS &&
1117 	    (fl_dump_key_val(skb, &key->enc_ipv4.src,
1118 			    TCA_FLOWER_KEY_ENC_IPV4_SRC, &mask->enc_ipv4.src,
1119 			    TCA_FLOWER_KEY_ENC_IPV4_SRC_MASK,
1120 			    sizeof(key->enc_ipv4.src)) ||
1121 	     fl_dump_key_val(skb, &key->enc_ipv4.dst,
1122 			     TCA_FLOWER_KEY_ENC_IPV4_DST, &mask->enc_ipv4.dst,
1123 			     TCA_FLOWER_KEY_ENC_IPV4_DST_MASK,
1124 			     sizeof(key->enc_ipv4.dst))))
1125 		goto nla_put_failure;
1126 	else if (key->enc_control.addr_type == FLOW_DISSECTOR_KEY_IPV6_ADDRS &&
1127 		 (fl_dump_key_val(skb, &key->enc_ipv6.src,
1128 			    TCA_FLOWER_KEY_ENC_IPV6_SRC, &mask->enc_ipv6.src,
1129 			    TCA_FLOWER_KEY_ENC_IPV6_SRC_MASK,
1130 			    sizeof(key->enc_ipv6.src)) ||
1131 		 fl_dump_key_val(skb, &key->enc_ipv6.dst,
1132 				 TCA_FLOWER_KEY_ENC_IPV6_DST,
1133 				 &mask->enc_ipv6.dst,
1134 				 TCA_FLOWER_KEY_ENC_IPV6_DST_MASK,
1135 			    sizeof(key->enc_ipv6.dst))))
1136 		goto nla_put_failure;
1137 
1138 	if (fl_dump_key_val(skb, &key->enc_key_id, TCA_FLOWER_KEY_ENC_KEY_ID,
1139 			    &mask->enc_key_id, TCA_FLOWER_UNSPEC,
1140 			    sizeof(key->enc_key_id)) ||
1141 	    fl_dump_key_val(skb, &key->enc_tp.src,
1142 			    TCA_FLOWER_KEY_ENC_UDP_SRC_PORT,
1143 			    &mask->enc_tp.src,
1144 			    TCA_FLOWER_KEY_ENC_UDP_SRC_PORT_MASK,
1145 			    sizeof(key->enc_tp.src)) ||
1146 	    fl_dump_key_val(skb, &key->enc_tp.dst,
1147 			    TCA_FLOWER_KEY_ENC_UDP_DST_PORT,
1148 			    &mask->enc_tp.dst,
1149 			    TCA_FLOWER_KEY_ENC_UDP_DST_PORT_MASK,
1150 			    sizeof(key->enc_tp.dst)))
1151 		goto nla_put_failure;
1152 
1153 	if (fl_dump_key_flags(skb, key->control.flags, mask->control.flags))
1154 		goto nla_put_failure;
1155 
1156 	nla_put_u32(skb, TCA_FLOWER_FLAGS, f->flags);
1157 
1158 	if (tcf_exts_dump(skb, &f->exts))
1159 		goto nla_put_failure;
1160 
1161 	nla_nest_end(skb, nest);
1162 
1163 	if (tcf_exts_dump_stats(skb, &f->exts) < 0)
1164 		goto nla_put_failure;
1165 
1166 	return skb->len;
1167 
1168 nla_put_failure:
1169 	nla_nest_cancel(skb, nest);
1170 	return -1;
1171 }
1172 
1173 static struct tcf_proto_ops cls_fl_ops __read_mostly = {
1174 	.kind		= "flower",
1175 	.classify	= fl_classify,
1176 	.init		= fl_init,
1177 	.destroy	= fl_destroy,
1178 	.get		= fl_get,
1179 	.change		= fl_change,
1180 	.delete		= fl_delete,
1181 	.walk		= fl_walk,
1182 	.dump		= fl_dump,
1183 	.owner		= THIS_MODULE,
1184 };
1185 
1186 static int __init cls_fl_init(void)
1187 {
1188 	return register_tcf_proto_ops(&cls_fl_ops);
1189 }
1190 
1191 static void __exit cls_fl_exit(void)
1192 {
1193 	unregister_tcf_proto_ops(&cls_fl_ops);
1194 }
1195 
1196 module_init(cls_fl_init);
1197 module_exit(cls_fl_exit);
1198 
1199 MODULE_AUTHOR("Jiri Pirko <jiri@resnulli.us>");
1200 MODULE_DESCRIPTION("Flower classifier");
1201 MODULE_LICENSE("GPL v2");
1202