udp.c 24.4 KB
Newer Older
L
Linus Torvalds 已提交
1 2
/*
 *	UDP over IPv6
3
 *	Linux INET6 implementation
L
Linus Torvalds 已提交
4 5
 *
 *	Authors:
6
 *	Pedro Roque		<roque@di.fc.ul.pt>
L
Linus Torvalds 已提交
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
 *
 *	Based on linux/ipv4/udp.c
 *
 *	$Id: udp.c,v 1.65 2002/02/01 22:01:04 davem Exp $
 *
 *	Fixes:
 *	Hideaki YOSHIFUJI	:	sin6_scope_id support
 *	YOSHIFUJI Hideaki @USAGI and:	Support IPV6_V6ONLY socket option, which
 *	Alexey Kuznetsov		allow both IPv4 and IPv6 sockets to bind
 *					a single port at the same time.
 *      Kazunori MIYAZAWA @USAGI:       change process style to use ip6_append_data
 *      YOSHIFUJI Hideaki @USAGI:	convert /proc/net/udp6 to seq_file.
 *
 *	This program is free software; you can redistribute it and/or
 *      modify it under the terms of the GNU General Public License
 *      as published by the Free Software Foundation; either version
 *      2 of the License, or (at your option) any later version.
 */

#include <linux/errno.h>
#include <linux/types.h>
#include <linux/socket.h>
#include <linux/sockios.h>
#include <linux/net.h>
#include <linux/in6.h>
#include <linux/netdevice.h>
#include <linux/if_arp.h>
#include <linux/ipv6.h>
#include <linux/icmpv6.h>
#include <linux/init.h>
37
#include <linux/skbuff.h>
L
Linus Torvalds 已提交
38 39 40 41 42 43 44
#include <asm/uaccess.h>

#include <net/ndisc.h>
#include <net/protocol.h>
#include <net/transp_v6.h>
#include <net/ip6_route.h>
#include <net/raw.h>
45
#include <net/tcp_states.h>
L
Linus Torvalds 已提交
46 47 48 49 50
#include <net/ip6_checksum.h>
#include <net/xfrm.h>

#include <linux/proc_fs.h>
#include <linux/seq_file.h>
51
#include "udp_impl.h"
L
Linus Torvalds 已提交
52

53
DEFINE_SNMP_STAT(struct udp_mib, udp_stats_in6) __read_mostly;
L
Linus Torvalds 已提交
54

55
static inline int udp_v6_get_port(struct sock *sk, unsigned short snum)
L
Linus Torvalds 已提交
56
{
57
	return udp_get_port(sk, snum, ipv6_rcv_saddr_equal);
L
Linus Torvalds 已提交
58 59
}

60 61 62
static struct sock *__udp6_lib_lookup(struct in6_addr *saddr, __be16 sport,
				      struct in6_addr *daddr, __be16 dport,
				      int dif, struct hlist_head udptable[])
L
Linus Torvalds 已提交
63 64 65 66 67 68
{
	struct sock *sk, *result = NULL;
	struct hlist_node *node;
	unsigned short hnum = ntohs(dport);
	int badness = -1;

69
	read_lock(&udp_hash_lock);
70
	sk_for_each(sk, node, &udptable[hnum & (UDP_HTABLE_SIZE - 1)]) {
L
Linus Torvalds 已提交
71 72
		struct inet_sock *inet = inet_sk(sk);

73
		if (sk->sk_hash == hnum && sk->sk_family == PF_INET6) {
L
Linus Torvalds 已提交
74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95
			struct ipv6_pinfo *np = inet6_sk(sk);
			int score = 0;
			if (inet->dport) {
				if (inet->dport != sport)
					continue;
				score++;
			}
			if (!ipv6_addr_any(&np->rcv_saddr)) {
				if (!ipv6_addr_equal(&np->rcv_saddr, daddr))
					continue;
				score++;
			}
			if (!ipv6_addr_any(&np->daddr)) {
				if (!ipv6_addr_equal(&np->daddr, saddr))
					continue;
				score++;
			}
			if (sk->sk_bound_dev_if) {
				if (sk->sk_bound_dev_if != dif)
					continue;
				score++;
			}
S
Stephen Hemminger 已提交
96
			if (score == 4) {
L
Linus Torvalds 已提交
97 98
				result = sk;
				break;
S
Stephen Hemminger 已提交
99
			} else if (score > badness) {
L
Linus Torvalds 已提交
100 101 102 103 104 105 106
				result = sk;
				badness = score;
			}
		}
	}
	if (result)
		sock_hold(result);
107
	read_unlock(&udp_hash_lock);
L
Linus Torvalds 已提交
108 109 110 111 112 113 114 115
	return result;
}

/*
 * 	This should be easy, if there is something there we
 * 	return it, otherwise we block.
 */

