xref: /openbmc/linux/net/ipv6/reassembly.c (revision fc08c258)
11da177e4SLinus Torvalds /*
21da177e4SLinus Torvalds  *	IPv6 fragment reassembly
31da177e4SLinus Torvalds  *	Linux INET6 implementation
41da177e4SLinus Torvalds  *
51da177e4SLinus Torvalds  *	Authors:
61da177e4SLinus Torvalds  *	Pedro Roque		<roque@di.fc.ul.pt>
71da177e4SLinus Torvalds  *
81da177e4SLinus Torvalds  *	Based on: net/ipv4/ip_fragment.c
91da177e4SLinus Torvalds  *
101da177e4SLinus Torvalds  *	This program is free software; you can redistribute it and/or
111da177e4SLinus Torvalds  *      modify it under the terms of the GNU General Public License
121da177e4SLinus Torvalds  *      as published by the Free Software Foundation; either version
131da177e4SLinus Torvalds  *      2 of the License, or (at your option) any later version.
141da177e4SLinus Torvalds  */
151da177e4SLinus Torvalds 
161da177e4SLinus Torvalds /*
171da177e4SLinus Torvalds  *	Fixes:
181da177e4SLinus Torvalds  *	Andi Kleen	Make it work with multiple hosts.
191da177e4SLinus Torvalds  *			More RFC compliance.
201da177e4SLinus Torvalds  *
211da177e4SLinus Torvalds  *      Horst von Brand Add missing #include <linux/string.h>
221da177e4SLinus Torvalds  *	Alexey Kuznetsov	SMP races, threading, cleanup.
231da177e4SLinus Torvalds  *	Patrick McHardy		LRU queue of frag heads for evictor.
241da177e4SLinus Torvalds  *	Mitsuru KANDA @USAGI	Register inet6_protocol{}.
251da177e4SLinus Torvalds  *	David Stevens and
261da177e4SLinus Torvalds  *	YOSHIFUJI,H. @USAGI	Always remove fragment header to
271da177e4SLinus Torvalds  *				calculate ICV correctly.
281da177e4SLinus Torvalds  */
295a3da1feSHannes Frederic Sowa 
305a3da1feSHannes Frederic Sowa #define pr_fmt(fmt) "IPv6: " fmt
315a3da1feSHannes Frederic Sowa 
321da177e4SLinus Torvalds #include <linux/errno.h>
331da177e4SLinus Torvalds #include <linux/types.h>
341da177e4SLinus Torvalds #include <linux/string.h>
351da177e4SLinus Torvalds #include <linux/socket.h>
361da177e4SLinus Torvalds #include <linux/sockios.h>
371da177e4SLinus Torvalds #include <linux/jiffies.h>
381da177e4SLinus Torvalds #include <linux/net.h>
391da177e4SLinus Torvalds #include <linux/list.h>
401da177e4SLinus Torvalds #include <linux/netdevice.h>
411da177e4SLinus Torvalds #include <linux/in6.h>
421da177e4SLinus Torvalds #include <linux/ipv6.h>
431da177e4SLinus Torvalds #include <linux/icmpv6.h>
441da177e4SLinus Torvalds #include <linux/random.h>
451da177e4SLinus Torvalds #include <linux/jhash.h>
46f61944efSHerbert Xu #include <linux/skbuff.h>
475a0e3ad6STejun Heo #include <linux/slab.h>
48bc3b2d7fSPaul Gortmaker #include <linux/export.h>
491da177e4SLinus Torvalds 
501da177e4SLinus Torvalds #include <net/sock.h>
511da177e4SLinus Torvalds #include <net/snmp.h>
521da177e4SLinus Torvalds 
531da177e4SLinus Torvalds #include <net/ipv6.h>
54a11d206dSYOSHIFUJI Hideaki #include <net/ip6_route.h>
551da177e4SLinus Torvalds #include <net/protocol.h>
561da177e4SLinus Torvalds #include <net/transp_v6.h>
571da177e4SLinus Torvalds #include <net/rawv6.h>
581da177e4SLinus Torvalds #include <net/ndisc.h>
591da177e4SLinus Torvalds #include <net/addrconf.h>
605ab11c98SPavel Emelyanov #include <net/inet_frag.h>
61eec2e618SHannes Frederic Sowa #include <net/inet_ecn.h>
621da177e4SLinus Torvalds 
63d4ad4d22SNikolay Aleksandrov static const char ip6_frag_cache_name[] = "ip6-frags";
64d4ad4d22SNikolay Aleksandrov 
65cc24becaSIan Morris struct ip6frag_skb_cb {
661da177e4SLinus Torvalds 	struct inet6_skb_parm	h;
671da177e4SLinus Torvalds 	int			offset;
681da177e4SLinus Torvalds };
691da177e4SLinus Torvalds 
701da177e4SLinus Torvalds #define FRAG6_CB(skb)	((struct ip6frag_skb_cb *)((skb)->cb))
711da177e4SLinus Torvalds 
72fc08c258SFabian Frederick static u8 ip6_frag_ecn(const struct ipv6hdr *ipv6h)
73eec2e618SHannes Frederic Sowa {
74eec2e618SHannes Frederic Sowa 	return 1 << (ipv6_get_dsfield(ipv6h) & INET_ECN_MASK);
75eec2e618SHannes Frederic Sowa }
761da177e4SLinus Torvalds 
777eb95156SPavel Emelyanov static struct inet_frags ip6_frags;
781da177e4SLinus Torvalds 
79f61944efSHerbert Xu static int ip6_frag_reasm(struct frag_queue *fq, struct sk_buff *prev,
80f61944efSHerbert Xu 			  struct net_device *dev);
81f61944efSHerbert Xu 
82f6596f9dSZach Brown /*
83f6596f9dSZach Brown  * callers should be careful not to use the hash value outside the ipfrag_lock
84f6596f9dSZach Brown  * as doing so could race with ipfrag_hash_rnd being recalculated.
85f6596f9dSZach Brown  */
86b1190570SHannes Frederic Sowa static unsigned int inet6_hash_frag(__be32 id, const struct in6_addr *saddr,
87b1190570SHannes Frederic Sowa 				    const struct in6_addr *daddr)
881da177e4SLinus Torvalds {
89b1190570SHannes Frederic Sowa 	net_get_random_once(&ip6_frags.rnd, sizeof(ip6_frags.rnd));
90fb3cfe6eSFlorian Westphal 	return jhash_3words(ipv6_addr_hash(saddr), ipv6_addr_hash(daddr),
91b1190570SHannes Frederic Sowa 			    (__force u32)id, ip6_frags.rnd);
921da177e4SLinus Torvalds }
931da177e4SLinus Torvalds 
9436c77782SFlorian Westphal static unsigned int ip6_hashfn(const struct inet_frag_queue *q)
951da177e4SLinus Torvalds {
9636c77782SFlorian Westphal 	const struct frag_queue *fq;
971da177e4SLinus Torvalds 
98321a3a99SPavel Emelyanov 	fq = container_of(q, struct frag_queue, q);
99b1190570SHannes Frederic Sowa 	return inet6_hash_frag(fq->id, &fq->saddr, &fq->daddr);
1001da177e4SLinus Torvalds }
1011da177e4SLinus Torvalds 
10236c77782SFlorian Westphal bool ip6_frag_match(const struct inet_frag_queue *q, const void *a)
103abd6523dSPavel Emelyanov {
10436c77782SFlorian Westphal 	const struct frag_queue *fq;
10536c77782SFlorian Westphal 	const struct ip6_create_arg *arg = a;
106abd6523dSPavel Emelyanov 
107abd6523dSPavel Emelyanov 	fq = container_of(q, struct frag_queue, q);
108cbc264caSEric Dumazet 	return	fq->id == arg->id &&
109cbc264caSEric Dumazet 		fq->user == arg->user &&
110abd6523dSPavel Emelyanov 		ipv6_addr_equal(&fq->saddr, arg->src) &&
111cbc264caSEric Dumazet 		ipv6_addr_equal(&fq->daddr, arg->dst);
112abd6523dSPavel Emelyanov }
113abd6523dSPavel Emelyanov EXPORT_SYMBOL(ip6_frag_match);
114abd6523dSPavel Emelyanov 
11536c77782SFlorian Westphal void ip6_frag_init(struct inet_frag_queue *q, const void *a)
116c6fda282SPavel Emelyanov {
117c6fda282SPavel Emelyanov 	struct frag_queue *fq = container_of(q, struct frag_queue, q);
11836c77782SFlorian Westphal 	const struct ip6_create_arg *arg = a;
119c6fda282SPavel Emelyanov 
120c6fda282SPavel Emelyanov 	fq->id = arg->id;
1210b5ccb2eSPatrick McHardy 	fq->user = arg->user;
1224e3fd7a0SAlexey Dobriyan 	fq->saddr = *arg->src;
1234e3fd7a0SAlexey Dobriyan 	fq->daddr = *arg->dst;
124eec2e618SHannes Frederic Sowa 	fq->ecn = arg->ecn;
125c6fda282SPavel Emelyanov }
126c6fda282SPavel Emelyanov EXPORT_SYMBOL(ip6_frag_init);
127c6fda282SPavel Emelyanov 
128b836c99fSAmerigo Wang void ip6_expire_frag_queue(struct net *net, struct frag_queue *fq,
129b836c99fSAmerigo Wang 			   struct inet_frags *frags)
1301da177e4SLinus Torvalds {
131a11d206dSYOSHIFUJI Hideaki 	struct net_device *dev = NULL;
132e521db9dSPavel Emelyanov 
1335ab11c98SPavel Emelyanov 	spin_lock(&fq->q.lock);
1341da177e4SLinus Torvalds 
13506aa8b8aSNikolay Aleksandrov 	if (fq->q.flags & INET_FRAG_COMPLETE)
1361da177e4SLinus Torvalds 		goto out;
1371da177e4SLinus Torvalds 
138b836c99fSAmerigo Wang 	inet_frag_kill(&fq->q, frags);
1391da177e4SLinus Torvalds 
140a11d206dSYOSHIFUJI Hideaki 	rcu_read_lock();
14169df9d59SEric Dumazet 	dev = dev_get_by_index_rcu(net, fq->iif);
14269df9d59SEric Dumazet 	if (!dev)
14369df9d59SEric Dumazet 		goto out_rcu_unlock;
14469df9d59SEric Dumazet 
145483a47d2SDenis V. Lunev 	IP6_INC_STATS_BH(net, __in6_dev_get(dev), IPSTATS_MIB_REASMFAILS);
1461da177e4SLinus Torvalds 
1472e404f63SNikolay Aleksandrov 	if (fq->q.flags & INET_FRAG_EVICTED)
1482e404f63SNikolay Aleksandrov 		goto out_rcu_unlock;
1492e404f63SNikolay Aleksandrov 
1502e404f63SNikolay Aleksandrov 	IP6_INC_STATS_BH(net, __in6_dev_get(dev), IPSTATS_MIB_REASMTIMEOUT);
1512e404f63SNikolay Aleksandrov 
15278c784c4SIngo Oeser 	/* Don't send error if the first segment did not arrive. */
15306aa8b8aSNikolay Aleksandrov 	if (!(fq->q.flags & INET_FRAG_FIRST_IN) || !fq->q.fragments)
15469df9d59SEric Dumazet 		goto out_rcu_unlock;
15578c784c4SIngo Oeser 
1562e404f63SNikolay Aleksandrov 	/* But use as source device on which LAST ARRIVED
1572e404f63SNikolay Aleksandrov 	 * segment was received. And do not use fq->dev
1582e404f63SNikolay Aleksandrov 	 * pointer directly, device might already disappeared.
1591da177e4SLinus Torvalds 	 */
1605ab11c98SPavel Emelyanov 	fq->q.fragments->dev = dev;
1613ffe533cSAlexey Dobriyan 	icmpv6_send(fq->q.fragments, ICMPV6_TIME_EXCEED, ICMPV6_EXC_FRAGTIME, 0);
16269df9d59SEric Dumazet out_rcu_unlock:
16369df9d59SEric Dumazet 	rcu_read_unlock();
1641da177e4SLinus Torvalds out:
1655ab11c98SPavel Emelyanov 	spin_unlock(&fq->q.lock);
166b836c99fSAmerigo Wang 	inet_frag_put(&fq->q, frags);
167b836c99fSAmerigo Wang }
168b836c99fSAmerigo Wang EXPORT_SYMBOL(ip6_expire_frag_queue);
169b836c99fSAmerigo Wang 
170b836c99fSAmerigo Wang static void ip6_frag_expire(unsigned long data)
171b836c99fSAmerigo Wang {
172b836c99fSAmerigo Wang 	struct frag_queue *fq;
173b836c99fSAmerigo Wang 	struct net *net;
174b836c99fSAmerigo Wang 
175b836c99fSAmerigo Wang 	fq = container_of((struct inet_frag_queue *)data, struct frag_queue, q);
176b836c99fSAmerigo Wang 	net = container_of(fq->q.net, struct net, ipv6.frags);
177b836c99fSAmerigo Wang 
178b836c99fSAmerigo Wang 	ip6_expire_frag_queue(net, fq, &ip6_frags);
1791da177e4SLinus Torvalds }
1801da177e4SLinus Torvalds 
181fc08c258SFabian Frederick static struct frag_queue *
182eec2e618SHannes Frederic Sowa fq_find(struct net *net, __be32 id, const struct in6_addr *src,
183eec2e618SHannes Frederic Sowa 	const struct in6_addr *dst, u8 ecn)
1841da177e4SLinus Torvalds {
185c6fda282SPavel Emelyanov 	struct inet_frag_queue *q;
186c6fda282SPavel Emelyanov 	struct ip6_create_arg arg;
187abd6523dSPavel Emelyanov 	unsigned int hash;
1881da177e4SLinus Torvalds 
189c6fda282SPavel Emelyanov 	arg.id = id;
1900b5ccb2eSPatrick McHardy 	arg.user = IP6_DEFRAG_LOCAL_DELIVER;
191c6fda282SPavel Emelyanov 	arg.src = src;
192c6fda282SPavel Emelyanov 	arg.dst = dst;
193eec2e618SHannes Frederic Sowa 	arg.ecn = ecn;
1949a375803SPavel Emelyanov 
195b1190570SHannes Frederic Sowa 	hash = inet6_hash_frag(id, src, dst);
196c6fda282SPavel Emelyanov 
197ac18e750SPavel Emelyanov 	q = inet_frag_find(&net->ipv6.frags, &ip6_frags, &arg, hash);
1985a3da1feSHannes Frederic Sowa 	if (IS_ERR_OR_NULL(q)) {
1995a3da1feSHannes Frederic Sowa 		inet_frag_maybe_warn_overflow(q, pr_fmt());
2009546377cSShan Wei 		return NULL;
2015a3da1feSHannes Frederic Sowa 	}
202c6fda282SPavel Emelyanov 	return container_of(q, struct frag_queue, q);
2031da177e4SLinus Torvalds }
2041da177e4SLinus Torvalds 
205f61944efSHerbert Xu static int ip6_frag_queue(struct frag_queue *fq, struct sk_buff *skb,
2061da177e4SLinus Torvalds 			   struct frag_hdr *fhdr, int nhoff)
2071da177e4SLinus Torvalds {
2081da177e4SLinus Torvalds 	struct sk_buff *prev, *next;
209f61944efSHerbert Xu 	struct net_device *dev;
2101da177e4SLinus Torvalds 	int offset, end;
211adf30907SEric Dumazet 	struct net *net = dev_net(skb_dst(skb)->dev);
212eec2e618SHannes Frederic Sowa 	u8 ecn;
2131da177e4SLinus Torvalds 
21406aa8b8aSNikolay Aleksandrov 	if (fq->q.flags & INET_FRAG_COMPLETE)
2151da177e4SLinus Torvalds 		goto err;
2161da177e4SLinus Torvalds 
2171da177e4SLinus Torvalds 	offset = ntohs(fhdr->frag_off) & ~0x7;
2180660e03fSArnaldo Carvalho de Melo 	end = offset + (ntohs(ipv6_hdr(skb)->payload_len) -
2190660e03fSArnaldo Carvalho de Melo 			((u8 *)(fhdr + 1) - (u8 *)(ipv6_hdr(skb) + 1)));
2201da177e4SLinus Torvalds 
2211da177e4SLinus Torvalds 	if ((unsigned int)end > IPV6_MAXPLEN) {
222adf30907SEric Dumazet 		IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)),
223a11d206dSYOSHIFUJI Hideaki 				 IPSTATS_MIB_INHDRERRORS);
224d56f90a7SArnaldo Carvalho de Melo 		icmpv6_param_prob(skb, ICMPV6_HDR_FIELD,
225d56f90a7SArnaldo Carvalho de Melo 				  ((u8 *)&fhdr->frag_off -
226d56f90a7SArnaldo Carvalho de Melo 				   skb_network_header(skb)));
227f61944efSHerbert Xu 		return -1;
2281da177e4SLinus Torvalds 	}
2291da177e4SLinus Torvalds 
230eec2e618SHannes Frederic Sowa 	ecn = ip6_frag_ecn(ipv6_hdr(skb));
231eec2e618SHannes Frederic Sowa 
232d56f90a7SArnaldo Carvalho de Melo 	if (skb->ip_summed == CHECKSUM_COMPLETE) {
233d56f90a7SArnaldo Carvalho de Melo 		const unsigned char *nh = skb_network_header(skb);
2341da177e4SLinus Torvalds 		skb->csum = csum_sub(skb->csum,
235d56f90a7SArnaldo Carvalho de Melo 				     csum_partial(nh, (u8 *)(fhdr + 1) - nh,
236d56f90a7SArnaldo Carvalho de Melo 						  0));
237d56f90a7SArnaldo Carvalho de Melo 	}
2381da177e4SLinus Torvalds 
2391da177e4SLinus Torvalds 	/* Is this the final fragment? */
2401da177e4SLinus Torvalds 	if (!(fhdr->frag_off & htons(IP6_MF))) {
2411da177e4SLinus Torvalds 		/* If we already have some bits beyond end
2421da177e4SLinus Torvalds 		 * or have different end, the segment is corrupted.
2431da177e4SLinus Torvalds 		 */
2445ab11c98SPavel Emelyanov 		if (end < fq->q.len ||
24506aa8b8aSNikolay Aleksandrov 		    ((fq->q.flags & INET_FRAG_LAST_IN) && end != fq->q.len))
2461da177e4SLinus Torvalds 			goto err;
24706aa8b8aSNikolay Aleksandrov 		fq->q.flags |= INET_FRAG_LAST_IN;
2485ab11c98SPavel Emelyanov 		fq->q.len = end;
2491da177e4SLinus Torvalds 	} else {
2501da177e4SLinus Torvalds 		/* Check if the fragment is rounded to 8 bytes.
2511da177e4SLinus Torvalds 		 * Required by the RFC.
2521da177e4SLinus Torvalds 		 */
2531da177e4SLinus Torvalds 		if (end & 0x7) {
2541da177e4SLinus Torvalds 			/* RFC2460 says always send parameter problem in
2551da177e4SLinus Torvalds 			 * this case. -DaveM
2561da177e4SLinus Torvalds 			 */
257adf30907SEric Dumazet 			IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)),
258a11d206dSYOSHIFUJI Hideaki 					 IPSTATS_MIB_INHDRERRORS);
2591da177e4SLinus Torvalds 			icmpv6_param_prob(skb, ICMPV6_HDR_FIELD,
2601da177e4SLinus Torvalds 					  offsetof(struct ipv6hdr, payload_len));
261f61944efSHerbert Xu 			return -1;
2621da177e4SLinus Torvalds 		}
2635ab11c98SPavel Emelyanov 		if (end > fq->q.len) {
2641da177e4SLinus Torvalds 			/* Some bits beyond end -> corruption. */
26506aa8b8aSNikolay Aleksandrov 			if (fq->q.flags & INET_FRAG_LAST_IN)
2661da177e4SLinus Torvalds 				goto err;
2675ab11c98SPavel Emelyanov 			fq->q.len = end;
2681da177e4SLinus Torvalds 		}
2691da177e4SLinus Torvalds 	}
2701da177e4SLinus Torvalds 
2711da177e4SLinus Torvalds 	if (end == offset)
2721da177e4SLinus Torvalds 		goto err;
2731da177e4SLinus Torvalds 
2741da177e4SLinus Torvalds 	/* Point into the IP datagram 'data' part. */
2751da177e4SLinus Torvalds 	if (!pskb_pull(skb, (u8 *) (fhdr + 1) - skb->data))
2761da177e4SLinus Torvalds 		goto err;
27742ca89c1SStephen Hemminger 
27842ca89c1SStephen Hemminger 	if (pskb_trim_rcsum(skb, end - offset))
2791da177e4SLinus Torvalds 		goto err;
2801da177e4SLinus Torvalds 
2811da177e4SLinus Torvalds 	/* Find out which fragments are in front and at the back of us
2821da177e4SLinus Torvalds 	 * in the chain of fragments so far.  We must know where to put
2831da177e4SLinus Torvalds 	 * this fragment, right?
2841da177e4SLinus Torvalds 	 */
285d6bebca9SChangli Gao 	prev = fq->q.fragments_tail;
286d6bebca9SChangli Gao 	if (!prev || FRAG6_CB(prev)->offset < offset) {
287d6bebca9SChangli Gao 		next = NULL;
288d6bebca9SChangli Gao 		goto found;
289d6bebca9SChangli Gao 	}
2901da177e4SLinus Torvalds 	prev = NULL;
2915ab11c98SPavel Emelyanov 	for (next = fq->q.fragments; next != NULL; next = next->next) {
2921da177e4SLinus Torvalds 		if (FRAG6_CB(next)->offset >= offset)
2931da177e4SLinus Torvalds 			break;	/* bingo! */
2941da177e4SLinus Torvalds 		prev = next;
2951da177e4SLinus Torvalds 	}
2961da177e4SLinus Torvalds 
297d6bebca9SChangli Gao found:
2985de658f8SEric Dumazet 	/* RFC5722, Section 4, amended by Errata ID : 3089
29970789d70SNicolas Dichtel 	 *                          When reassembling an IPv6 datagram, if
30070789d70SNicolas Dichtel 	 *   one or more its constituent fragments is determined to be an
30170789d70SNicolas Dichtel 	 *   overlapping fragment, the entire datagram (and any constituent
3025de658f8SEric Dumazet 	 *   fragments) MUST be silently discarded.
3031da177e4SLinus Torvalds 	 */
3041da177e4SLinus Torvalds 
30570789d70SNicolas Dichtel 	/* Check for overlap with preceding fragment. */
30670789d70SNicolas Dichtel 	if (prev &&
307f4642141SShan Wei 	    (FRAG6_CB(prev)->offset + prev->len) > offset)
30870789d70SNicolas Dichtel 		goto discard_fq;
3091da177e4SLinus Torvalds 
31070789d70SNicolas Dichtel 	/* Look for overlap with succeeding segment. */
31170789d70SNicolas Dichtel 	if (next && FRAG6_CB(next)->offset < end)
31270789d70SNicolas Dichtel 		goto discard_fq;
3131da177e4SLinus Torvalds 
3141da177e4SLinus Torvalds 	FRAG6_CB(skb)->offset = offset;
3151da177e4SLinus Torvalds 
3161da177e4SLinus Torvalds 	/* Insert this fragment in the chain of fragments. */
3171da177e4SLinus Torvalds 	skb->next = next;
318d6bebca9SChangli Gao 	if (!next)
319d6bebca9SChangli Gao 		fq->q.fragments_tail = skb;
3201da177e4SLinus Torvalds 	if (prev)
3211da177e4SLinus Torvalds 		prev->next = skb;
3221da177e4SLinus Torvalds 	else
3235ab11c98SPavel Emelyanov 		fq->q.fragments = skb;
3241da177e4SLinus Torvalds 
325f61944efSHerbert Xu 	dev = skb->dev;
326f61944efSHerbert Xu 	if (dev) {
327f61944efSHerbert Xu 		fq->iif = dev->ifindex;
3281da177e4SLinus Torvalds 		skb->dev = NULL;
329f61944efSHerbert Xu 	}
3305ab11c98SPavel Emelyanov 	fq->q.stamp = skb->tstamp;
3315ab11c98SPavel Emelyanov 	fq->q.meat += skb->len;
332eec2e618SHannes Frederic Sowa 	fq->ecn |= ecn;
333d433673eSJesper Dangaard Brouer 	add_frag_mem_limit(&fq->q, skb->truesize);
3341da177e4SLinus Torvalds 
3351da177e4SLinus Torvalds 	/* The first fragment.
3361da177e4SLinus Torvalds 	 * nhoffset is obtained from the first fragment, of course.
3371da177e4SLinus Torvalds 	 */
3381da177e4SLinus Torvalds 	if (offset == 0) {
3391da177e4SLinus Torvalds 		fq->nhoffset = nhoff;
34006aa8b8aSNikolay Aleksandrov 		fq->q.flags |= INET_FRAG_FIRST_IN;
3411da177e4SLinus Torvalds 	}
342f61944efSHerbert Xu 
34306aa8b8aSNikolay Aleksandrov 	if (fq->q.flags == (INET_FRAG_FIRST_IN | INET_FRAG_LAST_IN) &&
34497599dc7SEric Dumazet 	    fq->q.meat == fq->q.len) {
34597599dc7SEric Dumazet 		int res;
34697599dc7SEric Dumazet 		unsigned long orefdst = skb->_skb_refdst;
347f61944efSHerbert Xu 
34897599dc7SEric Dumazet 		skb->_skb_refdst = 0UL;
34997599dc7SEric Dumazet 		res = ip6_frag_reasm(fq, prev, dev);
35097599dc7SEric Dumazet 		skb->_skb_refdst = orefdst;
35197599dc7SEric Dumazet 		return res;
35297599dc7SEric Dumazet 	}
35397599dc7SEric Dumazet 
35497599dc7SEric Dumazet 	skb_dst_drop(skb);
355f61944efSHerbert Xu 	return -1;
3561da177e4SLinus Torvalds 
35770789d70SNicolas Dichtel discard_fq:
358b836c99fSAmerigo Wang 	inet_frag_kill(&fq->q, &ip6_frags);
3591da177e4SLinus Torvalds err:
360d2373862SNikolay Aleksandrov 	IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)),
3613bd653c8SDenis V. Lunev 			 IPSTATS_MIB_REASMFAILS);
3621da177e4SLinus Torvalds 	kfree_skb(skb);
363f61944efSHerbert Xu 	return -1;
3641da177e4SLinus Torvalds }
3651da177e4SLinus Torvalds 
3661da177e4SLinus Torvalds /*
3671da177e4SLinus Torvalds  *	Check if this packet is complete.
3681da177e4SLinus Torvalds  *	Returns NULL on failure by any reason, and pointer
3691da177e4SLinus Torvalds  *	to current nexthdr field in reassembled frame.
3701da177e4SLinus Torvalds  *
3711da177e4SLinus Torvalds  *	It is called with locked fq, and caller must check that
3721da177e4SLinus Torvalds  *	queue is eligible for reassembly i.e. it is not COMPLETE,
3731da177e4SLinus Torvalds  *	the last and the first frames arrived and all the bits are here.
3741da177e4SLinus Torvalds  */
375f61944efSHerbert Xu static int ip6_frag_reasm(struct frag_queue *fq, struct sk_buff *prev,
3761da177e4SLinus Torvalds 			  struct net_device *dev)
3771da177e4SLinus Torvalds {
3782bad35b7SJorge Boncompte [DTI2] 	struct net *net = container_of(fq->q.net, struct net, ipv6.frags);
3795ab11c98SPavel Emelyanov 	struct sk_buff *fp, *head = fq->q.fragments;
3801da177e4SLinus Torvalds 	int    payload_len;
3811da177e4SLinus Torvalds 	unsigned int nhoff;
382ec16439eSEric Dumazet 	int sum_truesize;
383eec2e618SHannes Frederic Sowa 	u8 ecn;
3841da177e4SLinus Torvalds 
385b836c99fSAmerigo Wang 	inet_frag_kill(&fq->q, &ip6_frags);
3861da177e4SLinus Torvalds 
387eec2e618SHannes Frederic Sowa 	ecn = ip_frag_ecn_table[fq->ecn];
388eec2e618SHannes Frederic Sowa 	if (unlikely(ecn == 0xff))
389eec2e618SHannes Frederic Sowa 		goto out_fail;
390eec2e618SHannes Frederic Sowa 
391f61944efSHerbert Xu 	/* Make the one we just received the head. */
392f61944efSHerbert Xu 	if (prev) {
393f61944efSHerbert Xu 		head = prev->next;
394f61944efSHerbert Xu 		fp = skb_clone(head, GFP_ATOMIC);
395f61944efSHerbert Xu 
396f61944efSHerbert Xu 		if (!fp)
397f61944efSHerbert Xu 			goto out_oom;
398f61944efSHerbert Xu 
399f61944efSHerbert Xu 		fp->next = head->next;
400d6bebca9SChangli Gao 		if (!fp->next)
401d6bebca9SChangli Gao 			fq->q.fragments_tail = fp;
402f61944efSHerbert Xu 		prev->next = fp;
403f61944efSHerbert Xu 
4045ab11c98SPavel Emelyanov 		skb_morph(head, fq->q.fragments);
4055ab11c98SPavel Emelyanov 		head->next = fq->q.fragments->next;
406f61944efSHerbert Xu 
407808db80aSEric Dumazet 		consume_skb(fq->q.fragments);
4085ab11c98SPavel Emelyanov 		fq->q.fragments = head;
409f61944efSHerbert Xu 	}
410f61944efSHerbert Xu 
411547b792cSIlpo Järvinen 	WARN_ON(head == NULL);
412547b792cSIlpo Järvinen 	WARN_ON(FRAG6_CB(head)->offset != 0);
4131da177e4SLinus Torvalds 
4141da177e4SLinus Torvalds 	/* Unfragmented part is taken from the first segment. */
415d56f90a7SArnaldo Carvalho de Melo 	payload_len = ((head->data - skb_network_header(head)) -
4165ab11c98SPavel Emelyanov 		       sizeof(struct ipv6hdr) + fq->q.len -
417d56f90a7SArnaldo Carvalho de Melo 		       sizeof(struct frag_hdr));
4181da177e4SLinus Torvalds 	if (payload_len > IPV6_MAXPLEN)
4191da177e4SLinus Torvalds 		goto out_oversize;
4201da177e4SLinus Torvalds 
4211da177e4SLinus Torvalds 	/* Head of list must not be cloned. */
42214bbd6a5SPravin B Shelar 	if (skb_unclone(head, GFP_ATOMIC))
4231da177e4SLinus Torvalds 		goto out_oom;
4241da177e4SLinus Torvalds 
4251da177e4SLinus Torvalds 	/* If the first fragment is fragmented itself, we split
4261da177e4SLinus Torvalds 	 * it to two chunks: the first with data and paged part
4271da177e4SLinus Torvalds 	 * and the second, holding only fragments. */
42821dc3301SDavid S. Miller 	if (skb_has_frag_list(head)) {
4291da177e4SLinus Torvalds 		struct sk_buff *clone;
4301da177e4SLinus Torvalds 		int i, plen = 0;
4311da177e4SLinus Torvalds 
4321da177e4SLinus Torvalds 		if ((clone = alloc_skb(0, GFP_ATOMIC)) == NULL)
4331da177e4SLinus Torvalds 			goto out_oom;
4341da177e4SLinus Torvalds 		clone->next = head->next;
4351da177e4SLinus Torvalds 		head->next = clone;
4361da177e4SLinus Torvalds 		skb_shinfo(clone)->frag_list = skb_shinfo(head)->frag_list;
4374d9092bbSDavid S. Miller 		skb_frag_list_init(head);
4381da177e4SLinus Torvalds 		for (i = 0; i < skb_shinfo(head)->nr_frags; i++)
4399e903e08SEric Dumazet 			plen += skb_frag_size(&skb_shinfo(head)->frags[i]);
4401da177e4SLinus Torvalds 		clone->len = clone->data_len = head->data_len - plen;
4411da177e4SLinus Torvalds 		head->data_len -= clone->len;
4421da177e4SLinus Torvalds 		head->len -= clone->len;
4431da177e4SLinus Torvalds 		clone->csum = 0;
4441da177e4SLinus Torvalds 		clone->ip_summed = head->ip_summed;
445d433673eSJesper Dangaard Brouer 		add_frag_mem_limit(&fq->q, clone->truesize);
4461da177e4SLinus Torvalds 	}
4471da177e4SLinus Torvalds 
4481da177e4SLinus Torvalds 	/* We have to remove fragment header from datagram and to relocate
4491da177e4SLinus Torvalds 	 * header in order to calculate ICV correctly. */
4501da177e4SLinus Torvalds 	nhoff = fq->nhoffset;
451b0e380b1SArnaldo Carvalho de Melo 	skb_network_header(head)[nhoff] = skb_transport_header(head)[0];
4521da177e4SLinus Torvalds 	memmove(head->head + sizeof(struct frag_hdr), head->head,
4531da177e4SLinus Torvalds 		(head->data - head->head) - sizeof(struct frag_hdr));
454b0e380b1SArnaldo Carvalho de Melo 	head->mac_header += sizeof(struct frag_hdr);
455b0e380b1SArnaldo Carvalho de Melo 	head->network_header += sizeof(struct frag_hdr);
4561da177e4SLinus Torvalds 
457badff6d0SArnaldo Carvalho de Melo 	skb_reset_transport_header(head);
458d56f90a7SArnaldo Carvalho de Melo 	skb_push(head, head->data - skb_network_header(head));
4591da177e4SLinus Torvalds 
460ec16439eSEric Dumazet 	sum_truesize = head->truesize;
461ec16439eSEric Dumazet 	for (fp = head->next; fp;) {
462ec16439eSEric Dumazet 		bool headstolen;
463ec16439eSEric Dumazet 		int delta;
464ec16439eSEric Dumazet 		struct sk_buff *next = fp->next;
465ec16439eSEric Dumazet 
466ec16439eSEric Dumazet 		sum_truesize += fp->truesize;
4671da177e4SLinus Torvalds 		if (head->ip_summed != fp->ip_summed)
4681da177e4SLinus Torvalds 			head->ip_summed = CHECKSUM_NONE;
46984fa7933SPatrick McHardy 		else if (head->ip_summed == CHECKSUM_COMPLETE)
4701da177e4SLinus Torvalds 			head->csum = csum_add(head->csum, fp->csum);
471ec16439eSEric Dumazet 
472ec16439eSEric Dumazet 		if (skb_try_coalesce(head, fp, &headstolen, &delta)) {
473ec16439eSEric Dumazet 			kfree_skb_partial(fp, headstolen);
474ec16439eSEric Dumazet 		} else {
475ec16439eSEric Dumazet 			if (!skb_shinfo(head)->frag_list)
476ec16439eSEric Dumazet 				skb_shinfo(head)->frag_list = fp;
477ec16439eSEric Dumazet 			head->data_len += fp->len;
478ec16439eSEric Dumazet 			head->len += fp->len;
4791da177e4SLinus Torvalds 			head->truesize += fp->truesize;
4801da177e4SLinus Torvalds 		}
481ec16439eSEric Dumazet 		fp = next;
482ec16439eSEric Dumazet 	}
483d433673eSJesper Dangaard Brouer 	sub_frag_mem_limit(&fq->q, sum_truesize);
4841da177e4SLinus Torvalds 
4851da177e4SLinus Torvalds 	head->next = NULL;
4861da177e4SLinus Torvalds 	head->dev = dev;
4875ab11c98SPavel Emelyanov 	head->tstamp = fq->q.stamp;
4880660e03fSArnaldo Carvalho de Melo 	ipv6_hdr(head)->payload_len = htons(payload_len);
489eec2e618SHannes Frederic Sowa 	ipv6_change_dsfield(ipv6_hdr(head), 0xff, ecn);
490951dbc8aSPatrick McHardy 	IP6CB(head)->nhoff = nhoff;
491f46078cfSHannes Frederic Sowa 	IP6CB(head)->flags |= IP6SKB_FRAGMENTED;
4921da177e4SLinus Torvalds 
4931da177e4SLinus Torvalds 	/* Yes, and fold redundant checksum back. 8) */
49484fa7933SPatrick McHardy 	if (head->ip_summed == CHECKSUM_COMPLETE)
495d56f90a7SArnaldo Carvalho de Melo 		head->csum = csum_partial(skb_network_header(head),
496cfe1fc77SArnaldo Carvalho de Melo 					  skb_network_header_len(head),
497d56f90a7SArnaldo Carvalho de Melo 					  head->csum);
4981da177e4SLinus Torvalds 
499a11d206dSYOSHIFUJI Hideaki 	rcu_read_lock();
5002bad35b7SJorge Boncompte [DTI2] 	IP6_INC_STATS_BH(net, __in6_dev_get(dev), IPSTATS_MIB_REASMOKS);
501a11d206dSYOSHIFUJI Hideaki 	rcu_read_unlock();
5025ab11c98SPavel Emelyanov 	fq->q.fragments = NULL;
503d6bebca9SChangli Gao 	fq->q.fragments_tail = NULL;
5041da177e4SLinus Torvalds 	return 1;
5051da177e4SLinus Torvalds 
5061da177e4SLinus Torvalds out_oversize:
507e87cc472SJoe Perches 	net_dbg_ratelimited("ip6_frag_reasm: payload len = %d\n", payload_len);
5081da177e4SLinus Torvalds 	goto out_fail;
5091da177e4SLinus Torvalds out_oom:
510e87cc472SJoe Perches 	net_dbg_ratelimited("ip6_frag_reasm: no memory for reassembly\n");
5111da177e4SLinus Torvalds out_fail:
512a11d206dSYOSHIFUJI Hideaki 	rcu_read_lock();
5132bad35b7SJorge Boncompte [DTI2] 	IP6_INC_STATS_BH(net, __in6_dev_get(dev), IPSTATS_MIB_REASMFAILS);
514a11d206dSYOSHIFUJI Hideaki 	rcu_read_unlock();
5151da177e4SLinus Torvalds 	return -1;
5161da177e4SLinus Torvalds }
5171da177e4SLinus Torvalds 
518e5bbef20SHerbert Xu static int ipv6_frag_rcv(struct sk_buff *skb)
5191da177e4SLinus Torvalds {
5201da177e4SLinus Torvalds 	struct frag_hdr *fhdr;
5211da177e4SLinus Torvalds 	struct frag_queue *fq;
522b71d1d42SEric Dumazet 	const struct ipv6hdr *hdr = ipv6_hdr(skb);
523adf30907SEric Dumazet 	struct net *net = dev_net(skb_dst(skb)->dev);
5241da177e4SLinus Torvalds 
525f46078cfSHannes Frederic Sowa 	if (IP6CB(skb)->flags & IP6SKB_FRAGMENTED)
526f46078cfSHannes Frederic Sowa 		goto fail_hdr;
527f46078cfSHannes Frederic Sowa 
528adf30907SEric Dumazet 	IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)), IPSTATS_MIB_REASMREQDS);
5291da177e4SLinus Torvalds 
5301da177e4SLinus Torvalds 	/* Jumbo payload inhibits frag. header */
53198b3377cSDenis V. Lunev 	if (hdr->payload_len == 0)
53298b3377cSDenis V. Lunev 		goto fail_hdr;
53398b3377cSDenis V. Lunev 
534ea2ae17dSArnaldo Carvalho de Melo 	if (!pskb_may_pull(skb, (skb_transport_offset(skb) +
53598b3377cSDenis V. Lunev 				 sizeof(struct frag_hdr))))
53698b3377cSDenis V. Lunev 		goto fail_hdr;
5371da177e4SLinus Torvalds 
5380660e03fSArnaldo Carvalho de Melo 	hdr = ipv6_hdr(skb);
5399c70220bSArnaldo Carvalho de Melo 	fhdr = (struct frag_hdr *)skb_transport_header(skb);
5401da177e4SLinus Torvalds 
5411da177e4SLinus Torvalds 	if (!(fhdr->frag_off & htons(0xFFF9))) {
5421da177e4SLinus Torvalds 		/* It is not a fragmented frame */
543b0e380b1SArnaldo Carvalho de Melo 		skb->transport_header += sizeof(struct frag_hdr);
544483a47d2SDenis V. Lunev 		IP6_INC_STATS_BH(net,
545adf30907SEric Dumazet 				 ip6_dst_idev(skb_dst(skb)), IPSTATS_MIB_REASMOKS);
5461da177e4SLinus Torvalds 
547d56f90a7SArnaldo Carvalho de Melo 		IP6CB(skb)->nhoff = (u8 *)fhdr - skb_network_header(skb);
548f46078cfSHannes Frederic Sowa 		IP6CB(skb)->flags |= IP6SKB_FRAGMENTED;
5491da177e4SLinus Torvalds 		return 1;
5501da177e4SLinus Torvalds 	}
5511da177e4SLinus Torvalds 
552eec2e618SHannes Frederic Sowa 	fq = fq_find(net, fhdr->identification, &hdr->saddr, &hdr->daddr,
553eec2e618SHannes Frederic Sowa 		     ip6_frag_ecn(hdr));
5549546377cSShan Wei 	if (fq != NULL) {
555f61944efSHerbert Xu 		int ret;
5561da177e4SLinus Torvalds 
5575ab11c98SPavel Emelyanov 		spin_lock(&fq->q.lock);
5581da177e4SLinus Torvalds 
559f61944efSHerbert Xu 		ret = ip6_frag_queue(fq, skb, fhdr, IP6CB(skb)->nhoff);
5601da177e4SLinus Torvalds 
5615ab11c98SPavel Emelyanov 		spin_unlock(&fq->q.lock);
562b836c99fSAmerigo Wang 		inet_frag_put(&fq->q, &ip6_frags);
5631da177e4SLinus Torvalds 		return ret;
5641da177e4SLinus Torvalds 	}
5651da177e4SLinus Torvalds 
566adf30907SEric Dumazet 	IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)), IPSTATS_MIB_REASMFAILS);
5671da177e4SLinus Torvalds 	kfree_skb(skb);
5681da177e4SLinus Torvalds 	return -1;
56998b3377cSDenis V. Lunev 
57098b3377cSDenis V. Lunev fail_hdr:
571d2373862SNikolay Aleksandrov 	IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)),
572d2373862SNikolay Aleksandrov 			 IPSTATS_MIB_INHDRERRORS);
57398b3377cSDenis V. Lunev 	icmpv6_param_prob(skb, ICMPV6_HDR_FIELD, skb_network_header_len(skb));
57498b3377cSDenis V. Lunev 	return -1;
5751da177e4SLinus Torvalds }
5761da177e4SLinus Torvalds 
577cc24becaSIan Morris static const struct inet6_protocol frag_protocol = {
5781da177e4SLinus Torvalds 	.handler	=	ipv6_frag_rcv,
5791da177e4SLinus Torvalds 	.flags		=	INET6_PROTO_NOPOLICY,
5801da177e4SLinus Torvalds };
5811da177e4SLinus Torvalds 
5828d8354d2SPavel Emelyanov #ifdef CONFIG_SYSCTL
5831bab4c75SNikolay Aleksandrov static int zero;
5841bab4c75SNikolay Aleksandrov 
5850a64b4b8SPavel Emelyanov static struct ctl_table ip6_frags_ns_ctl_table[] = {
586e71e0349SDaniel Lezcano 	{
5878d8354d2SPavel Emelyanov 		.procname	= "ip6frag_high_thresh",
588e31e0bdcSPavel Emelyanov 		.data		= &init_net.ipv6.frags.high_thresh,
5898d8354d2SPavel Emelyanov 		.maxlen		= sizeof(int),
5908d8354d2SPavel Emelyanov 		.mode		= 0644,
5911bab4c75SNikolay Aleksandrov 		.proc_handler	= proc_dointvec_minmax,
5921bab4c75SNikolay Aleksandrov 		.extra1		= &init_net.ipv6.frags.low_thresh
5938d8354d2SPavel Emelyanov 	},
5948d8354d2SPavel Emelyanov 	{
5958d8354d2SPavel Emelyanov 		.procname	= "ip6frag_low_thresh",
596e31e0bdcSPavel Emelyanov 		.data		= &init_net.ipv6.frags.low_thresh,
5978d8354d2SPavel Emelyanov 		.maxlen		= sizeof(int),
5988d8354d2SPavel Emelyanov 		.mode		= 0644,
5991bab4c75SNikolay Aleksandrov 		.proc_handler	= proc_dointvec_minmax,
6001bab4c75SNikolay Aleksandrov 		.extra1		= &zero,
6011bab4c75SNikolay Aleksandrov 		.extra2		= &init_net.ipv6.frags.high_thresh
6028d8354d2SPavel Emelyanov 	},
6038d8354d2SPavel Emelyanov 	{
6048d8354d2SPavel Emelyanov 		.procname	= "ip6frag_time",
605b2fd5321SPavel Emelyanov 		.data		= &init_net.ipv6.frags.timeout,
6068d8354d2SPavel Emelyanov 		.maxlen		= sizeof(int),
6078d8354d2SPavel Emelyanov 		.mode		= 0644,
6086d9f239aSAlexey Dobriyan 		.proc_handler	= proc_dointvec_jiffies,
6098d8354d2SPavel Emelyanov 	},
6107d291ebbSPavel Emelyanov 	{ }
6117d291ebbSPavel Emelyanov };
6127d291ebbSPavel Emelyanov 
613e3a57d18SFlorian Westphal /* secret interval has been deprecated */
614e3a57d18SFlorian Westphal static int ip6_frags_secret_interval_unused;
6157d291ebbSPavel Emelyanov static struct ctl_table ip6_frags_ctl_table[] = {
6168d8354d2SPavel Emelyanov 	{
6178d8354d2SPavel Emelyanov 		.procname	= "ip6frag_secret_interval",
618e3a57d18SFlorian Westphal 		.data		= &ip6_frags_secret_interval_unused,
6198d8354d2SPavel Emelyanov 		.maxlen		= sizeof(int),
6208d8354d2SPavel Emelyanov 		.mode		= 0644,
6216d9f239aSAlexey Dobriyan 		.proc_handler	= proc_dointvec_jiffies,
6228d8354d2SPavel Emelyanov 	},
6238d8354d2SPavel Emelyanov 	{ }
6248d8354d2SPavel Emelyanov };
6257d460db9SDaniel Lezcano 
6262c8c1e72SAlexey Dobriyan static int __net_init ip6_frags_ns_sysctl_register(struct net *net)
6278d8354d2SPavel Emelyanov {
628e4a2d5c2SPavel Emelyanov 	struct ctl_table *table;
6298d8354d2SPavel Emelyanov 	struct ctl_table_header *hdr;
6308d8354d2SPavel Emelyanov 
6310a64b4b8SPavel Emelyanov 	table = ip6_frags_ns_ctl_table;
63209ad9bc7SOctavian Purdila 	if (!net_eq(net, &init_net)) {
6330a64b4b8SPavel Emelyanov 		table = kmemdup(table, sizeof(ip6_frags_ns_ctl_table), GFP_KERNEL);
634e4a2d5c2SPavel Emelyanov 		if (table == NULL)
635e4a2d5c2SPavel Emelyanov 			goto err_alloc;
636e4a2d5c2SPavel Emelyanov 
637e31e0bdcSPavel Emelyanov 		table[0].data = &net->ipv6.frags.high_thresh;
6381bab4c75SNikolay Aleksandrov 		table[0].extra1 = &net->ipv6.frags.low_thresh;
6391bab4c75SNikolay Aleksandrov 		table[0].extra2 = &init_net.ipv6.frags.high_thresh;
640e31e0bdcSPavel Emelyanov 		table[1].data = &net->ipv6.frags.low_thresh;
6411bab4c75SNikolay Aleksandrov 		table[1].extra2 = &net->ipv6.frags.high_thresh;
642b2fd5321SPavel Emelyanov 		table[2].data = &net->ipv6.frags.timeout;
643464dc801SEric W. Biederman 
644464dc801SEric W. Biederman 		/* Don't export sysctls to unprivileged users */
645464dc801SEric W. Biederman 		if (net->user_ns != &init_user_ns)
646464dc801SEric W. Biederman 			table[0].procname = NULL;
647e4a2d5c2SPavel Emelyanov 	}
648e4a2d5c2SPavel Emelyanov 
649ec8f23ceSEric W. Biederman 	hdr = register_net_sysctl(net, "net/ipv6", table);
650e4a2d5c2SPavel Emelyanov 	if (hdr == NULL)
651e4a2d5c2SPavel Emelyanov 		goto err_reg;
652e4a2d5c2SPavel Emelyanov 
653e4a2d5c2SPavel Emelyanov 	net->ipv6.sysctl.frags_hdr = hdr;
654e4a2d5c2SPavel Emelyanov 	return 0;
655e4a2d5c2SPavel Emelyanov 
656e4a2d5c2SPavel Emelyanov err_reg:
65709ad9bc7SOctavian Purdila 	if (!net_eq(net, &init_net))
658e4a2d5c2SPavel Emelyanov 		kfree(table);
659e4a2d5c2SPavel Emelyanov err_alloc:
660e4a2d5c2SPavel Emelyanov 	return -ENOMEM;
661e4a2d5c2SPavel Emelyanov }
662e4a2d5c2SPavel Emelyanov 
6632c8c1e72SAlexey Dobriyan static void __net_exit ip6_frags_ns_sysctl_unregister(struct net *net)
664e4a2d5c2SPavel Emelyanov {
665e4a2d5c2SPavel Emelyanov 	struct ctl_table *table;
666e4a2d5c2SPavel Emelyanov 
667e4a2d5c2SPavel Emelyanov 	table = net->ipv6.sysctl.frags_hdr->ctl_table_arg;
668e4a2d5c2SPavel Emelyanov 	unregister_net_sysctl_table(net->ipv6.sysctl.frags_hdr);
6693705e11aSYang Hongyang 	if (!net_eq(net, &init_net))
670e4a2d5c2SPavel Emelyanov 		kfree(table);
6718d8354d2SPavel Emelyanov }
6727d291ebbSPavel Emelyanov 
6737d291ebbSPavel Emelyanov static struct ctl_table_header *ip6_ctl_header;
6747d291ebbSPavel Emelyanov 
6757d291ebbSPavel Emelyanov static int ip6_frags_sysctl_register(void)
6767d291ebbSPavel Emelyanov {
67743444757SEric W. Biederman 	ip6_ctl_header = register_net_sysctl(&init_net, "net/ipv6",
6787d291ebbSPavel Emelyanov 			ip6_frags_ctl_table);
6797d291ebbSPavel Emelyanov 	return ip6_ctl_header == NULL ? -ENOMEM : 0;
6807d291ebbSPavel Emelyanov }
6817d291ebbSPavel Emelyanov 
6827d291ebbSPavel Emelyanov static void ip6_frags_sysctl_unregister(void)
6837d291ebbSPavel Emelyanov {
6847d291ebbSPavel Emelyanov 	unregister_net_sysctl_table(ip6_ctl_header);
6857d291ebbSPavel Emelyanov }
6868d8354d2SPavel Emelyanov #else
687fc08c258SFabian Frederick static int ip6_frags_ns_sysctl_register(struct net *net)
6888d8354d2SPavel Emelyanov {
6898d8354d2SPavel Emelyanov 	return 0;
6908d8354d2SPavel Emelyanov }
691e4a2d5c2SPavel Emelyanov 
692fc08c258SFabian Frederick static void ip6_frags_ns_sysctl_unregister(struct net *net)
693e4a2d5c2SPavel Emelyanov {
694e4a2d5c2SPavel Emelyanov }
6957d291ebbSPavel Emelyanov 
696fc08c258SFabian Frederick static int ip6_frags_sysctl_register(void)
6977d291ebbSPavel Emelyanov {
6987d291ebbSPavel Emelyanov 	return 0;
6997d291ebbSPavel Emelyanov }
7007d291ebbSPavel Emelyanov 
701fc08c258SFabian Frederick static void ip6_frags_sysctl_unregister(void)
7027d291ebbSPavel Emelyanov {
7037d291ebbSPavel Emelyanov }
7048d8354d2SPavel Emelyanov #endif
7058d8354d2SPavel Emelyanov 
7062c8c1e72SAlexey Dobriyan static int __net_init ipv6_frags_init_net(struct net *net)
7078d8354d2SPavel Emelyanov {
7087c070aa9SShan Wei 	net->ipv6.frags.high_thresh = IPV6_FRAG_HIGH_THRESH;
7097c070aa9SShan Wei 	net->ipv6.frags.low_thresh = IPV6_FRAG_LOW_THRESH;
710b2fd5321SPavel Emelyanov 	net->ipv6.frags.timeout = IPV6_FRAG_TIMEOUT;
7118d8354d2SPavel Emelyanov 
712e5a2bb84SPavel Emelyanov 	inet_frags_init_net(&net->ipv6.frags);
713e5a2bb84SPavel Emelyanov 
7140a64b4b8SPavel Emelyanov 	return ip6_frags_ns_sysctl_register(net);
715e71e0349SDaniel Lezcano }
716e71e0349SDaniel Lezcano 
7172c8c1e72SAlexey Dobriyan static void __net_exit ipv6_frags_exit_net(struct net *net)
71881566e83SPavel Emelyanov {
7190a64b4b8SPavel Emelyanov 	ip6_frags_ns_sysctl_unregister(net);
72081566e83SPavel Emelyanov 	inet_frags_exit_net(&net->ipv6.frags, &ip6_frags);
72181566e83SPavel Emelyanov }
72281566e83SPavel Emelyanov 
72381566e83SPavel Emelyanov static struct pernet_operations ip6_frags_ops = {
72481566e83SPavel Emelyanov 	.init = ipv6_frags_init_net,
72581566e83SPavel Emelyanov 	.exit = ipv6_frags_exit_net,
72681566e83SPavel Emelyanov };
72781566e83SPavel Emelyanov 
728853cbbaaSDaniel Lezcano int __init ipv6_frag_init(void)
7291da177e4SLinus Torvalds {
730853cbbaaSDaniel Lezcano 	int ret;
7311da177e4SLinus Torvalds 
732853cbbaaSDaniel Lezcano 	ret = inet6_add_protocol(&frag_protocol, IPPROTO_FRAGMENT);
733853cbbaaSDaniel Lezcano 	if (ret)
734853cbbaaSDaniel Lezcano 		goto out;
735e71e0349SDaniel Lezcano 
7367d291ebbSPavel Emelyanov 	ret = ip6_frags_sysctl_register();
7377d291ebbSPavel Emelyanov 	if (ret)
7387d291ebbSPavel Emelyanov 		goto err_sysctl;
7397d291ebbSPavel Emelyanov 
7400002c630SPavel Emelyanov 	ret = register_pernet_subsys(&ip6_frags_ops);
7410002c630SPavel Emelyanov 	if (ret)
7420002c630SPavel Emelyanov 		goto err_pernet;
7438d8354d2SPavel Emelyanov 
744321a3a99SPavel Emelyanov 	ip6_frags.hashfn = ip6_hashfn;
745c6fda282SPavel Emelyanov 	ip6_frags.constructor = ip6_frag_init;
746c9547709SPavel Emelyanov 	ip6_frags.destructor = NULL;
7471e4b8287SPavel Emelyanov 	ip6_frags.skb_free = NULL;
7481e4b8287SPavel Emelyanov 	ip6_frags.qsize = sizeof(struct frag_queue);
749abd6523dSPavel Emelyanov 	ip6_frags.match = ip6_frag_match;
750e521db9dSPavel Emelyanov 	ip6_frags.frag_expire = ip6_frag_expire;
751d4ad4d22SNikolay Aleksandrov 	ip6_frags.frags_cache_name = ip6_frag_cache_name;
752d4ad4d22SNikolay Aleksandrov 	ret = inet_frags_init(&ip6_frags);
753d4ad4d22SNikolay Aleksandrov 	if (ret)
754d4ad4d22SNikolay Aleksandrov 		goto err_pernet;
755853cbbaaSDaniel Lezcano out:
756853cbbaaSDaniel Lezcano 	return ret;
7570002c630SPavel Emelyanov 
7580002c630SPavel Emelyanov err_pernet:
7597d291ebbSPavel Emelyanov 	ip6_frags_sysctl_unregister();
7607d291ebbSPavel Emelyanov err_sysctl:
7610002c630SPavel Emelyanov 	inet6_del_protocol(&frag_protocol, IPPROTO_FRAGMENT);
7620002c630SPavel Emelyanov 	goto out;
763853cbbaaSDaniel Lezcano }
764853cbbaaSDaniel Lezcano 
765853cbbaaSDaniel Lezcano void ipv6_frag_exit(void)
766853cbbaaSDaniel Lezcano {
767853cbbaaSDaniel Lezcano 	inet_frags_fini(&ip6_frags);
7687d291ebbSPavel Emelyanov 	ip6_frags_sysctl_unregister();
76981566e83SPavel Emelyanov 	unregister_pernet_subsys(&ip6_frags_ops);
770853cbbaaSDaniel Lezcano 	inet6_del_protocol(&frag_protocol, IPPROTO_FRAGMENT);
7711da177e4SLinus Torvalds }
772