xref: /openbmc/linux/net/sched/cls_flower.c (revision a17922de)
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 #include <linux/mpls.h>
22 
23 #include <net/sch_generic.h>
24 #include <net/pkt_cls.h>
25 #include <net/ip.h>
26 #include <net/flow_dissector.h>
27 
28 #include <net/dst.h>
29 #include <net/dst_metadata.h>
30 
31 struct fl_flow_key {
32 	int	indev_ifindex;
33 	struct flow_dissector_key_control control;
34 	struct flow_dissector_key_control enc_control;
35 	struct flow_dissector_key_basic basic;
36 	struct flow_dissector_key_eth_addrs eth;
37 	struct flow_dissector_key_vlan vlan;
38 	struct flow_dissector_key_vlan cvlan;
39 	union {
40 		struct flow_dissector_key_ipv4_addrs ipv4;
41 		struct flow_dissector_key_ipv6_addrs ipv6;
42 	};
43 	struct flow_dissector_key_ports tp;
44 	struct flow_dissector_key_icmp icmp;
45 	struct flow_dissector_key_arp arp;
46 	struct flow_dissector_key_keyid enc_key_id;
47 	union {
48 		struct flow_dissector_key_ipv4_addrs enc_ipv4;
49 		struct flow_dissector_key_ipv6_addrs enc_ipv6;
50 	};
51 	struct flow_dissector_key_ports enc_tp;
52 	struct flow_dissector_key_mpls mpls;
53 	struct flow_dissector_key_tcp tcp;
54 	struct flow_dissector_key_ip ip;
55 	struct flow_dissector_key_ip enc_ip;
56 } __aligned(BITS_PER_LONG / 8); /* Ensure that we can do comparisons as longs. */
57 
58 struct fl_flow_mask_range {
59 	unsigned short int start;
60 	unsigned short int end;
61 };
62 
63 struct fl_flow_mask {
64 	struct fl_flow_key key;
65 	struct fl_flow_mask_range range;
66 	struct rhash_head ht_node;
67 	struct rhashtable ht;
68 	struct rhashtable_params filter_ht_params;
69 	struct flow_dissector dissector;
70 	struct list_head filters;
71 	struct rcu_work rwork;
72 	struct list_head list;
73 };
74 
75 struct fl_flow_tmplt {
76 	struct fl_flow_key dummy_key;
77 	struct fl_flow_key mask;
78 	struct flow_dissector dissector;
79 	struct tcf_chain *chain;
80 };
81 
82 struct cls_fl_head {
83 	struct rhashtable ht;
84 	struct list_head masks;
85 	struct rcu_work rwork;
86 	struct idr handle_idr;
87 };
88 
89 struct cls_fl_filter {
90 	struct fl_flow_mask *mask;
91 	struct rhash_head ht_node;
92 	struct fl_flow_key mkey;
93 	struct tcf_exts exts;
94 	struct tcf_result res;
95 	struct fl_flow_key key;
96 	struct list_head list;
97 	u32 handle;
98 	u32 flags;
99 	unsigned int in_hw_count;
100 	struct rcu_work rwork;
101 	struct net_device *hw_dev;
102 };
103 
104 static const struct rhashtable_params mask_ht_params = {
105 	.key_offset = offsetof(struct fl_flow_mask, key),
106 	.key_len = sizeof(struct fl_flow_key),
107 	.head_offset = offsetof(struct fl_flow_mask, ht_node),
108 	.automatic_shrinking = true,
109 };
110 
111 static unsigned short int fl_mask_range(const struct fl_flow_mask *mask)
112 {
113 	return mask->range.end - mask->range.start;
114 }
115 
116 static void fl_mask_update_range(struct fl_flow_mask *mask)
117 {
118 	const u8 *bytes = (const u8 *) &mask->key;
119 	size_t size = sizeof(mask->key);
120 	size_t i, first = 0, last;
121 
122 	for (i = 0; i < size; i++) {
123 		if (bytes[i]) {
124 			first = i;
125 			break;
126 		}
127 	}
128 	last = first;
129 	for (i = size - 1; i != first; i--) {
130 		if (bytes[i]) {
131 			last = i;
132 			break;
133 		}
134 	}
135 	mask->range.start = rounddown(first, sizeof(long));
136 	mask->range.end = roundup(last + 1, sizeof(long));
137 }
138 
139 static void *fl_key_get_start(struct fl_flow_key *key,
140 			      const struct fl_flow_mask *mask)
141 {
142 	return (u8 *) key + mask->range.start;
143 }
144 
145 static void fl_set_masked_key(struct fl_flow_key *mkey, struct fl_flow_key *key,
146 			      struct fl_flow_mask *mask)
147 {
148 	const long *lkey = fl_key_get_start(key, mask);
149 	const long *lmask = fl_key_get_start(&mask->key, mask);
150 	long *lmkey = fl_key_get_start(mkey, mask);
151 	int i;
152 
153 	for (i = 0; i < fl_mask_range(mask); i += sizeof(long))
154 		*lmkey++ = *lkey++ & *lmask++;
155 }
156 
157 static bool fl_mask_fits_tmplt(struct fl_flow_tmplt *tmplt,
158 			       struct fl_flow_mask *mask)
159 {
160 	const long *lmask = fl_key_get_start(&mask->key, mask);
161 	const long *ltmplt;
162 	int i;
163 
164 	if (!tmplt)
165 		return true;
166 	ltmplt = fl_key_get_start(&tmplt->mask, mask);
167 	for (i = 0; i < fl_mask_range(mask); i += sizeof(long)) {
168 		if (~*ltmplt++ & *lmask++)
169 			return false;
170 	}
171 	return true;
172 }
173 
174 static void fl_clear_masked_range(struct fl_flow_key *key,
175 				  struct fl_flow_mask *mask)
176 {
177 	memset(fl_key_get_start(key, mask), 0, fl_mask_range(mask));
178 }
179 
180 static struct cls_fl_filter *fl_lookup(struct fl_flow_mask *mask,
181 				       struct fl_flow_key *mkey)
182 {
183 	return rhashtable_lookup_fast(&mask->ht, fl_key_get_start(mkey, mask),
184 				      mask->filter_ht_params);
185 }
186 
187 static int fl_classify(struct sk_buff *skb, const struct tcf_proto *tp,
188 		       struct tcf_result *res)
189 {
190 	struct cls_fl_head *head = rcu_dereference_bh(tp->root);
191 	struct cls_fl_filter *f;
192 	struct fl_flow_mask *mask;
193 	struct fl_flow_key skb_key;
194 	struct fl_flow_key skb_mkey;
195 
196 	list_for_each_entry_rcu(mask, &head->masks, list) {
197 		fl_clear_masked_range(&skb_key, mask);
198 
199 		skb_key.indev_ifindex = skb->skb_iif;
200 		/* skb_flow_dissect() does not set n_proto in case an unknown
201 		 * protocol, so do it rather here.
202 		 */
203 		skb_key.basic.n_proto = skb->protocol;
204 		skb_flow_dissect_tunnel_info(skb, &mask->dissector, &skb_key);
205 		skb_flow_dissect(skb, &mask->dissector, &skb_key, 0);
206 
207 		fl_set_masked_key(&skb_mkey, &skb_key, mask);
208 
209 		f = fl_lookup(mask, &skb_mkey);
210 		if (f && !tc_skip_sw(f->flags)) {
211 			*res = f->res;
212 			return tcf_exts_exec(skb, &f->exts, res);
213 		}
214 	}
215 	return -1;
216 }
217 
218 static int fl_init(struct tcf_proto *tp)
219 {
220 	struct cls_fl_head *head;
221 
222 	head = kzalloc(sizeof(*head), GFP_KERNEL);
223 	if (!head)
224 		return -ENOBUFS;
225 
226 	INIT_LIST_HEAD_RCU(&head->masks);
227 	rcu_assign_pointer(tp->root, head);
228 	idr_init(&head->handle_idr);
229 
230 	return rhashtable_init(&head->ht, &mask_ht_params);
231 }
232 
233 static void fl_mask_free(struct fl_flow_mask *mask)
234 {
235 	rhashtable_destroy(&mask->ht);
236 	kfree(mask);
237 }
238 
239 static void fl_mask_free_work(struct work_struct *work)
240 {
241 	struct fl_flow_mask *mask = container_of(to_rcu_work(work),
242 						 struct fl_flow_mask, rwork);
243 
244 	fl_mask_free(mask);
245 }
246 
247 static bool fl_mask_put(struct cls_fl_head *head, struct fl_flow_mask *mask,
248 			bool async)
249 {
250 	if (!list_empty(&mask->filters))
251 		return false;
252 
253 	rhashtable_remove_fast(&head->ht, &mask->ht_node, mask_ht_params);
254 	list_del_rcu(&mask->list);
255 	if (async)
256 		tcf_queue_work(&mask->rwork, fl_mask_free_work);
257 	else
258 		fl_mask_free(mask);
259 
260 	return true;
261 }
262 
263 static void __fl_destroy_filter(struct cls_fl_filter *f)
264 {
265 	tcf_exts_destroy(&f->exts);
266 	tcf_exts_put_net(&f->exts);
267 	kfree(f);
268 }
269 
270 static void fl_destroy_filter_work(struct work_struct *work)
271 {
272 	struct cls_fl_filter *f = container_of(to_rcu_work(work),
273 					struct cls_fl_filter, rwork);
274 
275 	rtnl_lock();
276 	__fl_destroy_filter(f);
277 	rtnl_unlock();
278 }
279 
280 static void fl_hw_destroy_filter(struct tcf_proto *tp, struct cls_fl_filter *f,
281 				 struct netlink_ext_ack *extack)
282 {
283 	struct tc_cls_flower_offload cls_flower = {};
284 	struct tcf_block *block = tp->chain->block;
285 
286 	tc_cls_common_offload_init(&cls_flower.common, tp, f->flags, extack);
287 	cls_flower.command = TC_CLSFLOWER_DESTROY;
288 	cls_flower.cookie = (unsigned long) f;
289 
290 	tc_setup_cb_call(block, &f->exts, TC_SETUP_CLSFLOWER,
291 			 &cls_flower, false);
292 	tcf_block_offload_dec(block, &f->flags);
293 }
294 
295 static int fl_hw_replace_filter(struct tcf_proto *tp,
296 				struct cls_fl_filter *f,
297 				struct netlink_ext_ack *extack)
298 {
299 	struct tc_cls_flower_offload cls_flower = {};
300 	struct tcf_block *block = tp->chain->block;
301 	bool skip_sw = tc_skip_sw(f->flags);
302 	int err;
303 
304 	tc_cls_common_offload_init(&cls_flower.common, tp, f->flags, extack);
305 	cls_flower.command = TC_CLSFLOWER_REPLACE;
306 	cls_flower.cookie = (unsigned long) f;
307 	cls_flower.dissector = &f->mask->dissector;
308 	cls_flower.mask = &f->mask->key;
309 	cls_flower.key = &f->mkey;
310 	cls_flower.exts = &f->exts;
311 	cls_flower.classid = f->res.classid;
312 
313 	err = tc_setup_cb_call(block, &f->exts, TC_SETUP_CLSFLOWER,
314 			       &cls_flower, skip_sw);
315 	if (err < 0) {
316 		fl_hw_destroy_filter(tp, f, NULL);
317 		return err;
318 	} else if (err > 0) {
319 		f->in_hw_count = err;
320 		tcf_block_offload_inc(block, &f->flags);
321 	}
322 
323 	if (skip_sw && !(f->flags & TCA_CLS_FLAGS_IN_HW))
324 		return -EINVAL;
325 
326 	return 0;
327 }
328 
329 static void fl_hw_update_stats(struct tcf_proto *tp, struct cls_fl_filter *f)
330 {
331 	struct tc_cls_flower_offload cls_flower = {};
332 	struct tcf_block *block = tp->chain->block;
333 
334 	tc_cls_common_offload_init(&cls_flower.common, tp, f->flags, NULL);
335 	cls_flower.command = TC_CLSFLOWER_STATS;
336 	cls_flower.cookie = (unsigned long) f;
337 	cls_flower.exts = &f->exts;
338 	cls_flower.classid = f->res.classid;
339 
340 	tc_setup_cb_call(block, &f->exts, TC_SETUP_CLSFLOWER,
341 			 &cls_flower, false);
342 }
343 
344 static bool __fl_delete(struct tcf_proto *tp, struct cls_fl_filter *f,
345 			struct netlink_ext_ack *extack)
346 {
347 	struct cls_fl_head *head = rtnl_dereference(tp->root);
348 	bool async = tcf_exts_get_net(&f->exts);
349 	bool last;
350 
351 	idr_remove(&head->handle_idr, f->handle);
352 	list_del_rcu(&f->list);
353 	last = fl_mask_put(head, f->mask, async);
354 	if (!tc_skip_hw(f->flags))
355 		fl_hw_destroy_filter(tp, f, extack);
356 	tcf_unbind_filter(tp, &f->res);
357 	if (async)
358 		tcf_queue_work(&f->rwork, fl_destroy_filter_work);
359 	else
360 		__fl_destroy_filter(f);
361 
362 	return last;
363 }
364 
365 static void fl_destroy_sleepable(struct work_struct *work)
366 {
367 	struct cls_fl_head *head = container_of(to_rcu_work(work),
368 						struct cls_fl_head,
369 						rwork);
370 
371 	rhashtable_destroy(&head->ht);
372 	kfree(head);
373 	module_put(THIS_MODULE);
374 }
375 
376 static void fl_destroy(struct tcf_proto *tp, struct netlink_ext_ack *extack)
377 {
378 	struct cls_fl_head *head = rtnl_dereference(tp->root);
379 	struct fl_flow_mask *mask, *next_mask;
380 	struct cls_fl_filter *f, *next;
381 
382 	list_for_each_entry_safe(mask, next_mask, &head->masks, list) {
383 		list_for_each_entry_safe(f, next, &mask->filters, list) {
384 			if (__fl_delete(tp, f, extack))
385 				break;
386 		}
387 	}
388 	idr_destroy(&head->handle_idr);
389 
390 	__module_get(THIS_MODULE);
391 	tcf_queue_work(&head->rwork, fl_destroy_sleepable);
392 }
393 
394 static void *fl_get(struct tcf_proto *tp, u32 handle)
395 {
396 	struct cls_fl_head *head = rtnl_dereference(tp->root);
397 
398 	return idr_find(&head->handle_idr, handle);
399 }
400 
401 static const struct nla_policy fl_policy[TCA_FLOWER_MAX + 1] = {
402 	[TCA_FLOWER_UNSPEC]		= { .type = NLA_UNSPEC },
403 	[TCA_FLOWER_CLASSID]		= { .type = NLA_U32 },
404 	[TCA_FLOWER_INDEV]		= { .type = NLA_STRING,
405 					    .len = IFNAMSIZ },
406 	[TCA_FLOWER_KEY_ETH_DST]	= { .len = ETH_ALEN },
407 	[TCA_FLOWER_KEY_ETH_DST_MASK]	= { .len = ETH_ALEN },
408 	[TCA_FLOWER_KEY_ETH_SRC]	= { .len = ETH_ALEN },
409 	[TCA_FLOWER_KEY_ETH_SRC_MASK]	= { .len = ETH_ALEN },
410 	[TCA_FLOWER_KEY_ETH_TYPE]	= { .type = NLA_U16 },
411 	[TCA_FLOWER_KEY_IP_PROTO]	= { .type = NLA_U8 },
412 	[TCA_FLOWER_KEY_IPV4_SRC]	= { .type = NLA_U32 },
413 	[TCA_FLOWER_KEY_IPV4_SRC_MASK]	= { .type = NLA_U32 },
414 	[TCA_FLOWER_KEY_IPV4_DST]	= { .type = NLA_U32 },
415 	[TCA_FLOWER_KEY_IPV4_DST_MASK]	= { .type = NLA_U32 },
416 	[TCA_FLOWER_KEY_IPV6_SRC]	= { .len = sizeof(struct in6_addr) },
417 	[TCA_FLOWER_KEY_IPV6_SRC_MASK]	= { .len = sizeof(struct in6_addr) },
418 	[TCA_FLOWER_KEY_IPV6_DST]	= { .len = sizeof(struct in6_addr) },
419 	[TCA_FLOWER_KEY_IPV6_DST_MASK]	= { .len = sizeof(struct in6_addr) },
420 	[TCA_FLOWER_KEY_TCP_SRC]	= { .type = NLA_U16 },
421 	[TCA_FLOWER_KEY_TCP_DST]	= { .type = NLA_U16 },
422 	[TCA_FLOWER_KEY_UDP_SRC]	= { .type = NLA_U16 },
423 	[TCA_FLOWER_KEY_UDP_DST]	= { .type = NLA_U16 },
424 	[TCA_FLOWER_KEY_VLAN_ID]	= { .type = NLA_U16 },
425 	[TCA_FLOWER_KEY_VLAN_PRIO]	= { .type = NLA_U8 },
426 	[TCA_FLOWER_KEY_VLAN_ETH_TYPE]	= { .type = NLA_U16 },
427 	[TCA_FLOWER_KEY_ENC_KEY_ID]	= { .type = NLA_U32 },
428 	[TCA_FLOWER_KEY_ENC_IPV4_SRC]	= { .type = NLA_U32 },
429 	[TCA_FLOWER_KEY_ENC_IPV4_SRC_MASK] = { .type = NLA_U32 },
430 	[TCA_FLOWER_KEY_ENC_IPV4_DST]	= { .type = NLA_U32 },
431 	[TCA_FLOWER_KEY_ENC_IPV4_DST_MASK] = { .type = NLA_U32 },
432 	[TCA_FLOWER_KEY_ENC_IPV6_SRC]	= { .len = sizeof(struct in6_addr) },
433 	[TCA_FLOWER_KEY_ENC_IPV6_SRC_MASK] = { .len = sizeof(struct in6_addr) },
434 	[TCA_FLOWER_KEY_ENC_IPV6_DST]	= { .len = sizeof(struct in6_addr) },
435 	[TCA_FLOWER_KEY_ENC_IPV6_DST_MASK] = { .len = sizeof(struct in6_addr) },
436 	[TCA_FLOWER_KEY_TCP_SRC_MASK]	= { .type = NLA_U16 },
437 	[TCA_FLOWER_KEY_TCP_DST_MASK]	= { .type = NLA_U16 },
438 	[TCA_FLOWER_KEY_UDP_SRC_MASK]	= { .type = NLA_U16 },
439 	[TCA_FLOWER_KEY_UDP_DST_MASK]	= { .type = NLA_U16 },
440 	[TCA_FLOWER_KEY_SCTP_SRC_MASK]	= { .type = NLA_U16 },
441 	[TCA_FLOWER_KEY_SCTP_DST_MASK]	= { .type = NLA_U16 },
442 	[TCA_FLOWER_KEY_SCTP_SRC]	= { .type = NLA_U16 },
443 	[TCA_FLOWER_KEY_SCTP_DST]	= { .type = NLA_U16 },
444 	[TCA_FLOWER_KEY_ENC_UDP_SRC_PORT]	= { .type = NLA_U16 },
445 	[TCA_FLOWER_KEY_ENC_UDP_SRC_PORT_MASK]	= { .type = NLA_U16 },
446 	[TCA_FLOWER_KEY_ENC_UDP_DST_PORT]	= { .type = NLA_U16 },
447 	[TCA_FLOWER_KEY_ENC_UDP_DST_PORT_MASK]	= { .type = NLA_U16 },
448 	[TCA_FLOWER_KEY_FLAGS]		= { .type = NLA_U32 },
449 	[TCA_FLOWER_KEY_FLAGS_MASK]	= { .type = NLA_U32 },
450 	[TCA_FLOWER_KEY_ICMPV4_TYPE]	= { .type = NLA_U8 },
451 	[TCA_FLOWER_KEY_ICMPV4_TYPE_MASK] = { .type = NLA_U8 },
452 	[TCA_FLOWER_KEY_ICMPV4_CODE]	= { .type = NLA_U8 },
453 	[TCA_FLOWER_KEY_ICMPV4_CODE_MASK] = { .type = NLA_U8 },
454 	[TCA_FLOWER_KEY_ICMPV6_TYPE]	= { .type = NLA_U8 },
455 	[TCA_FLOWER_KEY_ICMPV6_TYPE_MASK] = { .type = NLA_U8 },
456 	[TCA_FLOWER_KEY_ICMPV6_CODE]	= { .type = NLA_U8 },
457 	[TCA_FLOWER_KEY_ICMPV6_CODE_MASK] = { .type = NLA_U8 },
458 	[TCA_FLOWER_KEY_ARP_SIP]	= { .type = NLA_U32 },
459 	[TCA_FLOWER_KEY_ARP_SIP_MASK]	= { .type = NLA_U32 },
460 	[TCA_FLOWER_KEY_ARP_TIP]	= { .type = NLA_U32 },
461 	[TCA_FLOWER_KEY_ARP_TIP_MASK]	= { .type = NLA_U32 },
462 	[TCA_FLOWER_KEY_ARP_OP]		= { .type = NLA_U8 },
463 	[TCA_FLOWER_KEY_ARP_OP_MASK]	= { .type = NLA_U8 },
464 	[TCA_FLOWER_KEY_ARP_SHA]	= { .len = ETH_ALEN },
465 	[TCA_FLOWER_KEY_ARP_SHA_MASK]	= { .len = ETH_ALEN },
466 	[TCA_FLOWER_KEY_ARP_THA]	= { .len = ETH_ALEN },
467 	[TCA_FLOWER_KEY_ARP_THA_MASK]	= { .len = ETH_ALEN },
468 	[TCA_FLOWER_KEY_MPLS_TTL]	= { .type = NLA_U8 },
469 	[TCA_FLOWER_KEY_MPLS_BOS]	= { .type = NLA_U8 },
470 	[TCA_FLOWER_KEY_MPLS_TC]	= { .type = NLA_U8 },
471 	[TCA_FLOWER_KEY_MPLS_LABEL]	= { .type = NLA_U32 },
472 	[TCA_FLOWER_KEY_TCP_FLAGS]	= { .type = NLA_U16 },
473 	[TCA_FLOWER_KEY_TCP_FLAGS_MASK]	= { .type = NLA_U16 },
474 	[TCA_FLOWER_KEY_IP_TOS]		= { .type = NLA_U8 },
475 	[TCA_FLOWER_KEY_IP_TOS_MASK]	= { .type = NLA_U8 },
476 	[TCA_FLOWER_KEY_IP_TTL]		= { .type = NLA_U8 },
477 	[TCA_FLOWER_KEY_IP_TTL_MASK]	= { .type = NLA_U8 },
478 	[TCA_FLOWER_KEY_CVLAN_ID]	= { .type = NLA_U16 },
479 	[TCA_FLOWER_KEY_CVLAN_PRIO]	= { .type = NLA_U8 },
480 	[TCA_FLOWER_KEY_CVLAN_ETH_TYPE]	= { .type = NLA_U16 },
481 	[TCA_FLOWER_KEY_ENC_IP_TOS]	= { .type = NLA_U8 },
482 	[TCA_FLOWER_KEY_ENC_IP_TOS_MASK] = { .type = NLA_U8 },
483 	[TCA_FLOWER_KEY_ENC_IP_TTL]	 = { .type = NLA_U8 },
484 	[TCA_FLOWER_KEY_ENC_IP_TTL_MASK] = { .type = NLA_U8 },
485 };
486 
487 static void fl_set_key_val(struct nlattr **tb,
488 			   void *val, int val_type,
489 			   void *mask, int mask_type, int len)
490 {
491 	if (!tb[val_type])
492 		return;
493 	memcpy(val, nla_data(tb[val_type]), len);
494 	if (mask_type == TCA_FLOWER_UNSPEC || !tb[mask_type])
495 		memset(mask, 0xff, len);
496 	else
497 		memcpy(mask, nla_data(tb[mask_type]), len);
498 }
499 
500 static int fl_set_key_mpls(struct nlattr **tb,
501 			   struct flow_dissector_key_mpls *key_val,
502 			   struct flow_dissector_key_mpls *key_mask)
503 {
504 	if (tb[TCA_FLOWER_KEY_MPLS_TTL]) {
505 		key_val->mpls_ttl = nla_get_u8(tb[TCA_FLOWER_KEY_MPLS_TTL]);
506 		key_mask->mpls_ttl = MPLS_TTL_MASK;
507 	}
508 	if (tb[TCA_FLOWER_KEY_MPLS_BOS]) {
509 		u8 bos = nla_get_u8(tb[TCA_FLOWER_KEY_MPLS_BOS]);
510 
511 		if (bos & ~MPLS_BOS_MASK)
512 			return -EINVAL;
513 		key_val->mpls_bos = bos;
514 		key_mask->mpls_bos = MPLS_BOS_MASK;
515 	}
516 	if (tb[TCA_FLOWER_KEY_MPLS_TC]) {
517 		u8 tc = nla_get_u8(tb[TCA_FLOWER_KEY_MPLS_TC]);
518 
519 		if (tc & ~MPLS_TC_MASK)
520 			return -EINVAL;
521 		key_val->mpls_tc = tc;
522 		key_mask->mpls_tc = MPLS_TC_MASK;
523 	}
524 	if (tb[TCA_FLOWER_KEY_MPLS_LABEL]) {
525 		u32 label = nla_get_u32(tb[TCA_FLOWER_KEY_MPLS_LABEL]);
526 
527 		if (label & ~MPLS_LABEL_MASK)
528 			return -EINVAL;
529 		key_val->mpls_label = label;
530 		key_mask->mpls_label = MPLS_LABEL_MASK;
531 	}
532 	return 0;
533 }
534 
535 static void fl_set_key_vlan(struct nlattr **tb,
536 			    __be16 ethertype,
537 			    int vlan_id_key, int vlan_prio_key,
538 			    struct flow_dissector_key_vlan *key_val,
539 			    struct flow_dissector_key_vlan *key_mask)
540 {
541 #define VLAN_PRIORITY_MASK	0x7
542 
543 	if (tb[vlan_id_key]) {
544 		key_val->vlan_id =
545 			nla_get_u16(tb[vlan_id_key]) & VLAN_VID_MASK;
546 		key_mask->vlan_id = VLAN_VID_MASK;
547 	}
548 	if (tb[vlan_prio_key]) {
549 		key_val->vlan_priority =
550 			nla_get_u8(tb[vlan_prio_key]) &
551 			VLAN_PRIORITY_MASK;
552 		key_mask->vlan_priority = VLAN_PRIORITY_MASK;
553 	}
554 	key_val->vlan_tpid = ethertype;
555 	key_mask->vlan_tpid = cpu_to_be16(~0);
556 }
557 
558 static void fl_set_key_flag(u32 flower_key, u32 flower_mask,
559 			    u32 *dissector_key, u32 *dissector_mask,
560 			    u32 flower_flag_bit, u32 dissector_flag_bit)
561 {
562 	if (flower_mask & flower_flag_bit) {
563 		*dissector_mask |= dissector_flag_bit;
564 		if (flower_key & flower_flag_bit)
565 			*dissector_key |= dissector_flag_bit;
566 	}
567 }
568 
569 static int fl_set_key_flags(struct nlattr **tb,
570 			    u32 *flags_key, u32 *flags_mask)
571 {
572 	u32 key, mask;
573 
574 	/* mask is mandatory for flags */
575 	if (!tb[TCA_FLOWER_KEY_FLAGS_MASK])
576 		return -EINVAL;
577 
578 	key = be32_to_cpu(nla_get_u32(tb[TCA_FLOWER_KEY_FLAGS]));
579 	mask = be32_to_cpu(nla_get_u32(tb[TCA_FLOWER_KEY_FLAGS_MASK]));
580 
581 	*flags_key  = 0;
582 	*flags_mask = 0;
583 
584 	fl_set_key_flag(key, mask, flags_key, flags_mask,
585 			TCA_FLOWER_KEY_FLAGS_IS_FRAGMENT, FLOW_DIS_IS_FRAGMENT);
586 	fl_set_key_flag(key, mask, flags_key, flags_mask,
587 			TCA_FLOWER_KEY_FLAGS_FRAG_IS_FIRST,
588 			FLOW_DIS_FIRST_FRAG);
589 
590 	return 0;
591 }
592 
593 static void fl_set_key_ip(struct nlattr **tb, bool encap,
594 			  struct flow_dissector_key_ip *key,
595 			  struct flow_dissector_key_ip *mask)
596 {
597 	int tos_key = encap ? TCA_FLOWER_KEY_ENC_IP_TOS : TCA_FLOWER_KEY_IP_TOS;
598 	int ttl_key = encap ? TCA_FLOWER_KEY_ENC_IP_TTL : TCA_FLOWER_KEY_IP_TTL;
599 	int tos_mask = encap ? TCA_FLOWER_KEY_ENC_IP_TOS_MASK : TCA_FLOWER_KEY_IP_TOS_MASK;
600 	int ttl_mask = encap ? TCA_FLOWER_KEY_ENC_IP_TTL_MASK : TCA_FLOWER_KEY_IP_TTL_MASK;
601 
602 	fl_set_key_val(tb, &key->tos, tos_key, &mask->tos, tos_mask, sizeof(key->tos));
603 	fl_set_key_val(tb, &key->ttl, ttl_key, &mask->ttl, ttl_mask, sizeof(key->ttl));
604 }
605 
606 static int fl_set_key(struct net *net, struct nlattr **tb,
607 		      struct fl_flow_key *key, struct fl_flow_key *mask,
608 		      struct netlink_ext_ack *extack)
609 {
610 	__be16 ethertype;
611 	int ret = 0;
612 #ifdef CONFIG_NET_CLS_IND
613 	if (tb[TCA_FLOWER_INDEV]) {
614 		int err = tcf_change_indev(net, tb[TCA_FLOWER_INDEV], extack);
615 		if (err < 0)
616 			return err;
617 		key->indev_ifindex = err;
618 		mask->indev_ifindex = 0xffffffff;
619 	}
620 #endif
621 
622 	fl_set_key_val(tb, key->eth.dst, TCA_FLOWER_KEY_ETH_DST,
623 		       mask->eth.dst, TCA_FLOWER_KEY_ETH_DST_MASK,
624 		       sizeof(key->eth.dst));
625 	fl_set_key_val(tb, key->eth.src, TCA_FLOWER_KEY_ETH_SRC,
626 		       mask->eth.src, TCA_FLOWER_KEY_ETH_SRC_MASK,
627 		       sizeof(key->eth.src));
628 
629 	if (tb[TCA_FLOWER_KEY_ETH_TYPE]) {
630 		ethertype = nla_get_be16(tb[TCA_FLOWER_KEY_ETH_TYPE]);
631 
632 		if (eth_type_vlan(ethertype)) {
633 			fl_set_key_vlan(tb, ethertype, TCA_FLOWER_KEY_VLAN_ID,
634 					TCA_FLOWER_KEY_VLAN_PRIO, &key->vlan,
635 					&mask->vlan);
636 
637 			if (tb[TCA_FLOWER_KEY_VLAN_ETH_TYPE]) {
638 				ethertype = nla_get_be16(tb[TCA_FLOWER_KEY_VLAN_ETH_TYPE]);
639 				if (eth_type_vlan(ethertype)) {
640 					fl_set_key_vlan(tb, ethertype,
641 							TCA_FLOWER_KEY_CVLAN_ID,
642 							TCA_FLOWER_KEY_CVLAN_PRIO,
643 							&key->cvlan, &mask->cvlan);
644 					fl_set_key_val(tb, &key->basic.n_proto,
645 						       TCA_FLOWER_KEY_CVLAN_ETH_TYPE,
646 						       &mask->basic.n_proto,
647 						       TCA_FLOWER_UNSPEC,
648 						       sizeof(key->basic.n_proto));
649 				} else {
650 					key->basic.n_proto = ethertype;
651 					mask->basic.n_proto = cpu_to_be16(~0);
652 				}
653 			}
654 		} else {
655 			key->basic.n_proto = ethertype;
656 			mask->basic.n_proto = cpu_to_be16(~0);
657 		}
658 	}
659 
660 	if (key->basic.n_proto == htons(ETH_P_IP) ||
661 	    key->basic.n_proto == htons(ETH_P_IPV6)) {
662 		fl_set_key_val(tb, &key->basic.ip_proto, TCA_FLOWER_KEY_IP_PROTO,
663 			       &mask->basic.ip_proto, TCA_FLOWER_UNSPEC,
664 			       sizeof(key->basic.ip_proto));
665 		fl_set_key_ip(tb, false, &key->ip, &mask->ip);
666 	}
667 
668 	if (tb[TCA_FLOWER_KEY_IPV4_SRC] || tb[TCA_FLOWER_KEY_IPV4_DST]) {
669 		key->control.addr_type = FLOW_DISSECTOR_KEY_IPV4_ADDRS;
670 		mask->control.addr_type = ~0;
671 		fl_set_key_val(tb, &key->ipv4.src, TCA_FLOWER_KEY_IPV4_SRC,
672 			       &mask->ipv4.src, TCA_FLOWER_KEY_IPV4_SRC_MASK,
673 			       sizeof(key->ipv4.src));
674 		fl_set_key_val(tb, &key->ipv4.dst, TCA_FLOWER_KEY_IPV4_DST,
675 			       &mask->ipv4.dst, TCA_FLOWER_KEY_IPV4_DST_MASK,
676 			       sizeof(key->ipv4.dst));
677 	} else if (tb[TCA_FLOWER_KEY_IPV6_SRC] || tb[TCA_FLOWER_KEY_IPV6_DST]) {
678 		key->control.addr_type = FLOW_DISSECTOR_KEY_IPV6_ADDRS;
679 		mask->control.addr_type = ~0;
680 		fl_set_key_val(tb, &key->ipv6.src, TCA_FLOWER_KEY_IPV6_SRC,
681 			       &mask->ipv6.src, TCA_FLOWER_KEY_IPV6_SRC_MASK,
682 			       sizeof(key->ipv6.src));
683 		fl_set_key_val(tb, &key->ipv6.dst, TCA_FLOWER_KEY_IPV6_DST,
684 			       &mask->ipv6.dst, TCA_FLOWER_KEY_IPV6_DST_MASK,
685 			       sizeof(key->ipv6.dst));
686 	}
687 
688 	if (key->basic.ip_proto == IPPROTO_TCP) {
689 		fl_set_key_val(tb, &key->tp.src, TCA_FLOWER_KEY_TCP_SRC,
690 			       &mask->tp.src, TCA_FLOWER_KEY_TCP_SRC_MASK,
691 			       sizeof(key->tp.src));
692 		fl_set_key_val(tb, &key->tp.dst, TCA_FLOWER_KEY_TCP_DST,
693 			       &mask->tp.dst, TCA_FLOWER_KEY_TCP_DST_MASK,
694 			       sizeof(key->tp.dst));
695 		fl_set_key_val(tb, &key->tcp.flags, TCA_FLOWER_KEY_TCP_FLAGS,
696 			       &mask->tcp.flags, TCA_FLOWER_KEY_TCP_FLAGS_MASK,
697 			       sizeof(key->tcp.flags));
698 	} else if (key->basic.ip_proto == IPPROTO_UDP) {
699 		fl_set_key_val(tb, &key->tp.src, TCA_FLOWER_KEY_UDP_SRC,
700 			       &mask->tp.src, TCA_FLOWER_KEY_UDP_SRC_MASK,
701 			       sizeof(key->tp.src));
702 		fl_set_key_val(tb, &key->tp.dst, TCA_FLOWER_KEY_UDP_DST,
703 			       &mask->tp.dst, TCA_FLOWER_KEY_UDP_DST_MASK,
704 			       sizeof(key->tp.dst));
705 	} else if (key->basic.ip_proto == IPPROTO_SCTP) {
706 		fl_set_key_val(tb, &key->tp.src, TCA_FLOWER_KEY_SCTP_SRC,
707 			       &mask->tp.src, TCA_FLOWER_KEY_SCTP_SRC_MASK,
708 			       sizeof(key->tp.src));
709 		fl_set_key_val(tb, &key->tp.dst, TCA_FLOWER_KEY_SCTP_DST,
710 			       &mask->tp.dst, TCA_FLOWER_KEY_SCTP_DST_MASK,
711 			       sizeof(key->tp.dst));
712 	} else if (key->basic.n_proto == htons(ETH_P_IP) &&
713 		   key->basic.ip_proto == IPPROTO_ICMP) {
714 		fl_set_key_val(tb, &key->icmp.type, TCA_FLOWER_KEY_ICMPV4_TYPE,
715 			       &mask->icmp.type,
716 			       TCA_FLOWER_KEY_ICMPV4_TYPE_MASK,
717 			       sizeof(key->icmp.type));
718 		fl_set_key_val(tb, &key->icmp.code, TCA_FLOWER_KEY_ICMPV4_CODE,
719 			       &mask->icmp.code,
720 			       TCA_FLOWER_KEY_ICMPV4_CODE_MASK,
721 			       sizeof(key->icmp.code));
722 	} else if (key->basic.n_proto == htons(ETH_P_IPV6) &&
723 		   key->basic.ip_proto == IPPROTO_ICMPV6) {
724 		fl_set_key_val(tb, &key->icmp.type, TCA_FLOWER_KEY_ICMPV6_TYPE,
725 			       &mask->icmp.type,
726 			       TCA_FLOWER_KEY_ICMPV6_TYPE_MASK,
727 			       sizeof(key->icmp.type));
728 		fl_set_key_val(tb, &key->icmp.code, TCA_FLOWER_KEY_ICMPV6_CODE,
729 			       &mask->icmp.code,
730 			       TCA_FLOWER_KEY_ICMPV6_CODE_MASK,
731 			       sizeof(key->icmp.code));
732 	} else if (key->basic.n_proto == htons(ETH_P_MPLS_UC) ||
733 		   key->basic.n_proto == htons(ETH_P_MPLS_MC)) {
734 		ret = fl_set_key_mpls(tb, &key->mpls, &mask->mpls);
735 		if (ret)
736 			return ret;
737 	} else if (key->basic.n_proto == htons(ETH_P_ARP) ||
738 		   key->basic.n_proto == htons(ETH_P_RARP)) {
739 		fl_set_key_val(tb, &key->arp.sip, TCA_FLOWER_KEY_ARP_SIP,
740 			       &mask->arp.sip, TCA_FLOWER_KEY_ARP_SIP_MASK,
741 			       sizeof(key->arp.sip));
742 		fl_set_key_val(tb, &key->arp.tip, TCA_FLOWER_KEY_ARP_TIP,
743 			       &mask->arp.tip, TCA_FLOWER_KEY_ARP_TIP_MASK,
744 			       sizeof(key->arp.tip));
745 		fl_set_key_val(tb, &key->arp.op, TCA_FLOWER_KEY_ARP_OP,
746 			       &mask->arp.op, TCA_FLOWER_KEY_ARP_OP_MASK,
747 			       sizeof(key->arp.op));
748 		fl_set_key_val(tb, key->arp.sha, TCA_FLOWER_KEY_ARP_SHA,
749 			       mask->arp.sha, TCA_FLOWER_KEY_ARP_SHA_MASK,
750 			       sizeof(key->arp.sha));
751 		fl_set_key_val(tb, key->arp.tha, TCA_FLOWER_KEY_ARP_THA,
752 			       mask->arp.tha, TCA_FLOWER_KEY_ARP_THA_MASK,
753 			       sizeof(key->arp.tha));
754 	}
755 
756 	if (tb[TCA_FLOWER_KEY_ENC_IPV4_SRC] ||
757 	    tb[TCA_FLOWER_KEY_ENC_IPV4_DST]) {
758 		key->enc_control.addr_type = FLOW_DISSECTOR_KEY_IPV4_ADDRS;
759 		mask->enc_control.addr_type = ~0;
760 		fl_set_key_val(tb, &key->enc_ipv4.src,
761 			       TCA_FLOWER_KEY_ENC_IPV4_SRC,
762 			       &mask->enc_ipv4.src,
763 			       TCA_FLOWER_KEY_ENC_IPV4_SRC_MASK,
764 			       sizeof(key->enc_ipv4.src));
765 		fl_set_key_val(tb, &key->enc_ipv4.dst,
766 			       TCA_FLOWER_KEY_ENC_IPV4_DST,
767 			       &mask->enc_ipv4.dst,
768 			       TCA_FLOWER_KEY_ENC_IPV4_DST_MASK,
769 			       sizeof(key->enc_ipv4.dst));
770 	}
771 
772 	if (tb[TCA_FLOWER_KEY_ENC_IPV6_SRC] ||
773 	    tb[TCA_FLOWER_KEY_ENC_IPV6_DST]) {
774 		key->enc_control.addr_type = FLOW_DISSECTOR_KEY_IPV6_ADDRS;
775 		mask->enc_control.addr_type = ~0;
776 		fl_set_key_val(tb, &key->enc_ipv6.src,
777 			       TCA_FLOWER_KEY_ENC_IPV6_SRC,
778 			       &mask->enc_ipv6.src,
779 			       TCA_FLOWER_KEY_ENC_IPV6_SRC_MASK,
780 			       sizeof(key->enc_ipv6.src));
781 		fl_set_key_val(tb, &key->enc_ipv6.dst,
782 			       TCA_FLOWER_KEY_ENC_IPV6_DST,
783 			       &mask->enc_ipv6.dst,
784 			       TCA_FLOWER_KEY_ENC_IPV6_DST_MASK,
785 			       sizeof(key->enc_ipv6.dst));
786 	}
787 
788 	fl_set_key_val(tb, &key->enc_key_id.keyid, TCA_FLOWER_KEY_ENC_KEY_ID,
789 		       &mask->enc_key_id.keyid, TCA_FLOWER_UNSPEC,
790 		       sizeof(key->enc_key_id.keyid));
791 
792 	fl_set_key_val(tb, &key->enc_tp.src, TCA_FLOWER_KEY_ENC_UDP_SRC_PORT,
793 		       &mask->enc_tp.src, TCA_FLOWER_KEY_ENC_UDP_SRC_PORT_MASK,
794 		       sizeof(key->enc_tp.src));
795 
796 	fl_set_key_val(tb, &key->enc_tp.dst, TCA_FLOWER_KEY_ENC_UDP_DST_PORT,
797 		       &mask->enc_tp.dst, TCA_FLOWER_KEY_ENC_UDP_DST_PORT_MASK,
798 		       sizeof(key->enc_tp.dst));
799 
800 	fl_set_key_ip(tb, true, &key->enc_ip, &mask->enc_ip);
801 
802 	if (tb[TCA_FLOWER_KEY_FLAGS])
803 		ret = fl_set_key_flags(tb, &key->control.flags, &mask->control.flags);
804 
805 	return ret;
806 }
807 
808 static void fl_mask_copy(struct fl_flow_mask *dst,
809 			 struct fl_flow_mask *src)
810 {
811 	const void *psrc = fl_key_get_start(&src->key, src);
812 	void *pdst = fl_key_get_start(&dst->key, src);
813 
814 	memcpy(pdst, psrc, fl_mask_range(src));
815 	dst->range = src->range;
816 }
817 
818 static const struct rhashtable_params fl_ht_params = {
819 	.key_offset = offsetof(struct cls_fl_filter, mkey), /* base offset */
820 	.head_offset = offsetof(struct cls_fl_filter, ht_node),
821 	.automatic_shrinking = true,
822 };
823 
824 static int fl_init_mask_hashtable(struct fl_flow_mask *mask)
825 {
826 	mask->filter_ht_params = fl_ht_params;
827 	mask->filter_ht_params.key_len = fl_mask_range(mask);
828 	mask->filter_ht_params.key_offset += mask->range.start;
829 
830 	return rhashtable_init(&mask->ht, &mask->filter_ht_params);
831 }
832 
833 #define FL_KEY_MEMBER_OFFSET(member) offsetof(struct fl_flow_key, member)
834 #define FL_KEY_MEMBER_SIZE(member) (sizeof(((struct fl_flow_key *) 0)->member))
835 
836 #define FL_KEY_IS_MASKED(mask, member)						\
837 	memchr_inv(((char *)mask) + FL_KEY_MEMBER_OFFSET(member),		\
838 		   0, FL_KEY_MEMBER_SIZE(member))				\
839 
840 #define FL_KEY_SET(keys, cnt, id, member)					\
841 	do {									\
842 		keys[cnt].key_id = id;						\
843 		keys[cnt].offset = FL_KEY_MEMBER_OFFSET(member);		\
844 		cnt++;								\
845 	} while(0);
846 
847 #define FL_KEY_SET_IF_MASKED(mask, keys, cnt, id, member)			\
848 	do {									\
849 		if (FL_KEY_IS_MASKED(mask, member))				\
850 			FL_KEY_SET(keys, cnt, id, member);			\
851 	} while(0);
852 
853 static void fl_init_dissector(struct flow_dissector *dissector,
854 			      struct fl_flow_key *mask)
855 {
856 	struct flow_dissector_key keys[FLOW_DISSECTOR_KEY_MAX];
857 	size_t cnt = 0;
858 
859 	FL_KEY_SET(keys, cnt, FLOW_DISSECTOR_KEY_CONTROL, control);
860 	FL_KEY_SET(keys, cnt, FLOW_DISSECTOR_KEY_BASIC, basic);
861 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
862 			     FLOW_DISSECTOR_KEY_ETH_ADDRS, eth);
863 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
864 			     FLOW_DISSECTOR_KEY_IPV4_ADDRS, ipv4);
865 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
866 			     FLOW_DISSECTOR_KEY_IPV6_ADDRS, ipv6);
867 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
868 			     FLOW_DISSECTOR_KEY_PORTS, tp);
869 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
870 			     FLOW_DISSECTOR_KEY_IP, ip);
871 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
872 			     FLOW_DISSECTOR_KEY_TCP, tcp);
873 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
874 			     FLOW_DISSECTOR_KEY_ICMP, icmp);
875 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
876 			     FLOW_DISSECTOR_KEY_ARP, arp);
877 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
878 			     FLOW_DISSECTOR_KEY_MPLS, mpls);
879 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
880 			     FLOW_DISSECTOR_KEY_VLAN, vlan);
881 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
882 			     FLOW_DISSECTOR_KEY_CVLAN, cvlan);
883 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
884 			     FLOW_DISSECTOR_KEY_ENC_KEYID, enc_key_id);
885 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
886 			     FLOW_DISSECTOR_KEY_ENC_IPV4_ADDRS, enc_ipv4);
887 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
888 			     FLOW_DISSECTOR_KEY_ENC_IPV6_ADDRS, enc_ipv6);
889 	if (FL_KEY_IS_MASKED(mask, enc_ipv4) ||
890 	    FL_KEY_IS_MASKED(mask, enc_ipv6))
891 		FL_KEY_SET(keys, cnt, FLOW_DISSECTOR_KEY_ENC_CONTROL,
892 			   enc_control);
893 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
894 			     FLOW_DISSECTOR_KEY_ENC_PORTS, enc_tp);
895 	FL_KEY_SET_IF_MASKED(mask, keys, cnt,
896 			     FLOW_DISSECTOR_KEY_ENC_IP, enc_ip);
897 
898 	skb_flow_dissector_init(dissector, keys, cnt);
899 }
900 
901 static struct fl_flow_mask *fl_create_new_mask(struct cls_fl_head *head,
902 					       struct fl_flow_mask *mask)
903 {
904 	struct fl_flow_mask *newmask;
905 	int err;
906 
907 	newmask = kzalloc(sizeof(*newmask), GFP_KERNEL);
908 	if (!newmask)
909 		return ERR_PTR(-ENOMEM);
910 
911 	fl_mask_copy(newmask, mask);
912 
913 	err = fl_init_mask_hashtable(newmask);
914 	if (err)
915 		goto errout_free;
916 
917 	fl_init_dissector(&newmask->dissector, &newmask->key);
918 
919 	INIT_LIST_HEAD_RCU(&newmask->filters);
920 
921 	err = rhashtable_insert_fast(&head->ht, &newmask->ht_node,
922 				     mask_ht_params);
923 	if (err)
924 		goto errout_destroy;
925 
926 	list_add_tail_rcu(&newmask->list, &head->masks);
927 
928 	return newmask;
929 
930 errout_destroy:
931 	rhashtable_destroy(&newmask->ht);
932 errout_free:
933 	kfree(newmask);
934 
935 	return ERR_PTR(err);
936 }
937 
938 static int fl_check_assign_mask(struct cls_fl_head *head,
939 				struct cls_fl_filter *fnew,
940 				struct cls_fl_filter *fold,
941 				struct fl_flow_mask *mask)
942 {
943 	struct fl_flow_mask *newmask;
944 
945 	fnew->mask = rhashtable_lookup_fast(&head->ht, mask, mask_ht_params);
946 	if (!fnew->mask) {
947 		if (fold)
948 			return -EINVAL;
949 
950 		newmask = fl_create_new_mask(head, mask);
951 		if (IS_ERR(newmask))
952 			return PTR_ERR(newmask);
953 
954 		fnew->mask = newmask;
955 	} else if (fold && fold->mask != fnew->mask) {
956 		return -EINVAL;
957 	}
958 
959 	return 0;
960 }
961 
962 static int fl_set_parms(struct net *net, struct tcf_proto *tp,
963 			struct cls_fl_filter *f, struct fl_flow_mask *mask,
964 			unsigned long base, struct nlattr **tb,
965 			struct nlattr *est, bool ovr,
966 			struct fl_flow_tmplt *tmplt,
967 			struct netlink_ext_ack *extack)
968 {
969 	int err;
970 
971 	err = tcf_exts_validate(net, tp, tb, est, &f->exts, ovr, extack);
972 	if (err < 0)
973 		return err;
974 
975 	if (tb[TCA_FLOWER_CLASSID]) {
976 		f->res.classid = nla_get_u32(tb[TCA_FLOWER_CLASSID]);
977 		tcf_bind_filter(tp, &f->res, base);
978 	}
979 
980 	err = fl_set_key(net, tb, &f->key, &mask->key, extack);
981 	if (err)
982 		return err;
983 
984 	fl_mask_update_range(mask);
985 	fl_set_masked_key(&f->mkey, &f->key, mask);
986 
987 	if (!fl_mask_fits_tmplt(tmplt, mask)) {
988 		NL_SET_ERR_MSG_MOD(extack, "Mask does not fit the template");
989 		return -EINVAL;
990 	}
991 
992 	return 0;
993 }
994 
995 static int fl_change(struct net *net, struct sk_buff *in_skb,
996 		     struct tcf_proto *tp, unsigned long base,
997 		     u32 handle, struct nlattr **tca,
998 		     void **arg, bool ovr, struct netlink_ext_ack *extack)
999 {
1000 	struct cls_fl_head *head = rtnl_dereference(tp->root);
1001 	struct cls_fl_filter *fold = *arg;
1002 	struct cls_fl_filter *fnew;
1003 	struct nlattr **tb;
1004 	struct fl_flow_mask mask = {};
1005 	int err;
1006 
1007 	if (!tca[TCA_OPTIONS])
1008 		return -EINVAL;
1009 
1010 	tb = kcalloc(TCA_FLOWER_MAX + 1, sizeof(struct nlattr *), GFP_KERNEL);
1011 	if (!tb)
1012 		return -ENOBUFS;
1013 
1014 	err = nla_parse_nested(tb, TCA_FLOWER_MAX, tca[TCA_OPTIONS],
1015 			       fl_policy, NULL);
1016 	if (err < 0)
1017 		goto errout_tb;
1018 
1019 	if (fold && handle && fold->handle != handle) {
1020 		err = -EINVAL;
1021 		goto errout_tb;
1022 	}
1023 
1024 	fnew = kzalloc(sizeof(*fnew), GFP_KERNEL);
1025 	if (!fnew) {
1026 		err = -ENOBUFS;
1027 		goto errout_tb;
1028 	}
1029 
1030 	err = tcf_exts_init(&fnew->exts, TCA_FLOWER_ACT, 0);
1031 	if (err < 0)
1032 		goto errout;
1033 
1034 	if (!handle) {
1035 		handle = 1;
1036 		err = idr_alloc_u32(&head->handle_idr, fnew, &handle,
1037 				    INT_MAX, GFP_KERNEL);
1038 	} else if (!fold) {
1039 		/* user specifies a handle and it doesn't exist */
1040 		err = idr_alloc_u32(&head->handle_idr, fnew, &handle,
1041 				    handle, GFP_KERNEL);
1042 	}
1043 	if (err)
1044 		goto errout;
1045 	fnew->handle = handle;
1046 
1047 	if (tb[TCA_FLOWER_FLAGS]) {
1048 		fnew->flags = nla_get_u32(tb[TCA_FLOWER_FLAGS]);
1049 
1050 		if (!tc_flags_valid(fnew->flags)) {
1051 			err = -EINVAL;
1052 			goto errout_idr;
1053 		}
1054 	}
1055 
1056 	err = fl_set_parms(net, tp, fnew, &mask, base, tb, tca[TCA_RATE], ovr,
1057 			   tp->chain->tmplt_priv, extack);
1058 	if (err)
1059 		goto errout_idr;
1060 
1061 	err = fl_check_assign_mask(head, fnew, fold, &mask);
1062 	if (err)
1063 		goto errout_idr;
1064 
1065 	if (!tc_skip_sw(fnew->flags)) {
1066 		if (!fold && fl_lookup(fnew->mask, &fnew->mkey)) {
1067 			err = -EEXIST;
1068 			goto errout_mask;
1069 		}
1070 
1071 		err = rhashtable_insert_fast(&fnew->mask->ht, &fnew->ht_node,
1072 					     fnew->mask->filter_ht_params);
1073 		if (err)
1074 			goto errout_mask;
1075 	}
1076 
1077 	if (!tc_skip_hw(fnew->flags)) {
1078 		err = fl_hw_replace_filter(tp, fnew, extack);
1079 		if (err)
1080 			goto errout_mask;
1081 	}
1082 
1083 	if (!tc_in_hw(fnew->flags))
1084 		fnew->flags |= TCA_CLS_FLAGS_NOT_IN_HW;
1085 
1086 	if (fold) {
1087 		if (!tc_skip_sw(fold->flags))
1088 			rhashtable_remove_fast(&fold->mask->ht,
1089 					       &fold->ht_node,
1090 					       fold->mask->filter_ht_params);
1091 		if (!tc_skip_hw(fold->flags))
1092 			fl_hw_destroy_filter(tp, fold, NULL);
1093 	}
1094 
1095 	*arg = fnew;
1096 
1097 	if (fold) {
1098 		idr_replace(&head->handle_idr, fnew, fnew->handle);
1099 		list_replace_rcu(&fold->list, &fnew->list);
1100 		tcf_unbind_filter(tp, &fold->res);
1101 		tcf_exts_get_net(&fold->exts);
1102 		tcf_queue_work(&fold->rwork, fl_destroy_filter_work);
1103 	} else {
1104 		list_add_tail_rcu(&fnew->list, &fnew->mask->filters);
1105 	}
1106 
1107 	kfree(tb);
1108 	return 0;
1109 
1110 errout_mask:
1111 	fl_mask_put(head, fnew->mask, false);
1112 
1113 errout_idr:
1114 	if (!fold)
1115 		idr_remove(&head->handle_idr, fnew->handle);
1116 errout:
1117 	tcf_exts_destroy(&fnew->exts);
1118 	kfree(fnew);
1119 errout_tb:
1120 	kfree(tb);
1121 	return err;
1122 }
1123 
1124 static int fl_delete(struct tcf_proto *tp, void *arg, bool *last,
1125 		     struct netlink_ext_ack *extack)
1126 {
1127 	struct cls_fl_head *head = rtnl_dereference(tp->root);
1128 	struct cls_fl_filter *f = arg;
1129 
1130 	if (!tc_skip_sw(f->flags))
1131 		rhashtable_remove_fast(&f->mask->ht, &f->ht_node,
1132 				       f->mask->filter_ht_params);
1133 	__fl_delete(tp, f, extack);
1134 	*last = list_empty(&head->masks);
1135 	return 0;
1136 }
1137 
1138 static void fl_walk(struct tcf_proto *tp, struct tcf_walker *arg)
1139 {
1140 	struct cls_fl_head *head = rtnl_dereference(tp->root);
1141 	struct cls_fl_filter *f;
1142 
1143 	arg->count = arg->skip;
1144 
1145 	while ((f = idr_get_next_ul(&head->handle_idr,
1146 				    &arg->cookie)) != NULL) {
1147 		if (arg->fn(tp, f, arg) < 0) {
1148 			arg->stop = 1;
1149 			break;
1150 		}
1151 		arg->cookie = f->handle + 1;
1152 		arg->count++;
1153 	}
1154 }
1155 
1156 static int fl_reoffload(struct tcf_proto *tp, bool add, tc_setup_cb_t *cb,
1157 			void *cb_priv, struct netlink_ext_ack *extack)
1158 {
1159 	struct cls_fl_head *head = rtnl_dereference(tp->root);
1160 	struct tc_cls_flower_offload cls_flower = {};
1161 	struct tcf_block *block = tp->chain->block;
1162 	struct fl_flow_mask *mask;
1163 	struct cls_fl_filter *f;
1164 	int err;
1165 
1166 	list_for_each_entry(mask, &head->masks, list) {
1167 		list_for_each_entry(f, &mask->filters, list) {
1168 			if (tc_skip_hw(f->flags))
1169 				continue;
1170 
1171 			tc_cls_common_offload_init(&cls_flower.common, tp,
1172 						   f->flags, extack);
1173 			cls_flower.command = add ?
1174 				TC_CLSFLOWER_REPLACE : TC_CLSFLOWER_DESTROY;
1175 			cls_flower.cookie = (unsigned long)f;
1176 			cls_flower.dissector = &mask->dissector;
1177 			cls_flower.mask = &f->mkey;
1178 			cls_flower.key = &f->key;
1179 			cls_flower.exts = &f->exts;
1180 			cls_flower.classid = f->res.classid;
1181 
1182 			err = cb(TC_SETUP_CLSFLOWER, &cls_flower, cb_priv);
1183 			if (err) {
1184 				if (add && tc_skip_sw(f->flags))
1185 					return err;
1186 				continue;
1187 			}
1188 
1189 			tc_cls_offload_cnt_update(block, &f->in_hw_count,
1190 						  &f->flags, add);
1191 		}
1192 	}
1193 
1194 	return 0;
1195 }
1196 
1197 static void fl_hw_create_tmplt(struct tcf_chain *chain,
1198 			       struct fl_flow_tmplt *tmplt)
1199 {
1200 	struct tc_cls_flower_offload cls_flower = {};
1201 	struct tcf_block *block = chain->block;
1202 	struct tcf_exts dummy_exts = { 0, };
1203 
1204 	cls_flower.common.chain_index = chain->index;
1205 	cls_flower.command = TC_CLSFLOWER_TMPLT_CREATE;
1206 	cls_flower.cookie = (unsigned long) tmplt;
1207 	cls_flower.dissector = &tmplt->dissector;
1208 	cls_flower.mask = &tmplt->mask;
1209 	cls_flower.key = &tmplt->dummy_key;
1210 	cls_flower.exts = &dummy_exts;
1211 
1212 	/* We don't care if driver (any of them) fails to handle this
1213 	 * call. It serves just as a hint for it.
1214 	 */
1215 	tc_setup_cb_call(block, NULL, TC_SETUP_CLSFLOWER,
1216 			 &cls_flower, false);
1217 }
1218 
1219 static void fl_hw_destroy_tmplt(struct tcf_chain *chain,
1220 				struct fl_flow_tmplt *tmplt)
1221 {
1222 	struct tc_cls_flower_offload cls_flower = {};
1223 	struct tcf_block *block = chain->block;
1224 
1225 	cls_flower.common.chain_index = chain->index;
1226 	cls_flower.command = TC_CLSFLOWER_TMPLT_DESTROY;
1227 	cls_flower.cookie = (unsigned long) tmplt;
1228 
1229 	tc_setup_cb_call(block, NULL, TC_SETUP_CLSFLOWER,
1230 			 &cls_flower, false);
1231 }
1232 
1233 static void *fl_tmplt_create(struct net *net, struct tcf_chain *chain,
1234 			     struct nlattr **tca,
1235 			     struct netlink_ext_ack *extack)
1236 {
1237 	struct fl_flow_tmplt *tmplt;
1238 	struct nlattr **tb;
1239 	int err;
1240 
1241 	if (!tca[TCA_OPTIONS])
1242 		return ERR_PTR(-EINVAL);
1243 
1244 	tb = kcalloc(TCA_FLOWER_MAX + 1, sizeof(struct nlattr *), GFP_KERNEL);
1245 	if (!tb)
1246 		return ERR_PTR(-ENOBUFS);
1247 	err = nla_parse_nested(tb, TCA_FLOWER_MAX, tca[TCA_OPTIONS],
1248 			       fl_policy, NULL);
1249 	if (err)
1250 		goto errout_tb;
1251 
1252 	tmplt = kzalloc(sizeof(*tmplt), GFP_KERNEL);
1253 	if (!tmplt)
1254 		goto errout_tb;
1255 	tmplt->chain = chain;
1256 	err = fl_set_key(net, tb, &tmplt->dummy_key, &tmplt->mask, extack);
1257 	if (err)
1258 		goto errout_tmplt;
1259 	kfree(tb);
1260 
1261 	fl_init_dissector(&tmplt->dissector, &tmplt->mask);
1262 
1263 	fl_hw_create_tmplt(chain, tmplt);
1264 
1265 	return tmplt;
1266 
1267 errout_tmplt:
1268 	kfree(tmplt);
1269 errout_tb:
1270 	kfree(tb);
1271 	return ERR_PTR(err);
1272 }
1273 
1274 static void fl_tmplt_destroy(void *tmplt_priv)
1275 {
1276 	struct fl_flow_tmplt *tmplt = tmplt_priv;
1277 
1278 	fl_hw_destroy_tmplt(tmplt->chain, tmplt);
1279 	kfree(tmplt);
1280 }
1281 
1282 static int fl_dump_key_val(struct sk_buff *skb,
1283 			   void *val, int val_type,
1284 			   void *mask, int mask_type, int len)
1285 {
1286 	int err;
1287 
1288 	if (!memchr_inv(mask, 0, len))
1289 		return 0;
1290 	err = nla_put(skb, val_type, len, val);
1291 	if (err)
1292 		return err;
1293 	if (mask_type != TCA_FLOWER_UNSPEC) {
1294 		err = nla_put(skb, mask_type, len, mask);
1295 		if (err)
1296 			return err;
1297 	}
1298 	return 0;
1299 }
1300 
1301 static int fl_dump_key_mpls(struct sk_buff *skb,
1302 			    struct flow_dissector_key_mpls *mpls_key,
1303 			    struct flow_dissector_key_mpls *mpls_mask)
1304 {
1305 	int err;
1306 
1307 	if (!memchr_inv(mpls_mask, 0, sizeof(*mpls_mask)))
1308 		return 0;
1309 	if (mpls_mask->mpls_ttl) {
1310 		err = nla_put_u8(skb, TCA_FLOWER_KEY_MPLS_TTL,
1311 				 mpls_key->mpls_ttl);
1312 		if (err)
1313 			return err;
1314 	}
1315 	if (mpls_mask->mpls_tc) {
1316 		err = nla_put_u8(skb, TCA_FLOWER_KEY_MPLS_TC,
1317 				 mpls_key->mpls_tc);
1318 		if (err)
1319 			return err;
1320 	}
1321 	if (mpls_mask->mpls_label) {
1322 		err = nla_put_u32(skb, TCA_FLOWER_KEY_MPLS_LABEL,
1323 				  mpls_key->mpls_label);
1324 		if (err)
1325 			return err;
1326 	}
1327 	if (mpls_mask->mpls_bos) {
1328 		err = nla_put_u8(skb, TCA_FLOWER_KEY_MPLS_BOS,
1329 				 mpls_key->mpls_bos);
1330 		if (err)
1331 			return err;
1332 	}
1333 	return 0;
1334 }
1335 
1336 static int fl_dump_key_ip(struct sk_buff *skb, bool encap,
1337 			  struct flow_dissector_key_ip *key,
1338 			  struct flow_dissector_key_ip *mask)
1339 {
1340 	int tos_key = encap ? TCA_FLOWER_KEY_ENC_IP_TOS : TCA_FLOWER_KEY_IP_TOS;
1341 	int ttl_key = encap ? TCA_FLOWER_KEY_ENC_IP_TTL : TCA_FLOWER_KEY_IP_TTL;
1342 	int tos_mask = encap ? TCA_FLOWER_KEY_ENC_IP_TOS_MASK : TCA_FLOWER_KEY_IP_TOS_MASK;
1343 	int ttl_mask = encap ? TCA_FLOWER_KEY_ENC_IP_TTL_MASK : TCA_FLOWER_KEY_IP_TTL_MASK;
1344 
1345 	if (fl_dump_key_val(skb, &key->tos, tos_key, &mask->tos, tos_mask, sizeof(key->tos)) ||
1346 	    fl_dump_key_val(skb, &key->ttl, ttl_key, &mask->ttl, ttl_mask, sizeof(key->ttl)))
1347 		return -1;
1348 
1349 	return 0;
1350 }
1351 
1352 static int fl_dump_key_vlan(struct sk_buff *skb,
1353 			    int vlan_id_key, int vlan_prio_key,
1354 			    struct flow_dissector_key_vlan *vlan_key,
1355 			    struct flow_dissector_key_vlan *vlan_mask)
1356 {
1357 	int err;
1358 
1359 	if (!memchr_inv(vlan_mask, 0, sizeof(*vlan_mask)))
1360 		return 0;
1361 	if (vlan_mask->vlan_id) {
1362 		err = nla_put_u16(skb, vlan_id_key,
1363 				  vlan_key->vlan_id);
1364 		if (err)
1365 			return err;
1366 	}
1367 	if (vlan_mask->vlan_priority) {
1368 		err = nla_put_u8(skb, vlan_prio_key,
1369 				 vlan_key->vlan_priority);
1370 		if (err)
1371 			return err;
1372 	}
1373 	return 0;
1374 }
1375 
1376 static void fl_get_key_flag(u32 dissector_key, u32 dissector_mask,
1377 			    u32 *flower_key, u32 *flower_mask,
1378 			    u32 flower_flag_bit, u32 dissector_flag_bit)
1379 {
1380 	if (dissector_mask & dissector_flag_bit) {
1381 		*flower_mask |= flower_flag_bit;
1382 		if (dissector_key & dissector_flag_bit)
1383 			*flower_key |= flower_flag_bit;
1384 	}
1385 }
1386 
1387 static int fl_dump_key_flags(struct sk_buff *skb, u32 flags_key, u32 flags_mask)
1388 {
1389 	u32 key, mask;
1390 	__be32 _key, _mask;
1391 	int err;
1392 
1393 	if (!memchr_inv(&flags_mask, 0, sizeof(flags_mask)))
1394 		return 0;
1395 
1396 	key = 0;
1397 	mask = 0;
1398 
1399 	fl_get_key_flag(flags_key, flags_mask, &key, &mask,
1400 			TCA_FLOWER_KEY_FLAGS_IS_FRAGMENT, FLOW_DIS_IS_FRAGMENT);
1401 	fl_get_key_flag(flags_key, flags_mask, &key, &mask,
1402 			TCA_FLOWER_KEY_FLAGS_FRAG_IS_FIRST,
1403 			FLOW_DIS_FIRST_FRAG);
1404 
1405 	_key = cpu_to_be32(key);
1406 	_mask = cpu_to_be32(mask);
1407 
1408 	err = nla_put(skb, TCA_FLOWER_KEY_FLAGS, 4, &_key);
1409 	if (err)
1410 		return err;
1411 
1412 	return nla_put(skb, TCA_FLOWER_KEY_FLAGS_MASK, 4, &_mask);
1413 }
1414 
1415 static int fl_dump_key(struct sk_buff *skb, struct net *net,
1416 		       struct fl_flow_key *key, struct fl_flow_key *mask)
1417 {
1418 	if (mask->indev_ifindex) {
1419 		struct net_device *dev;
1420 
1421 		dev = __dev_get_by_index(net, key->indev_ifindex);
1422 		if (dev && nla_put_string(skb, TCA_FLOWER_INDEV, dev->name))
1423 			goto nla_put_failure;
1424 	}
1425 
1426 	if (fl_dump_key_val(skb, key->eth.dst, TCA_FLOWER_KEY_ETH_DST,
1427 			    mask->eth.dst, TCA_FLOWER_KEY_ETH_DST_MASK,
1428 			    sizeof(key->eth.dst)) ||
1429 	    fl_dump_key_val(skb, key->eth.src, TCA_FLOWER_KEY_ETH_SRC,
1430 			    mask->eth.src, TCA_FLOWER_KEY_ETH_SRC_MASK,
1431 			    sizeof(key->eth.src)) ||
1432 	    fl_dump_key_val(skb, &key->basic.n_proto, TCA_FLOWER_KEY_ETH_TYPE,
1433 			    &mask->basic.n_proto, TCA_FLOWER_UNSPEC,
1434 			    sizeof(key->basic.n_proto)))
1435 		goto nla_put_failure;
1436 
1437 	if (fl_dump_key_mpls(skb, &key->mpls, &mask->mpls))
1438 		goto nla_put_failure;
1439 
1440 	if (fl_dump_key_vlan(skb, TCA_FLOWER_KEY_VLAN_ID,
1441 			     TCA_FLOWER_KEY_VLAN_PRIO, &key->vlan, &mask->vlan))
1442 		goto nla_put_failure;
1443 
1444 	if (fl_dump_key_vlan(skb, TCA_FLOWER_KEY_CVLAN_ID,
1445 			     TCA_FLOWER_KEY_CVLAN_PRIO,
1446 			     &key->cvlan, &mask->cvlan) ||
1447 	    (mask->cvlan.vlan_tpid &&
1448 	     nla_put_u16(skb, TCA_FLOWER_KEY_VLAN_ETH_TYPE,
1449 			 key->cvlan.vlan_tpid)))
1450 		goto nla_put_failure;
1451 
1452 	if (mask->basic.n_proto) {
1453 		if (mask->cvlan.vlan_tpid) {
1454 			if (nla_put_be16(skb, TCA_FLOWER_KEY_CVLAN_ETH_TYPE,
1455 					 key->basic.n_proto))
1456 				goto nla_put_failure;
1457 		} else if (mask->vlan.vlan_tpid) {
1458 			if (nla_put_be16(skb, TCA_FLOWER_KEY_VLAN_ETH_TYPE,
1459 					 key->basic.n_proto))
1460 				goto nla_put_failure;
1461 		}
1462 	}
1463 
1464 	if ((key->basic.n_proto == htons(ETH_P_IP) ||
1465 	     key->basic.n_proto == htons(ETH_P_IPV6)) &&
1466 	    (fl_dump_key_val(skb, &key->basic.ip_proto, TCA_FLOWER_KEY_IP_PROTO,
1467 			    &mask->basic.ip_proto, TCA_FLOWER_UNSPEC,
1468 			    sizeof(key->basic.ip_proto)) ||
1469 	    fl_dump_key_ip(skb, false, &key->ip, &mask->ip)))
1470 		goto nla_put_failure;
1471 
1472 	if (key->control.addr_type == FLOW_DISSECTOR_KEY_IPV4_ADDRS &&
1473 	    (fl_dump_key_val(skb, &key->ipv4.src, TCA_FLOWER_KEY_IPV4_SRC,
1474 			     &mask->ipv4.src, TCA_FLOWER_KEY_IPV4_SRC_MASK,
1475 			     sizeof(key->ipv4.src)) ||
1476 	     fl_dump_key_val(skb, &key->ipv4.dst, TCA_FLOWER_KEY_IPV4_DST,
1477 			     &mask->ipv4.dst, TCA_FLOWER_KEY_IPV4_DST_MASK,
1478 			     sizeof(key->ipv4.dst))))
1479 		goto nla_put_failure;
1480 	else if (key->control.addr_type == FLOW_DISSECTOR_KEY_IPV6_ADDRS &&
1481 		 (fl_dump_key_val(skb, &key->ipv6.src, TCA_FLOWER_KEY_IPV6_SRC,
1482 				  &mask->ipv6.src, TCA_FLOWER_KEY_IPV6_SRC_MASK,
1483 				  sizeof(key->ipv6.src)) ||
1484 		  fl_dump_key_val(skb, &key->ipv6.dst, TCA_FLOWER_KEY_IPV6_DST,
1485 				  &mask->ipv6.dst, TCA_FLOWER_KEY_IPV6_DST_MASK,
1486 				  sizeof(key->ipv6.dst))))
1487 		goto nla_put_failure;
1488 
1489 	if (key->basic.ip_proto == IPPROTO_TCP &&
1490 	    (fl_dump_key_val(skb, &key->tp.src, TCA_FLOWER_KEY_TCP_SRC,
1491 			     &mask->tp.src, TCA_FLOWER_KEY_TCP_SRC_MASK,
1492 			     sizeof(key->tp.src)) ||
1493 	     fl_dump_key_val(skb, &key->tp.dst, TCA_FLOWER_KEY_TCP_DST,
1494 			     &mask->tp.dst, TCA_FLOWER_KEY_TCP_DST_MASK,
1495 			     sizeof(key->tp.dst)) ||
1496 	     fl_dump_key_val(skb, &key->tcp.flags, TCA_FLOWER_KEY_TCP_FLAGS,
1497 			     &mask->tcp.flags, TCA_FLOWER_KEY_TCP_FLAGS_MASK,
1498 			     sizeof(key->tcp.flags))))
1499 		goto nla_put_failure;
1500 	else if (key->basic.ip_proto == IPPROTO_UDP &&
1501 		 (fl_dump_key_val(skb, &key->tp.src, TCA_FLOWER_KEY_UDP_SRC,
1502 				  &mask->tp.src, TCA_FLOWER_KEY_UDP_SRC_MASK,
1503 				  sizeof(key->tp.src)) ||
1504 		  fl_dump_key_val(skb, &key->tp.dst, TCA_FLOWER_KEY_UDP_DST,
1505 				  &mask->tp.dst, TCA_FLOWER_KEY_UDP_DST_MASK,
1506 				  sizeof(key->tp.dst))))
1507 		goto nla_put_failure;
1508 	else if (key->basic.ip_proto == IPPROTO_SCTP &&
1509 		 (fl_dump_key_val(skb, &key->tp.src, TCA_FLOWER_KEY_SCTP_SRC,
1510 				  &mask->tp.src, TCA_FLOWER_KEY_SCTP_SRC_MASK,
1511 				  sizeof(key->tp.src)) ||
1512 		  fl_dump_key_val(skb, &key->tp.dst, TCA_FLOWER_KEY_SCTP_DST,
1513 				  &mask->tp.dst, TCA_FLOWER_KEY_SCTP_DST_MASK,
1514 				  sizeof(key->tp.dst))))
1515 		goto nla_put_failure;
1516 	else if (key->basic.n_proto == htons(ETH_P_IP) &&
1517 		 key->basic.ip_proto == IPPROTO_ICMP &&
1518 		 (fl_dump_key_val(skb, &key->icmp.type,
1519 				  TCA_FLOWER_KEY_ICMPV4_TYPE, &mask->icmp.type,
1520 				  TCA_FLOWER_KEY_ICMPV4_TYPE_MASK,
1521 				  sizeof(key->icmp.type)) ||
1522 		  fl_dump_key_val(skb, &key->icmp.code,
1523 				  TCA_FLOWER_KEY_ICMPV4_CODE, &mask->icmp.code,
1524 				  TCA_FLOWER_KEY_ICMPV4_CODE_MASK,
1525 				  sizeof(key->icmp.code))))
1526 		goto nla_put_failure;
1527 	else if (key->basic.n_proto == htons(ETH_P_IPV6) &&
1528 		 key->basic.ip_proto == IPPROTO_ICMPV6 &&
1529 		 (fl_dump_key_val(skb, &key->icmp.type,
1530 				  TCA_FLOWER_KEY_ICMPV6_TYPE, &mask->icmp.type,
1531 				  TCA_FLOWER_KEY_ICMPV6_TYPE_MASK,
1532 				  sizeof(key->icmp.type)) ||
1533 		  fl_dump_key_val(skb, &key->icmp.code,
1534 				  TCA_FLOWER_KEY_ICMPV6_CODE, &mask->icmp.code,
1535 				  TCA_FLOWER_KEY_ICMPV6_CODE_MASK,
1536 				  sizeof(key->icmp.code))))
1537 		goto nla_put_failure;
1538 	else if ((key->basic.n_proto == htons(ETH_P_ARP) ||
1539 		  key->basic.n_proto == htons(ETH_P_RARP)) &&
1540 		 (fl_dump_key_val(skb, &key->arp.sip,
1541 				  TCA_FLOWER_KEY_ARP_SIP, &mask->arp.sip,
1542 				  TCA_FLOWER_KEY_ARP_SIP_MASK,
1543 				  sizeof(key->arp.sip)) ||
1544 		  fl_dump_key_val(skb, &key->arp.tip,
1545 				  TCA_FLOWER_KEY_ARP_TIP, &mask->arp.tip,
1546 				  TCA_FLOWER_KEY_ARP_TIP_MASK,
1547 				  sizeof(key->arp.tip)) ||
1548 		  fl_dump_key_val(skb, &key->arp.op,
1549 				  TCA_FLOWER_KEY_ARP_OP, &mask->arp.op,
1550 				  TCA_FLOWER_KEY_ARP_OP_MASK,
1551 				  sizeof(key->arp.op)) ||
1552 		  fl_dump_key_val(skb, key->arp.sha, TCA_FLOWER_KEY_ARP_SHA,
1553 				  mask->arp.sha, TCA_FLOWER_KEY_ARP_SHA_MASK,
1554 				  sizeof(key->arp.sha)) ||
1555 		  fl_dump_key_val(skb, key->arp.tha, TCA_FLOWER_KEY_ARP_THA,
1556 				  mask->arp.tha, TCA_FLOWER_KEY_ARP_THA_MASK,
1557 				  sizeof(key->arp.tha))))
1558 		goto nla_put_failure;
1559 
1560 	if (key->enc_control.addr_type == FLOW_DISSECTOR_KEY_IPV4_ADDRS &&
1561 	    (fl_dump_key_val(skb, &key->enc_ipv4.src,
1562 			    TCA_FLOWER_KEY_ENC_IPV4_SRC, &mask->enc_ipv4.src,
1563 			    TCA_FLOWER_KEY_ENC_IPV4_SRC_MASK,
1564 			    sizeof(key->enc_ipv4.src)) ||
1565 	     fl_dump_key_val(skb, &key->enc_ipv4.dst,
1566 			     TCA_FLOWER_KEY_ENC_IPV4_DST, &mask->enc_ipv4.dst,
1567 			     TCA_FLOWER_KEY_ENC_IPV4_DST_MASK,
1568 			     sizeof(key->enc_ipv4.dst))))
1569 		goto nla_put_failure;
1570 	else if (key->enc_control.addr_type == FLOW_DISSECTOR_KEY_IPV6_ADDRS &&
1571 		 (fl_dump_key_val(skb, &key->enc_ipv6.src,
1572 			    TCA_FLOWER_KEY_ENC_IPV6_SRC, &mask->enc_ipv6.src,
1573 			    TCA_FLOWER_KEY_ENC_IPV6_SRC_MASK,
1574 			    sizeof(key->enc_ipv6.src)) ||
1575 		 fl_dump_key_val(skb, &key->enc_ipv6.dst,
1576 				 TCA_FLOWER_KEY_ENC_IPV6_DST,
1577 				 &mask->enc_ipv6.dst,
1578 				 TCA_FLOWER_KEY_ENC_IPV6_DST_MASK,
1579 			    sizeof(key->enc_ipv6.dst))))
1580 		goto nla_put_failure;
1581 
1582 	if (fl_dump_key_val(skb, &key->enc_key_id, TCA_FLOWER_KEY_ENC_KEY_ID,
1583 			    &mask->enc_key_id, TCA_FLOWER_UNSPEC,
1584 			    sizeof(key->enc_key_id)) ||
1585 	    fl_dump_key_val(skb, &key->enc_tp.src,
1586 			    TCA_FLOWER_KEY_ENC_UDP_SRC_PORT,
1587 			    &mask->enc_tp.src,
1588 			    TCA_FLOWER_KEY_ENC_UDP_SRC_PORT_MASK,
1589 			    sizeof(key->enc_tp.src)) ||
1590 	    fl_dump_key_val(skb, &key->enc_tp.dst,
1591 			    TCA_FLOWER_KEY_ENC_UDP_DST_PORT,
1592 			    &mask->enc_tp.dst,
1593 			    TCA_FLOWER_KEY_ENC_UDP_DST_PORT_MASK,
1594 			    sizeof(key->enc_tp.dst)) ||
1595 	    fl_dump_key_ip(skb, true, &key->enc_ip, &mask->enc_ip))
1596 		goto nla_put_failure;
1597 
1598 	if (fl_dump_key_flags(skb, key->control.flags, mask->control.flags))
1599 		goto nla_put_failure;
1600 
1601 	return 0;
1602 
1603 nla_put_failure:
1604 	return -EMSGSIZE;
1605 }
1606 
1607 static int fl_dump(struct net *net, struct tcf_proto *tp, void *fh,
1608 		   struct sk_buff *skb, struct tcmsg *t)
1609 {
1610 	struct cls_fl_filter *f = fh;
1611 	struct nlattr *nest;
1612 	struct fl_flow_key *key, *mask;
1613 
1614 	if (!f)
1615 		return skb->len;
1616 
1617 	t->tcm_handle = f->handle;
1618 
1619 	nest = nla_nest_start(skb, TCA_OPTIONS);
1620 	if (!nest)
1621 		goto nla_put_failure;
1622 
1623 	if (f->res.classid &&
1624 	    nla_put_u32(skb, TCA_FLOWER_CLASSID, f->res.classid))
1625 		goto nla_put_failure;
1626 
1627 	key = &f->key;
1628 	mask = &f->mask->key;
1629 
1630 	if (fl_dump_key(skb, net, key, mask))
1631 		goto nla_put_failure;
1632 
1633 	if (!tc_skip_hw(f->flags))
1634 		fl_hw_update_stats(tp, f);
1635 
1636 	if (f->flags && nla_put_u32(skb, TCA_FLOWER_FLAGS, f->flags))
1637 		goto nla_put_failure;
1638 
1639 	if (tcf_exts_dump(skb, &f->exts))
1640 		goto nla_put_failure;
1641 
1642 	nla_nest_end(skb, nest);
1643 
1644 	if (tcf_exts_dump_stats(skb, &f->exts) < 0)
1645 		goto nla_put_failure;
1646 
1647 	return skb->len;
1648 
1649 nla_put_failure:
1650 	nla_nest_cancel(skb, nest);
1651 	return -1;
1652 }
1653 
1654 static int fl_tmplt_dump(struct sk_buff *skb, struct net *net, void *tmplt_priv)
1655 {
1656 	struct fl_flow_tmplt *tmplt = tmplt_priv;
1657 	struct fl_flow_key *key, *mask;
1658 	struct nlattr *nest;
1659 
1660 	nest = nla_nest_start(skb, TCA_OPTIONS);
1661 	if (!nest)
1662 		goto nla_put_failure;
1663 
1664 	key = &tmplt->dummy_key;
1665 	mask = &tmplt->mask;
1666 
1667 	if (fl_dump_key(skb, net, key, mask))
1668 		goto nla_put_failure;
1669 
1670 	nla_nest_end(skb, nest);
1671 
1672 	return skb->len;
1673 
1674 nla_put_failure:
1675 	nla_nest_cancel(skb, nest);
1676 	return -EMSGSIZE;
1677 }
1678 
1679 static void fl_bind_class(void *fh, u32 classid, unsigned long cl)
1680 {
1681 	struct cls_fl_filter *f = fh;
1682 
1683 	if (f && f->res.classid == classid)
1684 		f->res.class = cl;
1685 }
1686 
1687 static struct tcf_proto_ops cls_fl_ops __read_mostly = {
1688 	.kind		= "flower",
1689 	.classify	= fl_classify,
1690 	.init		= fl_init,
1691 	.destroy	= fl_destroy,
1692 	.get		= fl_get,
1693 	.change		= fl_change,
1694 	.delete		= fl_delete,
1695 	.walk		= fl_walk,
1696 	.reoffload	= fl_reoffload,
1697 	.dump		= fl_dump,
1698 	.bind_class	= fl_bind_class,
1699 	.tmplt_create	= fl_tmplt_create,
1700 	.tmplt_destroy	= fl_tmplt_destroy,
1701 	.tmplt_dump	= fl_tmplt_dump,
1702 	.owner		= THIS_MODULE,
1703 };
1704 
1705 static int __init cls_fl_init(void)
1706 {
1707 	return register_tcf_proto_ops(&cls_fl_ops);
1708 }
1709 
1710 static void __exit cls_fl_exit(void)
1711 {
1712 	unregister_tcf_proto_ops(&cls_fl_ops);
1713 }
1714 
1715 module_init(cls_fl_init);
1716 module_exit(cls_fl_exit);
1717 
1718 MODULE_AUTHOR("Jiri Pirko <jiri@resnulli.us>");
1719 MODULE_DESCRIPTION("Flower classifier");
1720 MODULE_LICENSE("GPL v2");
1721