116
int udpv6_recvmsg(struct kiocb *iocb, struct sock *sk,
L
Linus Torvalds 已提交
117 118 119 120 121
		  struct msghdr *msg, size_t len,
		  int noblock, int flags, int *addr_len)
{
	struct ipv6_pinfo *np = inet6_sk(sk);
	struct inet_sock *inet = inet_sk(sk);
122
	struct sk_buff *skb;
123 124 125
	unsigned int ulen, copied;
	int err;
	int is_udplite = IS_UDPLITE(sk);
L
Linus Torvalds 已提交
126

127 128 129
	if (addr_len)
		*addr_len=sizeof(struct sockaddr_in6);

L
Linus Torvalds 已提交
130 131 132 133 134 135 136 137
	if (flags & MSG_ERRQUEUE)
		return ipv6_recv_error(sk, msg, len);

try_again:
	skb = skb_recv_datagram(sk, flags, noblock, &err);
	if (!skb)
		goto out;

138 139 140 141 142
	ulen = skb->len - sizeof(struct udphdr);
	copied = len;
	if (copied > ulen)
		copied = ulen;
	else if (copied < ulen)
143
		msg->msg_flags |= MSG_TRUNC;
L
Linus Torvalds 已提交
144

145
	/*
146 147 148
	 * If checksum is needed at all, try to do it while copying the
	 * data.  If the data is truncated, or if we only want a partial
	 * coverage checksum (UDP-Lite), do it before the copy.
149 150
	 */

151 152
	if (copied < ulen || UDP_SKB_CB(skb)->partial_cov) {
		if (udp_lib_checksum_complete(skb))
L
Linus Torvalds 已提交
153
			goto csum_copy_err;
154 155
	}

156
	if (skb_csum_unnecessary(skb))
157 158 159
		err = skb_copy_datagram_iovec(skb, sizeof(struct udphdr),
					      msg->msg_iov, copied       );
	else {
L
Linus Torvalds 已提交
160 161 162 163 164 165 166 167 168 169 170 171
		err = skb_copy_and_csum_datagram_iovec(skb, sizeof(struct udphdr), msg->msg_iov);
		if (err == -EINVAL)
			goto csum_copy_err;
	}
	if (err)
		goto out_free;

	sock_recv_timestamp(msg, sk, skb);

	/* Copy the address. */
	if (msg->msg_name) {
		struct sockaddr_in6 *sin6;
172

L
Linus Torvalds 已提交
173 174
		sin6 = (struct sockaddr_in6 *) msg->msg_name;
		sin6->sin6_family = AF_INET6;
175
		sin6->sin6_port = udp_hdr(skb)->source;
L
Linus Torvalds 已提交
176 177 178 179 180
		sin6->sin6_flowinfo = 0;
		sin6->sin6_scope_id = 0;

		if (skb->protocol == htons(ETH_P_IP))
			ipv6_addr_set(&sin6->sin6_addr, 0, 0,
181
				      htonl(0xffff), ip_hdr(skb)->saddr);
L
Linus Torvalds 已提交
182
		else {
183 184
			ipv6_addr_copy(&sin6->sin6_addr,
				       &ipv6_hdr(skb)->saddr);
L
Linus Torvalds 已提交
185 186 187 188 189 190 191 192 193 194 195
			if (ipv6_addr_type(&sin6->sin6_addr) & IPV6_ADDR_LINKLOCAL)
				sin6->sin6_scope_id = IP6CB(skb)->iif;
		}

	}
	if (skb->protocol == htons(ETH_P_IP)) {
		if (inet->cmsg_flags)
			ip_cmsg_recv(msg, skb);
	} else {
		if (np->rxopt.all)
			datagram_recv_ctl(sk, msg, skb);
196
	}
L
Linus Torvalds 已提交
197 198 199

	err = copied;
	if (flags & MSG_TRUNC)
200
		err = ulen;
L
Linus Torvalds 已提交
201 202 203 204 205 206 207

out_free:
	skb_free_datagram(sk, skb);
out:
	return err;

csum_copy_err:
208
	skb_kill_datagram(sk, skb, flags);
L
Linus Torvalds 已提交
209 210

	if (flags & MSG_DONTWAIT) {
211
		UDP6_INC_STATS_USER(UDP_MIB_INERRORS, is_udplite);
L
Linus Torvalds 已提交
212 213 214 215 216
		return -EAGAIN;
	}
	goto try_again;
}

217 218 219
void __udp6_lib_err(struct sk_buff *skb, struct inet6_skb_parm *opt,
		    int type, int code, int offset, __be32 info,
		    struct hlist_head udptable[]                    )
L
Linus Torvalds 已提交
220 221 222 223 224 225 226 227 228
{
	struct ipv6_pinfo *np;
	struct ipv6hdr *hdr = (struct ipv6hdr*)skb->data;
	struct in6_addr *saddr = &hdr->saddr;
	struct in6_addr *daddr = &hdr->daddr;
	struct udphdr *uh = (struct udphdr*)(skb->data+offset);
	struct sock *sk;
	int err;

229 230
	sk = __udp6_lib_lookup(daddr, uh->dest,
			       saddr, uh->source, inet6_iif(skb), udptable);
L
Linus Torvalds 已提交
231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250
	if (sk == NULL)
		return;

	np = inet6_sk(sk);

	if (!icmpv6_err_convert(type, code, &err) && !np->recverr)
		goto out;

	if (sk->sk_state != TCP_ESTABLISHED && !np->recverr)
		goto out;

	if (np->recverr)
		ipv6_icmp_error(sk, skb, err, uh->dest, ntohl(info), (u8 *)(uh+1));

	sk->sk_err = err;
	sk->sk_error_report(sk);
out:
	sock_put(sk);
}

251 252
static __inline__ void udpv6_err(struct sk_buff *skb,
				 struct inet6_skb_parm *opt, int type,
253
				 int code, int offset, __be32 info     )
254 255 256 257 258
{
	return __udp6_lib_err(skb, opt, type, code, offset, info, udp_hash);
}

