Linux Audio

Check our new training course

Loading...
v4.17
 
  1/*
  2 *  SR-IPv6 implementation
  3 *
  4 *  Author:
  5 *  David Lebrun <david.lebrun@uclouvain.be>
  6 *
  7 *
  8 *  This program is free software; you can redistribute it and/or
  9 *	  modify it under the terms of the GNU General Public License
 10 *	  as published by the Free Software Foundation; either version
 11 *	  2 of the License, or (at your option) any later version.
 12 */
 13
 14#include <linux/errno.h>
 15#include <linux/types.h>
 16#include <linux/socket.h>
 17#include <linux/net.h>
 18#include <linux/in6.h>
 19#include <linux/slab.h>
 
 20
 21#include <net/ipv6.h>
 22#include <net/protocol.h>
 23
 24#include <net/seg6.h>
 25#include <net/genetlink.h>
 26#include <linux/seg6.h>
 27#include <linux/seg6_genl.h>
 28#ifdef CONFIG_IPV6_SEG6_HMAC
 29#include <net/seg6_hmac.h>
 30#endif
 31
 32bool seg6_validate_srh(struct ipv6_sr_hdr *srh, int len)
 33{
 34	int trailing;
 35	unsigned int tlv_offset;
 
 
 36
 37	if (srh->type != IPV6_SRCRT_TYPE_4)
 38		return false;
 39
 40	if (((srh->hdrlen + 1) << 3) != len)
 41		return false;
 42
 43	if (srh->segments_left > srh->first_segment)
 44		return false;
 
 
 
 
 
 
 
 
 
 45
 46	tlv_offset = sizeof(*srh) + ((srh->first_segment + 1) << 4);
 47
 48	trailing = len - tlv_offset;
 49	if (trailing < 0)
 50		return false;
 51
 52	while (trailing) {
 53		struct sr6_tlv *tlv;
 54		unsigned int tlv_len;
 55
 56		if (trailing < sizeof(*tlv))
 57			return false;
 58
 59		tlv = (struct sr6_tlv *)((unsigned char *)srh + tlv_offset);
 60		tlv_len = sizeof(*tlv) + tlv->len;
 61
 62		trailing -= tlv_len;
 63		if (trailing < 0)
 64			return false;
 65
 66		tlv_offset += tlv_len;
 67	}
 68
 69	return true;
 70}
 71
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 72static struct genl_family seg6_genl_family;
 73
 74static const struct nla_policy seg6_genl_policy[SEG6_ATTR_MAX + 1] = {
 75	[SEG6_ATTR_DST]				= { .type = NLA_BINARY,
 76		.len = sizeof(struct in6_addr) },
 77	[SEG6_ATTR_DSTLEN]			= { .type = NLA_S32, },
 78	[SEG6_ATTR_HMACKEYID]		= { .type = NLA_U32, },
 79	[SEG6_ATTR_SECRET]			= { .type = NLA_BINARY, },
 80	[SEG6_ATTR_SECRETLEN]		= { .type = NLA_U8, },
 81	[SEG6_ATTR_ALGID]			= { .type = NLA_U8, },
 82	[SEG6_ATTR_HMACINFO]		= { .type = NLA_NESTED, },
 83};
 84
 85#ifdef CONFIG_IPV6_SEG6_HMAC
 86
 87static int seg6_genl_sethmac(struct sk_buff *skb, struct genl_info *info)
 88{
 89	struct net *net = genl_info_net(info);
 90	struct seg6_pernet_data *sdata;
 91	struct seg6_hmac_info *hinfo;
 92	u32 hmackeyid;
 93	char *secret;
 94	int err = 0;
 95	u8 algid;
 96	u8 slen;
 97
 98	sdata = seg6_pernet(net);
 99
100	if (!info->attrs[SEG6_ATTR_HMACKEYID] ||
101	    !info->attrs[SEG6_ATTR_SECRETLEN] ||
102	    !info->attrs[SEG6_ATTR_ALGID])
103		return -EINVAL;
104
105	hmackeyid = nla_get_u32(info->attrs[SEG6_ATTR_HMACKEYID]);
106	slen = nla_get_u8(info->attrs[SEG6_ATTR_SECRETLEN]);
107	algid = nla_get_u8(info->attrs[SEG6_ATTR_ALGID]);
108
109	if (hmackeyid == 0)
110		return -EINVAL;
111
112	if (slen > SEG6_HMAC_SECRET_LEN)
113		return -EINVAL;
114
115	mutex_lock(&sdata->lock);
116	hinfo = seg6_hmac_info_lookup(net, hmackeyid);
117
118	if (!slen) {
119		if (!hinfo)
120			err = -ENOENT;
121
122		err = seg6_hmac_info_del(net, hmackeyid);
123
124		goto out_unlock;
125	}
126
127	if (!info->attrs[SEG6_ATTR_SECRET]) {
128		err = -EINVAL;
129		goto out_unlock;
130	}
131
 
 
 
 
 
132	if (hinfo) {
133		err = seg6_hmac_info_del(net, hmackeyid);
134		if (err)
135			goto out_unlock;
136	}
137
138	secret = (char *)nla_data(info->attrs[SEG6_ATTR_SECRET]);
139
140	hinfo = kzalloc(sizeof(*hinfo), GFP_KERNEL);
141	if (!hinfo) {
142		err = -ENOMEM;
143		goto out_unlock;
144	}
145
146	memcpy(hinfo->secret, secret, slen);
147	hinfo->slen = slen;
148	hinfo->alg_id = algid;
149	hinfo->hmackeyid = hmackeyid;
150
151	err = seg6_hmac_info_add(net, hmackeyid, hinfo);
152	if (err)
153		kfree(hinfo);
154
155out_unlock:
156	mutex_unlock(&sdata->lock);
157	return err;
158}
159
160#else
161
162static int seg6_genl_sethmac(struct sk_buff *skb, struct genl_info *info)
163{
164	return -ENOTSUPP;
165}
166
167#endif
168
169static int seg6_genl_set_tunsrc(struct sk_buff *skb, struct genl_info *info)
170{
171	struct net *net = genl_info_net(info);
172	struct in6_addr *val, *t_old, *t_new;
173	struct seg6_pernet_data *sdata;
174
175	sdata = seg6_pernet(net);
176
177	if (!info->attrs[SEG6_ATTR_DST])
178		return -EINVAL;
179
180	val = nla_data(info->attrs[SEG6_ATTR_DST]);
181	t_new = kmemdup(val, sizeof(*val), GFP_KERNEL);
182	if (!t_new)
183		return -ENOMEM;
184
185	mutex_lock(&sdata->lock);
186
187	t_old = sdata->tun_src;
188	rcu_assign_pointer(sdata->tun_src, t_new);
189
190	mutex_unlock(&sdata->lock);
191
192	synchronize_net();
193	kfree(t_old);
194
195	return 0;
196}
197
198static int seg6_genl_get_tunsrc(struct sk_buff *skb, struct genl_info *info)
199{
200	struct net *net = genl_info_net(info);
201	struct in6_addr *tun_src;
202	struct sk_buff *msg;
203	void *hdr;
204
205	msg = genlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
206	if (!msg)
207		return -ENOMEM;
208
209	hdr = genlmsg_put(msg, info->snd_portid, info->snd_seq,
210			  &seg6_genl_family, 0, SEG6_CMD_GET_TUNSRC);
211	if (!hdr)
212		goto free_msg;
213
214	rcu_read_lock();
215	tun_src = rcu_dereference(seg6_pernet(net)->tun_src);
216
217	if (nla_put(msg, SEG6_ATTR_DST, sizeof(struct in6_addr), tun_src))
218		goto nla_put_failure;
219
220	rcu_read_unlock();
221
222	genlmsg_end(msg, hdr);
223	genlmsg_reply(msg, info);
224
225	return 0;
226
227nla_put_failure:
228	rcu_read_unlock();
229	genlmsg_cancel(msg, hdr);
230free_msg:
231	nlmsg_free(msg);
232	return -ENOMEM;
233}
234
235#ifdef CONFIG_IPV6_SEG6_HMAC
236
237static int __seg6_hmac_fill_info(struct seg6_hmac_info *hinfo,
238				 struct sk_buff *msg)
239{
240	if (nla_put_u32(msg, SEG6_ATTR_HMACKEYID, hinfo->hmackeyid) ||
241	    nla_put_u8(msg, SEG6_ATTR_SECRETLEN, hinfo->slen) ||
242	    nla_put(msg, SEG6_ATTR_SECRET, hinfo->slen, hinfo->secret) ||
243	    nla_put_u8(msg, SEG6_ATTR_ALGID, hinfo->alg_id))
244		return -1;
245
246	return 0;
247}
248
249static int __seg6_genl_dumphmac_element(struct seg6_hmac_info *hinfo,
250					u32 portid, u32 seq, u32 flags,
251					struct sk_buff *skb, u8 cmd)
252{
253	void *hdr;
254
255	hdr = genlmsg_put(skb, portid, seq, &seg6_genl_family, flags, cmd);
256	if (!hdr)
257		return -ENOMEM;
258
259	if (__seg6_hmac_fill_info(hinfo, skb) < 0)
260		goto nla_put_failure;
261
262	genlmsg_end(skb, hdr);
263	return 0;
264
265nla_put_failure:
266	genlmsg_cancel(skb, hdr);
267	return -EMSGSIZE;
268}
269
270static int seg6_genl_dumphmac_start(struct netlink_callback *cb)
271{
272	struct net *net = sock_net(cb->skb->sk);
273	struct seg6_pernet_data *sdata;
274	struct rhashtable_iter *iter;
275
276	sdata = seg6_pernet(net);
277	iter = (struct rhashtable_iter *)cb->args[0];
278
279	if (!iter) {
280		iter = kmalloc(sizeof(*iter), GFP_KERNEL);
281		if (!iter)
282			return -ENOMEM;
283
284		cb->args[0] = (long)iter;
285	}
286
287	rhashtable_walk_enter(&sdata->hmac_infos, iter);
288
289	return 0;
290}
291
292static int seg6_genl_dumphmac_done(struct netlink_callback *cb)
293{
294	struct rhashtable_iter *iter = (struct rhashtable_iter *)cb->args[0];
295
296	rhashtable_walk_exit(iter);
297
298	kfree(iter);
299
300	return 0;
301}
302
303static int seg6_genl_dumphmac(struct sk_buff *skb, struct netlink_callback *cb)
304{
305	struct rhashtable_iter *iter = (struct rhashtable_iter *)cb->args[0];
306	struct seg6_hmac_info *hinfo;
307	int ret;
308
309	rhashtable_walk_start(iter);
310
311	for (;;) {
312		hinfo = rhashtable_walk_next(iter);
313
314		if (IS_ERR(hinfo)) {
315			if (PTR_ERR(hinfo) == -EAGAIN)
316				continue;
317			ret = PTR_ERR(hinfo);
318			goto done;
319		} else if (!hinfo) {
320			break;
321		}
322
323		ret = __seg6_genl_dumphmac_element(hinfo,
324						   NETLINK_CB(cb->skb).portid,
325						   cb->nlh->nlmsg_seq,
326						   NLM_F_MULTI,
327						   skb, SEG6_CMD_DUMPHMAC);
328		if (ret)
329			goto done;
330	}
331
332	ret = skb->len;
333
334done:
335	rhashtable_walk_stop(iter);
336	return ret;
337}
338
339#else
340
341static int seg6_genl_dumphmac_start(struct netlink_callback *cb)
342{
343	return 0;
344}
345
346static int seg6_genl_dumphmac_done(struct netlink_callback *cb)
347{
348	return 0;
349}
350
351static int seg6_genl_dumphmac(struct sk_buff *skb, struct netlink_callback *cb)
352{
353	return -ENOTSUPP;
354}
355
356#endif
357
358static int __net_init seg6_net_init(struct net *net)
359{
360	struct seg6_pernet_data *sdata;
361
362	sdata = kzalloc(sizeof(*sdata), GFP_KERNEL);
363	if (!sdata)
364		return -ENOMEM;
365
366	mutex_init(&sdata->lock);
367
368	sdata->tun_src = kzalloc(sizeof(*sdata->tun_src), GFP_KERNEL);
369	if (!sdata->tun_src) {
370		kfree(sdata);
371		return -ENOMEM;
372	}
373
374	net->ipv6.seg6_data = sdata;
375
376#ifdef CONFIG_IPV6_SEG6_HMAC
377	seg6_hmac_net_init(net);
378#endif
 
 
379
380	return 0;
381}
382
383static void __net_exit seg6_net_exit(struct net *net)
384{
385	struct seg6_pernet_data *sdata = seg6_pernet(net);
386
387#ifdef CONFIG_IPV6_SEG6_HMAC
388	seg6_hmac_net_exit(net);
389#endif
390
391	kfree(sdata->tun_src);
392	kfree(sdata);
393}
394
395static struct pernet_operations ip6_segments_ops = {
396	.init = seg6_net_init,
397	.exit = seg6_net_exit,
398};
399
400static const struct genl_ops seg6_genl_ops[] = {
401	{
402		.cmd	= SEG6_CMD_SETHMAC,
 
403		.doit	= seg6_genl_sethmac,
404		.policy	= seg6_genl_policy,
405		.flags	= GENL_ADMIN_PERM,
406	},
407	{
408		.cmd	= SEG6_CMD_DUMPHMAC,
 
409		.start	= seg6_genl_dumphmac_start,
410		.dumpit	= seg6_genl_dumphmac,
411		.done	= seg6_genl_dumphmac_done,
412		.policy	= seg6_genl_policy,
413		.flags	= GENL_ADMIN_PERM,
414	},
415	{
416		.cmd	= SEG6_CMD_SET_TUNSRC,
 
417		.doit	= seg6_genl_set_tunsrc,
418		.policy	= seg6_genl_policy,
419		.flags	= GENL_ADMIN_PERM,
420	},
421	{
422		.cmd	= SEG6_CMD_GET_TUNSRC,
 
423		.doit	= seg6_genl_get_tunsrc,
424		.policy = seg6_genl_policy,
425		.flags	= GENL_ADMIN_PERM,
426	},
427};
428
429static struct genl_family seg6_genl_family __ro_after_init = {
430	.hdrsize	= 0,
431	.name		= SEG6_GENL_NAME,
432	.version	= SEG6_GENL_VERSION,
433	.maxattr	= SEG6_ATTR_MAX,
 
434	.netnsok	= true,
435	.parallel_ops	= true,
436	.ops		= seg6_genl_ops,
437	.n_ops		= ARRAY_SIZE(seg6_genl_ops),
 
438	.module		= THIS_MODULE,
439};
440
441int __init seg6_init(void)
442{
443	int err = -ENOMEM;
444
445	err = genl_register_family(&seg6_genl_family);
446	if (err)
447		goto out;
448
449	err = register_pernet_subsys(&ip6_segments_ops);
450	if (err)
451		goto out_unregister_genl;
452
453#ifdef CONFIG_IPV6_SEG6_LWTUNNEL
454	err = seg6_iptunnel_init();
455	if (err)
456		goto out_unregister_pernet;
457
458	err = seg6_local_init();
459	if (err)
460		goto out_unregister_pernet;
461#endif
462
463#ifdef CONFIG_IPV6_SEG6_HMAC
464	err = seg6_hmac_init();
465	if (err)
466		goto out_unregister_iptun;
467#endif
468
469	pr_info("Segment Routing with IPv6\n");
470
471out:
472	return err;
473#ifdef CONFIG_IPV6_SEG6_HMAC
474out_unregister_iptun:
475#ifdef CONFIG_IPV6_SEG6_LWTUNNEL
476	seg6_local_exit();
 
477	seg6_iptunnel_exit();
478#endif
479#endif
480#ifdef CONFIG_IPV6_SEG6_LWTUNNEL
481out_unregister_pernet:
482	unregister_pernet_subsys(&ip6_segments_ops);
483#endif
484out_unregister_genl:
485	genl_unregister_family(&seg6_genl_family);
 
 
486	goto out;
487}
488
489void seg6_exit(void)
490{
491#ifdef CONFIG_IPV6_SEG6_HMAC
492	seg6_hmac_exit();
493#endif
494#ifdef CONFIG_IPV6_SEG6_LWTUNNEL
495	seg6_iptunnel_exit();
496#endif
497	unregister_pernet_subsys(&ip6_segments_ops);
498	genl_unregister_family(&seg6_genl_family);
 
499}
v6.13.7
  1// SPDX-License-Identifier: GPL-2.0-or-later
  2/*
  3 *  SR-IPv6 implementation
  4 *
  5 *  Author:
  6 *  David Lebrun <david.lebrun@uclouvain.be>
 
 
 
 
 
 
  7 */
  8
  9#include <linux/errno.h>
 10#include <linux/types.h>
 11#include <linux/socket.h>
 12#include <linux/net.h>
 13#include <linux/in6.h>
 14#include <linux/slab.h>
 15#include <linux/rhashtable.h>
 16
 17#include <net/ipv6.h>
 18#include <net/protocol.h>
 19
 20#include <net/seg6.h>
 21#include <net/genetlink.h>
 22#include <linux/seg6.h>
 23#include <linux/seg6_genl.h>
 
 24#include <net/seg6_hmac.h>
 
 25
 26bool seg6_validate_srh(struct ipv6_sr_hdr *srh, int len, bool reduced)
 27{
 
 28	unsigned int tlv_offset;
 29	int max_last_entry;
 30	int trailing;
 31
 32	if (srh->type != IPV6_SRCRT_TYPE_4)
 33		return false;
 34
 35	if (((srh->hdrlen + 1) << 3) != len)
 36		return false;
 37
 38	if (!reduced && srh->segments_left > srh->first_segment) {
 39		return false;
 40	} else {
 41		max_last_entry = (srh->hdrlen / 2) - 1;
 42
 43		if (srh->first_segment > max_last_entry)
 44			return false;
 45
 46		if (srh->segments_left > srh->first_segment + 1)
 47			return false;
 48	}
 49
 50	tlv_offset = sizeof(*srh) + ((srh->first_segment + 1) << 4);
 51
 52	trailing = len - tlv_offset;
 53	if (trailing < 0)
 54		return false;
 55
 56	while (trailing) {
 57		struct sr6_tlv *tlv;
 58		unsigned int tlv_len;
 59
 60		if (trailing < sizeof(*tlv))
 61			return false;
 62
 63		tlv = (struct sr6_tlv *)((unsigned char *)srh + tlv_offset);
 64		tlv_len = sizeof(*tlv) + tlv->len;
 65
 66		trailing -= tlv_len;
 67		if (trailing < 0)
 68			return false;
 69
 70		tlv_offset += tlv_len;
 71	}
 72
 73	return true;
 74}
 75
 76struct ipv6_sr_hdr *seg6_get_srh(struct sk_buff *skb, int flags)
 77{
 78	struct ipv6_sr_hdr *srh;
 79	int len, srhoff = 0;
 80
 81	if (ipv6_find_hdr(skb, &srhoff, IPPROTO_ROUTING, NULL, &flags) < 0)
 82		return NULL;
 83
 84	if (!pskb_may_pull(skb, srhoff + sizeof(*srh)))
 85		return NULL;
 86
 87	srh = (struct ipv6_sr_hdr *)(skb->data + srhoff);
 88
 89	len = (srh->hdrlen + 1) << 3;
 90
 91	if (!pskb_may_pull(skb, srhoff + len))
 92		return NULL;
 93
 94	/* note that pskb_may_pull may change pointers in header;
 95	 * for this reason it is necessary to reload them when needed.
 96	 */
 97	srh = (struct ipv6_sr_hdr *)(skb->data + srhoff);
 98
 99	if (!seg6_validate_srh(srh, len, true))
100		return NULL;
101
102	return srh;
103}
104
105/* Determine if an ICMP invoking packet contains a segment routing
106 * header.  If it does, extract the offset to the true destination
107 * address, which is in the first segment address.
108 */
109void seg6_icmp_srh(struct sk_buff *skb, struct inet6_skb_parm *opt)
110{
111	__u16 network_header = skb->network_header;
112	struct ipv6_sr_hdr *srh;
113
114	/* Update network header to point to the invoking packet
115	 * inside the ICMP packet, so we can use the seg6_get_srh()
116	 * helper.
117	 */
118	skb_reset_network_header(skb);
119
120	srh = seg6_get_srh(skb, 0);
121	if (!srh)
122		goto out;
123
124	if (srh->type != IPV6_SRCRT_TYPE_4)
125		goto out;
126
127	opt->flags |= IP6SKB_SEG6;
128	opt->srhoff = (unsigned char *)srh - skb->data;
129
130out:
131	/* Restore the network header back to the ICMP packet */
132	skb->network_header = network_header;
133}
134
135static struct genl_family seg6_genl_family;
136
137static const struct nla_policy seg6_genl_policy[SEG6_ATTR_MAX + 1] = {
138	[SEG6_ATTR_DST]				= { .type = NLA_BINARY,
139		.len = sizeof(struct in6_addr) },
140	[SEG6_ATTR_DSTLEN]			= { .type = NLA_S32, },
141	[SEG6_ATTR_HMACKEYID]		= { .type = NLA_U32, },
142	[SEG6_ATTR_SECRET]			= { .type = NLA_BINARY, },
143	[SEG6_ATTR_SECRETLEN]		= { .type = NLA_U8, },
144	[SEG6_ATTR_ALGID]			= { .type = NLA_U8, },
145	[SEG6_ATTR_HMACINFO]		= { .type = NLA_NESTED, },
146};
147
148#ifdef CONFIG_IPV6_SEG6_HMAC
149
150static int seg6_genl_sethmac(struct sk_buff *skb, struct genl_info *info)
151{
152	struct net *net = genl_info_net(info);
153	struct seg6_pernet_data *sdata;
154	struct seg6_hmac_info *hinfo;
155	u32 hmackeyid;
156	char *secret;
157	int err = 0;
158	u8 algid;
159	u8 slen;
160
161	sdata = seg6_pernet(net);
162
163	if (!info->attrs[SEG6_ATTR_HMACKEYID] ||
164	    !info->attrs[SEG6_ATTR_SECRETLEN] ||
165	    !info->attrs[SEG6_ATTR_ALGID])
166		return -EINVAL;
167
168	hmackeyid = nla_get_u32(info->attrs[SEG6_ATTR_HMACKEYID]);
169	slen = nla_get_u8(info->attrs[SEG6_ATTR_SECRETLEN]);
170	algid = nla_get_u8(info->attrs[SEG6_ATTR_ALGID]);
171
172	if (hmackeyid == 0)
173		return -EINVAL;
174
175	if (slen > SEG6_HMAC_SECRET_LEN)
176		return -EINVAL;
177
178	mutex_lock(&sdata->lock);
179	hinfo = seg6_hmac_info_lookup(net, hmackeyid);
180
181	if (!slen) {
 
 
 
182		err = seg6_hmac_info_del(net, hmackeyid);
183
184		goto out_unlock;
185	}
186
187	if (!info->attrs[SEG6_ATTR_SECRET]) {
188		err = -EINVAL;
189		goto out_unlock;
190	}
191
192	if (slen > nla_len(info->attrs[SEG6_ATTR_SECRET])) {
193		err = -EINVAL;
194		goto out_unlock;
195	}
196
197	if (hinfo) {
198		err = seg6_hmac_info_del(net, hmackeyid);
199		if (err)
200			goto out_unlock;
201	}
202
203	secret = (char *)nla_data(info->attrs[SEG6_ATTR_SECRET]);
204
205	hinfo = kzalloc(sizeof(*hinfo), GFP_KERNEL);
206	if (!hinfo) {
207		err = -ENOMEM;
208		goto out_unlock;
209	}
210
211	memcpy(hinfo->secret, secret, slen);
212	hinfo->slen = slen;
213	hinfo->alg_id = algid;
214	hinfo->hmackeyid = hmackeyid;
215
216	err = seg6_hmac_info_add(net, hmackeyid, hinfo);
217	if (err)
218		kfree(hinfo);
219
220out_unlock:
221	mutex_unlock(&sdata->lock);
222	return err;
223}
224
225#else
226
227static int seg6_genl_sethmac(struct sk_buff *skb, struct genl_info *info)
228{
229	return -ENOTSUPP;
230}
231
232#endif
233
234static int seg6_genl_set_tunsrc(struct sk_buff *skb, struct genl_info *info)
235{
236	struct net *net = genl_info_net(info);
237	struct in6_addr *val, *t_old, *t_new;
238	struct seg6_pernet_data *sdata;
239
240	sdata = seg6_pernet(net);
241
242	if (!info->attrs[SEG6_ATTR_DST])
243		return -EINVAL;
244
245	val = nla_data(info->attrs[SEG6_ATTR_DST]);
246	t_new = kmemdup(val, sizeof(*val), GFP_KERNEL);
247	if (!t_new)
248		return -ENOMEM;
249
250	mutex_lock(&sdata->lock);
251
252	t_old = sdata->tun_src;
253	rcu_assign_pointer(sdata->tun_src, t_new);
254
255	mutex_unlock(&sdata->lock);
256
257	synchronize_net();
258	kfree(t_old);
259
260	return 0;
261}
262
263static int seg6_genl_get_tunsrc(struct sk_buff *skb, struct genl_info *info)
264{
265	struct net *net = genl_info_net(info);
266	struct in6_addr *tun_src;
267	struct sk_buff *msg;
268	void *hdr;
269
270	msg = genlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
271	if (!msg)
272		return -ENOMEM;
273
274	hdr = genlmsg_put(msg, info->snd_portid, info->snd_seq,
275			  &seg6_genl_family, 0, SEG6_CMD_GET_TUNSRC);
276	if (!hdr)
277		goto free_msg;
278
279	rcu_read_lock();
280	tun_src = rcu_dereference(seg6_pernet(net)->tun_src);
281
282	if (nla_put(msg, SEG6_ATTR_DST, sizeof(struct in6_addr), tun_src))
283		goto nla_put_failure;
284
285	rcu_read_unlock();
286
287	genlmsg_end(msg, hdr);
288	return genlmsg_reply(msg, info);
 
 
289
290nla_put_failure:
291	rcu_read_unlock();
 
292free_msg:
293	nlmsg_free(msg);
294	return -ENOMEM;
295}
296
297#ifdef CONFIG_IPV6_SEG6_HMAC
298
299static int __seg6_hmac_fill_info(struct seg6_hmac_info *hinfo,
300				 struct sk_buff *msg)
301{
302	if (nla_put_u32(msg, SEG6_ATTR_HMACKEYID, hinfo->hmackeyid) ||
303	    nla_put_u8(msg, SEG6_ATTR_SECRETLEN, hinfo->slen) ||
304	    nla_put(msg, SEG6_ATTR_SECRET, hinfo->slen, hinfo->secret) ||
305	    nla_put_u8(msg, SEG6_ATTR_ALGID, hinfo->alg_id))
306		return -1;
307
308	return 0;
309}
310
311static int __seg6_genl_dumphmac_element(struct seg6_hmac_info *hinfo,
312					u32 portid, u32 seq, u32 flags,
313					struct sk_buff *skb, u8 cmd)
314{
315	void *hdr;
316
317	hdr = genlmsg_put(skb, portid, seq, &seg6_genl_family, flags, cmd);
318	if (!hdr)
319		return -ENOMEM;
320
321	if (__seg6_hmac_fill_info(hinfo, skb) < 0)
322		goto nla_put_failure;
323
324	genlmsg_end(skb, hdr);
325	return 0;
326
327nla_put_failure:
328	genlmsg_cancel(skb, hdr);
329	return -EMSGSIZE;
330}
331
332static int seg6_genl_dumphmac_start(struct netlink_callback *cb)
333{
334	struct net *net = sock_net(cb->skb->sk);
335	struct seg6_pernet_data *sdata;
336	struct rhashtable_iter *iter;
337
338	sdata = seg6_pernet(net);
339	iter = (struct rhashtable_iter *)cb->args[0];
340
341	if (!iter) {
342		iter = kmalloc(sizeof(*iter), GFP_KERNEL);
343		if (!iter)
344			return -ENOMEM;
345
346		cb->args[0] = (long)iter;
347	}
348
349	rhashtable_walk_enter(&sdata->hmac_infos, iter);
350
351	return 0;
352}
353
354static int seg6_genl_dumphmac_done(struct netlink_callback *cb)
355{
356	struct rhashtable_iter *iter = (struct rhashtable_iter *)cb->args[0];
357
358	rhashtable_walk_exit(iter);
359
360	kfree(iter);
361
362	return 0;
363}
364
365static int seg6_genl_dumphmac(struct sk_buff *skb, struct netlink_callback *cb)
366{
367	struct rhashtable_iter *iter = (struct rhashtable_iter *)cb->args[0];
368	struct seg6_hmac_info *hinfo;
369	int ret;
370
371	rhashtable_walk_start(iter);
372
373	for (;;) {
374		hinfo = rhashtable_walk_next(iter);
375
376		if (IS_ERR(hinfo)) {
377			if (PTR_ERR(hinfo) == -EAGAIN)
378				continue;
379			ret = PTR_ERR(hinfo);
380			goto done;
381		} else if (!hinfo) {
382			break;
383		}
384
385		ret = __seg6_genl_dumphmac_element(hinfo,
386						   NETLINK_CB(cb->skb).portid,
387						   cb->nlh->nlmsg_seq,
388						   NLM_F_MULTI,
389						   skb, SEG6_CMD_DUMPHMAC);
390		if (ret)
391			goto done;
392	}
393
394	ret = skb->len;
395
396done:
397	rhashtable_walk_stop(iter);
398	return ret;
399}
400
401#else
402
403static int seg6_genl_dumphmac_start(struct netlink_callback *cb)
404{
405	return 0;
406}
407
408static int seg6_genl_dumphmac_done(struct netlink_callback *cb)
409{
410	return 0;
411}
412
413static int seg6_genl_dumphmac(struct sk_buff *skb, struct netlink_callback *cb)
414{
415	return -ENOTSUPP;
416}
417
418#endif
419
420static int __net_init seg6_net_init(struct net *net)
421{
422	struct seg6_pernet_data *sdata;
423
424	sdata = kzalloc(sizeof(*sdata), GFP_KERNEL);
425	if (!sdata)
426		return -ENOMEM;
427
428	mutex_init(&sdata->lock);
429
430	sdata->tun_src = kzalloc(sizeof(*sdata->tun_src), GFP_KERNEL);
431	if (!sdata->tun_src) {
432		kfree(sdata);
433		return -ENOMEM;
434	}
435
436	net->ipv6.seg6_data = sdata;
437
438	if (seg6_hmac_net_init(net)) {
439		kfree(rcu_dereference_raw(sdata->tun_src));
440		kfree(sdata);
441		return -ENOMEM;
442	}
443
444	return 0;
445}
446
447static void __net_exit seg6_net_exit(struct net *net)
448{
449	struct seg6_pernet_data *sdata = seg6_pernet(net);
450
 
451	seg6_hmac_net_exit(net);
 
452
453	kfree(rcu_dereference_raw(sdata->tun_src));
454	kfree(sdata);
455}
456
457static struct pernet_operations ip6_segments_ops = {
458	.init = seg6_net_init,
459	.exit = seg6_net_exit,
460};
461
462static const struct genl_ops seg6_genl_ops[] = {
463	{
464		.cmd	= SEG6_CMD_SETHMAC,
465		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
466		.doit	= seg6_genl_sethmac,
 
467		.flags	= GENL_ADMIN_PERM,
468	},
469	{
470		.cmd	= SEG6_CMD_DUMPHMAC,
471		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
472		.start	= seg6_genl_dumphmac_start,
473		.dumpit	= seg6_genl_dumphmac,
474		.done	= seg6_genl_dumphmac_done,
 
475		.flags	= GENL_ADMIN_PERM,
476	},
477	{
478		.cmd	= SEG6_CMD_SET_TUNSRC,
479		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
480		.doit	= seg6_genl_set_tunsrc,
 
481		.flags	= GENL_ADMIN_PERM,
482	},
483	{
484		.cmd	= SEG6_CMD_GET_TUNSRC,
485		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
486		.doit	= seg6_genl_get_tunsrc,
 
487		.flags	= GENL_ADMIN_PERM,
488	},
489};
490
491static struct genl_family seg6_genl_family __ro_after_init = {
492	.hdrsize	= 0,
493	.name		= SEG6_GENL_NAME,
494	.version	= SEG6_GENL_VERSION,
495	.maxattr	= SEG6_ATTR_MAX,
496	.policy = seg6_genl_policy,
497	.netnsok	= true,
498	.parallel_ops	= true,
499	.ops		= seg6_genl_ops,
500	.n_ops		= ARRAY_SIZE(seg6_genl_ops),
501	.resv_start_op	= SEG6_CMD_GET_TUNSRC + 1,
502	.module		= THIS_MODULE,
503};
504
505int __init seg6_init(void)
506{
507	int err;
508
509	err = register_pernet_subsys(&ip6_segments_ops);
510	if (err)
511		goto out;
512
513	err = genl_register_family(&seg6_genl_family);
514	if (err)
515		goto out_unregister_pernet;
516
 
517	err = seg6_iptunnel_init();
518	if (err)
519		goto out_unregister_genl;
520
521	err = seg6_local_init();
522	if (err)
523		goto out_unregister_iptun;
 
524
 
525	err = seg6_hmac_init();
526	if (err)
527		goto out_unregister_seg6;
 
528
529	pr_info("Segment Routing with IPv6\n");
530
531out:
532	return err;
533out_unregister_seg6:
 
 
534	seg6_local_exit();
535out_unregister_iptun:
536	seg6_iptunnel_exit();
 
 
 
 
 
 
537out_unregister_genl:
538	genl_unregister_family(&seg6_genl_family);
539out_unregister_pernet:
540	unregister_pernet_subsys(&ip6_segments_ops);
541	goto out;
542}
543
544void seg6_exit(void)
545{
 
546	seg6_hmac_exit();
547	seg6_local_exit();
 
548	seg6_iptunnel_exit();
 
 
549	genl_unregister_family(&seg6_genl_family);
550	unregister_pernet_subsys(&ip6_segments_ops);
551}