ip6_output.c 40.1 KB
Newer Older
L
Linus Torvalds 已提交
1 2
/*
 *	IPv6 output functions
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
 *
 *	Based on linux/net/ipv4/ip_output.c
 *
 *	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.
 *
 *	Changes:
 *	A.N.Kuznetsov	:	airthmetics in fragmentation.
 *				extension headers are implemented.
 *				route changes now work.
 *				ip6_forward does not confuse sniffers.
 *				etc.
 *
 *      H. von Brand    :       Added missing #include <linux/string.h>
 *	Imran Patel	: 	frag id should be in NBO
 *      Kazunori MIYAZAWA @USAGI
 *			:       add ip6_append_data and related functions
 *				for datagram xmit
 */

#include <linux/errno.h>
H
Herbert Xu 已提交
30
#include <linux/kernel.h>
L
Linus Torvalds 已提交
31 32 33 34 35 36 37 38
#include <linux/string.h>
#include <linux/socket.h>
#include <linux/net.h>
#include <linux/netdevice.h>
#include <linux/if_arp.h>
#include <linux/in6.h>
#include <linux/tcp.h>
#include <linux/route.h>
39
#include <linux/module.h>
40
#include <linux/slab.h>
L
Linus Torvalds 已提交
41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56

#include <linux/netfilter.h>
#include <linux/netfilter_ipv6.h>

#include <net/sock.h>
#include <net/snmp.h>

#include <net/ipv6.h>
#include <net/ndisc.h>
#include <net/protocol.h>
#include <net/ip6_route.h>
#include <net/addrconf.h>
#include <net/rawv6.h>
#include <net/icmp.h>
#include <net/xfrm.h>
#include <net/checksum.h>
57
#include <linux/mroute6.h>
L
Linus Torvalds 已提交
58

59
int ip6_fragment(struct sk_buff *skb, int (*output)(struct sk_buff *));
L
Linus Torvalds 已提交
60

H
Herbert Xu 已提交
61 62 63 64 65 66 67 68 69
int __ip6_local_out(struct sk_buff *skb)
{
	int len;

	len = skb->len - sizeof(struct ipv6hdr);
	if (len > IPV6_MAXPLEN)
		len = 0;
	ipv6_hdr(skb)->payload_len = htons(len);

70 71
	return nf_hook(NFPROTO_IPV6, NF_INET_LOCAL_OUT, skb, NULL,
		       skb_dst(skb)->dev, dst_output);
H
Herbert Xu 已提交
72 73 74 75 76 77 78 79 80 81 82 83 84 85
}

int ip6_local_out(struct sk_buff *skb)
{
	int err;

	err = __ip6_local_out(skb);
	if (likely(err == 1))
		err = dst_output(skb);

	return err;
}
EXPORT_SYMBOL_GPL(ip6_local_out);

86
static int ip6_finish_output2(struct sk_buff *skb)
L
Linus Torvalds 已提交
87
{
E
Eric Dumazet 已提交
88
	struct dst_entry *dst = skb_dst(skb);
L
Linus Torvalds 已提交
89
	struct net_device *dev = dst->dev;
90
	struct neighbour *neigh;
91
	struct rt6_info *rt;
L
Linus Torvalds 已提交
92 93 94 95

	skb->protocol = htons(ETH_P_IPV6);
	skb->dev = dev;

96
	if (ipv6_addr_is_multicast(&ipv6_hdr(skb)->daddr)) {
E
Eric Dumazet 已提交
97
		struct inet6_dev *idev = ip6_dst_idev(skb_dst(skb));
L
Linus Torvalds 已提交
98

99
		if (!(dev->flags & IFF_LOOPBACK) && sk_mc_loop(skb->sk) &&
100
		    ((mroute6_socket(dev_net(dev), skb) &&
101
		     !(IP6CB(skb)->flags & IP6SKB_FORWARDED)) ||
102 103
		     ipv6_chk_mcast_addr(dev, &ipv6_hdr(skb)->daddr,
					 &ipv6_hdr(skb)->saddr))) {
L
Linus Torvalds 已提交
104 105 106 107 108 109
			struct sk_buff *newskb = skb_clone(skb, GFP_ATOMIC);

			/* Do not check for IFF_ALLMULTI; multicast routing
			   is not supported in any case.
			 */
			if (newskb)
110 111
				NF_HOOK(NFPROTO_IPV6, NF_INET_POST_ROUTING,
					newskb, NULL, newskb->dev,
112
					dev_loopback_xmit);
L
Linus Torvalds 已提交
113

114
			if (ipv6_hdr(skb)->hop_limit == 0) {
115 116
				IP6_INC_STATS(dev_net(dev), idev,
					      IPSTATS_MIB_OUTDISCARDS);
L
Linus Torvalds 已提交
117 118 119 120 121
				kfree_skb(skb);
				return 0;
			}
		}

122 123
		IP6_UPD_PO_STATS(dev_net(dev), idev, IPSTATS_MIB_OUTMCAST,
				skb->len);
L
Linus Torvalds 已提交
124 125
	}

126 127
	rt = (struct rt6_info *) dst;
	neigh = rt->n;
128 129
	if (neigh)
		return dst_neigh_output(dst, neigh, skb);
130

131 132 133 134
	IP6_INC_STATS_BH(dev_net(dst->dev),
			 ip6_dst_idev(dst), IPSTATS_MIB_OUTNOROUTES);
	kfree_skb(skb);
	return -EINVAL;
L
Linus Torvalds 已提交
135 136
}

137 138 139 140 141 142 143 144 145
static int ip6_finish_output(struct sk_buff *skb)
{
	if ((skb->len > ip6_skb_dst_mtu(skb) && !skb_is_gso(skb)) ||
	    dst_allfrag(skb_dst(skb)))
		return ip6_fragment(skb, ip6_finish_output2);
	else
		return ip6_finish_output2(skb);
}

L
Linus Torvalds 已提交
146 147
int ip6_output(struct sk_buff *skb)
{
148
	struct net_device *dev = skb_dst(skb)->dev;
E
Eric Dumazet 已提交
149
	struct inet6_dev *idev = ip6_dst_idev(skb_dst(skb));
150
	if (unlikely(idev->cnf.disable_ipv6)) {
151
		IP6_INC_STATS(dev_net(dev), idev,
152
			      IPSTATS_MIB_OUTDISCARDS);
153 154 155 156
		kfree_skb(skb);
		return 0;
	}

157 158 159
	return NF_HOOK_COND(NFPROTO_IPV6, NF_INET_POST_ROUTING, skb, NULL, dev,
			    ip6_finish_output,
			    !(IP6CB(skb)->flags & IP6SKB_REROUTED));
L
Linus Torvalds 已提交
160 161 162
}

/*
S
Shan Wei 已提交
163
 *	xmit an sk_buff (used by TCP, SCTP and DCCP)
L
Linus Torvalds 已提交
164 165
 */

166
int ip6_xmit(struct sock *sk, struct sk_buff *skb, struct flowi6 *fl6,
167
	     struct ipv6_txoptions *opt, int tclass)
