reassembly.c 19.5 KB
Newer Older
L
Linus Torvalds 已提交
1 2
/*
 *	IPv6 fragment reassembly
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
 *
 *	Based on: net/ipv4/ip_fragment.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.
 */

16 17
/*
 *	Fixes:
L
Linus Torvalds 已提交
18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
 *	Andi Kleen	Make it work with multiple hosts.
 *			More RFC compliance.
 *
 *      Horst von Brand Add missing #include <linux/string.h>
 *	Alexey Kuznetsov	SMP races, threading, cleanup.
 *	Patrick McHardy		LRU queue of frag heads for evictor.
 *	Mitsuru KANDA @USAGI	Register inet6_protocol{}.
 *	David Stevens and
 *	YOSHIFUJI,H. @USAGI	Always remove fragment header to
 *				calculate ICV correctly.
 */
#include <linux/errno.h>
#include <linux/types.h>
#include <linux/string.h>
#include <linux/socket.h>
#include <linux/sockios.h>
#include <linux/jiffies.h>
#include <linux/net.h>
#include <linux/list.h>
#include <linux/netdevice.h>
#include <linux/in6.h>
#include <linux/ipv6.h>
#include <linux/icmpv6.h>
#include <linux/random.h>
#include <linux/jhash.h>
43
#include <linux/skbuff.h>
L
Linus Torvalds 已提交
44 45 46 47 48

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

#include <net/ipv6.h>
49
#include <net/ip6_route.h>
L
Linus Torvalds 已提交
50 51 52 53 54
#include <net/protocol.h>
#include <net/transp_v6.h>
#include <net/rawv6.h>
#include <net/ndisc.h>
#include <net/addrconf.h>
55
#include <net/inet_frag.h>
L
Linus Torvalds 已提交
56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71

struct ip6frag_skb_cb
{
	struct inet6_skb_parm	h;
	int			offset;
};

#define FRAG6_CB(skb)	((struct ip6frag_skb_cb*)((skb)->cb))


/*
 *	Equivalent of ipv4 struct ipq
 */

struct frag_queue
{
72
	struct inet_frag_queue	q;
L
Linus Torvalds 已提交
73

A
Al Viro 已提交
74
	__be32			id;		/* fragment id		*/
L
Linus Torvalds 已提交
75 76 77 78 79 80 81 82
	struct in6_addr		saddr;
	struct in6_addr		daddr;

	int			iif;
	unsigned int		csum;
	__u16			nhoffset;
};

83
static struct inet_frags ip6_frags;
L
Linus Torvalds 已提交
84

85
int ip6_frag_nqueues(struct net *net)
86
{
87
	return net->ipv6.frags.nqueues;
88
}
L
Linus Torvalds 已提交
89

90
int ip6_frag_mem(struct net *net)
91
{
92
	return atomic_read(&net->ipv6.frags.mem);
93
}
L
Linus Torvalds 已提交
94

95 96 97
static int ip6_frag_reasm(struct frag_queue *fq, struct sk_buff *prev,
			  struct net_device *dev);

98 99 100 101
/*
 * callers should be careful not to use the hash value outside the ipfrag_lock
 * as doing so could race with ipfrag_hash_rnd being recalculated.
 */
A
Al Viro 已提交
102
static unsigned int ip6qhashfn(__be32 id, struct in6_addr *saddr,
L
Linus Torvalds 已提交
103 104 105 106
			       struct in6_addr *daddr)
{
	u32 a, b, c;

A
Al Viro 已提交
107 108 109
	a = (__force u32)saddr->s6_addr32[0];
	b = (__force u32)saddr->s6_addr32[1];
	c = (__force u32)saddr->s6_addr32[2];
L
Linus Torvalds 已提交
110 111 112

	a += JHASH_GOLDEN_RATIO;
	b += JHASH_GOLDEN_RATIO;
113
	c += ip6_frags.rnd;
L
Linus Torvalds 已提交
114 115
	__jhash_mix(a, b, c);

A
Al Viro 已提交
116 117 118
	a += (__force u32)saddr->s6_addr32[3];
	b += (__force u32)daddr->s6_addr32[0];
	c += (__force u32)daddr->s6_addr32[1];
L
Linus Torvalds 已提交
119 120
	__jhash_mix(a, b, c);

A
Al Viro 已提交
121 122 123
	a += (__force u32)daddr->s6_addr32[2];
	b += (__force u32)daddr->s6_addr32[3];
	c += (__force u32)id;
L
Linus Torvalds 已提交
124 125
	__jhash_mix(a, b, c);

126
	return c & (INETFRAGS_HASHSZ - 1);
L
Linus Torvalds 已提交
127 128
}

