udp_tunnel.c 5.72 KB
Newer Older
1 2 3 4 5 6
#include <linux/module.h>
#include <linux/errno.h>
#include <linux/socket.h>
#include <linux/udp.h>
#include <linux/types.h>
#include <linux/kernel.h>
7 8
#include <net/dst_metadata.h>
#include <net/net_namespace.h>
9 10 11
#include <net/udp.h>
#include <net/udp_tunnel.h>

12 13
int udp_sock_create4(struct net *net, struct udp_port_cfg *cfg,
		     struct socket **sockp)
14
{
15
	int err;
16
	struct socket *sock = NULL;
17
	struct sockaddr_in udp_addr;
18

19
	err = sock_create_kern(net, AF_INET, SOCK_DGRAM, 0, &sock);
20 21
	if (err < 0)
		goto error;
22

23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39
	if (cfg->bind_ifindex) {
		struct net_device *dev;

		dev = dev_get_by_index(net, cfg->bind_ifindex);
		if (!dev) {
			err = -ENODEV;
			goto error;
		}

		err = kernel_setsockopt(sock, SOL_SOCKET, SO_BINDTODEVICE,
					dev->name, strlen(dev->name) + 1);
		dev_put(dev);

		if (err < 0)
			goto error;
	}

40 41 42 43 44 45 46
	udp_addr.sin_family = AF_INET;
	udp_addr.sin_addr = cfg->local_ip;
	udp_addr.sin_port = cfg->local_udp_port;
	err = kernel_bind(sock, (struct sockaddr *)&udp_addr,
			  sizeof(udp_addr));
	if (err < 0)
		goto error;
47

48
	if (cfg->peer_udp_port) {
49
		udp_addr.sin_family = AF_INET;
50 51 52 53
		udp_addr.sin_addr = cfg->peer_ip;
		udp_addr.sin_port = cfg->peer_udp_port;
		err = kernel_connect(sock, (struct sockaddr *)&udp_addr,
				     sizeof(udp_addr), 0);
54 55 56 57
		if (err < 0)
			goto error;
	}

58
	sock->sk->sk_no_check_tx = !cfg->use_udp_checksums;
59 60 61 62 63 64 65

	*sockp = sock;
	return 0;

error:
	if (sock) {
		kernel_sock_shutdown(sock, SHUT_RDWR);
66
		sock_release(sock);
67 68 69 70
	}
	*sockp = NULL;
	return err;
}
71
EXPORT_SYMBOL(udp_sock_create4);
72

73 74 75 76 77 78 79 80 81
void setup_udp_tunnel_sock(struct net *net, struct socket *sock,
			   struct udp_tunnel_sock_cfg *cfg)
{
	struct sock *sk = sock->sk;

	/* Disable multicast loopback */
	inet_sk(sk)->mc_loop = 0;

	/* Enable CHECKSUM_UNNECESSARY to CHECKSUM_COMPLETE conversion */
82
	inet_inc_convert_csum(sk);
83 84 85 86 87

	rcu_assign_sk_user_data(sk, cfg->sk_user_data);

	udp_sk(sk)->encap_type = cfg->encap_type;
	udp_sk(sk)->encap_rcv = cfg->encap_rcv;
88
	udp_sk(sk)->encap_err_lookup = cfg->encap_err_lookup;
89
	udp_sk(sk)->encap_destroy = cfg->encap_destroy;
90 91
	udp_sk(sk)->gro_receive = cfg->gro_receive;
	udp_sk(sk)->gro_complete = cfg->gro_complete;
92 93 94 95 96

	udp_tunnel_encap_enable(sock);
}
EXPORT_SYMBOL_GPL(setup_udp_tunnel_sock);

97 98 99 100 101 102
void udp_tunnel_push_rx_port(struct net_device *dev, struct socket *sock,
			     unsigned short type)
{
	struct sock *sk = sock->sk;
	struct udp_tunnel_info ti;

103 104
	if (!dev->netdev_ops->ndo_udp_tunnel_add ||
	    !(dev->features & NETIF_F_RX_UDP_TUNNEL_PORT))
105 106
		return;

107 108 109 110
	ti.type = type;
	ti.sa_family = sk->sk_family;
	ti.port = inet_sk(sk)->inet_sport;

111
	dev->netdev_ops->ndo_udp_tunnel_add(dev, &ti);
112 113 114
}
EXPORT_SYMBOL_GPL(udp_tunnel_push_rx_port);