L
Linus Torvalds 已提交
168
{
169
	struct net *net = sock_net(sk);
170
	struct ipv6_pinfo *np = inet6_sk(sk);
171
	struct in6_addr *first_hop = &fl6->daddr;
E
Eric Dumazet 已提交
172
	struct dst_entry *dst = skb_dst(skb);
L
Linus Torvalds 已提交
173
	struct ipv6hdr *hdr;
174
	u8  proto = fl6->flowi6_proto;
L
Linus Torvalds 已提交
175
	int seg_len = skb->len;
G
Gerrit Renker 已提交
176
	int hlimit = -1;
L
Linus Torvalds 已提交
177 178 179
	u32 mtu;

	if (opt) {
180
		unsigned int head_room;
L
Linus Torvalds 已提交
181 182 183 184 185 186 187 188 189 190

		/* First: exthdrs may take lots of space (~8K for now)
		   MAX_HEADER is not enough.
		 */
		head_room = opt->opt_nflen + opt->opt_flen;
		seg_len += head_room;
		head_room += sizeof(struct ipv6hdr) + LL_RESERVED_SPACE(dst->dev);

		if (skb_headroom(skb) < head_room) {
			struct sk_buff *skb2 = skb_realloc_headroom(skb, head_room);
191
			if (skb2 == NULL) {
E
Eric Dumazet 已提交
192
				IP6_INC_STATS(net, ip6_dst_idev(skb_dst(skb)),
193 194
					      IPSTATS_MIB_OUTDISCARDS);
				kfree_skb(skb);
L
Linus Torvalds 已提交
195 196
				return -ENOBUFS;
			}
197
			consume_skb(skb);
198
			skb = skb2;
199
			skb_set_owner_w(skb, sk);
L
Linus Torvalds 已提交
200 201 202 203 204 205 206
		}
		if (opt->opt_flen)
			ipv6_push_frag_opts(skb, opt, &proto);
		if (opt->opt_nflen)
			ipv6_push_nfrag_opts(skb, opt, &proto, &first_hop);
	}

207 208
	skb_push(skb, sizeof(struct ipv6hdr));
	skb_reset_network_header(skb);
209
	hdr = ipv6_hdr(skb);
L
Linus Torvalds 已提交
210 211 212 213

	/*
	 *	Fill in the IPv6 header
	 */
214
	if (np)
L
Linus Torvalds 已提交
215 216
		hlimit = np->hop_limit;
	if (hlimit < 0)
217
		hlimit = ip6_dst_hoplimit(dst);
L
Linus Torvalds 已提交
218

219
	ip6_flow_hdr(hdr, tclass, fl6->flowlabel);
220

L
Linus Torvalds 已提交
221 222 223 224
	hdr->payload_len = htons(seg_len);
	hdr->nexthdr = proto;
	hdr->hop_limit = hlimit;

A
Alexey Dobriyan 已提交
225 226
	hdr->saddr = fl6->saddr;
	hdr->daddr = *first_hop;
L
Linus Torvalds 已提交
227

228
	skb->priority = sk->sk_priority;
229
	skb->mark = sk->sk_mark;
230

L
Linus Torvalds 已提交
231
	mtu = dst_mtu(dst);
232
	if ((skb->len <= mtu) || skb->local_df || skb_is_gso(skb)) {
E
Eric Dumazet 已提交
233
		IP6_UPD_PO_STATS(net, ip6_dst_idev(skb_dst(skb)),
234
			      IPSTATS_MIB_OUT, skb->len);
235 236
		return NF_HOOK(NFPROTO_IPV6, NF_INET_LOCAL_OUT, skb, NULL,
			       dst->dev, dst_output);
L
Linus Torvalds 已提交
237 238
	}

239
	net_dbg_ratelimited("IPv6: sending pkt_too_big to self\n");
L
Linus Torvalds 已提交
240
	skb->dev = dst->dev;
241
	icmpv6_send(skb, ICMPV6_PKT_TOOBIG, 0, mtu);
E
Eric Dumazet 已提交
242
	IP6_INC_STATS(net, ip6_dst_idev(skb_dst(skb)), IPSTATS_MIB_FRAGFAILS);
L
Linus Torvalds 已提交
243 244 245 246
	kfree_skb(skb);
	return -EMSGSIZE;
}

247 248
EXPORT_SYMBOL(ip6_xmit);

L
Linus Torvalds 已提交
249 250 251 252 253 254 255 256
/*
 *	To avoid extra problems ND packets are send through this
 *	routine. It's code duplication but I really want to avoid
 *	extra checks since ipv6_build_header is used by TCP (which
 *	is for us performance critical)
 */

int ip6_nd_hdr(struct sock *sk, struct sk_buff *skb, struct net_device *dev,
257
	       const struct in6_addr *saddr, const struct in6_addr *daddr,
L
Linus Torvalds 已提交
258 259 260 261 262 263 264 265
	       int proto, int len)
{
	struct ipv6_pinfo *np = inet6_sk(sk);
	struct ipv6hdr *hdr;

	skb->protocol = htons(ETH_P_IPV6);
	skb->dev = dev;

266 267
	skb_reset_network_header(skb);
	skb_put(skb, sizeof(struct ipv6hdr));
268
	hdr = ipv6_hdr(skb);
L
Linus Torvalds 已提交
269

270
	ip6_flow_hdr(hdr, 0, 0);
L
Linus Torvalds 已提交
271 272 273 274 275

	hdr->payload_len = htons(len);
	hdr->nexthdr = proto;
	hdr->hop_limit = np->hop_limit;

A
Alexey Dobriyan 已提交
276 277
	hdr->saddr = *saddr;
	hdr->daddr = *daddr;
L
Linus Torvalds 已提交
278 279 280 281 282 283 284 285 286 287 288 289

	return 0;
}

static int ip6_call_ra_chain(struct sk_buff *skb, int sel)
{
	struct ip6_ra_chain *ra;
	struct sock *last = NULL;

	read_lock(&ip6_ra_lock);
	for (ra = ip6_ra_chain; ra; ra = ra->next) {
		struct sock *sk = ra->sk;
290 291 292
		if (sk && ra->sel == sel &&
		    (!sk->sk_bound_dev_if ||
		     sk->sk_bound_dev_if == skb->dev->ifindex)) {
L
Linus Torvalds 已提交
293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310
			if (last) {
				struct sk_buff *skb2 = skb_clone(skb, GFP_ATOMIC);
				if (skb2)
					rawv6_rcv(last, skb2);
			}
			last = sk;
		}
	}

	if (last) {
		rawv6_rcv(last, skb);
		read_unlock(&ip6_ra_lock);
		return 1;
	}
	read_unlock(&ip6_ra_lock);
	return 0;
}

311 312
static int ip6_forward_proxy_check(struct sk_buff *skb)
{
313
	struct ipv6hdr *hdr = ipv6_hdr(skb);
314
	u8 nexthdr = hdr->nexthdr;
315
	__be16 frag_off;
316 317 318
	int offset;

	if (ipv6_ext_hdr(nexthdr)) {
319
		offset = ipv6_skip_exthdr(skb, sizeof(*hdr), &nexthdr, &frag_off);
320 321 322 323 324 325 326 327
		if (offset < 0)
			return 0;
	} else
		offset = sizeof(struct ipv6hdr);

	if (nexthdr == IPPROTO_ICMPV6) {
		struct icmp6hdr *icmp6;

328 329
		if (!pskb_may_pull(skb, (skb_network_header(skb) +
					 offset + 1 - skb->data)))
330 331
			return 0;

332
		icmp6 = (struct icmp6hdr *)(skb_network_header(skb) + offset);
333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349

		switch (icmp6->icmp6_type) {
		case NDISC_ROUTER_SOLICITATION:
		case NDISC_ROUTER_ADVERTISEMENT:
		case NDISC_NEIGHBOUR_SOLICITATION:
		case NDISC_NEIGHBOUR_ADVERTISEMENT:
		case NDISC_REDIRECT:
			/* For reaction involving unicast neighbor discovery
			 * message destined to the proxied address, pass it to
			 * input function.
			 */
			return 1;
		default:
			break;
		}
	}

350 351 352 353 354 355 356 357 358 359
	/*
	 * The proxying router can't forward traffic sent to a link-local
	 * address, so signal the sender and discard the packet. This
	 * behavior is clarified by the MIPv6 specification.
	 */
	if (ipv6_addr_type(&hdr->daddr) & IPV6_ADDR_LINKLOCAL) {
		dst_link_failure(skb);
		return -1;
	}

360 361 362
	return 0;
}

L
Linus Torvalds 已提交
363 364 365 366 367 368 369
static inline int ip6_forward_finish(struct sk_buff *skb)
{
	return dst_output(skb);
}