129
static unsigned int ip6_hashfn(struct inet_frag_queue *q)
L
Linus Torvalds 已提交
130
{
131
	struct frag_queue *fq;
L
Linus Torvalds 已提交
132

133 134
	fq = container_of(q, struct frag_queue, q);
	return ip6qhashfn(fq->id, &fq->saddr, &fq->daddr);
L
Linus Torvalds 已提交
135 136
}

137 138 139 140 141 142 143 144 145 146 147 148
int ip6_frag_match(struct inet_frag_queue *q, void *a)
{
	struct frag_queue *fq;
	struct ip6_create_arg *arg = a;

	fq = container_of(q, struct frag_queue, q);
	return (fq->id == arg->id &&
			ipv6_addr_equal(&fq->saddr, arg->src) &&
			ipv6_addr_equal(&fq->daddr, arg->dst));
}
EXPORT_SYMBOL(ip6_frag_match);

L
Linus Torvalds 已提交
149
/* Memory Tracking Functions. */
150 151
static inline void frag_kfree_skb(struct netns_frags *nf,
		struct sk_buff *skb, int *work)
L
Linus Torvalds 已提交
152 153 154
{
	if (work)
		*work -= skb->truesize;
155
	atomic_sub(skb->truesize, &nf->mem);
L
Linus Torvalds 已提交
156 157 158
	kfree_skb(skb);
}

159
void ip6_frag_init(struct inet_frag_queue *q, void *a)
L
Linus Torvalds 已提交
160
{
161 162 163 164 165 166
	struct frag_queue *fq = container_of(q, struct frag_queue, q);
	struct ip6_create_arg *arg = a;

	fq->id = arg->id;
	ipv6_addr_copy(&fq->saddr, arg->src);
	ipv6_addr_copy(&fq->daddr, arg->dst);
L
Linus Torvalds 已提交
167
}
168
EXPORT_SYMBOL(ip6_frag_init);
L
Linus Torvalds 已提交
169 170 171

/* Destruction primitives. */

172
static __inline__ void fq_put(struct frag_queue *fq)
L
Linus Torvalds 已提交
173
{
P
Pavel Emelyanov 已提交
174
	inet_frag_put(&fq->q, &ip6_frags);
L
Linus Torvalds 已提交
175 176 177 178 179 180 181
}

/* Kill fq entry. It is not destroyed immediately,
 * because caller (and someone more) holds reference count.
 */
static __inline__ void fq_kill(struct frag_queue *fq)
{
182
	inet_frag_kill(&fq->q, &ip6_frags);
L
Linus Torvalds 已提交
183 184
}

185
static void ip6_evictor(struct net *net, struct inet6_dev *idev)
L
Linus Torvalds 已提交
186
{
187 188
	int evicted;

189
	evicted = inet_frag_evictor(&net->ipv6.frags, &ip6_frags);
190 191
	if (evicted)
		IP6_ADD_STATS_BH(idev, IPSTATS_MIB_REASMFAILS, evicted);
L
Linus Torvalds 已提交
192 193 194 195
}