115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132
void udp_tunnel_drop_rx_port(struct net_device *dev, struct socket *sock,
			     unsigned short type)
{
	struct sock *sk = sock->sk;
	struct udp_tunnel_info ti;

	if (!dev->netdev_ops->ndo_udp_tunnel_del ||
	    !(dev->features & NETIF_F_RX_UDP_TUNNEL_PORT))
		return;

	ti.type = type;
	ti.sa_family = sk->sk_family;
	ti.port = inet_sk(sk)->inet_sport;

	dev->netdev_ops->ndo_udp_tunnel_del(dev, &ti);
}
EXPORT_SYMBOL_GPL(udp_tunnel_drop_rx_port);

133 134 135 136 137 138 139 140 141 142 143 144 145
/* Notify netdevs that UDP port started listening */
void udp_tunnel_notify_add_rx_port(struct socket *sock, unsigned short type)
{
	struct sock *sk = sock->sk;
	struct net *net = sock_net(sk);
	struct udp_tunnel_info ti;
	struct net_device *dev;

	ti.type = type;
	ti.sa_family = sk->sk_family;
	ti.port = inet_sk(sk)->inet_sport;

	rcu_read_lock();
146 147 148
	for_each_netdev_rcu(net, dev) {
		if (!dev->netdev_ops->ndo_udp_tunnel_add)
			continue;
149 150
		if (!(dev->features & NETIF_F_RX_UDP_TUNNEL_PORT))
			continue;
151 152
		dev->netdev_ops->ndo_udp_tunnel_add(dev, &ti);
	}
153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169
	rcu_read_unlock();
}
EXPORT_SYMBOL_GPL(udp_tunnel_notify_add_rx_port);

/* Notify netdevs that UDP port is no more listening */
void udp_tunnel_notify_del_rx_port(struct socket *sock, unsigned short type)
{
	struct sock *sk = sock->sk;
	struct net *net = sock_net(sk);
	struct udp_tunnel_info ti;
	struct net_device *dev;

	ti.type = type;
	ti.sa_family = sk->sk_family;
	ti.port = inet_sk(sk)->inet_sport;

	rcu_read_lock();
170 171 172
	for_each_netdev_rcu(net, dev) {
		if (!dev->netdev_ops->ndo_udp_tunnel_del)
			continue;
173 174
		if (!(dev->features & NETIF_F_RX_UDP_TUNNEL_PORT))
			continue;
175 176
		dev->netdev_ops->ndo_udp_tunnel_del(dev, &ti);
	}
177 178 179 180
	rcu_read_unlock();
}
EXPORT_SYMBOL_GPL(udp_tunnel_notify_del_rx_port);

181 182 183 184
void 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)
185 186 187 188 189 190 191 192 193 194 195
{
	struct udphdr *uh;

	__skb_push(skb, sizeof(*uh));
	skb_reset_transport_header(skb);
	uh = udp_hdr(skb);

	uh->dest = dst_port;
	uh->source = src_port;
	uh->len = htons(skb->len);

196 197
	memset(&(IPCB(skb)->opt), 0, sizeof(IPCB(skb)->opt));

198
	udp_set_csum(nocheck, skb, src, dst, skb->len);
199

200
	iptunnel_xmit(sk, rt, skb, src, dst, IPPROTO_UDP, tos, ttl, df, xnet);
201 202 203 204 205 206 207
}
EXPORT_SYMBOL_GPL(udp_tunnel_xmit_skb);

void udp_tunnel_sock_release(struct socket *sock)
{
	rcu_assign_sk_user_data(sock->sk, NULL);
	kernel_sock_shutdown(sock, SHUT_RDWR);
208
	sock_release(sock);
209 210 211
}
EXPORT_SYMBOL_GPL(udp_tunnel_sock_release);

212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233
struct metadata_dst *udp_tun_rx_dst(struct sk_buff *skb,  unsigned short family,
				    __be16 flags, __be64 tunnel_id, int md_size)
{
	struct metadata_dst *tun_dst;
	struct ip_tunnel_info *info;

	if (family == AF_INET)
		tun_dst = ip_tun_rx_dst(skb, flags, tunnel_id, md_size);
	else
		tun_dst = ipv6_tun_rx_dst(skb, flags, tunnel_id, md_size);
	if (!tun_dst)
		return NULL;

	info = &tun_dst->u.tun_info;
	info->key.tp_src = udp_hdr(skb)->source;
	info->key.tp_dst = udp_hdr(skb)->dest;
	if (udp_hdr(skb)->check)
		info->key.tun_flags |= TUNNEL_CSUM;
	return tun_dst;
}
EXPORT_SYMBOL_GPL(udp_tun_rx_dst);

234
MODULE_LICENSE("GPL");