int ip6_forward(struct sk_buff *skb)
{
E
Eric Dumazet 已提交
370
	struct dst_entry *dst = skb_dst(skb);
371
	struct ipv6hdr *hdr = ipv6_hdr(skb);
L
Linus Torvalds 已提交
372
	struct inet6_skb_parm *opt = IP6CB(skb);
373
	struct net *net = dev_net(dst->dev);
374
	u32 mtu;
375

376
	if (net->ipv6.devconf_all->forwarding == 0)
L
Linus Torvalds 已提交
377 378
		goto error;

379 380 381
	if (skb_warn_if_lro(skb))
		goto drop;

L
Linus Torvalds 已提交
382
	if (!xfrm6_policy_check(NULL, XFRM_POLICY_FWD, skb)) {
383
		IP6_INC_STATS(net, ip6_dst_idev(dst), IPSTATS_MIB_INDISCARDS);
L
Linus Torvalds 已提交
384 385 386
		goto drop;
	}

387 388 389
	if (skb->pkt_type != PACKET_HOST)
		goto drop;

390
	skb_forward_csum(skb);
L
Linus Torvalds 已提交
391 392 393 394 395 396 397 398 399 400 401 402 403 404 405

	/*
	 *	We DO NOT make any processing on
	 *	RA packets, pushing them to user level AS IS
	 *	without ane WARRANTY that application will be able
	 *	to interpret them. The reason is that we
	 *	cannot make anything clever here.
	 *
	 *	We are not end-node, so that if packet contains
	 *	AH/ESP, we cannot make anything.
	 *	Defragmentation also would be mistake, RA packets
	 *	cannot be fragmented, because there is no warranty
	 *	that different fragments will go along one path. --ANK
	 */
	if (opt->ra) {
406
		u8 *ptr = skb_network_header(skb) + opt->ra;
L
Linus Torvalds 已提交
407 408 409 410 411 412 413 414 415 416
		if (ip6_call_ra_chain(skb, (ptr[2]<<8) + ptr[3]))
			return 0;
	}

	/*
	 *	check and decrement ttl
	 */
	if (hdr->hop_limit <= 1) {
		/* Force OUTPUT device used as source address */
		skb->dev = dst->dev;
417
		icmpv6_send(skb, ICMPV6_TIME_EXCEED, ICMPV6_EXC_HOPLIMIT, 0);
418 419
		IP6_INC_STATS_BH(net,
				 ip6_dst_idev(dst), IPSTATS_MIB_INHDRERRORS);
L
Linus Torvalds 已提交
420 421 422 423 424

		kfree_skb(skb);
		return -ETIMEDOUT;
	}

425
	/* XXX: idev->cnf.proxy_ndp? */
426
	if (net->ipv6.devconf_all->proxy_ndp &&
427
	    pneigh_lookup(&nd_tbl, net, &hdr->daddr, skb->dev, 0)) {
428 429
		int proxied = ip6_forward_proxy_check(skb);
		if (proxied > 0)
430
			return ip6_input(skb);
431
		else if (proxied < 0) {
432 433
			IP6_INC_STATS(net, ip6_dst_idev(dst),
				      IPSTATS_MIB_INDISCARDS);
434 435
			goto drop;
		}
436 437
	}

L
Linus Torvalds 已提交
438
	if (!xfrm6_route_forward(skb)) {
439
		IP6_INC_STATS(net, ip6_dst_idev(dst), IPSTATS_MIB_INDISCARDS);
L
Linus Torvalds 已提交
440 441
		goto drop;
	}
E
Eric Dumazet 已提交
442
	dst = skb_dst(skb);
L
Linus Torvalds 已提交
443 444 445

	/* IPv6 specs say nothing about it, but it is clear that we cannot
	   send redirects to source routed frames.
446
	   We don't send redirects to frames decapsulated from IPsec.
L
Linus Torvalds 已提交
447
	 */
448
	if (skb->dev == dst->dev && opt->srcrt == 0 && !skb_sec_path(skb)) {
L
Linus Torvalds 已提交
449
		struct in6_addr *target = NULL;
450
		struct inet_peer *peer;
L
Linus Torvalds 已提交
451 452 453 454 455 456 457 458
		struct rt6_info *rt;

		/*
		 *	incoming and outgoing devices are the same
		 *	send a redirect.
		 */

		rt = (struct rt6_info *) dst;
459 460
		if (rt->rt6i_flags & RTF_GATEWAY)
			target = &rt->rt6i_gateway;
L
Linus Torvalds 已提交
461 462 463
		else
			target = &hdr->daddr;

464
		peer = inet_getpeer_v6(net->ipv6.peers, &rt->rt6i_dst.addr, 1);
465

L
Linus Torvalds 已提交
466 467 468
		/* Limit redirects both by destination (here)
		   and by source (inside ndisc_send_redirect)
		 */
469
		if (inet_peer_xrlim_allow(peer, 1*HZ))
470
			ndisc_send_redirect(skb, target);
471 472
		if (peer)
			inet_putpeer(peer);
473 474 475
	} else {
		int addrtype = ipv6_addr_type(&hdr->saddr);

L
Linus Torvalds 已提交
476
		/* This check is security critical. */
477 478
		if (addrtype == IPV6_ADDR_ANY ||
		    addrtype & (IPV6_ADDR_MULTICAST | IPV6_ADDR_LOOPBACK))
479 480 481
			goto error;
		if (addrtype & IPV6_ADDR_LINKLOCAL) {
			icmpv6_send(skb, ICMPV6_DEST_UNREACH,
482
				    ICMPV6_NOT_NEIGHBOUR, 0);
483 484
			goto error;
		}
L
Linus Torvalds 已提交
485 486
	}

487 488 489 490
	mtu = dst_mtu(dst);
	if (mtu < IPV6_MIN_MTU)
		mtu = IPV6_MIN_MTU;

491 492
	if ((!skb->local_df && skb->len > mtu && !skb_is_gso(skb)) ||
	    (IP6CB(skb)->frag_max_size && IP6CB(skb)->frag_max_size > mtu)) {
L
Linus Torvalds 已提交
493 494
		/* Again, force OUTPUT device used as source address */
		skb->dev = dst->dev;
495
		icmpv6_send(skb, ICMPV6_PKT_TOOBIG, 0, mtu);
496 497 498 499
		IP6_INC_STATS_BH(net,
				 ip6_dst_idev(dst), IPSTATS_MIB_INTOOBIGERRORS);
		IP6_INC_STATS_BH(net,
				 ip6_dst_idev(dst), IPSTATS_MIB_FRAGFAILS);
L
Linus Torvalds 已提交
500 501 502 503 504
		kfree_skb(skb);
		return -EMSGSIZE;
	}

	if (skb_cow(skb, dst->dev->hard_header_len)) {
505
		IP6_INC_STATS(net, ip6_dst_idev(dst), IPSTATS_MIB_OUTDISCARDS);
L
Linus Torvalds 已提交
506 507 508
		goto drop;
	}

509
	hdr = ipv6_hdr(skb);
L
Linus Torvalds 已提交
510 511

	/* Mangling hops number delayed to point after skb COW */
512

L
Linus Torvalds 已提交
513 514
	hdr->hop_limit--;

515
	IP6_INC_STATS_BH(net, ip6_dst_idev(dst), IPSTATS_MIB_OUTFORWDATAGRAMS);
516
	IP6_ADD_STATS_BH(net, ip6_dst_idev(dst), IPSTATS_MIB_OUTOCTETS, skb->len);
517
	return NF_HOOK(NFPROTO_IPV6, NF_INET_FORWARD, skb, skb->dev, dst->dev,
518
		       ip6_forward_finish);
L
Linus Torvalds 已提交
519 520

error:
521
	IP6_INC_STATS_BH(net, ip6_dst_idev(dst), IPSTATS_MIB_INADDRERRORS);
L
Linus Torvalds 已提交
522 523 524 525 526 527 528 529 530 531
drop:
	kfree_skb(skb);
	return -EINVAL;
}

static void ip6_copy_metadata(struct sk_buff *to, struct sk_buff *from)
{
	to->pkt_type = from->pkt_type;
	to->priority = from->priority;
	to->protocol = from->protocol;
E
Eric Dumazet 已提交
532 533
	skb_dst_drop(to);
	skb_dst_set(to, dst_clone(skb_dst(from)));
L
Linus Torvalds 已提交
534
	to->dev = from->dev;
T
Thomas Graf 已提交
535
	to->mark = from->mark;
L
Linus Torvalds 已提交
536 537 538 539

#ifdef CONFIG_NET_SCHED
	to->tc_index = from->tc_index;
#endif
540
	nf_copy(to, from);
A
Amerigo Wang 已提交
541
#if IS_ENABLED(CONFIG_NETFILTER_XT_TARGET_TRACE)
542 543
	to->nf_trace = from->nf_trace;
#endif
544
	skb_copy_secmark(to, from);
L
Linus Torvalds 已提交
545 546
}