static void ip6_frag_expire(unsigned long data)
{
196
	struct frag_queue *fq;
197
	struct net_device *dev = NULL;
198
	struct net *net;
L
Linus Torvalds 已提交
199

200 201
	fq = container_of((struct inet_frag_queue *)data, struct frag_queue, q);

202
	spin_lock(&fq->q.lock);
L
Linus Torvalds 已提交
203

204
	if (fq->q.last_in & INET_FRAG_COMPLETE)
L
Linus Torvalds 已提交
205 206 207 208
		goto out;

	fq_kill(fq);

209 210
	net = container_of(fq->q.net, struct net, ipv6.frags);
	dev = dev_get_by_index(net, fq->iif);
211 212 213 214 215 216 217
	if (!dev)
		goto out;

	rcu_read_lock();
	IP6_INC_STATS_BH(__in6_dev_get(dev), IPSTATS_MIB_REASMTIMEOUT);
	IP6_INC_STATS_BH(__in6_dev_get(dev), IPSTATS_MIB_REASMFAILS);
	rcu_read_unlock();
L
Linus Torvalds 已提交
218

219
	/* Don't send error if the first segment did not arrive. */
220
	if (!(fq->q.last_in & INET_FRAG_FIRST_IN) || !fq->q.fragments)
221 222 223 224 225 226 227
		goto out;

	/*
	   But use as source device on which LAST ARRIVED
	   segment was received. And do not use fq->dev
	   pointer directly, device might already disappeared.
	 */
228 229
	fq->q.fragments->dev = dev;
	icmpv6_send(fq->q.fragments, ICMPV6_TIME_EXCEED, ICMPV6_EXC_FRAGTIME, 0, dev);
L
Linus Torvalds 已提交
230
out:
231 232
	if (dev)
		dev_put(dev);
233
	spin_unlock(&fq->q.lock);
234
	fq_put(fq);
L
Linus Torvalds 已提交
235 236
}

237
static __inline__ struct frag_queue *
238
fq_find(struct net *net, __be32 id, struct in6_addr *src, struct in6_addr *dst,
239
	struct inet6_dev *idev)
L
Linus Torvalds 已提交
240
{
241 242
	struct inet_frag_queue *q;
	struct ip6_create_arg arg;
243
	unsigned int hash;
L
Linus Torvalds 已提交
244

245 246 247
	arg.id = id;
	arg.src = src;
	arg.dst = dst;
248
	hash = ip6qhashfn(id, src, dst);
L
Linus Torvalds 已提交
249

250
	q = inet_frag_find(&net->ipv6.frags, &ip6_frags, &arg, hash);
251 252
	if (q == NULL)
		goto oom;
L
Linus Torvalds 已提交
253

254
	return container_of(q, struct frag_queue, q);
L
Linus Torvalds 已提交
255 256

oom:
257
	IP6_INC_STATS_BH(idev, IPSTATS_MIB_REASMFAILS);
L
Linus Torvalds 已提交
258 259 260
	return NULL;
}

261
static int ip6_frag_queue(struct frag_queue *fq, struct sk_buff *skb,
L
Linus Torvalds 已提交
262 263 264
			   struct frag_hdr *fhdr, int nhoff)
{
	struct sk_buff *prev, *next;
265
	struct net_device *dev;
L
Linus Torvalds 已提交
266 267
	int offset, end;

268
	if (fq->q.last_in & INET_FRAG_COMPLETE)
L
Linus Torvalds 已提交
269 270 271
		goto err;

	offset = ntohs(fhdr->frag_off) & ~0x7;
272 273
	end = offset + (ntohs(ipv6_hdr(skb)->payload_len) -
			((u8 *)(fhdr + 1) - (u8 *)(ipv6_hdr(skb) + 1)));
L
Linus Torvalds 已提交
274 275

	if ((unsigned int)end > IPV6_MAXPLEN) {
276 277
		IP6_INC_STATS_BH(ip6_dst_idev(skb->dst),
				 IPSTATS_MIB_INHDRERRORS);
278 279 280
		icmpv6_param_prob(skb, ICMPV6_HDR_FIELD,
				  ((u8 *)&fhdr->frag_off -
				   skb_network_header(skb)));
281
		return -1;
L
Linus Torvalds 已提交
282 283
	}

284 285
	if (skb->ip_summed == CHECKSUM_COMPLETE) {
		const unsigned char *nh = skb_network_header(skb);
286
		skb->csum = csum_sub(skb->csum,
287 288 289
				     csum_partial(nh, (u8 *)(fhdr + 1) - nh,
						  0));
	}
L
Linus Torvalds 已提交
290 291 292 293 294 295