int udpv6_queue_rcv_skb(struct sock * sk, struct sk_buff *skb)
L
Linus Torvalds 已提交
259
{
260
	struct udp_sock *up = udp_sk(sk);
261 262
	int rc;

263 264
	if (!xfrm6_policy_check(sk, XFRM_POLICY_IN, skb))
		goto drop;
L
Linus Torvalds 已提交
265

266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282
	/*
	 * UDP-Lite specific tests, ignored on UDP sockets (see net/ipv4/udp.c).
	 */
	if ((up->pcflag & UDPLITE_RECV_CC)  &&  UDP_SKB_CB(skb)->partial_cov) {

		if (up->pcrlen == 0) {          /* full coverage was set  */
			LIMIT_NETDEBUG(KERN_WARNING "UDPLITE6: partial coverage"
				" %d while full coverage %d requested\n",
				UDP_SKB_CB(skb)->cscov, skb->len);
			goto drop;
		}
		if (UDP_SKB_CB(skb)->cscov  <  up->pcrlen) {
			LIMIT_NETDEBUG(KERN_WARNING "UDPLITE6: coverage %d "
						    "too small, need min %d\n",
				       UDP_SKB_CB(skb)->cscov, up->pcrlen);
			goto drop;
		}
L
Linus Torvalds 已提交
283 284
	}

285 286 287 288
	if (sk->sk_filter) {
		if (udp_lib_checksum_complete(skb))
			goto drop;
	}
289

290 291 292
	if ((rc = sock_queue_rcv_skb(sk,skb)) < 0) {
		/* Note that an ENOMEM error is charged twice */
		if (rc == -ENOMEM)
293 294
			UDP6_INC_STATS_BH(UDP_MIB_RCVBUFERRORS, up->pcflag);
		goto drop;
L
Linus Torvalds 已提交
295
	}
296
	UDP6_INC_STATS_BH(UDP_MIB_INDATAGRAMS, up->pcflag);
L
Linus Torvalds 已提交
297
	return 0;
298 299 300 301
drop:
	UDP6_INC_STATS_BH(UDP_MIB_INERRORS, up->pcflag);
	kfree_skb(skb);
	return -1;
L
Linus Torvalds 已提交
302 303 304
}

static struct sock *udp_v6_mcast_next(struct sock *sk,
A
Al Viro 已提交
305 306
				      __be16 loc_port, struct in6_addr *loc_addr,
				      __be16 rmt_port, struct in6_addr *rmt_addr,
L
Linus Torvalds 已提交
307 308 309 310 311 312 313 314 315
				      int dif)
{
	struct hlist_node *node;
	struct sock *s = sk;
	unsigned short num = ntohs(loc_port);

	sk_for_each_from(s, node) {
		struct inet_sock *inet = inet_sk(s);

316
		if (s->sk_hash == num && s->sk_family == PF_INET6) {
L
Linus Torvalds 已提交
317 318 319 320 321 322 323 324 325 326 327 328 329
			struct ipv6_pinfo *np = inet6_sk(s);
			if (inet->dport) {
				if (inet->dport != rmt_port)
					continue;
			}
			if (!ipv6_addr_any(&np->daddr) &&
			    !ipv6_addr_equal(&np->daddr, rmt_addr))
				continue;

			if (s->sk_bound_dev_if && s->sk_bound_dev_if != dif)
				continue;

			if (!ipv6_addr_any(&np->rcv_saddr)) {
330 331
				if (!ipv6_addr_equal(&np->rcv_saddr, loc_addr))
					continue;
L
Linus Torvalds 已提交
332
			}
S
Stephen Hemminger 已提交
333
			if (!inet6_mc_check(s, loc_addr, rmt_addr))
L
Linus Torvalds 已提交
334 335 336 337 338 339 340 341 342 343 344
				continue;
			return s;
		}
	}
	return NULL;
}

/*
 * Note: called only from the BH handler context,
 * so we don't need to lock the hashes.
 */
345
static int __udp6_lib_mcast_deliver(struct sk_buff *skb, struct in6_addr *saddr,
346
			   struct in6_addr *daddr, struct hlist_head udptable[])
L
Linus Torvalds 已提交
347 348
{
	struct sock *sk, *sk2;
349
	const struct udphdr *uh = udp_hdr(skb);
L
Linus Torvalds 已提交
350 351 352
	int dif;

	read_lock(&udp_hash_lock);
353
	sk = sk_head(&udptable[ntohs(uh->dest) & (UDP_HTABLE_SIZE - 1)]);
354
	dif = inet6_iif(skb);
L
Linus Torvalds 已提交
355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370
	sk = udp_v6_mcast_next(sk, uh->dest, daddr, uh->source, saddr, dif);
	if (!sk) {
		kfree_skb(skb);
		goto out;
	}

	sk2 = sk;
	while ((sk2 = udp_v6_mcast_next(sk_next(sk2), uh->dest, daddr,
					uh->source, saddr, dif))) {
		struct sk_buff *buff = skb_clone(skb, GFP_ATOMIC);
		if (buff)
			udpv6_queue_rcv_skb(sk2, buff);
	}
	udpv6_queue_rcv_skb(sk, skb);
out:
	read_unlock(&udp_hash_lock);
371
	return 0;
L
Linus Torvalds 已提交
372 373
}

374 375
static inline int udp6_csum_init(struct sk_buff *skb, struct udphdr *uh,
				 int proto)