547
int ip6_fragment(struct sk_buff *skb, int (*output)(struct sk_buff *))
L
Linus Torvalds 已提交
548 549
{
	struct sk_buff *frag;
E
Eric Dumazet 已提交
550
	struct rt6_info *rt = (struct rt6_info*)skb_dst(skb);
551
	struct ipv6_pinfo *np = skb->sk ? inet6_sk(skb->sk) : NULL;
L
Linus Torvalds 已提交
552 553 554
	struct ipv6hdr *tmp_hdr;
	struct frag_hdr *fh;
	unsigned int mtu, hlen, left, len;
555
	int hroom, troom;
A
Al Viro 已提交
556
	__be32 frag_id = 0;
L
Linus Torvalds 已提交
557 558
	int ptr, offset = 0, err=0;
	u8 *prevhdr, nexthdr = 0;
E
Eric Dumazet 已提交
559
	struct net *net = dev_net(skb_dst(skb)->dev);
L
Linus Torvalds 已提交
560 561 562 563

	hlen = ip6_find_1stfragopt(skb, &prevhdr);
	nexthdr = *prevhdr;

J
John Heffner 已提交
564
	mtu = ip6_skb_dst_mtu(skb);
565 566

	/* We must not fragment if the socket is set to force MTU discovery
567
	 * or if the skb it not generated by a local socket.
568
	 */
569 570 571
	if (unlikely(!skb->local_df && skb->len > mtu) ||
		     (IP6CB(skb)->frag_max_size &&
		      IP6CB(skb)->frag_max_size > mtu)) {
572 573 574
		if (skb->sk && dst_allfrag(skb_dst(skb)))
			sk_nocaps_add(skb->sk, NETIF_F_GSO_MASK);

E
Eric Dumazet 已提交
575
		skb->dev = skb_dst(skb)->dev;
576
		icmpv6_send(skb, ICMPV6_PKT_TOOBIG, 0, mtu);
E
Eric Dumazet 已提交
577
		IP6_INC_STATS(net, ip6_dst_idev(skb_dst(skb)),
578
			      IPSTATS_MIB_FRAGFAILS);
579 580 581 582
		kfree_skb(skb);
		return -EMSGSIZE;
	}

583 584 585 586 587
	if (np && np->frag_size < mtu) {
		if (np->frag_size)
			mtu = np->frag_size;
	}
	mtu -= hlen + sizeof(struct frag_hdr);
L
Linus Torvalds 已提交
588

589
	if (skb_has_frag_list(skb)) {
L
Linus Torvalds 已提交
590
		int first_len = skb_pagelen(skb);
591
		struct sk_buff *frag2;
L
Linus Torvalds 已提交
592 593 594 595 596 597

		if (first_len - hlen > mtu ||
		    ((first_len - hlen) & 7) ||
		    skb_cloned(skb))
			goto slow_path;

598
		skb_walk_frags(skb, frag) {
L
Linus Torvalds 已提交
599 600 601 602
			/* Correct geometry. */
			if (frag->len > mtu ||
			    ((frag->len & 7) && frag->next) ||
			    skb_headroom(frag) < hlen)
603
				goto slow_path_clean;
L
Linus Torvalds 已提交
604 605 606

			/* Partially cloned skb? */
			if (skb_shared(frag))
607
				goto slow_path_clean;
608 609 610 611 612 613

			BUG_ON(frag->sk);
			if (skb->sk) {
				frag->sk = skb->sk;
				frag->destructor = sock_wfree;
			}
614
			skb->truesize -= frag->truesize;
L
Linus Torvalds 已提交
615 616 617 618 619
		}

		err = 0;
		offset = 0;
		frag = skb_shinfo(skb)->frag_list;
620
		skb_frag_list_init(skb);
L
Linus Torvalds 已提交
621 622
		/* BUILD HEADER */

Y
YOSHIFUJI Hideaki 已提交
623
		*prevhdr = NEXTHDR_FRAGMENT;
624
		tmp_hdr = kmemdup(skb_network_header(skb), hlen, GFP_ATOMIC);
L
Linus Torvalds 已提交
625
		if (!tmp_hdr) {
E
Eric Dumazet 已提交
626
			IP6_INC_STATS(net, ip6_dst_idev(skb_dst(skb)),
627
				      IPSTATS_MIB_FRAGFAILS);
L
Linus Torvalds 已提交
628 629 630 631 632
			return -ENOMEM;
		}

		__skb_pull(skb, hlen);
		fh = (struct frag_hdr*)__skb_push(skb, sizeof(struct frag_hdr));
633 634
		__skb_push(skb, hlen);
		skb_reset_network_header(skb);
635
		memcpy(skb_network_header(skb), tmp_hdr, hlen);
L
Linus Torvalds 已提交
636

637
		ipv6_select_ident(fh, rt);
L
Linus Torvalds 已提交
638 639 640 641 642 643 644 645
		fh->nexthdr = nexthdr;
		fh->reserved = 0;
		fh->frag_off = htons(IP6_MF);
		frag_id = fh->identification;

		first_len = skb_pagelen(skb);
		skb->data_len = first_len - skb_headlen(skb);
		skb->len = first_len;
646 647
		ipv6_hdr(skb)->payload_len = htons(first_len -
						   sizeof(struct ipv6hdr));
648

649
		dst_hold(&rt->dst);
L
Linus Torvalds 已提交
650 651 652 653 654 655

		for (;;) {
			/* Prepare header of the next frame,
			 * before previous one went down. */
			if (frag) {
				frag->ip_summed = CHECKSUM_NONE;
656
				skb_reset_transport_header(frag);
L
Linus Torvalds 已提交
657
				fh = (struct frag_hdr*)__skb_push(frag, sizeof(struct frag_hdr));
658 659
				__skb_push(frag, hlen);
				skb_reset_network_header(frag);
660 661
				memcpy(skb_network_header(frag), tmp_hdr,
				       hlen);
L
Linus Torvalds 已提交
662 663 664 665 666 667 668
				offset += skb->len - hlen - sizeof(struct frag_hdr);
				fh->nexthdr = nexthdr;
				fh->reserved = 0;
				fh->frag_off = htons(offset);
				if (frag->next != NULL)
					fh->frag_off |= htons(IP6_MF);
				fh->identification = frag_id;
669 670 671
				ipv6_hdr(frag)->payload_len =
						htons(frag->len -
						      sizeof(struct ipv6hdr));
L
Linus Torvalds 已提交
672 673
				ip6_copy_metadata(frag, skb);
			}
674

L
Linus Torvalds 已提交
675
			err = output(skb);
676
			if(!err)
677
				IP6_INC_STATS(net, ip6_dst_idev(&rt->dst),
678
					      IPSTATS_MIB_FRAGCREATES);
679

L
Linus Torvalds 已提交
680 681 682 683 684 685 686 687
			if (err || !frag)
				break;

			skb = frag;
			frag = skb->next;
			skb->next = NULL;
		}

J
Jesper Juhl 已提交
688
		kfree(tmp_hdr);
L
Linus Torvalds 已提交
689 690

		if (err == 0) {
691
			IP6_INC_STATS(net, ip6_dst_idev(&rt->dst),
692
				      IPSTATS_MIB_FRAGOKS);
A
Amerigo Wang 已提交
693
			ip6_rt_put(rt);
L
Linus Torvalds 已提交
694 695 696 697 698 699 700 701 702
			return 0;
		}

		while (frag) {
			skb = frag->next;
			kfree_skb(frag);
			frag = skb;
		}

703
		IP6_INC_STATS(net, ip6_dst_idev(&rt->dst),
704
			      IPSTATS_MIB_FRAGFAILS);
A
Amerigo Wang 已提交
705
		ip6_rt_put(rt);
L
Linus Torvalds 已提交
706
		return err;
707 708 709 710 711 712 713 714 715

slow_path_clean:
		skb_walk_frags(skb, frag2) {
			if (frag2 == frag)
				break;
			frag2->sk = NULL;
			frag2->destructor = NULL;
			skb->truesize += frag2->truesize;
		}
L
Linus Torvalds 已提交
716 717 718
	}

slow_path:
719 720 721 722
	if ((skb->ip_summed == CHECKSUM_PARTIAL) &&
	    skb_checksum_help(skb))
		goto fail;

L
Linus Torvalds 已提交
723 724 725 726 727 728 729 730
	left = skb->len - hlen;		/* Space per frame */
	ptr = hlen;			/* Where to start from */

	/*
	 *	Fragment the datagram.
	 */

	*prevhdr = NEXTHDR_FRAGMENT;
731 732
	hroom = LL_RESERVED_SPACE(rt->dst.dev);
	troom = rt->dst.dev->needed_tailroom;
L
Linus Torvalds 已提交
733 734 735 736 737 738 739 740 741

	/*
	 *	Keep copying data until we run out.
	 */
	while(left > 0)	{
		len = left;
		/* IF: it doesn't fit, use 'mtu' - the data space left */
		if (len > mtu)
			len = mtu;
L
Lucas De Marchi 已提交
742
		/* IF: we are not sending up to and including the packet end
L
Linus Torvalds 已提交
743 744 745 746 747 748 749 750
		   then align the next start on an eight byte boundary */
		if (len < left)	{
			len &= ~7;
		}
		/*
		 *	Allocate buffer.
		 */

751 752
		if ((frag = alloc_skb(len + hlen + sizeof(struct frag_hdr) +
				      hroom + troom, GFP_ATOMIC)) == NULL) {
753
			NETDEBUG(KERN_INFO "IPv6: frag: no memory for new fragment!\n");
E
Eric Dumazet 已提交
754
			IP6_INC_STATS(net, ip6_dst_idev(skb_dst(skb)),
755
				      IPSTATS_MIB_FRAGFAILS);
L
Linus Torvalds 已提交
756 757 758 759 760 761 762 763 764
			err = -ENOMEM;
			goto fail;
		}

		/*
		 *	Set up data on packet
		 */

		ip6_copy_metadata(frag, skb);
765
		skb_reserve(frag, hroom);
L
Linus Torvalds 已提交
766
		skb_put(frag, len + hlen + sizeof(struct frag_hdr));
767
		skb_reset_network_header(frag);
768
		fh = (struct frag_hdr *)(skb_network_header(frag) + hlen);
769 770
		frag->transport_header = (frag->network_header + hlen +
					  sizeof(struct frag_hdr));
L
Linus Torvalds 已提交
771 772 773 774 775 776 777 778 779 780 781

		/*
		 *	Charge the memory for the fragment to any owner
		 *	it might possess
		 */
		if (skb->sk)
			skb_set_owner_w(frag, skb->sk);

		/*
		 *	Copy the packet header into the new buffer.
		 */
782
		skb_copy_from_linear_data(skb, skb_network_header(frag), hlen);
L
Linus Torvalds 已提交
783 784 785 786 787 788

		/*
		 *	Build fragment header.
		 */
		fh->nexthdr = nexthdr;
		fh->reserved = 0;
789
		if (!frag_id) {
790
			ipv6_select_ident(fh, rt);
L
Linus Torvalds 已提交
791 792 793 794 795 796 797
			frag_id = fh->identification;
		} else
			fh->identification = frag_id;

		/*
		 *	Copy a block of the IP datagram.
		 */
798
		if (skb_copy_bits(skb, ptr, skb_transport_header(frag), len))
L
Linus Torvalds 已提交
799 800 801 802 803 804
			BUG();
		left -= len;

		fh->frag_off = htons(offset);
		if (left > 0)
			fh->frag_off |= htons(IP6_MF);
805 806
		ipv6_hdr(frag)->payload_len = htons(frag->len -
						    sizeof(struct ipv6hdr));
L
Linus Torvalds 已提交
807 808 809 810 811 812 813 814 815 816

		ptr += len;
		offset += len;

		/*
		 *	Put this fragment into the sending queue.
		 */
		err = output(frag);
		if (err)
			goto fail;
817

E
Eric Dumazet 已提交
818
		IP6_INC_STATS(net, ip6_dst_idev(skb_dst(skb)),
819
			      IPSTATS_MIB_FRAGCREATES);
L
Linus Torvalds 已提交
820
	}
E
Eric Dumazet 已提交
821
	IP6_INC_STATS(net, ip6_dst_idev(skb_dst(skb)),
822
		      IPSTATS_MIB_FRAGOKS);
823
	consume_skb(skb);
L
Linus Torvalds 已提交
824 825 826
	return err;

fail:
E
Eric Dumazet 已提交
827
	IP6_INC_STATS(net, ip6_dst_idev(skb_dst(skb)),
828
		      IPSTATS_MIB_FRAGFAILS);
829
	kfree_skb(skb);
L
Linus Torvalds 已提交
830 831 832
	return err;
}