	/* Is this the final fragment? */
	if (!(fhdr->frag_off & htons(IP6_MF))) {
		/* If we already have some bits beyond end
		 * or have different end, the segment is corrupted.
		 */
296
		if (end < fq->q.len ||
297
		    ((fq->q.last_in & INET_FRAG_LAST_IN) && end != fq->q.len))
L
Linus Torvalds 已提交
298
			goto err;
299
		fq->q.last_in |= INET_FRAG_LAST_IN;
300
		fq->q.len = end;
L
Linus Torvalds 已提交
301 302 303 304 305 306 307 308
	} else {
		/* Check if the fragment is rounded to 8 bytes.
		 * Required by the RFC.
		 */
		if (end & 0x7) {
			/* RFC2460 says always send parameter problem in
			 * this case. -DaveM
			 */
309 310
			IP6_INC_STATS_BH(ip6_dst_idev(skb->dst),
					 IPSTATS_MIB_INHDRERRORS);
311
			icmpv6_param_prob(skb, ICMPV6_HDR_FIELD,
L
Linus Torvalds 已提交
312
					  offsetof(struct ipv6hdr, payload_len));
313
			return -1;
L
Linus Torvalds 已提交
314
		}
315
		if (end > fq->q.len) {
L
Linus Torvalds 已提交
316
			/* Some bits beyond end -> corruption. */
317
			if (fq->q.last_in & INET_FRAG_LAST_IN)
L
Linus Torvalds 已提交
318
				goto err;
319
			fq->q.len = end;
L
Linus Torvalds 已提交
320 321 322 323 324 325 326 327 328
		}
	}

	if (end == offset)
		goto err;

	/* Point into the IP datagram 'data' part. */
	if (!pskb_pull(skb, (u8 *) (fhdr + 1) - skb->data))
		goto err;
329

330 331
	if (pskb_trim_rcsum(skb, end - offset))
		goto err;
L
Linus Torvalds 已提交
332 333 334 335 336 337

	/* Find out which fragments are in front and at the back of us
	 * in the chain of fragments so far.  We must know where to put
	 * this fragment, right?
	 */
	prev = NULL;
338
	for(next = fq->q.fragments; next != NULL; next = next->next) {
L
Linus Torvalds 已提交
339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374
		if (FRAG6_CB(next)->offset >= offset)
			break;	/* bingo! */
		prev = next;
	}

	/* We found where to put this one.  Check for overlap with
	 * preceding fragment, and, if needed, align things so that
	 * any overlaps are eliminated.
	 */
	if (prev) {
		int i = (FRAG6_CB(prev)->offset + prev->len) - offset;

		if (i > 0) {
			offset += i;
			if (end <= offset)
				goto err;
			if (!pskb_pull(skb, i))
				goto err;
			if (skb->ip_summed != CHECKSUM_UNNECESSARY)
				skb->ip_summed = CHECKSUM_NONE;
		}
	}

	/* Look for overlap with succeeding segments.
	 * If we can merge fragments, do it.
	 */
	while (next && FRAG6_CB(next)->offset < end) {
		int i = end - FRAG6_CB(next)->offset; /* overlap is 'i' bytes */

		if (i < next->len) {
			/* Eat head of the next overlapped fragment
			 * and leave the loop. The next ones cannot overlap.
			 */
			if (!pskb_pull(next, i))
				goto err;
			FRAG6_CB(next)->offset += i;	/* next fragment */
375
			fq->q.meat -= i;
L
Linus Torvalds 已提交
376 377 378 379 380 381 382 383 384 385 386 387 388 389
			if (next->ip_summed != CHECKSUM_UNNECESSARY)
				next->ip_summed = CHECKSUM_NONE;
			break;
		} else {
			struct sk_buff *free_it = next;

			/* Old fragment is completely overridden with
			 * new one drop it.
			 */
			next = next->next;

			if (prev)
				prev->next = next;
			else
390
				fq->q.fragments = next;
L
Linus Torvalds 已提交
391

392
			fq->q.meat -= free_it->len;
393
			frag_kfree_skb(fq->q.net, free_it, NULL);
L
Linus Torvalds 已提交
394 395 396 397 398 399 400 401 402 403
		}
	}

	FRAG6_CB(skb)->offset = offset;

	/* Insert this fragment in the chain of fragments. */
	skb->next = next;
	if (prev)
		prev->next = skb;
	else
404
		fq->q.fragments = skb;
L
Linus Torvalds 已提交
405

406 407 408 409 410
	dev = skb->dev;
	if (dev) {
		fq->iif = dev->ifindex;
		skb->dev = NULL;
	}
411 412
	fq->q.stamp = skb->tstamp;
	fq->q.meat += skb->len;
413
	atomic_add(skb->truesize, &fq->q.net->mem);
L
Linus Torvalds 已提交
414 415 416 417 418 419