376
{
377 378 379 380 381 382 383 384 385 386 387
	int err;

	UDP_SKB_CB(skb)->partial_cov = 0;
	UDP_SKB_CB(skb)->cscov = skb->len;

	if (proto == IPPROTO_UDPLITE) {
		err = udplite_checksum_init(skb, uh);
		if (err)
			return err;
	}

388 389 390 391 392 393 394 395
	if (uh->check == 0) {
		/* RFC 2460 section 8.1 says that we SHOULD log
		   this error. Well, it is reasonable.
		 */
		LIMIT_NETDEBUG(KERN_INFO "IPv6: udp checksum is 0\n");
		return 1;
	}
	if (skb->ip_summed == CHECKSUM_COMPLETE &&
396
	    !csum_ipv6_magic(&ipv6_hdr(skb)->saddr, &ipv6_hdr(skb)->daddr,
397
			     skb->len, proto, skb->csum))
398 399
		skb->ip_summed = CHECKSUM_UNNECESSARY;

400
	if (!skb_csum_unnecessary(skb))
401 402
		skb->csum = ~csum_unfold(csum_ipv6_magic(&ipv6_hdr(skb)->saddr,
							 &ipv6_hdr(skb)->daddr,
403
							 skb->len, proto, 0));
404

405
	return 0;
406 407 408
}

int __udp6_lib_rcv(struct sk_buff **pskb, struct hlist_head udptable[],
409
		   int proto)
L
Linus Torvalds 已提交
410 411 412
{
	struct sk_buff *skb = *pskb;
	struct sock *sk;
413
	struct udphdr *uh;
L
Linus Torvalds 已提交
414 415 416 417 418 419 420
	struct net_device *dev = skb->dev;
	struct in6_addr *saddr, *daddr;
	u32 ulen = 0;

	if (!pskb_may_pull(skb, sizeof(struct udphdr)))
		goto short_packet;

421 422
	saddr = &ipv6_hdr(skb)->saddr;
	daddr = &ipv6_hdr(skb)->daddr;
423
	uh = udp_hdr(skb);
L
Linus Torvalds 已提交
424 425

	ulen = ntohs(uh->len);
426 427
	if (ulen > skb->len)
		goto short_packet;
L
Linus Torvalds 已提交
428

429 430
	if (proto == IPPROTO_UDP) {
		/* UDP validates ulen. */
L
Linus Torvalds 已提交
431

432 433 434
		/* Check for jumbo payload */
		if (ulen == 0)
			ulen = skb->len;
L
Linus Torvalds 已提交
435

436 437
		if (ulen < sizeof(*uh))
			goto short_packet;
L
Linus Torvalds 已提交
438

439 440 441
		if (ulen < skb->len) {
			if (pskb_trim_rcsum(skb, ulen))
				goto short_packet;
442 443
			saddr = &ipv6_hdr(skb)->saddr;
			daddr = &ipv6_hdr(skb)->daddr;
444
			uh = udp_hdr(skb);
445 446
		}
	}
L
Linus Torvalds 已提交
447

448 449 450
	if (udp6_csum_init(skb, uh, proto))
		goto discard;

451 452
	/*
	 *	Multicast receive code
L
Linus Torvalds 已提交
453
	 */
454 455
	if (ipv6_addr_is_multicast(daddr))
		return __udp6_lib_mcast_deliver(skb, saddr, daddr, udptable);
L
Linus Torvalds 已提交
456 457 458

	/* Unicast */

459
	/*
L
Linus Torvalds 已提交
460 461 462
	 * check socket cache ... must talk to Alan about his plans
	 * for sock caches... i'll skip this for now.
	 */
463 464
	sk = __udp6_lib_lookup(saddr, uh->source,
			       daddr, uh->dest, inet6_iif(skb), udptable);
L
Linus Torvalds 已提交
465 466 467 468 469

	if (sk == NULL) {
		if (!xfrm6_policy_check(NULL, XFRM_POLICY_IN, skb))
			goto discard;

470
		if (udp_lib_checksum_complete(skb))
L
Linus Torvalds 已提交
471
			goto discard;
472
		UDP6_INC_STATS_BH(UDP_MIB_NOPORTS, proto == IPPROTO_UDPLITE);
L
Linus Torvalds 已提交
473 474 475 476

		icmpv6_send(skb, ICMPV6_DEST_UNREACH, ICMPV6_PORT_UNREACH, 0, dev);

		kfree_skb(skb);
S
Stephen Hemminger 已提交
477
		return 0;
L
Linus Torvalds 已提交
478
	}
479

L
Linus Torvalds 已提交
480
	/* deliver */
481

L
Linus Torvalds 已提交
482 483
	udpv6_queue_rcv_skb(sk, skb);
	sock_put(sk);
S
Stephen Hemminger 已提交
484
	return 0;
L
Linus Torvalds 已提交
485

486
short_packet:
487
	LIMIT_NETDEBUG(KERN_DEBUG "UDP%sv6: short packet: %d/%u\n",
488 489
		       proto == IPPROTO_UDPLITE ? "-Lite" : "",
		       ulen, skb->len);
L
Linus Torvalds 已提交
490 491

discard:
492
	UDP6_INC_STATS_BH(UDP_MIB_INERRORS, proto == IPPROTO_UDPLITE);
L
Linus Torvalds 已提交
493
	kfree_skb(skb);
S
Stephen Hemminger 已提交
494
	return 0;
L
Linus Torvalds 已提交
495
}
496 497 498

static __inline__ int udpv6_rcv(struct sk_buff **pskb)
{
499
	return __udp6_lib_rcv(pskb, udp_hash, IPPROTO_UDP);
500 501
}

L
Linus Torvalds 已提交
502 503 504 505 506 507 508 509 510 511 512
/*
 * Throw away all pending data and cancel the corking. Socket is locked.
 */
static void udp_v6_flush_pending_frames(struct sock *sk)
{
	struct udp_sock *up = udp_sk(sk);

	if (up->pending) {
		up->len = 0;
		up->pending = 0;
		ip6_flush_pending_frames(sk);
513
	}
L
Linus Torvalds 已提交
514 515 516 517 518 519
}

