xref: /openbmc/linux/net/ipv6/reassembly.c (revision d4ad4d22)
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 
651da177e4SLinus Torvalds struct ip6frag_skb_cb
661da177e4SLinus Torvalds {
671da177e4SLinus Torvalds 	struct inet6_skb_parm	h;
681da177e4SLinus Torvalds 	int			offset;
691da177e4SLinus Torvalds };
701da177e4SLinus Torvalds 
711da177e4SLinus Torvalds #define FRAG6_CB(skb)	((struct ip6frag_skb_cb*)((skb)->cb))
721da177e4SLinus Torvalds 
73eec2e618SHannes Frederic Sowa static inline u8 ip6_frag_ecn(const struct ipv6hdr *ipv6h)
74eec2e618SHannes Frederic Sowa {
75eec2e618SHannes Frederic Sowa 	return 1 << (ipv6_get_dsfield(ipv6h) & INET_ECN_MASK);
76eec2e618SHannes Frederic Sowa }
771da177e4SLinus Torvalds 
787eb95156SPavel Emelyanov static struct inet_frags ip6_frags;
791da177e4SLinus Torvalds 
80f61944efSHerbert Xu static int ip6_frag_reasm(struct frag_queue *fq, struct sk_buff *prev,
81f61944efSHerbert Xu 			  struct net_device *dev);
82f61944efSHerbert Xu 
83f6596f9dSZach Brown /*
84f6596f9dSZach Brown  * callers should be careful not to use the hash value outside the ipfrag_lock
85f6596f9dSZach Brown  * as doing so could race with ipfrag_hash_rnd being recalculated.
86f6596f9dSZach Brown  */
87b1190570SHannes Frederic Sowa static unsigned int inet6_hash_frag(__be32 id, const struct in6_addr *saddr,
88b1190570SHannes Frederic Sowa 				    const struct in6_addr *daddr)
891da177e4SLinus Torvalds {
90b1190570SHannes Frederic Sowa 	net_get_random_once(&ip6_frags.rnd, sizeof(ip6_frags.rnd));
91fb3cfe6eSFlorian Westphal 	return jhash_3words(ipv6_addr_hash(saddr), ipv6_addr_hash(daddr),
92b1190570SHannes Frederic Sowa 			    (__force u32)id, ip6_frags.rnd);
931da177e4SLinus Torvalds }
941da177e4SLinus Torvalds 
9536c77782SFlorian Westphal static unsigned int ip6_hashfn(const struct inet_frag_queue *q)
961da177e4SLinus Torvalds {
9736c77782SFlorian Westphal 	const struct frag_queue *fq;
981da177e4SLinus Torvalds 
99321a3a99SPavel Emelyanov 	fq = container_of(q, struct frag_queue, q);
100b1190570SHannes Frederic Sowa 	return inet6_hash_frag(fq->id, &fq->saddr, &fq->daddr);
1011da177e4SLinus Torvalds }
1021da177e4SLinus Torvalds 
10336c77782SFlorian Westphal bool ip6_frag_match(const struct inet_frag_queue *q, const void *a)
104abd6523dSPavel Emelyanov {
10536c77782SFlorian Westphal 	const struct frag_queue *fq;
10636c77782SFlorian Westphal 	const struct ip6_create_arg *arg = a;
107abd6523dSPavel Emelyanov 
108abd6523dSPavel Emelyanov 	fq = container_of(q, struct frag_queue, q);
109cbc264caSEric Dumazet 	return	fq->id == arg->id &&
110cbc264caSEric Dumazet 		fq->user == arg->user &&
111abd6523dSPavel Emelyanov 		ipv6_addr_equal(&fq->saddr, arg->src) &&
112cbc264caSEric Dumazet 		ipv6_addr_equal(&fq->daddr, arg->dst);
113abd6523dSPavel Emelyanov }
114abd6523dSPavel Emelyanov EXPORT_SYMBOL(ip6_frag_match);
115abd6523dSPavel Emelyanov 
11636c77782SFlorian Westphal void ip6_frag_init(struct inet_frag_queue *q, const void *a)
117c6fda282SPavel Emelyanov {
118c6fda282SPavel Emelyanov 	struct frag_queue *fq = container_of(q, struct frag_queue, q);
11936c77782SFlorian Westphal 	const struct ip6_create_arg *arg = a;
120c6fda282SPavel Emelyanov 
121c6fda282SPavel Emelyanov 	fq->id = arg->id;
1220b5ccb2eSPatrick McHardy 	fq->user = arg->user;
1234e3fd7a0SAlexey Dobriyan 	fq->saddr = *arg->src;
1244e3fd7a0SAlexey Dobriyan 	fq->daddr = *arg->dst;
125eec2e618SHannes Frederic Sowa 	fq->ecn = arg->ecn;
126c6fda282SPavel Emelyanov }
127c6fda282SPavel Emelyanov EXPORT_SYMBOL(ip6_frag_init);
128c6fda282SPavel Emelyanov 
129b836c99fSAmerigo Wang void ip6_expire_frag_queue(struct net *net, struct frag_queue *fq,
130b836c99fSAmerigo Wang 			   struct inet_frags *frags)
1311da177e4SLinus Torvalds {
132a11d206dSYOSHIFUJI Hideaki 	struct net_device *dev = NULL;
133e521db9dSPavel Emelyanov 
1345ab11c98SPavel Emelyanov 	spin_lock(&fq->q.lock);
1351da177e4SLinus Torvalds 
13606aa8b8aSNikolay Aleksandrov 	if (fq->q.flags & INET_FRAG_COMPLETE)
1371da177e4SLinus Torvalds 		goto out;
1381da177e4SLinus Torvalds 
139b836c99fSAmerigo Wang 	inet_frag_kill(&fq->q, frags);
1401da177e4SLinus Torvalds 
141a11d206dSYOSHIFUJI Hideaki 	rcu_read_lock();
14269df9d59SEric Dumazet 	dev = dev_get_by_index_rcu(net, fq->iif);
14369df9d59SEric Dumazet 	if (!dev)
14469df9d59SEric Dumazet 		goto out_rcu_unlock;
14569df9d59SEric Dumazet 
146483a47d2SDenis V. Lunev 	IP6_INC_STATS_BH(net, __in6_dev_get(dev), IPSTATS_MIB_REASMFAILS);
1471da177e4SLinus Torvalds 
1482e404f63SNikolay Aleksandrov 	if (fq->q.flags & INET_FRAG_EVICTED)
1492e404f63SNikolay Aleksandrov 		goto out_rcu_unlock;
1502e404f63SNikolay Aleksandrov 
1512e404f63SNikolay Aleksandrov 	IP6_INC_STATS_BH(net, __in6_dev_get(dev), IPSTATS_MIB_REASMTIMEOUT);
1522e404f63SNikolay Aleksandrov 
15378c784c4SIngo Oeser 	/* Don't send error if the first segment did not arrive. */
15406aa8b8aSNikolay Aleksandrov 	if (!(fq->q.flags & INET_FRAG_FIRST_IN) || !fq->q.fragments)
15569df9d59SEric Dumazet 		goto out_rcu_unlock;
15678c784c4SIngo Oeser 
1572e404f63SNikolay Aleksandrov 	/* But use as source device on which LAST ARRIVED
1582e404f63SNikolay Aleksandrov 	 * segment was received. And do not use fq->dev
1592e404f63SNikolay Aleksandrov 	 * pointer directly, device might already disappeared.
1601da177e4SLinus Torvalds 	 */
1615ab11c98SPavel Emelyanov 	fq->q.fragments->dev = dev;
1623ffe533cSAlexey Dobriyan 	icmpv6_send(fq->q.fragments, ICMPV6_TIME_EXCEED, ICMPV6_EXC_FRAGTIME, 0);
16369df9d59SEric Dumazet out_rcu_unlock:
16469df9d59SEric Dumazet 	rcu_read_unlock();
1651da177e4SLinus Torvalds out:
1665ab11c98SPavel Emelyanov 	spin_unlock(&fq->q.lock);
167b836c99fSAmerigo Wang 	inet_frag_put(&fq->q, frags);
168b836c99fSAmerigo Wang }
169b836c99fSAmerigo Wang EXPORT_SYMBOL(ip6_expire_frag_queue);
170b836c99fSAmerigo Wang 
171b836c99fSAmerigo Wang static void ip6_frag_expire(unsigned long data)
172b836c99fSAmerigo Wang {
173b836c99fSAmerigo Wang 	struct frag_queue *fq;
174b836c99fSAmerigo Wang 	struct net *net;
175b836c99fSAmerigo Wang 
176b836c99fSAmerigo Wang 	fq = container_of((struct inet_frag_queue *)data, struct frag_queue, q);
177b836c99fSAmerigo Wang 	net = container_of(fq->q.net, struct net, ipv6.frags);
178b836c99fSAmerigo Wang 
179b836c99fSAmerigo Wang 	ip6_expire_frag_queue(net, fq, &ip6_frags);
1801da177e4SLinus Torvalds }
1811da177e4SLinus Torvalds 
182abd6523dSPavel Emelyanov static __inline__ struct frag_queue *
183eec2e618SHannes Frederic Sowa fq_find(struct net *net, __be32 id, const struct in6_addr *src,
184eec2e618SHannes Frederic Sowa 	const struct in6_addr *dst, u8 ecn)
1851da177e4SLinus Torvalds {
186c6fda282SPavel Emelyanov 	struct inet_frag_queue *q;
187c6fda282SPavel Emelyanov 	struct ip6_create_arg arg;
188abd6523dSPavel Emelyanov 	unsigned int hash;
1891da177e4SLinus Torvalds 
190c6fda282SPavel Emelyanov 	arg.id = id;
1910b5ccb2eSPatrick McHardy 	arg.user = IP6_DEFRAG_LOCAL_DELIVER;
192c6fda282SPavel Emelyanov 	arg.src = src;
193c6fda282SPavel Emelyanov 	arg.dst = dst;
194eec2e618SHannes Frederic Sowa 	arg.ecn = ecn;
1959a375803SPavel Emelyanov 
196b1190570SHannes Frederic Sowa 	hash = inet6_hash_frag(id, src, dst);
197c6fda282SPavel Emelyanov 
198ac18e750SPavel Emelyanov 	q = inet_frag_find(&net->ipv6.frags, &ip6_frags, &arg, hash);
1995a3da1feSHannes Frederic Sowa 	if (IS_ERR_OR_NULL(q)) {
2005a3da1feSHannes Frederic Sowa 		inet_frag_maybe_warn_overflow(q, pr_fmt());
2019546377cSShan Wei 		return NULL;
2025a3da1feSHannes Frederic Sowa 	}
203c6fda282SPavel Emelyanov 	return container_of(q, struct frag_queue, q);
2041da177e4SLinus Torvalds }
2051da177e4SLinus Torvalds 
206f61944efSHerbert Xu static int ip6_frag_queue(struct frag_queue *fq, struct sk_buff *skb,
2071da177e4SLinus Torvalds 			   struct frag_hdr *fhdr, int nhoff)
2081da177e4SLinus Torvalds {
2091da177e4SLinus Torvalds 	struct sk_buff *prev, *next;
210f61944efSHerbert Xu 	struct net_device *dev;
2111da177e4SLinus Torvalds 	int offset, end;
212adf30907SEric Dumazet 	struct net *net = dev_net(skb_dst(skb)->dev);
213eec2e618SHannes Frederic Sowa 	u8 ecn;
2141da177e4SLinus Torvalds 
21506aa8b8aSNikolay Aleksandrov 	if (fq->q.flags & INET_FRAG_COMPLETE)
2161da177e4SLinus Torvalds 		goto err;
2171da177e4SLinus Torvalds 
2181da177e4SLinus Torvalds 	offset = ntohs(fhdr->frag_off) & ~0x7;
2190660e03fSArnaldo Carvalho de Melo 	end = offset + (ntohs(ipv6_hdr(skb)->payload_len) -
2200660e03fSArnaldo Carvalho de Melo 			((u8 *)(fhdr + 1) - (u8 *)(ipv6_hdr(skb) + 1)));
2211da177e4SLinus Torvalds 
2221da177e4SLinus Torvalds 	if ((unsigned int)end > IPV6_MAXPLEN) {
223adf30907SEric Dumazet 		IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)),
224a11d206dSYOSHIFUJI Hideaki 				 IPSTATS_MIB_INHDRERRORS);
225d56f90a7SArnaldo Carvalho de Melo 		icmpv6_param_prob(skb, ICMPV6_HDR_FIELD,
226d56f90a7SArnaldo Carvalho de Melo 				  ((u8 *)&fhdr->frag_off -
227d56f90a7SArnaldo Carvalho de Melo 				   skb_network_header(skb)));
228f61944efSHerbert Xu 		return -1;
2291da177e4SLinus Torvalds 	}
2301da177e4SLinus Torvalds 
231eec2e618SHannes Frederic Sowa 	ecn = ip6_frag_ecn(ipv6_hdr(skb));
232eec2e618SHannes Frederic Sowa 
233d56f90a7SArnaldo Carvalho de Melo 	if (skb->ip_summed == CHECKSUM_COMPLETE) {
234d56f90a7SArnaldo Carvalho de Melo 		const unsigned char *nh = skb_network_header(skb);
2351da177e4SLinus Torvalds 		skb->csum = csum_sub(skb->csum,
236d56f90a7SArnaldo Carvalho de Melo 				     csum_partial(nh, (u8 *)(fhdr + 1) - nh,
237d56f90a7SArnaldo Carvalho de Melo 						  0));
238d56f90a7SArnaldo Carvalho de Melo 	}
2391da177e4SLinus Torvalds 
2401da177e4SLinus Torvalds 	/* Is this the final fragment? */
2411da177e4SLinus Torvalds 	if (!(fhdr->frag_off & htons(IP6_MF))) {
2421da177e4SLinus Torvalds 		/* If we already have some bits beyond end
2431da177e4SLinus Torvalds 		 * or have different end, the segment is corrupted.
2441da177e4SLinus Torvalds 		 */
2455ab11c98SPavel Emelyanov 		if (end < fq->q.len ||
24606aa8b8aSNikolay Aleksandrov 		    ((fq->q.flags & INET_FRAG_LAST_IN) && end != fq->q.len))
2471da177e4SLinus Torvalds 			goto err;
24806aa8b8aSNikolay Aleksandrov 		fq->q.flags |= INET_FRAG_LAST_IN;
2495ab11c98SPavel Emelyanov 		fq->q.len = end;
2501da177e4SLinus Torvalds 	} else {
2511da177e4SLinus Torvalds 		/* Check if the fragment is rounded to 8 bytes.
2521da177e4SLinus Torvalds 		 * Required by the RFC.
2531da177e4SLinus Torvalds 		 */
2541da177e4SLinus Torvalds 		if (end & 0x7) {
2551da177e4SLinus Torvalds 			/* RFC2460 says always send parameter problem in
2561da177e4SLinus Torvalds 			 * this case. -DaveM
2571da177e4SLinus Torvalds 			 */
258adf30907SEric Dumazet 			IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)),
259a11d206dSYOSHIFUJI Hideaki 					 IPSTATS_MIB_INHDRERRORS);
2601da177e4SLinus Torvalds 			icmpv6_param_prob(skb, ICMPV6_HDR_FIELD,
2611da177e4SLinus Torvalds 					  offsetof(struct ipv6hdr, payload_len));
262f61944efSHerbert Xu 			return -1;
2631da177e4SLinus Torvalds 		}
2645ab11c98SPavel Emelyanov 		if (end > fq->q.len) {
2651da177e4SLinus Torvalds 			/* Some bits beyond end -> corruption. */
26606aa8b8aSNikolay Aleksandrov 			if (fq->q.flags & INET_FRAG_LAST_IN)
2671da177e4SLinus Torvalds 				goto err;
2685ab11c98SPavel Emelyanov 			fq->q.len = end;
2691da177e4SLinus Torvalds 		}
2701da177e4SLinus Torvalds 	}
2711da177e4SLinus Torvalds 
2721da177e4SLinus Torvalds 	if (end == offset)
2731da177e4SLinus Torvalds 		goto err;
2741da177e4SLinus Torvalds 
2751da177e4SLinus Torvalds 	/* Point into the IP datagram 'data' part. */
2761da177e4SLinus Torvalds 	if (!pskb_pull(skb, (u8 *) (fhdr + 1) - skb->data))
2771da177e4SLinus Torvalds 		goto err;
27842ca89c1SStephen Hemminger 
27942ca89c1SStephen Hemminger 	if (pskb_trim_rcsum(skb, end - offset))
2801da177e4SLinus Torvalds 		goto err;
2811da177e4SLinus Torvalds 
2821da177e4SLinus Torvalds 	/* Find out which fragments are in front and at the back of us
2831da177e4SLinus Torvalds 	 * in the chain of fragments so far.  We must know where to put
2841da177e4SLinus Torvalds 	 * this fragment, right?
2851da177e4SLinus Torvalds 	 */
286d6bebca9SChangli Gao 	prev = fq->q.fragments_tail;
287d6bebca9SChangli Gao 	if (!prev || FRAG6_CB(prev)->offset < offset) {
288d6bebca9SChangli Gao 		next = NULL;
289d6bebca9SChangli Gao 		goto found;
290d6bebca9SChangli Gao 	}
2911da177e4SLinus Torvalds 	prev = NULL;
2925ab11c98SPavel Emelyanov 	for(next = fq->q.fragments; next != NULL; next = next->next) {
2931da177e4SLinus Torvalds 		if (FRAG6_CB(next)->offset >= offset)
2941da177e4SLinus Torvalds 			break;	/* bingo! */
2951da177e4SLinus Torvalds 		prev = next;
2961da177e4SLinus Torvalds 	}
2971da177e4SLinus Torvalds 
298d6bebca9SChangli Gao found:
2995de658f8SEric Dumazet 	/* RFC5722, Section 4, amended by Errata ID : 3089
30070789d70SNicolas Dichtel 	 *                          When reassembling an IPv6 datagram, if
30170789d70SNicolas Dichtel 	 *   one or more its constituent fragments is determined to be an
30270789d70SNicolas Dichtel 	 *   overlapping fragment, the entire datagram (and any constituent
3035de658f8SEric Dumazet 	 *   fragments) MUST be silently discarded.
3041da177e4SLinus Torvalds 	 */
3051da177e4SLinus Torvalds 
30670789d70SNicolas Dichtel 	/* Check for overlap with preceding fragment. */
30770789d70SNicolas Dichtel 	if (prev &&
308f4642141SShan Wei 	    (FRAG6_CB(prev)->offset + prev->len) > offset)
30970789d70SNicolas Dichtel 		goto discard_fq;
3101da177e4SLinus Torvalds 
31170789d70SNicolas Dichtel 	/* Look for overlap with succeeding segment. */
31270789d70SNicolas Dichtel 	if (next && FRAG6_CB(next)->offset < end)
31370789d70SNicolas Dichtel 		goto discard_fq;
3141da177e4SLinus Torvalds 
3151da177e4SLinus Torvalds 	FRAG6_CB(skb)->offset = offset;
3161da177e4SLinus Torvalds 
3171da177e4SLinus Torvalds 	/* Insert this fragment in the chain of fragments. */
3181da177e4SLinus Torvalds 	skb->next = next;
319d6bebca9SChangli Gao 	if (!next)
320d6bebca9SChangli Gao 		fq->q.fragments_tail = skb;
3211da177e4SLinus Torvalds 	if (prev)
3221da177e4SLinus Torvalds 		prev->next = skb;
3231da177e4SLinus Torvalds 	else
3245ab11c98SPavel Emelyanov 		fq->q.fragments = skb;
3251da177e4SLinus Torvalds 
326f61944efSHerbert Xu 	dev = skb->dev;
327f61944efSHerbert Xu 	if (dev) {
328f61944efSHerbert Xu 		fq->iif = dev->ifindex;
3291da177e4SLinus Torvalds 		skb->dev = NULL;
330f61944efSHerbert Xu 	}
3315ab11c98SPavel Emelyanov 	fq->q.stamp = skb->tstamp;
3325ab11c98SPavel Emelyanov 	fq->q.meat += skb->len;
333eec2e618SHannes Frederic Sowa 	fq->ecn |= ecn;
334d433673eSJesper Dangaard Brouer 	add_frag_mem_limit(&fq->q, skb->truesize);
3351da177e4SLinus Torvalds 
3361da177e4SLinus Torvalds 	/* The first fragment.
3371da177e4SLinus Torvalds 	 * nhoffset is obtained from the first fragment, of course.
3381da177e4SLinus Torvalds 	 */
3391da177e4SLinus Torvalds 	if (offset == 0) {
3401da177e4SLinus Torvalds 		fq->nhoffset = nhoff;
34106aa8b8aSNikolay Aleksandrov 		fq->q.flags |= INET_FRAG_FIRST_IN;
3421da177e4SLinus Torvalds 	}
343f61944efSHerbert Xu 
34406aa8b8aSNikolay Aleksandrov 	if (fq->q.flags == (INET_FRAG_FIRST_IN | INET_FRAG_LAST_IN) &&
34597599dc7SEric Dumazet 	    fq->q.meat == fq->q.len) {
34697599dc7SEric Dumazet 		int res;
34797599dc7SEric Dumazet 		unsigned long orefdst = skb->_skb_refdst;
348f61944efSHerbert Xu 
34997599dc7SEric Dumazet 		skb->_skb_refdst = 0UL;
35097599dc7SEric Dumazet 		res = ip6_frag_reasm(fq, prev, dev);
35197599dc7SEric Dumazet 		skb->_skb_refdst = orefdst;
35297599dc7SEric Dumazet 		return res;
35397599dc7SEric Dumazet 	}
35497599dc7SEric Dumazet 
35597599dc7SEric Dumazet 	skb_dst_drop(skb);
356f61944efSHerbert Xu 	return -1;
3571da177e4SLinus Torvalds 
35870789d70SNicolas Dichtel discard_fq:
359b836c99fSAmerigo Wang 	inet_frag_kill(&fq->q, &ip6_frags);
3601da177e4SLinus Torvalds err:
361d2373862SNikolay Aleksandrov 	IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)),
3623bd653c8SDenis V. Lunev 			 IPSTATS_MIB_REASMFAILS);
3631da177e4SLinus Torvalds 	kfree_skb(skb);
364f61944efSHerbert Xu 	return -1;
3651da177e4SLinus Torvalds }
3661da177e4SLinus Torvalds 
3671da177e4SLinus Torvalds /*
3681da177e4SLinus Torvalds  *	Check if this packet is complete.
3691da177e4SLinus Torvalds  *	Returns NULL on failure by any reason, and pointer
3701da177e4SLinus Torvalds  *	to current nexthdr field in reassembled frame.
3711da177e4SLinus Torvalds  *
3721da177e4SLinus Torvalds  *	It is called with locked fq, and caller must check that
3731da177e4SLinus Torvalds  *	queue is eligible for reassembly i.e. it is not COMPLETE,
3741da177e4SLinus Torvalds  *	the last and the first frames arrived and all the bits are here.
3751da177e4SLinus Torvalds  */
376f61944efSHerbert Xu static int ip6_frag_reasm(struct frag_queue *fq, struct sk_buff *prev,
3771da177e4SLinus Torvalds 			  struct net_device *dev)
3781da177e4SLinus Torvalds {
3792bad35b7SJorge Boncompte [DTI2] 	struct net *net = container_of(fq->q.net, struct net, ipv6.frags);
3805ab11c98SPavel Emelyanov 	struct sk_buff *fp, *head = fq->q.fragments;
3811da177e4SLinus Torvalds 	int    payload_len;
3821da177e4SLinus Torvalds 	unsigned int nhoff;
383ec16439eSEric Dumazet 	int sum_truesize;
384eec2e618SHannes Frederic Sowa 	u8 ecn;
3851da177e4SLinus Torvalds 
386b836c99fSAmerigo Wang 	inet_frag_kill(&fq->q, &ip6_frags);
3871da177e4SLinus Torvalds 
388eec2e618SHannes Frederic Sowa 	ecn = ip_frag_ecn_table[fq->ecn];
389eec2e618SHannes Frederic Sowa 	if (unlikely(ecn == 0xff))
390eec2e618SHannes Frederic Sowa 		goto out_fail;
391eec2e618SHannes Frederic Sowa 
392f61944efSHerbert Xu 	/* Make the one we just received the head. */
393f61944efSHerbert Xu 	if (prev) {
394f61944efSHerbert Xu 		head = prev->next;
395f61944efSHerbert Xu 		fp = skb_clone(head, GFP_ATOMIC);
396f61944efSHerbert Xu 
397f61944efSHerbert Xu 		if (!fp)
398f61944efSHerbert Xu 			goto out_oom;
399f61944efSHerbert Xu 
400f61944efSHerbert Xu 		fp->next = head->next;
401d6bebca9SChangli Gao 		if (!fp->next)
402d6bebca9SChangli Gao 			fq->q.fragments_tail = fp;
403f61944efSHerbert Xu 		prev->next = fp;
404f61944efSHerbert Xu 
4055ab11c98SPavel Emelyanov 		skb_morph(head, fq->q.fragments);
4065ab11c98SPavel Emelyanov 		head->next = fq->q.fragments->next;
407f61944efSHerbert Xu 
408808db80aSEric Dumazet 		consume_skb(fq->q.fragments);
4095ab11c98SPavel Emelyanov 		fq->q.fragments = head;
410f61944efSHerbert Xu 	}
411f61944efSHerbert Xu 
412547b792cSIlpo Järvinen 	WARN_ON(head == NULL);
413547b792cSIlpo Järvinen 	WARN_ON(FRAG6_CB(head)->offset != 0);
4141da177e4SLinus Torvalds 
4151da177e4SLinus Torvalds 	/* Unfragmented part is taken from the first segment. */
416d56f90a7SArnaldo Carvalho de Melo 	payload_len = ((head->data - skb_network_header(head)) -
4175ab11c98SPavel Emelyanov 		       sizeof(struct ipv6hdr) + fq->q.len -
418d56f90a7SArnaldo Carvalho de Melo 		       sizeof(struct frag_hdr));
4191da177e4SLinus Torvalds 	if (payload_len > IPV6_MAXPLEN)
4201da177e4SLinus Torvalds 		goto out_oversize;
4211da177e4SLinus Torvalds 
4221da177e4SLinus Torvalds 	/* Head of list must not be cloned. */
42314bbd6a5SPravin B Shelar 	if (skb_unclone(head, GFP_ATOMIC))
4241da177e4SLinus Torvalds 		goto out_oom;
4251da177e4SLinus Torvalds 
4261da177e4SLinus Torvalds 	/* If the first fragment is fragmented itself, we split
4271da177e4SLinus Torvalds 	 * it to two chunks: the first with data and paged part
4281da177e4SLinus Torvalds 	 * and the second, holding only fragments. */
42921dc3301SDavid S. Miller 	if (skb_has_frag_list(head)) {
4301da177e4SLinus Torvalds 		struct sk_buff *clone;
4311da177e4SLinus Torvalds 		int i, plen = 0;
4321da177e4SLinus Torvalds 
4331da177e4SLinus Torvalds 		if ((clone = alloc_skb(0, GFP_ATOMIC)) == NULL)
4341da177e4SLinus Torvalds 			goto out_oom;
4351da177e4SLinus Torvalds 		clone->next = head->next;
4361da177e4SLinus Torvalds 		head->next = clone;
4371da177e4SLinus Torvalds 		skb_shinfo(clone)->frag_list = skb_shinfo(head)->frag_list;
4384d9092bbSDavid S. Miller 		skb_frag_list_init(head);
4391da177e4SLinus Torvalds 		for (i = 0; i < skb_shinfo(head)->nr_frags; i++)
4409e903e08SEric Dumazet 			plen += skb_frag_size(&skb_shinfo(head)->frags[i]);
4411da177e4SLinus Torvalds 		clone->len = clone->data_len = head->data_len - plen;
4421da177e4SLinus Torvalds 		head->data_len -= clone->len;
4431da177e4SLinus Torvalds 		head->len -= clone->len;
4441da177e4SLinus Torvalds 		clone->csum = 0;
4451da177e4SLinus Torvalds 		clone->ip_summed = head->ip_summed;
446d433673eSJesper Dangaard Brouer 		add_frag_mem_limit(&fq->q, clone->truesize);
4471da177e4SLinus Torvalds 	}
4481da177e4SLinus Torvalds 
4491da177e4SLinus Torvalds 	/* We have to remove fragment header from datagram and to relocate
4501da177e4SLinus Torvalds 	 * header in order to calculate ICV correctly. */
4511da177e4SLinus Torvalds 	nhoff = fq->nhoffset;
452b0e380b1SArnaldo Carvalho de Melo 	skb_network_header(head)[nhoff] = skb_transport_header(head)[0];
4531da177e4SLinus Torvalds 	memmove(head->head + sizeof(struct frag_hdr), head->head,
4541da177e4SLinus Torvalds 		(head->data - head->head) - sizeof(struct frag_hdr));
455b0e380b1SArnaldo Carvalho de Melo 	head->mac_header += sizeof(struct frag_hdr);
456b0e380b1SArnaldo Carvalho de Melo 	head->network_header += sizeof(struct frag_hdr);
4571da177e4SLinus Torvalds 
458badff6d0SArnaldo Carvalho de Melo 	skb_reset_transport_header(head);
459d56f90a7SArnaldo Carvalho de Melo 	skb_push(head, head->data - skb_network_header(head));
4601da177e4SLinus Torvalds 
461ec16439eSEric Dumazet 	sum_truesize = head->truesize;
462ec16439eSEric Dumazet 	for (fp = head->next; fp;) {
463ec16439eSEric Dumazet 		bool headstolen;
464ec16439eSEric Dumazet 		int delta;
465ec16439eSEric Dumazet 		struct sk_buff *next = fp->next;
466ec16439eSEric Dumazet 
467ec16439eSEric Dumazet 		sum_truesize += fp->truesize;
4681da177e4SLinus Torvalds 		if (head->ip_summed != fp->ip_summed)
4691da177e4SLinus Torvalds 			head->ip_summed = CHECKSUM_NONE;
47084fa7933SPatrick McHardy 		else if (head->ip_summed == CHECKSUM_COMPLETE)
4711da177e4SLinus Torvalds 			head->csum = csum_add(head->csum, fp->csum);
472ec16439eSEric Dumazet 
473ec16439eSEric Dumazet 		if (skb_try_coalesce(head, fp, &headstolen, &delta)) {
474ec16439eSEric Dumazet 			kfree_skb_partial(fp, headstolen);
475ec16439eSEric Dumazet 		} else {
476ec16439eSEric Dumazet 			if (!skb_shinfo(head)->frag_list)
477ec16439eSEric Dumazet 				skb_shinfo(head)->frag_list = fp;
478ec16439eSEric Dumazet 			head->data_len += fp->len;
479ec16439eSEric Dumazet 			head->len += fp->len;
4801da177e4SLinus Torvalds 			head->truesize += fp->truesize;
4811da177e4SLinus Torvalds 		}
482ec16439eSEric Dumazet 		fp = next;
483ec16439eSEric Dumazet 	}
484d433673eSJesper Dangaard Brouer 	sub_frag_mem_limit(&fq->q, sum_truesize);
4851da177e4SLinus Torvalds 
4861da177e4SLinus Torvalds 	head->next = NULL;
4871da177e4SLinus Torvalds 	head->dev = dev;
4885ab11c98SPavel Emelyanov 	head->tstamp = fq->q.stamp;
4890660e03fSArnaldo Carvalho de Melo 	ipv6_hdr(head)->payload_len = htons(payload_len);
490eec2e618SHannes Frederic Sowa 	ipv6_change_dsfield(ipv6_hdr(head), 0xff, ecn);
491951dbc8aSPatrick McHardy 	IP6CB(head)->nhoff = nhoff;
492f46078cfSHannes Frederic Sowa 	IP6CB(head)->flags |= IP6SKB_FRAGMENTED;
4931da177e4SLinus Torvalds 
4941da177e4SLinus Torvalds 	/* Yes, and fold redundant checksum back. 8) */
49584fa7933SPatrick McHardy 	if (head->ip_summed == CHECKSUM_COMPLETE)
496d56f90a7SArnaldo Carvalho de Melo 		head->csum = csum_partial(skb_network_header(head),
497cfe1fc77SArnaldo Carvalho de Melo 					  skb_network_header_len(head),
498d56f90a7SArnaldo Carvalho de Melo 					  head->csum);
4991da177e4SLinus Torvalds 
500a11d206dSYOSHIFUJI Hideaki 	rcu_read_lock();
5012bad35b7SJorge Boncompte [DTI2] 	IP6_INC_STATS_BH(net, __in6_dev_get(dev), IPSTATS_MIB_REASMOKS);
502a11d206dSYOSHIFUJI Hideaki 	rcu_read_unlock();
5035ab11c98SPavel Emelyanov 	fq->q.fragments = NULL;
504d6bebca9SChangli Gao 	fq->q.fragments_tail = NULL;
5051da177e4SLinus Torvalds 	return 1;
5061da177e4SLinus Torvalds 
5071da177e4SLinus Torvalds out_oversize:
508e87cc472SJoe Perches 	net_dbg_ratelimited("ip6_frag_reasm: payload len = %d\n", payload_len);
5091da177e4SLinus Torvalds 	goto out_fail;
5101da177e4SLinus Torvalds out_oom:
511e87cc472SJoe Perches 	net_dbg_ratelimited("ip6_frag_reasm: no memory for reassembly\n");
5121da177e4SLinus Torvalds out_fail:
513a11d206dSYOSHIFUJI Hideaki 	rcu_read_lock();
5142bad35b7SJorge Boncompte [DTI2] 	IP6_INC_STATS_BH(net, __in6_dev_get(dev), IPSTATS_MIB_REASMFAILS);
515a11d206dSYOSHIFUJI Hideaki 	rcu_read_unlock();
5161da177e4SLinus Torvalds 	return -1;
5171da177e4SLinus Torvalds }
5181da177e4SLinus Torvalds 
519e5bbef20SHerbert Xu static int ipv6_frag_rcv(struct sk_buff *skb)
5201da177e4SLinus Torvalds {
5211da177e4SLinus Torvalds 	struct frag_hdr *fhdr;
5221da177e4SLinus Torvalds 	struct frag_queue *fq;
523b71d1d42SEric Dumazet 	const struct ipv6hdr *hdr = ipv6_hdr(skb);
524adf30907SEric Dumazet 	struct net *net = dev_net(skb_dst(skb)->dev);
5251da177e4SLinus Torvalds 
526f46078cfSHannes Frederic Sowa 	if (IP6CB(skb)->flags & IP6SKB_FRAGMENTED)
527f46078cfSHannes Frederic Sowa 		goto fail_hdr;
528f46078cfSHannes Frederic Sowa 
529adf30907SEric Dumazet 	IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)), IPSTATS_MIB_REASMREQDS);
5301da177e4SLinus Torvalds 
5311da177e4SLinus Torvalds 	/* Jumbo payload inhibits frag. header */
53298b3377cSDenis V. Lunev 	if (hdr->payload_len==0)
53398b3377cSDenis V. Lunev 		goto fail_hdr;
53498b3377cSDenis V. Lunev 
535ea2ae17dSArnaldo Carvalho de Melo 	if (!pskb_may_pull(skb, (skb_transport_offset(skb) +
53698b3377cSDenis V. Lunev 				 sizeof(struct frag_hdr))))
53798b3377cSDenis V. Lunev 		goto fail_hdr;
5381da177e4SLinus Torvalds 
5390660e03fSArnaldo Carvalho de Melo 	hdr = ipv6_hdr(skb);
5409c70220bSArnaldo Carvalho de Melo 	fhdr = (struct frag_hdr *)skb_transport_header(skb);
5411da177e4SLinus Torvalds 
5421da177e4SLinus Torvalds 	if (!(fhdr->frag_off & htons(0xFFF9))) {
5431da177e4SLinus Torvalds 		/* It is not a fragmented frame */
544b0e380b1SArnaldo Carvalho de Melo 		skb->transport_header += sizeof(struct frag_hdr);
545483a47d2SDenis V. Lunev 		IP6_INC_STATS_BH(net,
546adf30907SEric Dumazet 				 ip6_dst_idev(skb_dst(skb)), IPSTATS_MIB_REASMOKS);
5471da177e4SLinus Torvalds 
548d56f90a7SArnaldo Carvalho de Melo 		IP6CB(skb)->nhoff = (u8 *)fhdr - skb_network_header(skb);
549f46078cfSHannes Frederic Sowa 		IP6CB(skb)->flags |= IP6SKB_FRAGMENTED;
5501da177e4SLinus Torvalds 		return 1;
5511da177e4SLinus Torvalds 	}
5521da177e4SLinus Torvalds 
553eec2e618SHannes Frederic Sowa 	fq = fq_find(net, fhdr->identification, &hdr->saddr, &hdr->daddr,
554eec2e618SHannes Frederic Sowa 		     ip6_frag_ecn(hdr));
5559546377cSShan Wei 	if (fq != NULL) {
556f61944efSHerbert Xu 		int ret;
5571da177e4SLinus Torvalds 
5585ab11c98SPavel Emelyanov 		spin_lock(&fq->q.lock);
5591da177e4SLinus Torvalds 
560f61944efSHerbert Xu 		ret = ip6_frag_queue(fq, skb, fhdr, IP6CB(skb)->nhoff);
5611da177e4SLinus Torvalds 
5625ab11c98SPavel Emelyanov 		spin_unlock(&fq->q.lock);
563b836c99fSAmerigo Wang 		inet_frag_put(&fq->q, &ip6_frags);
5641da177e4SLinus Torvalds 		return ret;
5651da177e4SLinus Torvalds 	}
5661da177e4SLinus Torvalds 
567adf30907SEric Dumazet 	IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)), IPSTATS_MIB_REASMFAILS);
5681da177e4SLinus Torvalds 	kfree_skb(skb);
5691da177e4SLinus Torvalds 	return -1;
57098b3377cSDenis V. Lunev 
57198b3377cSDenis V. Lunev fail_hdr:
572d2373862SNikolay Aleksandrov 	IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)),
573d2373862SNikolay Aleksandrov 			 IPSTATS_MIB_INHDRERRORS);
57498b3377cSDenis V. Lunev 	icmpv6_param_prob(skb, ICMPV6_HDR_FIELD, skb_network_header_len(skb));
57598b3377cSDenis V. Lunev 	return -1;
5761da177e4SLinus Torvalds }
5771da177e4SLinus Torvalds 
57841135cc8SAlexey Dobriyan static const struct inet6_protocol frag_protocol =
5791da177e4SLinus Torvalds {
5801da177e4SLinus Torvalds 	.handler	=	ipv6_frag_rcv,
5811da177e4SLinus Torvalds 	.flags		=	INET6_PROTO_NOPOLICY,
5821da177e4SLinus Torvalds };
5831da177e4SLinus Torvalds 
5848d8354d2SPavel Emelyanov #ifdef CONFIG_SYSCTL
5851bab4c75SNikolay Aleksandrov static int zero;
5861bab4c75SNikolay Aleksandrov 
5870a64b4b8SPavel Emelyanov static struct ctl_table ip6_frags_ns_ctl_table[] = {
588e71e0349SDaniel Lezcano 	{
5898d8354d2SPavel Emelyanov 		.procname	= "ip6frag_high_thresh",
590e31e0bdcSPavel Emelyanov 		.data		= &init_net.ipv6.frags.high_thresh,
5918d8354d2SPavel Emelyanov 		.maxlen		= sizeof(int),
5928d8354d2SPavel Emelyanov 		.mode		= 0644,
5931bab4c75SNikolay Aleksandrov 		.proc_handler	= proc_dointvec_minmax,
5941bab4c75SNikolay Aleksandrov 		.extra1		= &init_net.ipv6.frags.low_thresh
5958d8354d2SPavel Emelyanov 	},
5968d8354d2SPavel Emelyanov 	{
5978d8354d2SPavel Emelyanov 		.procname	= "ip6frag_low_thresh",
598e31e0bdcSPavel Emelyanov 		.data		= &init_net.ipv6.frags.low_thresh,
5998d8354d2SPavel Emelyanov 		.maxlen		= sizeof(int),
6008d8354d2SPavel Emelyanov 		.mode		= 0644,
6011bab4c75SNikolay Aleksandrov 		.proc_handler	= proc_dointvec_minmax,
6021bab4c75SNikolay Aleksandrov 		.extra1		= &zero,
6031bab4c75SNikolay Aleksandrov 		.extra2		= &init_net.ipv6.frags.high_thresh
6048d8354d2SPavel Emelyanov 	},
6058d8354d2SPavel Emelyanov 	{
6068d8354d2SPavel Emelyanov 		.procname	= "ip6frag_time",
607b2fd5321SPavel Emelyanov 		.data		= &init_net.ipv6.frags.timeout,
6088d8354d2SPavel Emelyanov 		.maxlen		= sizeof(int),
6098d8354d2SPavel Emelyanov 		.mode		= 0644,
6106d9f239aSAlexey Dobriyan 		.proc_handler	= proc_dointvec_jiffies,
6118d8354d2SPavel Emelyanov 	},
6127d291ebbSPavel Emelyanov 	{ }
6137d291ebbSPavel Emelyanov };
6147d291ebbSPavel Emelyanov 
615e3a57d18SFlorian Westphal /* secret interval has been deprecated */
616e3a57d18SFlorian Westphal static int ip6_frags_secret_interval_unused;
6177d291ebbSPavel Emelyanov static struct ctl_table ip6_frags_ctl_table[] = {
6188d8354d2SPavel Emelyanov 	{
6198d8354d2SPavel Emelyanov 		.procname	= "ip6frag_secret_interval",
620e3a57d18SFlorian Westphal 		.data		= &ip6_frags_secret_interval_unused,
6218d8354d2SPavel Emelyanov 		.maxlen		= sizeof(int),
6228d8354d2SPavel Emelyanov 		.mode		= 0644,
6236d9f239aSAlexey Dobriyan 		.proc_handler	= proc_dointvec_jiffies,
6248d8354d2SPavel Emelyanov 	},
6258d8354d2SPavel Emelyanov 	{ }
6268d8354d2SPavel Emelyanov };
6277d460db9SDaniel Lezcano 
6282c8c1e72SAlexey Dobriyan static int __net_init ip6_frags_ns_sysctl_register(struct net *net)
6298d8354d2SPavel Emelyanov {
630e4a2d5c2SPavel Emelyanov 	struct ctl_table *table;
6318d8354d2SPavel Emelyanov 	struct ctl_table_header *hdr;
6328d8354d2SPavel Emelyanov 
6330a64b4b8SPavel Emelyanov 	table = ip6_frags_ns_ctl_table;
63409ad9bc7SOctavian Purdila 	if (!net_eq(net, &init_net)) {
6350a64b4b8SPavel Emelyanov 		table = kmemdup(table, sizeof(ip6_frags_ns_ctl_table), GFP_KERNEL);
636e4a2d5c2SPavel Emelyanov 		if (table == NULL)
637e4a2d5c2SPavel Emelyanov 			goto err_alloc;
638e4a2d5c2SPavel Emelyanov 
639e31e0bdcSPavel Emelyanov 		table[0].data = &net->ipv6.frags.high_thresh;
6401bab4c75SNikolay Aleksandrov 		table[0].extra1 = &net->ipv6.frags.low_thresh;
6411bab4c75SNikolay Aleksandrov 		table[0].extra2 = &init_net.ipv6.frags.high_thresh;
642e31e0bdcSPavel Emelyanov 		table[1].data = &net->ipv6.frags.low_thresh;
6431bab4c75SNikolay Aleksandrov 		table[1].extra2 = &net->ipv6.frags.high_thresh;
644b2fd5321SPavel Emelyanov 		table[2].data = &net->ipv6.frags.timeout;
645464dc801SEric W. Biederman 
646464dc801SEric W. Biederman 		/* Don't export sysctls to unprivileged users */
647464dc801SEric W. Biederman 		if (net->user_ns != &init_user_ns)
648464dc801SEric W. Biederman 			table[0].procname = NULL;
649e4a2d5c2SPavel Emelyanov 	}
650e4a2d5c2SPavel Emelyanov 
651ec8f23ceSEric W. Biederman 	hdr = register_net_sysctl(net, "net/ipv6", table);
652e4a2d5c2SPavel Emelyanov 	if (hdr == NULL)
653e4a2d5c2SPavel Emelyanov 		goto err_reg;
654e4a2d5c2SPavel Emelyanov 
655e4a2d5c2SPavel Emelyanov 	net->ipv6.sysctl.frags_hdr = hdr;
656e4a2d5c2SPavel Emelyanov 	return 0;
657e4a2d5c2SPavel Emelyanov 
658e4a2d5c2SPavel Emelyanov err_reg:
65909ad9bc7SOctavian Purdila 	if (!net_eq(net, &init_net))
660e4a2d5c2SPavel Emelyanov 		kfree(table);
661e4a2d5c2SPavel Emelyanov err_alloc:
662e4a2d5c2SPavel Emelyanov 	return -ENOMEM;
663e4a2d5c2SPavel Emelyanov }
664e4a2d5c2SPavel Emelyanov 
6652c8c1e72SAlexey Dobriyan static void __net_exit ip6_frags_ns_sysctl_unregister(struct net *net)
666e4a2d5c2SPavel Emelyanov {
667e4a2d5c2SPavel Emelyanov 	struct ctl_table *table;
668e4a2d5c2SPavel Emelyanov 
669e4a2d5c2SPavel Emelyanov 	table = net->ipv6.sysctl.frags_hdr->ctl_table_arg;
670e4a2d5c2SPavel Emelyanov 	unregister_net_sysctl_table(net->ipv6.sysctl.frags_hdr);
6713705e11aSYang Hongyang 	if (!net_eq(net, &init_net))
672e4a2d5c2SPavel Emelyanov 		kfree(table);
6738d8354d2SPavel Emelyanov }
6747d291ebbSPavel Emelyanov 
6757d291ebbSPavel Emelyanov static struct ctl_table_header *ip6_ctl_header;
6767d291ebbSPavel Emelyanov 
6777d291ebbSPavel Emelyanov static int ip6_frags_sysctl_register(void)
6787d291ebbSPavel Emelyanov {
67943444757SEric W. Biederman 	ip6_ctl_header = register_net_sysctl(&init_net, "net/ipv6",
6807d291ebbSPavel Emelyanov 			ip6_frags_ctl_table);
6817d291ebbSPavel Emelyanov 	return ip6_ctl_header == NULL ? -ENOMEM : 0;
6827d291ebbSPavel Emelyanov }
6837d291ebbSPavel Emelyanov 
6847d291ebbSPavel Emelyanov static void ip6_frags_sysctl_unregister(void)
6857d291ebbSPavel Emelyanov {
6867d291ebbSPavel Emelyanov 	unregister_net_sysctl_table(ip6_ctl_header);
6877d291ebbSPavel Emelyanov }
6888d8354d2SPavel Emelyanov #else
6890a64b4b8SPavel Emelyanov static inline int ip6_frags_ns_sysctl_register(struct net *net)
6908d8354d2SPavel Emelyanov {
6918d8354d2SPavel Emelyanov 	return 0;
6928d8354d2SPavel Emelyanov }
693e4a2d5c2SPavel Emelyanov 
6940a64b4b8SPavel Emelyanov static inline void ip6_frags_ns_sysctl_unregister(struct net *net)
695e4a2d5c2SPavel Emelyanov {
696e4a2d5c2SPavel Emelyanov }
6977d291ebbSPavel Emelyanov 
6987d291ebbSPavel Emelyanov static inline int ip6_frags_sysctl_register(void)
6997d291ebbSPavel Emelyanov {
7007d291ebbSPavel Emelyanov 	return 0;
7017d291ebbSPavel Emelyanov }
7027d291ebbSPavel Emelyanov 
7037d291ebbSPavel Emelyanov static inline void ip6_frags_sysctl_unregister(void)
7047d291ebbSPavel Emelyanov {
7057d291ebbSPavel Emelyanov }
7068d8354d2SPavel Emelyanov #endif
7078d8354d2SPavel Emelyanov 
7082c8c1e72SAlexey Dobriyan static int __net_init ipv6_frags_init_net(struct net *net)
7098d8354d2SPavel Emelyanov {
7107c070aa9SShan Wei 	net->ipv6.frags.high_thresh = IPV6_FRAG_HIGH_THRESH;
7117c070aa9SShan Wei 	net->ipv6.frags.low_thresh = IPV6_FRAG_LOW_THRESH;
712b2fd5321SPavel Emelyanov 	net->ipv6.frags.timeout = IPV6_FRAG_TIMEOUT;
7138d8354d2SPavel Emelyanov 
714e5a2bb84SPavel Emelyanov 	inet_frags_init_net(&net->ipv6.frags);
715e5a2bb84SPavel Emelyanov 
7160a64b4b8SPavel Emelyanov 	return ip6_frags_ns_sysctl_register(net);
717e71e0349SDaniel Lezcano }
718e71e0349SDaniel Lezcano 
7192c8c1e72SAlexey Dobriyan static void __net_exit ipv6_frags_exit_net(struct net *net)
72081566e83SPavel Emelyanov {
7210a64b4b8SPavel Emelyanov 	ip6_frags_ns_sysctl_unregister(net);
72281566e83SPavel Emelyanov 	inet_frags_exit_net(&net->ipv6.frags, &ip6_frags);
72381566e83SPavel Emelyanov }
72481566e83SPavel Emelyanov 
72581566e83SPavel Emelyanov static struct pernet_operations ip6_frags_ops = {
72681566e83SPavel Emelyanov 	.init = ipv6_frags_init_net,
72781566e83SPavel Emelyanov 	.exit = ipv6_frags_exit_net,
72881566e83SPavel Emelyanov };
72981566e83SPavel Emelyanov 
730853cbbaaSDaniel Lezcano int __init ipv6_frag_init(void)
7311da177e4SLinus Torvalds {
732853cbbaaSDaniel Lezcano 	int ret;
7331da177e4SLinus Torvalds 
734853cbbaaSDaniel Lezcano 	ret = inet6_add_protocol(&frag_protocol, IPPROTO_FRAGMENT);
735853cbbaaSDaniel Lezcano 	if (ret)
736853cbbaaSDaniel Lezcano 		goto out;
737e71e0349SDaniel Lezcano 
7387d291ebbSPavel Emelyanov 	ret = ip6_frags_sysctl_register();
7397d291ebbSPavel Emelyanov 	if (ret)
7407d291ebbSPavel Emelyanov 		goto err_sysctl;
7417d291ebbSPavel Emelyanov 
7420002c630SPavel Emelyanov 	ret = register_pernet_subsys(&ip6_frags_ops);
7430002c630SPavel Emelyanov 	if (ret)
7440002c630SPavel Emelyanov 		goto err_pernet;
7458d8354d2SPavel Emelyanov 
746321a3a99SPavel Emelyanov 	ip6_frags.hashfn = ip6_hashfn;
747c6fda282SPavel Emelyanov 	ip6_frags.constructor = ip6_frag_init;
748c9547709SPavel Emelyanov 	ip6_frags.destructor = NULL;
7491e4b8287SPavel Emelyanov 	ip6_frags.skb_free = NULL;
7501e4b8287SPavel Emelyanov 	ip6_frags.qsize = sizeof(struct frag_queue);
751abd6523dSPavel Emelyanov 	ip6_frags.match = ip6_frag_match;
752e521db9dSPavel Emelyanov 	ip6_frags.frag_expire = ip6_frag_expire;
753d4ad4d22SNikolay Aleksandrov 	ip6_frags.frags_cache_name = ip6_frag_cache_name;
754d4ad4d22SNikolay Aleksandrov 	ret = inet_frags_init(&ip6_frags);
755d4ad4d22SNikolay Aleksandrov 	if (ret)
756d4ad4d22SNikolay Aleksandrov 		goto err_pernet;
757853cbbaaSDaniel Lezcano out:
758853cbbaaSDaniel Lezcano 	return ret;
7590002c630SPavel Emelyanov 
7600002c630SPavel Emelyanov err_pernet:
7617d291ebbSPavel Emelyanov 	ip6_frags_sysctl_unregister();
7627d291ebbSPavel Emelyanov err_sysctl:
7630002c630SPavel Emelyanov 	inet6_del_protocol(&frag_protocol, IPPROTO_FRAGMENT);
7640002c630SPavel Emelyanov 	goto out;
765853cbbaaSDaniel Lezcano }
766853cbbaaSDaniel Lezcano 
767853cbbaaSDaniel Lezcano void ipv6_frag_exit(void)
768853cbbaaSDaniel Lezcano {
769853cbbaaSDaniel Lezcano 	inet_frags_fini(&ip6_frags);
7707d291ebbSPavel Emelyanov 	ip6_frags_sysctl_unregister();
77181566e83SPavel Emelyanov 	unregister_pernet_subsys(&ip6_frags_ops);
772853cbbaaSDaniel Lezcano 	inet6_del_protocol(&frag_protocol, IPPROTO_FRAGMENT);
7731da177e4SLinus Torvalds }
774