833 834 835
static inline int ip6_rt_check(const struct rt6key *rt_key,
			       const struct in6_addr *fl_addr,
			       const struct in6_addr *addr_cache)
836
{
E
Eric Dumazet 已提交
837 838
	return (rt_key->plen != 128 || !ipv6_addr_equal(fl_addr, &rt_key->addr)) &&
		(addr_cache == NULL || !ipv6_addr_equal(fl_addr, addr_cache));
839 840
}

841 842
static struct dst_entry *ip6_sk_dst_check(struct sock *sk,
					  struct dst_entry *dst,
843
					  const struct flowi6 *fl6)
L
Linus Torvalds 已提交
844
{
845 846
	struct ipv6_pinfo *np = inet6_sk(sk);
	struct rt6_info *rt = (struct rt6_info *)dst;
L
Linus Torvalds 已提交
847

848 849 850 851 852 853 854 855
	if (!dst)
		goto out;

	/* Yes, checking route validity in not connected
	 * case is not very simple. Take into account,
	 * that we do not support routing by source, TOS,
	 * and MSG_DONTROUTE 		--ANK (980726)
	 *
856 857
	 * 1. ip6_rt_check(): If route was host route,
	 *    check that cached destination is current.
858 859 860 861 862 863 864 865 866 867
	 *    If it is network route, we still may
	 *    check its validity using saved pointer
	 *    to the last used address: daddr_cache.
	 *    We do not want to save whole address now,
	 *    (because main consumer of this service
	 *    is tcp, which has not this problem),
	 *    so that the last trick works only on connected
	 *    sockets.
	 * 2. oif also should be the same.
	 */
868
	if (ip6_rt_check(&rt->rt6i_dst, &fl6->daddr, np->daddr_cache) ||
869
#ifdef CONFIG_IPV6_SUBTREES
870
	    ip6_rt_check(&rt->rt6i_src, &fl6->saddr, np->saddr_cache) ||
871
#endif
872
	    (fl6->flowi6_oif && fl6->flowi6_oif != dst->dev->ifindex)) {
873 874
		dst_release(dst);
		dst = NULL;
L
Linus Torvalds 已提交
875 876
	}

877 878 879 880 881
out:
	return dst;
}

static int ip6_dst_lookup_tail(struct sock *sk,
882
			       struct dst_entry **dst, struct flowi6 *fl6)
883
{
884
	struct net *net = sock_net(sk);
885 886
#ifdef CONFIG_IPV6_OPTIMISTIC_DAD
	struct neighbour *n;
887
	struct rt6_info *rt;
888 889
#endif
	int err;
890

L
Linus Torvalds 已提交
891
	if (*dst == NULL)
892
		*dst = ip6_route_output(net, sk, fl6);
L
Linus Torvalds 已提交
893 894 895 896

	if ((err = (*dst)->error))
		goto out_err_release;

897
	if (ipv6_addr_any(&fl6->saddr)) {
898 899 900 901
		struct rt6_info *rt = (struct rt6_info *) *dst;
		err = ip6_route_get_saddr(net, rt, &fl6->daddr,
					  sk ? inet6_sk(sk)->srcprefs : 0,
					  &fl6->saddr);
902
		if (err)
L
Linus Torvalds 已提交
903 904 905
			goto out_err_release;
	}

906
#ifdef CONFIG_IPV6_OPTIMISTIC_DAD
907 908 909 910 911 912 913 914
	/*
	 * Here if the dst entry we've looked up
	 * has a neighbour entry that is in the INCOMPLETE
	 * state and the src address from the flow is
	 * marked as OPTIMISTIC, we release the found
	 * dst entry and replace it instead with the
	 * dst entry of the nexthop router
	 */
915
	rt = (struct rt6_info *) *dst;
916 917 918 919 920 921
	rcu_read_lock_bh();
	n = __ipv6_neigh_lookup_noref(rt->dst.dev, rt6_nexthop(rt, &fl6->daddr));
	err = n && !(n->nud_state & NUD_VALID) ? -EINVAL : 0;
	rcu_read_unlock_bh();

	if (err) {
922
		struct inet6_ifaddr *ifp;
923
		struct flowi6 fl_gw6;
924 925
		int redirect;

926
		ifp = ipv6_get_ifaddr(net, &fl6->saddr,
927 928 929 930 931 932 933 934 935 936 937 938
				      (*dst)->dev, 1);

		redirect = (ifp && ifp->flags & IFA_F_OPTIMISTIC);
		if (ifp)
			in6_ifa_put(ifp);

		if (redirect) {
			/*
			 * We need to get the dst entry for the
			 * default router instead
			 */
			dst_release(*dst);
939 940 941
			memcpy(&fl_gw6, fl6, sizeof(struct flowi6));
			memset(&fl_gw6.daddr, 0, sizeof(struct in6_addr));
			*dst = ip6_route_output(net, sk, &fl_gw6);
942 943
			if ((err = (*dst)->error))
				goto out_err_release;
944
		}
945
	}
946 947
#endif

L
Linus Torvalds 已提交
948 949 950
	return 0;

out_err_release:
951
	if (err == -ENETUNREACH)
952
		IP6_INC_STATS_BH(net, NULL, IPSTATS_MIB_OUTNOROUTES);
L
Linus Torvalds 已提交
953 954 955 956
	dst_release(*dst);
	*dst = NULL;
	return err;
}
A
Adrian Bunk 已提交
957

958 959 960 961
/**
 *	ip6_dst_lookup - perform route lookup on flow
 *	@sk: socket which provides route info
 *	@dst: pointer to dst_entry * for result
962
 *	@fl6: flow to lookup
963 964 965 966 967
 *
 *	This function performs a route lookup on the given flow.
 *
 *	It returns zero on success, or a standard errno code on error.
 */
968
int ip6_dst_lookup(struct sock *sk, struct dst_entry **dst, struct flowi6 *fl6)
969 970
{
	*dst = NULL;
971
	return ip6_dst_lookup_tail(sk, dst, fl6);
972
}
973 974
EXPORT_SYMBOL_GPL(ip6_dst_lookup);

975
/**
976 977
 *	ip6_dst_lookup_flow - perform route lookup on flow with ipsec
 *	@sk: socket which provides route info
978
 *	@fl6: flow to lookup
979
 *	@final_dst: final destination address for ipsec lookup
980
 *	@can_sleep: we are in a sleepable context
981 982 983 984 985 986
 *
 *	This function performs a route lookup on the given flow.
 *
 *	It returns a valid dst pointer on success, or a pointer encoded
 *	error code.
 */
987
struct dst_entry *ip6_dst_lookup_flow(struct sock *sk, struct flowi6 *fl6,
988
				      const struct in6_addr *final_dst,
989
				      bool can_sleep)
990 991 992 993
{
	struct dst_entry *dst = NULL;
	int err;

994
	err = ip6_dst_lookup_tail(sk, &dst, fl6);
995 996 997
	if (err)
		return ERR_PTR(err);
	if (final_dst)
A
Alexey Dobriyan 已提交
998
		fl6->daddr = *final_dst;
999
	if (can_sleep)
1000
		fl6->flowi6_flags |= FLOWI_FLAG_CAN_SLEEP;
1001

1002
	return xfrm_lookup(sock_net(sk), dst, flowi6_to_flowi(fl6), sk, 0);
1003 1004 1005 1006 1007
}
EXPORT_SYMBOL_GPL(ip6_dst_lookup_flow);

/**
 *	ip6_sk_dst_lookup_flow - perform socket cached route lookup on flow
1008
 *	@sk: socket which provides the dst cache and route info
1009
 *	@fl6: flow to lookup
1010
 *	@final_dst: final destination address for ipsec lookup
1011
 *	@can_sleep: we are in a sleepable context
1012 1013 1014 1015 1016 1017
 *
 *	This function performs a route lookup on the given flow with the
 *	possibility of using the cached route in the socket if it is valid.
 *	It will take the socket dst lock when operating on the dst cache.
 *	As a result, this function can only be used in process context.
 *
1018 1019
 *	It returns a valid dst pointer on success, or a pointer encoded
 *	error code.
1020
 */
