xref: /openbmc/linux/net/ipv6/seg6_local.c (revision 848f3c0d47694924536e2894cb349613201321c6)
12874c5fdSThomas Gleixner // SPDX-License-Identifier: GPL-2.0-or-later
2d1df6fd8SDavid Lebrun /*
3d1df6fd8SDavid Lebrun  *  SR-IPv6 implementation
4d1df6fd8SDavid Lebrun  *
5004d4b27SMathieu Xhonneux  *  Authors:
6d1df6fd8SDavid Lebrun  *  David Lebrun <david.lebrun@uclouvain.be>
7004d4b27SMathieu Xhonneux  *  eBPF support: Mathieu Xhonneux <m.xhonneux@gmail.com>
8d1df6fd8SDavid Lebrun  */
9d1df6fd8SDavid Lebrun 
10b6459415SJakub Kicinski #include <linux/filter.h>
11d1df6fd8SDavid Lebrun #include <linux/types.h>
12d1df6fd8SDavid Lebrun #include <linux/skbuff.h>
13d1df6fd8SDavid Lebrun #include <linux/net.h>
14d1df6fd8SDavid Lebrun #include <linux/module.h>
15d1df6fd8SDavid Lebrun #include <net/ip.h>
16d1df6fd8SDavid Lebrun #include <net/lwtunnel.h>
17d1df6fd8SDavid Lebrun #include <net/netevent.h>
18d1df6fd8SDavid Lebrun #include <net/netns/generic.h>
19d1df6fd8SDavid Lebrun #include <net/ip6_fib.h>
20d1df6fd8SDavid Lebrun #include <net/route.h>
21d1df6fd8SDavid Lebrun #include <net/seg6.h>
22d1df6fd8SDavid Lebrun #include <linux/seg6.h>
23d1df6fd8SDavid Lebrun #include <linux/seg6_local.h>
24d1df6fd8SDavid Lebrun #include <net/addrconf.h>
25d1df6fd8SDavid Lebrun #include <net/ip6_route.h>
26d1df6fd8SDavid Lebrun #include <net/dst_cache.h>
2762ebaeaeSYuki Taguchi #include <net/ip_tunnels.h>
28d1df6fd8SDavid Lebrun #ifdef CONFIG_IPV6_SEG6_HMAC
29d1df6fd8SDavid Lebrun #include <net/seg6_hmac.h>
30d1df6fd8SDavid Lebrun #endif
311c1e761eSMathieu Xhonneux #include <net/seg6_local.h>
32891ef8ddSDavid Lebrun #include <linux/etherdevice.h>
33004d4b27SMathieu Xhonneux #include <linux/bpf.h>
347a3f5b0dSRyoga Saito #include <linux/netfilter.h>
35d1df6fd8SDavid Lebrun 
36300a0fd8SAndrea Mayer #define SEG6_F_ATTR(i)		BIT(i)
37300a0fd8SAndrea Mayer 
38d1df6fd8SDavid Lebrun struct seg6_local_lwt;
39d1df6fd8SDavid Lebrun 
40cfdf64a0SAndrea Mayer /* callbacks used for customizing the creation and destruction of a behavior */
41cfdf64a0SAndrea Mayer struct seg6_local_lwtunnel_ops {
42cfdf64a0SAndrea Mayer 	int (*build_state)(struct seg6_local_lwt *slwt, const void *cfg,
43cfdf64a0SAndrea Mayer 			   struct netlink_ext_ack *extack);
44cfdf64a0SAndrea Mayer 	void (*destroy_state)(struct seg6_local_lwt *slwt);
45cfdf64a0SAndrea Mayer };
46cfdf64a0SAndrea Mayer 
47d1df6fd8SDavid Lebrun struct seg6_action_desc {
48d1df6fd8SDavid Lebrun 	int action;
49d1df6fd8SDavid Lebrun 	unsigned long attrs;
500a3021f1SAndrea Mayer 
510a3021f1SAndrea Mayer 	/* The optattrs field is used for specifying all the optional
520a3021f1SAndrea Mayer 	 * attributes supported by a specific behavior.
530a3021f1SAndrea Mayer 	 * It means that if one of these attributes is not provided in the
540a3021f1SAndrea Mayer 	 * netlink message during the behavior creation, no errors will be
550a3021f1SAndrea Mayer 	 * returned to the userspace.
560a3021f1SAndrea Mayer 	 *
570a3021f1SAndrea Mayer 	 * Each attribute can be only of two types (mutually exclusive):
580a3021f1SAndrea Mayer 	 * 1) required or 2) optional.
590a3021f1SAndrea Mayer 	 * Every user MUST obey to this rule! If you set an attribute as
600a3021f1SAndrea Mayer 	 * required the same attribute CANNOT be set as optional and vice
610a3021f1SAndrea Mayer 	 * versa.
620a3021f1SAndrea Mayer 	 */
630a3021f1SAndrea Mayer 	unsigned long optattrs;
640a3021f1SAndrea Mayer 
65d1df6fd8SDavid Lebrun 	int (*input)(struct sk_buff *skb, struct seg6_local_lwt *slwt);
66d1df6fd8SDavid Lebrun 	int static_headroom;
67cfdf64a0SAndrea Mayer 
68cfdf64a0SAndrea Mayer 	struct seg6_local_lwtunnel_ops slwt_ops;
69d1df6fd8SDavid Lebrun };
70d1df6fd8SDavid Lebrun 
71004d4b27SMathieu Xhonneux struct bpf_lwt_prog {
72004d4b27SMathieu Xhonneux 	struct bpf_prog *prog;
73004d4b27SMathieu Xhonneux 	char *name;
74004d4b27SMathieu Xhonneux };
75004d4b27SMathieu Xhonneux 
76*848f3c0dSAndrea Mayer /* default length values (expressed in bits) for both Locator-Block and
77*848f3c0dSAndrea Mayer  * Locator-Node Function.
78*848f3c0dSAndrea Mayer  *
79*848f3c0dSAndrea Mayer  * Both SEG6_LOCAL_LCBLOCK_DBITS and SEG6_LOCAL_LCNODE_FN_DBITS *must* be:
80*848f3c0dSAndrea Mayer  *    i) greater than 0;
81*848f3c0dSAndrea Mayer  *   ii) evenly divisible by 8. In other terms, the lengths of the
82*848f3c0dSAndrea Mayer  *	 Locator-Block and Locator-Node Function must be byte-aligned (we can
83*848f3c0dSAndrea Mayer  *	 relax this constraint in the future if really needed).
84*848f3c0dSAndrea Mayer  *
85*848f3c0dSAndrea Mayer  * Moreover, a third condition must hold:
86*848f3c0dSAndrea Mayer  *  iii) SEG6_LOCAL_LCBLOCK_DBITS + SEG6_LOCAL_LCNODE_FN_DBITS <= 128.
87*848f3c0dSAndrea Mayer  *
88*848f3c0dSAndrea Mayer  * The correctness of SEG6_LOCAL_LCBLOCK_DBITS and SEG6_LOCAL_LCNODE_FN_DBITS
89*848f3c0dSAndrea Mayer  * values are checked during the kernel compilation. If the compilation stops,
90*848f3c0dSAndrea Mayer  * check the value of these parameters to see if they meet conditions (i), (ii)
91*848f3c0dSAndrea Mayer  * and (iii).
92*848f3c0dSAndrea Mayer  */
93*848f3c0dSAndrea Mayer #define SEG6_LOCAL_LCBLOCK_DBITS	32
94*848f3c0dSAndrea Mayer #define SEG6_LOCAL_LCNODE_FN_DBITS	16
95*848f3c0dSAndrea Mayer 
96*848f3c0dSAndrea Mayer /* The following next_csid_chk_{cntr,lcblock,lcblock_fn}_bits macros can be
97*848f3c0dSAndrea Mayer  * used directly to check whether the lengths (in bits) of Locator-Block and
98*848f3c0dSAndrea Mayer  * Locator-Node Function are valid according to (i), (ii), (iii).
99*848f3c0dSAndrea Mayer  */
100*848f3c0dSAndrea Mayer #define next_csid_chk_cntr_bits(blen, flen)		\
101*848f3c0dSAndrea Mayer 	((blen) + (flen) > 128)
102*848f3c0dSAndrea Mayer 
103*848f3c0dSAndrea Mayer #define next_csid_chk_lcblock_bits(blen)		\
104*848f3c0dSAndrea Mayer ({							\
105*848f3c0dSAndrea Mayer 	typeof(blen) __tmp = blen;			\
106*848f3c0dSAndrea Mayer 	(!__tmp || __tmp > 120 || (__tmp & 0x07));	\
107*848f3c0dSAndrea Mayer })
108*848f3c0dSAndrea Mayer 
109*848f3c0dSAndrea Mayer #define next_csid_chk_lcnode_fn_bits(flen)		\
110*848f3c0dSAndrea Mayer 	next_csid_chk_lcblock_bits(flen)
111*848f3c0dSAndrea Mayer 
112*848f3c0dSAndrea Mayer /* Supported Flavor operations are reported in this bitmask */
113*848f3c0dSAndrea Mayer #define SEG6_LOCAL_FLV_SUPP_OPS	(BIT(SEG6_LOCAL_FLV_OP_NEXT_CSID))
114*848f3c0dSAndrea Mayer 
115*848f3c0dSAndrea Mayer struct seg6_flavors_info {
116*848f3c0dSAndrea Mayer 	/* Flavor operations */
117*848f3c0dSAndrea Mayer 	__u32 flv_ops;
118*848f3c0dSAndrea Mayer 
119*848f3c0dSAndrea Mayer 	/* Locator-Block length, expressed in bits */
120*848f3c0dSAndrea Mayer 	__u8 lcblock_bits;
121*848f3c0dSAndrea Mayer 	/* Locator-Node Function length, expressed in bits*/
122*848f3c0dSAndrea Mayer 	__u8 lcnode_func_bits;
123*848f3c0dSAndrea Mayer };
124*848f3c0dSAndrea Mayer 
125664d6f86SAndrea Mayer enum seg6_end_dt_mode {
126664d6f86SAndrea Mayer 	DT_INVALID_MODE	= -EINVAL,
127664d6f86SAndrea Mayer 	DT_LEGACY_MODE	= 0,
128664d6f86SAndrea Mayer 	DT_VRF_MODE	= 1,
129664d6f86SAndrea Mayer };
130664d6f86SAndrea Mayer 
131664d6f86SAndrea Mayer struct seg6_end_dt_info {
132664d6f86SAndrea Mayer 	enum seg6_end_dt_mode mode;
133664d6f86SAndrea Mayer 
134664d6f86SAndrea Mayer 	struct net *net;
135664d6f86SAndrea Mayer 	/* VRF device associated to the routing table used by the SRv6
136664d6f86SAndrea Mayer 	 * End.DT4/DT6 behavior for routing IPv4/IPv6 packets.
137664d6f86SAndrea Mayer 	 */
138664d6f86SAndrea Mayer 	int vrf_ifindex;
139664d6f86SAndrea Mayer 	int vrf_table;
140664d6f86SAndrea Mayer 
1418b532109SAndrea Mayer 	/* tunneled packet family (IPv4 or IPv6).
1428b532109SAndrea Mayer 	 * Protocol and header length are inferred from family.
1438b532109SAndrea Mayer 	 */
144664d6f86SAndrea Mayer 	u16 family;
145664d6f86SAndrea Mayer };
146664d6f86SAndrea Mayer 
14794604548SAndrea Mayer struct pcpu_seg6_local_counters {
14894604548SAndrea Mayer 	u64_stats_t packets;
14994604548SAndrea Mayer 	u64_stats_t bytes;
15094604548SAndrea Mayer 	u64_stats_t errors;
15194604548SAndrea Mayer 
15294604548SAndrea Mayer 	struct u64_stats_sync syncp;
15394604548SAndrea Mayer };
15494604548SAndrea Mayer 
15594604548SAndrea Mayer /* This struct groups all the SRv6 Behavior counters supported so far.
15694604548SAndrea Mayer  *
15794604548SAndrea Mayer  * put_nla_counters() makes use of this data structure to collect all counter
15894604548SAndrea Mayer  * values after the per-CPU counter evaluation has been performed.
15994604548SAndrea Mayer  * Finally, each counter value (in seg6_local_counters) is stored in the
16094604548SAndrea Mayer  * corresponding netlink attribute and sent to user space.
16194604548SAndrea Mayer  *
16294604548SAndrea Mayer  * NB: we don't want to expose this structure to user space!
16394604548SAndrea Mayer  */
16494604548SAndrea Mayer struct seg6_local_counters {
16594604548SAndrea Mayer 	__u64 packets;
16694604548SAndrea Mayer 	__u64 bytes;
16794604548SAndrea Mayer 	__u64 errors;
16894604548SAndrea Mayer };
16994604548SAndrea Mayer 
17094604548SAndrea Mayer #define seg6_local_alloc_pcpu_counters(__gfp)				\
17194604548SAndrea Mayer 	__netdev_alloc_pcpu_stats(struct pcpu_seg6_local_counters,	\
17294604548SAndrea Mayer 				  ((__gfp) | __GFP_ZERO))
17394604548SAndrea Mayer 
17494604548SAndrea Mayer #define SEG6_F_LOCAL_COUNTERS	SEG6_F_ATTR(SEG6_LOCAL_COUNTERS)
17594604548SAndrea Mayer 
176d1df6fd8SDavid Lebrun struct seg6_local_lwt {
177d1df6fd8SDavid Lebrun 	int action;
178d1df6fd8SDavid Lebrun 	struct ipv6_sr_hdr *srh;
179d1df6fd8SDavid Lebrun 	int table;
180d1df6fd8SDavid Lebrun 	struct in_addr nh4;
181d1df6fd8SDavid Lebrun 	struct in6_addr nh6;
182d1df6fd8SDavid Lebrun 	int iif;
183d1df6fd8SDavid Lebrun 	int oif;
184004d4b27SMathieu Xhonneux 	struct bpf_lwt_prog bpf;
185664d6f86SAndrea Mayer #ifdef CONFIG_NET_L3_MASTER_DEV
186664d6f86SAndrea Mayer 	struct seg6_end_dt_info dt_info;
187664d6f86SAndrea Mayer #endif
188*848f3c0dSAndrea Mayer 	struct seg6_flavors_info flv_info;
189*848f3c0dSAndrea Mayer 
19094604548SAndrea Mayer 	struct pcpu_seg6_local_counters __percpu *pcpu_counters;
191d1df6fd8SDavid Lebrun 
192d1df6fd8SDavid Lebrun 	int headroom;
193d1df6fd8SDavid Lebrun 	struct seg6_action_desc *desc;
1940a3021f1SAndrea Mayer 	/* unlike the required attrs, we have to track the optional attributes
1950a3021f1SAndrea Mayer 	 * that have been effectively parsed.
1960a3021f1SAndrea Mayer 	 */
1970a3021f1SAndrea Mayer 	unsigned long parsed_optattrs;
198d1df6fd8SDavid Lebrun };
199d1df6fd8SDavid Lebrun 
200d1df6fd8SDavid Lebrun static struct seg6_local_lwt *seg6_local_lwtunnel(struct lwtunnel_state *lwt)
201d1df6fd8SDavid Lebrun {
202d1df6fd8SDavid Lebrun 	return (struct seg6_local_lwt *)lwt->data;
203d1df6fd8SDavid Lebrun }
204d1df6fd8SDavid Lebrun 
205140f04c3SDavid Lebrun static struct ipv6_sr_hdr *get_and_validate_srh(struct sk_buff *skb)
206140f04c3SDavid Lebrun {
207140f04c3SDavid Lebrun 	struct ipv6_sr_hdr *srh;
208140f04c3SDavid Lebrun 
209fa55a7d7SAndrew Lunn 	srh = seg6_get_srh(skb, IP6_FH_F_SKIP_RH);
210140f04c3SDavid Lebrun 	if (!srh)
211140f04c3SDavid Lebrun 		return NULL;
212140f04c3SDavid Lebrun 
213140f04c3SDavid Lebrun #ifdef CONFIG_IPV6_SEG6_HMAC
214140f04c3SDavid Lebrun 	if (!seg6_hmac_validate_skb(skb))
215140f04c3SDavid Lebrun 		return NULL;
216140f04c3SDavid Lebrun #endif
217140f04c3SDavid Lebrun 
218140f04c3SDavid Lebrun 	return srh;
219140f04c3SDavid Lebrun }
220140f04c3SDavid Lebrun 
221d7a669ddSDavid Lebrun static bool decap_and_validate(struct sk_buff *skb, int proto)
222d7a669ddSDavid Lebrun {
223d7a669ddSDavid Lebrun 	struct ipv6_sr_hdr *srh;
224d7a669ddSDavid Lebrun 	unsigned int off = 0;
225d7a669ddSDavid Lebrun 
226fa55a7d7SAndrew Lunn 	srh = seg6_get_srh(skb, 0);
227d7a669ddSDavid Lebrun 	if (srh && srh->segments_left > 0)
228d7a669ddSDavid Lebrun 		return false;
229d7a669ddSDavid Lebrun 
230d7a669ddSDavid Lebrun #ifdef CONFIG_IPV6_SEG6_HMAC
231d7a669ddSDavid Lebrun 	if (srh && !seg6_hmac_validate_skb(skb))
232d7a669ddSDavid Lebrun 		return false;
233d7a669ddSDavid Lebrun #endif
234d7a669ddSDavid Lebrun 
235d7a669ddSDavid Lebrun 	if (ipv6_find_hdr(skb, &off, proto, NULL, NULL) < 0)
236d7a669ddSDavid Lebrun 		return false;
237d7a669ddSDavid Lebrun 
238d7a669ddSDavid Lebrun 	if (!pskb_pull(skb, off))
239d7a669ddSDavid Lebrun 		return false;
240d7a669ddSDavid Lebrun 
241d7a669ddSDavid Lebrun 	skb_postpull_rcsum(skb, skb_network_header(skb), off);
242d7a669ddSDavid Lebrun 
243d7a669ddSDavid Lebrun 	skb_reset_network_header(skb);
244d7a669ddSDavid Lebrun 	skb_reset_transport_header(skb);
24562ebaeaeSYuki Taguchi 	if (iptunnel_pull_offloads(skb))
24662ebaeaeSYuki Taguchi 		return false;
247d7a669ddSDavid Lebrun 
248d7a669ddSDavid Lebrun 	return true;
249d7a669ddSDavid Lebrun }
250d7a669ddSDavid Lebrun 
251d7a669ddSDavid Lebrun static void advance_nextseg(struct ipv6_sr_hdr *srh, struct in6_addr *daddr)
252d7a669ddSDavid Lebrun {
253d7a669ddSDavid Lebrun 	struct in6_addr *addr;
254d7a669ddSDavid Lebrun 
255d7a669ddSDavid Lebrun 	srh->segments_left--;
256d7a669ddSDavid Lebrun 	addr = srh->segments + srh->segments_left;
257d7a669ddSDavid Lebrun 	*daddr = *addr;
258d7a669ddSDavid Lebrun }
259d7a669ddSDavid Lebrun 
260fd1fef0cSAndrea Mayer static int
261fd1fef0cSAndrea Mayer seg6_lookup_any_nexthop(struct sk_buff *skb, struct in6_addr *nhaddr,
262fd1fef0cSAndrea Mayer 			u32 tbl_id, bool local_delivery)
263d7a669ddSDavid Lebrun {
264d7a669ddSDavid Lebrun 	struct net *net = dev_net(skb->dev);
265d7a669ddSDavid Lebrun 	struct ipv6hdr *hdr = ipv6_hdr(skb);
266d7a669ddSDavid Lebrun 	int flags = RT6_LOOKUP_F_HAS_SADDR;
267d7a669ddSDavid Lebrun 	struct dst_entry *dst = NULL;
268d7a669ddSDavid Lebrun 	struct rt6_info *rt;
269d7a669ddSDavid Lebrun 	struct flowi6 fl6;
270fd1fef0cSAndrea Mayer 	int dev_flags = 0;
271d7a669ddSDavid Lebrun 
272a3bd2102SAndrea Mayer 	memset(&fl6, 0, sizeof(fl6));
273d7a669ddSDavid Lebrun 	fl6.flowi6_iif = skb->dev->ifindex;
274d7a669ddSDavid Lebrun 	fl6.daddr = nhaddr ? *nhaddr : hdr->daddr;
275d7a669ddSDavid Lebrun 	fl6.saddr = hdr->saddr;
276d7a669ddSDavid Lebrun 	fl6.flowlabel = ip6_flowinfo(hdr);
277d7a669ddSDavid Lebrun 	fl6.flowi6_mark = skb->mark;
278d7a669ddSDavid Lebrun 	fl6.flowi6_proto = hdr->nexthdr;
279d7a669ddSDavid Lebrun 
280d7a669ddSDavid Lebrun 	if (nhaddr)
281d7a669ddSDavid Lebrun 		fl6.flowi6_flags = FLOWI_FLAG_KNOWN_NH;
282d7a669ddSDavid Lebrun 
283d7a669ddSDavid Lebrun 	if (!tbl_id) {
284b75cc8f9SDavid Ahern 		dst = ip6_route_input_lookup(net, skb->dev, &fl6, skb, flags);
285d7a669ddSDavid Lebrun 	} else {
286d7a669ddSDavid Lebrun 		struct fib6_table *table;
287d7a669ddSDavid Lebrun 
288d7a669ddSDavid Lebrun 		table = fib6_get_table(net, tbl_id);
289d7a669ddSDavid Lebrun 		if (!table)
290d7a669ddSDavid Lebrun 			goto out;
291d7a669ddSDavid Lebrun 
292b75cc8f9SDavid Ahern 		rt = ip6_pol_route(net, table, 0, &fl6, skb, flags);
293d7a669ddSDavid Lebrun 		dst = &rt->dst;
294d7a669ddSDavid Lebrun 	}
295d7a669ddSDavid Lebrun 
296fd1fef0cSAndrea Mayer 	/* we want to discard traffic destined for local packet processing,
297fd1fef0cSAndrea Mayer 	 * if @local_delivery is set to false.
298fd1fef0cSAndrea Mayer 	 */
299fd1fef0cSAndrea Mayer 	if (!local_delivery)
300fd1fef0cSAndrea Mayer 		dev_flags |= IFF_LOOPBACK;
301fd1fef0cSAndrea Mayer 
302fd1fef0cSAndrea Mayer 	if (dst && (dst->dev->flags & dev_flags) && !dst->error) {
303d7a669ddSDavid Lebrun 		dst_release(dst);
304d7a669ddSDavid Lebrun 		dst = NULL;
305d7a669ddSDavid Lebrun 	}
306d7a669ddSDavid Lebrun 
307d7a669ddSDavid Lebrun out:
308d7a669ddSDavid Lebrun 	if (!dst) {
309d7a669ddSDavid Lebrun 		rt = net->ipv6.ip6_blk_hole_entry;
310d7a669ddSDavid Lebrun 		dst = &rt->dst;
311d7a669ddSDavid Lebrun 		dst_hold(dst);
312d7a669ddSDavid Lebrun 	}
313d7a669ddSDavid Lebrun 
314d7a669ddSDavid Lebrun 	skb_dst_drop(skb);
315d7a669ddSDavid Lebrun 	skb_dst_set(skb, dst);
3161c1e761eSMathieu Xhonneux 	return dst->error;
317d7a669ddSDavid Lebrun }
318d7a669ddSDavid Lebrun 
319fd1fef0cSAndrea Mayer int seg6_lookup_nexthop(struct sk_buff *skb,
320fd1fef0cSAndrea Mayer 			struct in6_addr *nhaddr, u32 tbl_id)
321fd1fef0cSAndrea Mayer {
322fd1fef0cSAndrea Mayer 	return seg6_lookup_any_nexthop(skb, nhaddr, tbl_id, false);
323fd1fef0cSAndrea Mayer }
324fd1fef0cSAndrea Mayer 
325*848f3c0dSAndrea Mayer static __u8 seg6_flv_lcblock_octects(const struct seg6_flavors_info *finfo)
326*848f3c0dSAndrea Mayer {
327*848f3c0dSAndrea Mayer 	return finfo->lcblock_bits >> 3;
328*848f3c0dSAndrea Mayer }
329*848f3c0dSAndrea Mayer 
330*848f3c0dSAndrea Mayer static __u8 seg6_flv_lcnode_func_octects(const struct seg6_flavors_info *finfo)
331*848f3c0dSAndrea Mayer {
332*848f3c0dSAndrea Mayer 	return finfo->lcnode_func_bits >> 3;
333*848f3c0dSAndrea Mayer }
334*848f3c0dSAndrea Mayer 
335*848f3c0dSAndrea Mayer static bool seg6_next_csid_is_arg_zero(const struct in6_addr *addr,
336*848f3c0dSAndrea Mayer 				       const struct seg6_flavors_info *finfo)
337*848f3c0dSAndrea Mayer {
338*848f3c0dSAndrea Mayer 	__u8 fnc_octects = seg6_flv_lcnode_func_octects(finfo);
339*848f3c0dSAndrea Mayer 	__u8 blk_octects = seg6_flv_lcblock_octects(finfo);
340*848f3c0dSAndrea Mayer 	__u8 arg_octects;
341*848f3c0dSAndrea Mayer 	int i;
342*848f3c0dSAndrea Mayer 
343*848f3c0dSAndrea Mayer 	arg_octects = 16 - blk_octects - fnc_octects;
344*848f3c0dSAndrea Mayer 	for (i = 0; i < arg_octects; ++i) {
345*848f3c0dSAndrea Mayer 		if (addr->s6_addr[blk_octects + fnc_octects + i] != 0x00)
346*848f3c0dSAndrea Mayer 			return false;
347*848f3c0dSAndrea Mayer 	}
348*848f3c0dSAndrea Mayer 
349*848f3c0dSAndrea Mayer 	return true;
350*848f3c0dSAndrea Mayer }
351*848f3c0dSAndrea Mayer 
352*848f3c0dSAndrea Mayer /* assume that DA.Argument length > 0 */
353*848f3c0dSAndrea Mayer static void seg6_next_csid_advance_arg(struct in6_addr *addr,
354*848f3c0dSAndrea Mayer 				       const struct seg6_flavors_info *finfo)
355*848f3c0dSAndrea Mayer {
356*848f3c0dSAndrea Mayer 	__u8 fnc_octects = seg6_flv_lcnode_func_octects(finfo);
357*848f3c0dSAndrea Mayer 	__u8 blk_octects = seg6_flv_lcblock_octects(finfo);
358*848f3c0dSAndrea Mayer 
359*848f3c0dSAndrea Mayer 	/* advance DA.Argument */
360*848f3c0dSAndrea Mayer 	memmove(&addr->s6_addr[blk_octects],
361*848f3c0dSAndrea Mayer 		&addr->s6_addr[blk_octects + fnc_octects],
362*848f3c0dSAndrea Mayer 		16 - blk_octects - fnc_octects);
363*848f3c0dSAndrea Mayer 
364*848f3c0dSAndrea Mayer 	memset(&addr->s6_addr[16 - fnc_octects], 0x00, fnc_octects);
365*848f3c0dSAndrea Mayer }
366*848f3c0dSAndrea Mayer 
367*848f3c0dSAndrea Mayer static int input_action_end_core(struct sk_buff *skb,
368*848f3c0dSAndrea Mayer 				 struct seg6_local_lwt *slwt)
369140f04c3SDavid Lebrun {
370140f04c3SDavid Lebrun 	struct ipv6_sr_hdr *srh;
371140f04c3SDavid Lebrun 
372140f04c3SDavid Lebrun 	srh = get_and_validate_srh(skb);
373140f04c3SDavid Lebrun 	if (!srh)
374140f04c3SDavid Lebrun 		goto drop;
375140f04c3SDavid Lebrun 
376d7a669ddSDavid Lebrun 	advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
377140f04c3SDavid Lebrun 
3781c1e761eSMathieu Xhonneux 	seg6_lookup_nexthop(skb, NULL, 0);
379140f04c3SDavid Lebrun 
380140f04c3SDavid Lebrun 	return dst_input(skb);
381140f04c3SDavid Lebrun 
382140f04c3SDavid Lebrun drop:
383140f04c3SDavid Lebrun 	kfree_skb(skb);
384140f04c3SDavid Lebrun 	return -EINVAL;
385140f04c3SDavid Lebrun }
386140f04c3SDavid Lebrun 
387*848f3c0dSAndrea Mayer static int end_next_csid_core(struct sk_buff *skb, struct seg6_local_lwt *slwt)
388*848f3c0dSAndrea Mayer {
389*848f3c0dSAndrea Mayer 	const struct seg6_flavors_info *finfo = &slwt->flv_info;
390*848f3c0dSAndrea Mayer 	struct in6_addr *daddr = &ipv6_hdr(skb)->daddr;
391*848f3c0dSAndrea Mayer 
392*848f3c0dSAndrea Mayer 	if (seg6_next_csid_is_arg_zero(daddr, finfo))
393*848f3c0dSAndrea Mayer 		return input_action_end_core(skb, slwt);
394*848f3c0dSAndrea Mayer 
395*848f3c0dSAndrea Mayer 	/* update DA */
396*848f3c0dSAndrea Mayer 	seg6_next_csid_advance_arg(daddr, finfo);
397*848f3c0dSAndrea Mayer 
398*848f3c0dSAndrea Mayer 	seg6_lookup_nexthop(skb, NULL, 0);
399*848f3c0dSAndrea Mayer 
400*848f3c0dSAndrea Mayer 	return dst_input(skb);
401*848f3c0dSAndrea Mayer }
402*848f3c0dSAndrea Mayer 
403*848f3c0dSAndrea Mayer static bool seg6_next_csid_enabled(__u32 fops)
404*848f3c0dSAndrea Mayer {
405*848f3c0dSAndrea Mayer 	return fops & BIT(SEG6_LOCAL_FLV_OP_NEXT_CSID);
406*848f3c0dSAndrea Mayer }
407*848f3c0dSAndrea Mayer 
408*848f3c0dSAndrea Mayer /* regular endpoint function */
409*848f3c0dSAndrea Mayer static int input_action_end(struct sk_buff *skb, struct seg6_local_lwt *slwt)
410*848f3c0dSAndrea Mayer {
411*848f3c0dSAndrea Mayer 	const struct seg6_flavors_info *finfo = &slwt->flv_info;
412*848f3c0dSAndrea Mayer 
413*848f3c0dSAndrea Mayer 	if (seg6_next_csid_enabled(finfo->flv_ops))
414*848f3c0dSAndrea Mayer 		return end_next_csid_core(skb, slwt);
415*848f3c0dSAndrea Mayer 
416*848f3c0dSAndrea Mayer 	return input_action_end_core(skb, slwt);
417*848f3c0dSAndrea Mayer }
418*848f3c0dSAndrea Mayer 
419140f04c3SDavid Lebrun /* regular endpoint, and forward to specified nexthop */
420140f04c3SDavid Lebrun static int input_action_end_x(struct sk_buff *skb, struct seg6_local_lwt *slwt)
421140f04c3SDavid Lebrun {
422140f04c3SDavid Lebrun 	struct ipv6_sr_hdr *srh;
423140f04c3SDavid Lebrun 
424140f04c3SDavid Lebrun 	srh = get_and_validate_srh(skb);
425140f04c3SDavid Lebrun 	if (!srh)
426140f04c3SDavid Lebrun 		goto drop;
427140f04c3SDavid Lebrun 
428d7a669ddSDavid Lebrun 	advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
429140f04c3SDavid Lebrun 
4301c1e761eSMathieu Xhonneux 	seg6_lookup_nexthop(skb, &slwt->nh6, 0);
431140f04c3SDavid Lebrun 
432140f04c3SDavid Lebrun 	return dst_input(skb);
433140f04c3SDavid Lebrun 
434140f04c3SDavid Lebrun drop:
435140f04c3SDavid Lebrun 	kfree_skb(skb);
436140f04c3SDavid Lebrun 	return -EINVAL;
437140f04c3SDavid Lebrun }
438140f04c3SDavid Lebrun 
439891ef8ddSDavid Lebrun static int input_action_end_t(struct sk_buff *skb, struct seg6_local_lwt *slwt)
440891ef8ddSDavid Lebrun {
441891ef8ddSDavid Lebrun 	struct ipv6_sr_hdr *srh;
442891ef8ddSDavid Lebrun 
443891ef8ddSDavid Lebrun 	srh = get_and_validate_srh(skb);
444891ef8ddSDavid Lebrun 	if (!srh)
445891ef8ddSDavid Lebrun 		goto drop;
446891ef8ddSDavid Lebrun 
447891ef8ddSDavid Lebrun 	advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
448891ef8ddSDavid Lebrun 
4491c1e761eSMathieu Xhonneux 	seg6_lookup_nexthop(skb, NULL, slwt->table);
450891ef8ddSDavid Lebrun 
451891ef8ddSDavid Lebrun 	return dst_input(skb);
452891ef8ddSDavid Lebrun 
453891ef8ddSDavid Lebrun drop:
454891ef8ddSDavid Lebrun 	kfree_skb(skb);
455891ef8ddSDavid Lebrun 	return -EINVAL;
456891ef8ddSDavid Lebrun }
457891ef8ddSDavid Lebrun 
458891ef8ddSDavid Lebrun /* decapsulate and forward inner L2 frame on specified interface */
459891ef8ddSDavid Lebrun static int input_action_end_dx2(struct sk_buff *skb,
460891ef8ddSDavid Lebrun 				struct seg6_local_lwt *slwt)
461891ef8ddSDavid Lebrun {
462891ef8ddSDavid Lebrun 	struct net *net = dev_net(skb->dev);
463891ef8ddSDavid Lebrun 	struct net_device *odev;
464891ef8ddSDavid Lebrun 	struct ethhdr *eth;
465891ef8ddSDavid Lebrun 
46626776253SPaolo Lungaroni 	if (!decap_and_validate(skb, IPPROTO_ETHERNET))
467891ef8ddSDavid Lebrun 		goto drop;
468891ef8ddSDavid Lebrun 
469891ef8ddSDavid Lebrun 	if (!pskb_may_pull(skb, ETH_HLEN))
470891ef8ddSDavid Lebrun 		goto drop;
471891ef8ddSDavid Lebrun 
472891ef8ddSDavid Lebrun 	skb_reset_mac_header(skb);
473891ef8ddSDavid Lebrun 	eth = (struct ethhdr *)skb->data;
474891ef8ddSDavid Lebrun 
475891ef8ddSDavid Lebrun 	/* To determine the frame's protocol, we assume it is 802.3. This avoids
476891ef8ddSDavid Lebrun 	 * a call to eth_type_trans(), which is not really relevant for our
477891ef8ddSDavid Lebrun 	 * use case.
478891ef8ddSDavid Lebrun 	 */
479891ef8ddSDavid Lebrun 	if (!eth_proto_is_802_3(eth->h_proto))
480891ef8ddSDavid Lebrun 		goto drop;
481891ef8ddSDavid Lebrun 
482891ef8ddSDavid Lebrun 	odev = dev_get_by_index_rcu(net, slwt->oif);
483891ef8ddSDavid Lebrun 	if (!odev)
484891ef8ddSDavid Lebrun 		goto drop;
485891ef8ddSDavid Lebrun 
486891ef8ddSDavid Lebrun 	/* As we accept Ethernet frames, make sure the egress device is of
487891ef8ddSDavid Lebrun 	 * the correct type.
488891ef8ddSDavid Lebrun 	 */
489891ef8ddSDavid Lebrun 	if (odev->type != ARPHRD_ETHER)
490891ef8ddSDavid Lebrun 		goto drop;
491891ef8ddSDavid Lebrun 
492891ef8ddSDavid Lebrun 	if (!(odev->flags & IFF_UP) || !netif_carrier_ok(odev))
493891ef8ddSDavid Lebrun 		goto drop;
494891ef8ddSDavid Lebrun 
495891ef8ddSDavid Lebrun 	skb_orphan(skb);
496891ef8ddSDavid Lebrun 
497891ef8ddSDavid Lebrun 	if (skb_warn_if_lro(skb))
498891ef8ddSDavid Lebrun 		goto drop;
499891ef8ddSDavid Lebrun 
500891ef8ddSDavid Lebrun 	skb_forward_csum(skb);
501891ef8ddSDavid Lebrun 
502891ef8ddSDavid Lebrun 	if (skb->len - ETH_HLEN > odev->mtu)
503891ef8ddSDavid Lebrun 		goto drop;
504891ef8ddSDavid Lebrun 
505891ef8ddSDavid Lebrun 	skb->dev = odev;
506891ef8ddSDavid Lebrun 	skb->protocol = eth->h_proto;
507891ef8ddSDavid Lebrun 
508891ef8ddSDavid Lebrun 	return dev_queue_xmit(skb);
509891ef8ddSDavid Lebrun 
510891ef8ddSDavid Lebrun drop:
511891ef8ddSDavid Lebrun 	kfree_skb(skb);
512891ef8ddSDavid Lebrun 	return -EINVAL;
513891ef8ddSDavid Lebrun }
514891ef8ddSDavid Lebrun 
5157a3f5b0dSRyoga Saito static int input_action_end_dx6_finish(struct net *net, struct sock *sk,
5167a3f5b0dSRyoga Saito 				       struct sk_buff *skb)
5177a3f5b0dSRyoga Saito {
5187a3f5b0dSRyoga Saito 	struct dst_entry *orig_dst = skb_dst(skb);
5197a3f5b0dSRyoga Saito 	struct in6_addr *nhaddr = NULL;
5207a3f5b0dSRyoga Saito 	struct seg6_local_lwt *slwt;
5217a3f5b0dSRyoga Saito 
5227a3f5b0dSRyoga Saito 	slwt = seg6_local_lwtunnel(orig_dst->lwtstate);
5237a3f5b0dSRyoga Saito 
5247a3f5b0dSRyoga Saito 	/* The inner packet is not associated to any local interface,
5257a3f5b0dSRyoga Saito 	 * so we do not call netif_rx().
5267a3f5b0dSRyoga Saito 	 *
5277a3f5b0dSRyoga Saito 	 * If slwt->nh6 is set to ::, then lookup the nexthop for the
5287a3f5b0dSRyoga Saito 	 * inner packet's DA. Otherwise, use the specified nexthop.
5297a3f5b0dSRyoga Saito 	 */
5307a3f5b0dSRyoga Saito 	if (!ipv6_addr_any(&slwt->nh6))
5317a3f5b0dSRyoga Saito 		nhaddr = &slwt->nh6;
5327a3f5b0dSRyoga Saito 
5337a3f5b0dSRyoga Saito 	seg6_lookup_nexthop(skb, nhaddr, 0);
5347a3f5b0dSRyoga Saito 
5357a3f5b0dSRyoga Saito 	return dst_input(skb);
5367a3f5b0dSRyoga Saito }
5377a3f5b0dSRyoga Saito 
538140f04c3SDavid Lebrun /* decapsulate and forward to specified nexthop */
539140f04c3SDavid Lebrun static int input_action_end_dx6(struct sk_buff *skb,
540140f04c3SDavid Lebrun 				struct seg6_local_lwt *slwt)
541140f04c3SDavid Lebrun {
542140f04c3SDavid Lebrun 	/* this function accepts IPv6 encapsulated packets, with either
543140f04c3SDavid Lebrun 	 * an SRH with SL=0, or no SRH.
544140f04c3SDavid Lebrun 	 */
545140f04c3SDavid Lebrun 
546d7a669ddSDavid Lebrun 	if (!decap_and_validate(skb, IPPROTO_IPV6))
547140f04c3SDavid Lebrun 		goto drop;
548140f04c3SDavid Lebrun 
549d7a669ddSDavid Lebrun 	if (!pskb_may_pull(skb, sizeof(struct ipv6hdr)))
550140f04c3SDavid Lebrun 		goto drop;
551140f04c3SDavid Lebrun 
552c71644d0SAndrea Mayer 	skb_set_transport_header(skb, sizeof(struct ipv6hdr));
5537a3f5b0dSRyoga Saito 	nf_reset_ct(skb);
554c71644d0SAndrea Mayer 
5557a3f5b0dSRyoga Saito 	if (static_branch_unlikely(&nf_hooks_lwtunnel_enabled))
5567a3f5b0dSRyoga Saito 		return NF_HOOK(NFPROTO_IPV6, NF_INET_PRE_ROUTING,
5577a3f5b0dSRyoga Saito 			       dev_net(skb->dev), NULL, skb, NULL,
5587a3f5b0dSRyoga Saito 			       skb_dst(skb)->dev, input_action_end_dx6_finish);
559140f04c3SDavid Lebrun 
5607a3f5b0dSRyoga Saito 	return input_action_end_dx6_finish(dev_net(skb->dev), NULL, skb);
561140f04c3SDavid Lebrun drop:
562140f04c3SDavid Lebrun 	kfree_skb(skb);
563140f04c3SDavid Lebrun 	return -EINVAL;
564140f04c3SDavid Lebrun }
565140f04c3SDavid Lebrun 
5667a3f5b0dSRyoga Saito static int input_action_end_dx4_finish(struct net *net, struct sock *sk,
5677a3f5b0dSRyoga Saito 				       struct sk_buff *skb)
568891ef8ddSDavid Lebrun {
5697a3f5b0dSRyoga Saito 	struct dst_entry *orig_dst = skb_dst(skb);
5707a3f5b0dSRyoga Saito 	struct seg6_local_lwt *slwt;
571891ef8ddSDavid Lebrun 	struct iphdr *iph;
572891ef8ddSDavid Lebrun 	__be32 nhaddr;
573891ef8ddSDavid Lebrun 	int err;
574891ef8ddSDavid Lebrun 
5757a3f5b0dSRyoga Saito 	slwt = seg6_local_lwtunnel(orig_dst->lwtstate);
576891ef8ddSDavid Lebrun 
577891ef8ddSDavid Lebrun 	iph = ip_hdr(skb);
578891ef8ddSDavid Lebrun 
579891ef8ddSDavid Lebrun 	nhaddr = slwt->nh4.s_addr ?: iph->daddr;
580891ef8ddSDavid Lebrun 
581891ef8ddSDavid Lebrun 	skb_dst_drop(skb);
582891ef8ddSDavid Lebrun 
583891ef8ddSDavid Lebrun 	err = ip_route_input(skb, nhaddr, iph->saddr, 0, skb->dev);
5847a3f5b0dSRyoga Saito 	if (err) {
5857a3f5b0dSRyoga Saito 		kfree_skb(skb);
5867a3f5b0dSRyoga Saito 		return -EINVAL;
5877a3f5b0dSRyoga Saito 	}
588891ef8ddSDavid Lebrun 
589891ef8ddSDavid Lebrun 	return dst_input(skb);
5907a3f5b0dSRyoga Saito }
591891ef8ddSDavid Lebrun 
5927a3f5b0dSRyoga Saito static int input_action_end_dx4(struct sk_buff *skb,
5937a3f5b0dSRyoga Saito 				struct seg6_local_lwt *slwt)
5947a3f5b0dSRyoga Saito {
5957a3f5b0dSRyoga Saito 	if (!decap_and_validate(skb, IPPROTO_IPIP))
5967a3f5b0dSRyoga Saito 		goto drop;
5977a3f5b0dSRyoga Saito 
5987a3f5b0dSRyoga Saito 	if (!pskb_may_pull(skb, sizeof(struct iphdr)))
5997a3f5b0dSRyoga Saito 		goto drop;
6007a3f5b0dSRyoga Saito 
6017a3f5b0dSRyoga Saito 	skb->protocol = htons(ETH_P_IP);
6027a3f5b0dSRyoga Saito 	skb_set_transport_header(skb, sizeof(struct iphdr));
6037a3f5b0dSRyoga Saito 	nf_reset_ct(skb);
6047a3f5b0dSRyoga Saito 
6057a3f5b0dSRyoga Saito 	if (static_branch_unlikely(&nf_hooks_lwtunnel_enabled))
6067a3f5b0dSRyoga Saito 		return NF_HOOK(NFPROTO_IPV4, NF_INET_PRE_ROUTING,
6077a3f5b0dSRyoga Saito 			       dev_net(skb->dev), NULL, skb, NULL,
6087a3f5b0dSRyoga Saito 			       skb_dst(skb)->dev, input_action_end_dx4_finish);
6097a3f5b0dSRyoga Saito 
6107a3f5b0dSRyoga Saito 	return input_action_end_dx4_finish(dev_net(skb->dev), NULL, skb);
611891ef8ddSDavid Lebrun drop:
612891ef8ddSDavid Lebrun 	kfree_skb(skb);
613891ef8ddSDavid Lebrun 	return -EINVAL;
614891ef8ddSDavid Lebrun }
615891ef8ddSDavid Lebrun 
616664d6f86SAndrea Mayer #ifdef CONFIG_NET_L3_MASTER_DEV
617664d6f86SAndrea Mayer static struct net *fib6_config_get_net(const struct fib6_config *fib6_cfg)
618664d6f86SAndrea Mayer {
619664d6f86SAndrea Mayer 	const struct nl_info *nli = &fib6_cfg->fc_nlinfo;
620664d6f86SAndrea Mayer 
621664d6f86SAndrea Mayer 	return nli->nl_net;
622664d6f86SAndrea Mayer }
623664d6f86SAndrea Mayer 
624664d6f86SAndrea Mayer static int __seg6_end_dt_vrf_build(struct seg6_local_lwt *slwt, const void *cfg,
625664d6f86SAndrea Mayer 				   u16 family, struct netlink_ext_ack *extack)
626664d6f86SAndrea Mayer {
627664d6f86SAndrea Mayer 	struct seg6_end_dt_info *info = &slwt->dt_info;
628664d6f86SAndrea Mayer 	int vrf_ifindex;
629664d6f86SAndrea Mayer 	struct net *net;
630664d6f86SAndrea Mayer 
631664d6f86SAndrea Mayer 	net = fib6_config_get_net(cfg);
632664d6f86SAndrea Mayer 
633664d6f86SAndrea Mayer 	/* note that vrf_table was already set by parse_nla_vrftable() */
634664d6f86SAndrea Mayer 	vrf_ifindex = l3mdev_ifindex_lookup_by_table_id(L3MDEV_TYPE_VRF, net,
635664d6f86SAndrea Mayer 							info->vrf_table);
636664d6f86SAndrea Mayer 	if (vrf_ifindex < 0) {
637664d6f86SAndrea Mayer 		if (vrf_ifindex == -EPERM) {
638664d6f86SAndrea Mayer 			NL_SET_ERR_MSG(extack,
639664d6f86SAndrea Mayer 				       "Strict mode for VRF is disabled");
640664d6f86SAndrea Mayer 		} else if (vrf_ifindex == -ENODEV) {
641664d6f86SAndrea Mayer 			NL_SET_ERR_MSG(extack,
642664d6f86SAndrea Mayer 				       "Table has no associated VRF device");
643664d6f86SAndrea Mayer 		} else {
644664d6f86SAndrea Mayer 			pr_debug("seg6local: SRv6 End.DT* creation error=%d\n",
645664d6f86SAndrea Mayer 				 vrf_ifindex);
646664d6f86SAndrea Mayer 		}
647664d6f86SAndrea Mayer 
648664d6f86SAndrea Mayer 		return vrf_ifindex;
649664d6f86SAndrea Mayer 	}
650664d6f86SAndrea Mayer 
651664d6f86SAndrea Mayer 	info->net = net;
652664d6f86SAndrea Mayer 	info->vrf_ifindex = vrf_ifindex;
653664d6f86SAndrea Mayer 
654664d6f86SAndrea Mayer 	info->family = family;
655664d6f86SAndrea Mayer 	info->mode = DT_VRF_MODE;
656664d6f86SAndrea Mayer 
657664d6f86SAndrea Mayer 	return 0;
658664d6f86SAndrea Mayer }
659664d6f86SAndrea Mayer 
660664d6f86SAndrea Mayer /* The SRv6 End.DT4/DT6 behavior extracts the inner (IPv4/IPv6) packet and
661664d6f86SAndrea Mayer  * routes the IPv4/IPv6 packet by looking at the configured routing table.
662664d6f86SAndrea Mayer  *
663664d6f86SAndrea Mayer  * In the SRv6 End.DT4/DT6 use case, we can receive traffic (IPv6+Segment
664664d6f86SAndrea Mayer  * Routing Header packets) from several interfaces and the outer IPv6
665664d6f86SAndrea Mayer  * destination address (DA) is used for retrieving the specific instance of the
666664d6f86SAndrea Mayer  * End.DT4/DT6 behavior that should process the packets.
667664d6f86SAndrea Mayer  *
668664d6f86SAndrea Mayer  * However, the inner IPv4/IPv6 packet is not really bound to any receiving
669664d6f86SAndrea Mayer  * interface and thus the End.DT4/DT6 sets the VRF (associated with the
670664d6f86SAndrea Mayer  * corresponding routing table) as the *receiving* interface.
671664d6f86SAndrea Mayer  * In other words, the End.DT4/DT6 processes a packet as if it has been received
672664d6f86SAndrea Mayer  * directly by the VRF (and not by one of its slave devices, if any).
673664d6f86SAndrea Mayer  * In this way, the VRF interface is used for routing the IPv4/IPv6 packet in
674664d6f86SAndrea Mayer  * according to the routing table configured by the End.DT4/DT6 instance.
675664d6f86SAndrea Mayer  *
676664d6f86SAndrea Mayer  * This design allows you to get some interesting features like:
677664d6f86SAndrea Mayer  *  1) the statistics on rx packets;
678664d6f86SAndrea Mayer  *  2) the possibility to install a packet sniffer on the receiving interface
679664d6f86SAndrea Mayer  *     (the VRF one) for looking at the incoming packets;
680664d6f86SAndrea Mayer  *  3) the possibility to leverage the netfilter prerouting hook for the inner
681664d6f86SAndrea Mayer  *     IPv4 packet.
682664d6f86SAndrea Mayer  *
683664d6f86SAndrea Mayer  * This function returns:
684664d6f86SAndrea Mayer  *  - the sk_buff* when the VRF rcv handler has processed the packet correctly;
685664d6f86SAndrea Mayer  *  - NULL when the skb is consumed by the VRF rcv handler;
686664d6f86SAndrea Mayer  *  - a pointer which encodes a negative error number in case of error.
687664d6f86SAndrea Mayer  *    Note that in this case, the function takes care of freeing the skb.
688664d6f86SAndrea Mayer  */
689664d6f86SAndrea Mayer static struct sk_buff *end_dt_vrf_rcv(struct sk_buff *skb, u16 family,
690664d6f86SAndrea Mayer 				      struct net_device *dev)
691664d6f86SAndrea Mayer {
692664d6f86SAndrea Mayer 	/* based on l3mdev_ip_rcv; we are only interested in the master */
693664d6f86SAndrea Mayer 	if (unlikely(!netif_is_l3_master(dev) && !netif_has_l3_rx_handler(dev)))
694664d6f86SAndrea Mayer 		goto drop;
695664d6f86SAndrea Mayer 
696664d6f86SAndrea Mayer 	if (unlikely(!dev->l3mdev_ops->l3mdev_l3_rcv))
697664d6f86SAndrea Mayer 		goto drop;
698664d6f86SAndrea Mayer 
699664d6f86SAndrea Mayer 	/* the decap packet IPv4/IPv6 does not come with any mac header info.
700664d6f86SAndrea Mayer 	 * We must unset the mac header to allow the VRF device to rebuild it,
701664d6f86SAndrea Mayer 	 * just in case there is a sniffer attached on the device.
702664d6f86SAndrea Mayer 	 */
703664d6f86SAndrea Mayer 	skb_unset_mac_header(skb);
704664d6f86SAndrea Mayer 
705664d6f86SAndrea Mayer 	skb = dev->l3mdev_ops->l3mdev_l3_rcv(dev, skb, family);
706664d6f86SAndrea Mayer 	if (!skb)
707664d6f86SAndrea Mayer 		/* the skb buffer was consumed by the handler */
708664d6f86SAndrea Mayer 		return NULL;
709664d6f86SAndrea Mayer 
710664d6f86SAndrea Mayer 	/* when a packet is received by a VRF or by one of its slaves, the
711664d6f86SAndrea Mayer 	 * master device reference is set into the skb.
712664d6f86SAndrea Mayer 	 */
713664d6f86SAndrea Mayer 	if (unlikely(skb->dev != dev || skb->skb_iif != dev->ifindex))
714664d6f86SAndrea Mayer 		goto drop;
715664d6f86SAndrea Mayer 
716664d6f86SAndrea Mayer 	return skb;
717664d6f86SAndrea Mayer 
718664d6f86SAndrea Mayer drop:
719664d6f86SAndrea Mayer 	kfree_skb(skb);
720664d6f86SAndrea Mayer 	return ERR_PTR(-EINVAL);
721664d6f86SAndrea Mayer }
722664d6f86SAndrea Mayer 
723664d6f86SAndrea Mayer static struct net_device *end_dt_get_vrf_rcu(struct sk_buff *skb,
724664d6f86SAndrea Mayer 					     struct seg6_end_dt_info *info)
725664d6f86SAndrea Mayer {
726664d6f86SAndrea Mayer 	int vrf_ifindex = info->vrf_ifindex;
727664d6f86SAndrea Mayer 	struct net *net = info->net;
728664d6f86SAndrea Mayer 
729664d6f86SAndrea Mayer 	if (unlikely(vrf_ifindex < 0))
730664d6f86SAndrea Mayer 		goto error;
731664d6f86SAndrea Mayer 
732664d6f86SAndrea Mayer 	if (unlikely(!net_eq(dev_net(skb->dev), net)))
733664d6f86SAndrea Mayer 		goto error;
734664d6f86SAndrea Mayer 
735664d6f86SAndrea Mayer 	return dev_get_by_index_rcu(net, vrf_ifindex);
736664d6f86SAndrea Mayer 
737664d6f86SAndrea Mayer error:
738664d6f86SAndrea Mayer 	return NULL;
739664d6f86SAndrea Mayer }
740664d6f86SAndrea Mayer 
741664d6f86SAndrea Mayer static struct sk_buff *end_dt_vrf_core(struct sk_buff *skb,
7428b532109SAndrea Mayer 				       struct seg6_local_lwt *slwt, u16 family)
743664d6f86SAndrea Mayer {
744664d6f86SAndrea Mayer 	struct seg6_end_dt_info *info = &slwt->dt_info;
745664d6f86SAndrea Mayer 	struct net_device *vrf;
7468b532109SAndrea Mayer 	__be16 protocol;
7478b532109SAndrea Mayer 	int hdrlen;
748664d6f86SAndrea Mayer 
749664d6f86SAndrea Mayer 	vrf = end_dt_get_vrf_rcu(skb, info);
750664d6f86SAndrea Mayer 	if (unlikely(!vrf))
751664d6f86SAndrea Mayer 		goto drop;
752664d6f86SAndrea Mayer 
7538b532109SAndrea Mayer 	switch (family) {
7548b532109SAndrea Mayer 	case AF_INET:
7558b532109SAndrea Mayer 		protocol = htons(ETH_P_IP);
7568b532109SAndrea Mayer 		hdrlen = sizeof(struct iphdr);
7578b532109SAndrea Mayer 		break;
7588b532109SAndrea Mayer 	case AF_INET6:
7598b532109SAndrea Mayer 		protocol = htons(ETH_P_IPV6);
7608b532109SAndrea Mayer 		hdrlen = sizeof(struct ipv6hdr);
7618b532109SAndrea Mayer 		break;
7628b532109SAndrea Mayer 	case AF_UNSPEC:
7638b532109SAndrea Mayer 		fallthrough;
7648b532109SAndrea Mayer 	default:
7658b532109SAndrea Mayer 		goto drop;
7668b532109SAndrea Mayer 	}
7678b532109SAndrea Mayer 
7688b532109SAndrea Mayer 	if (unlikely(info->family != AF_UNSPEC && info->family != family)) {
7698b532109SAndrea Mayer 		pr_warn_once("seg6local: SRv6 End.DT* family mismatch");
7708b532109SAndrea Mayer 		goto drop;
7718b532109SAndrea Mayer 	}
7728b532109SAndrea Mayer 
7738b532109SAndrea Mayer 	skb->protocol = protocol;
774664d6f86SAndrea Mayer 
775664d6f86SAndrea Mayer 	skb_dst_drop(skb);
776664d6f86SAndrea Mayer 
7778b532109SAndrea Mayer 	skb_set_transport_header(skb, hdrlen);
7787a3f5b0dSRyoga Saito 	nf_reset_ct(skb);
779664d6f86SAndrea Mayer 
7808b532109SAndrea Mayer 	return end_dt_vrf_rcv(skb, family, vrf);
781664d6f86SAndrea Mayer 
782664d6f86SAndrea Mayer drop:
783664d6f86SAndrea Mayer 	kfree_skb(skb);
784664d6f86SAndrea Mayer 	return ERR_PTR(-EINVAL);
785664d6f86SAndrea Mayer }
786664d6f86SAndrea Mayer 
787664d6f86SAndrea Mayer static int input_action_end_dt4(struct sk_buff *skb,
788664d6f86SAndrea Mayer 				struct seg6_local_lwt *slwt)
789664d6f86SAndrea Mayer {
790664d6f86SAndrea Mayer 	struct iphdr *iph;
791664d6f86SAndrea Mayer 	int err;
792664d6f86SAndrea Mayer 
793664d6f86SAndrea Mayer 	if (!decap_and_validate(skb, IPPROTO_IPIP))
794664d6f86SAndrea Mayer 		goto drop;
795664d6f86SAndrea Mayer 
796664d6f86SAndrea Mayer 	if (!pskb_may_pull(skb, sizeof(struct iphdr)))
797664d6f86SAndrea Mayer 		goto drop;
798664d6f86SAndrea Mayer 
7998b532109SAndrea Mayer 	skb = end_dt_vrf_core(skb, slwt, AF_INET);
800664d6f86SAndrea Mayer 	if (!skb)
801664d6f86SAndrea Mayer 		/* packet has been processed and consumed by the VRF */
802664d6f86SAndrea Mayer 		return 0;
803664d6f86SAndrea Mayer 
804664d6f86SAndrea Mayer 	if (IS_ERR(skb))
805664d6f86SAndrea Mayer 		return PTR_ERR(skb);
806664d6f86SAndrea Mayer 
807664d6f86SAndrea Mayer 	iph = ip_hdr(skb);
808664d6f86SAndrea Mayer 
809664d6f86SAndrea Mayer 	err = ip_route_input(skb, iph->daddr, iph->saddr, 0, skb->dev);
810664d6f86SAndrea Mayer 	if (unlikely(err))
811664d6f86SAndrea Mayer 		goto drop;
812664d6f86SAndrea Mayer 
813664d6f86SAndrea Mayer 	return dst_input(skb);
814664d6f86SAndrea Mayer 
815664d6f86SAndrea Mayer drop:
816664d6f86SAndrea Mayer 	kfree_skb(skb);
817664d6f86SAndrea Mayer 	return -EINVAL;
818664d6f86SAndrea Mayer }
819664d6f86SAndrea Mayer 
820664d6f86SAndrea Mayer static int seg6_end_dt4_build(struct seg6_local_lwt *slwt, const void *cfg,
821664d6f86SAndrea Mayer 			      struct netlink_ext_ack *extack)
822664d6f86SAndrea Mayer {
823664d6f86SAndrea Mayer 	return __seg6_end_dt_vrf_build(slwt, cfg, AF_INET, extack);
824664d6f86SAndrea Mayer }
82520a081b7SAndrea Mayer 
82620a081b7SAndrea Mayer static enum
82720a081b7SAndrea Mayer seg6_end_dt_mode seg6_end_dt6_parse_mode(struct seg6_local_lwt *slwt)
82820a081b7SAndrea Mayer {
82920a081b7SAndrea Mayer 	unsigned long parsed_optattrs = slwt->parsed_optattrs;
83020a081b7SAndrea Mayer 	bool legacy, vrfmode;
83120a081b7SAndrea Mayer 
832300a0fd8SAndrea Mayer 	legacy	= !!(parsed_optattrs & SEG6_F_ATTR(SEG6_LOCAL_TABLE));
833300a0fd8SAndrea Mayer 	vrfmode	= !!(parsed_optattrs & SEG6_F_ATTR(SEG6_LOCAL_VRFTABLE));
83420a081b7SAndrea Mayer 
83520a081b7SAndrea Mayer 	if (!(legacy ^ vrfmode))
83620a081b7SAndrea Mayer 		/* both are absent or present: invalid DT6 mode */
83720a081b7SAndrea Mayer 		return DT_INVALID_MODE;
83820a081b7SAndrea Mayer 
83920a081b7SAndrea Mayer 	return legacy ? DT_LEGACY_MODE : DT_VRF_MODE;
84020a081b7SAndrea Mayer }
84120a081b7SAndrea Mayer 
84220a081b7SAndrea Mayer static enum seg6_end_dt_mode seg6_end_dt6_get_mode(struct seg6_local_lwt *slwt)
84320a081b7SAndrea Mayer {
84420a081b7SAndrea Mayer 	struct seg6_end_dt_info *info = &slwt->dt_info;
84520a081b7SAndrea Mayer 
84620a081b7SAndrea Mayer 	return info->mode;
84720a081b7SAndrea Mayer }
84820a081b7SAndrea Mayer 
84920a081b7SAndrea Mayer static int seg6_end_dt6_build(struct seg6_local_lwt *slwt, const void *cfg,
85020a081b7SAndrea Mayer 			      struct netlink_ext_ack *extack)
85120a081b7SAndrea Mayer {
85220a081b7SAndrea Mayer 	enum seg6_end_dt_mode mode = seg6_end_dt6_parse_mode(slwt);
85320a081b7SAndrea Mayer 	struct seg6_end_dt_info *info = &slwt->dt_info;
85420a081b7SAndrea Mayer 
85520a081b7SAndrea Mayer 	switch (mode) {
85620a081b7SAndrea Mayer 	case DT_LEGACY_MODE:
85720a081b7SAndrea Mayer 		info->mode = DT_LEGACY_MODE;
85820a081b7SAndrea Mayer 		return 0;
85920a081b7SAndrea Mayer 	case DT_VRF_MODE:
86020a081b7SAndrea Mayer 		return __seg6_end_dt_vrf_build(slwt, cfg, AF_INET6, extack);
86120a081b7SAndrea Mayer 	default:
86220a081b7SAndrea Mayer 		NL_SET_ERR_MSG(extack, "table or vrftable must be specified");
86320a081b7SAndrea Mayer 		return -EINVAL;
86420a081b7SAndrea Mayer 	}
86520a081b7SAndrea Mayer }
866664d6f86SAndrea Mayer #endif
867664d6f86SAndrea Mayer 
868891ef8ddSDavid Lebrun static int input_action_end_dt6(struct sk_buff *skb,
869891ef8ddSDavid Lebrun 				struct seg6_local_lwt *slwt)
870891ef8ddSDavid Lebrun {
871891ef8ddSDavid Lebrun 	if (!decap_and_validate(skb, IPPROTO_IPV6))
872891ef8ddSDavid Lebrun 		goto drop;
873891ef8ddSDavid Lebrun 
874891ef8ddSDavid Lebrun 	if (!pskb_may_pull(skb, sizeof(struct ipv6hdr)))
875891ef8ddSDavid Lebrun 		goto drop;
876891ef8ddSDavid Lebrun 
87720a081b7SAndrea Mayer #ifdef CONFIG_NET_L3_MASTER_DEV
87820a081b7SAndrea Mayer 	if (seg6_end_dt6_get_mode(slwt) == DT_LEGACY_MODE)
87920a081b7SAndrea Mayer 		goto legacy_mode;
88020a081b7SAndrea Mayer 
88120a081b7SAndrea Mayer 	/* DT6_VRF_MODE */
8828b532109SAndrea Mayer 	skb = end_dt_vrf_core(skb, slwt, AF_INET6);
88320a081b7SAndrea Mayer 	if (!skb)
88420a081b7SAndrea Mayer 		/* packet has been processed and consumed by the VRF */
88520a081b7SAndrea Mayer 		return 0;
88620a081b7SAndrea Mayer 
88720a081b7SAndrea Mayer 	if (IS_ERR(skb))
88820a081b7SAndrea Mayer 		return PTR_ERR(skb);
88920a081b7SAndrea Mayer 
89020a081b7SAndrea Mayer 	/* note: this time we do not need to specify the table because the VRF
89120a081b7SAndrea Mayer 	 * takes care of selecting the correct table.
89220a081b7SAndrea Mayer 	 */
89320a081b7SAndrea Mayer 	seg6_lookup_any_nexthop(skb, NULL, 0, true);
89420a081b7SAndrea Mayer 
89520a081b7SAndrea Mayer 	return dst_input(skb);
89620a081b7SAndrea Mayer 
89720a081b7SAndrea Mayer legacy_mode:
89820a081b7SAndrea Mayer #endif
899c71644d0SAndrea Mayer 	skb_set_transport_header(skb, sizeof(struct ipv6hdr));
900c71644d0SAndrea Mayer 
901fd1fef0cSAndrea Mayer 	seg6_lookup_any_nexthop(skb, NULL, slwt->table, true);
902891ef8ddSDavid Lebrun 
903891ef8ddSDavid Lebrun 	return dst_input(skb);
904891ef8ddSDavid Lebrun 
905891ef8ddSDavid Lebrun drop:
906891ef8ddSDavid Lebrun 	kfree_skb(skb);
907891ef8ddSDavid Lebrun 	return -EINVAL;
908891ef8ddSDavid Lebrun }
909891ef8ddSDavid Lebrun 
9108b532109SAndrea Mayer #ifdef CONFIG_NET_L3_MASTER_DEV
9118b532109SAndrea Mayer static int seg6_end_dt46_build(struct seg6_local_lwt *slwt, const void *cfg,
9128b532109SAndrea Mayer 			       struct netlink_ext_ack *extack)
9138b532109SAndrea Mayer {
9148b532109SAndrea Mayer 	return __seg6_end_dt_vrf_build(slwt, cfg, AF_UNSPEC, extack);
9158b532109SAndrea Mayer }
9168b532109SAndrea Mayer 
9178b532109SAndrea Mayer static int input_action_end_dt46(struct sk_buff *skb,
9188b532109SAndrea Mayer 				 struct seg6_local_lwt *slwt)
9198b532109SAndrea Mayer {
9208b532109SAndrea Mayer 	unsigned int off = 0;
9218b532109SAndrea Mayer 	int nexthdr;
9228b532109SAndrea Mayer 
9238b532109SAndrea Mayer 	nexthdr = ipv6_find_hdr(skb, &off, -1, NULL, NULL);
9248b532109SAndrea Mayer 	if (unlikely(nexthdr < 0))
9258b532109SAndrea Mayer 		goto drop;
9268b532109SAndrea Mayer 
9278b532109SAndrea Mayer 	switch (nexthdr) {
9288b532109SAndrea Mayer 	case IPPROTO_IPIP:
9298b532109SAndrea Mayer 		return input_action_end_dt4(skb, slwt);
9308b532109SAndrea Mayer 	case IPPROTO_IPV6:
9318b532109SAndrea Mayer 		return input_action_end_dt6(skb, slwt);
9328b532109SAndrea Mayer 	}
9338b532109SAndrea Mayer 
9348b532109SAndrea Mayer drop:
9358b532109SAndrea Mayer 	kfree_skb(skb);
9368b532109SAndrea Mayer 	return -EINVAL;
9378b532109SAndrea Mayer }
9388b532109SAndrea Mayer #endif
9398b532109SAndrea Mayer 
940140f04c3SDavid Lebrun /* push an SRH on top of the current one */
941140f04c3SDavid Lebrun static int input_action_end_b6(struct sk_buff *skb, struct seg6_local_lwt *slwt)
942140f04c3SDavid Lebrun {
943140f04c3SDavid Lebrun 	struct ipv6_sr_hdr *srh;
944140f04c3SDavid Lebrun 	int err = -EINVAL;
945140f04c3SDavid Lebrun 
946140f04c3SDavid Lebrun 	srh = get_and_validate_srh(skb);
947140f04c3SDavid Lebrun 	if (!srh)
948140f04c3SDavid Lebrun 		goto drop;
949140f04c3SDavid Lebrun 
950140f04c3SDavid Lebrun 	err = seg6_do_srh_inline(skb, slwt->srh);
951140f04c3SDavid Lebrun 	if (err)
952140f04c3SDavid Lebrun 		goto drop;
953140f04c3SDavid Lebrun 
954140f04c3SDavid Lebrun 	skb_set_transport_header(skb, sizeof(struct ipv6hdr));
955140f04c3SDavid Lebrun 
9561c1e761eSMathieu Xhonneux 	seg6_lookup_nexthop(skb, NULL, 0);
957140f04c3SDavid Lebrun 
958140f04c3SDavid Lebrun 	return dst_input(skb);
959140f04c3SDavid Lebrun 
960140f04c3SDavid Lebrun drop:
961140f04c3SDavid Lebrun 	kfree_skb(skb);
962140f04c3SDavid Lebrun 	return err;
963140f04c3SDavid Lebrun }
964140f04c3SDavid Lebrun 
965140f04c3SDavid Lebrun /* encapsulate within an outer IPv6 header and a specified SRH */
966140f04c3SDavid Lebrun static int input_action_end_b6_encap(struct sk_buff *skb,
967140f04c3SDavid Lebrun 				     struct seg6_local_lwt *slwt)
968140f04c3SDavid Lebrun {
969140f04c3SDavid Lebrun 	struct ipv6_sr_hdr *srh;
970140f04c3SDavid Lebrun 	int err = -EINVAL;
971140f04c3SDavid Lebrun 
972140f04c3SDavid Lebrun 	srh = get_and_validate_srh(skb);
973140f04c3SDavid Lebrun 	if (!srh)
974140f04c3SDavid Lebrun 		goto drop;
975140f04c3SDavid Lebrun 
976d7a669ddSDavid Lebrun 	advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
977140f04c3SDavid Lebrun 
978140f04c3SDavid Lebrun 	skb_reset_inner_headers(skb);
979140f04c3SDavid Lebrun 	skb->encapsulation = 1;
980140f04c3SDavid Lebrun 
98132d99d0bSDavid Lebrun 	err = seg6_do_srh_encap(skb, slwt->srh, IPPROTO_IPV6);
982140f04c3SDavid Lebrun 	if (err)
983140f04c3SDavid Lebrun 		goto drop;
984140f04c3SDavid Lebrun 
985140f04c3SDavid Lebrun 	skb_set_transport_header(skb, sizeof(struct ipv6hdr));
986140f04c3SDavid Lebrun 
9871c1e761eSMathieu Xhonneux 	seg6_lookup_nexthop(skb, NULL, 0);
988140f04c3SDavid Lebrun 
989140f04c3SDavid Lebrun 	return dst_input(skb);
990140f04c3SDavid Lebrun 
991140f04c3SDavid Lebrun drop:
992140f04c3SDavid Lebrun 	kfree_skb(skb);
993140f04c3SDavid Lebrun 	return err;
994140f04c3SDavid Lebrun }
995140f04c3SDavid Lebrun 
996fe94cc29SMathieu Xhonneux DEFINE_PER_CPU(struct seg6_bpf_srh_state, seg6_bpf_srh_states);
997fe94cc29SMathieu Xhonneux 
998486cdf21SMathieu Xhonneux bool seg6_bpf_has_valid_srh(struct sk_buff *skb)
999486cdf21SMathieu Xhonneux {
1000486cdf21SMathieu Xhonneux 	struct seg6_bpf_srh_state *srh_state =
1001486cdf21SMathieu Xhonneux 		this_cpu_ptr(&seg6_bpf_srh_states);
1002486cdf21SMathieu Xhonneux 	struct ipv6_sr_hdr *srh = srh_state->srh;
1003486cdf21SMathieu Xhonneux 
1004486cdf21SMathieu Xhonneux 	if (unlikely(srh == NULL))
1005486cdf21SMathieu Xhonneux 		return false;
1006486cdf21SMathieu Xhonneux 
1007486cdf21SMathieu Xhonneux 	if (unlikely(!srh_state->valid)) {
1008486cdf21SMathieu Xhonneux 		if ((srh_state->hdrlen & 7) != 0)
1009486cdf21SMathieu Xhonneux 			return false;
1010486cdf21SMathieu Xhonneux 
1011486cdf21SMathieu Xhonneux 		srh->hdrlen = (u8)(srh_state->hdrlen >> 3);
1012bb986a50SAhmed Abdelsalam 		if (!seg6_validate_srh(srh, (srh->hdrlen + 1) << 3, true))
1013486cdf21SMathieu Xhonneux 			return false;
1014486cdf21SMathieu Xhonneux 
1015486cdf21SMathieu Xhonneux 		srh_state->valid = true;
1016486cdf21SMathieu Xhonneux 	}
1017486cdf21SMathieu Xhonneux 
1018486cdf21SMathieu Xhonneux 	return true;
1019486cdf21SMathieu Xhonneux }
1020486cdf21SMathieu Xhonneux 
1021004d4b27SMathieu Xhonneux static int input_action_end_bpf(struct sk_buff *skb,
1022004d4b27SMathieu Xhonneux 				struct seg6_local_lwt *slwt)
1023004d4b27SMathieu Xhonneux {
1024004d4b27SMathieu Xhonneux 	struct seg6_bpf_srh_state *srh_state =
1025004d4b27SMathieu Xhonneux 		this_cpu_ptr(&seg6_bpf_srh_states);
1026004d4b27SMathieu Xhonneux 	struct ipv6_sr_hdr *srh;
1027004d4b27SMathieu Xhonneux 	int ret;
1028004d4b27SMathieu Xhonneux 
1029004d4b27SMathieu Xhonneux 	srh = get_and_validate_srh(skb);
1030486cdf21SMathieu Xhonneux 	if (!srh) {
1031486cdf21SMathieu Xhonneux 		kfree_skb(skb);
1032486cdf21SMathieu Xhonneux 		return -EINVAL;
1033486cdf21SMathieu Xhonneux 	}
1034004d4b27SMathieu Xhonneux 	advance_nextseg(srh, &ipv6_hdr(skb)->daddr);
1035004d4b27SMathieu Xhonneux 
1036004d4b27SMathieu Xhonneux 	/* preempt_disable is needed to protect the per-CPU buffer srh_state,
1037004d4b27SMathieu Xhonneux 	 * which is also accessed by the bpf_lwt_seg6_* helpers
1038004d4b27SMathieu Xhonneux 	 */
1039004d4b27SMathieu Xhonneux 	preempt_disable();
1040486cdf21SMathieu Xhonneux 	srh_state->srh = srh;
1041004d4b27SMathieu Xhonneux 	srh_state->hdrlen = srh->hdrlen << 3;
1042486cdf21SMathieu Xhonneux 	srh_state->valid = true;
1043004d4b27SMathieu Xhonneux 
1044004d4b27SMathieu Xhonneux 	rcu_read_lock();
1045004d4b27SMathieu Xhonneux 	bpf_compute_data_pointers(skb);
1046004d4b27SMathieu Xhonneux 	ret = bpf_prog_run_save_cb(slwt->bpf.prog, skb);
1047004d4b27SMathieu Xhonneux 	rcu_read_unlock();
1048004d4b27SMathieu Xhonneux 
1049004d4b27SMathieu Xhonneux 	switch (ret) {
1050004d4b27SMathieu Xhonneux 	case BPF_OK:
1051004d4b27SMathieu Xhonneux 	case BPF_REDIRECT:
1052004d4b27SMathieu Xhonneux 		break;
1053004d4b27SMathieu Xhonneux 	case BPF_DROP:
1054004d4b27SMathieu Xhonneux 		goto drop;
1055004d4b27SMathieu Xhonneux 	default:
1056004d4b27SMathieu Xhonneux 		pr_warn_once("bpf-seg6local: Illegal return value %u\n", ret);
1057004d4b27SMathieu Xhonneux 		goto drop;
1058004d4b27SMathieu Xhonneux 	}
1059004d4b27SMathieu Xhonneux 
1060486cdf21SMathieu Xhonneux 	if (srh_state->srh && !seg6_bpf_has_valid_srh(skb))
1061004d4b27SMathieu Xhonneux 		goto drop;
1062004d4b27SMathieu Xhonneux 
1063486cdf21SMathieu Xhonneux 	preempt_enable();
1064004d4b27SMathieu Xhonneux 	if (ret != BPF_REDIRECT)
1065004d4b27SMathieu Xhonneux 		seg6_lookup_nexthop(skb, NULL, 0);
1066004d4b27SMathieu Xhonneux 
1067004d4b27SMathieu Xhonneux 	return dst_input(skb);
1068004d4b27SMathieu Xhonneux 
1069004d4b27SMathieu Xhonneux drop:
1070486cdf21SMathieu Xhonneux 	preempt_enable();
1071004d4b27SMathieu Xhonneux 	kfree_skb(skb);
1072004d4b27SMathieu Xhonneux 	return -EINVAL;
1073004d4b27SMathieu Xhonneux }
1074004d4b27SMathieu Xhonneux 
1075d1df6fd8SDavid Lebrun static struct seg6_action_desc seg6_action_table[] = {
1076d1df6fd8SDavid Lebrun 	{
1077d1df6fd8SDavid Lebrun 		.action		= SEG6_LOCAL_ACTION_END,
1078d1df6fd8SDavid Lebrun 		.attrs		= 0,
1079*848f3c0dSAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS |
1080*848f3c0dSAndrea Mayer 				  SEG6_F_ATTR(SEG6_LOCAL_FLAVORS),
1081140f04c3SDavid Lebrun 		.input		= input_action_end,
1082d1df6fd8SDavid Lebrun 	},
1083140f04c3SDavid Lebrun 	{
1084140f04c3SDavid Lebrun 		.action		= SEG6_LOCAL_ACTION_END_X,
1085300a0fd8SAndrea Mayer 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_NH6),
108694604548SAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1087140f04c3SDavid Lebrun 		.input		= input_action_end_x,
1088140f04c3SDavid Lebrun 	},
1089140f04c3SDavid Lebrun 	{
1090891ef8ddSDavid Lebrun 		.action		= SEG6_LOCAL_ACTION_END_T,
1091300a0fd8SAndrea Mayer 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_TABLE),
109294604548SAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1093891ef8ddSDavid Lebrun 		.input		= input_action_end_t,
1094891ef8ddSDavid Lebrun 	},
1095891ef8ddSDavid Lebrun 	{
1096891ef8ddSDavid Lebrun 		.action		= SEG6_LOCAL_ACTION_END_DX2,
1097300a0fd8SAndrea Mayer 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_OIF),
109894604548SAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1099891ef8ddSDavid Lebrun 		.input		= input_action_end_dx2,
1100891ef8ddSDavid Lebrun 	},
1101891ef8ddSDavid Lebrun 	{
1102140f04c3SDavid Lebrun 		.action		= SEG6_LOCAL_ACTION_END_DX6,
1103300a0fd8SAndrea Mayer 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_NH6),
110494604548SAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1105140f04c3SDavid Lebrun 		.input		= input_action_end_dx6,
1106140f04c3SDavid Lebrun 	},
1107140f04c3SDavid Lebrun 	{
1108891ef8ddSDavid Lebrun 		.action		= SEG6_LOCAL_ACTION_END_DX4,
1109300a0fd8SAndrea Mayer 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_NH4),
111094604548SAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1111891ef8ddSDavid Lebrun 		.input		= input_action_end_dx4,
1112891ef8ddSDavid Lebrun 	},
1113891ef8ddSDavid Lebrun 	{
1114664d6f86SAndrea Mayer 		.action		= SEG6_LOCAL_ACTION_END_DT4,
1115300a0fd8SAndrea Mayer 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_VRFTABLE),
111694604548SAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1117664d6f86SAndrea Mayer #ifdef CONFIG_NET_L3_MASTER_DEV
1118664d6f86SAndrea Mayer 		.input		= input_action_end_dt4,
1119664d6f86SAndrea Mayer 		.slwt_ops	= {
1120664d6f86SAndrea Mayer 					.build_state = seg6_end_dt4_build,
1121664d6f86SAndrea Mayer 				  },
1122664d6f86SAndrea Mayer #endif
1123664d6f86SAndrea Mayer 	},
1124664d6f86SAndrea Mayer 	{
1125891ef8ddSDavid Lebrun 		.action		= SEG6_LOCAL_ACTION_END_DT6,
112620a081b7SAndrea Mayer #ifdef CONFIG_NET_L3_MASTER_DEV
112720a081b7SAndrea Mayer 		.attrs		= 0,
112894604548SAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS		|
112994604548SAndrea Mayer 				  SEG6_F_ATTR(SEG6_LOCAL_TABLE) |
1130300a0fd8SAndrea Mayer 				  SEG6_F_ATTR(SEG6_LOCAL_VRFTABLE),
113120a081b7SAndrea Mayer 		.slwt_ops	= {
113220a081b7SAndrea Mayer 					.build_state = seg6_end_dt6_build,
113320a081b7SAndrea Mayer 				  },
113420a081b7SAndrea Mayer #else
1135300a0fd8SAndrea Mayer 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_TABLE),
113694604548SAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
113720a081b7SAndrea Mayer #endif
1138891ef8ddSDavid Lebrun 		.input		= input_action_end_dt6,
1139891ef8ddSDavid Lebrun 	},
1140891ef8ddSDavid Lebrun 	{
11418b532109SAndrea Mayer 		.action		= SEG6_LOCAL_ACTION_END_DT46,
11428b532109SAndrea Mayer 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_VRFTABLE),
11438b532109SAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
11448b532109SAndrea Mayer #ifdef CONFIG_NET_L3_MASTER_DEV
11458b532109SAndrea Mayer 		.input		= input_action_end_dt46,
11468b532109SAndrea Mayer 		.slwt_ops	= {
11478b532109SAndrea Mayer 					.build_state = seg6_end_dt46_build,
11488b532109SAndrea Mayer 				  },
11498b532109SAndrea Mayer #endif
11508b532109SAndrea Mayer 	},
11518b532109SAndrea Mayer 	{
1152140f04c3SDavid Lebrun 		.action		= SEG6_LOCAL_ACTION_END_B6,
1153300a0fd8SAndrea Mayer 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_SRH),
115494604548SAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1155140f04c3SDavid Lebrun 		.input		= input_action_end_b6,
1156140f04c3SDavid Lebrun 	},
1157140f04c3SDavid Lebrun 	{
1158140f04c3SDavid Lebrun 		.action		= SEG6_LOCAL_ACTION_END_B6_ENCAP,
1159300a0fd8SAndrea Mayer 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_SRH),
116094604548SAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1161140f04c3SDavid Lebrun 		.input		= input_action_end_b6_encap,
1162140f04c3SDavid Lebrun 		.static_headroom	= sizeof(struct ipv6hdr),
1163004d4b27SMathieu Xhonneux 	},
1164004d4b27SMathieu Xhonneux 	{
1165004d4b27SMathieu Xhonneux 		.action		= SEG6_LOCAL_ACTION_END_BPF,
1166300a0fd8SAndrea Mayer 		.attrs		= SEG6_F_ATTR(SEG6_LOCAL_BPF),
116794604548SAndrea Mayer 		.optattrs	= SEG6_F_LOCAL_COUNTERS,
1168004d4b27SMathieu Xhonneux 		.input		= input_action_end_bpf,
1169004d4b27SMathieu Xhonneux 	},
1170004d4b27SMathieu Xhonneux 
1171d1df6fd8SDavid Lebrun };
1172d1df6fd8SDavid Lebrun 
1173d1df6fd8SDavid Lebrun static struct seg6_action_desc *__get_action_desc(int action)
1174d1df6fd8SDavid Lebrun {
1175d1df6fd8SDavid Lebrun 	struct seg6_action_desc *desc;
1176d1df6fd8SDavid Lebrun 	int i, count;
1177d1df6fd8SDavid Lebrun 
1178709af180SColin Ian King 	count = ARRAY_SIZE(seg6_action_table);
1179d1df6fd8SDavid Lebrun 	for (i = 0; i < count; i++) {
1180d1df6fd8SDavid Lebrun 		desc = &seg6_action_table[i];
1181d1df6fd8SDavid Lebrun 		if (desc->action == action)
1182d1df6fd8SDavid Lebrun 			return desc;
1183d1df6fd8SDavid Lebrun 	}
1184d1df6fd8SDavid Lebrun 
1185d1df6fd8SDavid Lebrun 	return NULL;
1186d1df6fd8SDavid Lebrun }
1187d1df6fd8SDavid Lebrun 
118894604548SAndrea Mayer static bool seg6_lwtunnel_counters_enabled(struct seg6_local_lwt *slwt)
118994604548SAndrea Mayer {
119094604548SAndrea Mayer 	return slwt->parsed_optattrs & SEG6_F_LOCAL_COUNTERS;
119194604548SAndrea Mayer }
119294604548SAndrea Mayer 
119394604548SAndrea Mayer static void seg6_local_update_counters(struct seg6_local_lwt *slwt,
119494604548SAndrea Mayer 				       unsigned int len, int err)
119594604548SAndrea Mayer {
119694604548SAndrea Mayer 	struct pcpu_seg6_local_counters *pcounters;
119794604548SAndrea Mayer 
119894604548SAndrea Mayer 	pcounters = this_cpu_ptr(slwt->pcpu_counters);
119994604548SAndrea Mayer 	u64_stats_update_begin(&pcounters->syncp);
120094604548SAndrea Mayer 
120194604548SAndrea Mayer 	if (likely(!err)) {
120294604548SAndrea Mayer 		u64_stats_inc(&pcounters->packets);
120394604548SAndrea Mayer 		u64_stats_add(&pcounters->bytes, len);
120494604548SAndrea Mayer 	} else {
120594604548SAndrea Mayer 		u64_stats_inc(&pcounters->errors);
120694604548SAndrea Mayer 	}
120794604548SAndrea Mayer 
120894604548SAndrea Mayer 	u64_stats_update_end(&pcounters->syncp);
120994604548SAndrea Mayer }
121094604548SAndrea Mayer 
12117a3f5b0dSRyoga Saito static int seg6_local_input_core(struct net *net, struct sock *sk,
12127a3f5b0dSRyoga Saito 				 struct sk_buff *skb)
1213d1df6fd8SDavid Lebrun {
1214d1df6fd8SDavid Lebrun 	struct dst_entry *orig_dst = skb_dst(skb);
1215d1df6fd8SDavid Lebrun 	struct seg6_action_desc *desc;
1216d1df6fd8SDavid Lebrun 	struct seg6_local_lwt *slwt;
121794604548SAndrea Mayer 	unsigned int len = skb->len;
121894604548SAndrea Mayer 	int rc;
1219d1df6fd8SDavid Lebrun 
1220d1df6fd8SDavid Lebrun 	slwt = seg6_local_lwtunnel(orig_dst->lwtstate);
1221d1df6fd8SDavid Lebrun 	desc = slwt->desc;
1222d1df6fd8SDavid Lebrun 
122394604548SAndrea Mayer 	rc = desc->input(skb, slwt);
122494604548SAndrea Mayer 
122594604548SAndrea Mayer 	if (!seg6_lwtunnel_counters_enabled(slwt))
122694604548SAndrea Mayer 		return rc;
122794604548SAndrea Mayer 
122894604548SAndrea Mayer 	seg6_local_update_counters(slwt, len, rc);
122994604548SAndrea Mayer 
123094604548SAndrea Mayer 	return rc;
1231d1df6fd8SDavid Lebrun }
1232d1df6fd8SDavid Lebrun 
12337a3f5b0dSRyoga Saito static int seg6_local_input(struct sk_buff *skb)
12347a3f5b0dSRyoga Saito {
12357a3f5b0dSRyoga Saito 	if (skb->protocol != htons(ETH_P_IPV6)) {
12367a3f5b0dSRyoga Saito 		kfree_skb(skb);
12377a3f5b0dSRyoga Saito 		return -EINVAL;
12387a3f5b0dSRyoga Saito 	}
12397a3f5b0dSRyoga Saito 
12407a3f5b0dSRyoga Saito 	if (static_branch_unlikely(&nf_hooks_lwtunnel_enabled))
12417a3f5b0dSRyoga Saito 		return NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_IN,
12427a3f5b0dSRyoga Saito 			       dev_net(skb->dev), NULL, skb, skb->dev, NULL,
12437a3f5b0dSRyoga Saito 			       seg6_local_input_core);
12447a3f5b0dSRyoga Saito 
12457a3f5b0dSRyoga Saito 	return seg6_local_input_core(dev_net(skb->dev), NULL, skb);
12467a3f5b0dSRyoga Saito }
12477a3f5b0dSRyoga Saito 
1248d1df6fd8SDavid Lebrun static const struct nla_policy seg6_local_policy[SEG6_LOCAL_MAX + 1] = {
1249d1df6fd8SDavid Lebrun 	[SEG6_LOCAL_ACTION]	= { .type = NLA_U32 },
1250d1df6fd8SDavid Lebrun 	[SEG6_LOCAL_SRH]	= { .type = NLA_BINARY },
1251d1df6fd8SDavid Lebrun 	[SEG6_LOCAL_TABLE]	= { .type = NLA_U32 },
1252664d6f86SAndrea Mayer 	[SEG6_LOCAL_VRFTABLE]	= { .type = NLA_U32 },
1253d1df6fd8SDavid Lebrun 	[SEG6_LOCAL_NH4]	= { .type = NLA_BINARY,
1254d1df6fd8SDavid Lebrun 				    .len = sizeof(struct in_addr) },
1255d1df6fd8SDavid Lebrun 	[SEG6_LOCAL_NH6]	= { .type = NLA_BINARY,
1256d1df6fd8SDavid Lebrun 				    .len = sizeof(struct in6_addr) },
1257d1df6fd8SDavid Lebrun 	[SEG6_LOCAL_IIF]	= { .type = NLA_U32 },
1258d1df6fd8SDavid Lebrun 	[SEG6_LOCAL_OIF]	= { .type = NLA_U32 },
1259004d4b27SMathieu Xhonneux 	[SEG6_LOCAL_BPF]	= { .type = NLA_NESTED },
126094604548SAndrea Mayer 	[SEG6_LOCAL_COUNTERS]	= { .type = NLA_NESTED },
1261*848f3c0dSAndrea Mayer 	[SEG6_LOCAL_FLAVORS]	= { .type = NLA_NESTED },
1262d1df6fd8SDavid Lebrun };
1263d1df6fd8SDavid Lebrun 
1264e2a8ecc4SAndrea Mayer static int parse_nla_srh(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1265e2a8ecc4SAndrea Mayer 			 struct netlink_ext_ack *extack)
12662d9cc60aSDavid Lebrun {
12672d9cc60aSDavid Lebrun 	struct ipv6_sr_hdr *srh;
12682d9cc60aSDavid Lebrun 	int len;
12692d9cc60aSDavid Lebrun 
12702d9cc60aSDavid Lebrun 	srh = nla_data(attrs[SEG6_LOCAL_SRH]);
12712d9cc60aSDavid Lebrun 	len = nla_len(attrs[SEG6_LOCAL_SRH]);
12722d9cc60aSDavid Lebrun 
12732d9cc60aSDavid Lebrun 	/* SRH must contain at least one segment */
12742d9cc60aSDavid Lebrun 	if (len < sizeof(*srh) + sizeof(struct in6_addr))
12752d9cc60aSDavid Lebrun 		return -EINVAL;
12762d9cc60aSDavid Lebrun 
1277bb986a50SAhmed Abdelsalam 	if (!seg6_validate_srh(srh, len, false))
12782d9cc60aSDavid Lebrun 		return -EINVAL;
12792d9cc60aSDavid Lebrun 
12807fa41efaSYueHaibing 	slwt->srh = kmemdup(srh, len, GFP_KERNEL);
12812d9cc60aSDavid Lebrun 	if (!slwt->srh)
12822d9cc60aSDavid Lebrun 		return -ENOMEM;
12832d9cc60aSDavid Lebrun 
12842d9cc60aSDavid Lebrun 	slwt->headroom += len;
12852d9cc60aSDavid Lebrun 
12862d9cc60aSDavid Lebrun 	return 0;
12872d9cc60aSDavid Lebrun }
12882d9cc60aSDavid Lebrun 
12892d9cc60aSDavid Lebrun static int put_nla_srh(struct sk_buff *skb, struct seg6_local_lwt *slwt)
12902d9cc60aSDavid Lebrun {
12912d9cc60aSDavid Lebrun 	struct ipv6_sr_hdr *srh;
12922d9cc60aSDavid Lebrun 	struct nlattr *nla;
12932d9cc60aSDavid Lebrun 	int len;
12942d9cc60aSDavid Lebrun 
12952d9cc60aSDavid Lebrun 	srh = slwt->srh;
12962d9cc60aSDavid Lebrun 	len = (srh->hdrlen + 1) << 3;
12972d9cc60aSDavid Lebrun 
12982d9cc60aSDavid Lebrun 	nla = nla_reserve(skb, SEG6_LOCAL_SRH, len);
12992d9cc60aSDavid Lebrun 	if (!nla)
13002d9cc60aSDavid Lebrun 		return -EMSGSIZE;
13012d9cc60aSDavid Lebrun 
13022d9cc60aSDavid Lebrun 	memcpy(nla_data(nla), srh, len);
13032d9cc60aSDavid Lebrun 
13042d9cc60aSDavid Lebrun 	return 0;
13052d9cc60aSDavid Lebrun }
13062d9cc60aSDavid Lebrun 
13072d9cc60aSDavid Lebrun static int cmp_nla_srh(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
13082d9cc60aSDavid Lebrun {
13092d9cc60aSDavid Lebrun 	int len = (a->srh->hdrlen + 1) << 3;
13102d9cc60aSDavid Lebrun 
13112d9cc60aSDavid Lebrun 	if (len != ((b->srh->hdrlen + 1) << 3))
13122d9cc60aSDavid Lebrun 		return 1;
13132d9cc60aSDavid Lebrun 
13142d9cc60aSDavid Lebrun 	return memcmp(a->srh, b->srh, len);
13152d9cc60aSDavid Lebrun }
13162d9cc60aSDavid Lebrun 
1317964adce5SAndrea Mayer static void destroy_attr_srh(struct seg6_local_lwt *slwt)
1318964adce5SAndrea Mayer {
1319964adce5SAndrea Mayer 	kfree(slwt->srh);
1320964adce5SAndrea Mayer }
1321964adce5SAndrea Mayer 
1322e2a8ecc4SAndrea Mayer static int parse_nla_table(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1323e2a8ecc4SAndrea Mayer 			   struct netlink_ext_ack *extack)
13242d9cc60aSDavid Lebrun {
13252d9cc60aSDavid Lebrun 	slwt->table = nla_get_u32(attrs[SEG6_LOCAL_TABLE]);
13262d9cc60aSDavid Lebrun 
13272d9cc60aSDavid Lebrun 	return 0;
13282d9cc60aSDavid Lebrun }
13292d9cc60aSDavid Lebrun 
13302d9cc60aSDavid Lebrun static int put_nla_table(struct sk_buff *skb, struct seg6_local_lwt *slwt)
13312d9cc60aSDavid Lebrun {
13322d9cc60aSDavid Lebrun 	if (nla_put_u32(skb, SEG6_LOCAL_TABLE, slwt->table))
13332d9cc60aSDavid Lebrun 		return -EMSGSIZE;
13342d9cc60aSDavid Lebrun 
13352d9cc60aSDavid Lebrun 	return 0;
13362d9cc60aSDavid Lebrun }
13372d9cc60aSDavid Lebrun 
13382d9cc60aSDavid Lebrun static int cmp_nla_table(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
13392d9cc60aSDavid Lebrun {
13402d9cc60aSDavid Lebrun 	if (a->table != b->table)
13412d9cc60aSDavid Lebrun 		return 1;
13422d9cc60aSDavid Lebrun 
13432d9cc60aSDavid Lebrun 	return 0;
13442d9cc60aSDavid Lebrun }
13452d9cc60aSDavid Lebrun 
1346664d6f86SAndrea Mayer static struct
1347664d6f86SAndrea Mayer seg6_end_dt_info *seg6_possible_end_dt_info(struct seg6_local_lwt *slwt)
1348664d6f86SAndrea Mayer {
1349664d6f86SAndrea Mayer #ifdef CONFIG_NET_L3_MASTER_DEV
1350664d6f86SAndrea Mayer 	return &slwt->dt_info;
1351664d6f86SAndrea Mayer #else
1352664d6f86SAndrea Mayer 	return ERR_PTR(-EOPNOTSUPP);
1353664d6f86SAndrea Mayer #endif
1354664d6f86SAndrea Mayer }
1355664d6f86SAndrea Mayer 
1356664d6f86SAndrea Mayer static int parse_nla_vrftable(struct nlattr **attrs,
1357e2a8ecc4SAndrea Mayer 			      struct seg6_local_lwt *slwt,
1358e2a8ecc4SAndrea Mayer 			      struct netlink_ext_ack *extack)
1359664d6f86SAndrea Mayer {
1360664d6f86SAndrea Mayer 	struct seg6_end_dt_info *info = seg6_possible_end_dt_info(slwt);
1361664d6f86SAndrea Mayer 
1362664d6f86SAndrea Mayer 	if (IS_ERR(info))
1363664d6f86SAndrea Mayer 		return PTR_ERR(info);
1364664d6f86SAndrea Mayer 
1365664d6f86SAndrea Mayer 	info->vrf_table = nla_get_u32(attrs[SEG6_LOCAL_VRFTABLE]);
1366664d6f86SAndrea Mayer 
1367664d6f86SAndrea Mayer 	return 0;
1368664d6f86SAndrea Mayer }
1369664d6f86SAndrea Mayer 
1370664d6f86SAndrea Mayer static int put_nla_vrftable(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1371664d6f86SAndrea Mayer {
1372664d6f86SAndrea Mayer 	struct seg6_end_dt_info *info = seg6_possible_end_dt_info(slwt);
1373664d6f86SAndrea Mayer 
1374664d6f86SAndrea Mayer 	if (IS_ERR(info))
1375664d6f86SAndrea Mayer 		return PTR_ERR(info);
1376664d6f86SAndrea Mayer 
1377664d6f86SAndrea Mayer 	if (nla_put_u32(skb, SEG6_LOCAL_VRFTABLE, info->vrf_table))
1378664d6f86SAndrea Mayer 		return -EMSGSIZE;
1379664d6f86SAndrea Mayer 
1380664d6f86SAndrea Mayer 	return 0;
1381664d6f86SAndrea Mayer }
1382664d6f86SAndrea Mayer 
1383664d6f86SAndrea Mayer static int cmp_nla_vrftable(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1384664d6f86SAndrea Mayer {
1385664d6f86SAndrea Mayer 	struct seg6_end_dt_info *info_a = seg6_possible_end_dt_info(a);
1386664d6f86SAndrea Mayer 	struct seg6_end_dt_info *info_b = seg6_possible_end_dt_info(b);
1387664d6f86SAndrea Mayer 
1388664d6f86SAndrea Mayer 	if (info_a->vrf_table != info_b->vrf_table)
1389664d6f86SAndrea Mayer 		return 1;
1390664d6f86SAndrea Mayer 
1391664d6f86SAndrea Mayer 	return 0;
1392664d6f86SAndrea Mayer }
1393664d6f86SAndrea Mayer 
1394e2a8ecc4SAndrea Mayer static int parse_nla_nh4(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1395e2a8ecc4SAndrea Mayer 			 struct netlink_ext_ack *extack)
13962d9cc60aSDavid Lebrun {
13972d9cc60aSDavid Lebrun 	memcpy(&slwt->nh4, nla_data(attrs[SEG6_LOCAL_NH4]),
13982d9cc60aSDavid Lebrun 	       sizeof(struct in_addr));
13992d9cc60aSDavid Lebrun 
14002d9cc60aSDavid Lebrun 	return 0;
14012d9cc60aSDavid Lebrun }
14022d9cc60aSDavid Lebrun 
14032d9cc60aSDavid Lebrun static int put_nla_nh4(struct sk_buff *skb, struct seg6_local_lwt *slwt)
14042d9cc60aSDavid Lebrun {
14052d9cc60aSDavid Lebrun 	struct nlattr *nla;
14062d9cc60aSDavid Lebrun 
14072d9cc60aSDavid Lebrun 	nla = nla_reserve(skb, SEG6_LOCAL_NH4, sizeof(struct in_addr));
14082d9cc60aSDavid Lebrun 	if (!nla)
14092d9cc60aSDavid Lebrun 		return -EMSGSIZE;
14102d9cc60aSDavid Lebrun 
14112d9cc60aSDavid Lebrun 	memcpy(nla_data(nla), &slwt->nh4, sizeof(struct in_addr));
14122d9cc60aSDavid Lebrun 
14132d9cc60aSDavid Lebrun 	return 0;
14142d9cc60aSDavid Lebrun }
14152d9cc60aSDavid Lebrun 
14162d9cc60aSDavid Lebrun static int cmp_nla_nh4(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
14172d9cc60aSDavid Lebrun {
14182d9cc60aSDavid Lebrun 	return memcmp(&a->nh4, &b->nh4, sizeof(struct in_addr));
14192d9cc60aSDavid Lebrun }
14202d9cc60aSDavid Lebrun 
1421e2a8ecc4SAndrea Mayer static int parse_nla_nh6(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1422e2a8ecc4SAndrea Mayer 			 struct netlink_ext_ack *extack)
14232d9cc60aSDavid Lebrun {
14242d9cc60aSDavid Lebrun 	memcpy(&slwt->nh6, nla_data(attrs[SEG6_LOCAL_NH6]),
14252d9cc60aSDavid Lebrun 	       sizeof(struct in6_addr));
14262d9cc60aSDavid Lebrun 
14272d9cc60aSDavid Lebrun 	return 0;
14282d9cc60aSDavid Lebrun }
14292d9cc60aSDavid Lebrun 
14302d9cc60aSDavid Lebrun static int put_nla_nh6(struct sk_buff *skb, struct seg6_local_lwt *slwt)
14312d9cc60aSDavid Lebrun {
14322d9cc60aSDavid Lebrun 	struct nlattr *nla;
14332d9cc60aSDavid Lebrun 
14342d9cc60aSDavid Lebrun 	nla = nla_reserve(skb, SEG6_LOCAL_NH6, sizeof(struct in6_addr));
14352d9cc60aSDavid Lebrun 	if (!nla)
14362d9cc60aSDavid Lebrun 		return -EMSGSIZE;
14372d9cc60aSDavid Lebrun 
14382d9cc60aSDavid Lebrun 	memcpy(nla_data(nla), &slwt->nh6, sizeof(struct in6_addr));
14392d9cc60aSDavid Lebrun 
14402d9cc60aSDavid Lebrun 	return 0;
14412d9cc60aSDavid Lebrun }
14422d9cc60aSDavid Lebrun 
14432d9cc60aSDavid Lebrun static int cmp_nla_nh6(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
14442d9cc60aSDavid Lebrun {
14452d9cc60aSDavid Lebrun 	return memcmp(&a->nh6, &b->nh6, sizeof(struct in6_addr));
14462d9cc60aSDavid Lebrun }
14472d9cc60aSDavid Lebrun 
1448e2a8ecc4SAndrea Mayer static int parse_nla_iif(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1449e2a8ecc4SAndrea Mayer 			 struct netlink_ext_ack *extack)
14502d9cc60aSDavid Lebrun {
14512d9cc60aSDavid Lebrun 	slwt->iif = nla_get_u32(attrs[SEG6_LOCAL_IIF]);
14522d9cc60aSDavid Lebrun 
14532d9cc60aSDavid Lebrun 	return 0;
14542d9cc60aSDavid Lebrun }
14552d9cc60aSDavid Lebrun 
14562d9cc60aSDavid Lebrun static int put_nla_iif(struct sk_buff *skb, struct seg6_local_lwt *slwt)
14572d9cc60aSDavid Lebrun {
14582d9cc60aSDavid Lebrun 	if (nla_put_u32(skb, SEG6_LOCAL_IIF, slwt->iif))
14592d9cc60aSDavid Lebrun 		return -EMSGSIZE;
14602d9cc60aSDavid Lebrun 
14612d9cc60aSDavid Lebrun 	return 0;
14622d9cc60aSDavid Lebrun }
14632d9cc60aSDavid Lebrun 
14642d9cc60aSDavid Lebrun static int cmp_nla_iif(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
14652d9cc60aSDavid Lebrun {
14662d9cc60aSDavid Lebrun 	if (a->iif != b->iif)
14672d9cc60aSDavid Lebrun 		return 1;
14682d9cc60aSDavid Lebrun 
14692d9cc60aSDavid Lebrun 	return 0;
14702d9cc60aSDavid Lebrun }
14712d9cc60aSDavid Lebrun 
1472e2a8ecc4SAndrea Mayer static int parse_nla_oif(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1473e2a8ecc4SAndrea Mayer 			 struct netlink_ext_ack *extack)
14742d9cc60aSDavid Lebrun {
14752d9cc60aSDavid Lebrun 	slwt->oif = nla_get_u32(attrs[SEG6_LOCAL_OIF]);
14762d9cc60aSDavid Lebrun 
14772d9cc60aSDavid Lebrun 	return 0;
14782d9cc60aSDavid Lebrun }
14792d9cc60aSDavid Lebrun 
14802d9cc60aSDavid Lebrun static int put_nla_oif(struct sk_buff *skb, struct seg6_local_lwt *slwt)
14812d9cc60aSDavid Lebrun {
14822d9cc60aSDavid Lebrun 	if (nla_put_u32(skb, SEG6_LOCAL_OIF, slwt->oif))
14832d9cc60aSDavid Lebrun 		return -EMSGSIZE;
14842d9cc60aSDavid Lebrun 
14852d9cc60aSDavid Lebrun 	return 0;
14862d9cc60aSDavid Lebrun }
14872d9cc60aSDavid Lebrun 
14882d9cc60aSDavid Lebrun static int cmp_nla_oif(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
14892d9cc60aSDavid Lebrun {
14902d9cc60aSDavid Lebrun 	if (a->oif != b->oif)
14912d9cc60aSDavid Lebrun 		return 1;
14922d9cc60aSDavid Lebrun 
14932d9cc60aSDavid Lebrun 	return 0;
14942d9cc60aSDavid Lebrun }
14952d9cc60aSDavid Lebrun 
1496004d4b27SMathieu Xhonneux #define MAX_PROG_NAME 256
1497004d4b27SMathieu Xhonneux static const struct nla_policy bpf_prog_policy[SEG6_LOCAL_BPF_PROG_MAX + 1] = {
1498004d4b27SMathieu Xhonneux 	[SEG6_LOCAL_BPF_PROG]	   = { .type = NLA_U32, },
1499004d4b27SMathieu Xhonneux 	[SEG6_LOCAL_BPF_PROG_NAME] = { .type = NLA_NUL_STRING,
1500004d4b27SMathieu Xhonneux 				       .len = MAX_PROG_NAME },
1501004d4b27SMathieu Xhonneux };
1502004d4b27SMathieu Xhonneux 
1503e2a8ecc4SAndrea Mayer static int parse_nla_bpf(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1504e2a8ecc4SAndrea Mayer 			 struct netlink_ext_ack *extack)
1505004d4b27SMathieu Xhonneux {
1506004d4b27SMathieu Xhonneux 	struct nlattr *tb[SEG6_LOCAL_BPF_PROG_MAX + 1];
1507004d4b27SMathieu Xhonneux 	struct bpf_prog *p;
1508004d4b27SMathieu Xhonneux 	int ret;
1509004d4b27SMathieu Xhonneux 	u32 fd;
1510004d4b27SMathieu Xhonneux 
15118cb08174SJohannes Berg 	ret = nla_parse_nested_deprecated(tb, SEG6_LOCAL_BPF_PROG_MAX,
15128cb08174SJohannes Berg 					  attrs[SEG6_LOCAL_BPF],
15138cb08174SJohannes Berg 					  bpf_prog_policy, NULL);
1514004d4b27SMathieu Xhonneux 	if (ret < 0)
1515004d4b27SMathieu Xhonneux 		return ret;
1516004d4b27SMathieu Xhonneux 
1517004d4b27SMathieu Xhonneux 	if (!tb[SEG6_LOCAL_BPF_PROG] || !tb[SEG6_LOCAL_BPF_PROG_NAME])
1518004d4b27SMathieu Xhonneux 		return -EINVAL;
1519004d4b27SMathieu Xhonneux 
1520004d4b27SMathieu Xhonneux 	slwt->bpf.name = nla_memdup(tb[SEG6_LOCAL_BPF_PROG_NAME], GFP_KERNEL);
1521004d4b27SMathieu Xhonneux 	if (!slwt->bpf.name)
1522004d4b27SMathieu Xhonneux 		return -ENOMEM;
1523004d4b27SMathieu Xhonneux 
1524004d4b27SMathieu Xhonneux 	fd = nla_get_u32(tb[SEG6_LOCAL_BPF_PROG]);
1525004d4b27SMathieu Xhonneux 	p = bpf_prog_get_type(fd, BPF_PROG_TYPE_LWT_SEG6LOCAL);
1526004d4b27SMathieu Xhonneux 	if (IS_ERR(p)) {
1527004d4b27SMathieu Xhonneux 		kfree(slwt->bpf.name);
1528004d4b27SMathieu Xhonneux 		return PTR_ERR(p);
1529004d4b27SMathieu Xhonneux 	}
1530004d4b27SMathieu Xhonneux 
1531004d4b27SMathieu Xhonneux 	slwt->bpf.prog = p;
1532004d4b27SMathieu Xhonneux 	return 0;
1533004d4b27SMathieu Xhonneux }
1534004d4b27SMathieu Xhonneux 
1535004d4b27SMathieu Xhonneux static int put_nla_bpf(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1536004d4b27SMathieu Xhonneux {
1537004d4b27SMathieu Xhonneux 	struct nlattr *nest;
1538004d4b27SMathieu Xhonneux 
1539004d4b27SMathieu Xhonneux 	if (!slwt->bpf.prog)
1540004d4b27SMathieu Xhonneux 		return 0;
1541004d4b27SMathieu Xhonneux 
1542ae0be8deSMichal Kubecek 	nest = nla_nest_start_noflag(skb, SEG6_LOCAL_BPF);
1543004d4b27SMathieu Xhonneux 	if (!nest)
1544004d4b27SMathieu Xhonneux 		return -EMSGSIZE;
1545004d4b27SMathieu Xhonneux 
1546004d4b27SMathieu Xhonneux 	if (nla_put_u32(skb, SEG6_LOCAL_BPF_PROG, slwt->bpf.prog->aux->id))
1547004d4b27SMathieu Xhonneux 		return -EMSGSIZE;
1548004d4b27SMathieu Xhonneux 
1549004d4b27SMathieu Xhonneux 	if (slwt->bpf.name &&
1550004d4b27SMathieu Xhonneux 	    nla_put_string(skb, SEG6_LOCAL_BPF_PROG_NAME, slwt->bpf.name))
1551004d4b27SMathieu Xhonneux 		return -EMSGSIZE;
1552004d4b27SMathieu Xhonneux 
1553004d4b27SMathieu Xhonneux 	return nla_nest_end(skb, nest);
1554004d4b27SMathieu Xhonneux }
1555004d4b27SMathieu Xhonneux 
1556004d4b27SMathieu Xhonneux static int cmp_nla_bpf(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1557004d4b27SMathieu Xhonneux {
1558004d4b27SMathieu Xhonneux 	if (!a->bpf.name && !b->bpf.name)
1559004d4b27SMathieu Xhonneux 		return 0;
1560004d4b27SMathieu Xhonneux 
1561004d4b27SMathieu Xhonneux 	if (!a->bpf.name || !b->bpf.name)
1562004d4b27SMathieu Xhonneux 		return 1;
1563004d4b27SMathieu Xhonneux 
1564004d4b27SMathieu Xhonneux 	return strcmp(a->bpf.name, b->bpf.name);
1565004d4b27SMathieu Xhonneux }
1566004d4b27SMathieu Xhonneux 
1567964adce5SAndrea Mayer static void destroy_attr_bpf(struct seg6_local_lwt *slwt)
1568964adce5SAndrea Mayer {
1569964adce5SAndrea Mayer 	kfree(slwt->bpf.name);
1570964adce5SAndrea Mayer 	if (slwt->bpf.prog)
1571964adce5SAndrea Mayer 		bpf_prog_put(slwt->bpf.prog);
1572964adce5SAndrea Mayer }
1573964adce5SAndrea Mayer 
157494604548SAndrea Mayer static const struct
157594604548SAndrea Mayer nla_policy seg6_local_counters_policy[SEG6_LOCAL_CNT_MAX + 1] = {
157694604548SAndrea Mayer 	[SEG6_LOCAL_CNT_PACKETS]	= { .type = NLA_U64 },
157794604548SAndrea Mayer 	[SEG6_LOCAL_CNT_BYTES]		= { .type = NLA_U64 },
157894604548SAndrea Mayer 	[SEG6_LOCAL_CNT_ERRORS]		= { .type = NLA_U64 },
157994604548SAndrea Mayer };
158094604548SAndrea Mayer 
158194604548SAndrea Mayer static int parse_nla_counters(struct nlattr **attrs,
1582e2a8ecc4SAndrea Mayer 			      struct seg6_local_lwt *slwt,
1583e2a8ecc4SAndrea Mayer 			      struct netlink_ext_ack *extack)
158494604548SAndrea Mayer {
158594604548SAndrea Mayer 	struct pcpu_seg6_local_counters __percpu *pcounters;
158694604548SAndrea Mayer 	struct nlattr *tb[SEG6_LOCAL_CNT_MAX + 1];
158794604548SAndrea Mayer 	int ret;
158894604548SAndrea Mayer 
158994604548SAndrea Mayer 	ret = nla_parse_nested_deprecated(tb, SEG6_LOCAL_CNT_MAX,
159094604548SAndrea Mayer 					  attrs[SEG6_LOCAL_COUNTERS],
159194604548SAndrea Mayer 					  seg6_local_counters_policy, NULL);
159294604548SAndrea Mayer 	if (ret < 0)
159394604548SAndrea Mayer 		return ret;
159494604548SAndrea Mayer 
159594604548SAndrea Mayer 	/* basic support for SRv6 Behavior counters requires at least:
159694604548SAndrea Mayer 	 * packets, bytes and errors.
159794604548SAndrea Mayer 	 */
159894604548SAndrea Mayer 	if (!tb[SEG6_LOCAL_CNT_PACKETS] || !tb[SEG6_LOCAL_CNT_BYTES] ||
159994604548SAndrea Mayer 	    !tb[SEG6_LOCAL_CNT_ERRORS])
160094604548SAndrea Mayer 		return -EINVAL;
160194604548SAndrea Mayer 
160294604548SAndrea Mayer 	/* counters are always zero initialized */
160394604548SAndrea Mayer 	pcounters = seg6_local_alloc_pcpu_counters(GFP_KERNEL);
160494604548SAndrea Mayer 	if (!pcounters)
160594604548SAndrea Mayer 		return -ENOMEM;
160694604548SAndrea Mayer 
160794604548SAndrea Mayer 	slwt->pcpu_counters = pcounters;
160894604548SAndrea Mayer 
160994604548SAndrea Mayer 	return 0;
161094604548SAndrea Mayer }
161194604548SAndrea Mayer 
161294604548SAndrea Mayer static int seg6_local_fill_nla_counters(struct sk_buff *skb,
161394604548SAndrea Mayer 					struct seg6_local_counters *counters)
161494604548SAndrea Mayer {
161594604548SAndrea Mayer 	if (nla_put_u64_64bit(skb, SEG6_LOCAL_CNT_PACKETS, counters->packets,
161694604548SAndrea Mayer 			      SEG6_LOCAL_CNT_PAD))
161794604548SAndrea Mayer 		return -EMSGSIZE;
161894604548SAndrea Mayer 
161994604548SAndrea Mayer 	if (nla_put_u64_64bit(skb, SEG6_LOCAL_CNT_BYTES, counters->bytes,
162094604548SAndrea Mayer 			      SEG6_LOCAL_CNT_PAD))
162194604548SAndrea Mayer 		return -EMSGSIZE;
162294604548SAndrea Mayer 
162394604548SAndrea Mayer 	if (nla_put_u64_64bit(skb, SEG6_LOCAL_CNT_ERRORS, counters->errors,
162494604548SAndrea Mayer 			      SEG6_LOCAL_CNT_PAD))
162594604548SAndrea Mayer 		return -EMSGSIZE;
162694604548SAndrea Mayer 
162794604548SAndrea Mayer 	return 0;
162894604548SAndrea Mayer }
162994604548SAndrea Mayer 
163094604548SAndrea Mayer static int put_nla_counters(struct sk_buff *skb, struct seg6_local_lwt *slwt)
163194604548SAndrea Mayer {
163294604548SAndrea Mayer 	struct seg6_local_counters counters = { 0, 0, 0 };
163394604548SAndrea Mayer 	struct nlattr *nest;
163494604548SAndrea Mayer 	int rc, i;
163594604548SAndrea Mayer 
163694604548SAndrea Mayer 	nest = nla_nest_start(skb, SEG6_LOCAL_COUNTERS);
163794604548SAndrea Mayer 	if (!nest)
163894604548SAndrea Mayer 		return -EMSGSIZE;
163994604548SAndrea Mayer 
164094604548SAndrea Mayer 	for_each_possible_cpu(i) {
164194604548SAndrea Mayer 		struct pcpu_seg6_local_counters *pcounters;
164294604548SAndrea Mayer 		u64 packets, bytes, errors;
164394604548SAndrea Mayer 		unsigned int start;
164494604548SAndrea Mayer 
164594604548SAndrea Mayer 		pcounters = per_cpu_ptr(slwt->pcpu_counters, i);
164694604548SAndrea Mayer 		do {
164794604548SAndrea Mayer 			start = u64_stats_fetch_begin_irq(&pcounters->syncp);
164894604548SAndrea Mayer 
164994604548SAndrea Mayer 			packets = u64_stats_read(&pcounters->packets);
165094604548SAndrea Mayer 			bytes = u64_stats_read(&pcounters->bytes);
165194604548SAndrea Mayer 			errors = u64_stats_read(&pcounters->errors);
165294604548SAndrea Mayer 
165394604548SAndrea Mayer 		} while (u64_stats_fetch_retry_irq(&pcounters->syncp, start));
165494604548SAndrea Mayer 
165594604548SAndrea Mayer 		counters.packets += packets;
165694604548SAndrea Mayer 		counters.bytes += bytes;
165794604548SAndrea Mayer 		counters.errors += errors;
165894604548SAndrea Mayer 	}
165994604548SAndrea Mayer 
166094604548SAndrea Mayer 	rc = seg6_local_fill_nla_counters(skb, &counters);
166194604548SAndrea Mayer 	if (rc < 0) {
166294604548SAndrea Mayer 		nla_nest_cancel(skb, nest);
166394604548SAndrea Mayer 		return rc;
166494604548SAndrea Mayer 	}
166594604548SAndrea Mayer 
166694604548SAndrea Mayer 	return nla_nest_end(skb, nest);
166794604548SAndrea Mayer }
166894604548SAndrea Mayer 
166994604548SAndrea Mayer static int cmp_nla_counters(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
167094604548SAndrea Mayer {
167194604548SAndrea Mayer 	/* a and b are equal if both have pcpu_counters set or not */
167294604548SAndrea Mayer 	return (!!((unsigned long)a->pcpu_counters)) ^
167394604548SAndrea Mayer 		(!!((unsigned long)b->pcpu_counters));
167494604548SAndrea Mayer }
167594604548SAndrea Mayer 
167694604548SAndrea Mayer static void destroy_attr_counters(struct seg6_local_lwt *slwt)
167794604548SAndrea Mayer {
167894604548SAndrea Mayer 	free_percpu(slwt->pcpu_counters);
167994604548SAndrea Mayer }
168094604548SAndrea Mayer 
1681*848f3c0dSAndrea Mayer static const
1682*848f3c0dSAndrea Mayer struct nla_policy seg6_local_flavors_policy[SEG6_LOCAL_FLV_MAX + 1] = {
1683*848f3c0dSAndrea Mayer 	[SEG6_LOCAL_FLV_OPERATION]	= { .type = NLA_U32 },
1684*848f3c0dSAndrea Mayer 	[SEG6_LOCAL_FLV_LCBLOCK_BITS]	= { .type = NLA_U8 },
1685*848f3c0dSAndrea Mayer 	[SEG6_LOCAL_FLV_LCNODE_FN_BITS]	= { .type = NLA_U8 },
1686*848f3c0dSAndrea Mayer };
1687*848f3c0dSAndrea Mayer 
1688*848f3c0dSAndrea Mayer /* check whether the lengths of the Locator-Block and Locator-Node Function
1689*848f3c0dSAndrea Mayer  * are compatible with the dimension of a C-SID container.
1690*848f3c0dSAndrea Mayer  */
1691*848f3c0dSAndrea Mayer static int seg6_chk_next_csid_cfg(__u8 block_len, __u8 func_len)
1692*848f3c0dSAndrea Mayer {
1693*848f3c0dSAndrea Mayer 	/* Locator-Block and Locator-Node Function cannot exceed 128 bits
1694*848f3c0dSAndrea Mayer 	 * (i.e. C-SID container lenghts).
1695*848f3c0dSAndrea Mayer 	 */
1696*848f3c0dSAndrea Mayer 	if (next_csid_chk_cntr_bits(block_len, func_len))
1697*848f3c0dSAndrea Mayer 		return -EINVAL;
1698*848f3c0dSAndrea Mayer 
1699*848f3c0dSAndrea Mayer 	/* Locator-Block length must be greater than zero and evenly divisible
1700*848f3c0dSAndrea Mayer 	 * by 8. There must be room for a Locator-Node Function, at least.
1701*848f3c0dSAndrea Mayer 	 */
1702*848f3c0dSAndrea Mayer 	if (next_csid_chk_lcblock_bits(block_len))
1703*848f3c0dSAndrea Mayer 		return -EINVAL;
1704*848f3c0dSAndrea Mayer 
1705*848f3c0dSAndrea Mayer 	/* Locator-Node Function length must be greater than zero and evenly
1706*848f3c0dSAndrea Mayer 	 * divisible by 8. There must be room for the Locator-Block.
1707*848f3c0dSAndrea Mayer 	 */
1708*848f3c0dSAndrea Mayer 	if (next_csid_chk_lcnode_fn_bits(func_len))
1709*848f3c0dSAndrea Mayer 		return -EINVAL;
1710*848f3c0dSAndrea Mayer 
1711*848f3c0dSAndrea Mayer 	return 0;
1712*848f3c0dSAndrea Mayer }
1713*848f3c0dSAndrea Mayer 
1714*848f3c0dSAndrea Mayer static int seg6_parse_nla_next_csid_cfg(struct nlattr **tb,
1715*848f3c0dSAndrea Mayer 					struct seg6_flavors_info *finfo,
1716*848f3c0dSAndrea Mayer 					struct netlink_ext_ack *extack)
1717*848f3c0dSAndrea Mayer {
1718*848f3c0dSAndrea Mayer 	__u8 func_len = SEG6_LOCAL_LCNODE_FN_DBITS;
1719*848f3c0dSAndrea Mayer 	__u8 block_len = SEG6_LOCAL_LCBLOCK_DBITS;
1720*848f3c0dSAndrea Mayer 	int rc;
1721*848f3c0dSAndrea Mayer 
1722*848f3c0dSAndrea Mayer 	if (tb[SEG6_LOCAL_FLV_LCBLOCK_BITS])
1723*848f3c0dSAndrea Mayer 		block_len = nla_get_u8(tb[SEG6_LOCAL_FLV_LCBLOCK_BITS]);
1724*848f3c0dSAndrea Mayer 
1725*848f3c0dSAndrea Mayer 	if (tb[SEG6_LOCAL_FLV_LCNODE_FN_BITS])
1726*848f3c0dSAndrea Mayer 		func_len = nla_get_u8(tb[SEG6_LOCAL_FLV_LCNODE_FN_BITS]);
1727*848f3c0dSAndrea Mayer 
1728*848f3c0dSAndrea Mayer 	rc = seg6_chk_next_csid_cfg(block_len, func_len);
1729*848f3c0dSAndrea Mayer 	if (rc < 0) {
1730*848f3c0dSAndrea Mayer 		NL_SET_ERR_MSG(extack,
1731*848f3c0dSAndrea Mayer 			       "Invalid Locator Block/Node Function lengths");
1732*848f3c0dSAndrea Mayer 		return rc;
1733*848f3c0dSAndrea Mayer 	}
1734*848f3c0dSAndrea Mayer 
1735*848f3c0dSAndrea Mayer 	finfo->lcblock_bits = block_len;
1736*848f3c0dSAndrea Mayer 	finfo->lcnode_func_bits = func_len;
1737*848f3c0dSAndrea Mayer 
1738*848f3c0dSAndrea Mayer 	return 0;
1739*848f3c0dSAndrea Mayer }
1740*848f3c0dSAndrea Mayer 
1741*848f3c0dSAndrea Mayer static int parse_nla_flavors(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1742*848f3c0dSAndrea Mayer 			     struct netlink_ext_ack *extack)
1743*848f3c0dSAndrea Mayer {
1744*848f3c0dSAndrea Mayer 	struct seg6_flavors_info *finfo = &slwt->flv_info;
1745*848f3c0dSAndrea Mayer 	struct nlattr *tb[SEG6_LOCAL_FLV_MAX + 1];
1746*848f3c0dSAndrea Mayer 	unsigned long fops;
1747*848f3c0dSAndrea Mayer 	int rc;
1748*848f3c0dSAndrea Mayer 
1749*848f3c0dSAndrea Mayer 	rc = nla_parse_nested_deprecated(tb, SEG6_LOCAL_FLV_MAX,
1750*848f3c0dSAndrea Mayer 					 attrs[SEG6_LOCAL_FLAVORS],
1751*848f3c0dSAndrea Mayer 					 seg6_local_flavors_policy, NULL);
1752*848f3c0dSAndrea Mayer 	if (rc < 0)
1753*848f3c0dSAndrea Mayer 		return rc;
1754*848f3c0dSAndrea Mayer 
1755*848f3c0dSAndrea Mayer 	/* this attribute MUST always be present since it represents the Flavor
1756*848f3c0dSAndrea Mayer 	 * operation(s) to be carried out.
1757*848f3c0dSAndrea Mayer 	 */
1758*848f3c0dSAndrea Mayer 	if (!tb[SEG6_LOCAL_FLV_OPERATION])
1759*848f3c0dSAndrea Mayer 		return -EINVAL;
1760*848f3c0dSAndrea Mayer 
1761*848f3c0dSAndrea Mayer 	fops = nla_get_u32(tb[SEG6_LOCAL_FLV_OPERATION]);
1762*848f3c0dSAndrea Mayer 	if (fops & ~SEG6_LOCAL_FLV_SUPP_OPS) {
1763*848f3c0dSAndrea Mayer 		NL_SET_ERR_MSG(extack, "Unsupported Flavor operation(s)");
1764*848f3c0dSAndrea Mayer 		return -EOPNOTSUPP;
1765*848f3c0dSAndrea Mayer 	}
1766*848f3c0dSAndrea Mayer 
1767*848f3c0dSAndrea Mayer 	finfo->flv_ops = fops;
1768*848f3c0dSAndrea Mayer 
1769*848f3c0dSAndrea Mayer 	if (seg6_next_csid_enabled(fops)) {
1770*848f3c0dSAndrea Mayer 		/* Locator-Block and Locator-Node Function lengths can be
1771*848f3c0dSAndrea Mayer 		 * provided by the user space. Otherwise, default values are
1772*848f3c0dSAndrea Mayer 		 * applied.
1773*848f3c0dSAndrea Mayer 		 */
1774*848f3c0dSAndrea Mayer 		rc = seg6_parse_nla_next_csid_cfg(tb, finfo, extack);
1775*848f3c0dSAndrea Mayer 		if (rc < 0)
1776*848f3c0dSAndrea Mayer 			return rc;
1777*848f3c0dSAndrea Mayer 	}
1778*848f3c0dSAndrea Mayer 
1779*848f3c0dSAndrea Mayer 	return 0;
1780*848f3c0dSAndrea Mayer }
1781*848f3c0dSAndrea Mayer 
1782*848f3c0dSAndrea Mayer static int seg6_fill_nla_next_csid_cfg(struct sk_buff *skb,
1783*848f3c0dSAndrea Mayer 				       struct seg6_flavors_info *finfo)
1784*848f3c0dSAndrea Mayer {
1785*848f3c0dSAndrea Mayer 	if (nla_put_u8(skb, SEG6_LOCAL_FLV_LCBLOCK_BITS, finfo->lcblock_bits))
1786*848f3c0dSAndrea Mayer 		return -EMSGSIZE;
1787*848f3c0dSAndrea Mayer 
1788*848f3c0dSAndrea Mayer 	if (nla_put_u8(skb, SEG6_LOCAL_FLV_LCNODE_FN_BITS,
1789*848f3c0dSAndrea Mayer 		       finfo->lcnode_func_bits))
1790*848f3c0dSAndrea Mayer 		return -EMSGSIZE;
1791*848f3c0dSAndrea Mayer 
1792*848f3c0dSAndrea Mayer 	return 0;
1793*848f3c0dSAndrea Mayer }
1794*848f3c0dSAndrea Mayer 
1795*848f3c0dSAndrea Mayer static int put_nla_flavors(struct sk_buff *skb, struct seg6_local_lwt *slwt)
1796*848f3c0dSAndrea Mayer {
1797*848f3c0dSAndrea Mayer 	struct seg6_flavors_info *finfo = &slwt->flv_info;
1798*848f3c0dSAndrea Mayer 	__u32 fops = finfo->flv_ops;
1799*848f3c0dSAndrea Mayer 	struct nlattr *nest;
1800*848f3c0dSAndrea Mayer 	int rc;
1801*848f3c0dSAndrea Mayer 
1802*848f3c0dSAndrea Mayer 	nest = nla_nest_start(skb, SEG6_LOCAL_FLAVORS);
1803*848f3c0dSAndrea Mayer 	if (!nest)
1804*848f3c0dSAndrea Mayer 		return -EMSGSIZE;
1805*848f3c0dSAndrea Mayer 
1806*848f3c0dSAndrea Mayer 	if (nla_put_u32(skb, SEG6_LOCAL_FLV_OPERATION, fops)) {
1807*848f3c0dSAndrea Mayer 		rc = -EMSGSIZE;
1808*848f3c0dSAndrea Mayer 		goto err;
1809*848f3c0dSAndrea Mayer 	}
1810*848f3c0dSAndrea Mayer 
1811*848f3c0dSAndrea Mayer 	if (seg6_next_csid_enabled(fops)) {
1812*848f3c0dSAndrea Mayer 		rc = seg6_fill_nla_next_csid_cfg(skb, finfo);
1813*848f3c0dSAndrea Mayer 		if (rc < 0)
1814*848f3c0dSAndrea Mayer 			goto err;
1815*848f3c0dSAndrea Mayer 	}
1816*848f3c0dSAndrea Mayer 
1817*848f3c0dSAndrea Mayer 	return nla_nest_end(skb, nest);
1818*848f3c0dSAndrea Mayer 
1819*848f3c0dSAndrea Mayer err:
1820*848f3c0dSAndrea Mayer 	nla_nest_cancel(skb, nest);
1821*848f3c0dSAndrea Mayer 	return rc;
1822*848f3c0dSAndrea Mayer }
1823*848f3c0dSAndrea Mayer 
1824*848f3c0dSAndrea Mayer static int seg6_cmp_nla_next_csid_cfg(struct seg6_flavors_info *finfo_a,
1825*848f3c0dSAndrea Mayer 				      struct seg6_flavors_info *finfo_b)
1826*848f3c0dSAndrea Mayer {
1827*848f3c0dSAndrea Mayer 	if (finfo_a->lcblock_bits != finfo_b->lcblock_bits)
1828*848f3c0dSAndrea Mayer 		return 1;
1829*848f3c0dSAndrea Mayer 
1830*848f3c0dSAndrea Mayer 	if (finfo_a->lcnode_func_bits != finfo_b->lcnode_func_bits)
1831*848f3c0dSAndrea Mayer 		return 1;
1832*848f3c0dSAndrea Mayer 
1833*848f3c0dSAndrea Mayer 	return 0;
1834*848f3c0dSAndrea Mayer }
1835*848f3c0dSAndrea Mayer 
1836*848f3c0dSAndrea Mayer static int cmp_nla_flavors(struct seg6_local_lwt *a, struct seg6_local_lwt *b)
1837*848f3c0dSAndrea Mayer {
1838*848f3c0dSAndrea Mayer 	struct seg6_flavors_info *finfo_a = &a->flv_info;
1839*848f3c0dSAndrea Mayer 	struct seg6_flavors_info *finfo_b = &b->flv_info;
1840*848f3c0dSAndrea Mayer 
1841*848f3c0dSAndrea Mayer 	if (finfo_a->flv_ops != finfo_b->flv_ops)
1842*848f3c0dSAndrea Mayer 		return 1;
1843*848f3c0dSAndrea Mayer 
1844*848f3c0dSAndrea Mayer 	if (seg6_next_csid_enabled(finfo_a->flv_ops)) {
1845*848f3c0dSAndrea Mayer 		if (seg6_cmp_nla_next_csid_cfg(finfo_a, finfo_b))
1846*848f3c0dSAndrea Mayer 			return 1;
1847*848f3c0dSAndrea Mayer 	}
1848*848f3c0dSAndrea Mayer 
1849*848f3c0dSAndrea Mayer 	return 0;
1850*848f3c0dSAndrea Mayer }
1851*848f3c0dSAndrea Mayer 
1852*848f3c0dSAndrea Mayer static int encap_size_flavors(struct seg6_local_lwt *slwt)
1853*848f3c0dSAndrea Mayer {
1854*848f3c0dSAndrea Mayer 	struct seg6_flavors_info *finfo = &slwt->flv_info;
1855*848f3c0dSAndrea Mayer 	int nlsize;
1856*848f3c0dSAndrea Mayer 
1857*848f3c0dSAndrea Mayer 	nlsize = nla_total_size(0) +	/* nest SEG6_LOCAL_FLAVORS */
1858*848f3c0dSAndrea Mayer 		 nla_total_size(4);	/* SEG6_LOCAL_FLV_OPERATION */
1859*848f3c0dSAndrea Mayer 
1860*848f3c0dSAndrea Mayer 	if (seg6_next_csid_enabled(finfo->flv_ops))
1861*848f3c0dSAndrea Mayer 		nlsize += nla_total_size(1) + /* SEG6_LOCAL_FLV_LCBLOCK_BITS */
1862*848f3c0dSAndrea Mayer 			  nla_total_size(1); /* SEG6_LOCAL_FLV_LCNODE_FN_BITS */
1863*848f3c0dSAndrea Mayer 
1864*848f3c0dSAndrea Mayer 	return nlsize;
1865*848f3c0dSAndrea Mayer }
1866*848f3c0dSAndrea Mayer 
1867d1df6fd8SDavid Lebrun struct seg6_action_param {
1868e2a8ecc4SAndrea Mayer 	int (*parse)(struct nlattr **attrs, struct seg6_local_lwt *slwt,
1869e2a8ecc4SAndrea Mayer 		     struct netlink_ext_ack *extack);
1870d1df6fd8SDavid Lebrun 	int (*put)(struct sk_buff *skb, struct seg6_local_lwt *slwt);
1871d1df6fd8SDavid Lebrun 	int (*cmp)(struct seg6_local_lwt *a, struct seg6_local_lwt *b);
1872964adce5SAndrea Mayer 
1873964adce5SAndrea Mayer 	/* optional destroy() callback useful for releasing resources which
1874964adce5SAndrea Mayer 	 * have been previously acquired in the corresponding parse()
1875964adce5SAndrea Mayer 	 * function.
1876964adce5SAndrea Mayer 	 */
1877964adce5SAndrea Mayer 	void (*destroy)(struct seg6_local_lwt *slwt);
1878d1df6fd8SDavid Lebrun };
1879d1df6fd8SDavid Lebrun 
1880d1df6fd8SDavid Lebrun static struct seg6_action_param seg6_action_params[SEG6_LOCAL_MAX + 1] = {
18812d9cc60aSDavid Lebrun 	[SEG6_LOCAL_SRH]	= { .parse = parse_nla_srh,
18822d9cc60aSDavid Lebrun 				    .put = put_nla_srh,
1883964adce5SAndrea Mayer 				    .cmp = cmp_nla_srh,
1884964adce5SAndrea Mayer 				    .destroy = destroy_attr_srh },
1885d1df6fd8SDavid Lebrun 
18862d9cc60aSDavid Lebrun 	[SEG6_LOCAL_TABLE]	= { .parse = parse_nla_table,
18872d9cc60aSDavid Lebrun 				    .put = put_nla_table,
18882d9cc60aSDavid Lebrun 				    .cmp = cmp_nla_table },
1889d1df6fd8SDavid Lebrun 
18902d9cc60aSDavid Lebrun 	[SEG6_LOCAL_NH4]	= { .parse = parse_nla_nh4,
18912d9cc60aSDavid Lebrun 				    .put = put_nla_nh4,
18922d9cc60aSDavid Lebrun 				    .cmp = cmp_nla_nh4 },
1893d1df6fd8SDavid Lebrun 
18942d9cc60aSDavid Lebrun 	[SEG6_LOCAL_NH6]	= { .parse = parse_nla_nh6,
18952d9cc60aSDavid Lebrun 				    .put = put_nla_nh6,
18962d9cc60aSDavid Lebrun 				    .cmp = cmp_nla_nh6 },
1897d1df6fd8SDavid Lebrun 
18982d9cc60aSDavid Lebrun 	[SEG6_LOCAL_IIF]	= { .parse = parse_nla_iif,
18992d9cc60aSDavid Lebrun 				    .put = put_nla_iif,
19002d9cc60aSDavid Lebrun 				    .cmp = cmp_nla_iif },
1901d1df6fd8SDavid Lebrun 
19022d9cc60aSDavid Lebrun 	[SEG6_LOCAL_OIF]	= { .parse = parse_nla_oif,
19032d9cc60aSDavid Lebrun 				    .put = put_nla_oif,
19042d9cc60aSDavid Lebrun 				    .cmp = cmp_nla_oif },
1905004d4b27SMathieu Xhonneux 
1906004d4b27SMathieu Xhonneux 	[SEG6_LOCAL_BPF]	= { .parse = parse_nla_bpf,
1907004d4b27SMathieu Xhonneux 				    .put = put_nla_bpf,
1908964adce5SAndrea Mayer 				    .cmp = cmp_nla_bpf,
1909964adce5SAndrea Mayer 				    .destroy = destroy_attr_bpf },
1910004d4b27SMathieu Xhonneux 
1911664d6f86SAndrea Mayer 	[SEG6_LOCAL_VRFTABLE]	= { .parse = parse_nla_vrftable,
1912664d6f86SAndrea Mayer 				    .put = put_nla_vrftable,
1913664d6f86SAndrea Mayer 				    .cmp = cmp_nla_vrftable },
1914664d6f86SAndrea Mayer 
191594604548SAndrea Mayer 	[SEG6_LOCAL_COUNTERS]	= { .parse = parse_nla_counters,
191694604548SAndrea Mayer 				    .put = put_nla_counters,
191794604548SAndrea Mayer 				    .cmp = cmp_nla_counters,
191894604548SAndrea Mayer 				    .destroy = destroy_attr_counters },
1919*848f3c0dSAndrea Mayer 
1920*848f3c0dSAndrea Mayer 	[SEG6_LOCAL_FLAVORS]	= { .parse = parse_nla_flavors,
1921*848f3c0dSAndrea Mayer 				    .put = put_nla_flavors,
1922*848f3c0dSAndrea Mayer 				    .cmp = cmp_nla_flavors },
1923d1df6fd8SDavid Lebrun };
1924d1df6fd8SDavid Lebrun 
1925964adce5SAndrea Mayer /* call the destroy() callback (if available) for each set attribute in
19260a3021f1SAndrea Mayer  * @parsed_attrs, starting from the first attribute up to the @max_parsed
19270a3021f1SAndrea Mayer  * (excluded) attribute.
1928964adce5SAndrea Mayer  */
19290a3021f1SAndrea Mayer static void __destroy_attrs(unsigned long parsed_attrs, int max_parsed,
19300a3021f1SAndrea Mayer 			    struct seg6_local_lwt *slwt)
1931964adce5SAndrea Mayer {
1932964adce5SAndrea Mayer 	struct seg6_action_param *param;
1933964adce5SAndrea Mayer 	int i;
1934964adce5SAndrea Mayer 
1935964adce5SAndrea Mayer 	/* Every required seg6local attribute is identified by an ID which is
1936964adce5SAndrea Mayer 	 * encoded as a flag (i.e: 1 << ID) in the 'attrs' bitmask;
1937964adce5SAndrea Mayer 	 *
19380a3021f1SAndrea Mayer 	 * We scan the 'parsed_attrs' bitmask, starting from the first attribute
1939964adce5SAndrea Mayer 	 * up to the @max_parsed (excluded) attribute.
1940964adce5SAndrea Mayer 	 * For each set attribute, we retrieve the corresponding destroy()
1941964adce5SAndrea Mayer 	 * callback. If the callback is not available, then we skip to the next
1942964adce5SAndrea Mayer 	 * attribute; otherwise, we call the destroy() callback.
1943964adce5SAndrea Mayer 	 */
1944ac0dbed9SNick Desaulniers 	for (i = SEG6_LOCAL_SRH; i < max_parsed; ++i) {
1945300a0fd8SAndrea Mayer 		if (!(parsed_attrs & SEG6_F_ATTR(i)))
1946964adce5SAndrea Mayer 			continue;
1947964adce5SAndrea Mayer 
1948964adce5SAndrea Mayer 		param = &seg6_action_params[i];
1949964adce5SAndrea Mayer 
1950964adce5SAndrea Mayer 		if (param->destroy)
1951964adce5SAndrea Mayer 			param->destroy(slwt);
1952964adce5SAndrea Mayer 	}
1953964adce5SAndrea Mayer }
1954964adce5SAndrea Mayer 
1955964adce5SAndrea Mayer /* release all the resources that may have been acquired during parsing
1956964adce5SAndrea Mayer  * operations.
1957964adce5SAndrea Mayer  */
1958964adce5SAndrea Mayer static void destroy_attrs(struct seg6_local_lwt *slwt)
1959964adce5SAndrea Mayer {
19600a3021f1SAndrea Mayer 	unsigned long attrs = slwt->desc->attrs | slwt->parsed_optattrs;
19610a3021f1SAndrea Mayer 
19620a3021f1SAndrea Mayer 	__destroy_attrs(attrs, SEG6_LOCAL_MAX + 1, slwt);
19630a3021f1SAndrea Mayer }
19640a3021f1SAndrea Mayer 
19650a3021f1SAndrea Mayer static int parse_nla_optional_attrs(struct nlattr **attrs,
1966e2a8ecc4SAndrea Mayer 				    struct seg6_local_lwt *slwt,
1967e2a8ecc4SAndrea Mayer 				    struct netlink_ext_ack *extack)
19680a3021f1SAndrea Mayer {
19690a3021f1SAndrea Mayer 	struct seg6_action_desc *desc = slwt->desc;
19700a3021f1SAndrea Mayer 	unsigned long parsed_optattrs = 0;
19710a3021f1SAndrea Mayer 	struct seg6_action_param *param;
19720a3021f1SAndrea Mayer 	int err, i;
19730a3021f1SAndrea Mayer 
1974ac0dbed9SNick Desaulniers 	for (i = SEG6_LOCAL_SRH; i < SEG6_LOCAL_MAX + 1; ++i) {
1975300a0fd8SAndrea Mayer 		if (!(desc->optattrs & SEG6_F_ATTR(i)) || !attrs[i])
19760a3021f1SAndrea Mayer 			continue;
19770a3021f1SAndrea Mayer 
19780a3021f1SAndrea Mayer 		/* once here, the i-th attribute is provided by the
19790a3021f1SAndrea Mayer 		 * userspace AND it is identified optional as well.
19800a3021f1SAndrea Mayer 		 */
19810a3021f1SAndrea Mayer 		param = &seg6_action_params[i];
19820a3021f1SAndrea Mayer 
1983e2a8ecc4SAndrea Mayer 		err = param->parse(attrs, slwt, extack);
19840a3021f1SAndrea Mayer 		if (err < 0)
19850a3021f1SAndrea Mayer 			goto parse_optattrs_err;
19860a3021f1SAndrea Mayer 
19870a3021f1SAndrea Mayer 		/* current attribute has been correctly parsed */
1988300a0fd8SAndrea Mayer 		parsed_optattrs |= SEG6_F_ATTR(i);
19890a3021f1SAndrea Mayer 	}
19900a3021f1SAndrea Mayer 
19910a3021f1SAndrea Mayer 	/* store in the tunnel state all the optional attributed successfully
19920a3021f1SAndrea Mayer 	 * parsed.
19930a3021f1SAndrea Mayer 	 */
19940a3021f1SAndrea Mayer 	slwt->parsed_optattrs = parsed_optattrs;
19950a3021f1SAndrea Mayer 
19960a3021f1SAndrea Mayer 	return 0;
19970a3021f1SAndrea Mayer 
19980a3021f1SAndrea Mayer parse_optattrs_err:
19990a3021f1SAndrea Mayer 	__destroy_attrs(parsed_optattrs, i, slwt);
20000a3021f1SAndrea Mayer 
20010a3021f1SAndrea Mayer 	return err;
2002964adce5SAndrea Mayer }
2003964adce5SAndrea Mayer 
2004cfdf64a0SAndrea Mayer /* call the custom constructor of the behavior during its initialization phase
2005cfdf64a0SAndrea Mayer  * and after that all its attributes have been parsed successfully.
2006cfdf64a0SAndrea Mayer  */
2007cfdf64a0SAndrea Mayer static int
2008cfdf64a0SAndrea Mayer seg6_local_lwtunnel_build_state(struct seg6_local_lwt *slwt, const void *cfg,
2009cfdf64a0SAndrea Mayer 				struct netlink_ext_ack *extack)
2010cfdf64a0SAndrea Mayer {
2011cfdf64a0SAndrea Mayer 	struct seg6_action_desc *desc = slwt->desc;
2012cfdf64a0SAndrea Mayer 	struct seg6_local_lwtunnel_ops *ops;
2013cfdf64a0SAndrea Mayer 
2014cfdf64a0SAndrea Mayer 	ops = &desc->slwt_ops;
2015cfdf64a0SAndrea Mayer 	if (!ops->build_state)
2016cfdf64a0SAndrea Mayer 		return 0;
2017cfdf64a0SAndrea Mayer 
2018cfdf64a0SAndrea Mayer 	return ops->build_state(slwt, cfg, extack);
2019cfdf64a0SAndrea Mayer }
2020cfdf64a0SAndrea Mayer 
2021cfdf64a0SAndrea Mayer /* call the custom destructor of the behavior which is invoked before the
2022cfdf64a0SAndrea Mayer  * tunnel is going to be destroyed.
2023cfdf64a0SAndrea Mayer  */
2024cfdf64a0SAndrea Mayer static void seg6_local_lwtunnel_destroy_state(struct seg6_local_lwt *slwt)
2025cfdf64a0SAndrea Mayer {
2026cfdf64a0SAndrea Mayer 	struct seg6_action_desc *desc = slwt->desc;
2027cfdf64a0SAndrea Mayer 	struct seg6_local_lwtunnel_ops *ops;
2028cfdf64a0SAndrea Mayer 
2029cfdf64a0SAndrea Mayer 	ops = &desc->slwt_ops;
2030cfdf64a0SAndrea Mayer 	if (!ops->destroy_state)
2031cfdf64a0SAndrea Mayer 		return;
2032cfdf64a0SAndrea Mayer 
2033cfdf64a0SAndrea Mayer 	ops->destroy_state(slwt);
2034cfdf64a0SAndrea Mayer }
2035cfdf64a0SAndrea Mayer 
2036e2a8ecc4SAndrea Mayer static int parse_nla_action(struct nlattr **attrs, struct seg6_local_lwt *slwt,
2037e2a8ecc4SAndrea Mayer 			    struct netlink_ext_ack *extack)
2038d1df6fd8SDavid Lebrun {
2039d1df6fd8SDavid Lebrun 	struct seg6_action_param *param;
2040d1df6fd8SDavid Lebrun 	struct seg6_action_desc *desc;
20410a3021f1SAndrea Mayer 	unsigned long invalid_attrs;
2042d1df6fd8SDavid Lebrun 	int i, err;
2043d1df6fd8SDavid Lebrun 
2044d1df6fd8SDavid Lebrun 	desc = __get_action_desc(slwt->action);
2045d1df6fd8SDavid Lebrun 	if (!desc)
2046d1df6fd8SDavid Lebrun 		return -EINVAL;
2047d1df6fd8SDavid Lebrun 
2048d1df6fd8SDavid Lebrun 	if (!desc->input)
2049d1df6fd8SDavid Lebrun 		return -EOPNOTSUPP;
2050d1df6fd8SDavid Lebrun 
2051d1df6fd8SDavid Lebrun 	slwt->desc = desc;
2052d1df6fd8SDavid Lebrun 	slwt->headroom += desc->static_headroom;
2053d1df6fd8SDavid Lebrun 
20540a3021f1SAndrea Mayer 	/* Forcing the desc->optattrs *set* and the desc->attrs *set* to be
20550a3021f1SAndrea Mayer 	 * disjoined, this allow us to release acquired resources by optional
20560a3021f1SAndrea Mayer 	 * attributes and by required attributes independently from each other
20570d770360SAndrea Mayer 	 * without any interference.
20580a3021f1SAndrea Mayer 	 * In other terms, we are sure that we do not release some the acquired
20590a3021f1SAndrea Mayer 	 * resources twice.
20600a3021f1SAndrea Mayer 	 *
20610a3021f1SAndrea Mayer 	 * Note that if an attribute is configured both as required and as
20620a3021f1SAndrea Mayer 	 * optional, it means that the user has messed something up in the
20630a3021f1SAndrea Mayer 	 * seg6_action_table. Therefore, this check is required for SRv6
20640a3021f1SAndrea Mayer 	 * behaviors to work properly.
20650a3021f1SAndrea Mayer 	 */
20660a3021f1SAndrea Mayer 	invalid_attrs = desc->attrs & desc->optattrs;
20670a3021f1SAndrea Mayer 	if (invalid_attrs) {
20680a3021f1SAndrea Mayer 		WARN_ONCE(1,
20690a3021f1SAndrea Mayer 			  "An attribute cannot be both required AND optional");
20700a3021f1SAndrea Mayer 		return -EINVAL;
20710a3021f1SAndrea Mayer 	}
20720a3021f1SAndrea Mayer 
20730a3021f1SAndrea Mayer 	/* parse the required attributes */
2074ac0dbed9SNick Desaulniers 	for (i = SEG6_LOCAL_SRH; i < SEG6_LOCAL_MAX + 1; i++) {
2075300a0fd8SAndrea Mayer 		if (desc->attrs & SEG6_F_ATTR(i)) {
2076d1df6fd8SDavid Lebrun 			if (!attrs[i])
2077d1df6fd8SDavid Lebrun 				return -EINVAL;
2078d1df6fd8SDavid Lebrun 
2079d1df6fd8SDavid Lebrun 			param = &seg6_action_params[i];
2080d1df6fd8SDavid Lebrun 
2081e2a8ecc4SAndrea Mayer 			err = param->parse(attrs, slwt, extack);
2082d1df6fd8SDavid Lebrun 			if (err < 0)
20830a3021f1SAndrea Mayer 				goto parse_attrs_err;
2084d1df6fd8SDavid Lebrun 		}
2085d1df6fd8SDavid Lebrun 	}
2086d1df6fd8SDavid Lebrun 
20870a3021f1SAndrea Mayer 	/* parse the optional attributes, if any */
2088e2a8ecc4SAndrea Mayer 	err = parse_nla_optional_attrs(attrs, slwt, extack);
20890a3021f1SAndrea Mayer 	if (err < 0)
20900a3021f1SAndrea Mayer 		goto parse_attrs_err;
20910a3021f1SAndrea Mayer 
2092d1df6fd8SDavid Lebrun 	return 0;
2093964adce5SAndrea Mayer 
20940a3021f1SAndrea Mayer parse_attrs_err:
2095964adce5SAndrea Mayer 	/* release any resource that may have been acquired during the i-1
2096964adce5SAndrea Mayer 	 * parse() operations.
2097964adce5SAndrea Mayer 	 */
20980a3021f1SAndrea Mayer 	__destroy_attrs(desc->attrs, i, slwt);
2099964adce5SAndrea Mayer 
2100964adce5SAndrea Mayer 	return err;
2101d1df6fd8SDavid Lebrun }
2102d1df6fd8SDavid Lebrun 
2103faee6769SAlexander Aring static int seg6_local_build_state(struct net *net, struct nlattr *nla,
2104faee6769SAlexander Aring 				  unsigned int family, const void *cfg,
2105faee6769SAlexander Aring 				  struct lwtunnel_state **ts,
2106d1df6fd8SDavid Lebrun 				  struct netlink_ext_ack *extack)
2107d1df6fd8SDavid Lebrun {
2108d1df6fd8SDavid Lebrun 	struct nlattr *tb[SEG6_LOCAL_MAX + 1];
2109d1df6fd8SDavid Lebrun 	struct lwtunnel_state *newts;
2110d1df6fd8SDavid Lebrun 	struct seg6_local_lwt *slwt;
2111d1df6fd8SDavid Lebrun 	int err;
2112d1df6fd8SDavid Lebrun 
21136285217fSDavid Lebrun 	if (family != AF_INET6)
21146285217fSDavid Lebrun 		return -EINVAL;
21156285217fSDavid Lebrun 
21168cb08174SJohannes Berg 	err = nla_parse_nested_deprecated(tb, SEG6_LOCAL_MAX, nla,
21178cb08174SJohannes Berg 					  seg6_local_policy, extack);
2118d1df6fd8SDavid Lebrun 
2119d1df6fd8SDavid Lebrun 	if (err < 0)
2120d1df6fd8SDavid Lebrun 		return err;
2121d1df6fd8SDavid Lebrun 
2122d1df6fd8SDavid Lebrun 	if (!tb[SEG6_LOCAL_ACTION])
2123d1df6fd8SDavid Lebrun 		return -EINVAL;
2124d1df6fd8SDavid Lebrun 
2125d1df6fd8SDavid Lebrun 	newts = lwtunnel_state_alloc(sizeof(*slwt));
2126d1df6fd8SDavid Lebrun 	if (!newts)
2127d1df6fd8SDavid Lebrun 		return -ENOMEM;
2128d1df6fd8SDavid Lebrun 
2129d1df6fd8SDavid Lebrun 	slwt = seg6_local_lwtunnel(newts);
2130d1df6fd8SDavid Lebrun 	slwt->action = nla_get_u32(tb[SEG6_LOCAL_ACTION]);
2131d1df6fd8SDavid Lebrun 
2132e2a8ecc4SAndrea Mayer 	err = parse_nla_action(tb, slwt, extack);
2133d1df6fd8SDavid Lebrun 	if (err < 0)
2134d1df6fd8SDavid Lebrun 		goto out_free;
2135d1df6fd8SDavid Lebrun 
2136cfdf64a0SAndrea Mayer 	err = seg6_local_lwtunnel_build_state(slwt, cfg, extack);
2137cfdf64a0SAndrea Mayer 	if (err < 0)
2138cfdf64a0SAndrea Mayer 		goto out_destroy_attrs;
2139cfdf64a0SAndrea Mayer 
2140d1df6fd8SDavid Lebrun 	newts->type = LWTUNNEL_ENCAP_SEG6_LOCAL;
2141d1df6fd8SDavid Lebrun 	newts->flags = LWTUNNEL_STATE_INPUT_REDIRECT;
2142d1df6fd8SDavid Lebrun 	newts->headroom = slwt->headroom;
2143d1df6fd8SDavid Lebrun 
2144d1df6fd8SDavid Lebrun 	*ts = newts;
2145d1df6fd8SDavid Lebrun 
2146d1df6fd8SDavid Lebrun 	return 0;
2147d1df6fd8SDavid Lebrun 
2148cfdf64a0SAndrea Mayer out_destroy_attrs:
2149cfdf64a0SAndrea Mayer 	destroy_attrs(slwt);
2150d1df6fd8SDavid Lebrun out_free:
2151d1df6fd8SDavid Lebrun 	kfree(newts);
2152d1df6fd8SDavid Lebrun 	return err;
2153d1df6fd8SDavid Lebrun }
2154d1df6fd8SDavid Lebrun 
2155d1df6fd8SDavid Lebrun static void seg6_local_destroy_state(struct lwtunnel_state *lwt)
2156d1df6fd8SDavid Lebrun {
2157d1df6fd8SDavid Lebrun 	struct seg6_local_lwt *slwt = seg6_local_lwtunnel(lwt);
2158d1df6fd8SDavid Lebrun 
2159cfdf64a0SAndrea Mayer 	seg6_local_lwtunnel_destroy_state(slwt);
2160cfdf64a0SAndrea Mayer 
2161964adce5SAndrea Mayer 	destroy_attrs(slwt);
2162004d4b27SMathieu Xhonneux 
2163004d4b27SMathieu Xhonneux 	return;
2164d1df6fd8SDavid Lebrun }
2165d1df6fd8SDavid Lebrun 
2166d1df6fd8SDavid Lebrun static int seg6_local_fill_encap(struct sk_buff *skb,
2167d1df6fd8SDavid Lebrun 				 struct lwtunnel_state *lwt)
2168d1df6fd8SDavid Lebrun {
2169d1df6fd8SDavid Lebrun 	struct seg6_local_lwt *slwt = seg6_local_lwtunnel(lwt);
2170d1df6fd8SDavid Lebrun 	struct seg6_action_param *param;
21710a3021f1SAndrea Mayer 	unsigned long attrs;
2172d1df6fd8SDavid Lebrun 	int i, err;
2173d1df6fd8SDavid Lebrun 
2174d1df6fd8SDavid Lebrun 	if (nla_put_u32(skb, SEG6_LOCAL_ACTION, slwt->action))
2175d1df6fd8SDavid Lebrun 		return -EMSGSIZE;
2176d1df6fd8SDavid Lebrun 
21770a3021f1SAndrea Mayer 	attrs = slwt->desc->attrs | slwt->parsed_optattrs;
21780a3021f1SAndrea Mayer 
2179ac0dbed9SNick Desaulniers 	for (i = SEG6_LOCAL_SRH; i < SEG6_LOCAL_MAX + 1; i++) {
2180300a0fd8SAndrea Mayer 		if (attrs & SEG6_F_ATTR(i)) {
2181d1df6fd8SDavid Lebrun 			param = &seg6_action_params[i];
2182d1df6fd8SDavid Lebrun 			err = param->put(skb, slwt);
2183d1df6fd8SDavid Lebrun 			if (err < 0)
2184d1df6fd8SDavid Lebrun 				return err;
2185d1df6fd8SDavid Lebrun 		}
2186d1df6fd8SDavid Lebrun 	}
2187d1df6fd8SDavid Lebrun 
2188d1df6fd8SDavid Lebrun 	return 0;
2189d1df6fd8SDavid Lebrun }
2190d1df6fd8SDavid Lebrun 
2191d1df6fd8SDavid Lebrun static int seg6_local_get_encap_size(struct lwtunnel_state *lwt)
2192d1df6fd8SDavid Lebrun {
2193d1df6fd8SDavid Lebrun 	struct seg6_local_lwt *slwt = seg6_local_lwtunnel(lwt);
2194d1df6fd8SDavid Lebrun 	unsigned long attrs;
2195d1df6fd8SDavid Lebrun 	int nlsize;
2196d1df6fd8SDavid Lebrun 
2197d1df6fd8SDavid Lebrun 	nlsize = nla_total_size(4); /* action */
2198d1df6fd8SDavid Lebrun 
21990a3021f1SAndrea Mayer 	attrs = slwt->desc->attrs | slwt->parsed_optattrs;
2200d1df6fd8SDavid Lebrun 
2201300a0fd8SAndrea Mayer 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_SRH))
2202d1df6fd8SDavid Lebrun 		nlsize += nla_total_size((slwt->srh->hdrlen + 1) << 3);
2203d1df6fd8SDavid Lebrun 
2204300a0fd8SAndrea Mayer 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_TABLE))
2205d1df6fd8SDavid Lebrun 		nlsize += nla_total_size(4);
2206d1df6fd8SDavid Lebrun 
2207300a0fd8SAndrea Mayer 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_NH4))
2208d1df6fd8SDavid Lebrun 		nlsize += nla_total_size(4);
2209d1df6fd8SDavid Lebrun 
2210300a0fd8SAndrea Mayer 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_NH6))
2211d1df6fd8SDavid Lebrun 		nlsize += nla_total_size(16);
2212d1df6fd8SDavid Lebrun 
2213300a0fd8SAndrea Mayer 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_IIF))
2214d1df6fd8SDavid Lebrun 		nlsize += nla_total_size(4);
2215d1df6fd8SDavid Lebrun 
2216300a0fd8SAndrea Mayer 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_OIF))
2217d1df6fd8SDavid Lebrun 		nlsize += nla_total_size(4);
2218d1df6fd8SDavid Lebrun 
2219300a0fd8SAndrea Mayer 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_BPF))
2220004d4b27SMathieu Xhonneux 		nlsize += nla_total_size(sizeof(struct nlattr)) +
2221004d4b27SMathieu Xhonneux 		       nla_total_size(MAX_PROG_NAME) +
2222004d4b27SMathieu Xhonneux 		       nla_total_size(4);
2223004d4b27SMathieu Xhonneux 
2224300a0fd8SAndrea Mayer 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_VRFTABLE))
2225664d6f86SAndrea Mayer 		nlsize += nla_total_size(4);
2226664d6f86SAndrea Mayer 
222794604548SAndrea Mayer 	if (attrs & SEG6_F_LOCAL_COUNTERS)
222894604548SAndrea Mayer 		nlsize += nla_total_size(0) + /* nest SEG6_LOCAL_COUNTERS */
222994604548SAndrea Mayer 			  /* SEG6_LOCAL_CNT_PACKETS */
223094604548SAndrea Mayer 			  nla_total_size_64bit(sizeof(__u64)) +
223194604548SAndrea Mayer 			  /* SEG6_LOCAL_CNT_BYTES */
223294604548SAndrea Mayer 			  nla_total_size_64bit(sizeof(__u64)) +
223394604548SAndrea Mayer 			  /* SEG6_LOCAL_CNT_ERRORS */
223494604548SAndrea Mayer 			  nla_total_size_64bit(sizeof(__u64));
223594604548SAndrea Mayer 
2236*848f3c0dSAndrea Mayer 	if (attrs & SEG6_F_ATTR(SEG6_LOCAL_FLAVORS))
2237*848f3c0dSAndrea Mayer 		nlsize += encap_size_flavors(slwt);
2238*848f3c0dSAndrea Mayer 
2239d1df6fd8SDavid Lebrun 	return nlsize;
2240d1df6fd8SDavid Lebrun }
2241d1df6fd8SDavid Lebrun 
2242d1df6fd8SDavid Lebrun static int seg6_local_cmp_encap(struct lwtunnel_state *a,
2243d1df6fd8SDavid Lebrun 				struct lwtunnel_state *b)
2244d1df6fd8SDavid Lebrun {
2245d1df6fd8SDavid Lebrun 	struct seg6_local_lwt *slwt_a, *slwt_b;
2246d1df6fd8SDavid Lebrun 	struct seg6_action_param *param;
22470a3021f1SAndrea Mayer 	unsigned long attrs_a, attrs_b;
2248d1df6fd8SDavid Lebrun 	int i;
2249d1df6fd8SDavid Lebrun 
2250d1df6fd8SDavid Lebrun 	slwt_a = seg6_local_lwtunnel(a);
2251d1df6fd8SDavid Lebrun 	slwt_b = seg6_local_lwtunnel(b);
2252d1df6fd8SDavid Lebrun 
2253d1df6fd8SDavid Lebrun 	if (slwt_a->action != slwt_b->action)
2254d1df6fd8SDavid Lebrun 		return 1;
2255d1df6fd8SDavid Lebrun 
22560a3021f1SAndrea Mayer 	attrs_a = slwt_a->desc->attrs | slwt_a->parsed_optattrs;
22570a3021f1SAndrea Mayer 	attrs_b = slwt_b->desc->attrs | slwt_b->parsed_optattrs;
22580a3021f1SAndrea Mayer 
22590a3021f1SAndrea Mayer 	if (attrs_a != attrs_b)
2260d1df6fd8SDavid Lebrun 		return 1;
2261d1df6fd8SDavid Lebrun 
2262ac0dbed9SNick Desaulniers 	for (i = SEG6_LOCAL_SRH; i < SEG6_LOCAL_MAX + 1; i++) {
2263300a0fd8SAndrea Mayer 		if (attrs_a & SEG6_F_ATTR(i)) {
2264d1df6fd8SDavid Lebrun 			param = &seg6_action_params[i];
2265d1df6fd8SDavid Lebrun 			if (param->cmp(slwt_a, slwt_b))
2266d1df6fd8SDavid Lebrun 				return 1;
2267d1df6fd8SDavid Lebrun 		}
2268d1df6fd8SDavid Lebrun 	}
2269d1df6fd8SDavid Lebrun 
2270d1df6fd8SDavid Lebrun 	return 0;
2271d1df6fd8SDavid Lebrun }
2272d1df6fd8SDavid Lebrun 
2273d1df6fd8SDavid Lebrun static const struct lwtunnel_encap_ops seg6_local_ops = {
2274d1df6fd8SDavid Lebrun 	.build_state	= seg6_local_build_state,
2275d1df6fd8SDavid Lebrun 	.destroy_state	= seg6_local_destroy_state,
2276d1df6fd8SDavid Lebrun 	.input		= seg6_local_input,
2277d1df6fd8SDavid Lebrun 	.fill_encap	= seg6_local_fill_encap,
2278d1df6fd8SDavid Lebrun 	.get_encap_size	= seg6_local_get_encap_size,
2279d1df6fd8SDavid Lebrun 	.cmp_encap	= seg6_local_cmp_encap,
2280d1df6fd8SDavid Lebrun 	.owner		= THIS_MODULE,
2281d1df6fd8SDavid Lebrun };
2282d1df6fd8SDavid Lebrun 
2283d1df6fd8SDavid Lebrun int __init seg6_local_init(void)
2284d1df6fd8SDavid Lebrun {
2285300a0fd8SAndrea Mayer 	/* If the max total number of defined attributes is reached, then your
2286300a0fd8SAndrea Mayer 	 * kernel build stops here.
2287300a0fd8SAndrea Mayer 	 *
2288300a0fd8SAndrea Mayer 	 * This check is required to avoid arithmetic overflows when processing
2289300a0fd8SAndrea Mayer 	 * behavior attributes and the maximum number of defined attributes
2290300a0fd8SAndrea Mayer 	 * exceeds the allowed value.
2291300a0fd8SAndrea Mayer 	 */
2292300a0fd8SAndrea Mayer 	BUILD_BUG_ON(SEG6_LOCAL_MAX + 1 > BITS_PER_TYPE(unsigned long));
2293300a0fd8SAndrea Mayer 
2294*848f3c0dSAndrea Mayer 	/* If the default NEXT-C-SID Locator-Block/Node Function lengths (in
2295*848f3c0dSAndrea Mayer 	 * bits) have been changed with invalid values, kernel build stops
2296*848f3c0dSAndrea Mayer 	 * here.
2297*848f3c0dSAndrea Mayer 	 */
2298*848f3c0dSAndrea Mayer 	BUILD_BUG_ON(next_csid_chk_cntr_bits(SEG6_LOCAL_LCBLOCK_DBITS,
2299*848f3c0dSAndrea Mayer 					     SEG6_LOCAL_LCNODE_FN_DBITS));
2300*848f3c0dSAndrea Mayer 	BUILD_BUG_ON(next_csid_chk_lcblock_bits(SEG6_LOCAL_LCBLOCK_DBITS));
2301*848f3c0dSAndrea Mayer 	BUILD_BUG_ON(next_csid_chk_lcnode_fn_bits(SEG6_LOCAL_LCNODE_FN_DBITS));
2302*848f3c0dSAndrea Mayer 
2303d1df6fd8SDavid Lebrun 	return lwtunnel_encap_add_ops(&seg6_local_ops,
2304d1df6fd8SDavid Lebrun 				      LWTUNNEL_ENCAP_SEG6_LOCAL);
2305d1df6fd8SDavid Lebrun }
2306d1df6fd8SDavid Lebrun 
2307d1df6fd8SDavid Lebrun void seg6_local_exit(void)
2308d1df6fd8SDavid Lebrun {
2309d1df6fd8SDavid Lebrun 	lwtunnel_encap_del_ops(&seg6_local_ops, LWTUNNEL_ENCAP_SEG6_LOCAL);
2310d1df6fd8SDavid Lebrun }
2311