tjh.dev
Linux kernel mirror (for testing)
git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
kernel
os
linux
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
13int 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
53error:
54 if (sock) {
55 kernel_sock_shutdown(sock, SHUT_RDWR);
56 sock_release(sock);
57 }
58 *sockp = NULL;
59 return err;
60}
61EXPORT_SYMBOL(udp_sock_create4);
62
63void 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}
85EXPORT_SYMBOL_GPL(setup_udp_tunnel_sock);
86
87void 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}
99EXPORT_SYMBOL_GPL(udp_tunnel_push_rx_port);
100
101void 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}
113EXPORT_SYMBOL_GPL(udp_tunnel_drop_rx_port);
114
115/* Notify netdevs that UDP port started listening */
116void 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}
133EXPORT_SYMBOL_GPL(udp_tunnel_notify_add_rx_port);
134
135/* Notify netdevs that UDP port is no more listening */
136void 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}
153EXPORT_SYMBOL_GPL(udp_tunnel_notify_del_rx_port);
154
155void 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}
176EXPORT_SYMBOL_GPL(udp_tunnel_xmit_skb);
177
178void 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}
184EXPORT_SYMBOL_GPL(udp_tunnel_sock_release);
185
186struct 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}
206EXPORT_SYMBOL_GPL(udp_tun_rx_dst);
207
208MODULE_LICENSE("GPL");