• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /* net/tipc/udp_media.c: IP bearer support for TIPC
2  *
3  * Copyright (c) 2015, Ericsson AB
4  * All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions are met:
8  *
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  * 3. Neither the names of the copyright holders nor the names of its
15  *    contributors may be used to endorse or promote products derived from
16  *    this software without specific prior written permission.
17  *
18  * Alternatively, this software may be distributed under the terms of the
19  * GNU General Public License ("GPL") version 2 as published by the Free
20  * Software Foundation.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
23  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
26  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
27  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
28  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
29  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
30  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
31  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
32  * POSSIBILITY OF SUCH DAMAGE.
33  */
34 
35 #include <linux/socket.h>
36 #include <linux/ip.h>
37 #include <linux/udp.h>
38 #include <linux/inet.h>
39 #include <linux/inetdevice.h>
40 #include <linux/igmp.h>
41 #include <linux/kernel.h>
42 #include <linux/workqueue.h>
43 #include <linux/list.h>
44 #include <net/sock.h>
45 #include <net/ip.h>
46 #include <net/udp_tunnel.h>
47 #include <net/addrconf.h>
48 #include <linux/tipc_netlink.h>
49 #include "core.h"
50 #include "addr.h"
51 #include "net.h"
52 #include "bearer.h"
53 #include "netlink.h"
54 #include "msg.h"
55 
56 /* IANA assigned UDP port */
57 #define UDP_PORT_DEFAULT	6118
58 
59 #define UDP_MIN_HEADROOM        48
60 
61 /**
62  * struct udp_media_addr - IP/UDP addressing information
63  *
64  * This is the bearer level originating address used in neighbor discovery
65  * messages, and all fields should be in network byte order
66  */
67 struct udp_media_addr {
68 	__be16	proto;
69 	__be16	port;
70 	union {
71 		struct in_addr ipv4;
72 		struct in6_addr ipv6;
73 	};
74 };
75 
76 /* struct udp_replicast - container for UDP remote addresses */
77 struct udp_replicast {
78 	struct udp_media_addr addr;
79 	struct rcu_head rcu;
80 	struct list_head list;
81 };
82 
83 /**
84  * struct udp_bearer - ip/udp bearer data structure
85  * @bearer:	associated generic tipc bearer
86  * @ubsock:	bearer associated socket
87  * @ifindex:	local address scope
88  * @work:	used to schedule deferred work on a bearer
89  */
90 struct udp_bearer {
91 	struct tipc_bearer __rcu *bearer;
92 	struct socket *ubsock;
93 	u32 ifindex;
94 	struct work_struct work;
95 	struct udp_replicast rcast;
96 };
97 
tipc_udp_is_mcast_addr(struct udp_media_addr * addr)98 static int tipc_udp_is_mcast_addr(struct udp_media_addr *addr)
99 {
100 	if (ntohs(addr->proto) == ETH_P_IP)
101 		return ipv4_is_multicast(addr->ipv4.s_addr);
102 #if IS_ENABLED(CONFIG_IPV6)
103 	else
104 		return ipv6_addr_is_multicast(&addr->ipv6);
105 #endif
106 	return 0;
107 }
108 
109 /* udp_media_addr_set - convert a ip/udp address to a TIPC media address */
tipc_udp_media_addr_set(struct tipc_media_addr * addr,struct udp_media_addr * ua)110 static void tipc_udp_media_addr_set(struct tipc_media_addr *addr,
111 				    struct udp_media_addr *ua)
112 {
113 	memset(addr, 0, sizeof(struct tipc_media_addr));
114 	addr->media_id = TIPC_MEDIA_TYPE_UDP;
115 	memcpy(addr->value, ua, sizeof(struct udp_media_addr));
116 
117 	if (tipc_udp_is_mcast_addr(ua))
118 		addr->broadcast = TIPC_BROADCAST_SUPPORT;
119 }
120 
121 /* tipc_udp_addr2str - convert ip/udp address to string */
tipc_udp_addr2str(struct tipc_media_addr * a,char * buf,int size)122 static int tipc_udp_addr2str(struct tipc_media_addr *a, char *buf, int size)
123 {
124 	struct udp_media_addr *ua = (struct udp_media_addr *)&a->value;
125 
126 	if (ntohs(ua->proto) == ETH_P_IP)
127 		snprintf(buf, size, "%pI4:%u", &ua->ipv4, ntohs(ua->port));
128 	else if (ntohs(ua->proto) == ETH_P_IPV6)
129 		snprintf(buf, size, "%pI6:%u", &ua->ipv6, ntohs(ua->port));
130 	else
131 		pr_err("Invalid UDP media address\n");
132 	return 0;
133 }
134 
135 /* tipc_udp_msg2addr - extract an ip/udp address from a TIPC ndisc message */
tipc_udp_msg2addr(struct tipc_bearer * b,struct tipc_media_addr * a,char * msg)136 static int tipc_udp_msg2addr(struct tipc_bearer *b, struct tipc_media_addr *a,
137 			     char *msg)
138 {
139 	struct udp_media_addr *ua;
140 
141 	ua = (struct udp_media_addr *) (msg + TIPC_MEDIA_ADDR_OFFSET);
142 	if (msg[TIPC_MEDIA_TYPE_OFFSET] != TIPC_MEDIA_TYPE_UDP)
143 		return -EINVAL;
144 	tipc_udp_media_addr_set(a, ua);
145 	return 0;
146 }
147 
148 /* tipc_udp_addr2msg - write an ip/udp address to a TIPC ndisc message */
tipc_udp_addr2msg(char * msg,struct tipc_media_addr * a)149 static int tipc_udp_addr2msg(char *msg, struct tipc_media_addr *a)
150 {
151 	memset(msg, 0, TIPC_MEDIA_INFO_SIZE);
152 	msg[TIPC_MEDIA_TYPE_OFFSET] = TIPC_MEDIA_TYPE_UDP;
153 	memcpy(msg + TIPC_MEDIA_ADDR_OFFSET, a->value,
154 	       sizeof(struct udp_media_addr));
155 	return 0;
156 }
157 
158 /* tipc_send_msg - enqueue a send request */
tipc_udp_xmit(struct net * net,struct sk_buff * skb,struct udp_bearer * ub,struct udp_media_addr * src,struct udp_media_addr * dst)159 static int tipc_udp_xmit(struct net *net, struct sk_buff *skb,
160 			 struct udp_bearer *ub, struct udp_media_addr *src,
161 			 struct udp_media_addr *dst)
162 {
163 	int ttl, err = 0;
164 	struct rtable *rt;
165 
166 	if (dst->proto == htons(ETH_P_IP)) {
167 		struct flowi4 fl = {
168 			.daddr = dst->ipv4.s_addr,
169 			.saddr = src->ipv4.s_addr,
170 			.flowi4_mark = skb->mark,
171 			.flowi4_proto = IPPROTO_UDP
172 		};
173 		rt = ip_route_output_key(net, &fl);
174 		if (IS_ERR(rt)) {
175 			err = PTR_ERR(rt);
176 			goto tx_error;
177 		}
178 
179 		ttl = ip4_dst_hoplimit(&rt->dst);
180 		udp_tunnel_xmit_skb(rt, ub->ubsock->sk, skb, src->ipv4.s_addr,
181 				    dst->ipv4.s_addr, 0, ttl, 0, src->port,
182 				    dst->port, false, true);
183 #if IS_ENABLED(CONFIG_IPV6)
184 	} else {
185 		struct dst_entry *ndst;
186 		struct flowi6 fl6 = {
187 			.flowi6_oif = ub->ifindex,
188 			.daddr = dst->ipv6,
189 			.saddr = src->ipv6,
190 			.flowi6_proto = IPPROTO_UDP
191 		};
192 		ndst = ipv6_stub->ipv6_dst_lookup_flow(net,
193 						       ub->ubsock->sk,
194 						       &fl6, NULL);
195 		if (IS_ERR(ndst)) {
196 			err = PTR_ERR(ndst);
197 			goto tx_error;
198 		}
199 		ttl = ip6_dst_hoplimit(ndst);
200 		err = udp_tunnel6_xmit_skb(ndst, ub->ubsock->sk, skb, NULL,
201 					   &src->ipv6, &dst->ipv6, 0, ttl, 0,
202 					   src->port, dst->port, false);
203 #endif
204 	}
205 	return err;
206 
207 tx_error:
208 	kfree_skb(skb);
209 	return err;
210 }
211 
tipc_udp_send_msg(struct net * net,struct sk_buff * skb,struct tipc_bearer * b,struct tipc_media_addr * addr)212 static int tipc_udp_send_msg(struct net *net, struct sk_buff *skb,
213 			     struct tipc_bearer *b,
214 			     struct tipc_media_addr *addr)
215 {
216 	struct udp_media_addr *src = (struct udp_media_addr *)&b->addr.value;
217 	struct udp_media_addr *dst = (struct udp_media_addr *)&addr->value;
218 	struct udp_replicast *rcast;
219 	struct udp_bearer *ub;
220 	int err = 0;
221 
222 	if (skb_headroom(skb) < UDP_MIN_HEADROOM) {
223 		err = pskb_expand_head(skb, UDP_MIN_HEADROOM, 0, GFP_ATOMIC);
224 		if (err)
225 			goto out;
226 	}
227 
228 	skb_set_inner_protocol(skb, htons(ETH_P_TIPC));
229 	ub = rcu_dereference_rtnl(b->media_ptr);
230 	if (!ub) {
231 		err = -ENODEV;
232 		goto out;
233 	}
234 
235 	if (addr->broadcast != TIPC_REPLICAST_SUPPORT)
236 		return tipc_udp_xmit(net, skb, ub, src, dst);
237 
238 	/* Replicast, send an skb to each configured IP address */
239 	list_for_each_entry_rcu(rcast, &ub->rcast.list, list) {
240 		struct sk_buff *_skb;
241 
242 		_skb = pskb_copy(skb, GFP_ATOMIC);
243 		if (!_skb) {
244 			err = -ENOMEM;
245 			goto out;
246 		}
247 
248 		err = tipc_udp_xmit(net, _skb, ub, src, &rcast->addr);
249 		if (err)
250 			goto out;
251 	}
252 	err = 0;
253 out:
254 	kfree_skb(skb);
255 	return err;
256 }
257 
tipc_udp_is_known_peer(struct tipc_bearer * b,struct udp_media_addr * addr)258 static bool tipc_udp_is_known_peer(struct tipc_bearer *b,
259 				   struct udp_media_addr *addr)
260 {
261 	struct udp_replicast *rcast, *tmp;
262 	struct udp_bearer *ub;
263 
264 	ub = rcu_dereference_rtnl(b->media_ptr);
265 	if (!ub) {
266 		pr_err_ratelimited("UDP bearer instance not found\n");
267 		return false;
268 	}
269 
270 	list_for_each_entry_safe(rcast, tmp, &ub->rcast.list, list) {
271 		if (!memcmp(&rcast->addr, addr, sizeof(struct udp_media_addr)))
272 			return true;
273 	}
274 
275 	return false;
276 }
277 
tipc_udp_rcast_add(struct tipc_bearer * b,struct udp_media_addr * addr)278 static int tipc_udp_rcast_add(struct tipc_bearer *b,
279 			      struct udp_media_addr *addr)
280 {
281 	struct udp_replicast *rcast;
282 	struct udp_bearer *ub;
283 
284 	ub = rcu_dereference_rtnl(b->media_ptr);
285 	if (!ub)
286 		return -ENODEV;
287 
288 	rcast = kmalloc(sizeof(*rcast), GFP_ATOMIC);
289 	if (!rcast)
290 		return -ENOMEM;
291 
292 	memcpy(&rcast->addr, addr, sizeof(struct udp_media_addr));
293 
294 	if (ntohs(addr->proto) == ETH_P_IP)
295 		pr_info("New replicast peer: %pI4\n", &rcast->addr.ipv4);
296 #if IS_ENABLED(CONFIG_IPV6)
297 	else if (ntohs(addr->proto) == ETH_P_IPV6)
298 		pr_info("New replicast peer: %pI6\n", &rcast->addr.ipv6);
299 #endif
300 	b->bcast_addr.broadcast = TIPC_REPLICAST_SUPPORT;
301 	list_add_rcu(&rcast->list, &ub->rcast.list);
302 	return 0;
303 }
304 
tipc_udp_rcast_disc(struct tipc_bearer * b,struct sk_buff * skb)305 static int tipc_udp_rcast_disc(struct tipc_bearer *b, struct sk_buff *skb)
306 {
307 	struct udp_media_addr src = {0};
308 	struct udp_media_addr *dst;
309 
310 	dst = (struct udp_media_addr *)&b->bcast_addr.value;
311 	if (tipc_udp_is_mcast_addr(dst))
312 		return 0;
313 
314 	src.port = udp_hdr(skb)->source;
315 
316 	if (ip_hdr(skb)->version == 4) {
317 		struct iphdr *iphdr = ip_hdr(skb);
318 
319 		src.proto = htons(ETH_P_IP);
320 		src.ipv4.s_addr = iphdr->saddr;
321 		if (ipv4_is_multicast(iphdr->daddr))
322 			return 0;
323 #if IS_ENABLED(CONFIG_IPV6)
324 	} else if (ip_hdr(skb)->version == 6) {
325 		struct ipv6hdr *iphdr = ipv6_hdr(skb);
326 
327 		src.proto = htons(ETH_P_IPV6);
328 		src.ipv6 = iphdr->saddr;
329 		if (ipv6_addr_is_multicast(&iphdr->daddr))
330 			return 0;
331 #endif
332 	} else {
333 		return 0;
334 	}
335 
336 	if (likely(tipc_udp_is_known_peer(b, &src)))
337 		return 0;
338 
339 	return tipc_udp_rcast_add(b, &src);
340 }
341 
342 /* tipc_udp_recv - read data from bearer socket */
tipc_udp_recv(struct sock * sk,struct sk_buff * skb)343 static int tipc_udp_recv(struct sock *sk, struct sk_buff *skb)
344 {
345 	struct udp_bearer *ub;
346 	struct tipc_bearer *b;
347 	struct tipc_msg *hdr;
348 	int err;
349 
350 	ub = rcu_dereference_sk_user_data(sk);
351 	if (!ub) {
352 		pr_err_ratelimited("Failed to get UDP bearer reference");
353 		goto out;
354 	}
355 	skb_pull(skb, sizeof(struct udphdr));
356 	hdr = buf_msg(skb);
357 
358 	rcu_read_lock();
359 	b = rcu_dereference_rtnl(ub->bearer);
360 	if (!b)
361 		goto rcu_out;
362 
363 	if (b && test_bit(0, &b->up)) {
364 		tipc_rcv(sock_net(sk), skb, b);
365 		rcu_read_unlock();
366 		return 0;
367 	}
368 
369 	if (unlikely(msg_user(hdr) == LINK_CONFIG)) {
370 		err = tipc_udp_rcast_disc(b, skb);
371 		if (err)
372 			goto rcu_out;
373 	}
374 
375 rcu_out:
376 	rcu_read_unlock();
377 out:
378 	kfree_skb(skb);
379 	return 0;
380 }
381 
enable_mcast(struct udp_bearer * ub,struct udp_media_addr * remote)382 static int enable_mcast(struct udp_bearer *ub, struct udp_media_addr *remote)
383 {
384 	int err = 0;
385 	struct ip_mreqn mreqn;
386 	struct sock *sk = ub->ubsock->sk;
387 
388 	if (ntohs(remote->proto) == ETH_P_IP) {
389 		mreqn.imr_multiaddr = remote->ipv4;
390 		mreqn.imr_ifindex = ub->ifindex;
391 		err = ip_mc_join_group(sk, &mreqn);
392 #if IS_ENABLED(CONFIG_IPV6)
393 	} else {
394 		err = ipv6_stub->ipv6_sock_mc_join(sk, ub->ifindex,
395 						   &remote->ipv6);
396 #endif
397 	}
398 	return err;
399 }
400 
__tipc_nl_add_udp_addr(struct sk_buff * skb,struct udp_media_addr * addr,int nla_t)401 static int __tipc_nl_add_udp_addr(struct sk_buff *skb,
402 				  struct udp_media_addr *addr, int nla_t)
403 {
404 	if (ntohs(addr->proto) == ETH_P_IP) {
405 		struct sockaddr_in ip4;
406 
407 		memset(&ip4, 0, sizeof(ip4));
408 		ip4.sin_family = AF_INET;
409 		ip4.sin_port = addr->port;
410 		ip4.sin_addr.s_addr = addr->ipv4.s_addr;
411 		if (nla_put(skb, nla_t, sizeof(ip4), &ip4))
412 			return -EMSGSIZE;
413 
414 #if IS_ENABLED(CONFIG_IPV6)
415 	} else if (ntohs(addr->proto) == ETH_P_IPV6) {
416 		struct sockaddr_in6 ip6;
417 
418 		memset(&ip6, 0, sizeof(ip6));
419 		ip6.sin6_family = AF_INET6;
420 		ip6.sin6_port  = addr->port;
421 		memcpy(&ip6.sin6_addr, &addr->ipv6, sizeof(struct in6_addr));
422 		if (nla_put(skb, nla_t, sizeof(ip6), &ip6))
423 			return -EMSGSIZE;
424 #endif
425 	}
426 
427 	return 0;
428 }
429 
tipc_udp_nl_dump_remoteip(struct sk_buff * skb,struct netlink_callback * cb)430 int tipc_udp_nl_dump_remoteip(struct sk_buff *skb, struct netlink_callback *cb)
431 {
432 	u32 bid = cb->args[0];
433 	u32 skip_cnt = cb->args[1];
434 	u32 portid = NETLINK_CB(cb->skb).portid;
435 	struct udp_replicast *rcast, *tmp;
436 	struct tipc_bearer *b;
437 	struct udp_bearer *ub;
438 	void *hdr;
439 	int err;
440 	int i;
441 
442 	if (!bid && !skip_cnt) {
443 		struct net *net = sock_net(skb->sk);
444 		struct nlattr *battrs[TIPC_NLA_BEARER_MAX + 1];
445 		struct nlattr **attrs;
446 		char *bname;
447 
448 		err = tipc_nlmsg_parse(cb->nlh, &attrs);
449 		if (err)
450 			return err;
451 
452 		if (!attrs[TIPC_NLA_BEARER])
453 			return -EINVAL;
454 
455 		err = nla_parse_nested(battrs, TIPC_NLA_BEARER_MAX,
456 				       attrs[TIPC_NLA_BEARER],
457 				       tipc_nl_bearer_policy, NULL);
458 		if (err)
459 			return err;
460 
461 		if (!battrs[TIPC_NLA_BEARER_NAME])
462 			return -EINVAL;
463 
464 		bname = nla_data(battrs[TIPC_NLA_BEARER_NAME]);
465 
466 		rtnl_lock();
467 		b = tipc_bearer_find(net, bname);
468 		if (!b) {
469 			rtnl_unlock();
470 			return -EINVAL;
471 		}
472 		bid = b->identity;
473 	} else {
474 		struct net *net = sock_net(skb->sk);
475 		struct tipc_net *tn = net_generic(net, tipc_net_id);
476 
477 		rtnl_lock();
478 		b = rtnl_dereference(tn->bearer_list[bid]);
479 		if (!b) {
480 			rtnl_unlock();
481 			return -EINVAL;
482 		}
483 	}
484 
485 	ub = rcu_dereference_rtnl(b->media_ptr);
486 	if (!ub) {
487 		rtnl_unlock();
488 		return -EINVAL;
489 	}
490 
491 	i = 0;
492 	list_for_each_entry_safe(rcast, tmp, &ub->rcast.list, list) {
493 		if (i < skip_cnt)
494 			goto count;
495 
496 		hdr = genlmsg_put(skb, portid, cb->nlh->nlmsg_seq,
497 				  &tipc_genl_family, NLM_F_MULTI,
498 				  TIPC_NL_BEARER_GET);
499 		if (!hdr)
500 			goto done;
501 
502 		err = __tipc_nl_add_udp_addr(skb, &rcast->addr,
503 					     TIPC_NLA_UDP_REMOTE);
504 		if (err) {
505 			genlmsg_cancel(skb, hdr);
506 			goto done;
507 		}
508 		genlmsg_end(skb, hdr);
509 count:
510 		i++;
511 	}
512 done:
513 	rtnl_unlock();
514 	cb->args[0] = bid;
515 	cb->args[1] = i;
516 
517 	return skb->len;
518 }
519 
tipc_udp_nl_add_bearer_data(struct tipc_nl_msg * msg,struct tipc_bearer * b)520 int tipc_udp_nl_add_bearer_data(struct tipc_nl_msg *msg, struct tipc_bearer *b)
521 {
522 	struct udp_media_addr *src = (struct udp_media_addr *)&b->addr.value;
523 	struct udp_media_addr *dst;
524 	struct udp_bearer *ub;
525 	struct nlattr *nest;
526 
527 	ub = rcu_dereference_rtnl(b->media_ptr);
528 	if (!ub)
529 		return -ENODEV;
530 
531 	nest = nla_nest_start(msg->skb, TIPC_NLA_BEARER_UDP_OPTS);
532 	if (!nest)
533 		goto msg_full;
534 
535 	if (__tipc_nl_add_udp_addr(msg->skb, src, TIPC_NLA_UDP_LOCAL))
536 		goto msg_full;
537 
538 	dst = (struct udp_media_addr *)&b->bcast_addr.value;
539 	if (__tipc_nl_add_udp_addr(msg->skb, dst, TIPC_NLA_UDP_REMOTE))
540 		goto msg_full;
541 
542 	if (!list_empty(&ub->rcast.list)) {
543 		if (nla_put_flag(msg->skb, TIPC_NLA_UDP_MULTI_REMOTEIP))
544 			goto msg_full;
545 	}
546 
547 	nla_nest_end(msg->skb, nest);
548 	return 0;
549 msg_full:
550 	nla_nest_cancel(msg->skb, nest);
551 	return -EMSGSIZE;
552 }
553 
554 /**
555  * tipc_parse_udp_addr - build udp media address from netlink data
556  * @nlattr:	netlink attribute containing sockaddr storage aligned address
557  * @addr:	tipc media address to fill with address, port and protocol type
558  * @scope_id:	IPv6 scope id pointer, not NULL indicates it's required
559  */
560 
tipc_parse_udp_addr(struct nlattr * nla,struct udp_media_addr * addr,u32 * scope_id)561 static int tipc_parse_udp_addr(struct nlattr *nla, struct udp_media_addr *addr,
562 			       u32 *scope_id)
563 {
564 	struct sockaddr_storage sa;
565 
566 	nla_memcpy(&sa, nla, sizeof(sa));
567 	if (sa.ss_family == AF_INET) {
568 		struct sockaddr_in *ip4 = (struct sockaddr_in *)&sa;
569 
570 		addr->proto = htons(ETH_P_IP);
571 		addr->port = ip4->sin_port;
572 		addr->ipv4.s_addr = ip4->sin_addr.s_addr;
573 		return 0;
574 
575 #if IS_ENABLED(CONFIG_IPV6)
576 	} else if (sa.ss_family == AF_INET6) {
577 		struct sockaddr_in6 *ip6 = (struct sockaddr_in6 *)&sa;
578 
579 		addr->proto = htons(ETH_P_IPV6);
580 		addr->port = ip6->sin6_port;
581 		memcpy(&addr->ipv6, &ip6->sin6_addr, sizeof(struct in6_addr));
582 
583 		/* Scope ID is only interesting for local addresses */
584 		if (scope_id) {
585 			int atype;
586 
587 			atype = ipv6_addr_type(&ip6->sin6_addr);
588 			if (__ipv6_addr_needs_scope_id(atype) &&
589 			    !ip6->sin6_scope_id) {
590 				return -EINVAL;
591 			}
592 
593 			*scope_id = ip6->sin6_scope_id ? : 0;
594 		}
595 
596 		return 0;
597 #endif
598 	}
599 	return -EADDRNOTAVAIL;
600 }
601 
tipc_udp_nl_bearer_add(struct tipc_bearer * b,struct nlattr * attr)602 int tipc_udp_nl_bearer_add(struct tipc_bearer *b, struct nlattr *attr)
603 {
604 	int err;
605 	struct udp_media_addr addr = {0};
606 	struct nlattr *opts[TIPC_NLA_UDP_MAX + 1];
607 	struct udp_media_addr *dst;
608 
609 	if (nla_parse_nested(opts, TIPC_NLA_UDP_MAX, attr,
610 			     tipc_nl_udp_policy, NULL))
611 		return -EINVAL;
612 
613 	if (!opts[TIPC_NLA_UDP_REMOTE])
614 		return -EINVAL;
615 
616 	err = tipc_parse_udp_addr(opts[TIPC_NLA_UDP_REMOTE], &addr, NULL);
617 	if (err)
618 		return err;
619 
620 	dst = (struct udp_media_addr *)&b->bcast_addr.value;
621 	if (tipc_udp_is_mcast_addr(dst)) {
622 		pr_err("Can't add remote ip to TIPC UDP multicast bearer\n");
623 		return -EINVAL;
624 	}
625 
626 	if (tipc_udp_is_known_peer(b, &addr))
627 		return 0;
628 
629 	return tipc_udp_rcast_add(b, &addr);
630 }
631 
632 /**
633  * tipc_udp_enable - callback to create a new udp bearer instance
634  * @net:	network namespace
635  * @b:		pointer to generic tipc_bearer
636  * @attrs:	netlink bearer configuration
637  *
638  * validate the bearer parameters and initialize the udp bearer
639  * rtnl_lock should be held
640  */
tipc_udp_enable(struct net * net,struct tipc_bearer * b,struct nlattr * attrs[])641 static int tipc_udp_enable(struct net *net, struct tipc_bearer *b,
642 			   struct nlattr *attrs[])
643 {
644 	int err = -EINVAL;
645 	struct udp_bearer *ub;
646 	struct udp_media_addr remote = {0};
647 	struct udp_media_addr local = {0};
648 	struct udp_port_cfg udp_conf = {0};
649 	struct udp_tunnel_sock_cfg tuncfg = {NULL};
650 	struct nlattr *opts[TIPC_NLA_UDP_MAX + 1];
651 	u8 node_id[NODE_ID_LEN] = {0,};
652 
653 	ub = kzalloc(sizeof(*ub), GFP_ATOMIC);
654 	if (!ub)
655 		return -ENOMEM;
656 
657 	INIT_LIST_HEAD(&ub->rcast.list);
658 
659 	if (!attrs[TIPC_NLA_BEARER_UDP_OPTS])
660 		goto err;
661 
662 	if (nla_parse_nested(opts, TIPC_NLA_UDP_MAX,
663 			     attrs[TIPC_NLA_BEARER_UDP_OPTS],
664 			     tipc_nl_udp_policy, NULL))
665 		goto err;
666 
667 	if (!opts[TIPC_NLA_UDP_LOCAL] || !opts[TIPC_NLA_UDP_REMOTE]) {
668 		pr_err("Invalid UDP bearer configuration");
669 		err = -EINVAL;
670 		goto err;
671 	}
672 
673 	err = tipc_parse_udp_addr(opts[TIPC_NLA_UDP_LOCAL], &local,
674 				  &ub->ifindex);
675 	if (err)
676 		goto err;
677 
678 	err = tipc_parse_udp_addr(opts[TIPC_NLA_UDP_REMOTE], &remote, NULL);
679 	if (err)
680 		goto err;
681 
682 	if (remote.proto != local.proto) {
683 		err = -EINVAL;
684 		goto err;
685 	}
686 
687 	/* Autoconfigure own node identity if needed */
688 	if (!tipc_own_id(net)) {
689 		memcpy(node_id, local.ipv6.in6_u.u6_addr8, 16);
690 		tipc_net_init(net, node_id, 0);
691 	}
692 	if (!tipc_own_id(net)) {
693 		pr_warn("Failed to set node id, please configure manually\n");
694 		err = -EINVAL;
695 		goto err;
696 	}
697 
698 	b->bcast_addr.media_id = TIPC_MEDIA_TYPE_UDP;
699 	b->bcast_addr.broadcast = TIPC_BROADCAST_SUPPORT;
700 	rcu_assign_pointer(b->media_ptr, ub);
701 	rcu_assign_pointer(ub->bearer, b);
702 	tipc_udp_media_addr_set(&b->addr, &local);
703 	if (local.proto == htons(ETH_P_IP)) {
704 		struct net_device *dev;
705 
706 		dev = __ip_dev_find(net, local.ipv4.s_addr, false);
707 		if (!dev) {
708 			err = -ENODEV;
709 			goto err;
710 		}
711 		udp_conf.family = AF_INET;
712 		udp_conf.local_ip.s_addr = htonl(INADDR_ANY);
713 		udp_conf.use_udp_checksums = false;
714 		ub->ifindex = dev->ifindex;
715 		if (tipc_mtu_bad(dev, sizeof(struct iphdr) +
716 				      sizeof(struct udphdr))) {
717 			err = -EINVAL;
718 			goto err;
719 		}
720 		b->mtu = b->media->mtu;
721 #if IS_ENABLED(CONFIG_IPV6)
722 	} else if (local.proto == htons(ETH_P_IPV6)) {
723 		udp_conf.family = AF_INET6;
724 		udp_conf.use_udp6_tx_checksums = true;
725 		udp_conf.use_udp6_rx_checksums = true;
726 		udp_conf.local_ip6 = in6addr_any;
727 		b->mtu = 1280;
728 #endif
729 	} else {
730 		err = -EAFNOSUPPORT;
731 		goto err;
732 	}
733 	udp_conf.local_udp_port = local.port;
734 	err = udp_sock_create(net, &udp_conf, &ub->ubsock);
735 	if (err)
736 		goto err;
737 	tuncfg.sk_user_data = ub;
738 	tuncfg.encap_type = 1;
739 	tuncfg.encap_rcv = tipc_udp_recv;
740 	tuncfg.encap_destroy = NULL;
741 	setup_udp_tunnel_sock(net, ub->ubsock, &tuncfg);
742 
743 	/**
744 	 * The bcast media address port is used for all peers and the ip
745 	 * is used if it's a multicast address.
746 	 */
747 	memcpy(&b->bcast_addr.value, &remote, sizeof(remote));
748 	if (tipc_udp_is_mcast_addr(&remote))
749 		err = enable_mcast(ub, &remote);
750 	else
751 		err = tipc_udp_rcast_add(b, &remote);
752 	if (err)
753 		goto err;
754 
755 	return 0;
756 err:
757 	if (ub->ubsock)
758 		udp_tunnel_sock_release(ub->ubsock);
759 	kfree(ub);
760 	return err;
761 }
762 
763 /* cleanup_bearer - break the socket/bearer association */
cleanup_bearer(struct work_struct * work)764 static void cleanup_bearer(struct work_struct *work)
765 {
766 	struct udp_bearer *ub = container_of(work, struct udp_bearer, work);
767 	struct udp_replicast *rcast, *tmp;
768 
769 	list_for_each_entry_safe(rcast, tmp, &ub->rcast.list, list) {
770 		list_del_rcu(&rcast->list);
771 		kfree_rcu(rcast, rcu);
772 	}
773 
774 	if (ub->ubsock)
775 		udp_tunnel_sock_release(ub->ubsock);
776 	synchronize_net();
777 	kfree(ub);
778 }
779 
780 /* tipc_udp_disable - detach bearer from socket */
tipc_udp_disable(struct tipc_bearer * b)781 static void tipc_udp_disable(struct tipc_bearer *b)
782 {
783 	struct udp_bearer *ub;
784 
785 	ub = rcu_dereference_rtnl(b->media_ptr);
786 	if (!ub) {
787 		pr_err("UDP bearer instance not found\n");
788 		return;
789 	}
790 	if (ub->ubsock)
791 		sock_set_flag(ub->ubsock->sk, SOCK_DEAD);
792 	RCU_INIT_POINTER(ub->bearer, NULL);
793 
794 	/* sock_release need to be done outside of rtnl lock */
795 	INIT_WORK(&ub->work, cleanup_bearer);
796 	schedule_work(&ub->work);
797 }
798 
799 struct tipc_media udp_media_info = {
800 	.send_msg	= tipc_udp_send_msg,
801 	.enable_media	= tipc_udp_enable,
802 	.disable_media	= tipc_udp_disable,
803 	.addr2str	= tipc_udp_addr2str,
804 	.addr2msg	= tipc_udp_addr2msg,
805 	.msg2addr	= tipc_udp_msg2addr,
806 	.priority	= TIPC_DEF_LINK_PRI,
807 	.tolerance	= TIPC_DEF_LINK_TOL,
808 	.window		= TIPC_DEF_LINK_WIN,
809 	.mtu		= TIPC_DEF_LINK_UDP_MTU,
810 	.type_id	= TIPC_MEDIA_TYPE_UDP,
811 	.hwaddr_len	= 0,
812 	.name		= "udp"
813 };
814