/*
 *	Sending
 */

520
static int udp_v6_push_pending_frames(struct sock *sk)
L
Linus Torvalds 已提交
521 522 523
{
	struct sk_buff *skb;
	struct udphdr *uh;
524
	struct udp_sock  *up = udp_sk(sk);
L
Linus Torvalds 已提交
525 526 527
	struct inet_sock *inet = inet_sk(sk);
	struct flowi *fl = &inet->cork.fl;
	int err = 0;
528
	__wsum csum = 0;
L
Linus Torvalds 已提交
529 530 531 532 533 534 535 536

	/* Grab the skbuff where UDP header space exists. */
	if ((skb = skb_peek(&sk->sk_write_queue)) == NULL)
		goto out;

	/*
	 * Create a UDP header
	 */
537
	uh = udp_hdr(skb);
L
Linus Torvalds 已提交
538 539 540 541 542
	uh->source = fl->fl_ip_sport;
	uh->dest = fl->fl_ip_dport;
	uh->len = htons(up->len);
	uh->check = 0;

543 544 545 546
	if (up->pcflag)
		csum = udplite_csum_outgoing(sk, skb);
	 else
		csum = udp_csum_outgoing(sk, skb);
L
Linus Torvalds 已提交
547

548 549 550
	/* add protocol-dependent pseudo-header */
	uh->check = csum_ipv6_magic(&fl->fl6_src, &fl->fl6_dst,
				    up->len, fl->proto, csum   );
L
Linus Torvalds 已提交
551
	if (uh->check == 0)
552
		uh->check = CSUM_MANGLED_0;
L
Linus Torvalds 已提交
553 554 555 556 557

	err = ip6_push_pending_frames(sk);
out:
	up->len = 0;
	up->pending = 0;
558 559
	if (!err)
		UDP6_INC_STATS_USER(UDP_MIB_OUTDATAGRAMS, up->pcflag);
L
Linus Torvalds 已提交
560 561 562
	return err;
}

563
int udpv6_sendmsg(struct kiocb *iocb, struct sock *sk,
L
Linus Torvalds 已提交
564 565 566 567 568 569 570 571 572 573
		  struct msghdr *msg, size_t len)
{
	struct ipv6_txoptions opt_space;
	struct udp_sock *up = udp_sk(sk);
	struct inet_sock *inet = inet_sk(sk);
	struct ipv6_pinfo *np = inet6_sk(sk);
	struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *) msg->msg_name;
	struct in6_addr *daddr, *final_p = NULL, final;
	struct ipv6_txoptions *opt = NULL;
	struct ip6_flowlabel *flowlabel = NULL;
H
Herbert Xu 已提交
574
	struct flowi fl;
L
Linus Torvalds 已提交
575 576 577 578
	struct dst_entry *dst;
	int addr_len = msg->msg_namelen;
	int ulen = len;
	int hlimit = -1;
579
	int tclass = -1;
L
Linus Torvalds 已提交
580 581
	int corkreq = up->corkflag || msg->msg_flags&MSG_MORE;
	int err;
582
	int connected = 0;
583 584
	int is_udplite = up->pcflag;
	int (*getfrag)(void *, char *, int, int, int, struct sk_buff *);
L
Linus Torvalds 已提交
585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610

	/* destination address check */
	if (sin6) {
		if (addr_len < offsetof(struct sockaddr, sa_data))
			return -EINVAL;

		switch (sin6->sin6_family) {
		case AF_INET6:
			if (addr_len < SIN6_LEN_RFC2133)
				return -EINVAL;
			daddr = &sin6->sin6_addr;
			break;
		case AF_INET:
			goto do_udp_sendmsg;
		case AF_UNSPEC:
			msg->msg_name = sin6 = NULL;
			msg->msg_namelen = addr_len = 0;
			daddr = NULL;
			break;
		default:
			return -EINVAL;
		}
	} else if (!up->pending) {
		if (sk->sk_state != TCP_ESTABLISHED)
			return -EDESTADDRREQ;
		daddr = &np->daddr;
611
	} else
L
Linus Torvalds 已提交
612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632
		daddr = NULL;

	if (daddr) {
		if (ipv6_addr_type(daddr) == IPV6_ADDR_MAPPED) {
			struct sockaddr_in sin;
			sin.sin_family = AF_INET;
			sin.sin_port = sin6 ? sin6->sin6_port : inet->dport;
			sin.sin_addr.s_addr = daddr->s6_addr32[3];
			msg->msg_name = &sin;
			msg->msg_namelen = sizeof(sin);
do_udp_sendmsg:
			if (__ipv6_only_sock(sk))
				return -ENETUNREACH;
			return udp_sendmsg(iocb, sk, msg, len);
		}
	}

	if (up->pending == AF_INET)
		return udp_sendmsg(iocb, sk, msg, len);

	/* Rough check on arithmetic overflow,
633
	   better check is made in ip6_append_data().
L
Linus Torvalds 已提交
634 635 636
	   */
	if (len > INT_MAX - sizeof(struct udphdr))
		return -EMSGSIZE;
637

L
Linus Torvalds 已提交
638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655
	if (up->pending) {
		/*
		 * There are pending frames.
		 * The socket lock must be held while it's corked.
		 */
		lock_sock(sk);
		if (likely(up->pending)) {
			if (unlikely(up->pending != AF_INET6)) {
				release_sock(sk);
				return -EAFNOSUPPORT;
			}
			dst = NULL;
			goto do_append_data;
		}
		release_sock(sk);
	}
	ulen += sizeof(struct udphdr);