	/* The first fragment.
	 * nhoffset is obtained from the first fragment, of course.
	 */
	if (offset == 0) {
		fq->nhoffset = nhoff;
420
		fq->q.last_in |= INET_FRAG_FIRST_IN;
L
Linus Torvalds 已提交
421
	}
422

423 424
	if (fq->q.last_in == (INET_FRAG_FIRST_IN | INET_FRAG_LAST_IN) &&
	    fq->q.meat == fq->q.len)
425 426
		return ip6_frag_reasm(fq, prev, dev);

427
	write_lock(&ip6_frags.lock);
428
	list_move_tail(&fq->q.lru_list, &fq->q.net->lru_list);
429
	write_unlock(&ip6_frags.lock);
430
	return -1;
L
Linus Torvalds 已提交
431 432

err:
433
	IP6_INC_STATS(ip6_dst_idev(skb->dst), IPSTATS_MIB_REASMFAILS);
L
Linus Torvalds 已提交
434
	kfree_skb(skb);
435
	return -1;
L
Linus Torvalds 已提交
436 437 438 439 440 441 442 443 444 445 446
}

/*
 *	Check if this packet is complete.
 *	Returns NULL on failure by any reason, and pointer
 *	to current nexthdr field in reassembled frame.
 *
 *	It is called with locked fq, and caller must check that
 *	queue is eligible for reassembly i.e. it is not COMPLETE,
 *	the last and the first frames arrived and all the bits are here.
 */
447
static int ip6_frag_reasm(struct frag_queue *fq, struct sk_buff *prev,
L
Linus Torvalds 已提交
448 449
			  struct net_device *dev)
{
450
	struct sk_buff *fp, *head = fq->q.fragments;
L
Linus Torvalds 已提交
451 452 453 454 455
	int    payload_len;
	unsigned int nhoff;

	fq_kill(fq);

456 457 458 459 460 461 462 463 464 465 466
	/* Make the one we just received the head. */
	if (prev) {
		head = prev->next;
		fp = skb_clone(head, GFP_ATOMIC);

		if (!fp)
			goto out_oom;

		fp->next = head->next;
		prev->next = fp;

467 468
		skb_morph(head, fq->q.fragments);
		head->next = fq->q.fragments->next;
469

470 471
		kfree_skb(fq->q.fragments);
		fq->q.fragments = head;
472 473
	}

L
Linus Torvalds 已提交
474 475 476 477
	BUG_TRAP(head != NULL);
	BUG_TRAP(FRAG6_CB(head)->offset == 0);

	/* Unfragmented part is taken from the first segment. */
478
	payload_len = ((head->data - skb_network_header(head)) -
479
		       sizeof(struct ipv6hdr) + fq->q.len -
480
		       sizeof(struct frag_hdr));
L
Linus Torvalds 已提交
481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507
	if (payload_len > IPV6_MAXPLEN)
		goto out_oversize;

	/* Head of list must not be cloned. */
	if (skb_cloned(head) && pskb_expand_head(head, 0, 0, GFP_ATOMIC))
		goto out_oom;

	/* If the first fragment is fragmented itself, we split
	 * it to two chunks: the first with data and paged part
	 * and the second, holding only fragments. */
	if (skb_shinfo(head)->frag_list) {
		struct sk_buff *clone;
		int i, plen = 0;

		if ((clone = alloc_skb(0, GFP_ATOMIC)) == NULL)
			goto out_oom;
		clone->next = head->next;
		head->next = clone;
		skb_shinfo(clone)->frag_list = skb_shinfo(head)->frag_list;
		skb_shinfo(head)->frag_list = NULL;
		for (i=0; i<skb_shinfo(head)->nr_frags; i++)
			plen += skb_shinfo(head)->frags[i].size;
		clone->len = clone->data_len = head->data_len - plen;
		head->data_len -= clone->len;
		head->len -= clone->len;
		clone->csum = 0;
		clone->ip_summed = head->ip_summed;
508
		atomic_add(clone->truesize, &fq->q.net->mem);
L
Linus Torvalds 已提交
509 510 511 512 513
	}

