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->ip_summed == CHECKSUM_UNNECESSARY)
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 175 176 177 178 179 180
		sin6 = (struct sockaddr_in6 *) msg->msg_name;
		sin6->sin6_family = AF_INET6;
		sin6->sin6_port = skb->h.uh->source;
		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 = skb->h.uh;
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 400
		skb->ip_summed = CHECKSUM_UNNECESSARY;

	if (skb->ip_summed != CHECKSUM_UNNECESSARY)
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;
L
Linus Torvalds 已提交
423 424 425
	uh = skb->h.uh;

	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 445 446
			uh = skb->h.uh;
		}
	}
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 537 538 539 540 541 542

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

	/*
	 * Create a UDP header
	 */
	uh = skb->h.uh;
	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 558 559 560

	err = ip6_push_pending_frames(sk);
out:
	up->len = 0;
	up->pending = 0;
	return err;
}

561
int udpv6_sendmsg(struct kiocb *iocb, struct sock *sk,
L
Linus Torvalds 已提交
562 563 564 565 566 567 568 569 570 571
		  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 已提交
572
	struct flowi fl;
L
Linus Torvalds 已提交
573 574 575 576
	struct dst_entry *dst;
	int addr_len = msg->msg_namelen;
	int ulen = len;
	int hlimit = -1;
577
	int tclass = -1;
L
Linus Torvalds 已提交
578 579
	int corkreq = up->corkflag || msg->msg_flags&MSG_MORE;
	int err;
580
	int connected = 0;
581 582
	int is_udplite = up->pcflag;
	int (*getfrag)(void *, char *, int, int, int, struct sk_buff *);
L
Linus Torvalds 已提交
583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608

	/* 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;
609
	} else
L
Linus Torvalds 已提交
610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630
		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,
631
	   better check is made in ip6_append_data().
L
Linus Torvalds 已提交
632 633 634
	   */
	if (len > INT_MAX - sizeof(struct udphdr))
		return -EMSGSIZE;
635

L
Linus Torvalds 已提交
636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653
	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 已提交
654
	memset(&fl, 0, sizeof(fl));
L
Linus Torvalds 已提交
655 656 657 658 659

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

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

		if (np->sndflow) {
H
Herbert Xu 已提交
664 665 666
			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 已提交
667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683
				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 已提交
684
			fl.oif = sin6->sin6_scope_id;
L
Linus Torvalds 已提交
685 686 687 688
	} else {
		if (sk->sk_state != TCP_ESTABLISHED)
			return -EDESTADDRREQ;

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

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

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

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

723
	fl.proto = sk->sk_protocol;
H
Herbert Xu 已提交
724 725 726 727
	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;
728

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

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

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

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

751
	if ((err = xfrm_lookup(&dst, &fl, sk, 1)) < 0)
L
Linus Torvalds 已提交
752 753 754
		goto out;

	if (hlimit < 0) {
H
Herbert Xu 已提交
755
		if (ipv6_addr_is_multicast(&fl.fl6_dst))
L
Linus Torvalds 已提交
756 757 758 759 760 761 762 763 764
			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);
	}

765 766 767 768 769 770
	if (tclass < 0) {
		tclass = np->tclass;
		if (tclass < 0)
			tclass = 0;
	}

L
Linus Torvalds 已提交
771 772 773 774 775 776 777 778 779 780
	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);

781
		LIMIT_NETDEBUG(KERN_DEBUG "udp cork app bug 2\n");
L
Linus Torvalds 已提交
782 783 784 785 786 787 788 789
		err = -EINVAL;
		goto out;
	}

	up->pending = AF_INET6;

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

802 803 804
	if (dst) {
		if (connected) {
			ip6_dst_store(sk, dst,
H
Herbert Xu 已提交
805
				      ipv6_addr_equal(&fl.fl6_dst, &np->daddr) ?
806 807
				      &np->daddr : NULL,
#ifdef CONFIG_IPV6_SUBTREES
H
Herbert Xu 已提交
808
				      ipv6_addr_equal(&fl.fl6_src, &np->saddr) ?
809 810 811
				      &np->saddr :
#endif
				      NULL);
812 813 814 815 816
		} else {
			dst_release(dst);
		}
	}

L
Linus Torvalds 已提交
817 818 819 820 821 822
	if (err > 0)
		err = np->recverr ? net_xmit_errno(err) : 0;
	release_sock(sk);
out:
	fl6_sock_release(flowlabel);
	if (!err) {
823
		UDP6_INC_STATS_USER(UDP_MIB_OUTDATAGRAMS, is_udplite);
L
Linus Torvalds 已提交
824 825
		return len;
	}
826 827 828 829 830 831 832 833
	/*
	 * 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)) {
834
		UDP6_INC_STATS_USER(UDP_MIB_SNDBUFERRORS, is_udplite);
835
	}
L
Linus Torvalds 已提交
836 837 838 839 840 841 842 843 844 845
	return err;

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

846
int udpv6_destroy_sock(struct sock *sk)
L
Linus Torvalds 已提交
847 848 849 850 851 852 853 854 855 856 857 858 859
{
	lock_sock(sk);
	udp_v6_flush_pending_frames(sk);
	release_sock(sk);

	inet6_destroy_sock(sk);

	return 0;
}

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

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

880 881
int udpv6_getsockopt(struct sock *sk, int level, int optname,
		     char __user *optval, int __user *optlen)
882
{
883
	if (level == SOL_UDP  ||  level == SOL_UDPLITE)
884
		return udp_lib_getsockopt(sk, level, optname, optval, optlen);
885
	return ipv6_getsockopt(sk, level, optname, optval, optlen);
886 887 888
}

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

L
Linus Torvalds 已提交
898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925
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,
926
		   sp->sk_state,
L
Linus Torvalds 已提交
927 928 929 930 931 932 933 934
		   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);
}

935
int udp6_seq_show(struct seq_file *seq, void *v)
L
Linus Torvalds 已提交
936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953
{
	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,
954
	.hashtable	= udp_hash,
L
Linus Torvalds 已提交
955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971
	.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 = {
972 973
	.name		   = "UDPv6",
	.owner		   = THIS_MODULE,
974
	.close		   = udp_lib_close,
975 976 977 978 979 980 981 982 983
	.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,
984 985
	.hash		   = udp_lib_hash,
	.unhash		   = udp_lib_unhash,
986 987
	.get_port	   = udp_v6_get_port,
	.obj_size	   = sizeof(struct udp6_sock),
988
#ifdef CONFIG_COMPAT
989 990
	.compat_setsockopt = compat_udpv6_setsockopt,
	.compat_getsockopt = compat_udpv6_getsockopt,
991
#endif
L
Linus Torvalds 已提交
992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010
};

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);
}