H
Herbert Xu 已提交
656
	memset(&fl, 0, sizeof(fl));
L
Linus Torvalds 已提交
657 658 659 660 661

	if (sin6) {
		if (sin6->sin6_port == 0)
			return -EINVAL;

H
Herbert Xu 已提交
662
		fl.fl_ip_dport = sin6->sin6_port;
L
Linus Torvalds 已提交
663 664 665
		daddr = &sin6->sin6_addr;

		if (np->sndflow) {
H
Herbert Xu 已提交
666 667 668
			fl.fl6_flowlabel = sin6->sin6_flowinfo&IPV6_FLOWINFO_MASK;
			if (fl.fl6_flowlabel&IPV6_FLOWLABEL_MASK) {
				flowlabel = fl6_sock_lookup(sk, fl.fl6_flowlabel);
L
Linus Torvalds 已提交
669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685
				if (flowlabel == NULL)
					return -EINVAL;
				daddr = &flowlabel->dst;
			}
		}

		/*
		 * Otherwise it will be difficult to maintain
		 * sk->sk_dst_cache.
		 */
		if (sk->sk_state == TCP_ESTABLISHED &&
		    ipv6_addr_equal(daddr, &np->daddr))
			daddr = &np->daddr;

		if (addr_len >= sizeof(struct sockaddr_in6) &&
		    sin6->sin6_scope_id &&
		    ipv6_addr_type(daddr)&IPV6_ADDR_LINKLOCAL)
H
Herbert Xu 已提交
686
			fl.oif = sin6->sin6_scope_id;
L
Linus Torvalds 已提交
687 688 689 690
	} else {
		if (sk->sk_state != TCP_ESTABLISHED)
			return -EDESTADDRREQ;

H
Herbert Xu 已提交
691
		fl.fl_ip_dport = inet->dport;
L
Linus Torvalds 已提交
692
		daddr = &np->daddr;
H
Herbert Xu 已提交
693
		fl.fl6_flowlabel = np->flow_label;
694
		connected = 1;
L
Linus Torvalds 已提交
695 696
	}

H
Herbert Xu 已提交
697 698
	if (!fl.oif)
		fl.oif = sk->sk_bound_dev_if;
L
Linus Torvalds 已提交
699 700 701 702 703 704

	if (msg->msg_controllen) {
		opt = &opt_space;
		memset(opt, 0, sizeof(struct ipv6_txoptions));
		opt->tot_len = sizeof(*opt);

H
Herbert Xu 已提交
705
		err = datagram_send_ctl(msg, &fl, opt, &hlimit, &tclass);
L
Linus Torvalds 已提交
706 707 708 709
		if (err < 0) {
			fl6_sock_release(flowlabel);
			return err;
		}
H
Herbert Xu 已提交
710 711
		if ((fl.fl6_flowlabel&IPV6_FLOWLABEL_MASK) && !flowlabel) {
			flowlabel = fl6_sock_lookup(sk, fl.fl6_flowlabel);
L
Linus Torvalds 已提交
712 713 714 715 716
			if (flowlabel == NULL)
				return -EINVAL;
		}
		if (!(opt->opt_nflen|opt->opt_flen))
			opt = NULL;
717
		connected = 0;
L
Linus Torvalds 已提交
718 719 720
	}
	if (opt == NULL)
		opt = np->opt;
721 722 723
	if (flowlabel)
		opt = fl6_merge_options(&opt_space, flowlabel, opt);
	opt = ipv6_fixup_options(&opt_space, opt);
L
Linus Torvalds 已提交
724

725
	fl.proto = sk->sk_protocol;
H
Herbert Xu 已提交
726 727 728 729
	ipv6_addr_copy(&fl.fl6_dst, daddr);
	if (ipv6_addr_any(&fl.fl6_src) && !ipv6_addr_any(&np->saddr))
		ipv6_addr_copy(&fl.fl6_src, &np->saddr);
	fl.fl_ip_sport = inet->sport;
730

L
Linus Torvalds 已提交
731 732 733
	/* merge ip6_build_xmit from ip6_output */
	if (opt && opt->srcrt) {
		struct rt0_hdr *rt0 = (struct rt0_hdr *) opt->srcrt;
H
Herbert Xu 已提交
734 735
		ipv6_addr_copy(&final, &fl.fl6_dst);
		ipv6_addr_copy(&fl.fl6_dst, rt0->addr);
L
Linus Torvalds 已提交
736
		final_p = &final;
737
		connected = 0;
L
Linus Torvalds 已提交
738 739
	}

H
Herbert Xu 已提交
740 741
	if (!fl.oif && ipv6_addr_is_multicast(&fl.fl6_dst)) {
		fl.oif = np->mcast_oif;
742 743
		connected = 0;
	}
L
Linus Torvalds 已提交
744

H
Herbert Xu 已提交
745
	security_sk_classify_flow(sk, &fl);
V
Venkat Yekkirala 已提交
746

H
Herbert Xu 已提交
747
	err = ip6_sk_dst_lookup(sk, &dst, &fl);
L
Linus Torvalds 已提交
748 749 750
	if (err)
		goto out;
	if (final_p)
H
Herbert Xu 已提交
751
		ipv6_addr_copy(&fl.fl6_dst, final_p);
L
Linus Torvalds 已提交
752

753 754 755 756 757 758
	if ((err = __xfrm_lookup(&dst, &fl, sk, 1)) < 0) {
		if (err == -EREMOTE)
			err = ip6_dst_blackhole(sk, &dst, &fl);
		if (err < 0)
			goto out;
	}