	/* We have to remove fragment header from datagram and to relocate
	 * header in order to calculate ICV correctly. */
	nhoff = fq->nhoffset;
514
	skb_network_header(head)[nhoff] = skb_transport_header(head)[0];
515
	memmove(head->head + sizeof(struct frag_hdr), head->head,
L
Linus Torvalds 已提交
516
		(head->data - head->head) - sizeof(struct frag_hdr));
517 518
	head->mac_header += sizeof(struct frag_hdr);
	head->network_header += sizeof(struct frag_hdr);
L
Linus Torvalds 已提交
519 520

	skb_shinfo(head)->frag_list = head->next;
521
	skb_reset_transport_header(head);
522
	skb_push(head, head->data - skb_network_header(head));
523
	atomic_sub(head->truesize, &fq->q.net->mem);
L
Linus Torvalds 已提交
524 525 526 527 528 529

	for (fp=head->next; fp; fp = fp->next) {
		head->data_len += fp->len;
		head->len += fp->len;
		if (head->ip_summed != fp->ip_summed)
			head->ip_summed = CHECKSUM_NONE;
530
		else if (head->ip_summed == CHECKSUM_COMPLETE)
L
Linus Torvalds 已提交
531 532
			head->csum = csum_add(head->csum, fp->csum);
		head->truesize += fp->truesize;
533
		atomic_sub(fp->truesize, &fq->q.net->mem);
L
Linus Torvalds 已提交
534 535 536 537
	}

	head->next = NULL;
	head->dev = dev;
538
	head->tstamp = fq->q.stamp;
539
	ipv6_hdr(head)->payload_len = htons(payload_len);
540
	IP6CB(head)->nhoff = nhoff;
L
Linus Torvalds 已提交
541 542

	/* Yes, and fold redundant checksum back. 8) */
543
	if (head->ip_summed == CHECKSUM_COMPLETE)
544
		head->csum = csum_partial(skb_network_header(head),
545
					  skb_network_header_len(head),
546
					  head->csum);
L
Linus Torvalds 已提交
547

548 549 550
	rcu_read_lock();
	IP6_INC_STATS_BH(__in6_dev_get(dev), IPSTATS_MIB_REASMOKS);
	rcu_read_unlock();
551
	fq->q.fragments = NULL;
L
Linus Torvalds 已提交
552 553 554 555 556 557 558 559 560 561
	return 1;

out_oversize:
	if (net_ratelimit())
		printk(KERN_DEBUG "ip6_frag_reasm: payload len = %d\n", payload_len);
	goto out_fail;
out_oom:
	if (net_ratelimit())
		printk(KERN_DEBUG "ip6_frag_reasm: no memory for reassembly\n");
out_fail:
562 563 564
	rcu_read_lock();
	IP6_INC_STATS_BH(__in6_dev_get(dev), IPSTATS_MIB_REASMFAILS);
	rcu_read_unlock();
L
Linus Torvalds 已提交
565 566 567
	return -1;
}

