1 // SPDX-License-Identifier: GPL-2.0-only
2 #include <linux/module.h>
3 #include <linux/errno.h>
4 #include <linux/socket.h>
5 #include <linux/udp.h>
6 #include <linux/types.h>
7 #include <linux/kernel.h>
8 #include <net/dst_metadata.h>
9 #include <net/net_namespace.h>
10 #include <net/udp.h>
11 #include <net/udp_tunnel.h>
12
udp_sock_create4(struct net * net,struct udp_port_cfg * cfg,struct socket ** sockp)13 int udp_sock_create4(struct net *net, struct udp_port_cfg *cfg,
14 struct socket **sockp)
15 {
16 int err;
17 struct socket *sock = NULL;
18 struct sockaddr_in udp_addr;
19
20 err = sock_create_kern(net, AF_INET, SOCK_DGRAM, 0, &sock);
21 if (err < 0)
22 goto error;
23
24 if (cfg->bind_ifindex) {
25 err = sock_bindtoindex(sock->sk, cfg->bind_ifindex, true);
26 if (err < 0)
27 goto error;
28 }
29
30 udp_addr.sin_family = AF_INET;
31 udp_addr.sin_addr = cfg->local_ip;
32 udp_addr.sin_port = cfg->local_udp_port;
33 err = kernel_bind(sock, (struct sockaddr *)&udp_addr,
34 sizeof(udp_addr));
35 if (err < 0)
36 goto error;
37
38 if (cfg->peer_udp_port) {
39 udp_addr.sin_family = AF_INET;
40 udp_addr.sin_addr = cfg->peer_ip;
41 udp_addr.sin_port = cfg->peer_udp_port;
42 err = kernel_connect(sock, (struct sockaddr *)&udp_addr,
43 sizeof(udp_addr), 0);
44 if (err < 0)
45 goto error;
46 }
47
48 sock->sk->sk_no_check_tx = !cfg->use_udp_checksums;
49
50 *sockp = sock;
51 return 0;
52
53 error:
54 if (sock) {
55 kernel_sock_shutdown(sock, SHUT_RDWR);
56 sock_release(sock);
57 }
58 *sockp = NULL;
59 return err;
60 }
61 EXPORT_SYMBOL(udp_sock_create4);
62
setup_udp_tunnel_sock(struct net * net,struct socket * sock,struct udp_tunnel_sock_cfg * cfg)63 void setup_udp_tunnel_sock(struct net *net, struct socket *sock,
64 struct udp_tunnel_sock_cfg *cfg)
65 {
66 struct sock *sk = sock->sk;
67
68 /* Disable multicast loopback */
69 inet_sk(sk)->mc_loop = 0;
70
71 /* Enable CHECKSUM_UNNECESSARY to CHECKSUM_COMPLETE conversion */
72 inet_inc_convert_csum(sk);
73
74 rcu_assign_sk_user_data(sk, cfg->sk_user_data);
75
76 udp_sk(sk)->encap_type = cfg->encap_type;
77 udp_sk(sk)->encap_rcv = cfg->encap_rcv;
78 udp_sk(sk)->encap_err_lookup = cfg->encap_err_lookup;
79 udp_sk(sk)->encap_destroy = cfg->encap_destroy;
80 udp_sk(sk)->gro_receive = cfg->gro_receive;
81 udp_sk(sk)->gro_complete = cfg->gro_complete;
82
83 udp_tunnel_encap_enable(sock);
84 }
85 EXPORT_SYMBOL_GPL(setup_udp_tunnel_sock);
86
udp_tunnel_push_rx_port(struct net_device * dev,struct socket * sock,unsigned short type)87 void udp_tunnel_push_rx_port(struct net_device *dev, struct socket *sock,
88 unsigned short type)
89 {
90 struct sock *sk = sock->sk;
91 struct udp_tunnel_info ti;
92
93 ti.type = type;
94 ti.sa_family = sk->sk_family;
95 ti.port = inet_sk(sk)->inet_sport;
96
97 udp_tunnel_nic_add_port(dev, &ti);
98 }
99 EXPORT_SYMBOL_GPL(udp_tunnel_push_rx_port);
100
udp_tunnel_drop_rx_port(struct net_device * dev,struct socket * sock,unsigned short type)101 void udp_tunnel_drop_rx_port(struct net_device *dev, struct socket *sock,
102 unsigned short type)
103 {
104 struct sock *sk = sock->sk;
105 struct udp_tunnel_info ti;
106
107 ti.type = type;
108 ti.sa_family = sk->sk_family;
109 ti.port = inet_sk(sk)->inet_sport;
110
111 udp_tunnel_nic_del_port(dev, &ti);
112 }
113 EXPORT_SYMBOL_GPL(udp_tunnel_drop_rx_port);
114
115 /* Notify netdevs that UDP port started listening */
udp_tunnel_notify_add_rx_port(struct socket * sock,unsigned short type)116 void udp_tunnel_notify_add_rx_port(struct socket *sock, unsigned short type)
117 {
118 struct sock *sk = sock->sk;
119 struct net *net = sock_net(sk);
120 struct udp_tunnel_info ti;
121 struct net_device *dev;
122
123 ti.type = type;
124 ti.sa_family = sk->sk_family;
125 ti.port = inet_sk(sk)->inet_sport;
126
127 rcu_read_lock();
128 for_each_netdev_rcu(net, dev) {
129 udp_tunnel_nic_add_port(dev, &ti);
130 }
131 rcu_read_unlock();
132 }
133 EXPORT_SYMBOL_GPL(udp_tunnel_notify_add_rx_port);
134
135 /* Notify netdevs that UDP port is no more listening */
udp_tunnel_notify_del_rx_port(struct socket * sock,unsigned short type)136 void udp_tunnel_notify_del_rx_port(struct socket *sock, unsigned short type)
137 {
138 struct sock *sk = sock->sk;
139 struct net *net = sock_net(sk);
140 struct udp_tunnel_info ti;
141 struct net_device *dev;
142
143 ti.type = type;
144 ti.sa_family = sk->sk_family;
145 ti.port = inet_sk(sk)->inet_sport;
146
147 rcu_read_lock();
148 for_each_netdev_rcu(net, dev) {
149 udp_tunnel_nic_del_port(dev, &ti);
150 }
151 rcu_read_unlock();
152 }
153 EXPORT_SYMBOL_GPL(udp_tunnel_notify_del_rx_port);
154
udp_tunnel_xmit_skb(struct rtable * rt,struct sock * sk,struct sk_buff * skb,__be32 src,__be32 dst,__u8 tos,__u8 ttl,__be16 df,__be16 src_port,__be16 dst_port,bool xnet,bool nocheck)155 void udp_tunnel_xmit_skb(struct rtable *rt, struct sock *sk, struct sk_buff *skb,
156 __be32 src, __be32 dst, __u8 tos, __u8 ttl,
157 __be16 df, __be16 src_port, __be16 dst_port,
158 bool xnet, bool nocheck)
159 {
160 struct udphdr *uh;
161
162 __skb_push(skb, sizeof(*uh));
163 skb_reset_transport_header(skb);
164 uh = udp_hdr(skb);
165
166 uh->dest = dst_port;
167 uh->source = src_port;
168 uh->len = htons(skb->len);
169
170 memset(&(IPCB(skb)->opt), 0, sizeof(IPCB(skb)->opt));
171
172 udp_set_csum(nocheck, skb, src, dst, skb->len);
173
174 iptunnel_xmit(sk, rt, skb, src, dst, IPPROTO_UDP, tos, ttl, df, xnet);
175 }
176 EXPORT_SYMBOL_GPL(udp_tunnel_xmit_skb);
177
udp_tunnel_sock_release(struct socket * sock)178 void udp_tunnel_sock_release(struct socket *sock)
179 {
180 rcu_assign_sk_user_data(sock->sk, NULL);
181 kernel_sock_shutdown(sock, SHUT_RDWR);
182 sock_release(sock);
183 }
184 EXPORT_SYMBOL_GPL(udp_tunnel_sock_release);
185
udp_tun_rx_dst(struct sk_buff * skb,unsigned short family,__be16 flags,__be64 tunnel_id,int md_size)186 struct metadata_dst *udp_tun_rx_dst(struct sk_buff *skb, unsigned short family,
187 __be16 flags, __be64 tunnel_id, int md_size)
188 {
189 struct metadata_dst *tun_dst;
190 struct ip_tunnel_info *info;
191
192 if (family == AF_INET)
193 tun_dst = ip_tun_rx_dst(skb, flags, tunnel_id, md_size);
194 else
195 tun_dst = ipv6_tun_rx_dst(skb, flags, tunnel_id, md_size);
196 if (!tun_dst)
197 return NULL;
198
199 info = &tun_dst->u.tun_info;
200 info->key.tp_src = udp_hdr(skb)->source;
201 info->key.tp_dst = udp_hdr(skb)->dest;
202 if (udp_hdr(skb)->check)
203 info->key.tun_flags |= TUNNEL_CSUM;
204 return tun_dst;
205 }
206 EXPORT_SYMBOL_GPL(udp_tun_rx_dst);
207
208 MODULE_LICENSE("GPL");
209