L
Linus Torvalds 已提交
759 760

	if (hlimit < 0) {
H
Herbert Xu 已提交
761
		if (ipv6_addr_is_multicast(&fl.fl6_dst))
L
Linus Torvalds 已提交
762 763 764 765 766 767 768 769 770
			hlimit = np->mcast_hops;
		else
			hlimit = np->hop_limit;
		if (hlimit < 0)
			hlimit = dst_metric(dst, RTAX_HOPLIMIT);
		if (hlimit < 0)
			hlimit = ipv6_get_hoplimit(dst->dev);
	}

771 772 773 774 775 776
	if (tclass < 0) {
		tclass = np->tclass;
		if (tclass < 0)
			tclass = 0;
	}

L
Linus Torvalds 已提交
777 778 779 780 781 782 783 784 785 786
	if (msg->msg_flags&MSG_CONFIRM)
		goto do_confirm;
back_from_confirm:

	lock_sock(sk);
	if (unlikely(up->pending)) {
		/* The socket is already corked while preparing it. */
		/* ... which is an evident application bug. --ANK */
		release_sock(sk);

787
		LIMIT_NETDEBUG(KERN_DEBUG "udp cork app bug 2\n");
L
Linus Torvalds 已提交
788 789 790 791 792 793 794 795
		err = -EINVAL;
		goto out;
	}

	up->pending = AF_INET6;

do_append_data:
	up->len += ulen;
796 797
	getfrag  =  is_udplite ?  udplite_getfrag : ip_generic_getfrag;
	err = ip6_append_data(sk, getfrag, msg->msg_iov, ulen,
H
Herbert Xu 已提交
798
		sizeof(struct udphdr), hlimit, tclass, opt, &fl,
799 800
		(struct rt6_info*)dst,
		corkreq ? msg->msg_flags|MSG_MORE : msg->msg_flags);
L
Linus Torvalds 已提交
801 802 803
	if (err)
		udp_v6_flush_pending_frames(sk);
	else if (!corkreq)
804
		err = udp_v6_push_pending_frames(sk);
H
Herbert Xu 已提交
805 806
	else if (unlikely(skb_queue_empty(&sk->sk_write_queue)))
		up->pending = 0;
L
Linus Torvalds 已提交
807

808 809 810
	if (dst) {
		if (connected) {
			ip6_dst_store(sk, dst,
H
Herbert Xu 已提交
811
				      ipv6_addr_equal(&fl.fl6_dst, &np->daddr) ?
812 813
				      &np->daddr : NULL,
#ifdef CONFIG_IPV6_SUBTREES
H
Herbert Xu 已提交
814
				      ipv6_addr_equal(&fl.fl6_src, &np->saddr) ?
815 816 817
				      &np->saddr :
#endif
				      NULL);
818 819 820 821 822
		} else {
			dst_release(dst);
		}
	}

L
Linus Torvalds 已提交
823 824 825 826 827
	if (err > 0)
		err = np->recverr ? net_xmit_errno(err) : 0;
	release_sock(sk);
out:
	fl6_sock_release(flowlabel);
828
	if (!err)
L
Linus Torvalds 已提交
829
		return len;
830 831 832 833 834 835 836 837
	/*
	 * ENOBUFS = no kernel mem, SOCK_NOSPACE = no sndbuf space.  Reporting
	 * ENOBUFS might not be good (it's not tunable per se), but otherwise
	 * we don't have a good statistic (IpOutDiscards but it can be too many
	 * things).  We could add another new stat but at least for now that
	 * seems like overkill.
	 */
	if (err == -ENOBUFS || test_bit(SOCK_NOSPACE, &sk->sk_socket->flags)) {
838
		UDP6_INC_STATS_USER(UDP_MIB_SNDBUFERRORS, is_udplite);
839
	}
L
Linus Torvalds 已提交
840 841 842 843 844 845 846 847 848 849
	return err;

do_confirm:
	dst_confirm(dst);
	if (!(msg->msg_flags&MSG_PROBE) || len)
		goto back_from_confirm;
	err = 0;
	goto out;
}

850
int udpv6_destroy_sock(struct sock *sk)
L
Linus Torvalds 已提交
851 852 853 854 855 856 857 858 859 860 861 862 863
{
	lock_sock(sk);
	udp_v6_flush_pending_frames(sk);
	release_sock(sk);

	inet6_destroy_sock(sk);

	return 0;
}

/*
 *	Socket option code for UDP
 */
864 865
int udpv6_setsockopt(struct sock *sk, int level, int optname,
		     char __user *optval, int optlen)
866
{
867
	if (level == SOL_UDP  ||  level == SOL_UDPLITE)
868 869
		return udp_lib_setsockopt(sk, level, optname, optval, optlen,
					  udp_v6_push_pending_frames);
870
	return ipv6_setsockopt(sk, level, optname, optval, optlen);
871 872 873
}

#ifdef CONFIG_COMPAT
874 875
int compat_udpv6_setsockopt(struct sock *sk, int level, int optname,
			    char __user *optval, int optlen)
876
{
877
	if (level == SOL_UDP  ||  level == SOL_UDPLITE)
878 879
		return udp_lib_setsockopt(sk, level, optname, optval, optlen,
					  udp_v6_push_pending_frames);
880
	return compat_ipv6_setsockopt(sk, level, optname, optval, optlen);
881 882 883
}
#endif

884 885
int udpv6_getsockopt(struct sock *sk, int level, int optname,
		     char __user *optval, int __user *optlen)