568
static int ipv6_frag_rcv(struct sk_buff *skb)
L
Linus Torvalds 已提交
569 570 571
{
	struct frag_hdr *fhdr;
	struct frag_queue *fq;
572
	struct ipv6hdr *hdr = ipv6_hdr(skb);
573
	struct net *net;
L
Linus Torvalds 已提交
574

575
	IP6_INC_STATS_BH(ip6_dst_idev(skb->dst), IPSTATS_MIB_REASMREQDS);
L
Linus Torvalds 已提交
576 577 578

	/* Jumbo payload inhibits frag. header */
	if (hdr->payload_len==0) {
579
		IP6_INC_STATS(ip6_dst_idev(skb->dst), IPSTATS_MIB_INHDRERRORS);
580 581
		icmpv6_param_prob(skb, ICMPV6_HDR_FIELD,
				  skb_network_header_len(skb));
L
Linus Torvalds 已提交
582 583
		return -1;
	}
584 585
	if (!pskb_may_pull(skb, (skb_transport_offset(skb) +
				 sizeof(struct frag_hdr)))) {
586
		IP6_INC_STATS(ip6_dst_idev(skb->dst), IPSTATS_MIB_INHDRERRORS);
587 588
		icmpv6_param_prob(skb, ICMPV6_HDR_FIELD,
				  skb_network_header_len(skb));
L
Linus Torvalds 已提交
589 590 591
		return -1;
	}

592
	hdr = ipv6_hdr(skb);
593
	fhdr = (struct frag_hdr *)skb_transport_header(skb);
L
Linus Torvalds 已提交
594 595 596

	if (!(fhdr->frag_off & htons(0xFFF9))) {
		/* It is not a fragmented frame */
597
		skb->transport_header += sizeof(struct frag_hdr);
598
		IP6_INC_STATS_BH(ip6_dst_idev(skb->dst), IPSTATS_MIB_REASMOKS);
L
Linus Torvalds 已提交
599

600
		IP6CB(skb)->nhoff = (u8 *)fhdr - skb_network_header(skb);
L
Linus Torvalds 已提交
601 602 603
		return 1;
	}

604
	net = dev_net(skb->dev);
605
	if (atomic_read(&net->ipv6.frags.mem) > net->ipv6.frags.high_thresh)
606
		ip6_evictor(net, ip6_dst_idev(skb->dst));
L
Linus Torvalds 已提交
607

608
	if ((fq = fq_find(net, fhdr->identification, &hdr->saddr, &hdr->daddr,
609
			  ip6_dst_idev(skb->dst))) != NULL) {
610
		int ret;
L
Linus Torvalds 已提交
611

612
		spin_lock(&fq->q.lock);
L
Linus Torvalds 已提交
613

614
		ret = ip6_frag_queue(fq, skb, fhdr, IP6CB(skb)->nhoff);
L
Linus Torvalds 已提交
615

616
		spin_unlock(&fq->q.lock);
617
		fq_put(fq);
L
Linus Torvalds 已提交
618 619 620
		return ret;
	}

621
	IP6_INC_STATS_BH(ip6_dst_idev(skb->dst), IPSTATS_MIB_REASMFAILS);
L
Linus Torvalds 已提交
622 623 624 625 626 627 628 629 630 631
	kfree_skb(skb);
	return -1;
}

static struct inet6_protocol frag_protocol =
{
	.handler	=	ipv6_frag_rcv,
	.flags		=	INET6_PROTO_NOPOLICY,
};

632
#ifdef CONFIG_SYSCTL
633
static struct ctl_table ip6_frags_ns_ctl_table[] = {
634 635 636
	{
		.ctl_name	= NET_IPV6_IP6FRAG_HIGH_THRESH,
		.procname	= "ip6frag_high_thresh",
637
		.data		= &init_net.ipv6.frags.high_thresh,
638 639 640 641 642 643 644
		.maxlen		= sizeof(int),
		.mode		= 0644,
		.proc_handler	= &proc_dointvec
	},
	{
		.ctl_name	= NET_IPV6_IP6FRAG_LOW_THRESH,
		.procname	= "ip6frag_low_thresh",
645
		.data		= &init_net.ipv6.frags.low_thresh,
646 647 648 649 650 651 652
		.maxlen		= sizeof(int),
		.mode		= 0644,
		.proc_handler	= &proc_dointvec
	},
	{
		.ctl_name	= NET_IPV6_IP6FRAG_TIME,
		.procname	= "ip6frag_time",
653
		.data		= &init_net.ipv6.frags.timeout,
654 655 656 657 658
		.maxlen		= sizeof(int),
		.mode		= 0644,
		.proc_handler	= &proc_dointvec_jiffies,
		.strategy	= &sysctl_jiffies,
	},
659 660 661 662
	{ }
};

static struct ctl_table ip6_frags_ctl_table[] = {
663 664 665
	{
		.ctl_name	= NET_IPV6_IP6FRAG_SECRET_INTERVAL,
		.procname	= "ip6frag_secret_interval",
666
		.data		= &ip6_frags.secret_interval,
667 668 669 670 671 672 673 674
		.maxlen		= sizeof(int),
		.mode		= 0644,
		.proc_handler	= &proc_dointvec_jiffies,
		.strategy	= &sysctl_jiffies
	},
	{ }
};

675
static int ip6_frags_ns_sysctl_register(struct net *net)
676
{
677
	struct ctl_table *table;
678 679
	struct ctl_table_header *hdr;

680
	table = ip6_frags_ns_ctl_table;
681
	if (net != &init_net) {
682
		table = kmemdup(table, sizeof(ip6_frags_ns_ctl_table), GFP_KERNEL);
683 684 685
		if (table == NULL)
			goto err_alloc;

686 687
		table[0].data = &net->ipv6.frags.high_thresh;
		table[1].data = &net->ipv6.frags.low_thresh;
688
		table[2].data = &net->ipv6.frags.timeout;
689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704
	}

	hdr = register_net_sysctl_table(net, net_ipv6_ctl_path, table);
	if (hdr == NULL)
		goto err_reg;

	net->ipv6.sysctl.frags_hdr = hdr;
	return 0;

err_reg:
	if (net != &init_net)
		kfree(table);
err_alloc:
	return -ENOMEM;
}

