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