xref: /linux/net/tipc/udp_media.c (revision fab183d632628381b466a41479489541ac0e29a0)
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/wait_bit.h>
44 #include <linux/list.h>
45 #include <net/sock.h>
46 #include <net/ip.h>
47 #include <net/udp_tunnel.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 #include "udp_media.h"
56 
57 /* IANA assigned UDP port */
58 #define UDP_PORT_DEFAULT	6118
59 
60 #define UDP_MIN_HEADROOM        48
61 
62 /**
63  * struct udp_media_addr - IP/UDP addressing information
64  *
65  * This is the bearer level originating address used in neighbor discovery
66  * messages, and all fields should be in network byte order
67  *
68  * @proto: Ethernet protocol in use
69  * @port: port being used
70  * @ipv4: IPv4 address of neighbor
71  * @ipv6: IPv6 address of neighbor
72  */
73 struct udp_media_addr {
74 	__be16	proto;
75 	__be16	port;
76 	union {
77 		struct in_addr ipv4;
78 		struct in6_addr ipv6;
79 	};
80 };
81 
82 /* struct udp_replicast - container for UDP remote addresses */
83 struct udp_replicast {
84 	struct udp_media_addr addr;
85 	struct dst_cache dst_cache;
86 	struct rcu_head rcu;
87 	struct list_head list;
88 };
89 
90 /**
91  * struct udp_bearer - ip/udp bearer data structure
92  * @bearer:	associated generic tipc bearer
93  * @sk:		bearer associated socket
94  * @ifindex:	local address scope
95  * @work:	used to schedule deferred work on a bearer
96  * @rcast:	associated udp_replicast container
97  * @rcast_lock:	serialize updates to @rcast.list against concurrent updaters
98  * @disabled:	bearer is being torn down; reject further @rcast.list additions
99  */
100 struct udp_bearer {
101 	struct tipc_bearer __rcu *bearer;
102 	struct sock *sk;
103 	u32 ifindex;
104 	struct work_struct work;
105 	struct udp_replicast rcast;
106 	spinlock_t rcast_lock;
107 	bool disabled;
108 };
109 
tipc_udp_is_mcast_addr(struct udp_media_addr * addr)110 static int tipc_udp_is_mcast_addr(struct udp_media_addr *addr)
111 {
112 	if (ntohs(addr->proto) == ETH_P_IP)
113 		return ipv4_is_multicast(addr->ipv4.s_addr);
114 #if IS_ENABLED(CONFIG_IPV6)
115 	else
116 		return ipv6_addr_is_multicast(&addr->ipv6);
117 #endif
118 	return 0;
119 }
120 
121 /* 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)122 static void tipc_udp_media_addr_set(struct tipc_media_addr *addr,
123 				    struct udp_media_addr *ua)
124 {
125 	memset(addr, 0, sizeof(struct tipc_media_addr));
126 	addr->media_id = TIPC_MEDIA_TYPE_UDP;
127 	memcpy(addr->value, ua, sizeof(struct udp_media_addr));
128 
129 	if (tipc_udp_is_mcast_addr(ua))
130 		addr->broadcast = TIPC_BROADCAST_SUPPORT;
131 }
132 
133 /* tipc_udp_addr2str - convert ip/udp address to string */
tipc_udp_addr2str(struct tipc_media_addr * a,char * buf,int size)134 static int tipc_udp_addr2str(struct tipc_media_addr *a, char *buf, int size)
135 {
136 	struct udp_media_addr *ua = (struct udp_media_addr *)&a->value;
137 
138 	if (ntohs(ua->proto) == ETH_P_IP)
139 		snprintf(buf, size, "%pI4:%u", &ua->ipv4, ntohs(ua->port));
140 	else if (ntohs(ua->proto) == ETH_P_IPV6)
141 		snprintf(buf, size, "%pI6:%u", &ua->ipv6, ntohs(ua->port));
142 	else {
143 		pr_err("Invalid UDP media address\n");
144 		return 1;
145 	}
146 
147 	return 0;
148 }
149 
150 /* 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)151 static int tipc_udp_msg2addr(struct tipc_bearer *b, struct tipc_media_addr *a,
152 			     char *msg)
153 {
154 	struct udp_media_addr *ua;
155 
156 	ua = (struct udp_media_addr *) (msg + TIPC_MEDIA_ADDR_OFFSET);
157 	if (msg[TIPC_MEDIA_TYPE_OFFSET] != TIPC_MEDIA_TYPE_UDP)
158 		return -EINVAL;
159 	tipc_udp_media_addr_set(a, ua);
160 	return 0;
161 }
162 
163 /* tipc_udp_addr2msg - write an ip/udp address to a TIPC ndisc message */
tipc_udp_addr2msg(char * msg,struct tipc_media_addr * a)164 static int tipc_udp_addr2msg(char *msg, struct tipc_media_addr *a)
165 {
166 	memset(msg, 0, TIPC_MEDIA_INFO_SIZE);
167 	msg[TIPC_MEDIA_TYPE_OFFSET] = TIPC_MEDIA_TYPE_UDP;
168 	memcpy(msg + TIPC_MEDIA_ADDR_OFFSET, a->value,
169 	       sizeof(struct udp_media_addr));
170 	return 0;
171 }
172 
173 /* 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,struct dst_cache * cache)174 static int tipc_udp_xmit(struct net *net, struct sk_buff *skb,
175 			 struct udp_bearer *ub, struct udp_media_addr *src,
176 			 struct udp_media_addr *dst, struct dst_cache *cache)
177 {
178 	struct dst_entry *ndst;
179 	int ttl, err;
180 
181 	local_bh_disable();
182 	ndst = dst_cache_get(cache);
183 	if (dst->proto == htons(ETH_P_IP)) {
184 		struct rtable *rt = dst_rtable(ndst);
185 
186 		if (!rt) {
187 			struct flowi4 fl = {
188 				.daddr = dst->ipv4.s_addr,
189 				.saddr = src->ipv4.s_addr,
190 				.flowi4_mark = skb->mark,
191 				.flowi4_proto = IPPROTO_UDP
192 			};
193 			rt = ip_route_output_key(net, &fl);
194 			if (IS_ERR(rt)) {
195 				err = PTR_ERR(rt);
196 				goto tx_error;
197 			}
198 			dst_cache_set_ip4(cache, &rt->dst, fl.saddr);
199 		}
200 
201 		ttl = ip4_dst_hoplimit(&rt->dst);
202 		udp_tunnel_xmit_skb(rt, ub->sk, skb, src->ipv4.s_addr,
203 				    dst->ipv4.s_addr, 0, ttl, 0, src->port,
204 				    dst->port, false, true, 0);
205 #if IS_ENABLED(CONFIG_IPV6)
206 	} else {
207 		if (!ndst) {
208 			struct flowi6 fl6 = {
209 				.flowi6_oif = ub->ifindex,
210 				.daddr = dst->ipv6,
211 				.saddr = src->ipv6,
212 				.flowi6_proto = IPPROTO_UDP
213 			};
214 			ndst = ip6_dst_lookup_flow(net, ub->sk,
215 						   &fl6, NULL);
216 			if (IS_ERR(ndst)) {
217 				err = PTR_ERR(ndst);
218 				goto tx_error;
219 			}
220 			dst_cache_set_ip6(cache, ndst, &fl6.saddr);
221 		}
222 		ttl = ip6_dst_hoplimit(ndst);
223 		udp_tunnel6_xmit_skb(ndst, ub->sk, skb, NULL,
224 				     &src->ipv6, &dst->ipv6, 0, ttl, 0,
225 				     src->port, dst->port, false, 0);
226 #endif
227 	}
228 	local_bh_enable();
229 	return 0;
230 
231 tx_error:
232 	local_bh_enable();
233 	kfree_skb(skb);
234 	return err;
235 }
236 
tipc_udp_send_msg(struct net * net,struct sk_buff * skb,struct tipc_bearer * b,struct tipc_media_addr * addr)237 static int tipc_udp_send_msg(struct net *net, struct sk_buff *skb,
238 			     struct tipc_bearer *b,
239 			     struct tipc_media_addr *addr)
240 {
241 	struct udp_media_addr *src = (struct udp_media_addr *)&b->addr.value;
242 	struct udp_media_addr *dst = (struct udp_media_addr *)&addr->value;
243 	struct udp_replicast *rcast;
244 	struct udp_bearer *ub;
245 	int err = 0;
246 
247 	if (skb_headroom(skb) < UDP_MIN_HEADROOM) {
248 		err = pskb_expand_head(skb, UDP_MIN_HEADROOM, 0, GFP_ATOMIC);
249 		if (err)
250 			goto out;
251 	}
252 
253 	skb_set_inner_protocol(skb, htons(ETH_P_TIPC));
254 	ub = rcu_dereference(b->media_ptr);
255 	if (!ub) {
256 		err = -ENODEV;
257 		goto out;
258 	}
259 
260 	if (addr->broadcast != TIPC_REPLICAST_SUPPORT)
261 		return tipc_udp_xmit(net, skb, ub, src, dst,
262 				     &ub->rcast.dst_cache);
263 
264 	/* Replicast, send an skb to each configured IP address */
265 	list_for_each_entry_rcu(rcast, &ub->rcast.list, list) {
266 		struct sk_buff *_skb;
267 
268 		_skb = pskb_copy(skb, GFP_ATOMIC);
269 		if (!_skb) {
270 			err = -ENOMEM;
271 			goto out;
272 		}
273 
274 		err = tipc_udp_xmit(net, _skb, ub, src, &rcast->addr,
275 				    &rcast->dst_cache);
276 		if (err)
277 			goto out;
278 	}
279 	err = 0;
280 out:
281 	kfree_skb(skb);
282 	return err;
283 }
284 
tipc_udp_rcast_add(struct tipc_bearer * b,struct udp_media_addr * addr)285 static int tipc_udp_rcast_add(struct tipc_bearer *b,
286 			      struct udp_media_addr *addr)
287 {
288 	struct udp_replicast *rcast;
289 	struct udp_bearer *ub;
290 
291 	ub = rcu_dereference_rtnl(b->media_ptr);
292 	if (!ub)
293 		return -ENODEV;
294 
295 	spin_lock_bh(&ub->rcast_lock);
296 	if (ub->disabled) {
297 		spin_unlock_bh(&ub->rcast_lock);
298 		return 0;
299 	}
300 	list_for_each_entry(rcast, &ub->rcast.list, list) {
301 		if (!memcmp(&rcast->addr, addr, sizeof(*addr))) {
302 			spin_unlock_bh(&ub->rcast_lock);
303 			return 0;
304 		}
305 	}
306 
307 	rcast = kmalloc_obj(*rcast, GFP_ATOMIC);
308 	if (!rcast) {
309 		spin_unlock_bh(&ub->rcast_lock);
310 		return -ENOMEM;
311 	}
312 
313 	if (dst_cache_init(&rcast->dst_cache, GFP_ATOMIC)) {
314 		spin_unlock_bh(&ub->rcast_lock);
315 		kfree(rcast);
316 		return -ENOMEM;
317 	}
318 
319 	memcpy(&rcast->addr, addr, sizeof(struct udp_media_addr));
320 	list_add_rcu(&rcast->list, &ub->rcast.list);
321 	b->bcast_addr.broadcast = TIPC_REPLICAST_SUPPORT;
322 	spin_unlock_bh(&ub->rcast_lock);
323 
324 	if (ntohs(addr->proto) == ETH_P_IP)
325 		pr_info("New replicast peer: %pI4\n", &rcast->addr.ipv4);
326 #if IS_ENABLED(CONFIG_IPV6)
327 	else if (ntohs(addr->proto) == ETH_P_IPV6)
328 		pr_info("New replicast peer: %pI6\n", &rcast->addr.ipv6);
329 #endif
330 	return 0;
331 }
332 
tipc_udp_rcast_disc(struct tipc_bearer * b,struct sk_buff * skb)333 static int tipc_udp_rcast_disc(struct tipc_bearer *b, struct sk_buff *skb)
334 {
335 	struct udp_media_addr src = {0};
336 	struct udp_media_addr *dst;
337 
338 	dst = (struct udp_media_addr *)&b->bcast_addr.value;
339 	if (tipc_udp_is_mcast_addr(dst))
340 		return 0;
341 
342 	src.port = udp_hdr(skb)->source;
343 
344 	if (ip_hdr(skb)->version == 4) {
345 		struct iphdr *iphdr = ip_hdr(skb);
346 
347 		src.proto = htons(ETH_P_IP);
348 		src.ipv4.s_addr = iphdr->saddr;
349 		if (ipv4_is_multicast(iphdr->daddr))
350 			return 0;
351 #if IS_ENABLED(CONFIG_IPV6)
352 	} else if (ip_hdr(skb)->version == 6) {
353 		struct ipv6hdr *iphdr = ipv6_hdr(skb);
354 
355 		src.proto = htons(ETH_P_IPV6);
356 		src.ipv6 = iphdr->saddr;
357 		if (ipv6_addr_is_multicast(&iphdr->daddr))
358 			return 0;
359 #endif
360 	} else {
361 		return 0;
362 	}
363 
364 	return tipc_udp_rcast_add(b, &src);
365 }
366 
367 /* tipc_udp_recv - read data from bearer socket */
tipc_udp_recv(struct sock * sk,struct sk_buff * skb)368 static int tipc_udp_recv(struct sock *sk, struct sk_buff *skb)
369 {
370 	struct udp_bearer *ub;
371 	struct tipc_bearer *b;
372 	struct tipc_msg *hdr;
373 	int err;
374 
375 	ub = rcu_dereference_sk_user_data(sk);
376 	if (!ub) {
377 		pr_err_ratelimited("Failed to get UDP bearer reference");
378 		goto out;
379 	}
380 	skb_pull(skb, sizeof(struct udphdr));
381 	hdr = buf_msg(skb);
382 
383 	b = rcu_dereference(ub->bearer);
384 	if (!b)
385 		goto out;
386 
387 	if (b && test_bit(0, &b->up)) {
388 		TIPC_SKB_CB(skb)->flags = 0;
389 		tipc_rcv(sock_net(sk), skb, b);
390 		return 0;
391 	}
392 
393 	if (unlikely(msg_user(hdr) == LINK_CONFIG)) {
394 		err = tipc_udp_rcast_disc(b, skb);
395 		if (err)
396 			goto out;
397 	}
398 
399 out:
400 	kfree_skb(skb);
401 	return 0;
402 }
403 
enable_mcast(struct udp_bearer * ub,struct udp_media_addr * remote)404 static int enable_mcast(struct udp_bearer *ub, struct udp_media_addr *remote)
405 {
406 	struct sock *sk = ub->sk;
407 	struct ip_mreqn mreqn;
408 	int err = 0;
409 
410 	if (ntohs(remote->proto) == ETH_P_IP) {
411 		mreqn.imr_multiaddr = remote->ipv4;
412 		mreqn.imr_ifindex = ub->ifindex;
413 		err = ip_mc_join_group(sk, &mreqn);
414 #if IS_ENABLED(CONFIG_IPV6)
415 	} else {
416 		lock_sock(sk);
417 		err = ipv6_sock_mc_join(sk, ub->ifindex, &remote->ipv6);
418 		release_sock(sk);
419 #endif
420 	}
421 	return err;
422 }
423 
__tipc_nl_add_udp_addr(struct sk_buff * skb,struct udp_media_addr * addr,int nla_t)424 static int __tipc_nl_add_udp_addr(struct sk_buff *skb,
425 				  struct udp_media_addr *addr, int nla_t)
426 {
427 	if (ntohs(addr->proto) == ETH_P_IP) {
428 		struct sockaddr_in ip4;
429 
430 		memset(&ip4, 0, sizeof(ip4));
431 		ip4.sin_family = AF_INET;
432 		ip4.sin_port = addr->port;
433 		ip4.sin_addr.s_addr = addr->ipv4.s_addr;
434 		if (nla_put(skb, nla_t, sizeof(ip4), &ip4))
435 			return -EMSGSIZE;
436 
437 #if IS_ENABLED(CONFIG_IPV6)
438 	} else if (ntohs(addr->proto) == ETH_P_IPV6) {
439 		struct sockaddr_in6 ip6;
440 
441 		memset(&ip6, 0, sizeof(ip6));
442 		ip6.sin6_family = AF_INET6;
443 		ip6.sin6_port  = addr->port;
444 		memcpy(&ip6.sin6_addr, &addr->ipv6, sizeof(struct in6_addr));
445 		if (nla_put(skb, nla_t, sizeof(ip6), &ip6))
446 			return -EMSGSIZE;
447 #endif
448 	}
449 
450 	return 0;
451 }
452 
tipc_udp_nl_dump_remoteip(struct sk_buff * skb,struct netlink_callback * cb)453 int tipc_udp_nl_dump_remoteip(struct sk_buff *skb, struct netlink_callback *cb)
454 {
455 	u32 bid = cb->args[0];
456 	u32 skip_cnt = cb->args[1];
457 	u32 portid = NETLINK_CB(cb->skb).portid;
458 	struct udp_replicast *rcast, *tmp;
459 	struct tipc_bearer *b;
460 	struct udp_bearer *ub;
461 	void *hdr;
462 	int err;
463 	int i;
464 
465 	if (!bid && !skip_cnt) {
466 		struct nlattr **attrs = genl_dumpit_info(cb)->info.attrs;
467 		struct net *net = sock_net(skb->sk);
468 		struct nlattr *battrs[TIPC_NLA_BEARER_MAX + 1];
469 		char *bname;
470 
471 		if (!attrs[TIPC_NLA_BEARER])
472 			return -EINVAL;
473 
474 		err = nla_parse_nested_deprecated(battrs, TIPC_NLA_BEARER_MAX,
475 						  attrs[TIPC_NLA_BEARER],
476 						  tipc_nl_bearer_policy, NULL);
477 		if (err)
478 			return err;
479 
480 		if (!battrs[TIPC_NLA_BEARER_NAME])
481 			return -EINVAL;
482 
483 		bname = nla_data(battrs[TIPC_NLA_BEARER_NAME]);
484 
485 		rtnl_lock();
486 		b = tipc_bearer_find(net, bname);
487 		if (!b || b->bcast_addr.media_id != TIPC_MEDIA_TYPE_UDP) {
488 			rtnl_unlock();
489 			return -EINVAL;
490 		}
491 		bid = b->identity;
492 	} else {
493 		struct net *net = sock_net(skb->sk);
494 		struct tipc_net *tn = net_generic(net, tipc_net_id);
495 
496 		rtnl_lock();
497 		b = rtnl_dereference(tn->bearer_list[bid]);
498 		if (!b || b->bcast_addr.media_id != TIPC_MEDIA_TYPE_UDP) {
499 			rtnl_unlock();
500 			return -EINVAL;
501 		}
502 	}
503 
504 	ub = rtnl_dereference(b->media_ptr);
505 	if (!ub) {
506 		rtnl_unlock();
507 		return -EINVAL;
508 	}
509 
510 	i = 0;
511 	list_for_each_entry_safe(rcast, tmp, &ub->rcast.list, list) {
512 		if (i < skip_cnt)
513 			goto count;
514 
515 		hdr = genlmsg_put(skb, portid, cb->nlh->nlmsg_seq,
516 				  &tipc_genl_family, NLM_F_MULTI,
517 				  TIPC_NL_BEARER_GET);
518 		if (!hdr)
519 			goto done;
520 
521 		err = __tipc_nl_add_udp_addr(skb, &rcast->addr,
522 					     TIPC_NLA_UDP_REMOTE);
523 		if (err) {
524 			genlmsg_cancel(skb, hdr);
525 			goto done;
526 		}
527 		genlmsg_end(skb, hdr);
528 count:
529 		i++;
530 	}
531 done:
532 	rtnl_unlock();
533 	cb->args[0] = bid;
534 	cb->args[1] = i;
535 
536 	return skb->len;
537 }
538 
tipc_udp_nl_add_bearer_data(struct tipc_nl_msg * msg,struct tipc_bearer * b)539 int tipc_udp_nl_add_bearer_data(struct tipc_nl_msg *msg, struct tipc_bearer *b)
540 {
541 	struct udp_media_addr *src = (struct udp_media_addr *)&b->addr.value;
542 	struct udp_media_addr *dst;
543 	struct udp_bearer *ub;
544 	struct nlattr *nest;
545 
546 	ub = rtnl_dereference(b->media_ptr);
547 	if (!ub)
548 		return -ENODEV;
549 
550 	nest = nla_nest_start_noflag(msg->skb, TIPC_NLA_BEARER_UDP_OPTS);
551 	if (!nest)
552 		goto msg_full;
553 
554 	if (__tipc_nl_add_udp_addr(msg->skb, src, TIPC_NLA_UDP_LOCAL))
555 		goto msg_full;
556 
557 	dst = (struct udp_media_addr *)&b->bcast_addr.value;
558 	if (__tipc_nl_add_udp_addr(msg->skb, dst, TIPC_NLA_UDP_REMOTE))
559 		goto msg_full;
560 
561 	if (!list_empty(&ub->rcast.list)) {
562 		if (nla_put_flag(msg->skb, TIPC_NLA_UDP_MULTI_REMOTEIP))
563 			goto msg_full;
564 	}
565 
566 	nla_nest_end(msg->skb, nest);
567 	return 0;
568 msg_full:
569 	nla_nest_cancel(msg->skb, nest);
570 	return -EMSGSIZE;
571 }
572 
573 /**
574  * tipc_parse_udp_addr - build udp media address from netlink data
575  * @nla:	netlink attribute containing sockaddr storage aligned address
576  * @addr:	tipc media address to fill with address, port and protocol type
577  * @scope_id:	IPv6 scope id pointer, not NULL indicates it's required
578  */
579 
tipc_parse_udp_addr(struct nlattr * nla,struct udp_media_addr * addr,u32 * scope_id)580 static int tipc_parse_udp_addr(struct nlattr *nla, struct udp_media_addr *addr,
581 			       u32 *scope_id)
582 {
583 	struct sockaddr_storage sa;
584 
585 	nla_memcpy(&sa, nla, sizeof(sa));
586 	if (sa.ss_family == AF_INET) {
587 		struct sockaddr_in *ip4 = (struct sockaddr_in *)&sa;
588 
589 		addr->proto = htons(ETH_P_IP);
590 		addr->port = ip4->sin_port;
591 		addr->ipv4.s_addr = ip4->sin_addr.s_addr;
592 		return 0;
593 
594 #if IS_ENABLED(CONFIG_IPV6)
595 	} else if (sa.ss_family == AF_INET6) {
596 		struct sockaddr_in6 *ip6 = (struct sockaddr_in6 *)&sa;
597 
598 		addr->proto = htons(ETH_P_IPV6);
599 		addr->port = ip6->sin6_port;
600 		memcpy(&addr->ipv6, &ip6->sin6_addr, sizeof(struct in6_addr));
601 
602 		/* Scope ID is only interesting for local addresses */
603 		if (scope_id) {
604 			int atype;
605 
606 			atype = ipv6_addr_type(&ip6->sin6_addr);
607 			if (__ipv6_addr_needs_scope_id(atype) &&
608 			    !ip6->sin6_scope_id) {
609 				return -EINVAL;
610 			}
611 
612 			*scope_id = ip6->sin6_scope_id ? : 0;
613 		}
614 
615 		return 0;
616 #endif
617 	}
618 	return -EADDRNOTAVAIL;
619 }
620 
tipc_udp_nl_bearer_add(struct tipc_bearer * b,struct nlattr * attr)621 int tipc_udp_nl_bearer_add(struct tipc_bearer *b, struct nlattr *attr)
622 {
623 	int err;
624 	struct udp_media_addr addr = {0};
625 	struct nlattr *opts[TIPC_NLA_UDP_MAX + 1];
626 	struct udp_media_addr *dst;
627 
628 	if (nla_parse_nested_deprecated(opts, TIPC_NLA_UDP_MAX, attr, tipc_nl_udp_policy, NULL))
629 		return -EINVAL;
630 
631 	if (!opts[TIPC_NLA_UDP_REMOTE])
632 		return -EINVAL;
633 
634 	err = tipc_parse_udp_addr(opts[TIPC_NLA_UDP_REMOTE], &addr, NULL);
635 	if (err)
636 		return err;
637 
638 	dst = (struct udp_media_addr *)&b->bcast_addr.value;
639 	if (tipc_udp_is_mcast_addr(dst)) {
640 		pr_err("Can't add remote ip to TIPC UDP multicast bearer\n");
641 		return -EINVAL;
642 	}
643 
644 	return tipc_udp_rcast_add(b, &addr);
645 }
646 
647 /**
648  * tipc_udp_enable - callback to create a new udp bearer instance
649  * @net:	network namespace
650  * @b:		pointer to generic tipc_bearer
651  * @attrs:	netlink bearer configuration
652  *
653  * validate the bearer parameters and initialize the udp bearer
654  * rtnl_lock should be held
655  */
tipc_udp_enable(struct net * net,struct tipc_bearer * b,struct nlattr * attrs[])656 static int tipc_udp_enable(struct net *net, struct tipc_bearer *b,
657 			   struct nlattr *attrs[])
658 {
659 	int err = -EINVAL;
660 	struct udp_bearer *ub;
661 	struct udp_media_addr remote = {0};
662 	struct udp_media_addr local = {0};
663 	struct udp_port_cfg udp_conf = {0};
664 	struct udp_tunnel_sock_cfg tuncfg = {NULL};
665 	struct nlattr *opts[TIPC_NLA_UDP_MAX + 1];
666 	u8 node_id[NODE_ID_LEN] = {0,};
667 	struct net_device *dev;
668 	struct socket *sock;
669 	int rmcast = 0;
670 
671 	ub = kzalloc_obj(*ub, GFP_ATOMIC);
672 	if (!ub)
673 		return -ENOMEM;
674 
675 	INIT_LIST_HEAD(&ub->rcast.list);
676 	spin_lock_init(&ub->rcast_lock);
677 
678 	if (!attrs[TIPC_NLA_BEARER_UDP_OPTS])
679 		goto err;
680 
681 	if (nla_parse_nested_deprecated(opts, TIPC_NLA_UDP_MAX, attrs[TIPC_NLA_BEARER_UDP_OPTS], tipc_nl_udp_policy, NULL))
682 		goto err;
683 
684 	if (!opts[TIPC_NLA_UDP_LOCAL] || !opts[TIPC_NLA_UDP_REMOTE]) {
685 		pr_err("Invalid UDP bearer configuration");
686 		err = -EINVAL;
687 		goto err;
688 	}
689 
690 	err = tipc_parse_udp_addr(opts[TIPC_NLA_UDP_LOCAL], &local,
691 				  &ub->ifindex);
692 	if (err)
693 		goto err;
694 
695 	err = tipc_parse_udp_addr(opts[TIPC_NLA_UDP_REMOTE], &remote, NULL);
696 	if (err)
697 		goto err;
698 
699 	if (remote.proto != local.proto) {
700 		err = -EINVAL;
701 		goto err;
702 	}
703 
704 	/* Checking remote ip address */
705 	rmcast = tipc_udp_is_mcast_addr(&remote);
706 
707 	/* Autoconfigure own node identity if needed */
708 	if (!tipc_own_id(net)) {
709 		memcpy(node_id, local.ipv6.in6_u.u6_addr8, 16);
710 		tipc_net_init(net, node_id, 0);
711 	}
712 	if (!tipc_own_id(net)) {
713 		pr_warn("Failed to set node id, please configure manually\n");
714 		err = -EINVAL;
715 		goto err;
716 	}
717 
718 	b->bcast_addr.media_id = TIPC_MEDIA_TYPE_UDP;
719 	b->bcast_addr.broadcast = TIPC_BROADCAST_SUPPORT;
720 	rcu_assign_pointer(b->media_ptr, ub);
721 	rcu_assign_pointer(ub->bearer, b);
722 	tipc_udp_media_addr_set(&b->addr, &local);
723 	if (local.proto == htons(ETH_P_IP)) {
724 		dev = __ip_dev_find(net, local.ipv4.s_addr, false);
725 		if (!dev) {
726 			err = -ENODEV;
727 			goto err;
728 		}
729 		udp_conf.family = AF_INET;
730 
731 		/* Switch to use ANY to receive packets from group */
732 		if (rmcast)
733 			udp_conf.local_ip.s_addr = htonl(INADDR_ANY);
734 		else
735 			udp_conf.local_ip.s_addr = local.ipv4.s_addr;
736 		udp_conf.use_udp_checksums = false;
737 		ub->ifindex = dev->ifindex;
738 		b->encap_hlen = sizeof(struct iphdr) + sizeof(struct udphdr);
739 		b->mtu = b->media->mtu;
740 #if IS_ENABLED(CONFIG_IPV6)
741 	} else if (local.proto == htons(ETH_P_IPV6)) {
742 		dev = ub->ifindex ? __dev_get_by_index(net, ub->ifindex) : NULL;
743 		dev = ipv6_dev_find(net, &local.ipv6, dev);
744 		if (!dev) {
745 			err = -ENODEV;
746 			goto err;
747 		}
748 		udp_conf.family = AF_INET6;
749 		udp_conf.use_udp6_tx_checksums = true;
750 		udp_conf.use_udp6_rx_checksums = true;
751 		if (rmcast)
752 			udp_conf.local_ip6 = in6addr_any;
753 		else
754 			udp_conf.local_ip6 = local.ipv6;
755 		ub->ifindex = dev->ifindex;
756 		b->encap_hlen = sizeof(struct ipv6hdr) + sizeof(struct udphdr);
757 		b->mtu = 1280;
758 #endif
759 	} else {
760 		err = -EAFNOSUPPORT;
761 		goto err;
762 	}
763 	udp_conf.local_udp_port = local.port;
764 	err = udp_sock_create(net, &udp_conf, &sock);
765 	if (err)
766 		goto err;
767 
768 	ub->sk = sock->sk;
769 	tuncfg.sk_user_data = ub;
770 	tuncfg.encap_type = 1;
771 	tuncfg.encap_rcv = tipc_udp_recv;
772 	tuncfg.encap_destroy = NULL;
773 	setup_udp_tunnel_sock(net, ub->sk, &tuncfg);
774 
775 	err = dst_cache_init(&ub->rcast.dst_cache, GFP_ATOMIC);
776 	if (err)
777 		goto free;
778 
779 	/*
780 	 * The bcast media address port is used for all peers and the ip
781 	 * is used if it's a multicast address.
782 	 */
783 	memcpy(&b->bcast_addr.value, &remote, sizeof(remote));
784 	if (rmcast)
785 		err = enable_mcast(ub, &remote);
786 	else
787 		err = tipc_udp_rcast_add(b, &remote);
788 	if (err)
789 		goto free;
790 
791 	return 0;
792 
793 free:
794 	dst_cache_destroy(&ub->rcast.dst_cache);
795 	udp_tunnel_sock_release(ub->sk);
796 	synchronize_rcu();
797 err:
798 	kfree(ub);
799 	return err;
800 }
801 
rcast_free_rcu(struct rcu_head * rcu)802 static void rcast_free_rcu(struct rcu_head *rcu)
803 {
804 	struct udp_replicast *rcast = container_of(rcu, struct udp_replicast, rcu);
805 
806 	dst_cache_destroy(&rcast->dst_cache);
807 	kfree(rcast);
808 }
809 
810 /* cleanup_bearer - break the socket/bearer association */
cleanup_bearer(struct work_struct * work)811 static void cleanup_bearer(struct work_struct *work)
812 {
813 	struct udp_bearer *ub = container_of(work, struct udp_bearer, work);
814 	struct udp_replicast *rcast, *tmp;
815 	struct tipc_net *tn;
816 
817 	spin_lock_bh(&ub->rcast_lock);
818 	list_for_each_entry_safe(rcast, tmp, &ub->rcast.list, list) {
819 		list_del_rcu(&rcast->list);
820 		call_rcu_hurry(&rcast->rcu, rcast_free_rcu);
821 	}
822 	ub->disabled = true;
823 	spin_unlock_bh(&ub->rcast_lock);
824 
825 	tn = tipc_net(sock_net(ub->sk));
826 
827 	udp_tunnel_sock_release(ub->sk);
828 
829 	synchronize_net();
830 
831 	dst_cache_destroy(&ub->rcast.dst_cache);
832 	if (atomic_dec_and_test(&tn->wq_count))
833 		wake_up_var(&tn->wq_count);
834 	kfree(ub);
835 }
836 
837 /* tipc_udp_disable - detach bearer from socket */
tipc_udp_disable(struct tipc_bearer * b)838 static void tipc_udp_disable(struct tipc_bearer *b)
839 {
840 	struct udp_bearer *ub;
841 
842 	ub = rtnl_dereference(b->media_ptr);
843 	if (!ub) {
844 		pr_err("UDP bearer instance not found\n");
845 		return;
846 	}
847 	sock_set_flag(ub->sk, SOCK_DEAD);
848 	RCU_INIT_POINTER(ub->bearer, NULL);
849 
850 	/* sock_release need to be done outside of rtnl lock */
851 	atomic_inc(&tipc_net(sock_net(ub->sk))->wq_count);
852 	INIT_WORK(&ub->work, cleanup_bearer);
853 	schedule_work(&ub->work);
854 }
855 
856 struct tipc_media udp_media_info = {
857 	.send_msg	= tipc_udp_send_msg,
858 	.enable_media	= tipc_udp_enable,
859 	.disable_media	= tipc_udp_disable,
860 	.addr2str	= tipc_udp_addr2str,
861 	.addr2msg	= tipc_udp_addr2msg,
862 	.msg2addr	= tipc_udp_msg2addr,
863 	.priority	= TIPC_DEF_LINK_PRI,
864 	.tolerance	= TIPC_DEF_LINK_TOL,
865 	.min_win	= TIPC_DEF_LINK_WIN,
866 	.max_win	= TIPC_DEF_LINK_WIN,
867 	.mtu		= TIPC_DEF_LINK_UDP_MTU,
868 	.type_id	= TIPC_MEDIA_TYPE_UDP,
869 	.hwaddr_len	= 0,
870 	.name		= "udp"
871 };
872