705
static void ip6_frags_ns_sysctl_unregister(struct net *net)
706 707 708 709 710 711
{
	struct ctl_table *table;

	table = net->ipv6.sysctl.frags_hdr->ctl_table_arg;
	unregister_net_sysctl_table(net->ipv6.sysctl.frags_hdr);
	kfree(table);
712
}
713 714 715 716 717 718 719 720 721 722 723 724 725 726

static struct ctl_table_header *ip6_ctl_header;

static int ip6_frags_sysctl_register(void)
{
	ip6_ctl_header = register_net_sysctl_rotable(net_ipv6_ctl_path,
			ip6_frags_ctl_table);
	return ip6_ctl_header == NULL ? -ENOMEM : 0;
}

static void ip6_frags_sysctl_unregister(void)
{
	unregister_net_sysctl_table(ip6_ctl_header);
}
727
#else
728
static inline int ip6_frags_ns_sysctl_register(struct net *net)
729
{
730 731
	return 0;
}
732

733
static inline void ip6_frags_ns_sysctl_unregister(struct net *net)
734 735
{
}
736 737 738 739 740 741 742 743 744

static inline int ip6_frags_sysctl_register(void)
{
	return 0;
}

static inline void ip6_frags_sysctl_unregister(void)
{
}
745
#endif
D
Daniel Lezcano 已提交
746

747 748
static int ipv6_frags_init_net(struct net *net)
{
749 750
	net->ipv6.frags.high_thresh = 256 * 1024;
	net->ipv6.frags.low_thresh = 192 * 1024;
751
	net->ipv6.frags.timeout = IPV6_FRAG_TIMEOUT;
752

753 754
	inet_frags_init_net(&net->ipv6.frags);

755
	return ip6_frags_ns_sysctl_register(net);
756 757
}

758 759
static void ipv6_frags_exit_net(struct net *net)
{
760
	ip6_frags_ns_sysctl_unregister(net);
761 762 763 764 765 766 767 768
	inet_frags_exit_net(&net->ipv6.frags, &ip6_frags);
}

static struct pernet_operations ip6_frags_ops = {
	.init = ipv6_frags_init_net,
	.exit = ipv6_frags_exit_net,
};

769
int __init ipv6_frag_init(void)
L
Linus Torvalds 已提交
770
{
771
	int ret;
L
Linus Torvalds 已提交
772

773 774 775
	ret = inet6_add_protocol(&frag_protocol, IPPROTO_FRAGMENT);
	if (ret)
		goto out;
776

777 778 779 780
	ret = ip6_frags_sysctl_register();
	if (ret)
		goto err_sysctl;

781 782 783
	ret = register_pernet_subsys(&ip6_frags_ops);
	if (ret)
		goto err_pernet;
784

785
	ip6_frags.hashfn = ip6_hashfn;
786
	ip6_frags.constructor = ip6_frag_init;
787
	ip6_frags.destructor = NULL;
788 789
	ip6_frags.skb_free = NULL;
	ip6_frags.qsize = sizeof(struct frag_queue);
790
	ip6_frags.match = ip6_frag_match;
791
	ip6_frags.frag_expire = ip6_frag_expire;
792
	ip6_frags.secret_interval = 10 * 60 * HZ;
793
	inet_frags_init(&ip6_frags);
794 795
out:
	return ret;
796 797

err_pernet:
798 799
	ip6_frags_sysctl_unregister();
err_sysctl:
800 801
	inet6_del_protocol(&frag_protocol, IPPROTO_FRAGMENT);
	goto out;
802 803 804 805 806
}

void ipv6_frag_exit(void)
{
	inet_frags_fini(&ip6_frags);
807
	ip6_frags_sysctl_unregister();
808
	unregister_pernet_subsys(&ip6_frags_ops);
809
	inet6_del_protocol(&frag_protocol, IPPROTO_FRAGMENT);
L
Linus Torvalds 已提交
810
}