1021
struct dst_entry *ip6_sk_dst_lookup_flow(struct sock *sk, struct flowi6 *fl6,
1022
					 const struct in6_addr *final_dst,
1023
					 bool can_sleep)
1024
{
1025 1026
	struct dst_entry *dst = sk_dst_check(sk, inet6_sk(sk)->dst_cookie);
	int err;
1027

1028
	dst = ip6_sk_dst_check(sk, dst, fl6);
1029

1030
	err = ip6_dst_lookup_tail(sk, &dst, fl6);
1031 1032 1033
	if (err)
		return ERR_PTR(err);
	if (final_dst)
A
Alexey Dobriyan 已提交
1034
		fl6->daddr = *final_dst;
1035
	if (can_sleep)
1036
		fl6->flowi6_flags |= FLOWI_FLAG_CAN_SLEEP;
1037

1038
	return xfrm_lookup(sock_net(sk), dst, flowi6_to_flowi(fl6), sk, 0);
1039
}
1040
EXPORT_SYMBOL_GPL(ip6_sk_dst_lookup_flow);
1041

A
Adrian Bunk 已提交
1042
static inline int ip6_ufo_append_data(struct sock *sk,
1043 1044 1045
			int getfrag(void *from, char *to, int offset, int len,
			int odd, struct sk_buff *skb),
			void *from, int length, int hh_len, int fragheaderlen,
1046 1047
			int transhdrlen, int mtu,unsigned int flags,
			struct rt6_info *rt)
1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061

{
	struct sk_buff *skb;
	int err;

	/* There is support for UDP large send offload by network
	 * device, so create one single skb packet containing complete
	 * udp datagram
	 */
	if ((skb = skb_peek_tail(&sk->sk_write_queue)) == NULL) {
		skb = sock_alloc_send_skb(sk,
			hh_len + fragheaderlen + transhdrlen + 20,
			(flags & MSG_DONTWAIT), &err);
		if (skb == NULL)
1062
			return err;
1063 1064 1065 1066 1067 1068 1069 1070

		/* reserve space for Hardware header */
		skb_reserve(skb, hh_len);

		/* create space for UDP/IP header */
		skb_put(skb,fragheaderlen + transhdrlen);

		/* initialize network header pointer */
1071
		skb_reset_network_header(skb);
1072 1073

		/* initialize protocol header pointer */
1074
		skb->transport_header = skb->network_header + fragheaderlen;
1075

1076
		skb->ip_summed = CHECKSUM_PARTIAL;
1077 1078 1079 1080 1081 1082 1083 1084
		skb->csum = 0;
	}

	err = skb_append_datato_frags(sk,skb, getfrag, from,
				      (length - transhdrlen));
	if (!err) {
		struct frag_hdr fhdr;

1085 1086 1087 1088 1089
		/* Specify the length of each IPv6 datagram fragment.
		 * It has to be a multiple of 8.
		 */
		skb_shinfo(skb)->gso_size = (mtu - fragheaderlen -
					     sizeof(struct frag_hdr)) & ~7;
H
Herbert Xu 已提交
1090
		skb_shinfo(skb)->gso_type = SKB_GSO_UDP;
1091
		ipv6_select_ident(&fhdr, rt);
1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103
		skb_shinfo(skb)->ip6_frag_id = fhdr.identification;
		__skb_queue_tail(&sk->sk_write_queue, skb);

		return 0;
	}
	/* There is not enough support do UPD LSO,
	 * so follow normal path
	 */
	kfree_skb(skb);

	return err;
}
L
Linus Torvalds 已提交
1104

1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116
static inline struct ipv6_opt_hdr *ip6_opt_dup(struct ipv6_opt_hdr *src,
					       gfp_t gfp)
{
	return src ? kmemdup(src, (src->hdrlen + 1) * 8, gfp) : NULL;
}

static inline struct ipv6_rt_hdr *ip6_rthdr_dup(struct ipv6_rt_hdr *src,
						gfp_t gfp)
{
	return src ? kmemdup(src, (src->hdrlen + 1) * 8, gfp) : NULL;
}

G
Gao feng 已提交
1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139
static void ip6_append_data_mtu(int *mtu,
				int *maxfraglen,
				unsigned int fragheaderlen,
				struct sk_buff *skb,
				struct rt6_info *rt)
{
	if (!(rt->dst.flags & DST_XFRM_TUNNEL)) {
		if (skb == NULL) {
			/* first fragment, reserve header_len */
			*mtu = *mtu - rt->dst.header_len;

		} else {
			/*
			 * this fragment is not first, the headers
			 * space is regarded as data space.
			 */
			*mtu = dst_mtu(rt->dst.path);
		}
		*maxfraglen = ((*mtu - fragheaderlen) & ~7)
			      + fragheaderlen - sizeof(struct frag_hdr);
	}
}

1140 1141 1142
int ip6_append_data(struct sock *sk, int getfrag(void *from, char *to,
	int offset, int len, int odd, struct sk_buff *skb),
	void *from, int length, int transhdrlen,
1143
	int hlimit, int tclass, struct ipv6_txoptions *opt, struct flowi6 *fl6,
1144
	struct rt6_info *rt, unsigned int flags, int dontfrag)
