xref: /openbmc/linux/net/ipv6/reassembly.c (revision 06aa8b8a)
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 
631da177e4SLinus Torvalds struct ip6frag_skb_cb
641da177e4SLinus Torvalds {
651da177e4SLinus Torvalds 	struct inet6_skb_parm	h;
661da177e4SLinus Torvalds 	int			offset;
671da177e4SLinus Torvalds };
681da177e4SLinus Torvalds 
691da177e4SLinus Torvalds #define FRAG6_CB(skb)	((struct ip6frag_skb_cb*)((skb)->cb))
701da177e4SLinus Torvalds 
71eec2e618SHannes Frederic Sowa static inline u8 ip6_frag_ecn(const struct ipv6hdr *ipv6h)
72eec2e618SHannes Frederic Sowa {
73eec2e618SHannes Frederic Sowa 	return 1 << (ipv6_get_dsfield(ipv6h) & INET_ECN_MASK);
74eec2e618SHannes Frederic Sowa }
751da177e4SLinus Torvalds 
767eb95156SPavel Emelyanov static struct inet_frags ip6_frags;
771da177e4SLinus Torvalds 
78f61944efSHerbert Xu static int ip6_frag_reasm(struct frag_queue *fq, struct sk_buff *prev,
79f61944efSHerbert Xu 			  struct net_device *dev);
80f61944efSHerbert Xu 
81f6596f9dSZach Brown /*
82f6596f9dSZach Brown  * callers should be careful not to use the hash value outside the ipfrag_lock
83f6596f9dSZach Brown  * as doing so could race with ipfrag_hash_rnd being recalculated.
84f6596f9dSZach Brown  */
85b1190570SHannes Frederic Sowa static unsigned int inet6_hash_frag(__be32 id, const struct in6_addr *saddr,
86b1190570SHannes Frederic Sowa 				    const struct in6_addr *daddr)
871da177e4SLinus Torvalds {
88b1190570SHannes Frederic Sowa 	net_get_random_once(&ip6_frags.rnd, sizeof(ip6_frags.rnd));
89fb3cfe6eSFlorian Westphal 	return jhash_3words(ipv6_addr_hash(saddr), ipv6_addr_hash(daddr),
90b1190570SHannes Frederic Sowa 			    (__force u32)id, ip6_frags.rnd);
911da177e4SLinus Torvalds }
921da177e4SLinus Torvalds 
9336c77782SFlorian Westphal static unsigned int ip6_hashfn(const struct inet_frag_queue *q)
941da177e4SLinus Torvalds {
9536c77782SFlorian Westphal 	const struct frag_queue *fq;
961da177e4SLinus Torvalds 
97321a3a99SPavel Emelyanov 	fq = container_of(q, struct frag_queue, q);
98b1190570SHannes Frederic Sowa 	return inet6_hash_frag(fq->id, &fq->saddr, &fq->daddr);
991da177e4SLinus Torvalds }
1001da177e4SLinus Torvalds 
10136c77782SFlorian Westphal bool ip6_frag_match(const struct inet_frag_queue *q, const void *a)
102abd6523dSPavel Emelyanov {
10336c77782SFlorian Westphal 	const struct frag_queue *fq;
10436c77782SFlorian Westphal 	const struct ip6_create_arg *arg = a;
105abd6523dSPavel Emelyanov 
106abd6523dSPavel Emelyanov 	fq = container_of(q, struct frag_queue, q);
107cbc264caSEric Dumazet 	return	fq->id == arg->id &&
108cbc264caSEric Dumazet 		fq->user == arg->user &&
109abd6523dSPavel Emelyanov 		ipv6_addr_equal(&fq->saddr, arg->src) &&
110cbc264caSEric Dumazet 		ipv6_addr_equal(&fq->daddr, arg->dst);
111abd6523dSPavel Emelyanov }
112abd6523dSPavel Emelyanov EXPORT_SYMBOL(ip6_frag_match);
113abd6523dSPavel Emelyanov 
11436c77782SFlorian Westphal void ip6_frag_init(struct inet_frag_queue *q, const void *a)
115c6fda282SPavel Emelyanov {
116c6fda282SPavel Emelyanov 	struct frag_queue *fq = container_of(q, struct frag_queue, q);
11736c77782SFlorian Westphal 	const struct ip6_create_arg *arg = a;
118c6fda282SPavel Emelyanov 
119c6fda282SPavel Emelyanov 	fq->id = arg->id;
1200b5ccb2eSPatrick McHardy 	fq->user = arg->user;
1214e3fd7a0SAlexey Dobriyan 	fq->saddr = *arg->src;
1224e3fd7a0SAlexey Dobriyan 	fq->daddr = *arg->dst;
123eec2e618SHannes Frederic Sowa 	fq->ecn = arg->ecn;
124c6fda282SPavel Emelyanov }
125c6fda282SPavel Emelyanov EXPORT_SYMBOL(ip6_frag_init);
126c6fda282SPavel Emelyanov 
127b836c99fSAmerigo Wang void ip6_expire_frag_queue(struct net *net, struct frag_queue *fq,
128b836c99fSAmerigo Wang 			   struct inet_frags *frags)
1291da177e4SLinus Torvalds {
130a11d206dSYOSHIFUJI Hideaki 	struct net_device *dev = NULL;
131e521db9dSPavel Emelyanov 
1325ab11c98SPavel Emelyanov 	spin_lock(&fq->q.lock);
1331da177e4SLinus Torvalds 
13406aa8b8aSNikolay Aleksandrov 	if (fq->q.flags & INET_FRAG_COMPLETE)
1351da177e4SLinus Torvalds 		goto out;
1361da177e4SLinus Torvalds 
137b836c99fSAmerigo Wang 	inet_frag_kill(&fq->q, frags);
1381da177e4SLinus Torvalds 
139a11d206dSYOSHIFUJI Hideaki 	rcu_read_lock();
14069df9d59SEric Dumazet 	dev = dev_get_by_index_rcu(net, fq->iif);
14169df9d59SEric Dumazet 	if (!dev)
14269df9d59SEric Dumazet 		goto out_rcu_unlock;
14369df9d59SEric Dumazet 
14406aa8b8aSNikolay Aleksandrov 	if (!(fq->q.flags & INET_FRAG_EVICTED))
145b13d3cbfSFlorian Westphal 		IP6_INC_STATS_BH(net, __in6_dev_get(dev),
146b13d3cbfSFlorian Westphal 				 IPSTATS_MIB_REASMTIMEOUT);
147483a47d2SDenis V. Lunev 	IP6_INC_STATS_BH(net, __in6_dev_get(dev), IPSTATS_MIB_REASMFAILS);
1481da177e4SLinus Torvalds 
14978c784c4SIngo Oeser 	/* Don't send error if the first segment did not arrive. */
15006aa8b8aSNikolay Aleksandrov 	if (!(fq->q.flags & INET_FRAG_FIRST_IN) || !fq->q.fragments)
15169df9d59SEric Dumazet 		goto out_rcu_unlock;
15278c784c4SIngo Oeser 
1531da177e4SLinus Torvalds 	/*
1541da177e4SLinus Torvalds 	   But use as source device on which LAST ARRIVED
1551da177e4SLinus Torvalds 	   segment was received. And do not use fq->dev
1561da177e4SLinus Torvalds 	   pointer directly, device might already disappeared.
1571da177e4SLinus Torvalds 	 */
1585ab11c98SPavel Emelyanov 	fq->q.fragments->dev = dev;
1593ffe533cSAlexey Dobriyan 	icmpv6_send(fq->q.fragments, ICMPV6_TIME_EXCEED, ICMPV6_EXC_FRAGTIME, 0);
16069df9d59SEric Dumazet out_rcu_unlock:
16169df9d59SEric Dumazet 	rcu_read_unlock();
1621da177e4SLinus Torvalds out:
1635ab11c98SPavel Emelyanov 	spin_unlock(&fq->q.lock);
164b836c99fSAmerigo Wang 	inet_frag_put(&fq->q, frags);
165b836c99fSAmerigo Wang }
166b836c99fSAmerigo Wang EXPORT_SYMBOL(ip6_expire_frag_queue);
167b836c99fSAmerigo Wang 
168b836c99fSAmerigo Wang static void ip6_frag_expire(unsigned long data)
169b836c99fSAmerigo Wang {
170b836c99fSAmerigo Wang 	struct frag_queue *fq;
171b836c99fSAmerigo Wang 	struct net *net;
172b836c99fSAmerigo Wang 
173b836c99fSAmerigo Wang 	fq = container_of((struct inet_frag_queue *)data, struct frag_queue, q);
174b836c99fSAmerigo Wang 	net = container_of(fq->q.net, struct net, ipv6.frags);
175b836c99fSAmerigo Wang 
176b836c99fSAmerigo Wang 	ip6_expire_frag_queue(net, fq, &ip6_frags);
1771da177e4SLinus Torvalds }
1781da177e4SLinus Torvalds 
179abd6523dSPavel Emelyanov static __inline__ struct frag_queue *
180eec2e618SHannes Frederic Sowa fq_find(struct net *net, __be32 id, const struct in6_addr *src,
181eec2e618SHannes Frederic Sowa 	const struct in6_addr *dst, u8 ecn)
1821da177e4SLinus Torvalds {
183c6fda282SPavel Emelyanov 	struct inet_frag_queue *q;
184c6fda282SPavel Emelyanov 	struct ip6_create_arg arg;
185abd6523dSPavel Emelyanov 	unsigned int hash;
1861da177e4SLinus Torvalds 
187c6fda282SPavel Emelyanov 	arg.id = id;
1880b5ccb2eSPatrick McHardy 	arg.user = IP6_DEFRAG_LOCAL_DELIVER;
189c6fda282SPavel Emelyanov 	arg.src = src;
190c6fda282SPavel Emelyanov 	arg.dst = dst;
191eec2e618SHannes Frederic Sowa 	arg.ecn = ecn;
1929a375803SPavel Emelyanov 
193b1190570SHannes Frederic Sowa 	hash = inet6_hash_frag(id, src, dst);
194c6fda282SPavel Emelyanov 
195ac18e750SPavel Emelyanov 	q = inet_frag_find(&net->ipv6.frags, &ip6_frags, &arg, hash);
1965a3da1feSHannes Frederic Sowa 	if (IS_ERR_OR_NULL(q)) {
1975a3da1feSHannes Frederic Sowa 		inet_frag_maybe_warn_overflow(q, pr_fmt());
1989546377cSShan Wei 		return NULL;
1995a3da1feSHannes Frederic Sowa 	}
200c6fda282SPavel Emelyanov 	return container_of(q, struct frag_queue, q);
2011da177e4SLinus Torvalds }
2021da177e4SLinus Torvalds 
203f61944efSHerbert Xu static int ip6_frag_queue(struct frag_queue *fq, struct sk_buff *skb,
2041da177e4SLinus Torvalds 			   struct frag_hdr *fhdr, int nhoff)
2051da177e4SLinus Torvalds {
2061da177e4SLinus Torvalds 	struct sk_buff *prev, *next;
207f61944efSHerbert Xu 	struct net_device *dev;
2081da177e4SLinus Torvalds 	int offset, end;
209adf30907SEric Dumazet 	struct net *net = dev_net(skb_dst(skb)->dev);
210eec2e618SHannes Frederic Sowa 	u8 ecn;
2111da177e4SLinus Torvalds 
21206aa8b8aSNikolay Aleksandrov 	if (fq->q.flags & INET_FRAG_COMPLETE)
2131da177e4SLinus Torvalds 		goto err;
2141da177e4SLinus Torvalds 
2151da177e4SLinus Torvalds 	offset = ntohs(fhdr->frag_off) & ~0x7;
2160660e03fSArnaldo Carvalho de Melo 	end = offset + (ntohs(ipv6_hdr(skb)->payload_len) -
2170660e03fSArnaldo Carvalho de Melo 			((u8 *)(fhdr + 1) - (u8 *)(ipv6_hdr(skb) + 1)));
2181da177e4SLinus Torvalds 
2191da177e4SLinus Torvalds 	if ((unsigned int)end > IPV6_MAXPLEN) {
220adf30907SEric Dumazet 		IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)),
221a11d206dSYOSHIFUJI Hideaki 				 IPSTATS_MIB_INHDRERRORS);
222d56f90a7SArnaldo Carvalho de Melo 		icmpv6_param_prob(skb, ICMPV6_HDR_FIELD,
223d56f90a7SArnaldo Carvalho de Melo 				  ((u8 *)&fhdr->frag_off -
224d56f90a7SArnaldo Carvalho de Melo 				   skb_network_header(skb)));
225f61944efSHerbert Xu 		return -1;
2261da177e4SLinus Torvalds 	}
2271da177e4SLinus Torvalds 
228eec2e618SHannes Frederic Sowa 	ecn = ip6_frag_ecn(ipv6_hdr(skb));
229eec2e618SHannes Frederic Sowa 
230d56f90a7SArnaldo Carvalho de Melo 	if (skb->ip_summed == CHECKSUM_COMPLETE) {
231d56f90a7SArnaldo Carvalho de Melo 		const unsigned char *nh = skb_network_header(skb);
2321da177e4SLinus Torvalds 		skb->csum = csum_sub(skb->csum,
233d56f90a7SArnaldo Carvalho de Melo 				     csum_partial(nh, (u8 *)(fhdr + 1) - nh,
234d56f90a7SArnaldo Carvalho de Melo 						  0));
235d56f90a7SArnaldo Carvalho de Melo 	}
2361da177e4SLinus Torvalds 
2371da177e4SLinus Torvalds 	/* Is this the final fragment? */
2381da177e4SLinus Torvalds 	if (!(fhdr->frag_off & htons(IP6_MF))) {
2391da177e4SLinus Torvalds 		/* If we already have some bits beyond end
2401da177e4SLinus Torvalds 		 * or have different end, the segment is corrupted.
2411da177e4SLinus Torvalds 		 */
2425ab11c98SPavel Emelyanov 		if (end < fq->q.len ||
24306aa8b8aSNikolay Aleksandrov 		    ((fq->q.flags & INET_FRAG_LAST_IN) && end != fq->q.len))
2441da177e4SLinus Torvalds 			goto err;
24506aa8b8aSNikolay Aleksandrov 		fq->q.flags |= INET_FRAG_LAST_IN;
2465ab11c98SPavel Emelyanov 		fq->q.len = end;
2471da177e4SLinus Torvalds 	} else {
2481da177e4SLinus Torvalds 		/* Check if the fragment is rounded to 8 bytes.
2491da177e4SLinus Torvalds 		 * Required by the RFC.
2501da177e4SLinus Torvalds 		 */
2511da177e4SLinus Torvalds 		if (end & 0x7) {
2521da177e4SLinus Torvalds 			/* RFC2460 says always send parameter problem in
2531da177e4SLinus Torvalds 			 * this case. -DaveM
2541da177e4SLinus Torvalds 			 */
255adf30907SEric Dumazet 			IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)),
256a11d206dSYOSHIFUJI Hideaki 					 IPSTATS_MIB_INHDRERRORS);
2571da177e4SLinus Torvalds 			icmpv6_param_prob(skb, ICMPV6_HDR_FIELD,
2581da177e4SLinus Torvalds 					  offsetof(struct ipv6hdr, payload_len));
259f61944efSHerbert Xu 			return -1;
2601da177e4SLinus Torvalds 		}
2615ab11c98SPavel Emelyanov 		if (end > fq->q.len) {
2621da177e4SLinus Torvalds 			/* Some bits beyond end -> corruption. */
26306aa8b8aSNikolay Aleksandrov 			if (fq->q.flags & INET_FRAG_LAST_IN)
2641da177e4SLinus Torvalds 				goto err;
2655ab11c98SPavel Emelyanov 			fq->q.len = end;
2661da177e4SLinus Torvalds 		}
2671da177e4SLinus Torvalds 	}
2681da177e4SLinus Torvalds 
2691da177e4SLinus Torvalds 	if (end == offset)
2701da177e4SLinus Torvalds 		goto err;
2711da177e4SLinus Torvalds 
2721da177e4SLinus Torvalds 	/* Point into the IP datagram 'data' part. */
2731da177e4SLinus Torvalds 	if (!pskb_pull(skb, (u8 *) (fhdr + 1) - skb->data))
2741da177e4SLinus Torvalds 		goto err;
27542ca89c1SStephen Hemminger 
27642ca89c1SStephen Hemminger 	if (pskb_trim_rcsum(skb, end - offset))
2771da177e4SLinus Torvalds 		goto err;
2781da177e4SLinus Torvalds 
2791da177e4SLinus Torvalds 	/* Find out which fragments are in front and at the back of us
2801da177e4SLinus Torvalds 	 * in the chain of fragments so far.  We must know where to put
2811da177e4SLinus Torvalds 	 * this fragment, right?
2821da177e4SLinus Torvalds 	 */
283d6bebca9SChangli Gao 	prev = fq->q.fragments_tail;
284d6bebca9SChangli Gao 	if (!prev || FRAG6_CB(prev)->offset < offset) {
285d6bebca9SChangli Gao 		next = NULL;
286d6bebca9SChangli Gao 		goto found;
287d6bebca9SChangli Gao 	}
2881da177e4SLinus Torvalds 	prev = NULL;
2895ab11c98SPavel Emelyanov 	for(next = fq->q.fragments; next != NULL; next = next->next) {
2901da177e4SLinus Torvalds 		if (FRAG6_CB(next)->offset >= offset)
2911da177e4SLinus Torvalds 			break;	/* bingo! */
2921da177e4SLinus Torvalds 		prev = next;
2931da177e4SLinus Torvalds 	}
2941da177e4SLinus Torvalds 
295d6bebca9SChangli Gao found:
2965de658f8SEric Dumazet 	/* RFC5722, Section 4, amended by Errata ID : 3089
29770789d70SNicolas Dichtel 	 *                          When reassembling an IPv6 datagram, if
29870789d70SNicolas Dichtel 	 *   one or more its constituent fragments is determined to be an
29970789d70SNicolas Dichtel 	 *   overlapping fragment, the entire datagram (and any constituent
3005de658f8SEric Dumazet 	 *   fragments) MUST be silently discarded.
3011da177e4SLinus Torvalds 	 */
3021da177e4SLinus Torvalds 
30370789d70SNicolas Dichtel 	/* Check for overlap with preceding fragment. */
30470789d70SNicolas Dichtel 	if (prev &&
305f4642141SShan Wei 	    (FRAG6_CB(prev)->offset + prev->len) > offset)
30670789d70SNicolas Dichtel 		goto discard_fq;
3071da177e4SLinus Torvalds 
30870789d70SNicolas Dichtel 	/* Look for overlap with succeeding segment. */
30970789d70SNicolas Dichtel 	if (next && FRAG6_CB(next)->offset < end)
31070789d70SNicolas Dichtel 		goto discard_fq;
3111da177e4SLinus Torvalds 
3121da177e4SLinus Torvalds 	FRAG6_CB(skb)->offset = offset;
3131da177e4SLinus Torvalds 
3141da177e4SLinus Torvalds 	/* Insert this fragment in the chain of fragments. */
3151da177e4SLinus Torvalds 	skb->next = next;
316d6bebca9SChangli Gao 	if (!next)
317d6bebca9SChangli Gao 		fq->q.fragments_tail = skb;
3181da177e4SLinus Torvalds 	if (prev)
3191da177e4SLinus Torvalds 		prev->next = skb;
3201da177e4SLinus Torvalds 	else
3215ab11c98SPavel Emelyanov 		fq->q.fragments = skb;
3221da177e4SLinus Torvalds 
323f61944efSHerbert Xu 	dev = skb->dev;
324f61944efSHerbert Xu 	if (dev) {
325f61944efSHerbert Xu 		fq->iif = dev->ifindex;
3261da177e4SLinus Torvalds 		skb->dev = NULL;
327f61944efSHerbert Xu 	}
3285ab11c98SPavel Emelyanov 	fq->q.stamp = skb->tstamp;
3295ab11c98SPavel Emelyanov 	fq->q.meat += skb->len;
330eec2e618SHannes Frederic Sowa 	fq->ecn |= ecn;
331d433673eSJesper Dangaard Brouer 	add_frag_mem_limit(&fq->q, skb->truesize);
3321da177e4SLinus Torvalds 
3331da177e4SLinus Torvalds 	/* The first fragment.
3341da177e4SLinus Torvalds 	 * nhoffset is obtained from the first fragment, of course.
3351da177e4SLinus Torvalds 	 */
3361da177e4SLinus Torvalds 	if (offset == 0) {
3371da177e4SLinus Torvalds 		fq->nhoffset = nhoff;
33806aa8b8aSNikolay Aleksandrov 		fq->q.flags |= INET_FRAG_FIRST_IN;
3391da177e4SLinus Torvalds 	}
340f61944efSHerbert Xu 
34106aa8b8aSNikolay Aleksandrov 	if (fq->q.flags == (INET_FRAG_FIRST_IN | INET_FRAG_LAST_IN) &&
34297599dc7SEric Dumazet 	    fq->q.meat == fq->q.len) {
34397599dc7SEric Dumazet 		int res;
34497599dc7SEric Dumazet 		unsigned long orefdst = skb->_skb_refdst;
345f61944efSHerbert Xu 
34697599dc7SEric Dumazet 		skb->_skb_refdst = 0UL;
34797599dc7SEric Dumazet 		res = ip6_frag_reasm(fq, prev, dev);
34897599dc7SEric Dumazet 		skb->_skb_refdst = orefdst;
34997599dc7SEric Dumazet 		return res;
35097599dc7SEric Dumazet 	}
35197599dc7SEric Dumazet 
35297599dc7SEric Dumazet 	skb_dst_drop(skb);
353f61944efSHerbert Xu 	return -1;
3541da177e4SLinus Torvalds 
35570789d70SNicolas Dichtel discard_fq:
356b836c99fSAmerigo Wang 	inet_frag_kill(&fq->q, &ip6_frags);
3571da177e4SLinus Torvalds err:
358d2373862SNikolay Aleksandrov 	IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)),
3593bd653c8SDenis V. Lunev 			 IPSTATS_MIB_REASMFAILS);
3601da177e4SLinus Torvalds 	kfree_skb(skb);
361f61944efSHerbert Xu 	return -1;
3621da177e4SLinus Torvalds }
3631da177e4SLinus Torvalds 
3641da177e4SLinus Torvalds /*
3651da177e4SLinus Torvalds  *	Check if this packet is complete.
3661da177e4SLinus Torvalds  *	Returns NULL on failure by any reason, and pointer
3671da177e4SLinus Torvalds  *	to current nexthdr field in reassembled frame.
3681da177e4SLinus Torvalds  *
3691da177e4SLinus Torvalds  *	It is called with locked fq, and caller must check that
3701da177e4SLinus Torvalds  *	queue is eligible for reassembly i.e. it is not COMPLETE,
3711da177e4SLinus Torvalds  *	the last and the first frames arrived and all the bits are here.
3721da177e4SLinus Torvalds  */
373f61944efSHerbert Xu static int ip6_frag_reasm(struct frag_queue *fq, struct sk_buff *prev,
3741da177e4SLinus Torvalds 			  struct net_device *dev)
3751da177e4SLinus Torvalds {
3762bad35b7SJorge Boncompte [DTI2] 	struct net *net = container_of(fq->q.net, struct net, ipv6.frags);
3775ab11c98SPavel Emelyanov 	struct sk_buff *fp, *head = fq->q.fragments;
3781da177e4SLinus Torvalds 	int    payload_len;
3791da177e4SLinus Torvalds 	unsigned int nhoff;
380ec16439eSEric Dumazet 	int sum_truesize;
381eec2e618SHannes Frederic Sowa 	u8 ecn;
3821da177e4SLinus Torvalds 
383b836c99fSAmerigo Wang 	inet_frag_kill(&fq->q, &ip6_frags);
3841da177e4SLinus Torvalds 
385eec2e618SHannes Frederic Sowa 	ecn = ip_frag_ecn_table[fq->ecn];
386eec2e618SHannes Frederic Sowa 	if (unlikely(ecn == 0xff))
387eec2e618SHannes Frederic Sowa 		goto out_fail;
388eec2e618SHannes Frederic Sowa 
389f61944efSHerbert Xu 	/* Make the one we just received the head. */
390f61944efSHerbert Xu 	if (prev) {
391f61944efSHerbert Xu 		head = prev->next;
392f61944efSHerbert Xu 		fp = skb_clone(head, GFP_ATOMIC);
393f61944efSHerbert Xu 
394f61944efSHerbert Xu 		if (!fp)
395f61944efSHerbert Xu 			goto out_oom;
396f61944efSHerbert Xu 
397f61944efSHerbert Xu 		fp->next = head->next;
398d6bebca9SChangli Gao 		if (!fp->next)
399d6bebca9SChangli Gao 			fq->q.fragments_tail = fp;
400f61944efSHerbert Xu 		prev->next = fp;
401f61944efSHerbert Xu 
4025ab11c98SPavel Emelyanov 		skb_morph(head, fq->q.fragments);
4035ab11c98SPavel Emelyanov 		head->next = fq->q.fragments->next;
404f61944efSHerbert Xu 
405808db80aSEric Dumazet 		consume_skb(fq->q.fragments);
4065ab11c98SPavel Emelyanov 		fq->q.fragments = head;
407f61944efSHerbert Xu 	}
408f61944efSHerbert Xu 
409547b792cSIlpo Järvinen 	WARN_ON(head == NULL);
410547b792cSIlpo Järvinen 	WARN_ON(FRAG6_CB(head)->offset != 0);
4111da177e4SLinus Torvalds 
4121da177e4SLinus Torvalds 	/* Unfragmented part is taken from the first segment. */
413d56f90a7SArnaldo Carvalho de Melo 	payload_len = ((head->data - skb_network_header(head)) -
4145ab11c98SPavel Emelyanov 		       sizeof(struct ipv6hdr) + fq->q.len -
415d56f90a7SArnaldo Carvalho de Melo 		       sizeof(struct frag_hdr));
4161da177e4SLinus Torvalds 	if (payload_len > IPV6_MAXPLEN)
4171da177e4SLinus Torvalds 		goto out_oversize;
4181da177e4SLinus Torvalds 
4191da177e4SLinus Torvalds 	/* Head of list must not be cloned. */
42014bbd6a5SPravin B Shelar 	if (skb_unclone(head, GFP_ATOMIC))
4211da177e4SLinus Torvalds 		goto out_oom;
4221da177e4SLinus Torvalds 
4231da177e4SLinus Torvalds 	/* If the first fragment is fragmented itself, we split
4241da177e4SLinus Torvalds 	 * it to two chunks: the first with data and paged part
4251da177e4SLinus Torvalds 	 * and the second, holding only fragments. */
42621dc3301SDavid S. Miller 	if (skb_has_frag_list(head)) {
4271da177e4SLinus Torvalds 		struct sk_buff *clone;
4281da177e4SLinus Torvalds 		int i, plen = 0;
4291da177e4SLinus Torvalds 
4301da177e4SLinus Torvalds 		if ((clone = alloc_skb(0, GFP_ATOMIC)) == NULL)
4311da177e4SLinus Torvalds 			goto out_oom;
4321da177e4SLinus Torvalds 		clone->next = head->next;
4331da177e4SLinus Torvalds 		head->next = clone;
4341da177e4SLinus Torvalds 		skb_shinfo(clone)->frag_list = skb_shinfo(head)->frag_list;
4354d9092bbSDavid S. Miller 		skb_frag_list_init(head);
4361da177e4SLinus Torvalds 		for (i = 0; i < skb_shinfo(head)->nr_frags; i++)
4379e903e08SEric Dumazet 			plen += skb_frag_size(&skb_shinfo(head)->frags[i]);
4381da177e4SLinus Torvalds 		clone->len = clone->data_len = head->data_len - plen;
4391da177e4SLinus Torvalds 		head->data_len -= clone->len;
4401da177e4SLinus Torvalds 		head->len -= clone->len;
4411da177e4SLinus Torvalds 		clone->csum = 0;
4421da177e4SLinus Torvalds 		clone->ip_summed = head->ip_summed;
443d433673eSJesper Dangaard Brouer 		add_frag_mem_limit(&fq->q, clone->truesize);
4441da177e4SLinus Torvalds 	}
4451da177e4SLinus Torvalds 
4461da177e4SLinus Torvalds 	/* We have to remove fragment header from datagram and to relocate
4471da177e4SLinus Torvalds 	 * header in order to calculate ICV correctly. */
4481da177e4SLinus Torvalds 	nhoff = fq->nhoffset;
449b0e380b1SArnaldo Carvalho de Melo 	skb_network_header(head)[nhoff] = skb_transport_header(head)[0];
4501da177e4SLinus Torvalds 	memmove(head->head + sizeof(struct frag_hdr), head->head,
4511da177e4SLinus Torvalds 		(head->data - head->head) - sizeof(struct frag_hdr));
452b0e380b1SArnaldo Carvalho de Melo 	head->mac_header += sizeof(struct frag_hdr);
453b0e380b1SArnaldo Carvalho de Melo 	head->network_header += sizeof(struct frag_hdr);
4541da177e4SLinus Torvalds 
455badff6d0SArnaldo Carvalho de Melo 	skb_reset_transport_header(head);
456d56f90a7SArnaldo Carvalho de Melo 	skb_push(head, head->data - skb_network_header(head));
4571da177e4SLinus Torvalds 
458ec16439eSEric Dumazet 	sum_truesize = head->truesize;
459ec16439eSEric Dumazet 	for (fp = head->next; fp;) {
460ec16439eSEric Dumazet 		bool headstolen;
461ec16439eSEric Dumazet 		int delta;
462ec16439eSEric Dumazet 		struct sk_buff *next = fp->next;
463ec16439eSEric Dumazet 
464ec16439eSEric Dumazet 		sum_truesize += fp->truesize;
4651da177e4SLinus Torvalds 		if (head->ip_summed != fp->ip_summed)
4661da177e4SLinus Torvalds 			head->ip_summed = CHECKSUM_NONE;
46784fa7933SPatrick McHardy 		else if (head->ip_summed == CHECKSUM_COMPLETE)
4681da177e4SLinus Torvalds 			head->csum = csum_add(head->csum, fp->csum);
469ec16439eSEric Dumazet 
470ec16439eSEric Dumazet 		if (skb_try_coalesce(head, fp, &headstolen, &delta)) {
471ec16439eSEric Dumazet 			kfree_skb_partial(fp, headstolen);
472ec16439eSEric Dumazet 		} else {
473ec16439eSEric Dumazet 			if (!skb_shinfo(head)->frag_list)
474ec16439eSEric Dumazet 				skb_shinfo(head)->frag_list = fp;
475ec16439eSEric Dumazet 			head->data_len += fp->len;
476ec16439eSEric Dumazet 			head->len += fp->len;
4771da177e4SLinus Torvalds 			head->truesize += fp->truesize;
4781da177e4SLinus Torvalds 		}
479ec16439eSEric Dumazet 		fp = next;
480ec16439eSEric Dumazet 	}
481d433673eSJesper Dangaard Brouer 	sub_frag_mem_limit(&fq->q, sum_truesize);
4821da177e4SLinus Torvalds 
4831da177e4SLinus Torvalds 	head->next = NULL;
4841da177e4SLinus Torvalds 	head->dev = dev;
4855ab11c98SPavel Emelyanov 	head->tstamp = fq->q.stamp;
4860660e03fSArnaldo Carvalho de Melo 	ipv6_hdr(head)->payload_len = htons(payload_len);
487eec2e618SHannes Frederic Sowa 	ipv6_change_dsfield(ipv6_hdr(head), 0xff, ecn);
488951dbc8aSPatrick McHardy 	IP6CB(head)->nhoff = nhoff;
489f46078cfSHannes Frederic Sowa 	IP6CB(head)->flags |= IP6SKB_FRAGMENTED;
4901da177e4SLinus Torvalds 
4911da177e4SLinus Torvalds 	/* Yes, and fold redundant checksum back. 8) */
49284fa7933SPatrick McHardy 	if (head->ip_summed == CHECKSUM_COMPLETE)
493d56f90a7SArnaldo Carvalho de Melo 		head->csum = csum_partial(skb_network_header(head),
494cfe1fc77SArnaldo Carvalho de Melo 					  skb_network_header_len(head),
495d56f90a7SArnaldo Carvalho de Melo 					  head->csum);
4961da177e4SLinus Torvalds 
497a11d206dSYOSHIFUJI Hideaki 	rcu_read_lock();
4982bad35b7SJorge Boncompte [DTI2] 	IP6_INC_STATS_BH(net, __in6_dev_get(dev), IPSTATS_MIB_REASMOKS);
499a11d206dSYOSHIFUJI Hideaki 	rcu_read_unlock();
5005ab11c98SPavel Emelyanov 	fq->q.fragments = NULL;
501d6bebca9SChangli Gao 	fq->q.fragments_tail = NULL;
5021da177e4SLinus Torvalds 	return 1;
5031da177e4SLinus Torvalds 
5041da177e4SLinus Torvalds out_oversize:
505e87cc472SJoe Perches 	net_dbg_ratelimited("ip6_frag_reasm: payload len = %d\n", payload_len);
5061da177e4SLinus Torvalds 	goto out_fail;
5071da177e4SLinus Torvalds out_oom:
508e87cc472SJoe Perches 	net_dbg_ratelimited("ip6_frag_reasm: no memory for reassembly\n");
5091da177e4SLinus Torvalds out_fail:
510a11d206dSYOSHIFUJI Hideaki 	rcu_read_lock();
5112bad35b7SJorge Boncompte [DTI2] 	IP6_INC_STATS_BH(net, __in6_dev_get(dev), IPSTATS_MIB_REASMFAILS);
512a11d206dSYOSHIFUJI Hideaki 	rcu_read_unlock();
5131da177e4SLinus Torvalds 	return -1;
5141da177e4SLinus Torvalds }
5151da177e4SLinus Torvalds 
516e5bbef20SHerbert Xu static int ipv6_frag_rcv(struct sk_buff *skb)
5171da177e4SLinus Torvalds {
5181da177e4SLinus Torvalds 	struct frag_hdr *fhdr;
5191da177e4SLinus Torvalds 	struct frag_queue *fq;
520b71d1d42SEric Dumazet 	const struct ipv6hdr *hdr = ipv6_hdr(skb);
521adf30907SEric Dumazet 	struct net *net = dev_net(skb_dst(skb)->dev);
5221da177e4SLinus Torvalds 
523f46078cfSHannes Frederic Sowa 	if (IP6CB(skb)->flags & IP6SKB_FRAGMENTED)
524f46078cfSHannes Frederic Sowa 		goto fail_hdr;
525f46078cfSHannes Frederic Sowa 
526adf30907SEric Dumazet 	IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)), IPSTATS_MIB_REASMREQDS);
5271da177e4SLinus Torvalds 
5281da177e4SLinus Torvalds 	/* Jumbo payload inhibits frag. header */
52998b3377cSDenis V. Lunev 	if (hdr->payload_len==0)
53098b3377cSDenis V. Lunev 		goto fail_hdr;
53198b3377cSDenis V. Lunev 
532ea2ae17dSArnaldo Carvalho de Melo 	if (!pskb_may_pull(skb, (skb_transport_offset(skb) +
53398b3377cSDenis V. Lunev 				 sizeof(struct frag_hdr))))
53498b3377cSDenis V. Lunev 		goto fail_hdr;
5351da177e4SLinus Torvalds 
5360660e03fSArnaldo Carvalho de Melo 	hdr = ipv6_hdr(skb);
5379c70220bSArnaldo Carvalho de Melo 	fhdr = (struct frag_hdr *)skb_transport_header(skb);
5381da177e4SLinus Torvalds 
5391da177e4SLinus Torvalds 	if (!(fhdr->frag_off & htons(0xFFF9))) {
5401da177e4SLinus Torvalds 		/* It is not a fragmented frame */
541b0e380b1SArnaldo Carvalho de Melo 		skb->transport_header += sizeof(struct frag_hdr);
542483a47d2SDenis V. Lunev 		IP6_INC_STATS_BH(net,
543adf30907SEric Dumazet 				 ip6_dst_idev(skb_dst(skb)), IPSTATS_MIB_REASMOKS);
5441da177e4SLinus Torvalds 
545d56f90a7SArnaldo Carvalho de Melo 		IP6CB(skb)->nhoff = (u8 *)fhdr - skb_network_header(skb);
546f46078cfSHannes Frederic Sowa 		IP6CB(skb)->flags |= IP6SKB_FRAGMENTED;
5471da177e4SLinus Torvalds 		return 1;
5481da177e4SLinus Torvalds 	}
5491da177e4SLinus Torvalds 
550eec2e618SHannes Frederic Sowa 	fq = fq_find(net, fhdr->identification, &hdr->saddr, &hdr->daddr,
551eec2e618SHannes Frederic Sowa 		     ip6_frag_ecn(hdr));
5529546377cSShan Wei 	if (fq != NULL) {
553f61944efSHerbert Xu 		int ret;
5541da177e4SLinus Torvalds 
5555ab11c98SPavel Emelyanov 		spin_lock(&fq->q.lock);
5561da177e4SLinus Torvalds 
557f61944efSHerbert Xu 		ret = ip6_frag_queue(fq, skb, fhdr, IP6CB(skb)->nhoff);
5581da177e4SLinus Torvalds 
5595ab11c98SPavel Emelyanov 		spin_unlock(&fq->q.lock);
560b836c99fSAmerigo Wang 		inet_frag_put(&fq->q, &ip6_frags);
5611da177e4SLinus Torvalds 		return ret;
5621da177e4SLinus Torvalds 	}
5631da177e4SLinus Torvalds 
564adf30907SEric Dumazet 	IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)), IPSTATS_MIB_REASMFAILS);
5651da177e4SLinus Torvalds 	kfree_skb(skb);
5661da177e4SLinus Torvalds 	return -1;
56798b3377cSDenis V. Lunev 
56898b3377cSDenis V. Lunev fail_hdr:
569d2373862SNikolay Aleksandrov 	IP6_INC_STATS_BH(net, ip6_dst_idev(skb_dst(skb)),
570d2373862SNikolay Aleksandrov 			 IPSTATS_MIB_INHDRERRORS);
57198b3377cSDenis V. Lunev 	icmpv6_param_prob(skb, ICMPV6_HDR_FIELD, skb_network_header_len(skb));
57298b3377cSDenis V. Lunev 	return -1;
5731da177e4SLinus Torvalds }
5741da177e4SLinus Torvalds 
57541135cc8SAlexey Dobriyan static const struct inet6_protocol frag_protocol =
5761da177e4SLinus Torvalds {
5771da177e4SLinus Torvalds 	.handler	=	ipv6_frag_rcv,
5781da177e4SLinus Torvalds 	.flags		=	INET6_PROTO_NOPOLICY,
5791da177e4SLinus Torvalds };
5801da177e4SLinus Torvalds 
5818d8354d2SPavel Emelyanov #ifdef CONFIG_SYSCTL
5821bab4c75SNikolay Aleksandrov static int zero;
5831bab4c75SNikolay Aleksandrov 
5840a64b4b8SPavel Emelyanov static struct ctl_table ip6_frags_ns_ctl_table[] = {
585e71e0349SDaniel Lezcano 	{
5868d8354d2SPavel Emelyanov 		.procname	= "ip6frag_high_thresh",
587e31e0bdcSPavel Emelyanov 		.data		= &init_net.ipv6.frags.high_thresh,
5888d8354d2SPavel Emelyanov 		.maxlen		= sizeof(int),
5898d8354d2SPavel Emelyanov 		.mode		= 0644,
5901bab4c75SNikolay Aleksandrov 		.proc_handler	= proc_dointvec_minmax,
5911bab4c75SNikolay Aleksandrov 		.extra1		= &init_net.ipv6.frags.low_thresh
5928d8354d2SPavel Emelyanov 	},
5938d8354d2SPavel Emelyanov 	{
5948d8354d2SPavel Emelyanov 		.procname	= "ip6frag_low_thresh",
595e31e0bdcSPavel Emelyanov 		.data		= &init_net.ipv6.frags.low_thresh,
5968d8354d2SPavel Emelyanov 		.maxlen		= sizeof(int),
5978d8354d2SPavel Emelyanov 		.mode		= 0644,
5981bab4c75SNikolay Aleksandrov 		.proc_handler	= proc_dointvec_minmax,
5991bab4c75SNikolay Aleksandrov 		.extra1		= &zero,
6001bab4c75SNikolay Aleksandrov 		.extra2		= &init_net.ipv6.frags.high_thresh
6018d8354d2SPavel Emelyanov 	},
6028d8354d2SPavel Emelyanov 	{
6038d8354d2SPavel Emelyanov 		.procname	= "ip6frag_time",
604b2fd5321SPavel Emelyanov 		.data		= &init_net.ipv6.frags.timeout,
6058d8354d2SPavel Emelyanov 		.maxlen		= sizeof(int),
6068d8354d2SPavel Emelyanov 		.mode		= 0644,
6076d9f239aSAlexey Dobriyan 		.proc_handler	= proc_dointvec_jiffies,
6088d8354d2SPavel Emelyanov 	},
6097d291ebbSPavel Emelyanov 	{ }
6107d291ebbSPavel Emelyanov };
6117d291ebbSPavel Emelyanov 
612e3a57d18SFlorian Westphal /* secret interval has been deprecated */
613e3a57d18SFlorian Westphal static int ip6_frags_secret_interval_unused;
6147d291ebbSPavel Emelyanov static struct ctl_table ip6_frags_ctl_table[] = {
6158d8354d2SPavel Emelyanov 	{
6168d8354d2SPavel Emelyanov 		.procname	= "ip6frag_secret_interval",
617e3a57d18SFlorian Westphal 		.data		= &ip6_frags_secret_interval_unused,
6188d8354d2SPavel Emelyanov 		.maxlen		= sizeof(int),
6198d8354d2SPavel Emelyanov 		.mode		= 0644,
6206d9f239aSAlexey Dobriyan 		.proc_handler	= proc_dointvec_jiffies,
6218d8354d2SPavel Emelyanov 	},
6228d8354d2SPavel Emelyanov 	{ }
6238d8354d2SPavel Emelyanov };
6247d460db9SDaniel Lezcano 
6252c8c1e72SAlexey Dobriyan static int __net_init ip6_frags_ns_sysctl_register(struct net *net)
6268d8354d2SPavel Emelyanov {
627e4a2d5c2SPavel Emelyanov 	struct ctl_table *table;
6288d8354d2SPavel Emelyanov 	struct ctl_table_header *hdr;
6298d8354d2SPavel Emelyanov 
6300a64b4b8SPavel Emelyanov 	table = ip6_frags_ns_ctl_table;
63109ad9bc7SOctavian Purdila 	if (!net_eq(net, &init_net)) {
6320a64b4b8SPavel Emelyanov 		table = kmemdup(table, sizeof(ip6_frags_ns_ctl_table), GFP_KERNEL);
633e4a2d5c2SPavel Emelyanov 		if (table == NULL)
634e4a2d5c2SPavel Emelyanov 			goto err_alloc;
635e4a2d5c2SPavel Emelyanov 
636e31e0bdcSPavel Emelyanov 		table[0].data = &net->ipv6.frags.high_thresh;
6371bab4c75SNikolay Aleksandrov 		table[0].extra1 = &net->ipv6.frags.low_thresh;
6381bab4c75SNikolay Aleksandrov 		table[0].extra2 = &init_net.ipv6.frags.high_thresh;
639e31e0bdcSPavel Emelyanov 		table[1].data = &net->ipv6.frags.low_thresh;
6401bab4c75SNikolay Aleksandrov 		table[1].extra2 = &net->ipv6.frags.high_thresh;
641b2fd5321SPavel Emelyanov 		table[2].data = &net->ipv6.frags.timeout;
642464dc801SEric W. Biederman 
643464dc801SEric W. Biederman 		/* Don't export sysctls to unprivileged users */
644464dc801SEric W. Biederman 		if (net->user_ns != &init_user_ns)
645464dc801SEric W. Biederman 			table[0].procname = NULL;
646e4a2d5c2SPavel Emelyanov 	}
647e4a2d5c2SPavel Emelyanov 
648ec8f23ceSEric W. Biederman 	hdr = register_net_sysctl(net, "net/ipv6", table);
649e4a2d5c2SPavel Emelyanov 	if (hdr == NULL)
650e4a2d5c2SPavel Emelyanov 		goto err_reg;
651e4a2d5c2SPavel Emelyanov 
652e4a2d5c2SPavel Emelyanov 	net->ipv6.sysctl.frags_hdr = hdr;
653e4a2d5c2SPavel Emelyanov 	return 0;
654e4a2d5c2SPavel Emelyanov 
655e4a2d5c2SPavel Emelyanov err_reg:
65609ad9bc7SOctavian Purdila 	if (!net_eq(net, &init_net))
657e4a2d5c2SPavel Emelyanov 		kfree(table);
658e4a2d5c2SPavel Emelyanov err_alloc:
659e4a2d5c2SPavel Emelyanov 	return -ENOMEM;
660e4a2d5c2SPavel Emelyanov }
661e4a2d5c2SPavel Emelyanov 
6622c8c1e72SAlexey Dobriyan static void __net_exit ip6_frags_ns_sysctl_unregister(struct net *net)
663e4a2d5c2SPavel Emelyanov {
664e4a2d5c2SPavel Emelyanov 	struct ctl_table *table;
665e4a2d5c2SPavel Emelyanov 
666e4a2d5c2SPavel Emelyanov 	table = net->ipv6.sysctl.frags_hdr->ctl_table_arg;
667e4a2d5c2SPavel Emelyanov 	unregister_net_sysctl_table(net->ipv6.sysctl.frags_hdr);
6683705e11aSYang Hongyang 	if (!net_eq(net, &init_net))
669e4a2d5c2SPavel Emelyanov 		kfree(table);
6708d8354d2SPavel Emelyanov }
6717d291ebbSPavel Emelyanov 
6727d291ebbSPavel Emelyanov static struct ctl_table_header *ip6_ctl_header;
6737d291ebbSPavel Emelyanov 
6747d291ebbSPavel Emelyanov static int ip6_frags_sysctl_register(void)
6757d291ebbSPavel Emelyanov {
67643444757SEric W. Biederman 	ip6_ctl_header = register_net_sysctl(&init_net, "net/ipv6",
6777d291ebbSPavel Emelyanov 			ip6_frags_ctl_table);
6787d291ebbSPavel Emelyanov 	return ip6_ctl_header == NULL ? -ENOMEM : 0;
6797d291ebbSPavel Emelyanov }
6807d291ebbSPavel Emelyanov 
6817d291ebbSPavel Emelyanov static void ip6_frags_sysctl_unregister(void)
6827d291ebbSPavel Emelyanov {
6837d291ebbSPavel Emelyanov 	unregister_net_sysctl_table(ip6_ctl_header);
6847d291ebbSPavel Emelyanov }
6858d8354d2SPavel Emelyanov #else
6860a64b4b8SPavel Emelyanov static inline int ip6_frags_ns_sysctl_register(struct net *net)
6878d8354d2SPavel Emelyanov {
6888d8354d2SPavel Emelyanov 	return 0;
6898d8354d2SPavel Emelyanov }
690e4a2d5c2SPavel Emelyanov 
6910a64b4b8SPavel Emelyanov static inline void ip6_frags_ns_sysctl_unregister(struct net *net)
692e4a2d5c2SPavel Emelyanov {
693e4a2d5c2SPavel Emelyanov }
6947d291ebbSPavel Emelyanov 
6957d291ebbSPavel Emelyanov static inline int ip6_frags_sysctl_register(void)
6967d291ebbSPavel Emelyanov {
6977d291ebbSPavel Emelyanov 	return 0;
6987d291ebbSPavel Emelyanov }
6997d291ebbSPavel Emelyanov 
7007d291ebbSPavel Emelyanov static inline void ip6_frags_sysctl_unregister(void)
7017d291ebbSPavel Emelyanov {
7027d291ebbSPavel Emelyanov }
7038d8354d2SPavel Emelyanov #endif
7048d8354d2SPavel Emelyanov 
7052c8c1e72SAlexey Dobriyan static int __net_init ipv6_frags_init_net(struct net *net)
7068d8354d2SPavel Emelyanov {
7077c070aa9SShan Wei 	net->ipv6.frags.high_thresh = IPV6_FRAG_HIGH_THRESH;
7087c070aa9SShan Wei 	net->ipv6.frags.low_thresh = IPV6_FRAG_LOW_THRESH;
709b2fd5321SPavel Emelyanov 	net->ipv6.frags.timeout = IPV6_FRAG_TIMEOUT;
7108d8354d2SPavel Emelyanov 
711e5a2bb84SPavel Emelyanov 	inet_frags_init_net(&net->ipv6.frags);
712e5a2bb84SPavel Emelyanov 
7130a64b4b8SPavel Emelyanov 	return ip6_frags_ns_sysctl_register(net);
714e71e0349SDaniel Lezcano }
715e71e0349SDaniel Lezcano 
7162c8c1e72SAlexey Dobriyan static void __net_exit ipv6_frags_exit_net(struct net *net)
71781566e83SPavel Emelyanov {
7180a64b4b8SPavel Emelyanov 	ip6_frags_ns_sysctl_unregister(net);
71981566e83SPavel Emelyanov 	inet_frags_exit_net(&net->ipv6.frags, &ip6_frags);
72081566e83SPavel Emelyanov }
72181566e83SPavel Emelyanov 
72281566e83SPavel Emelyanov static struct pernet_operations ip6_frags_ops = {
72381566e83SPavel Emelyanov 	.init = ipv6_frags_init_net,
72481566e83SPavel Emelyanov 	.exit = ipv6_frags_exit_net,
72581566e83SPavel Emelyanov };
72681566e83SPavel Emelyanov 
727853cbbaaSDaniel Lezcano int __init ipv6_frag_init(void)
7281da177e4SLinus Torvalds {
729853cbbaaSDaniel Lezcano 	int ret;
7301da177e4SLinus Torvalds 
731853cbbaaSDaniel Lezcano 	ret = inet6_add_protocol(&frag_protocol, IPPROTO_FRAGMENT);
732853cbbaaSDaniel Lezcano 	if (ret)
733853cbbaaSDaniel Lezcano 		goto out;
734e71e0349SDaniel Lezcano 
7357d291ebbSPavel Emelyanov 	ret = ip6_frags_sysctl_register();
7367d291ebbSPavel Emelyanov 	if (ret)
7377d291ebbSPavel Emelyanov 		goto err_sysctl;
7387d291ebbSPavel Emelyanov 
7390002c630SPavel Emelyanov 	ret = register_pernet_subsys(&ip6_frags_ops);
7400002c630SPavel Emelyanov 	if (ret)
7410002c630SPavel Emelyanov 		goto err_pernet;
7428d8354d2SPavel Emelyanov 
743321a3a99SPavel Emelyanov 	ip6_frags.hashfn = ip6_hashfn;
744c6fda282SPavel Emelyanov 	ip6_frags.constructor = ip6_frag_init;
745c9547709SPavel Emelyanov 	ip6_frags.destructor = NULL;
7461e4b8287SPavel Emelyanov 	ip6_frags.skb_free = NULL;
7471e4b8287SPavel Emelyanov 	ip6_frags.qsize = sizeof(struct frag_queue);
748abd6523dSPavel Emelyanov 	ip6_frags.match = ip6_frag_match;
749e521db9dSPavel Emelyanov 	ip6_frags.frag_expire = ip6_frag_expire;
7507eb95156SPavel Emelyanov 	inet_frags_init(&ip6_frags);
751853cbbaaSDaniel Lezcano out:
752853cbbaaSDaniel Lezcano 	return ret;
7530002c630SPavel Emelyanov 
7540002c630SPavel Emelyanov err_pernet:
7557d291ebbSPavel Emelyanov 	ip6_frags_sysctl_unregister();
7567d291ebbSPavel Emelyanov err_sysctl:
7570002c630SPavel Emelyanov 	inet6_del_protocol(&frag_protocol, IPPROTO_FRAGMENT);
7580002c630SPavel Emelyanov 	goto out;
759853cbbaaSDaniel Lezcano }
760853cbbaaSDaniel Lezcano 
761853cbbaaSDaniel Lezcano void ipv6_frag_exit(void)
762853cbbaaSDaniel Lezcano {
763853cbbaaSDaniel Lezcano 	inet_frags_fini(&ip6_frags);
7647d291ebbSPavel Emelyanov 	ip6_frags_sysctl_unregister();
76581566e83SPavel Emelyanov 	unregister_pernet_subsys(&ip6_frags_ops);
766853cbbaaSDaniel Lezcano 	inet6_del_protocol(&frag_protocol, IPPROTO_FRAGMENT);
7671da177e4SLinus Torvalds }
768