7413cbc9b638af492d6a592df957a3229fa6a3fa
[muen/linux.git] / net / tipc / udp_media.c
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/ipv6_stubs.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
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 */
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 */
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 */
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 */
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 */
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                 skb->dev = rt->dst.dev;
180                 ttl = ip4_dst_hoplimit(&rt->dst);
181                 udp_tunnel_xmit_skb(rt, ub->ubsock->sk, skb, src->ipv4.s_addr,
182                                     dst->ipv4.s_addr, 0, ttl, 0, src->port,
183                                     dst->port, false, true);
184 #if IS_ENABLED(CONFIG_IPV6)
185         } else {
186                 struct dst_entry *ndst;
187                 struct flowi6 fl6 = {
188                         .flowi6_oif = ub->ifindex,
189                         .daddr = dst->ipv6,
190                         .saddr = src->ipv6,
191                         .flowi6_proto = IPPROTO_UDP
192                 };
193                 err = ipv6_stub->ipv6_dst_lookup(net, ub->ubsock->sk, &ndst,
194                                                  &fl6);
195                 if (err)
196                         goto tx_error;
197                 ttl = ip6_dst_hoplimit(ndst);
198                 err = udp_tunnel6_xmit_skb(ndst, ub->ubsock->sk, skb,
199                                            ndst->dev, &src->ipv6,
200                                            &dst->ipv6, 0, ttl, 0, src->port,
201                                            dst->port, false);
202 #endif
203         }
204         return err;
205
206 tx_error:
207         kfree_skb(skb);
208         return err;
209 }
210
211 static int tipc_udp_send_msg(struct net *net, struct sk_buff *skb,
212                              struct tipc_bearer *b,
213                              struct tipc_media_addr *addr)
214 {
215         struct udp_media_addr *src = (struct udp_media_addr *)&b->addr.value;
216         struct udp_media_addr *dst = (struct udp_media_addr *)&addr->value;
217         struct udp_replicast *rcast;
218         struct udp_bearer *ub;
219         int err = 0;
220
221         if (skb_headroom(skb) < UDP_MIN_HEADROOM) {
222                 err = pskb_expand_head(skb, UDP_MIN_HEADROOM, 0, GFP_ATOMIC);
223                 if (err)
224                         goto out;
225         }
226
227         skb_set_inner_protocol(skb, htons(ETH_P_TIPC));
228         ub = rcu_dereference_rtnl(b->media_ptr);
229         if (!ub) {
230                 err = -ENODEV;
231                 goto out;
232         }
233
234         if (addr->broadcast != TIPC_REPLICAST_SUPPORT)
235                 return tipc_udp_xmit(net, skb, ub, src, dst);
236
237         /* Replicast, send an skb to each configured IP address */
238         list_for_each_entry_rcu(rcast, &ub->rcast.list, list) {
239                 struct sk_buff *_skb;
240
241                 _skb = pskb_copy(skb, GFP_ATOMIC);
242                 if (!_skb) {
243                         err = -ENOMEM;
244                         goto out;
245                 }
246
247                 err = tipc_udp_xmit(net, _skb, ub, src, &rcast->addr);
248                 if (err)
249                         goto out;
250         }
251         err = 0;
252 out:
253         kfree_skb(skb);
254         return err;
255 }
256
257 static bool tipc_udp_is_known_peer(struct tipc_bearer *b,
258                                    struct udp_media_addr *addr)
259 {
260         struct udp_replicast *rcast, *tmp;
261         struct udp_bearer *ub;
262
263         ub = rcu_dereference_rtnl(b->media_ptr);
264         if (!ub) {
265                 pr_err_ratelimited("UDP bearer instance not found\n");
266                 return false;
267         }
268
269         list_for_each_entry_safe(rcast, tmp, &ub->rcast.list, list) {
270                 if (!memcmp(&rcast->addr, addr, sizeof(struct udp_media_addr)))
271                         return true;
272         }
273
274         return false;
275 }
276
277 static int tipc_udp_rcast_add(struct tipc_bearer *b,
278                               struct udp_media_addr *addr)
279 {
280         struct udp_replicast *rcast;
281         struct udp_bearer *ub;
282
283         ub = rcu_dereference_rtnl(b->media_ptr);
284         if (!ub)
285                 return -ENODEV;
286
287         rcast = kmalloc(sizeof(*rcast), GFP_ATOMIC);
288         if (!rcast)
289                 return -ENOMEM;
290
291         memcpy(&rcast->addr, addr, sizeof(struct udp_media_addr));
292
293         if (ntohs(addr->proto) == ETH_P_IP)
294                 pr_info("New replicast peer: %pI4\n", &rcast->addr.ipv4);
295 #if IS_ENABLED(CONFIG_IPV6)
296         else if (ntohs(addr->proto) == ETH_P_IPV6)
297                 pr_info("New replicast peer: %pI6\n", &rcast->addr.ipv6);
298 #endif
299         b->bcast_addr.broadcast = TIPC_REPLICAST_SUPPORT;
300         list_add_rcu(&rcast->list, &ub->rcast.list);
301         return 0;
302 }
303
304 static int tipc_udp_rcast_disc(struct tipc_bearer *b, struct sk_buff *skb)
305 {
306         struct udp_media_addr src = {0};
307         struct udp_media_addr *dst;
308
309         dst = (struct udp_media_addr *)&b->bcast_addr.value;
310         if (tipc_udp_is_mcast_addr(dst))
311                 return 0;
312
313         src.port = udp_hdr(skb)->source;
314
315         if (ip_hdr(skb)->version == 4) {
316                 struct iphdr *iphdr = ip_hdr(skb);
317
318                 src.proto = htons(ETH_P_IP);
319                 src.ipv4.s_addr = iphdr->saddr;
320                 if (ipv4_is_multicast(iphdr->daddr))
321                         return 0;
322 #if IS_ENABLED(CONFIG_IPV6)
323         } else if (ip_hdr(skb)->version == 6) {
324                 struct ipv6hdr *iphdr = ipv6_hdr(skb);
325
326                 src.proto = htons(ETH_P_IPV6);
327                 src.ipv6 = iphdr->saddr;
328                 if (ipv6_addr_is_multicast(&iphdr->daddr))
329                         return 0;
330 #endif
331         } else {
332                 return 0;
333         }
334
335         if (likely(tipc_udp_is_known_peer(b, &src)))
336                 return 0;
337
338         return tipc_udp_rcast_add(b, &src);
339 }
340
341 /* tipc_udp_recv - read data from bearer socket */
342 static int tipc_udp_recv(struct sock *sk, struct sk_buff *skb)
343 {
344         struct udp_bearer *ub;
345         struct tipc_bearer *b;
346         struct tipc_msg *hdr;
347         int err;
348
349         ub = rcu_dereference_sk_user_data(sk);
350         if (!ub) {
351                 pr_err_ratelimited("Failed to get UDP bearer reference");
352                 goto out;
353         }
354         skb_pull(skb, sizeof(struct udphdr));
355         hdr = buf_msg(skb);
356
357         b = rcu_dereference(ub->bearer);
358         if (!b)
359                 goto out;
360
361         if (b && test_bit(0, &b->up)) {
362                 tipc_rcv(sock_net(sk), skb, b);
363                 rcu_read_unlock();
364                 return 0;
365         }
366
367         if (unlikely(msg_user(hdr) == LINK_CONFIG)) {
368                 err = tipc_udp_rcast_disc(b, skb);
369                 if (err)
370                         goto out;
371         }
372
373 out:
374         kfree_skb(skb);
375         return 0;
376 }
377
378 static int enable_mcast(struct udp_bearer *ub, struct udp_media_addr *remote)
379 {
380         int err = 0;
381         struct ip_mreqn mreqn;
382         struct sock *sk = ub->ubsock->sk;
383
384         if (ntohs(remote->proto) == ETH_P_IP) {
385                 mreqn.imr_multiaddr = remote->ipv4;
386                 mreqn.imr_ifindex = ub->ifindex;
387                 err = ip_mc_join_group(sk, &mreqn);
388 #if IS_ENABLED(CONFIG_IPV6)
389         } else {
390                 err = ipv6_stub->ipv6_sock_mc_join(sk, ub->ifindex,
391                                                    &remote->ipv6);
392 #endif
393         }
394         return err;
395 }
396
397 static int __tipc_nl_add_udp_addr(struct sk_buff *skb,
398                                   struct udp_media_addr *addr, int nla_t)
399 {
400         if (ntohs(addr->proto) == ETH_P_IP) {
401                 struct sockaddr_in ip4;
402
403                 memset(&ip4, 0, sizeof(ip4));
404                 ip4.sin_family = AF_INET;
405                 ip4.sin_port = addr->port;
406                 ip4.sin_addr.s_addr = addr->ipv4.s_addr;
407                 if (nla_put(skb, nla_t, sizeof(ip4), &ip4))
408                         return -EMSGSIZE;
409
410 #if IS_ENABLED(CONFIG_IPV6)
411         } else if (ntohs(addr->proto) == ETH_P_IPV6) {
412                 struct sockaddr_in6 ip6;
413
414                 memset(&ip6, 0, sizeof(ip6));
415                 ip6.sin6_family = AF_INET6;
416                 ip6.sin6_port  = addr->port;
417                 memcpy(&ip6.sin6_addr, &addr->ipv6, sizeof(struct in6_addr));
418                 if (nla_put(skb, nla_t, sizeof(ip6), &ip6))
419                         return -EMSGSIZE;
420 #endif
421         }
422
423         return 0;
424 }
425
426 int tipc_udp_nl_dump_remoteip(struct sk_buff *skb, struct netlink_callback *cb)
427 {
428         u32 bid = cb->args[0];
429         u32 skip_cnt = cb->args[1];
430         u32 portid = NETLINK_CB(cb->skb).portid;
431         struct udp_replicast *rcast, *tmp;
432         struct tipc_bearer *b;
433         struct udp_bearer *ub;
434         void *hdr;
435         int err;
436         int i;
437
438         if (!bid && !skip_cnt) {
439                 struct net *net = sock_net(skb->sk);
440                 struct nlattr *battrs[TIPC_NLA_BEARER_MAX + 1];
441                 struct nlattr **attrs;
442                 char *bname;
443
444                 err = tipc_nlmsg_parse(cb->nlh, &attrs);
445                 if (err)
446                         return err;
447
448                 if (!attrs[TIPC_NLA_BEARER])
449                         return -EINVAL;
450
451                 err = nla_parse_nested(battrs, TIPC_NLA_BEARER_MAX,
452                                        attrs[TIPC_NLA_BEARER],
453                                        tipc_nl_bearer_policy, NULL);
454                 if (err)
455                         return err;
456
457                 if (!battrs[TIPC_NLA_BEARER_NAME])
458                         return -EINVAL;
459
460                 bname = nla_data(battrs[TIPC_NLA_BEARER_NAME]);
461
462                 rtnl_lock();
463                 b = tipc_bearer_find(net, bname);
464                 if (!b) {
465                         rtnl_unlock();
466                         return -EINVAL;
467                 }
468                 bid = b->identity;
469         } else {
470                 struct net *net = sock_net(skb->sk);
471                 struct tipc_net *tn = net_generic(net, tipc_net_id);
472
473                 rtnl_lock();
474                 b = rtnl_dereference(tn->bearer_list[bid]);
475                 if (!b) {
476                         rtnl_unlock();
477                         return -EINVAL;
478                 }
479         }
480
481         ub = rcu_dereference_rtnl(b->media_ptr);
482         if (!ub) {
483                 rtnl_unlock();
484                 return -EINVAL;
485         }
486
487         i = 0;
488         list_for_each_entry_safe(rcast, tmp, &ub->rcast.list, list) {
489                 if (i < skip_cnt)
490                         goto count;
491
492                 hdr = genlmsg_put(skb, portid, cb->nlh->nlmsg_seq,
493                                   &tipc_genl_family, NLM_F_MULTI,
494                                   TIPC_NL_BEARER_GET);
495                 if (!hdr)
496                         goto done;
497
498                 err = __tipc_nl_add_udp_addr(skb, &rcast->addr,
499                                              TIPC_NLA_UDP_REMOTE);
500                 if (err) {
501                         genlmsg_cancel(skb, hdr);
502                         goto done;
503                 }
504                 genlmsg_end(skb, hdr);
505 count:
506                 i++;
507         }
508 done:
509         rtnl_unlock();
510         cb->args[0] = bid;
511         cb->args[1] = i;
512
513         return skb->len;
514 }
515
516 int tipc_udp_nl_add_bearer_data(struct tipc_nl_msg *msg, struct tipc_bearer *b)
517 {
518         struct udp_media_addr *src = (struct udp_media_addr *)&b->addr.value;
519         struct udp_media_addr *dst;
520         struct udp_bearer *ub;
521         struct nlattr *nest;
522
523         ub = rcu_dereference_rtnl(b->media_ptr);
524         if (!ub)
525                 return -ENODEV;
526
527         nest = nla_nest_start(msg->skb, TIPC_NLA_BEARER_UDP_OPTS);
528         if (!nest)
529                 goto msg_full;
530
531         if (__tipc_nl_add_udp_addr(msg->skb, src, TIPC_NLA_UDP_LOCAL))
532                 goto msg_full;
533
534         dst = (struct udp_media_addr *)&b->bcast_addr.value;
535         if (__tipc_nl_add_udp_addr(msg->skb, dst, TIPC_NLA_UDP_REMOTE))
536                 goto msg_full;
537
538         if (!list_empty(&ub->rcast.list)) {
539                 if (nla_put_flag(msg->skb, TIPC_NLA_UDP_MULTI_REMOTEIP))
540                         goto msg_full;
541         }
542
543         nla_nest_end(msg->skb, nest);
544         return 0;
545 msg_full:
546         nla_nest_cancel(msg->skb, nest);
547         return -EMSGSIZE;
548 }
549
550 /**
551  * tipc_parse_udp_addr - build udp media address from netlink data
552  * @nlattr:     netlink attribute containing sockaddr storage aligned address
553  * @addr:       tipc media address to fill with address, port and protocol type
554  * @scope_id:   IPv6 scope id pointer, not NULL indicates it's required
555  */
556
557 static int tipc_parse_udp_addr(struct nlattr *nla, struct udp_media_addr *addr,
558                                u32 *scope_id)
559 {
560         struct sockaddr_storage sa;
561
562         nla_memcpy(&sa, nla, sizeof(sa));
563         if (sa.ss_family == AF_INET) {
564                 struct sockaddr_in *ip4 = (struct sockaddr_in *)&sa;
565
566                 addr->proto = htons(ETH_P_IP);
567                 addr->port = ip4->sin_port;
568                 addr->ipv4.s_addr = ip4->sin_addr.s_addr;
569                 return 0;
570
571 #if IS_ENABLED(CONFIG_IPV6)
572         } else if (sa.ss_family == AF_INET6) {
573                 struct sockaddr_in6 *ip6 = (struct sockaddr_in6 *)&sa;
574
575                 addr->proto = htons(ETH_P_IPV6);
576                 addr->port = ip6->sin6_port;
577                 memcpy(&addr->ipv6, &ip6->sin6_addr, sizeof(struct in6_addr));
578
579                 /* Scope ID is only interesting for local addresses */
580                 if (scope_id) {
581                         int atype;
582
583                         atype = ipv6_addr_type(&ip6->sin6_addr);
584                         if (__ipv6_addr_needs_scope_id(atype) &&
585                             !ip6->sin6_scope_id) {
586                                 return -EINVAL;
587                         }
588
589                         *scope_id = ip6->sin6_scope_id ? : 0;
590                 }
591
592                 return 0;
593 #endif
594         }
595         return -EADDRNOTAVAIL;
596 }
597
598 int tipc_udp_nl_bearer_add(struct tipc_bearer *b, struct nlattr *attr)
599 {
600         int err;
601         struct udp_media_addr addr = {0};
602         struct nlattr *opts[TIPC_NLA_UDP_MAX + 1];
603         struct udp_media_addr *dst;
604
605         if (nla_parse_nested(opts, TIPC_NLA_UDP_MAX, attr,
606                              tipc_nl_udp_policy, NULL))
607                 return -EINVAL;
608
609         if (!opts[TIPC_NLA_UDP_REMOTE])
610                 return -EINVAL;
611
612         err = tipc_parse_udp_addr(opts[TIPC_NLA_UDP_REMOTE], &addr, NULL);
613         if (err)
614                 return err;
615
616         dst = (struct udp_media_addr *)&b->bcast_addr.value;
617         if (tipc_udp_is_mcast_addr(dst)) {
618                 pr_err("Can't add remote ip to TIPC UDP multicast bearer\n");
619                 return -EINVAL;
620         }
621
622         if (tipc_udp_is_known_peer(b, &addr))
623                 return 0;
624
625         return tipc_udp_rcast_add(b, &addr);
626 }
627
628 /**
629  * tipc_udp_enable - callback to create a new udp bearer instance
630  * @net:        network namespace
631  * @b:          pointer to generic tipc_bearer
632  * @attrs:      netlink bearer configuration
633  *
634  * validate the bearer parameters and initialize the udp bearer
635  * rtnl_lock should be held
636  */
637 static int tipc_udp_enable(struct net *net, struct tipc_bearer *b,
638                            struct nlattr *attrs[])
639 {
640         int err = -EINVAL;
641         struct udp_bearer *ub;
642         struct udp_media_addr remote = {0};
643         struct udp_media_addr local = {0};
644         struct udp_port_cfg udp_conf = {0};
645         struct udp_tunnel_sock_cfg tuncfg = {NULL};
646         struct nlattr *opts[TIPC_NLA_UDP_MAX + 1];
647         u8 node_id[NODE_ID_LEN] = {0,};
648         int rmcast = 0;
649
650         ub = kzalloc(sizeof(*ub), GFP_ATOMIC);
651         if (!ub)
652                 return -ENOMEM;
653
654         INIT_LIST_HEAD(&ub->rcast.list);
655
656         if (!attrs[TIPC_NLA_BEARER_UDP_OPTS])
657                 goto err;
658
659         if (nla_parse_nested(opts, TIPC_NLA_UDP_MAX,
660                              attrs[TIPC_NLA_BEARER_UDP_OPTS],
661                              tipc_nl_udp_policy, NULL))
662                 goto err;
663
664         if (!opts[TIPC_NLA_UDP_LOCAL] || !opts[TIPC_NLA_UDP_REMOTE]) {
665                 pr_err("Invalid UDP bearer configuration");
666                 err = -EINVAL;
667                 goto err;
668         }
669
670         err = tipc_parse_udp_addr(opts[TIPC_NLA_UDP_LOCAL], &local,
671                                   &ub->ifindex);
672         if (err)
673                 goto err;
674
675         err = tipc_parse_udp_addr(opts[TIPC_NLA_UDP_REMOTE], &remote, NULL);
676         if (err)
677                 goto err;
678
679         if (remote.proto != local.proto) {
680                 err = -EINVAL;
681                 goto err;
682         }
683
684         /* Checking remote ip address */
685         rmcast = tipc_udp_is_mcast_addr(&remote);
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
713                 /* Switch to use ANY to receive packets from group */
714                 if (rmcast)
715                         udp_conf.local_ip.s_addr = htonl(INADDR_ANY);
716                 else
717                         udp_conf.local_ip.s_addr = local.ipv4.s_addr;
718                 udp_conf.use_udp_checksums = false;
719                 ub->ifindex = dev->ifindex;
720                 if (tipc_mtu_bad(dev, sizeof(struct iphdr) +
721                                       sizeof(struct udphdr))) {
722                         err = -EINVAL;
723                         goto err;
724                 }
725                 b->mtu = b->media->mtu;
726 #if IS_ENABLED(CONFIG_IPV6)
727         } else if (local.proto == htons(ETH_P_IPV6)) {
728                 udp_conf.family = AF_INET6;
729                 udp_conf.use_udp6_tx_checksums = true;
730                 udp_conf.use_udp6_rx_checksums = true;
731                 if (rmcast)
732                         udp_conf.local_ip6 = in6addr_any;
733                 else
734                         udp_conf.local_ip6 = local.ipv6;
735                 b->mtu = 1280;
736 #endif
737         } else {
738                 err = -EAFNOSUPPORT;
739                 goto err;
740         }
741         udp_conf.local_udp_port = local.port;
742         err = udp_sock_create(net, &udp_conf, &ub->ubsock);
743         if (err)
744                 goto err;
745         tuncfg.sk_user_data = ub;
746         tuncfg.encap_type = 1;
747         tuncfg.encap_rcv = tipc_udp_recv;
748         tuncfg.encap_destroy = NULL;
749         setup_udp_tunnel_sock(net, ub->ubsock, &tuncfg);
750
751         /**
752          * The bcast media address port is used for all peers and the ip
753          * is used if it's a multicast address.
754          */
755         memcpy(&b->bcast_addr.value, &remote, sizeof(remote));
756         if (rmcast)
757                 err = enable_mcast(ub, &remote);
758         else
759                 err = tipc_udp_rcast_add(b, &remote);
760         if (err)
761                 goto err;
762
763         return 0;
764 err:
765         if (ub->ubsock)
766                 udp_tunnel_sock_release(ub->ubsock);
767         kfree(ub);
768         return err;
769 }
770
771 /* cleanup_bearer - break the socket/bearer association */
772 static void cleanup_bearer(struct work_struct *work)
773 {
774         struct udp_bearer *ub = container_of(work, struct udp_bearer, work);
775         struct udp_replicast *rcast, *tmp;
776
777         list_for_each_entry_safe(rcast, tmp, &ub->rcast.list, list) {
778                 list_del_rcu(&rcast->list);
779                 kfree_rcu(rcast, rcu);
780         }
781
782         if (ub->ubsock)
783                 udp_tunnel_sock_release(ub->ubsock);
784         synchronize_net();
785         kfree(ub);
786 }
787
788 /* tipc_udp_disable - detach bearer from socket */
789 static void tipc_udp_disable(struct tipc_bearer *b)
790 {
791         struct udp_bearer *ub;
792
793         ub = rcu_dereference_rtnl(b->media_ptr);
794         if (!ub) {
795                 pr_err("UDP bearer instance not found\n");
796                 return;
797         }
798         if (ub->ubsock)
799                 sock_set_flag(ub->ubsock->sk, SOCK_DEAD);
800         RCU_INIT_POINTER(ub->bearer, NULL);
801
802         /* sock_release need to be done outside of rtnl lock */
803         INIT_WORK(&ub->work, cleanup_bearer);
804         schedule_work(&ub->work);
805 }
806
807 struct tipc_media udp_media_info = {
808         .send_msg       = tipc_udp_send_msg,
809         .enable_media   = tipc_udp_enable,
810         .disable_media  = tipc_udp_disable,
811         .addr2str       = tipc_udp_addr2str,
812         .addr2msg       = tipc_udp_addr2msg,
813         .msg2addr       = tipc_udp_msg2addr,
814         .priority       = TIPC_DEF_LINK_PRI,
815         .tolerance      = TIPC_DEF_LINK_TOL,
816         .window         = TIPC_DEF_LINK_WIN,
817         .mtu            = TIPC_DEF_LINK_UDP_MTU,
818         .type_id        = TIPC_MEDIA_TYPE_UDP,
819         .hwaddr_len     = 0,
820         .name           = "udp"
821 };