xref: /openbmc/linux/net/ipv6/seg6_local.c (revision f052febd)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  *  SR-IPv6 implementation
4  *
5  *  Authors:
6  *  David Lebrun <david.lebrun@uclouvain.be>
7  *  eBPF support: Mathieu Xhonneux <m.xhonneux@gmail.com>
8  */
9 
10 #include <linux/filter.h>
11 #include <linux/types.h>
12 #include <linux/skbuff.h>
13 #include <linux/net.h>
14 #include <linux/module.h>
15 #include <net/ip.h>
16 #include <net/lwtunnel.h>
17 #include <net/netevent.h>
18 #include <net/netns/generic.h>
19 #include <net/ip6_fib.h>
20 #include <net/route.h>
21 #include <net/seg6.h>
22 #include <linux/seg6.h>
23 #include <linux/seg6_local.h>
24 #include <net/addrconf.h>
25 #include <net/ip6_route.h>
26 #include <net/dst_cache.h>
27 #include <net/ip_tunnels.h>
28 #ifdef CONFIG_IPV6_SEG6_HMAC
29 #include <net/seg6_hmac.h>
30 #endif
31 #include <net/seg6_local.h>
32 #include <linux/etherdevice.h>
33 #include <linux/bpf.h>
34 #include <linux/netfilter.h>
35 
36 #define SEG6_F_ATTR(i)		BIT(i)
37 
38 struct seg6_local_lwt;
39 
40 /* callbacks used for customizing the creation and destruction of a behavior */
41 struct seg6_local_lwtunnel_ops {
42 	int (*build_state)(struct seg6_local_lwt *slwt, const void *cfg,
43 			   struct netlink_ext_ack *extack);
44 	void (*destroy_state)(struct seg6_local_lwt *slwt);
45 };
46 
47 struct seg6_action_desc {
48 	int action;
49 	unsigned long attrs;
50 
51 	/* The optattrs field is used for specifying all the optional
52 	 * attributes supported by a specific behavior.
53 	 * It means that if one of these attributes is not provided in the
54 	 * netlink message during the behavior creation, no errors will be
55 	 * returned to the userspace.
56 	 *
57 	 * Each attribute can be only of two types (mutually exclusive):
58 	 * 1) required or 2) optional.
59 	 * Every user MUST obey to this rule! If you set an attribute as
60 	 * required the same attribute CANNOT be set as optional and vice
61 	 * versa.
62 	 */
63 	unsigned long optattrs;
64 
65 	int (*input)(struct sk_buff *skb, struct seg6_local_lwt *slwt);
66 	int static_headroom;
67 
68 	struct seg6_local_lwtunnel_ops slwt_ops;
69 };
70 
71 struct bpf_lwt_prog {
72 	struct bpf_prog *prog;
73 	char *name;
74 };
75 
76 /* default length values (expressed in bits) for both Locator-Block and
77  * Locator-Node Function.
78  *
79  * Both SEG6_LOCAL_LCBLOCK_DBITS and SEG6_LOCAL_LCNODE_FN_DBITS *must* be:
80  *    i) greater than 0;
81  *   ii) evenly divisible by 8. In other terms, the lengths of the
82  *	 Locator-Block and Locator-Node Function must be byte-aligned (we can
83  *	 relax this constraint in the future if really needed).
84  *
85  * Moreover, a third condition must hold:
86  *  iii) SEG6_LOCAL_LCBLOCK_DBITS + SEG6_LOCAL_LCNODE_FN_DBITS <= 128.
87  *
88  * The correctness of SEG6_LOCAL_LCBLOCK_DBITS and SEG6_LOCAL_LCNODE_FN_DBITS
89  * values are checked during the kernel compilation. If the compilation stops,
90  * check the value of these parameters to see if they meet conditions (i), (ii)
91  * and (iii).
92  */
93 #define SEG6_LOCAL_LCBLOCK_DBITS	32
94 #define SEG6_LOCAL_LCNODE_FN_DBITS	16
95 
96 /* The following next_csid_chk_{cntr,lcblock,lcblock_fn}_bits macros can be
97  * used directly to check whether the lengths (in bits) of Locator-Block and
98  * Locator-Node Function are valid according to (i), (ii), (iii).
99  */
100 #define next_csid_chk_cntr_bits(blen, flen)		\
101 	((blen) + (flen) > 128)
102 
103 #define next_csid_chk_lcblock_bits(blen)		\
104 ({							\
105 	typeof(blen) __tmp = blen;			\
106 	(!__tmp || __tmp > 120 || (__tmp & 0x07));	\
107 })
108 
109 #define next_csid_chk_lcnode_fn_bits(flen)		\
110 	next_csid_chk_lcblock_bits(flen)
111 
112 /* Supported Flavor operations are reported in this bitmask */
113 #define SEG6_LOCAL_FLV_SUPP_OPS	(BIT(SEG6_LOCAL_FLV_OP_NEXT_CSID))
114 
115 struct seg6_flavors_info {
116 	/* Flavor operations */
117 	__u32 flv_ops;
118 
119 	/* Locator-Block length, expressed in bits */
120 	__u8 lcblock_bits;
121 	/* Locator-Node Function length, expressed in bits*/
122 	__u8 lcnode_func_bits;
123 };
124 
125 enum seg6_end_dt_mode {
126 	DT_INVALID_MODE	= -EINVAL,
127 	DT_LEGACY_MODE	= 0,
128 	DT_VRF_MODE	= 1,
129 };
130 
131 struct seg6_end_dt_info {
132 	enum seg6_end_dt_mode mode;
133 
134 	struct net *net;
135 	/* VRF device associated to the routing table used by the SRv6
136 	 * End.DT4/DT6 behavior for routing IPv4/IPv6 packets.
137 	 */
138 	int vrf_ifindex;
139 	int vrf_table;
140 
141 	/* tunneled packet family (IPv4 or IPv6).
142 	 * Protocol and header length are inferred from family.
143 	 */
144 	u16 family;
145 };
146 
147 struct pcpu_seg6_local_counters {
148 	u64_stats_t packets;
149 	u64_stats_t bytes;
150 	u64_stats_t errors;
151 
152 	struct u64_stats_sync syncp;
153 };
154 
155 /* This struct groups all the SRv6 Behavior counters supported so far.
156  *
157  * put_nla_counters() makes use of this data structure to collect all counter
158  * values after the per-CPU counter evaluation has been performed.
159  * Finally, each counter value (in seg6_local_counters) is stored in the
160  * corresponding netlink attribute and sent to user space.
161  *
162  * NB: we don't want to expose this structure to user space!
163  */
164 struct seg6_local_counters {
165 	__u64 packets;
166 	__u64 bytes;
167 	__u64 errors;
168 };
169 
170 #define seg6_local_alloc_pcpu_counters(__gfp)				\
171 	__netdev_alloc_pcpu_stats(struct pcpu_seg6_local_counters,	\
172 				  ((__gfp) | __GFP_ZERO))
173 
174 #define SEG6_F_LOCAL_COUNTERS	SEG6_F_ATTR(SEG6_LOCAL_COUNTERS)
175 
176 struct seg6_local_lwt {
177 	int action;
178 	struct ipv6_sr_hdr *srh;
179 	int table;
180 	struct in_addr nh4;
181 	struct in6_addr nh6;
182 	int iif;
183 	int oif;
184 	struct bpf_lwt_prog bpf;
185 #ifdef CONFIG_NET_L3_MASTER_DEV
186 	struct seg6_end_dt_info dt_info;
187 #endif
188 	struct seg6_flavors_info flv_info;
189 
190 	struct pcpu_seg6_local_counters __percpu *pcpu_counters;
191 
192 	int headroom;
193 	struct seg6_action_desc *desc;
194 	/* unlike the required attrs, we have to track the optional attributes
195 	 * that have been effectively parsed.
196 	 */
197 	unsigned long parsed_optattrs;
198 };
199 
200 static struct seg6_local_lwt *seg6_local_lwtunnel(struct lwtunnel_state *lwt)
201 {
202 	return (struct seg6_local_lwt *)lwt->data;
203 }
204 
205 static struct ipv6_sr_hdr *get_and_validate_srh(struct sk_buff *skb)
206 {
207 	struct ipv6_sr_hdr *srh;
208 
209 	srh = seg6_get_srh(skb, IP6_FH_F_SKIP_RH);
210 	if (!srh)
211 		return NULL;
212 
213 #ifdef CONFIG_IPV6_SEG6_HMAC
214 	if (!seg6_hmac_validate_skb(skb))
215 		return NULL;
216 #endif
217 
218 	return srh;
219 }
220 
221 static bool decap_and_validate(struct sk_buff *skb, int proto)
222 {
223 	struct ipv6_sr_hdr *srh;
224 	unsigned int off = 0;
225 
226 	srh = seg6_get_srh(skb, 0);
227 	if (srh && srh->segments_left > 0)
228 		return false;
229 
230 #ifdef CONFIG_IPV6_SEG6_HMAC
231 	if (srh && !seg6_hmac_validate_skb(skb))
232 		return false;
233 #endif
234 
235 	if (ipv6_find_hdr(skb, &off, proto, NULL, NULL) < 0)
236 		return false;
237 
238 	if (!pskb_pull(skb, off))
239 		return false;
240 
241 	skb_postpull_rcsum(skb, skb_network_header(skb), off);
242 
243 	skb_reset_network_header(skb);
244 	skb_reset_transport_header(skb);
245 	if (iptunnel_pull_offloads(skb))
246 		return false;
247 
248 	return true;
249 }
250 
251 static void advance_nextseg(struct ipv6_sr_hdr *srh, struct in6_addr *daddr)
252 {
253 	struct in6_addr *addr;
254 
255 	srh->segments_left--;
256 	addr = srh->segments + srh->segments_left;
257 	*daddr = *addr;
258 }
259 
260 static int
261 seg6_lookup_any_nexthop(struct sk_buff *skb, struct in6_addr *nhaddr,
262 			u32 tbl_id, bool local_delivery)
263 {
264 	struct net *net = dev_net(skb->dev);
265 	struct ipv6hdr *hdr = ipv6_hdr(skb);
266 	int flags = RT6_LOOKUP_F_HAS_SADDR;
267 	struct dst_entry *dst = NULL;
268 	struct rt6_info *rt;
269 	struct flowi6 fl6;
270 	int dev_flags = 0;
271 
272 	memset(&fl6, 0, sizeof(fl6));
273 	fl6.flowi6_iif = skb->dev->ifindex;
274 	fl6.daddr = nhaddr ? *nhaddr : hdr->daddr;
275 	fl6.saddr = hdr->saddr;
276 	fl6.flowlabel = ip6_flowinfo(hdr);
277 	fl6.flowi6_mark = skb->mark;
278 	fl6.flowi6_proto = hdr->nexthdr;
279 
280 	if (nhaddr)
281 		fl6.flowi6_flags = FLOWI_FLAG_KNOWN_NH;
282 
283 	if (!tbl_id) {
284 		dst = ip6_route_input_lookup(net, skb->dev, &fl6, skb, flags);
285 	} else {
286 		struct fib6_table *table;
287 
288 		table = fib6_get_table(net, tbl_id);
289 		if (!table)
290 			goto out;
291 
292 		rt = ip6_pol_route(net, table, 0, &fl6, skb, flags);
293 		dst = &rt->dst;
294 	}
295 
296 	/* we want to discard traffic destined for local packet processing,
297 	 * if @local_delivery is set to false.
298 	 */
299 	if (!local_delivery)
300 		dev_flags |= IFF_LOOPBACK;
301 
302 	if (dst && (dst->dev->flags & dev_flags) && !dst->error) {
303 		dst_release(dst);
304 		dst = NULL;
305 	}
306 
307 out:
308 	if (!dst) {
309 		rt = net->ipv6.ip6_blk_hole_entry;
310 		dst = &rt->dst;
311 		dst_hold(dst);
312 	}
313 
314 	skb_dst_drop(skb);
315 	skb_dst_set(skb, dst);
316 	return dst->error;
317 }
318 
319 int seg6_lookup_nexthop(struct sk_buff *skb,
320 			struct in6_addr *nhaddr, u32 tbl_id)
321 {
322 	return seg6_lookup_any_nexthop(skb, nhaddr, tbl_id, false);
323 }
324 
325 static __u8 seg6_flv_lcblock_octects(const struct seg6_flavors_info *finfo)
326 {
327 	return finfo->lcblock_bits >> 3;
328 }
329 
330 static __u8 seg6_flv_lcnode_func_octects(const struct seg6_flavors_info *finfo)
331 {
332 	return finfo->lcnode_func_bits >> 3;
333 }
334 
335 static bool seg6_next_csid_is_arg_zero(const struct in6_addr *addr,
336 				       const struct seg6_flavors_info *finfo)
337 {
338 	__u8 fnc_octects = seg6_flv_lcnode_func_octects(finfo);
339 	__u8 blk_octects = seg6_flv_lcblock_octects(finfo);
340 	__u8 arg_octects;
341 	int i;
342 
343 	arg_octects = 16 - blk_octects - fnc_octects;
344 	for (i = 0; i < arg_octects; ++i) {
345 		if (addr->s6_addr[blk_octects + fnc_octects + i] != 0x00)
346 			return false;
347 	}
348 
349 	return true;
350 }
351 
352 /* assume that DA.Argument length > 0 */
353 static void seg6_next_csid_advance_arg(struct in6_addr *addr,
354 				       const struct seg6_flavors_info *finfo)
355 {
356 	__u8 fnc_octects = seg6_flv_lcnode_func_octects(finfo);
357 	__u8 blk_octects = seg6_flv_lcblock_octects(finfo);
358 
359 	/* advance DA.Argument */
360 	memmove(&addr->s6_addr[blk_octects],
361 		&addr->s6_addr[blk_octects + fnc_octects],
362 		16 - blk_octects - fnc_octects);
363 
364 	memset(&addr->s6_addr[16 - fnc_octects], 0x00, fnc_octects);
365 }
366 
367 static int input_action_end_core(struct sk_buff *skb,
368 				 struct seg6_local_lwt *slwt)
369 {
370 	struct ipv6_sr_hdr *srh;
371 
372 	srh = get_and_validate_srh(skb);
373 	if (!srh)
374 		goto drop;
375 
376 	advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
377 
378 	seg6_lookup_nexthop(skb, NULL, 0);
379 
380 	return dst_input(skb);
381 
382 drop:
383 	kfree_skb(skb);
384 	return -EINVAL;
385 }
386 
387 static int end_next_csid_core(struct sk_buff *skb, struct seg6_local_lwt *slwt)
388 {
389 	const struct seg6_flavors_info *finfo = &slwt->flv_info;
390 	struct in6_addr *daddr = &ipv6_hdr(skb)->daddr;
391 
392 	if (seg6_next_csid_is_arg_zero(daddr, finfo))
393 		return input_action_end_core(skb, slwt);
394 
395 	/* update DA */
396 	seg6_next_csid_advance_arg(daddr, finfo);
397 
398 	seg6_lookup_nexthop(skb, NULL, 0);
399 
400 	return dst_input(skb);
401 }
402 
403 static bool seg6_next_csid_enabled(__u32 fops)
404 {
405 	return fops & BIT(SEG6_LOCAL_FLV_OP_NEXT_CSID);
406 }
407 
408 /* regular endpoint function */
409 static int input_action_end(struct sk_buff *skb, struct seg6_local_lwt *slwt)
410 {
411 	const struct seg6_flavors_info *finfo = &slwt->flv_info;
412 
413 	if (seg6_next_csid_enabled(finfo->flv_ops))
414 		return end_next_csid_core(skb, slwt);
415 
416 	return input_action_end_core(skb, slwt);
417 }
418 
419 /* regular endpoint, and forward to specified nexthop */
420 static int input_action_end_x(struct sk_buff *skb, struct seg6_local_lwt *slwt)
421 {
422 	struct ipv6_sr_hdr *srh;
423 
424 	srh = get_and_validate_srh(skb);
425 	if (!srh)
426 		goto drop;
427 
428 	advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
429 
430 	seg6_lookup_nexthop(skb, &slwt->nh6, 0);
431 
432 	return dst_input(skb);
433 
434 drop:
435 	kfree_skb(skb);
436 	return -EINVAL;
437 }
438 
439 static int input_action_end_t(struct sk_buff *skb, struct seg6_local_lwt *slwt)
440 {
441 	struct ipv6_sr_hdr *srh;
442 
443 	srh = get_and_validate_srh(skb);
444 	if (!srh)
445 		goto drop;
446 
447 	advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
448 
449 	seg6_lookup_nexthop(skb, NULL, slwt->table);
450 
451 	return dst_input(skb);
452 
453 drop:
454 	kfree_skb(skb);
455 	return -EINVAL;
456 }
457 
458 /* decapsulate and forward inner L2 frame on specified interface */
459 static int input_action_end_dx2(struct sk_buff *skb,
460 				struct seg6_local_lwt *slwt)
461 {
462 	struct net *net = dev_net(skb->dev);
463 	struct net_device *odev;
464 	struct ethhdr *eth;
465 
466 	if (!decap_and_validate(skb, IPPROTO_ETHERNET))
467 		goto drop;
468 
469 	if (!pskb_may_pull(skb, ETH_HLEN))
470 		goto drop;
471 
472 	skb_reset_mac_header(skb);
473 	eth = (struct ethhdr *)skb->data;
474 
475 	/* To determine the frame's protocol, we assume it is 802.3. This avoids
476 	 * a call to eth_type_trans(), which is not really relevant for our
477 	 * use case.
478 	 */
479 	if (!eth_proto_is_802_3(eth->h_proto))
480 		goto drop;
481 
482 	odev = dev_get_by_index_rcu(net, slwt->oif);
483 	if (!odev)
484 		goto drop;
485 
486 	/* As we accept Ethernet frames, make sure the egress device is of
487 	 * the correct type.
488 	 */
489 	if (odev->type != ARPHRD_ETHER)
490 		goto drop;
491 
492 	if (!(odev->flags & IFF_UP) || !netif_carrier_ok(odev))
493 		goto drop;
494 
495 	skb_orphan(skb);
496 
497 	if (skb_warn_if_lro(skb))
498 		goto drop;
499 
500 	skb_forward_csum(skb);
501 
502 	if (skb->len - ETH_HLEN > odev->mtu)
503 		goto drop;
504 
505 	skb->dev = odev;
506 	skb->protocol = eth->h_proto;
507 
508 	return dev_queue_xmit(skb);
509 
510 drop:
511 	kfree_skb(skb);
512 	return -EINVAL;
513 }
514 
515 static int input_action_end_dx6_finish(struct net *net, struct sock *sk,
516 				       struct sk_buff *skb)
517 {
518 	struct dst_entry *orig_dst = skb_dst(skb);
519 	struct in6_addr *nhaddr = NULL;
520 	struct seg6_local_lwt *slwt;
521 
522 	slwt = seg6_local_lwtunnel(orig_dst->lwtstate);
523 
524 	/* The inner packet is not associated to any local interface,
525 	 * so we do not call netif_rx().
526 	 *
527 	 * If slwt->nh6 is set to ::, then lookup the nexthop for the
528 	 * inner packet's DA. Otherwise, use the specified nexthop.
529 	 */
530 	if (!ipv6_addr_any(&slwt->nh6))
531 		nhaddr = &slwt->nh6;
532 
533 	seg6_lookup_nexthop(skb, nhaddr, 0);
534 
535 	return dst_input(skb);
536 }
537 
538 /* decapsulate and forward to specified nexthop */
539 static int input_action_end_dx6(struct sk_buff *skb,
540 				struct seg6_local_lwt *slwt)
541 {
542 	/* this function accepts IPv6 encapsulated packets, with either
543 	 * an SRH with SL=0, or no SRH.
544 	 */
545 
546 	if (!decap_and_validate(skb, IPPROTO_IPV6))
547 		goto drop;
548 
549 	if (!pskb_may_pull(skb, sizeof(struct ipv6hdr)))
550 		goto drop;
551 
552 	skb_set_transport_header(skb, sizeof(struct ipv6hdr));
553 	nf_reset_ct(skb);
554 
555 	if (static_branch_unlikely(&nf_hooks_lwtunnel_enabled))
556 		return NF_HOOK(NFPROTO_IPV6, NF_INET_PRE_ROUTING,
557 			       dev_net(skb->dev), NULL, skb, NULL,
558 			       skb_dst(skb)->dev, input_action_end_dx6_finish);
559 
560 	return input_action_end_dx6_finish(dev_net(skb->dev), NULL, skb);
561 drop:
562 	kfree_skb(skb);
563 	return -EINVAL;
564 }
565 
566 static int input_action_end_dx4_finish(struct net *net, struct sock *sk,
567 				       struct sk_buff *skb)
568 {
569 	struct dst_entry *orig_dst = skb_dst(skb);
570 	struct seg6_local_lwt *slwt;
571 	struct iphdr *iph;
572 	__be32 nhaddr;
573 	int err;
574 
575 	slwt = seg6_local_lwtunnel(orig_dst->lwtstate);
576 
577 	iph = ip_hdr(skb);
578 
579 	nhaddr = slwt->nh4.s_addr ?: iph->daddr;
580 
581 	skb_dst_drop(skb);
582 
583 	err = ip_route_input(skb, nhaddr, iph->saddr, 0, skb->dev);
584 	if (err) {
585 		kfree_skb(skb);
586 		return -EINVAL;
587 	}
588 
589 	return dst_input(skb);
590 }
591 
592 static int input_action_end_dx4(struct sk_buff *skb,
593 				struct seg6_local_lwt *slwt)
594 {
595 	if (!decap_and_validate(skb, IPPROTO_IPIP))
596 		goto drop;
597 
598 	if (!pskb_may_pull(skb, sizeof(struct iphdr)))
599 		goto drop;
600 
601 	skb->protocol = htons(ETH_P_IP);
602 	skb_set_transport_header(skb, sizeof(struct iphdr));
603 	nf_reset_ct(skb);
604 
605 	if (static_branch_unlikely(&nf_hooks_lwtunnel_enabled))
606 		return NF_HOOK(NFPROTO_IPV4, NF_INET_PRE_ROUTING,
607 			       dev_net(skb->dev), NULL, skb, NULL,
608 			       skb_dst(skb)->dev, input_action_end_dx4_finish);
609 
610 	return input_action_end_dx4_finish(dev_net(skb->dev), NULL, skb);
611 drop:
612 	kfree_skb(skb);
613 	return -EINVAL;
614 }
615 
616 #ifdef CONFIG_NET_L3_MASTER_DEV
617 static struct net *fib6_config_get_net(const struct fib6_config *fib6_cfg)
618 {
619 	const struct nl_info *nli = &fib6_cfg->fc_nlinfo;
620 
621 	return nli->nl_net;
622 }
623 
624 static int __seg6_end_dt_vrf_build(struct seg6_local_lwt *slwt, const void *cfg,
625 				   u16 family, struct netlink_ext_ack *extack)
626 {
627 	struct seg6_end_dt_info *info = &slwt->dt_info;
628 	int vrf_ifindex;
629 	struct net *net;
630 
631 	net = fib6_config_get_net(cfg);
632 
633 	/* note that vrf_table was already set by parse_nla_vrftable() */
634 	vrf_ifindex = l3mdev_ifindex_lookup_by_table_id(L3MDEV_TYPE_VRF, net,
635 							info->vrf_table);
636 	if (vrf_ifindex < 0) {
637 		if (vrf_ifindex == -EPERM) {
638 			NL_SET_ERR_MSG(extack,
639 				       "Strict mode for VRF is disabled");
640 		} else if (vrf_ifindex == -ENODEV) {
641 			NL_SET_ERR_MSG(extack,
642 				       "Table has no associated VRF device");
643 		} else {
644 			pr_debug("seg6local: SRv6 End.DT* creation error=%d\n",
645 				 vrf_ifindex);
646 		}
647 
648 		return vrf_ifindex;
649 	}
650 
651 	info->net = net;
652 	info->vrf_ifindex = vrf_ifindex;
653 
654 	info->family = family;
655 	info->mode = DT_VRF_MODE;
656 
657 	return 0;
658 }
659 
660 /* The SRv6 End.DT4/DT6 behavior extracts the inner (IPv4/IPv6) packet and
661  * routes the IPv4/IPv6 packet by looking at the configured routing table.
662  *
663  * In the SRv6 End.DT4/DT6 use case, we can receive traffic (IPv6+Segment
664  * Routing Header packets) from several interfaces and the outer IPv6
665  * destination address (DA) is used for retrieving the specific instance of the
666  * End.DT4/DT6 behavior that should process the packets.
667  *
668  * However, the inner IPv4/IPv6 packet is not really bound to any receiving
669  * interface and thus the End.DT4/DT6 sets the VRF (associated with the
670  * corresponding routing table) as the *receiving* interface.
671  * In other words, the End.DT4/DT6 processes a packet as if it has been received
672  * directly by the VRF (and not by one of its slave devices, if any).
673  * In this way, the VRF interface is used for routing the IPv4/IPv6 packet in
674  * according to the routing table configured by the End.DT4/DT6 instance.
675  *
676  * This design allows you to get some interesting features like:
677  *  1) the statistics on rx packets;
678  *  2) the possibility to install a packet sniffer on the receiving interface
679  *     (the VRF one) for looking at the incoming packets;
680  *  3) the possibility to leverage the netfilter prerouting hook for the inner
681  *     IPv4 packet.
682  *
683  * This function returns:
684  *  - the sk_buff* when the VRF rcv handler has processed the packet correctly;
685  *  - NULL when the skb is consumed by the VRF rcv handler;
686  *  - a pointer which encodes a negative error number in case of error.
687  *    Note that in this case, the function takes care of freeing the skb.
688  */
689 static struct sk_buff *end_dt_vrf_rcv(struct sk_buff *skb, u16 family,
690 				      struct net_device *dev)
691 {
692 	/* based on l3mdev_ip_rcv; we are only interested in the master */
693 	if (unlikely(!netif_is_l3_master(dev) && !netif_has_l3_rx_handler(dev)))
694 		goto drop;
695 
696 	if (unlikely(!dev->l3mdev_ops->l3mdev_l3_rcv))
697 		goto drop;
698 
699 	/* the decap packet IPv4/IPv6 does not come with any mac header info.
700 	 * We must unset the mac header to allow the VRF device to rebuild it,
701 	 * just in case there is a sniffer attached on the device.
702 	 */
703 	skb_unset_mac_header(skb);
704 
705 	skb = dev->l3mdev_ops->l3mdev_l3_rcv(dev, skb, family);
706 	if (!skb)
707 		/* the skb buffer was consumed by the handler */
708 		return NULL;
709 
710 	/* when a packet is received by a VRF or by one of its slaves, the
711 	 * master device reference is set into the skb.
712 	 */
713 	if (unlikely(skb->dev != dev || skb->skb_iif != dev->ifindex))
714 		goto drop;
715 
716 	return skb;
717 
718 drop:
719 	kfree_skb(skb);
720 	return ERR_PTR(-EINVAL);
721 }
722 
723 static struct net_device *end_dt_get_vrf_rcu(struct sk_buff *skb,
724 					     struct seg6_end_dt_info *info)
725 {
726 	int vrf_ifindex = info->vrf_ifindex;
727 	struct net *net = info->net;
728 
729 	if (unlikely(vrf_ifindex < 0))
730 		goto error;
731 
732 	if (unlikely(!net_eq(dev_net(skb->dev), net)))
733 		goto error;
734 
735 	return dev_get_by_index_rcu(net, vrf_ifindex);
736 
737 error:
738 	return NULL;
739 }
740 
741 static struct sk_buff *end_dt_vrf_core(struct sk_buff *skb,
742 				       struct seg6_local_lwt *slwt, u16 family)
743 {
744 	struct seg6_end_dt_info *info = &slwt->dt_info;
745 	struct net_device *vrf;
746 	__be16 protocol;
747 	int hdrlen;
748 
749 	vrf = end_dt_get_vrf_rcu(skb, info);
750 	if (unlikely(!vrf))
751 		goto drop;
752 
753 	switch (family) {
754 	case AF_INET:
755 		protocol = htons(ETH_P_IP);
756 		hdrlen = sizeof(struct iphdr);
757 		break;
758 	case AF_INET6:
759 		protocol = htons(ETH_P_IPV6);
760 		hdrlen = sizeof(struct ipv6hdr);
761 		break;
762 	case AF_UNSPEC:
763 		fallthrough;
764 	default:
765 		goto drop;
766 	}
767 
768 	if (unlikely(info->family != AF_UNSPEC && info->family != family)) {
769 		pr_warn_once("seg6local: SRv6 End.DT* family mismatch");
770 		goto drop;
771 	}
772 
773 	skb->protocol = protocol;
774 
775 	skb_dst_drop(skb);
776 
777 	skb_set_transport_header(skb, hdrlen);
778 	nf_reset_ct(skb);
779 
780 	return end_dt_vrf_rcv(skb, family, vrf);
781 
782 drop:
783 	kfree_skb(skb);
784 	return ERR_PTR(-EINVAL);
785 }
786 
787 static int input_action_end_dt4(struct sk_buff *skb,
788 				struct seg6_local_lwt *slwt)
789 {
790 	struct iphdr *iph;
791 	int err;
792 
793 	if (!decap_and_validate(skb, IPPROTO_IPIP))
794 		goto drop;
795 
796 	if (!pskb_may_pull(skb, sizeof(struct iphdr)))
797 		goto drop;
798 
799 	skb = end_dt_vrf_core(skb, slwt, AF_INET);
800 	if (!skb)
801 		/* packet has been processed and consumed by the VRF */
802 		return 0;
803 
804 	if (IS_ERR(skb))
805 		return PTR_ERR(skb);
806 
807 	iph = ip_hdr(skb);
808 
809 	err = ip_route_input(skb, iph->daddr, iph->saddr, 0, skb->dev);
810 	if (unlikely(err))
811 		goto drop;
812 
813 	return dst_input(skb);
814 
815 drop:
816 	kfree_skb(skb);
817 	return -EINVAL;
818 }
819 
820 static int seg6_end_dt4_build(struct seg6_local_lwt *slwt, const void *cfg,
821 			      struct netlink_ext_ack *extack)
822 {
823 	return __seg6_end_dt_vrf_build(slwt, cfg, AF_INET, extack);
824 }
825 
826 static enum
827 seg6_end_dt_mode seg6_end_dt6_parse_mode(struct seg6_local_lwt *slwt)
828 {
829 	unsigned long parsed_optattrs = slwt->parsed_optattrs;
830 	bool legacy, vrfmode;
831 
832 	legacy	= !!(parsed_optattrs & SEG6_F_ATTR(SEG6_LOCAL_TABLE));
833 	vrfmode	= !!(parsed_optattrs & SEG6_F_ATTR(SEG6_LOCAL_VRFTABLE));
834 
835 	if (!(legacy ^ vrfmode))
836 		/* both are absent or present: invalid DT6 mode */
837 		return DT_INVALID_MODE;
838 
839 	return legacy ? DT_LEGACY_MODE : DT_VRF_MODE;
840 }
841 
842 static enum seg6_end_dt_mode seg6_end_dt6_get_mode(struct seg6_local_lwt *slwt)
843 {
844 	struct seg6_end_dt_info *info = &slwt->dt_info;
845 
846 	return info->mode;
847 }
848 
849 static int seg6_end_dt6_build(struct seg6_local_lwt *slwt, const void *cfg,
850 			      struct netlink_ext_ack *extack)
851 {
852 	enum seg6_end_dt_mode mode = seg6_end_dt6_parse_mode(slwt);
853 	struct seg6_end_dt_info *info = &slwt->dt_info;
854 
855 	switch (mode) {
856 	case DT_LEGACY_MODE:
857 		info->mode = DT_LEGACY_MODE;
858 		return 0;
859 	case DT_VRF_MODE:
860 		return __seg6_end_dt_vrf_build(slwt, cfg, AF_INET6, extack);
861 	default:
862 		NL_SET_ERR_MSG(extack, "table or vrftable must be specified");
863 		return -EINVAL;
864 	}
865 }
866 #endif
867 
868 static int input_action_end_dt6(struct sk_buff *skb,
869 				struct seg6_local_lwt *slwt)
870 {
871 	if (!decap_and_validate(skb, IPPROTO_IPV6))
872 		goto drop;
873 
874 	if (!pskb_may_pull(skb, sizeof(struct ipv6hdr)))
875 		goto drop;
876 
877 #ifdef CONFIG_NET_L3_MASTER_DEV
878 	if (seg6_end_dt6_get_mode(slwt) == DT_LEGACY_MODE)
879 		goto legacy_mode;
880 
881 	/* DT6_VRF_MODE */
882 	skb = end_dt_vrf_core(skb, slwt, AF_INET6);
883 	if (!skb)
884 		/* packet has been processed and consumed by the VRF */
885 		return 0;
886 
887 	if (IS_ERR(skb))
888 		return PTR_ERR(skb);
889 
890 	/* note: this time we do not need to specify the table because the VRF
891 	 * takes care of selecting the correct table.
892 	 */
893 	seg6_lookup_any_nexthop(skb, NULL, 0, true);
894 
895 	return dst_input(skb);
896 
897 legacy_mode:
898 #endif
899 	skb_set_transport_header(skb, sizeof(struct ipv6hdr));
900 
901 	seg6_lookup_any_nexthop(skb, NULL, slwt->table, true);
902 
903 	return dst_input(skb);
904 
905 drop:
906 	kfree_skb(skb);
907 	return -EINVAL;
908 }
909 
910 #ifdef CONFIG_NET_L3_MASTER_DEV
911 static int seg6_end_dt46_build(struct seg6_local_lwt *slwt, const void *cfg,
912 			       struct netlink_ext_ack *extack)
913 {
914 	return __seg6_end_dt_vrf_build(slwt, cfg, AF_UNSPEC, extack);
915 }
916 
917 static int input_action_end_dt46(struct sk_buff *skb,
918 				 struct seg6_local_lwt *slwt)
919 {
920 	unsigned int off = 0;
921 	int nexthdr;
922 
923 	nexthdr = ipv6_find_hdr(skb, &off, -1, NULL, NULL);
924 	if (unlikely(nexthdr < 0))
925 		goto drop;
926 
927 	switch (nexthdr) {
928 	case IPPROTO_IPIP:
929 		return input_action_end_dt4(skb, slwt);
930 	case IPPROTO_IPV6:
931 		return input_action_end_dt6(skb, slwt);
932 	}
933 
934 drop:
935 	kfree_skb(skb);
936 	return -EINVAL;
937 }
938 #endif
939 
940 /* push an SRH on top of the current one */
941 static int input_action_end_b6(struct sk_buff *skb, struct seg6_local_lwt *slwt)
942 {
943 	struct ipv6_sr_hdr *srh;
944 	int err = -EINVAL;
945 
946 	srh = get_and_validate_srh(skb);
947 	if (!srh)
948 		goto drop;
949 
950 	err = seg6_do_srh_inline(skb, slwt->srh);
951 	if (err)
952 		goto drop;
953 
954 	skb_set_transport_header(skb, sizeof(struct ipv6hdr));
955 
956 	seg6_lookup_nexthop(skb, NULL, 0);
957 
958 	return dst_input(skb);
959 
960 drop:
961 	kfree_skb(skb);
962 	return err;
963 }
964 
965 /* encapsulate within an outer IPv6 header and a specified SRH */
966 static int input_action_end_b6_encap(struct sk_buff *skb,
967 				     struct seg6_local_lwt *slwt)
968 {
969 	struct ipv6_sr_hdr *srh;
970 	int err = -EINVAL;
971 
972 	srh = get_and_validate_srh(skb);
973 	if (!srh)
974 		goto drop;
975 
976 	advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
977 
978 	skb_reset_inner_headers(skb);
979 	skb->encapsulation = 1;
980 
981 	err = seg6_do_srh_encap(skb, slwt->srh, IPPROTO_IPV6);
982 	if (err)
983 		goto drop;
984 
985 	skb_set_transport_header(skb, sizeof(struct ipv6hdr));
986 
987 	seg6_lookup_nexthop(skb, NULL, 0);
988 
989 	return dst_input(skb);
990 
991 drop:
992 	kfree_skb(skb);
993 	return err;
994 }
995 
996 DEFINE_PER_CPU(struct seg6_bpf_srh_state, seg6_bpf_srh_states);
997 
998 bool seg6_bpf_has_valid_srh(struct sk_buff *skb)
999 {
1000 	struct seg6_bpf_srh_state *srh_state =
1001 		this_cpu_ptr(&seg6_bpf_srh_states);
1002 	struct ipv6_sr_hdr *srh = srh_state->srh;
1003 
1004 	if (unlikely(srh == NULL))
1005 		return false;
1006 
1007 	if (unlikely(!srh_state->valid)) {
1008 		if ((srh_state->hdrlen & 7) != 0)
1009 			return false;
1010 
1011 		srh->hdrlen = (u8)(srh_state->hdrlen >> 3);
1012 		if (!seg6_validate_srh(srh, (srh->hdrlen + 1) << 3, true))
1013 			return false;
1014 
1015 		srh_state->valid = true;
1016 	}
1017 
1018 	return true;
1019 }
1020 
1021 static int input_action_end_bpf(struct sk_buff *skb,
1022 				struct seg6_local_lwt *slwt)
1023 {
1024 	struct seg6_bpf_srh_state *srh_state =
1025 		this_cpu_ptr(&seg6_bpf_srh_states);
1026 	struct ipv6_sr_hdr *srh;
1027 	int ret;
1028 
1029 	srh = get_and_validate_srh(skb);
1030 	if (!srh) {
1031 		kfree_skb(skb);
1032 		return -EINVAL;
1033 	}
1034 	advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
1035 
1036 	/* preempt_disable is needed to protect the per-CPU buffer srh_state,
1037 	 * which is also accessed by the bpf_lwt_seg6_* helpers
1038 	 */
1039 	preempt_disable();
1040 	srh_state->srh = srh;
1041 	srh_state->hdrlen = srh->hdrlen << 3;
1042 	srh_state->valid = true;
1043 
1044 	rcu_read_lock();
1045 	bpf_compute_data_pointers(skb);
1046 	ret = bpf_prog_run_save_cb(slwt->bpf.prog, skb);
1047 	rcu_read_unlock();
1048 
1049 	switch (ret) {
1050 	case BPF_OK:
1051 	case BPF_REDIRECT:
1052 		break;
1053 	case BPF_DROP:
1054 		goto drop;
1055 	default:
1056 		pr_warn_once("bpf-seg6local: Illegal return value %u\n", ret);
1057 		goto drop;
1058 	}
1059 
1060 	if (srh_state->srh && !seg6_bpf_has_valid_srh(skb))
1061 		goto drop;
1062 
1063 	preempt_enable();
1064 	if (ret != BPF_REDIRECT)
1065 		seg6_lookup_nexthop(skb, NULL, 0);
1066 
1067 	return dst_input(skb);
1068 
1069 drop:
1070 	preempt_enable();
1071 	kfree_skb(skb);
1072 	return -EINVAL;
1073 }
1074 
1075 static struct seg6_action_desc seg6_action_table[] = {
1076 	{
1077 		.action		= SEG6_LOCAL_ACTION_END,
1078 		.attrs		= 0,
1079 		.optattrs	= SEG6_F_LOCAL_COUNTERS |
1080 				  SEG6_F_ATTR(SEG6_LOCAL_FLAVORS),
1081 		.input		= input_action_end,
1082 	},
1083 	{
1084 		.action		= SEG6_LOCAL_ACTION_END_X,
1085 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_NH6),
1086 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1087 		.input		= input_action_end_x,
1088 	},
1089 	{
1090 		.action		= SEG6_LOCAL_ACTION_END_T,
1091 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_TABLE),
1092 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1093 		.input		= input_action_end_t,
1094 	},
1095 	{
1096 		.action		= SEG6_LOCAL_ACTION_END_DX2,
1097 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_OIF),
1098 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1099 		.input		= input_action_end_dx2,
1100 	},
1101 	{
1102 		.action		= SEG6_LOCAL_ACTION_END_DX6,
1103 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_NH6),
1104 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1105 		.input		= input_action_end_dx6,
1106 	},
1107 	{
1108 		.action		= SEG6_LOCAL_ACTION_END_DX4,
1109 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_NH4),
1110 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1111 		.input		= input_action_end_dx4,
1112 	},
1113 	{
1114 		.action		= SEG6_LOCAL_ACTION_END_DT4,
1115 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_VRFTABLE),
1116 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1117 #ifdef CONFIG_NET_L3_MASTER_DEV
1118 		.input		= input_action_end_dt4,
1119 		.slwt_ops	= {
1120 					.build_state = seg6_end_dt4_build,
1121 				  },
1122 #endif
1123 	},
1124 	{
1125 		.action		= SEG6_LOCAL_ACTION_END_DT6,
1126 #ifdef CONFIG_NET_L3_MASTER_DEV
1127 		.attrs		= 0,
1128 		.optattrs	= SEG6_F_LOCAL_COUNTERS		|
1129 				  SEG6_F_ATTR(SEG6_LOCAL_TABLE) |
1130 				  SEG6_F_ATTR(SEG6_LOCAL_VRFTABLE),
1131 		.slwt_ops	= {
1132 					.build_state = seg6_end_dt6_build,
1133 				  },
1134 #else
1135 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_TABLE),
1136 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1137 #endif
1138 		.input		= input_action_end_dt6,
1139 	},
1140 	{
1141 		.action		= SEG6_LOCAL_ACTION_END_DT46,
1142 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_VRFTABLE),
1143 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1144 #ifdef CONFIG_NET_L3_MASTER_DEV
1145 		.input		= input_action_end_dt46,
1146 		.slwt_ops	= {
1147 					.build_state = seg6_end_dt46_build,
1148 				  },
1149 #endif
1150 	},
1151 	{
1152 		.action		= SEG6_LOCAL_ACTION_END_B6,
1153 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_SRH),
1154 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1155 		.input		= input_action_end_b6,
1156 	},
1157 	{
1158 		.action		= SEG6_LOCAL_ACTION_END_B6_ENCAP,
1159 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_SRH),
1160 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1161 		.input		= input_action_end_b6_encap,
1162 		.static_headroom	= sizeof(struct ipv6hdr),
1163 	},
1164 	{
1165 		.action		= SEG6_LOCAL_ACTION_END_BPF,
1166 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_BPF),
1167 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1168 		.input		= input_action_end_bpf,
1169 	},
1170 
1171 };
1172 
1173 static struct seg6_action_desc *__get_action_desc(int action)
1174 {
1175 	struct seg6_action_desc *desc;
1176 	int i, count;
1177 
1178 	count = ARRAY_SIZE(seg6_action_table);
1179 	for (i = 0; i < count; i++) {
1180 		desc = &seg6_action_table[i];
1181 		if (desc->action == action)
1182 			return desc;
1183 	}
1184 
1185 	return NULL;
1186 }
1187 
1188 static bool seg6_lwtunnel_counters_enabled(struct seg6_local_lwt *slwt)
1189 {
1190 	return slwt->parsed_optattrs & SEG6_F_LOCAL_COUNTERS;
1191 }
1192 
1193 static void seg6_local_update_counters(struct seg6_local_lwt *slwt,
1194 				       unsigned int len, int err)
1195 {
1196 	struct pcpu_seg6_local_counters *pcounters;
1197 
1198 	pcounters = this_cpu_ptr(slwt->pcpu_counters);
1199 	u64_stats_update_begin(&pcounters->syncp);
1200 
1201 	if (likely(!err)) {
1202 		u64_stats_inc(&pcounters->packets);
1203 		u64_stats_add(&pcounters->bytes, len);
1204 	} else {
1205 		u64_stats_inc(&pcounters->errors);
1206 	}
1207 
1208 	u64_stats_update_end(&pcounters->syncp);
1209 }
1210 
1211 static int seg6_local_input_core(struct net *net, struct sock *sk,
1212 				 struct sk_buff *skb)
1213 {
1214 	struct dst_entry *orig_dst = skb_dst(skb);
1215 	struct seg6_action_desc *desc;
1216 	struct seg6_local_lwt *slwt;
1217 	unsigned int len = skb->len;
1218 	int rc;
1219 
1220 	slwt = seg6_local_lwtunnel(orig_dst->lwtstate);
1221 	desc = slwt->desc;
1222 
1223 	rc = desc->input(skb, slwt);
1224 
1225 	if (!seg6_lwtunnel_counters_enabled(slwt))
1226 		return rc;
1227 
1228 	seg6_local_update_counters(slwt, len, rc);
1229 
1230 	return rc;
1231 }
1232 
1233 static int seg6_local_input(struct sk_buff *skb)
1234 {
1235 	if (skb->protocol != htons(ETH_P_IPV6)) {
1236 		kfree_skb(skb);
1237 		return -EINVAL;
1238 	}
1239 
1240 	if (static_branch_unlikely(&nf_hooks_lwtunnel_enabled))
1241 		return NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_IN,
1242 			       dev_net(skb->dev), NULL, skb, skb->dev, NULL,
1243 			       seg6_local_input_core);
1244 
1245 	return seg6_local_input_core(dev_net(skb->dev), NULL, skb);
1246 }
1247 
1248 static const struct nla_policy seg6_local_policy[SEG6_LOCAL_MAX + 1] = {
1249 	[SEG6_LOCAL_ACTION]	= { .type = NLA_U32 },
1250 	[SEG6_LOCAL_SRH]	= { .type = NLA_BINARY },
1251 	[SEG6_LOCAL_TABLE]	= { .type = NLA_U32 },
1252 	[SEG6_LOCAL_VRFTABLE]	= { .type = NLA_U32 },
1253 	[SEG6_LOCAL_NH4]	= { .type = NLA_BINARY,
1254 				    .len = sizeof(struct in_addr) },
1255 	[SEG6_LOCAL_NH6]	= { .type = NLA_BINARY,
1256 				    .len = sizeof(struct in6_addr) },
1257 	[SEG6_LOCAL_IIF]	= { .type = NLA_U32 },
1258 	[SEG6_LOCAL_OIF]	= { .type = NLA_U32 },
1259 	[SEG6_LOCAL_BPF]	= { .type = NLA_NESTED },
1260 	[SEG6_LOCAL_COUNTERS]	= { .type = NLA_NESTED },
1261 	[SEG6_LOCAL_FLAVORS]	= { .type = NLA_NESTED },
1262 };
1263 
1264 static int parse_nla_srh(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1265 			 struct netlink_ext_ack *extack)
1266 {
1267 	struct ipv6_sr_hdr *srh;
1268 	int len;
1269 
1270 	srh = nla_data(attrs[SEG6_LOCAL_SRH]);
1271 	len = nla_len(attrs[SEG6_LOCAL_SRH]);
1272 
1273 	/* SRH must contain at least one segment */
1274 	if (len < sizeof(*srh) + sizeof(struct in6_addr))
1275 		return -EINVAL;
1276 
1277 	if (!seg6_validate_srh(srh, len, false))
1278 		return -EINVAL;
1279 
1280 	slwt->srh = kmemdup(srh, len, GFP_KERNEL);
1281 	if (!slwt->srh)
1282 		return -ENOMEM;
1283 
1284 	slwt->headroom += len;
1285 
1286 	return 0;
1287 }
1288 
1289 static int put_nla_srh(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1290 {
1291 	struct ipv6_sr_hdr *srh;
1292 	struct nlattr *nla;
1293 	int len;
1294 
1295 	srh = slwt->srh;
1296 	len = (srh->hdrlen + 1) << 3;
1297 
1298 	nla = nla_reserve(skb, SEG6_LOCAL_SRH, len);
1299 	if (!nla)
1300 		return -EMSGSIZE;
1301 
1302 	memcpy(nla_data(nla), srh, len);
1303 
1304 	return 0;
1305 }
1306 
1307 static int cmp_nla_srh(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1308 {
1309 	int len = (a->srh->hdrlen + 1) << 3;
1310 
1311 	if (len != ((b->srh->hdrlen + 1) << 3))
1312 		return 1;
1313 
1314 	return memcmp(a->srh, b->srh, len);
1315 }
1316 
1317 static void destroy_attr_srh(struct seg6_local_lwt *slwt)
1318 {
1319 	kfree(slwt->srh);
1320 }
1321 
1322 static int parse_nla_table(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1323 			   struct netlink_ext_ack *extack)
1324 {
1325 	slwt->table = nla_get_u32(attrs[SEG6_LOCAL_TABLE]);
1326 
1327 	return 0;
1328 }
1329 
1330 static int put_nla_table(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1331 {
1332 	if (nla_put_u32(skb, SEG6_LOCAL_TABLE, slwt->table))
1333 		return -EMSGSIZE;
1334 
1335 	return 0;
1336 }
1337 
1338 static int cmp_nla_table(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1339 {
1340 	if (a->table != b->table)
1341 		return 1;
1342 
1343 	return 0;
1344 }
1345 
1346 static struct
1347 seg6_end_dt_info *seg6_possible_end_dt_info(struct seg6_local_lwt *slwt)
1348 {
1349 #ifdef CONFIG_NET_L3_MASTER_DEV
1350 	return &slwt->dt_info;
1351 #else
1352 	return ERR_PTR(-EOPNOTSUPP);
1353 #endif
1354 }
1355 
1356 static int parse_nla_vrftable(struct nlattr **attrs,
1357 			      struct seg6_local_lwt *slwt,
1358 			      struct netlink_ext_ack *extack)
1359 {
1360 	struct seg6_end_dt_info *info = seg6_possible_end_dt_info(slwt);
1361 
1362 	if (IS_ERR(info))
1363 		return PTR_ERR(info);
1364 
1365 	info->vrf_table = nla_get_u32(attrs[SEG6_LOCAL_VRFTABLE]);
1366 
1367 	return 0;
1368 }
1369 
1370 static int put_nla_vrftable(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1371 {
1372 	struct seg6_end_dt_info *info = seg6_possible_end_dt_info(slwt);
1373 
1374 	if (IS_ERR(info))
1375 		return PTR_ERR(info);
1376 
1377 	if (nla_put_u32(skb, SEG6_LOCAL_VRFTABLE, info->vrf_table))
1378 		return -EMSGSIZE;
1379 
1380 	return 0;
1381 }
1382 
1383 static int cmp_nla_vrftable(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1384 {
1385 	struct seg6_end_dt_info *info_a = seg6_possible_end_dt_info(a);
1386 	struct seg6_end_dt_info *info_b = seg6_possible_end_dt_info(b);
1387 
1388 	if (info_a->vrf_table != info_b->vrf_table)
1389 		return 1;
1390 
1391 	return 0;
1392 }
1393 
1394 static int parse_nla_nh4(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1395 			 struct netlink_ext_ack *extack)
1396 {
1397 	memcpy(&slwt->nh4, nla_data(attrs[SEG6_LOCAL_NH4]),
1398 	       sizeof(struct in_addr));
1399 
1400 	return 0;
1401 }
1402 
1403 static int put_nla_nh4(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1404 {
1405 	struct nlattr *nla;
1406 
1407 	nla = nla_reserve(skb, SEG6_LOCAL_NH4, sizeof(struct in_addr));
1408 	if (!nla)
1409 		return -EMSGSIZE;
1410 
1411 	memcpy(nla_data(nla), &slwt->nh4, sizeof(struct in_addr));
1412 
1413 	return 0;
1414 }
1415 
1416 static int cmp_nla_nh4(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1417 {
1418 	return memcmp(&a->nh4, &b->nh4, sizeof(struct in_addr));
1419 }
1420 
1421 static int parse_nla_nh6(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1422 			 struct netlink_ext_ack *extack)
1423 {
1424 	memcpy(&slwt->nh6, nla_data(attrs[SEG6_LOCAL_NH6]),
1425 	       sizeof(struct in6_addr));
1426 
1427 	return 0;
1428 }
1429 
1430 static int put_nla_nh6(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1431 {
1432 	struct nlattr *nla;
1433 
1434 	nla = nla_reserve(skb, SEG6_LOCAL_NH6, sizeof(struct in6_addr));
1435 	if (!nla)
1436 		return -EMSGSIZE;
1437 
1438 	memcpy(nla_data(nla), &slwt->nh6, sizeof(struct in6_addr));
1439 
1440 	return 0;
1441 }
1442 
1443 static int cmp_nla_nh6(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1444 {
1445 	return memcmp(&a->nh6, &b->nh6, sizeof(struct in6_addr));
1446 }
1447 
1448 static int parse_nla_iif(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1449 			 struct netlink_ext_ack *extack)
1450 {
1451 	slwt->iif = nla_get_u32(attrs[SEG6_LOCAL_IIF]);
1452 
1453 	return 0;
1454 }
1455 
1456 static int put_nla_iif(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1457 {
1458 	if (nla_put_u32(skb, SEG6_LOCAL_IIF, slwt->iif))
1459 		return -EMSGSIZE;
1460 
1461 	return 0;
1462 }
1463 
1464 static int cmp_nla_iif(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1465 {
1466 	if (a->iif != b->iif)
1467 		return 1;
1468 
1469 	return 0;
1470 }
1471 
1472 static int parse_nla_oif(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1473 			 struct netlink_ext_ack *extack)
1474 {
1475 	slwt->oif = nla_get_u32(attrs[SEG6_LOCAL_OIF]);
1476 
1477 	return 0;
1478 }
1479 
1480 static int put_nla_oif(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1481 {
1482 	if (nla_put_u32(skb, SEG6_LOCAL_OIF, slwt->oif))
1483 		return -EMSGSIZE;
1484 
1485 	return 0;
1486 }
1487 
1488 static int cmp_nla_oif(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1489 {
1490 	if (a->oif != b->oif)
1491 		return 1;
1492 
1493 	return 0;
1494 }
1495 
1496 #define MAX_PROG_NAME 256
1497 static const struct nla_policy bpf_prog_policy[SEG6_LOCAL_BPF_PROG_MAX + 1] = {
1498 	[SEG6_LOCAL_BPF_PROG]	   = { .type = NLA_U32, },
1499 	[SEG6_LOCAL_BPF_PROG_NAME] = { .type = NLA_NUL_STRING,
1500 				       .len = MAX_PROG_NAME },
1501 };
1502 
1503 static int parse_nla_bpf(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1504 			 struct netlink_ext_ack *extack)
1505 {
1506 	struct nlattr *tb[SEG6_LOCAL_BPF_PROG_MAX + 1];
1507 	struct bpf_prog *p;
1508 	int ret;
1509 	u32 fd;
1510 
1511 	ret = nla_parse_nested_deprecated(tb, SEG6_LOCAL_BPF_PROG_MAX,
1512 					  attrs[SEG6_LOCAL_BPF],
1513 					  bpf_prog_policy, NULL);
1514 	if (ret < 0)
1515 		return ret;
1516 
1517 	if (!tb[SEG6_LOCAL_BPF_PROG] || !tb[SEG6_LOCAL_BPF_PROG_NAME])
1518 		return -EINVAL;
1519 
1520 	slwt->bpf.name = nla_memdup(tb[SEG6_LOCAL_BPF_PROG_NAME], GFP_KERNEL);
1521 	if (!slwt->bpf.name)
1522 		return -ENOMEM;
1523 
1524 	fd = nla_get_u32(tb[SEG6_LOCAL_BPF_PROG]);
1525 	p = bpf_prog_get_type(fd, BPF_PROG_TYPE_LWT_SEG6LOCAL);
1526 	if (IS_ERR(p)) {
1527 		kfree(slwt->bpf.name);
1528 		return PTR_ERR(p);
1529 	}
1530 
1531 	slwt->bpf.prog = p;
1532 	return 0;
1533 }
1534 
1535 static int put_nla_bpf(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1536 {
1537 	struct nlattr *nest;
1538 
1539 	if (!slwt->bpf.prog)
1540 		return 0;
1541 
1542 	nest = nla_nest_start_noflag(skb, SEG6_LOCAL_BPF);
1543 	if (!nest)
1544 		return -EMSGSIZE;
1545 
1546 	if (nla_put_u32(skb, SEG6_LOCAL_BPF_PROG, slwt->bpf.prog->aux->id))
1547 		return -EMSGSIZE;
1548 
1549 	if (slwt->bpf.name &&
1550 	    nla_put_string(skb, SEG6_LOCAL_BPF_PROG_NAME, slwt->bpf.name))
1551 		return -EMSGSIZE;
1552 
1553 	return nla_nest_end(skb, nest);
1554 }
1555 
1556 static int cmp_nla_bpf(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1557 {
1558 	if (!a->bpf.name && !b->bpf.name)
1559 		return 0;
1560 
1561 	if (!a->bpf.name || !b->bpf.name)
1562 		return 1;
1563 
1564 	return strcmp(a->bpf.name, b->bpf.name);
1565 }
1566 
1567 static void destroy_attr_bpf(struct seg6_local_lwt *slwt)
1568 {
1569 	kfree(slwt->bpf.name);
1570 	if (slwt->bpf.prog)
1571 		bpf_prog_put(slwt->bpf.prog);
1572 }
1573 
1574 static const struct
1575 nla_policy seg6_local_counters_policy[SEG6_LOCAL_CNT_MAX + 1] = {
1576 	[SEG6_LOCAL_CNT_PACKETS]	= { .type = NLA_U64 },
1577 	[SEG6_LOCAL_CNT_BYTES]		= { .type = NLA_U64 },
1578 	[SEG6_LOCAL_CNT_ERRORS]		= { .type = NLA_U64 },
1579 };
1580 
1581 static int parse_nla_counters(struct nlattr **attrs,
1582 			      struct seg6_local_lwt *slwt,
1583 			      struct netlink_ext_ack *extack)
1584 {
1585 	struct pcpu_seg6_local_counters __percpu *pcounters;
1586 	struct nlattr *tb[SEG6_LOCAL_CNT_MAX + 1];
1587 	int ret;
1588 
1589 	ret = nla_parse_nested_deprecated(tb, SEG6_LOCAL_CNT_MAX,
1590 					  attrs[SEG6_LOCAL_COUNTERS],
1591 					  seg6_local_counters_policy, NULL);
1592 	if (ret < 0)
1593 		return ret;
1594 
1595 	/* basic support for SRv6 Behavior counters requires at least:
1596 	 * packets, bytes and errors.
1597 	 */
1598 	if (!tb[SEG6_LOCAL_CNT_PACKETS] || !tb[SEG6_LOCAL_CNT_BYTES] ||
1599 	    !tb[SEG6_LOCAL_CNT_ERRORS])
1600 		return -EINVAL;
1601 
1602 	/* counters are always zero initialized */
1603 	pcounters = seg6_local_alloc_pcpu_counters(GFP_KERNEL);
1604 	if (!pcounters)
1605 		return -ENOMEM;
1606 
1607 	slwt->pcpu_counters = pcounters;
1608 
1609 	return 0;
1610 }
1611 
1612 static int seg6_local_fill_nla_counters(struct sk_buff *skb,
1613 					struct seg6_local_counters *counters)
1614 {
1615 	if (nla_put_u64_64bit(skb, SEG6_LOCAL_CNT_PACKETS, counters->packets,
1616 			      SEG6_LOCAL_CNT_PAD))
1617 		return -EMSGSIZE;
1618 
1619 	if (nla_put_u64_64bit(skb, SEG6_LOCAL_CNT_BYTES, counters->bytes,
1620 			      SEG6_LOCAL_CNT_PAD))
1621 		return -EMSGSIZE;
1622 
1623 	if (nla_put_u64_64bit(skb, SEG6_LOCAL_CNT_ERRORS, counters->errors,
1624 			      SEG6_LOCAL_CNT_PAD))
1625 		return -EMSGSIZE;
1626 
1627 	return 0;
1628 }
1629 
1630 static int put_nla_counters(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1631 {
1632 	struct seg6_local_counters counters = { 0, 0, 0 };
1633 	struct nlattr *nest;
1634 	int rc, i;
1635 
1636 	nest = nla_nest_start(skb, SEG6_LOCAL_COUNTERS);
1637 	if (!nest)
1638 		return -EMSGSIZE;
1639 
1640 	for_each_possible_cpu(i) {
1641 		struct pcpu_seg6_local_counters *pcounters;
1642 		u64 packets, bytes, errors;
1643 		unsigned int start;
1644 
1645 		pcounters = per_cpu_ptr(slwt->pcpu_counters, i);
1646 		do {
1647 			start = u64_stats_fetch_begin_irq(&pcounters->syncp);
1648 
1649 			packets = u64_stats_read(&pcounters->packets);
1650 			bytes = u64_stats_read(&pcounters->bytes);
1651 			errors = u64_stats_read(&pcounters->errors);
1652 
1653 		} while (u64_stats_fetch_retry_irq(&pcounters->syncp, start));
1654 
1655 		counters.packets += packets;
1656 		counters.bytes += bytes;
1657 		counters.errors += errors;
1658 	}
1659 
1660 	rc = seg6_local_fill_nla_counters(skb, &counters);
1661 	if (rc < 0) {
1662 		nla_nest_cancel(skb, nest);
1663 		return rc;
1664 	}
1665 
1666 	return nla_nest_end(skb, nest);
1667 }
1668 
1669 static int cmp_nla_counters(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1670 {
1671 	/* a and b are equal if both have pcpu_counters set or not */
1672 	return (!!((unsigned long)a->pcpu_counters)) ^
1673 		(!!((unsigned long)b->pcpu_counters));
1674 }
1675 
1676 static void destroy_attr_counters(struct seg6_local_lwt *slwt)
1677 {
1678 	free_percpu(slwt->pcpu_counters);
1679 }
1680 
1681 static const
1682 struct nla_policy seg6_local_flavors_policy[SEG6_LOCAL_FLV_MAX + 1] = {
1683 	[SEG6_LOCAL_FLV_OPERATION]	= { .type = NLA_U32 },
1684 	[SEG6_LOCAL_FLV_LCBLOCK_BITS]	= { .type = NLA_U8 },
1685 	[SEG6_LOCAL_FLV_LCNODE_FN_BITS]	= { .type = NLA_U8 },
1686 };
1687 
1688 /* check whether the lengths of the Locator-Block and Locator-Node Function
1689  * are compatible with the dimension of a C-SID container.
1690  */
1691 static int seg6_chk_next_csid_cfg(__u8 block_len, __u8 func_len)
1692 {
1693 	/* Locator-Block and Locator-Node Function cannot exceed 128 bits
1694 	 * (i.e. C-SID container lenghts).
1695 	 */
1696 	if (next_csid_chk_cntr_bits(block_len, func_len))
1697 		return -EINVAL;
1698 
1699 	/* Locator-Block length must be greater than zero and evenly divisible
1700 	 * by 8. There must be room for a Locator-Node Function, at least.
1701 	 */
1702 	if (next_csid_chk_lcblock_bits(block_len))
1703 		return -EINVAL;
1704 
1705 	/* Locator-Node Function length must be greater than zero and evenly
1706 	 * divisible by 8. There must be room for the Locator-Block.
1707 	 */
1708 	if (next_csid_chk_lcnode_fn_bits(func_len))
1709 		return -EINVAL;
1710 
1711 	return 0;
1712 }
1713 
1714 static int seg6_parse_nla_next_csid_cfg(struct nlattr **tb,
1715 					struct seg6_flavors_info *finfo,
1716 					struct netlink_ext_ack *extack)
1717 {
1718 	__u8 func_len = SEG6_LOCAL_LCNODE_FN_DBITS;
1719 	__u8 block_len = SEG6_LOCAL_LCBLOCK_DBITS;
1720 	int rc;
1721 
1722 	if (tb[SEG6_LOCAL_FLV_LCBLOCK_BITS])
1723 		block_len = nla_get_u8(tb[SEG6_LOCAL_FLV_LCBLOCK_BITS]);
1724 
1725 	if (tb[SEG6_LOCAL_FLV_LCNODE_FN_BITS])
1726 		func_len = nla_get_u8(tb[SEG6_LOCAL_FLV_LCNODE_FN_BITS]);
1727 
1728 	rc = seg6_chk_next_csid_cfg(block_len, func_len);
1729 	if (rc < 0) {
1730 		NL_SET_ERR_MSG(extack,
1731 			       "Invalid Locator Block/Node Function lengths");
1732 		return rc;
1733 	}
1734 
1735 	finfo->lcblock_bits = block_len;
1736 	finfo->lcnode_func_bits = func_len;
1737 
1738 	return 0;
1739 }
1740 
1741 static int parse_nla_flavors(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1742 			     struct netlink_ext_ack *extack)
1743 {
1744 	struct seg6_flavors_info *finfo = &slwt->flv_info;
1745 	struct nlattr *tb[SEG6_LOCAL_FLV_MAX + 1];
1746 	unsigned long fops;
1747 	int rc;
1748 
1749 	rc = nla_parse_nested_deprecated(tb, SEG6_LOCAL_FLV_MAX,
1750 					 attrs[SEG6_LOCAL_FLAVORS],
1751 					 seg6_local_flavors_policy, NULL);
1752 	if (rc < 0)
1753 		return rc;
1754 
1755 	/* this attribute MUST always be present since it represents the Flavor
1756 	 * operation(s) to be carried out.
1757 	 */
1758 	if (!tb[SEG6_LOCAL_FLV_OPERATION])
1759 		return -EINVAL;
1760 
1761 	fops = nla_get_u32(tb[SEG6_LOCAL_FLV_OPERATION]);
1762 	if (fops & ~SEG6_LOCAL_FLV_SUPP_OPS) {
1763 		NL_SET_ERR_MSG(extack, "Unsupported Flavor operation(s)");
1764 		return -EOPNOTSUPP;
1765 	}
1766 
1767 	finfo->flv_ops = fops;
1768 
1769 	if (seg6_next_csid_enabled(fops)) {
1770 		/* Locator-Block and Locator-Node Function lengths can be
1771 		 * provided by the user space. Otherwise, default values are
1772 		 * applied.
1773 		 */
1774 		rc = seg6_parse_nla_next_csid_cfg(tb, finfo, extack);
1775 		if (rc < 0)
1776 			return rc;
1777 	}
1778 
1779 	return 0;
1780 }
1781 
1782 static int seg6_fill_nla_next_csid_cfg(struct sk_buff *skb,
1783 				       struct seg6_flavors_info *finfo)
1784 {
1785 	if (nla_put_u8(skb, SEG6_LOCAL_FLV_LCBLOCK_BITS, finfo->lcblock_bits))
1786 		return -EMSGSIZE;
1787 
1788 	if (nla_put_u8(skb, SEG6_LOCAL_FLV_LCNODE_FN_BITS,
1789 		       finfo->lcnode_func_bits))
1790 		return -EMSGSIZE;
1791 
1792 	return 0;
1793 }
1794 
1795 static int put_nla_flavors(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1796 {
1797 	struct seg6_flavors_info *finfo = &slwt->flv_info;
1798 	__u32 fops = finfo->flv_ops;
1799 	struct nlattr *nest;
1800 	int rc;
1801 
1802 	nest = nla_nest_start(skb, SEG6_LOCAL_FLAVORS);
1803 	if (!nest)
1804 		return -EMSGSIZE;
1805 
1806 	if (nla_put_u32(skb, SEG6_LOCAL_FLV_OPERATION, fops)) {
1807 		rc = -EMSGSIZE;
1808 		goto err;
1809 	}
1810 
1811 	if (seg6_next_csid_enabled(fops)) {
1812 		rc = seg6_fill_nla_next_csid_cfg(skb, finfo);
1813 		if (rc < 0)
1814 			goto err;
1815 	}
1816 
1817 	return nla_nest_end(skb, nest);
1818 
1819 err:
1820 	nla_nest_cancel(skb, nest);
1821 	return rc;
1822 }
1823 
1824 static int seg6_cmp_nla_next_csid_cfg(struct seg6_flavors_info *finfo_a,
1825 				      struct seg6_flavors_info *finfo_b)
1826 {
1827 	if (finfo_a->lcblock_bits != finfo_b->lcblock_bits)
1828 		return 1;
1829 
1830 	if (finfo_a->lcnode_func_bits != finfo_b->lcnode_func_bits)
1831 		return 1;
1832 
1833 	return 0;
1834 }
1835 
1836 static int cmp_nla_flavors(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1837 {
1838 	struct seg6_flavors_info *finfo_a = &a->flv_info;
1839 	struct seg6_flavors_info *finfo_b = &b->flv_info;
1840 
1841 	if (finfo_a->flv_ops != finfo_b->flv_ops)
1842 		return 1;
1843 
1844 	if (seg6_next_csid_enabled(finfo_a->flv_ops)) {
1845 		if (seg6_cmp_nla_next_csid_cfg(finfo_a, finfo_b))
1846 			return 1;
1847 	}
1848 
1849 	return 0;
1850 }
1851 
1852 static int encap_size_flavors(struct seg6_local_lwt *slwt)
1853 {
1854 	struct seg6_flavors_info *finfo = &slwt->flv_info;
1855 	int nlsize;
1856 
1857 	nlsize = nla_total_size(0) +	/* nest SEG6_LOCAL_FLAVORS */
1858 		 nla_total_size(4);	/* SEG6_LOCAL_FLV_OPERATION */
1859 
1860 	if (seg6_next_csid_enabled(finfo->flv_ops))
1861 		nlsize += nla_total_size(1) + /* SEG6_LOCAL_FLV_LCBLOCK_BITS */
1862 			  nla_total_size(1); /* SEG6_LOCAL_FLV_LCNODE_FN_BITS */
1863 
1864 	return nlsize;
1865 }
1866 
1867 struct seg6_action_param {
1868 	int (*parse)(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1869 		     struct netlink_ext_ack *extack);
1870 	int (*put)(struct sk_buff *skb, struct seg6_local_lwt *slwt);
1871 	int (*cmp)(struct seg6_local_lwt *a, struct seg6_local_lwt *b);
1872 
1873 	/* optional destroy() callback useful for releasing resources which
1874 	 * have been previously acquired in the corresponding parse()
1875 	 * function.
1876 	 */
1877 	void (*destroy)(struct seg6_local_lwt *slwt);
1878 };
1879 
1880 static struct seg6_action_param seg6_action_params[SEG6_LOCAL_MAX + 1] = {
1881 	[SEG6_LOCAL_SRH]	= { .parse = parse_nla_srh,
1882 				    .put = put_nla_srh,
1883 				    .cmp = cmp_nla_srh,
1884 				    .destroy = destroy_attr_srh },
1885 
1886 	[SEG6_LOCAL_TABLE]	= { .parse = parse_nla_table,
1887 				    .put = put_nla_table,
1888 				    .cmp = cmp_nla_table },
1889 
1890 	[SEG6_LOCAL_NH4]	= { .parse = parse_nla_nh4,
1891 				    .put = put_nla_nh4,
1892 				    .cmp = cmp_nla_nh4 },
1893 
1894 	[SEG6_LOCAL_NH6]	= { .parse = parse_nla_nh6,
1895 				    .put = put_nla_nh6,
1896 				    .cmp = cmp_nla_nh6 },
1897 
1898 	[SEG6_LOCAL_IIF]	= { .parse = parse_nla_iif,
1899 				    .put = put_nla_iif,
1900 				    .cmp = cmp_nla_iif },
1901 
1902 	[SEG6_LOCAL_OIF]	= { .parse = parse_nla_oif,
1903 				    .put = put_nla_oif,
1904 				    .cmp = cmp_nla_oif },
1905 
1906 	[SEG6_LOCAL_BPF]	= { .parse = parse_nla_bpf,
1907 				    .put = put_nla_bpf,
1908 				    .cmp = cmp_nla_bpf,
1909 				    .destroy = destroy_attr_bpf },
1910 
1911 	[SEG6_LOCAL_VRFTABLE]	= { .parse = parse_nla_vrftable,
1912 				    .put = put_nla_vrftable,
1913 				    .cmp = cmp_nla_vrftable },
1914 
1915 	[SEG6_LOCAL_COUNTERS]	= { .parse = parse_nla_counters,
1916 				    .put = put_nla_counters,
1917 				    .cmp = cmp_nla_counters,
1918 				    .destroy = destroy_attr_counters },
1919 
1920 	[SEG6_LOCAL_FLAVORS]	= { .parse = parse_nla_flavors,
1921 				    .put = put_nla_flavors,
1922 				    .cmp = cmp_nla_flavors },
1923 };
1924 
1925 /* call the destroy() callback (if available) for each set attribute in
1926  * @parsed_attrs, starting from the first attribute up to the @max_parsed
1927  * (excluded) attribute.
1928  */
1929 static void __destroy_attrs(unsigned long parsed_attrs, int max_parsed,
1930 			    struct seg6_local_lwt *slwt)
1931 {
1932 	struct seg6_action_param *param;
1933 	int i;
1934 
1935 	/* Every required seg6local attribute is identified by an ID which is
1936 	 * encoded as a flag (i.e: 1 << ID) in the 'attrs' bitmask;
1937 	 *
1938 	 * We scan the 'parsed_attrs' bitmask, starting from the first attribute
1939 	 * up to the @max_parsed (excluded) attribute.
1940 	 * For each set attribute, we retrieve the corresponding destroy()
1941 	 * callback. If the callback is not available, then we skip to the next
1942 	 * attribute; otherwise, we call the destroy() callback.
1943 	 */
1944 	for (i = SEG6_LOCAL_SRH; i < max_parsed; ++i) {
1945 		if (!(parsed_attrs & SEG6_F_ATTR(i)))
1946 			continue;
1947 
1948 		param = &seg6_action_params[i];
1949 
1950 		if (param->destroy)
1951 			param->destroy(slwt);
1952 	}
1953 }
1954 
1955 /* release all the resources that may have been acquired during parsing
1956  * operations.
1957  */
1958 static void destroy_attrs(struct seg6_local_lwt *slwt)
1959 {
1960 	unsigned long attrs = slwt->desc->attrs | slwt->parsed_optattrs;
1961 
1962 	__destroy_attrs(attrs, SEG6_LOCAL_MAX + 1, slwt);
1963 }
1964 
1965 static int parse_nla_optional_attrs(struct nlattr **attrs,
1966 				    struct seg6_local_lwt *slwt,
1967 				    struct netlink_ext_ack *extack)
1968 {
1969 	struct seg6_action_desc *desc = slwt->desc;
1970 	unsigned long parsed_optattrs = 0;
1971 	struct seg6_action_param *param;
1972 	int err, i;
1973 
1974 	for (i = SEG6_LOCAL_SRH; i < SEG6_LOCAL_MAX + 1; ++i) {
1975 		if (!(desc->optattrs & SEG6_F_ATTR(i)) || !attrs[i])
1976 			continue;
1977 
1978 		/* once here, the i-th attribute is provided by the
1979 		 * userspace AND it is identified optional as well.
1980 		 */
1981 		param = &seg6_action_params[i];
1982 
1983 		err = param->parse(attrs, slwt, extack);
1984 		if (err < 0)
1985 			goto parse_optattrs_err;
1986 
1987 		/* current attribute has been correctly parsed */
1988 		parsed_optattrs |= SEG6_F_ATTR(i);
1989 	}
1990 
1991 	/* store in the tunnel state all the optional attributed successfully
1992 	 * parsed.
1993 	 */
1994 	slwt->parsed_optattrs = parsed_optattrs;
1995 
1996 	return 0;
1997 
1998 parse_optattrs_err:
1999 	__destroy_attrs(parsed_optattrs, i, slwt);
2000 
2001 	return err;
2002 }
2003 
2004 /* call the custom constructor of the behavior during its initialization phase
2005  * and after that all its attributes have been parsed successfully.
2006  */
2007 static int
2008 seg6_local_lwtunnel_build_state(struct seg6_local_lwt *slwt, const void *cfg,
2009 				struct netlink_ext_ack *extack)
2010 {
2011 	struct seg6_action_desc *desc = slwt->desc;
2012 	struct seg6_local_lwtunnel_ops *ops;
2013 
2014 	ops = &desc->slwt_ops;
2015 	if (!ops->build_state)
2016 		return 0;
2017 
2018 	return ops->build_state(slwt, cfg, extack);
2019 }
2020 
2021 /* call the custom destructor of the behavior which is invoked before the
2022  * tunnel is going to be destroyed.
2023  */
2024 static void seg6_local_lwtunnel_destroy_state(struct seg6_local_lwt *slwt)
2025 {
2026 	struct seg6_action_desc *desc = slwt->desc;
2027 	struct seg6_local_lwtunnel_ops *ops;
2028 
2029 	ops = &desc->slwt_ops;
2030 	if (!ops->destroy_state)
2031 		return;
2032 
2033 	ops->destroy_state(slwt);
2034 }
2035 
2036 static int parse_nla_action(struct nlattr **attrs, struct seg6_local_lwt *slwt,
2037 			    struct netlink_ext_ack *extack)
2038 {
2039 	struct seg6_action_param *param;
2040 	struct seg6_action_desc *desc;
2041 	unsigned long invalid_attrs;
2042 	int i, err;
2043 
2044 	desc = __get_action_desc(slwt->action);
2045 	if (!desc)
2046 		return -EINVAL;
2047 
2048 	if (!desc->input)
2049 		return -EOPNOTSUPP;
2050 
2051 	slwt->desc = desc;
2052 	slwt->headroom += desc->static_headroom;
2053 
2054 	/* Forcing the desc->optattrs *set* and the desc->attrs *set* to be
2055 	 * disjoined, this allow us to release acquired resources by optional
2056 	 * attributes and by required attributes independently from each other
2057 	 * without any interference.
2058 	 * In other terms, we are sure that we do not release some the acquired
2059 	 * resources twice.
2060 	 *
2061 	 * Note that if an attribute is configured both as required and as
2062 	 * optional, it means that the user has messed something up in the
2063 	 * seg6_action_table. Therefore, this check is required for SRv6
2064 	 * behaviors to work properly.
2065 	 */
2066 	invalid_attrs = desc->attrs & desc->optattrs;
2067 	if (invalid_attrs) {
2068 		WARN_ONCE(1,
2069 			  "An attribute cannot be both required AND optional");
2070 		return -EINVAL;
2071 	}
2072 
2073 	/* parse the required attributes */
2074 	for (i = SEG6_LOCAL_SRH; i < SEG6_LOCAL_MAX + 1; i++) {
2075 		if (desc->attrs & SEG6_F_ATTR(i)) {
2076 			if (!attrs[i])
2077 				return -EINVAL;
2078 
2079 			param = &seg6_action_params[i];
2080 
2081 			err = param->parse(attrs, slwt, extack);
2082 			if (err < 0)
2083 				goto parse_attrs_err;
2084 		}
2085 	}
2086 
2087 	/* parse the optional attributes, if any */
2088 	err = parse_nla_optional_attrs(attrs, slwt, extack);
2089 	if (err < 0)
2090 		goto parse_attrs_err;
2091 
2092 	return 0;
2093 
2094 parse_attrs_err:
2095 	/* release any resource that may have been acquired during the i-1
2096 	 * parse() operations.
2097 	 */
2098 	__destroy_attrs(desc->attrs, i, slwt);
2099 
2100 	return err;
2101 }
2102 
2103 static int seg6_local_build_state(struct net *net, struct nlattr *nla,
2104 				  unsigned int family, const void *cfg,
2105 				  struct lwtunnel_state **ts,
2106 				  struct netlink_ext_ack *extack)
2107 {
2108 	struct nlattr *tb[SEG6_LOCAL_MAX + 1];
2109 	struct lwtunnel_state *newts;
2110 	struct seg6_local_lwt *slwt;
2111 	int err;
2112 
2113 	if (family != AF_INET6)
2114 		return -EINVAL;
2115 
2116 	err = nla_parse_nested_deprecated(tb, SEG6_LOCAL_MAX, nla,
2117 					  seg6_local_policy, extack);
2118 
2119 	if (err < 0)
2120 		return err;
2121 
2122 	if (!tb[SEG6_LOCAL_ACTION])
2123 		return -EINVAL;
2124 
2125 	newts = lwtunnel_state_alloc(sizeof(*slwt));
2126 	if (!newts)
2127 		return -ENOMEM;
2128 
2129 	slwt = seg6_local_lwtunnel(newts);
2130 	slwt->action = nla_get_u32(tb[SEG6_LOCAL_ACTION]);
2131 
2132 	err = parse_nla_action(tb, slwt, extack);
2133 	if (err < 0)
2134 		goto out_free;
2135 
2136 	err = seg6_local_lwtunnel_build_state(slwt, cfg, extack);
2137 	if (err < 0)
2138 		goto out_destroy_attrs;
2139 
2140 	newts->type = LWTUNNEL_ENCAP_SEG6_LOCAL;
2141 	newts->flags = LWTUNNEL_STATE_INPUT_REDIRECT;
2142 	newts->headroom = slwt->headroom;
2143 
2144 	*ts = newts;
2145 
2146 	return 0;
2147 
2148 out_destroy_attrs:
2149 	destroy_attrs(slwt);
2150 out_free:
2151 	kfree(newts);
2152 	return err;
2153 }
2154 
2155 static void seg6_local_destroy_state(struct lwtunnel_state *lwt)
2156 {
2157 	struct seg6_local_lwt *slwt = seg6_local_lwtunnel(lwt);
2158 
2159 	seg6_local_lwtunnel_destroy_state(slwt);
2160 
2161 	destroy_attrs(slwt);
2162 
2163 	return;
2164 }
2165 
2166 static int seg6_local_fill_encap(struct sk_buff *skb,
2167 				 struct lwtunnel_state *lwt)
2168 {
2169 	struct seg6_local_lwt *slwt = seg6_local_lwtunnel(lwt);
2170 	struct seg6_action_param *param;
2171 	unsigned long attrs;
2172 	int i, err;
2173 
2174 	if (nla_put_u32(skb, SEG6_LOCAL_ACTION, slwt->action))
2175 		return -EMSGSIZE;
2176 
2177 	attrs = slwt->desc->attrs | slwt->parsed_optattrs;
2178 
2179 	for (i = SEG6_LOCAL_SRH; i < SEG6_LOCAL_MAX + 1; i++) {
2180 		if (attrs & SEG6_F_ATTR(i)) {
2181 			param = &seg6_action_params[i];
2182 			err = param->put(skb, slwt);
2183 			if (err < 0)
2184 				return err;
2185 		}
2186 	}
2187 
2188 	return 0;
2189 }
2190 
2191 static int seg6_local_get_encap_size(struct lwtunnel_state *lwt)
2192 {
2193 	struct seg6_local_lwt *slwt = seg6_local_lwtunnel(lwt);
2194 	unsigned long attrs;
2195 	int nlsize;
2196 
2197 	nlsize = nla_total_size(4); /* action */
2198 
2199 	attrs = slwt->desc->attrs | slwt->parsed_optattrs;
2200 
2201 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_SRH))
2202 		nlsize += nla_total_size((slwt->srh->hdrlen + 1) << 3);
2203 
2204 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_TABLE))
2205 		nlsize += nla_total_size(4);
2206 
2207 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_NH4))
2208 		nlsize += nla_total_size(4);
2209 
2210 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_NH6))
2211 		nlsize += nla_total_size(16);
2212 
2213 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_IIF))
2214 		nlsize += nla_total_size(4);
2215 
2216 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_OIF))
2217 		nlsize += nla_total_size(4);
2218 
2219 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_BPF))
2220 		nlsize += nla_total_size(sizeof(struct nlattr)) +
2221 		       nla_total_size(MAX_PROG_NAME) +
2222 		       nla_total_size(4);
2223 
2224 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_VRFTABLE))
2225 		nlsize += nla_total_size(4);
2226 
2227 	if (attrs & SEG6_F_LOCAL_COUNTERS)
2228 		nlsize += nla_total_size(0) + /* nest SEG6_LOCAL_COUNTERS */
2229 			  /* SEG6_LOCAL_CNT_PACKETS */
2230 			  nla_total_size_64bit(sizeof(__u64)) +
2231 			  /* SEG6_LOCAL_CNT_BYTES */
2232 			  nla_total_size_64bit(sizeof(__u64)) +
2233 			  /* SEG6_LOCAL_CNT_ERRORS */
2234 			  nla_total_size_64bit(sizeof(__u64));
2235 
2236 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_FLAVORS))
2237 		nlsize += encap_size_flavors(slwt);
2238 
2239 	return nlsize;
2240 }
2241 
2242 static int seg6_local_cmp_encap(struct lwtunnel_state *a,
2243 				struct lwtunnel_state *b)
2244 {
2245 	struct seg6_local_lwt *slwt_a, *slwt_b;
2246 	struct seg6_action_param *param;
2247 	unsigned long attrs_a, attrs_b;
2248 	int i;
2249 
2250 	slwt_a = seg6_local_lwtunnel(a);
2251 	slwt_b = seg6_local_lwtunnel(b);
2252 
2253 	if (slwt_a->action != slwt_b->action)
2254 		return 1;
2255 
2256 	attrs_a = slwt_a->desc->attrs | slwt_a->parsed_optattrs;
2257 	attrs_b = slwt_b->desc->attrs | slwt_b->parsed_optattrs;
2258 
2259 	if (attrs_a != attrs_b)
2260 		return 1;
2261 
2262 	for (i = SEG6_LOCAL_SRH; i < SEG6_LOCAL_MAX + 1; i++) {
2263 		if (attrs_a & SEG6_F_ATTR(i)) {
2264 			param = &seg6_action_params[i];
2265 			if (param->cmp(slwt_a, slwt_b))
2266 				return 1;
2267 		}
2268 	}
2269 
2270 	return 0;
2271 }
2272 
2273 static const struct lwtunnel_encap_ops seg6_local_ops = {
2274 	.build_state	= seg6_local_build_state,
2275 	.destroy_state	= seg6_local_destroy_state,
2276 	.input		= seg6_local_input,
2277 	.fill_encap	= seg6_local_fill_encap,
2278 	.get_encap_size	= seg6_local_get_encap_size,
2279 	.cmp_encap	= seg6_local_cmp_encap,
2280 	.owner		= THIS_MODULE,
2281 };
2282 
2283 int __init seg6_local_init(void)
2284 {
2285 	/* If the max total number of defined attributes is reached, then your
2286 	 * kernel build stops here.
2287 	 *
2288 	 * This check is required to avoid arithmetic overflows when processing
2289 	 * behavior attributes and the maximum number of defined attributes
2290 	 * exceeds the allowed value.
2291 	 */
2292 	BUILD_BUG_ON(SEG6_LOCAL_MAX + 1 > BITS_PER_TYPE(unsigned long));
2293 
2294 	/* If the default NEXT-C-SID Locator-Block/Node Function lengths (in
2295 	 * bits) have been changed with invalid values, kernel build stops
2296 	 * here.
2297 	 */
2298 	BUILD_BUG_ON(next_csid_chk_cntr_bits(SEG6_LOCAL_LCBLOCK_DBITS,
2299 					     SEG6_LOCAL_LCNODE_FN_DBITS));
2300 	BUILD_BUG_ON(next_csid_chk_lcblock_bits(SEG6_LOCAL_LCBLOCK_DBITS));
2301 	BUILD_BUG_ON(next_csid_chk_lcnode_fn_bits(SEG6_LOCAL_LCNODE_FN_DBITS));
2302 
2303 	return lwtunnel_encap_add_ops(&seg6_local_ops,
2304 				      LWTUNNEL_ENCAP_SEG6_LOCAL);
2305 }
2306 
2307 void seg6_local_exit(void)
2308 {
2309 	lwtunnel_encap_del_ops(&seg6_local_ops, LWTUNNEL_ENCAP_SEG6_LOCAL);
2310 }
2311