886
{
887
	if (level == SOL_UDP  ||  level == SOL_UDPLITE)
888
		return udp_lib_getsockopt(sk, level, optname, optval, optlen);
889
	return ipv6_getsockopt(sk, level, optname, optval, optlen);
890 891 892
}

#ifdef CONFIG_COMPAT
893 894
int compat_udpv6_getsockopt(struct sock *sk, int level, int optname,
			    char __user *optval, int __user *optlen)
895
{
896
	if (level == SOL_UDP  ||  level == SOL_UDPLITE)
897
		return udp_lib_getsockopt(sk, level, optname, optval, optlen);
898
	return compat_ipv6_getsockopt(sk, level, optname, optval, optlen);
899 900 901
}
#endif

L
Linus Torvalds 已提交
902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929
static struct inet6_protocol udpv6_protocol = {
	.handler	=	udpv6_rcv,
	.err_handler	=	udpv6_err,
	.flags		=	INET6_PROTO_NOPOLICY|INET6_PROTO_FINAL,
};

/* ------------------------------------------------------------------------ */
#ifdef CONFIG_PROC_FS

static void udp6_sock_seq_show(struct seq_file *seq, struct sock *sp, int bucket)
{
	struct inet_sock *inet = inet_sk(sp);
	struct ipv6_pinfo *np = inet6_sk(sp);
	struct in6_addr *dest, *src;
	__u16 destp, srcp;

	dest  = &np->daddr;
	src   = &np->rcv_saddr;
	destp = ntohs(inet->dport);
	srcp  = ntohs(inet->sport);
	seq_printf(seq,
		   "%4d: %08X%08X%08X%08X:%04X %08X%08X%08X%08X:%04X "
		   "%02X %08X:%08X %02X:%08lX %08X %5d %8d %lu %d %p\n",
		   bucket,
		   src->s6_addr32[0], src->s6_addr32[1],
		   src->s6_addr32[2], src->s6_addr32[3], srcp,
		   dest->s6_addr32[0], dest->s6_addr32[1],
		   dest->s6_addr32[2], dest->s6_addr32[3], destp,
930
		   sp->sk_state,
L
Linus Torvalds 已提交
931 932 933 934 935 936 937 938
		   atomic_read(&sp->sk_wmem_alloc),
		   atomic_read(&sp->sk_rmem_alloc),
		   0, 0L, 0,
		   sock_i_uid(sp), 0,
		   sock_i_ino(sp),
		   atomic_read(&sp->sk_refcnt), sp);
}

939
int udp6_seq_show(struct seq_file *seq, void *v)
L
Linus Torvalds 已提交
940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957
{
	if (v == SEQ_START_TOKEN)
		seq_printf(seq,
			   "  sl  "
			   "local_address                         "
			   "remote_address                        "
			   "st tx_queue rx_queue tr tm->when retrnsmt"
			   "   uid  timeout inode\n");
	else
		udp6_sock_seq_show(seq, v, ((struct udp_iter_state *)seq->private)->bucket);
	return 0;
}

static struct file_operations udp6_seq_fops;
static struct udp_seq_afinfo udp6_seq_afinfo = {
	.owner		= THIS_MODULE,
	.name		= "udp6",
	.family		= AF_INET6,
958
	.hashtable	= udp_hash,
L
Linus Torvalds 已提交
959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975
	.seq_show	= udp6_seq_show,
	.seq_fops	= &udp6_seq_fops,
};

int __init udp6_proc_init(void)
{
	return udp_proc_register(&udp6_seq_afinfo);
}

void udp6_proc_exit(void) {
	udp_proc_unregister(&udp6_seq_afinfo);
}
#endif /* CONFIG_PROC_FS */

/* ------------------------------------------------------------------------ */

struct proto udpv6_prot = {
976 977
	.name		   = "UDPv6",
	.owner		   = THIS_MODULE,
978
	.close		   = udp_lib_close,
979 980 981 982 983 984 985 986 987
	.connect	   = ip6_datagram_connect,
	.disconnect	   = udp_disconnect,
	.ioctl		   = udp_ioctl,
	.destroy	   = udpv6_destroy_sock,
	.setsockopt	   = udpv6_setsockopt,
	.getsockopt	   = udpv6_getsockopt,
	.sendmsg	   = udpv6_sendmsg,
	.recvmsg	   = udpv6_recvmsg,
	.backlog_rcv	   = udpv6_queue_rcv_skb,
988 989
	.hash		   = udp_lib_hash,
	.unhash		   = udp_lib_unhash,
990 991
	.get_port	   = udp_v6_get_port,
	.obj_size	   = sizeof(struct udp6_sock),
992
#ifdef CONFIG_COMPAT
993 994
	.compat_setsockopt = compat_udpv6_setsockopt,
	.compat_getsockopt = compat_udpv6_getsockopt,
995
#endif
L
Linus Torvalds 已提交
996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014
};

static struct inet_protosw udpv6_protosw = {
	.type =      SOCK_DGRAM,
	.protocol =  IPPROTO_UDP,
	.prot =      &udpv6_prot,
	.ops =       &inet6_dgram_ops,
	.capability =-1,
	.no_check =  UDP_CSUM_DEFAULT,
	.flags =     INET_PROTOSW_PERMANENT,
};


void __init udpv6_init(void)
{
	if (inet6_add_protocol(&udpv6_protocol, IPPROTO_UDP) < 0)
		printk(KERN_ERR "udpv6_init: Could not register protocol\n");
	inet6_register_protosw(&udpv6_protosw);
}