L
Linus Torvalds 已提交
1145 1146 1147
{
	struct inet_sock *inet = inet_sk(sk);
	struct ipv6_pinfo *np = inet6_sk(sk);
1148
	struct inet_cork *cork;
G
Gao feng 已提交
1149
	struct sk_buff *skb, *skb_prev = NULL;
L
Linus Torvalds 已提交
1150 1151
	unsigned int maxfraglen, fragheaderlen;
	int exthdrlen;
1152
	int dst_exthdrlen;
L
Linus Torvalds 已提交
1153 1154 1155 1156 1157
	int hh_len;
	int mtu;
	int copy;
	int err;
	int offset = 0;
1158
	__u8 tx_flags = 0;
L
Linus Torvalds 已提交
1159 1160 1161

	if (flags&MSG_PROBE)
		return 0;
1162
	cork = &inet->cork.base;
L
Linus Torvalds 已提交
1163 1164 1165 1166 1167
	if (skb_queue_empty(&sk->sk_write_queue)) {
		/*
		 * setup for corking
		 */
		if (opt) {
1168
			if (WARN_ON(np->cork.opt))
L
Linus Torvalds 已提交
1169
				return -EINVAL;
1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198

			np->cork.opt = kmalloc(opt->tot_len, sk->sk_allocation);
			if (unlikely(np->cork.opt == NULL))
				return -ENOBUFS;

			np->cork.opt->tot_len = opt->tot_len;
			np->cork.opt->opt_flen = opt->opt_flen;
			np->cork.opt->opt_nflen = opt->opt_nflen;

			np->cork.opt->dst0opt = ip6_opt_dup(opt->dst0opt,
							    sk->sk_allocation);
			if (opt->dst0opt && !np->cork.opt->dst0opt)
				return -ENOBUFS;

			np->cork.opt->dst1opt = ip6_opt_dup(opt->dst1opt,
							    sk->sk_allocation);
			if (opt->dst1opt && !np->cork.opt->dst1opt)
				return -ENOBUFS;

			np->cork.opt->hopopt = ip6_opt_dup(opt->hopopt,
							   sk->sk_allocation);
			if (opt->hopopt && !np->cork.opt->hopopt)
				return -ENOBUFS;

			np->cork.opt->srcrt = ip6_rthdr_dup(opt->srcrt,
							    sk->sk_allocation);
			if (opt->srcrt && !np->cork.opt->srcrt)
				return -ENOBUFS;

L
Linus Torvalds 已提交
1199 1200
			/* need source address above miyazawa*/
		}
1201
		dst_hold(&rt->dst);
1202
		cork->dst = &rt->dst;
1203
		inet->cork.fl.u.ip6 = *fl6;
L
Linus Torvalds 已提交
1204
		np->cork.hop_limit = hlimit;
1205
		np->cork.tclass = tclass;
G
Gao feng 已提交
1206 1207 1208 1209 1210 1211
		if (rt->dst.flags & DST_XFRM_TUNNEL)
			mtu = np->pmtudisc == IPV6_PMTUDISC_PROBE ?
			      rt->dst.dev->mtu : dst_mtu(&rt->dst);
		else
			mtu = np->pmtudisc == IPV6_PMTUDISC_PROBE ?
			      rt->dst.dev->mtu : dst_mtu(rt->dst.path);
1212
		if (np->frag_size < mtu) {
1213 1214 1215
			if (np->frag_size)
				mtu = np->frag_size;
		}
1216
		cork->fragsize = mtu;
1217
		if (dst_allfrag(rt->dst.path))
1218 1219
			cork->flags |= IPCORK_ALLFRAG;
		cork->length = 0;
1220
		exthdrlen = (opt ? opt->opt_flen : 0) - rt->rt6i_nfheader_len;
L
Linus Torvalds 已提交
1221 1222
		length += exthdrlen;
		transhdrlen += exthdrlen;
1223
		dst_exthdrlen = rt->dst.header_len;
L
Linus Torvalds 已提交
1224
	} else {
1225
		rt = (struct rt6_info *)cork->dst;
1226
		fl6 = &inet->cork.fl.u.ip6;
1227
		opt = np->cork.opt;
L
Linus Torvalds 已提交
1228 1229
		transhdrlen = 0;
		exthdrlen = 0;
1230
		dst_exthdrlen = 0;
1231
		mtu = cork->fragsize;
L
Linus Torvalds 已提交
1232 1233
	}

1234
	hh_len = LL_RESERVED_SPACE(rt->dst.dev);
L
Linus Torvalds 已提交
1235

1236
	fragheaderlen = sizeof(struct ipv6hdr) + rt->rt6i_nfheader_len +
1237
			(opt ? opt->opt_nflen : 0);
L
Linus Torvalds 已提交
1238 1239 1240
	maxfraglen = ((mtu - fragheaderlen) & ~7) + fragheaderlen - sizeof(struct frag_hdr);

	if (mtu <= sizeof(struct ipv6hdr) + IPV6_MAXPLEN) {
1241
		if (cork->length + length > sizeof(struct ipv6hdr) + IPV6_MAXPLEN - fragheaderlen) {
1242
			ipv6_local_error(sk, EMSGSIZE, fl6, mtu-exthdrlen);
L
Linus Torvalds 已提交
1243 1244 1245 1246
			return -EMSGSIZE;
		}
	}

1247 1248 1249 1250 1251 1252 1253
	/* For UDP, check if TX timestamp is enabled */
	if (sk->sk_type == SOCK_DGRAM) {
		err = sock_tx_timestamp(sk, &tx_flags);
		if (err)
			goto error;
	}

L
Linus Torvalds 已提交
1254 1255 1256 1257 1258 1259 1260
	/*
	 * Let's try using as much space as possible.
	 * Use MTU if total length of the message fits into the MTU.
	 * Otherwise, we need to reserve fragment header and
	 * fragment alignment (= 8-15 octects, in total).
	 *
	 * Note that we may need to "move" the data from the tail of
1261
	 * of the buffer to the new fragment when we split
L
Linus Torvalds 已提交
1262 1263
	 * the message.
	 *
1264
	 * FIXME: It may be fragmented into multiple chunks
L
Linus Torvalds 已提交
1265 1266
	 *        at once if non-fragmentable extension headers
	 *        are too large.
1267
	 * --yoshfuji
L
Linus Torvalds 已提交
1268 1269
	 */

1270
	cork->length += length;
1271 1272 1273
	if (length > mtu) {
		int proto = sk->sk_protocol;
		if (dontfrag && (proto == IPPROTO_UDP || proto == IPPROTO_RAW)){
1274
			ipv6_local_rxpmtu(sk, fl6, mtu-exthdrlen);
1275 1276
			return -EMSGSIZE;
		}
1277

1278
		if (proto == IPPROTO_UDP &&
1279
		    (rt->dst.dev->features & NETIF_F_UFO)) {
1280 1281 1282

			err = ip6_ufo_append_data(sk, getfrag, from, length,
						  hh_len, fragheaderlen,
1283
						  transhdrlen, mtu, flags, rt);
1284 1285 1286 1287
			if (err)
				goto error;
			return 0;
		}
1288
	}
L
Linus Torvalds 已提交
1289 1290 1291 1292 1293 1294

	if ((skb = skb_peek_tail(&sk->sk_write_queue)) == NULL)
		goto alloc_new_skb;

	while (length > 0) {
		/* Check if the remaining data fits into current packet. */
1295
		copy = (cork->length <= mtu && !(cork->flags & IPCORK_ALLFRAG) ? mtu : maxfraglen) - skb->len;
L
Linus Torvalds 已提交
1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306
		if (copy < length)
			copy = maxfraglen - skb->len;

		if (copy <= 0) {
			char *data;
			unsigned int datalen;
			unsigned int fraglen;
			unsigned int fraggap;
			unsigned int alloclen;
alloc_new_skb:
			/* There's no room in the current skb */
G
Gao feng 已提交
1307 1308
			if (skb)
				fraggap = skb->len - maxfraglen;
L
Linus Torvalds 已提交
1309 1310
			else
				fraggap = 0;
G
Gao feng 已提交
1311 1312 1313 1314 1315 1316
			/* update mtu and maxfraglen if necessary */
			if (skb == NULL || skb_prev == NULL)
				ip6_append_data_mtu(&mtu, &maxfraglen,
						    fragheaderlen, skb, rt);

			skb_prev = skb;
L
Linus Torvalds 已提交
1317 1318 1319 1320 1321 1322 1323

			/*
			 * If remaining data exceeds the mtu,
			 * we know we need more fragment(s).
			 */
			datalen = length + fraggap;

G
Gao feng 已提交
1324 1325
			if (datalen > (cork->length <= mtu && !(cork->flags & IPCORK_ALLFRAG) ? mtu : maxfraglen) - fragheaderlen)
				datalen = maxfraglen - fragheaderlen - rt->dst.trailer_len;
L
Linus Torvalds 已提交
1326
			if ((flags & MSG_MORE) &&
1327
			    !(rt->dst.dev->features&NETIF_F_SG))
L
Linus Torvalds 已提交
1328 1329 1330 1331
				alloclen = mtu;
			else
				alloclen = datalen + fragheaderlen;

1332 1333
			alloclen += dst_exthdrlen;

G
Gao feng 已提交
1334 1335 1336 1337 1338 1339 1340 1341 1342 1343
			if (datalen != length + fraggap) {
				/*
				 * this is not the last fragment, the trailer
				 * space is regarded as data space.
				 */
				datalen += rt->dst.trailer_len;
			}

			alloclen += rt->dst.trailer_len;
			fraglen = datalen + fragheaderlen;
L
Linus Torvalds 已提交
1344 1345 1346

			/*
			 * We just reserve space for fragment header.
1347
			 * Note: this may be overallocation if the message
L
Linus Torvalds 已提交
1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364
			 * (without MSG_MORE) fits into the MTU.
			 */
			alloclen += sizeof(struct frag_hdr);

			if (transhdrlen) {
				skb = sock_alloc_send_skb(sk,
						alloclen + hh_len,
						(flags & MSG_DONTWAIT), &err);
			} else {
				skb = NULL;
				if (atomic_read(&sk->sk_wmem_alloc) <=
				    2 * sk->sk_sndbuf)
					skb = sock_wmalloc(sk,
							   alloclen + hh_len, 1,
							   sk->sk_allocation);
				if (unlikely(skb == NULL))
					err = -ENOBUFS;
1365 1366 1367 1368 1369 1370
				else {
					/* Only the initial fragment
					 * is time stamped.
					 */
					tx_flags = 0;
				}
L
Linus Torvalds 已提交
1371 1372 1373 1374 1375 1376
			}
			if (skb == NULL)
				goto error;
			/*
			 *	Fill in the control structures
			 */
1377
			skb->ip_summed = CHECKSUM_NONE;
L
Linus Torvalds 已提交
1378
			skb->csum = 0;
1379 1380 1381
			/* reserve for fragmentation and ipsec header */
			skb_reserve(skb, hh_len + sizeof(struct frag_hdr) +
				    dst_exthdrlen);
L
Linus Torvalds 已提交
1382

1383 1384 1385
			if (sk->sk_type == SOCK_DGRAM)
				skb_shinfo(skb)->tx_flags = tx_flags;

L
Linus Torvalds 已提交
1386 1387 1388
			/*
			 *	Find where to start putting bytes
			 */
1389 1390 1391
			data = skb_put(skb, fraglen);
			skb_set_network_header(skb, exthdrlen);
			data += fragheaderlen;
1392 1393
			skb->transport_header = (skb->network_header +
						 fragheaderlen);
L
Linus Torvalds 已提交
1394 1395 1396 1397 1398 1399 1400
			if (fraggap) {
				skb->csum = skb_copy_and_csum_bits(
					skb_prev, maxfraglen,
					data + transhdrlen, fraggap, 0);
				skb_prev->csum = csum_sub(skb_prev->csum,
							  skb->csum);
				data += fraggap;
1401
				pskb_trim_unique(skb_prev, maxfraglen);
L
Linus Torvalds 已提交
1402 1403
			}
			copy = datalen - transhdrlen - fraggap;
1404

L
Linus Torvalds 已提交
1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418
			if (copy < 0) {
				err = -EINVAL;
				kfree_skb(skb);
				goto error;
			} else if (copy > 0 && getfrag(from, data + transhdrlen, offset, copy, fraggap, skb) < 0) {
				err = -EFAULT;
				kfree_skb(skb);
				goto error;
			}

			offset += copy;
			length -= datalen - fraggap;
			transhdrlen = 0;
			exthdrlen = 0;
1419
			dst_exthdrlen = 0;
L
Linus Torvalds 已提交
1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430

			/*
			 * Put the packet on the pending queue
			 */
			__skb_queue_tail(&sk->sk_write_queue, skb);
			continue;
		}

		if (copy > length)
			copy = length;

1431
		if (!(rt->dst.dev->features&NETIF_F_SG)) {
L
Linus Torvalds 已提交
1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442
			unsigned int off;

			off = skb->len;
			if (getfrag(from, skb_put(skb, copy),
						offset, copy, off, skb) < 0) {
				__skb_trim(skb, off);
				err = -EFAULT;
				goto error;
			}
		} else {
			int i = skb_shinfo(skb)->nr_frags;
1443
			struct page_frag *pfrag = sk_page_frag(sk);
L
Linus Torvalds 已提交
1444

1445 1446
			err = -ENOMEM;
			if (!sk_page_frag_refill(sk, pfrag))
L
Linus Torvalds 已提交
1447
				goto error;
1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458

			if (!skb_can_coalesce(skb, i, pfrag->page,
					      pfrag->offset)) {
				err = -EMSGSIZE;
				if (i == MAX_SKB_FRAGS)
					goto error;

				__skb_fill_page_desc(skb, i, pfrag->page,
						     pfrag->offset, 0);
				skb_shinfo(skb)->nr_frags = ++i;
				get_page(pfrag->page);
L
Linus Torvalds 已提交
1459
			}
1460
			copy = min_t(int, copy, pfrag->size - pfrag->offset);
E
Eric Dumazet 已提交
1461
			if (getfrag(from,
1462 1463 1464 1465 1466 1467
				    page_address(pfrag->page) + pfrag->offset,
				    offset, copy, skb->len, skb) < 0)
				goto error_efault;

			pfrag->offset += copy;
			skb_frag_size_add(&skb_shinfo(skb)->frags[i - 1], copy);
L
Linus Torvalds 已提交
1468 1469
			skb->len += copy;
			skb->data_len += copy;
1470 1471
			skb->truesize += copy;
			atomic_add(copy, &sk->sk_wmem_alloc);
L
Linus Torvalds 已提交
1472 1473 1474 1475
		}
		offset += copy;
		length -= copy;
	}
1476

L
Linus Torvalds 已提交
1477
	return 0;
1478 1479 1480

error_efault:
	err = -EFAULT;
L
Linus Torvalds 已提交
1481
error:
1482
	cork->length -= length;
1483
	IP6_INC_STATS(sock_net(sk), rt->rt6i_idev, IPSTATS_MIB_OUTDISCARDS);
L
Linus Torvalds 已提交
1484 1485
	return err;
}
1486
EXPORT_SYMBOL_GPL(ip6_append_data);
L
Linus Torvalds 已提交
1487

1488 1489
static void ip6_cork_release(struct inet_sock *inet, struct ipv6_pinfo *np)
{
1490 1491 1492 1493 1494 1495 1496 1497 1498
	if (np->cork.opt) {
		kfree(np->cork.opt->dst0opt);
		kfree(np->cork.opt->dst1opt);
		kfree(np->cork.opt->hopopt);
		kfree(np->cork.opt->srcrt);
		kfree(np->cork.opt);
		np->cork.opt = NULL;
	}

1499 1500 1501 1502
	if (inet->cork.base.dst) {
		dst_release(inet->cork.base.dst);
		inet->cork.base.dst = NULL;
		inet->cork.base.flags &= ~IPCORK_ALLFRAG;
1503 1504 1505 1506
	}
	memset(&inet->cork.fl, 0, sizeof(inet->cork.fl));
}

L
Linus Torvalds 已提交
1507 1508 1509 1510 1511 1512 1513
int ip6_push_pending_frames(struct sock *sk)
{
	struct sk_buff *skb, *tmp_skb;
	struct sk_buff **tail_skb;
	struct in6_addr final_dst_buf, *final_dst = &final_dst_buf;
	struct inet_sock *inet = inet_sk(sk);
	struct ipv6_pinfo *np = inet6_sk(sk);
1514
	struct net *net = sock_net(sk);
L
Linus Torvalds 已提交
1515 1516
	struct ipv6hdr *hdr;
	struct ipv6_txoptions *opt = np->cork.opt;
1517
	struct rt6_info *rt = (struct rt6_info *)inet->cork.base.dst;
1518 1519
	struct flowi6 *fl6 = &inet->cork.fl.u.ip6;
	unsigned char proto = fl6->flowi6_proto;
L
Linus Torvalds 已提交
1520 1521 1522 1523 1524 1525 1526
	int err = 0;

	if ((skb = __skb_dequeue(&sk->sk_write_queue)) == NULL)
		goto out;
	tail_skb = &(skb_shinfo(skb)->frag_list);

	/* move skb->data to ip header from ext header */
1527
	if (skb->data < skb_network_header(skb))
1528
		__skb_pull(skb, skb_network_offset(skb));
L
Linus Torvalds 已提交
1529
	while ((tmp_skb = __skb_dequeue(&sk->sk_write_queue)) != NULL) {
1530
		__skb_pull(tmp_skb, skb_network_header_len(skb));
L
Linus Torvalds 已提交
1531 1532 1533 1534 1535 1536 1537 1538 1539
		*tail_skb = tmp_skb;
		tail_skb = &(tmp_skb->next);
		skb->len += tmp_skb->len;
		skb->data_len += tmp_skb->len;
		skb->truesize += tmp_skb->truesize;
		tmp_skb->destructor = NULL;
		tmp_skb->sk = NULL;
	}

1540
	/* Allow local fragmentation. */
1541
	if (np->pmtudisc < IPV6_PMTUDISC_DO)
1542 1543
		skb->local_df = 1;

A
Alexey Dobriyan 已提交
1544
	*final_dst = fl6->daddr;
1545
	__skb_pull(skb, skb_network_header_len(skb));
L
Linus Torvalds 已提交
1546 1547 1548 1549 1550
	if (opt && opt->opt_flen)
		ipv6_push_frag_opts(skb, opt, &proto);
	if (opt && opt->opt_nflen)
		ipv6_push_nfrag_opts(skb, opt, &proto, &final_dst);

1551 1552
	skb_push(skb, sizeof(struct ipv6hdr));
	skb_reset_network_header(skb);
1553
	hdr = ipv6_hdr(skb);
1554

1555
	ip6_flow_hdr(hdr, np->cork.tclass, fl6->flowlabel);
L
Linus Torvalds 已提交
1556 1557
	hdr->hop_limit = np->cork.hop_limit;
	hdr->nexthdr = proto;
A
Alexey Dobriyan 已提交
1558 1559
	hdr->saddr = fl6->saddr;
	hdr->daddr = *final_dst;
L
Linus Torvalds 已提交
1560

1561
	skb->priority = sk->sk_priority;
1562
	skb->mark = sk->sk_mark;
1563

1564
	skb_dst_set(skb, dst_clone(&rt->dst));
1565
	IP6_UPD_PO_STATS(net, rt->rt6i_idev, IPSTATS_MIB_OUT, skb->len);
1566
	if (proto == IPPROTO_ICMPV6) {
E
Eric Dumazet 已提交
1567
		struct inet6_dev *idev = ip6_dst_idev(skb_dst(skb));
1568

1569
		ICMP6MSGOUT_INC_STATS_BH(net, idev, icmp6_hdr(skb)->icmp6_type);
1570
		ICMP6_INC_STATS_BH(net, idev, ICMP6_MIB_OUTMSGS);
1571 1572
	}

H
Herbert Xu 已提交
1573
	err = ip6_local_out(skb);
L
Linus Torvalds 已提交
1574 1575
	if (err) {
		if (err > 0)
E
Eric Dumazet 已提交
1576
			err = net_xmit_errno(err);
L
Linus Torvalds 已提交
1577 1578 1579 1580 1581
		if (err)
			goto error;
	}

out:
1582
	ip6_cork_release(inet, np);
L
Linus Torvalds 已提交
1583 1584
	return err;
error:
1585
	IP6_INC_STATS(net, rt->rt6i_idev, IPSTATS_MIB_OUTDISCARDS);
L
Linus Torvalds 已提交
1586 1587
	goto out;
}
1588
EXPORT_SYMBOL_GPL(ip6_push_pending_frames);
L
Linus Torvalds 已提交
1589 1590 1591 1592 1593 1594

void ip6_flush_pending_frames(struct sock *sk)
{
	struct sk_buff *skb;

	while ((skb = __skb_dequeue_tail(&sk->sk_write_queue)) != NULL) {
E
Eric Dumazet 已提交
1595 1596
		if (skb_dst(skb))
			IP6_INC_STATS(sock_net(sk), ip6_dst_idev(skb_dst(skb)),
1597
				      IPSTATS_MIB_OUTDISCARDS);
L
Linus Torvalds 已提交
1598 1599 1600
		kfree_skb(skb);
	}

1601
	ip6_cork_release(inet_sk(sk), inet6_sk(sk));
L
Linus Torvalds 已提交
1602
}
1603
EXPORT_SYMBOL_GPL(ip6_flush_pending_frames);