]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - sys/netinet6/frag6.c
MFC r329514:
[FreeBSD/FreeBSD.git] / sys / netinet6 / frag6.c
1 /*-
2  * Copyright (C) 1995, 1996, 1997, and 1998 WIDE Project.
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. Neither the name of the project nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  *
29  *      $KAME: frag6.c,v 1.33 2002/01/07 11:34:48 kjc Exp $
30  */
31
32 #include <sys/cdefs.h>
33 __FBSDID("$FreeBSD$");
34
35 #include "opt_rss.h"
36
37 #include <sys/param.h>
38 #include <sys/systm.h>
39 #include <sys/malloc.h>
40 #include <sys/mbuf.h>
41 #include <sys/domain.h>
42 #include <sys/eventhandler.h>
43 #include <sys/protosw.h>
44 #include <sys/socket.h>
45 #include <sys/errno.h>
46 #include <sys/time.h>
47 #include <sys/kernel.h>
48 #include <sys/syslog.h>
49
50 #include <net/if.h>
51 #include <net/if_var.h>
52 #include <net/netisr.h>
53 #include <net/route.h>
54 #include <net/vnet.h>
55
56 #include <netinet/in.h>
57 #include <netinet/in_var.h>
58 #include <netinet/ip6.h>
59 #include <netinet6/ip6_var.h>
60 #include <netinet/icmp6.h>
61 #include <netinet/in_systm.h>   /* for ECN definitions */
62 #include <netinet/ip.h>         /* for ECN definitions */
63
64 #include <security/mac/mac_framework.h>
65
66 static void frag6_enq(struct ip6asfrag *, struct ip6asfrag *);
67 static void frag6_deq(struct ip6asfrag *);
68 static void frag6_insque(struct ip6q *, struct ip6q *);
69 static void frag6_remque(struct ip6q *);
70 static void frag6_freef(struct ip6q *);
71
72 static struct mtx ip6qlock;
73 /*
74  * These fields all protected by ip6qlock.
75  */
76 static VNET_DEFINE(u_int, frag6_nfragpackets);
77 static VNET_DEFINE(u_int, frag6_nfrags);
78 static VNET_DEFINE(struct ip6q, ip6q);  /* ip6 reassemble queue */
79
80 #define V_frag6_nfragpackets            VNET(frag6_nfragpackets)
81 #define V_frag6_nfrags                  VNET(frag6_nfrags)
82 #define V_ip6q                          VNET(ip6q)
83
84 #define IP6Q_LOCK_INIT()        mtx_init(&ip6qlock, "ip6qlock", NULL, MTX_DEF);
85 #define IP6Q_LOCK()             mtx_lock(&ip6qlock)
86 #define IP6Q_TRYLOCK()          mtx_trylock(&ip6qlock)
87 #define IP6Q_LOCK_ASSERT()      mtx_assert(&ip6qlock, MA_OWNED)
88 #define IP6Q_UNLOCK()           mtx_unlock(&ip6qlock)
89
90 static MALLOC_DEFINE(M_FTABLE, "fragment", "fragment reassembly header");
91
92 /*
93  * Initialise reassembly queue and fragment identifier.
94  */
95 static void
96 frag6_change(void *tag)
97 {
98
99         V_ip6_maxfragpackets = nmbclusters / 4;
100         V_ip6_maxfrags = nmbclusters / 4;
101 }
102
103 void
104 frag6_init(void)
105 {
106
107         V_ip6_maxfragpackets = nmbclusters / 4;
108         V_ip6_maxfrags = nmbclusters / 4;
109         V_ip6q.ip6q_next = V_ip6q.ip6q_prev = &V_ip6q;
110
111         if (!IS_DEFAULT_VNET(curvnet))
112                 return;
113
114         EVENTHANDLER_REGISTER(nmbclusters_change,
115             frag6_change, NULL, EVENTHANDLER_PRI_ANY);
116
117         IP6Q_LOCK_INIT();
118 }
119
120 /*
121  * In RFC2460, fragment and reassembly rule do not agree with each other,
122  * in terms of next header field handling in fragment header.
123  * While the sender will use the same value for all of the fragmented packets,
124  * receiver is suggested not to check the consistency.
125  *
126  * fragment rule (p20):
127  *      (2) A Fragment header containing:
128  *      The Next Header value that identifies the first header of
129  *      the Fragmentable Part of the original packet.
130  *              -> next header field is same for all fragments
131  *
132  * reassembly rule (p21):
133  *      The Next Header field of the last header of the Unfragmentable
134  *      Part is obtained from the Next Header field of the first
135  *      fragment's Fragment header.
136  *              -> should grab it from the first fragment only
137  *
138  * The following note also contradicts with fragment rule - no one is going to
139  * send different fragment with different next header field.
140  *
141  * additional note (p22):
142  *      The Next Header values in the Fragment headers of different
143  *      fragments of the same original packet may differ.  Only the value
144  *      from the Offset zero fragment packet is used for reassembly.
145  *              -> should grab it from the first fragment only
146  *
147  * There is no explicit reason given in the RFC.  Historical reason maybe?
148  */
149 /*
150  * Fragment input
151  */
152 int
153 frag6_input(struct mbuf **mp, int *offp, int proto)
154 {
155         struct mbuf *m = *mp, *t;
156         struct ip6_hdr *ip6;
157         struct ip6_frag *ip6f;
158         struct ip6q *q6;
159         struct ip6asfrag *af6, *ip6af, *af6dwn;
160         struct in6_ifaddr *ia;
161         int offset = *offp, nxt, i, next;
162         int first_frag = 0;
163         int fragoff, frgpartlen;        /* must be larger than u_int16_t */
164         struct ifnet *dstifp;
165         u_int8_t ecn, ecn0;
166 #ifdef RSS
167         struct m_tag *mtag;
168         struct ip6_direct_ctx *ip6dc;
169 #endif
170
171 #if 0
172         char ip6buf[INET6_ADDRSTRLEN];
173 #endif
174
175         ip6 = mtod(m, struct ip6_hdr *);
176 #ifndef PULLDOWN_TEST
177         IP6_EXTHDR_CHECK(m, offset, sizeof(struct ip6_frag), IPPROTO_DONE);
178         ip6f = (struct ip6_frag *)((caddr_t)ip6 + offset);
179 #else
180         IP6_EXTHDR_GET(ip6f, struct ip6_frag *, m, offset, sizeof(*ip6f));
181         if (ip6f == NULL)
182                 return (IPPROTO_DONE);
183 #endif
184
185         dstifp = NULL;
186         /* find the destination interface of the packet. */
187         ia = in6ifa_ifwithaddr(&ip6->ip6_dst, 0 /* XXX */);
188         if (ia != NULL) {
189                 dstifp = ia->ia_ifp;
190                 ifa_free(&ia->ia_ifa);
191         }
192         /* jumbo payload can't contain a fragment header */
193         if (ip6->ip6_plen == 0) {
194                 icmp6_error(m, ICMP6_PARAM_PROB, ICMP6_PARAMPROB_HEADER, offset);
195                 in6_ifstat_inc(dstifp, ifs6_reass_fail);
196                 return IPPROTO_DONE;
197         }
198
199         /*
200          * check whether fragment packet's fragment length is
201          * multiple of 8 octets.
202          * sizeof(struct ip6_frag) == 8
203          * sizeof(struct ip6_hdr) = 40
204          */
205         if ((ip6f->ip6f_offlg & IP6F_MORE_FRAG) &&
206             (((ntohs(ip6->ip6_plen) - offset) & 0x7) != 0)) {
207                 icmp6_error(m, ICMP6_PARAM_PROB, ICMP6_PARAMPROB_HEADER,
208                     offsetof(struct ip6_hdr, ip6_plen));
209                 in6_ifstat_inc(dstifp, ifs6_reass_fail);
210                 return IPPROTO_DONE;
211         }
212
213         IP6STAT_INC(ip6s_fragments);
214         in6_ifstat_inc(dstifp, ifs6_reass_reqd);
215
216         /* offset now points to data portion */
217         offset += sizeof(struct ip6_frag);
218
219         /*
220          * RFC 6946: Handle "atomic" fragments (offset and m bit set to 0)
221          * upfront, unrelated to any reassembly.  Just skip the fragment header.
222          */
223         if ((ip6f->ip6f_offlg & ~IP6F_RESERVED_MASK) == 0) {
224                 /* XXX-BZ we want dedicated counters for this. */
225                 IP6STAT_INC(ip6s_reassembled);
226                 in6_ifstat_inc(dstifp, ifs6_reass_ok);
227                 *offp = offset;
228                 m->m_flags |= M_FRAGMENTED;
229                 return (ip6f->ip6f_nxt);
230         }
231
232         IP6Q_LOCK();
233
234         /*
235          * Enforce upper bound on number of fragments.
236          * If maxfrag is 0, never accept fragments.
237          * If maxfrag is -1, accept all fragments without limitation.
238          */
239         if (V_ip6_maxfrags < 0)
240                 ;
241         else if (V_frag6_nfrags >= (u_int)V_ip6_maxfrags)
242                 goto dropfrag;
243
244         for (q6 = V_ip6q.ip6q_next; q6 != &V_ip6q; q6 = q6->ip6q_next)
245                 if (ip6f->ip6f_ident == q6->ip6q_ident &&
246                     IN6_ARE_ADDR_EQUAL(&ip6->ip6_src, &q6->ip6q_src) &&
247                     IN6_ARE_ADDR_EQUAL(&ip6->ip6_dst, &q6->ip6q_dst)
248 #ifdef MAC
249                     && mac_ip6q_match(m, q6)
250 #endif
251                     )
252                         break;
253
254         if (q6 == &V_ip6q) {
255                 /*
256                  * the first fragment to arrive, create a reassembly queue.
257                  */
258                 first_frag = 1;
259
260                 /*
261                  * Enforce upper bound on number of fragmented packets
262                  * for which we attempt reassembly;
263                  * If maxfragpackets is 0, never accept fragments.
264                  * If maxfragpackets is -1, accept all fragments without
265                  * limitation.
266                  */
267                 if (V_ip6_maxfragpackets < 0)
268                         ;
269                 else if (V_frag6_nfragpackets >= (u_int)V_ip6_maxfragpackets)
270                         goto dropfrag;
271                 V_frag6_nfragpackets++;
272                 q6 = (struct ip6q *)malloc(sizeof(struct ip6q), M_FTABLE,
273                     M_NOWAIT);
274                 if (q6 == NULL)
275                         goto dropfrag;
276                 bzero(q6, sizeof(*q6));
277 #ifdef MAC
278                 if (mac_ip6q_init(q6, M_NOWAIT) != 0) {
279                         free(q6, M_FTABLE);
280                         goto dropfrag;
281                 }
282                 mac_ip6q_create(m, q6);
283 #endif
284                 frag6_insque(q6, &V_ip6q);
285
286                 /* ip6q_nxt will be filled afterwards, from 1st fragment */
287                 q6->ip6q_down   = q6->ip6q_up = (struct ip6asfrag *)q6;
288 #ifdef notyet
289                 q6->ip6q_nxtp   = (u_char *)nxtp;
290 #endif
291                 q6->ip6q_ident  = ip6f->ip6f_ident;
292                 q6->ip6q_ttl    = IPV6_FRAGTTL;
293                 q6->ip6q_src    = ip6->ip6_src;
294                 q6->ip6q_dst    = ip6->ip6_dst;
295                 q6->ip6q_ecn    =
296                     (ntohl(ip6->ip6_flow) >> 20) & IPTOS_ECN_MASK;
297                 q6->ip6q_unfrglen = -1; /* The 1st fragment has not arrived. */
298
299                 q6->ip6q_nfrag = 0;
300         }
301
302         /*
303          * If it's the 1st fragment, record the length of the
304          * unfragmentable part and the next header of the fragment header.
305          */
306         fragoff = ntohs(ip6f->ip6f_offlg & IP6F_OFF_MASK);
307         if (fragoff == 0) {
308                 q6->ip6q_unfrglen = offset - sizeof(struct ip6_hdr) -
309                     sizeof(struct ip6_frag);
310                 q6->ip6q_nxt = ip6f->ip6f_nxt;
311         }
312
313         /*
314          * Check that the reassembled packet would not exceed 65535 bytes
315          * in size.
316          * If it would exceed, discard the fragment and return an ICMP error.
317          */
318         frgpartlen = sizeof(struct ip6_hdr) + ntohs(ip6->ip6_plen) - offset;
319         if (q6->ip6q_unfrglen >= 0) {
320                 /* The 1st fragment has already arrived. */
321                 if (q6->ip6q_unfrglen + fragoff + frgpartlen > IPV6_MAXPACKET) {
322                         icmp6_error(m, ICMP6_PARAM_PROB, ICMP6_PARAMPROB_HEADER,
323                             offset - sizeof(struct ip6_frag) +
324                             offsetof(struct ip6_frag, ip6f_offlg));
325                         IP6Q_UNLOCK();
326                         return (IPPROTO_DONE);
327                 }
328         } else if (fragoff + frgpartlen > IPV6_MAXPACKET) {
329                 icmp6_error(m, ICMP6_PARAM_PROB, ICMP6_PARAMPROB_HEADER,
330                     offset - sizeof(struct ip6_frag) +
331                     offsetof(struct ip6_frag, ip6f_offlg));
332                 IP6Q_UNLOCK();
333                 return (IPPROTO_DONE);
334         }
335         /*
336          * If it's the first fragment, do the above check for each
337          * fragment already stored in the reassembly queue.
338          */
339         if (fragoff == 0) {
340                 for (af6 = q6->ip6q_down; af6 != (struct ip6asfrag *)q6;
341                      af6 = af6dwn) {
342                         af6dwn = af6->ip6af_down;
343
344                         if (q6->ip6q_unfrglen + af6->ip6af_off + af6->ip6af_frglen >
345                             IPV6_MAXPACKET) {
346                                 struct mbuf *merr = IP6_REASS_MBUF(af6);
347                                 struct ip6_hdr *ip6err;
348                                 int erroff = af6->ip6af_offset;
349
350                                 /* dequeue the fragment. */
351                                 frag6_deq(af6);
352                                 free(af6, M_FTABLE);
353
354                                 /* adjust pointer. */
355                                 ip6err = mtod(merr, struct ip6_hdr *);
356
357                                 /*
358                                  * Restore source and destination addresses
359                                  * in the erroneous IPv6 header.
360                                  */
361                                 ip6err->ip6_src = q6->ip6q_src;
362                                 ip6err->ip6_dst = q6->ip6q_dst;
363
364                                 icmp6_error(merr, ICMP6_PARAM_PROB,
365                                     ICMP6_PARAMPROB_HEADER,
366                                     erroff - sizeof(struct ip6_frag) +
367                                     offsetof(struct ip6_frag, ip6f_offlg));
368                         }
369                 }
370         }
371
372         ip6af = (struct ip6asfrag *)malloc(sizeof(struct ip6asfrag), M_FTABLE,
373             M_NOWAIT);
374         if (ip6af == NULL)
375                 goto dropfrag;
376         bzero(ip6af, sizeof(*ip6af));
377         ip6af->ip6af_mff = ip6f->ip6f_offlg & IP6F_MORE_FRAG;
378         ip6af->ip6af_off = fragoff;
379         ip6af->ip6af_frglen = frgpartlen;
380         ip6af->ip6af_offset = offset;
381         IP6_REASS_MBUF(ip6af) = m;
382
383         if (first_frag) {
384                 af6 = (struct ip6asfrag *)q6;
385                 goto insert;
386         }
387
388         /*
389          * Handle ECN by comparing this segment with the first one;
390          * if CE is set, do not lose CE.
391          * drop if CE and not-ECT are mixed for the same packet.
392          */
393         ecn = (ntohl(ip6->ip6_flow) >> 20) & IPTOS_ECN_MASK;
394         ecn0 = q6->ip6q_ecn;
395         if (ecn == IPTOS_ECN_CE) {
396                 if (ecn0 == IPTOS_ECN_NOTECT) {
397                         free(ip6af, M_FTABLE);
398                         goto dropfrag;
399                 }
400                 if (ecn0 != IPTOS_ECN_CE)
401                         q6->ip6q_ecn = IPTOS_ECN_CE;
402         }
403         if (ecn == IPTOS_ECN_NOTECT && ecn0 != IPTOS_ECN_NOTECT) {
404                 free(ip6af, M_FTABLE);
405                 goto dropfrag;
406         }
407
408         /*
409          * Find a segment which begins after this one does.
410          */
411         for (af6 = q6->ip6q_down; af6 != (struct ip6asfrag *)q6;
412              af6 = af6->ip6af_down)
413                 if (af6->ip6af_off > ip6af->ip6af_off)
414                         break;
415
416 #if 0
417         /*
418          * If there is a preceding segment, it may provide some of
419          * our data already.  If so, drop the data from the incoming
420          * segment.  If it provides all of our data, drop us.
421          */
422         if (af6->ip6af_up != (struct ip6asfrag *)q6) {
423                 i = af6->ip6af_up->ip6af_off + af6->ip6af_up->ip6af_frglen
424                         - ip6af->ip6af_off;
425                 if (i > 0) {
426                         if (i >= ip6af->ip6af_frglen)
427                                 goto dropfrag;
428                         m_adj(IP6_REASS_MBUF(ip6af), i);
429                         ip6af->ip6af_off += i;
430                         ip6af->ip6af_frglen -= i;
431                 }
432         }
433
434         /*
435          * While we overlap succeeding segments trim them or,
436          * if they are completely covered, dequeue them.
437          */
438         while (af6 != (struct ip6asfrag *)q6 &&
439                ip6af->ip6af_off + ip6af->ip6af_frglen > af6->ip6af_off) {
440                 i = (ip6af->ip6af_off + ip6af->ip6af_frglen) - af6->ip6af_off;
441                 if (i < af6->ip6af_frglen) {
442                         af6->ip6af_frglen -= i;
443                         af6->ip6af_off += i;
444                         m_adj(IP6_REASS_MBUF(af6), i);
445                         break;
446                 }
447                 af6 = af6->ip6af_down;
448                 m_freem(IP6_REASS_MBUF(af6->ip6af_up));
449                 frag6_deq(af6->ip6af_up);
450         }
451 #else
452         /*
453          * If the incoming framgent overlaps some existing fragments in
454          * the reassembly queue, drop it, since it is dangerous to override
455          * existing fragments from a security point of view.
456          * We don't know which fragment is the bad guy - here we trust
457          * fragment that came in earlier, with no real reason.
458          *
459          * Note: due to changes after disabling this part, mbuf passed to
460          * m_adj() below now does not meet the requirement.
461          */
462         if (af6->ip6af_up != (struct ip6asfrag *)q6) {
463                 i = af6->ip6af_up->ip6af_off + af6->ip6af_up->ip6af_frglen
464                         - ip6af->ip6af_off;
465                 if (i > 0) {
466 #if 0                           /* suppress the noisy log */
467                         log(LOG_ERR, "%d bytes of a fragment from %s "
468                             "overlaps the previous fragment\n",
469                             i, ip6_sprintf(ip6buf, &q6->ip6q_src));
470 #endif
471                         free(ip6af, M_FTABLE);
472                         goto dropfrag;
473                 }
474         }
475         if (af6 != (struct ip6asfrag *)q6) {
476                 i = (ip6af->ip6af_off + ip6af->ip6af_frglen) - af6->ip6af_off;
477                 if (i > 0) {
478 #if 0                           /* suppress the noisy log */
479                         log(LOG_ERR, "%d bytes of a fragment from %s "
480                             "overlaps the succeeding fragment",
481                             i, ip6_sprintf(ip6buf, &q6->ip6q_src));
482 #endif
483                         free(ip6af, M_FTABLE);
484                         goto dropfrag;
485                 }
486         }
487 #endif
488
489 insert:
490 #ifdef MAC
491         if (!first_frag)
492                 mac_ip6q_update(m, q6);
493 #endif
494
495         /*
496          * Stick new segment in its place;
497          * check for complete reassembly.
498          * Move to front of packet queue, as we are
499          * the most recently active fragmented packet.
500          */
501         frag6_enq(ip6af, af6->ip6af_up);
502         V_frag6_nfrags++;
503         q6->ip6q_nfrag++;
504 #if 0 /* xxx */
505         if (q6 != V_ip6q.ip6q_next) {
506                 frag6_remque(q6);
507                 frag6_insque(q6, &V_ip6q);
508         }
509 #endif
510         next = 0;
511         for (af6 = q6->ip6q_down; af6 != (struct ip6asfrag *)q6;
512              af6 = af6->ip6af_down) {
513                 if (af6->ip6af_off != next) {
514                         IP6Q_UNLOCK();
515                         return IPPROTO_DONE;
516                 }
517                 next += af6->ip6af_frglen;
518         }
519         if (af6->ip6af_up->ip6af_mff) {
520                 IP6Q_UNLOCK();
521                 return IPPROTO_DONE;
522         }
523
524         /*
525          * Reassembly is complete; concatenate fragments.
526          */
527         ip6af = q6->ip6q_down;
528         t = m = IP6_REASS_MBUF(ip6af);
529         af6 = ip6af->ip6af_down;
530         frag6_deq(ip6af);
531         while (af6 != (struct ip6asfrag *)q6) {
532                 m->m_pkthdr.csum_flags &=
533                     IP6_REASS_MBUF(af6)->m_pkthdr.csum_flags;
534                 m->m_pkthdr.csum_data +=
535                     IP6_REASS_MBUF(af6)->m_pkthdr.csum_data;
536
537                 af6dwn = af6->ip6af_down;
538                 frag6_deq(af6);
539                 while (t->m_next)
540                         t = t->m_next;
541                 m_adj(IP6_REASS_MBUF(af6), af6->ip6af_offset);
542                 m_cat(t, IP6_REASS_MBUF(af6));
543                 free(af6, M_FTABLE);
544                 af6 = af6dwn;
545         }
546
547         while (m->m_pkthdr.csum_data & 0xffff0000)
548                 m->m_pkthdr.csum_data = (m->m_pkthdr.csum_data & 0xffff) +
549                     (m->m_pkthdr.csum_data >> 16);
550
551         /* adjust offset to point where the original next header starts */
552         offset = ip6af->ip6af_offset - sizeof(struct ip6_frag);
553         free(ip6af, M_FTABLE);
554         ip6 = mtod(m, struct ip6_hdr *);
555         ip6->ip6_plen = htons((u_short)next + offset - sizeof(struct ip6_hdr));
556         if (q6->ip6q_ecn == IPTOS_ECN_CE)
557                 ip6->ip6_flow |= htonl(IPTOS_ECN_CE << 20);
558         nxt = q6->ip6q_nxt;
559 #ifdef notyet
560         *q6->ip6q_nxtp = (u_char)(nxt & 0xff);
561 #endif
562
563         if (ip6_deletefraghdr(m, offset, M_NOWAIT) != 0) {
564                 frag6_remque(q6);
565                 V_frag6_nfrags -= q6->ip6q_nfrag;
566 #ifdef MAC
567                 mac_ip6q_destroy(q6);
568 #endif
569                 free(q6, M_FTABLE);
570                 V_frag6_nfragpackets--;
571
572                 goto dropfrag;
573         }
574
575         /*
576          * Store NXT to the original.
577          */
578         m_copyback(m, ip6_get_prevhdr(m, offset), sizeof(uint8_t),
579             (caddr_t)&nxt);
580
581         frag6_remque(q6);
582         V_frag6_nfrags -= q6->ip6q_nfrag;
583 #ifdef MAC
584         mac_ip6q_reassemble(q6, m);
585         mac_ip6q_destroy(q6);
586 #endif
587         free(q6, M_FTABLE);
588         V_frag6_nfragpackets--;
589
590         if (m->m_flags & M_PKTHDR) { /* Isn't it always true? */
591                 int plen = 0;
592                 for (t = m; t; t = t->m_next)
593                         plen += t->m_len;
594                 m->m_pkthdr.len = plen;
595         }
596
597 #ifdef RSS
598         mtag = m_tag_alloc(MTAG_ABI_IPV6, IPV6_TAG_DIRECT, sizeof(*ip6dc),
599             M_NOWAIT);
600         if (mtag == NULL)
601                 goto dropfrag;
602
603         ip6dc = (struct ip6_direct_ctx *)(mtag + 1);
604         ip6dc->ip6dc_nxt = nxt;
605         ip6dc->ip6dc_off = offset;
606
607         m_tag_prepend(m, mtag);
608 #endif
609
610         IP6Q_UNLOCK();
611         IP6STAT_INC(ip6s_reassembled);
612         in6_ifstat_inc(dstifp, ifs6_reass_ok);
613
614 #ifdef RSS
615         /*
616          * Queue/dispatch for reprocessing.
617          */
618         netisr_dispatch(NETISR_IPV6_DIRECT, m);
619         return IPPROTO_DONE;
620 #endif
621
622         /*
623          * Tell launch routine the next header
624          */
625
626         *mp = m;
627         *offp = offset;
628
629         return nxt;
630
631  dropfrag:
632         IP6Q_UNLOCK();
633         in6_ifstat_inc(dstifp, ifs6_reass_fail);
634         IP6STAT_INC(ip6s_fragdropped);
635         m_freem(m);
636         return IPPROTO_DONE;
637 }
638
639 /*
640  * Free a fragment reassembly header and all
641  * associated datagrams.
642  */
643 void
644 frag6_freef(struct ip6q *q6)
645 {
646         struct ip6asfrag *af6, *down6;
647
648         IP6Q_LOCK_ASSERT();
649
650         for (af6 = q6->ip6q_down; af6 != (struct ip6asfrag *)q6;
651              af6 = down6) {
652                 struct mbuf *m = IP6_REASS_MBUF(af6);
653
654                 down6 = af6->ip6af_down;
655                 frag6_deq(af6);
656
657                 /*
658                  * Return ICMP time exceeded error for the 1st fragment.
659                  * Just free other fragments.
660                  */
661                 if (af6->ip6af_off == 0) {
662                         struct ip6_hdr *ip6;
663
664                         /* adjust pointer */
665                         ip6 = mtod(m, struct ip6_hdr *);
666
667                         /* restore source and destination addresses */
668                         ip6->ip6_src = q6->ip6q_src;
669                         ip6->ip6_dst = q6->ip6q_dst;
670
671                         icmp6_error(m, ICMP6_TIME_EXCEEDED,
672                                     ICMP6_TIME_EXCEED_REASSEMBLY, 0);
673                 } else
674                         m_freem(m);
675                 free(af6, M_FTABLE);
676         }
677         frag6_remque(q6);
678         V_frag6_nfrags -= q6->ip6q_nfrag;
679 #ifdef MAC
680         mac_ip6q_destroy(q6);
681 #endif
682         free(q6, M_FTABLE);
683         V_frag6_nfragpackets--;
684 }
685
686 /*
687  * Put an ip fragment on a reassembly chain.
688  * Like insque, but pointers in middle of structure.
689  */
690 void
691 frag6_enq(struct ip6asfrag *af6, struct ip6asfrag *up6)
692 {
693
694         IP6Q_LOCK_ASSERT();
695
696         af6->ip6af_up = up6;
697         af6->ip6af_down = up6->ip6af_down;
698         up6->ip6af_down->ip6af_up = af6;
699         up6->ip6af_down = af6;
700 }
701
702 /*
703  * To frag6_enq as remque is to insque.
704  */
705 void
706 frag6_deq(struct ip6asfrag *af6)
707 {
708
709         IP6Q_LOCK_ASSERT();
710
711         af6->ip6af_up->ip6af_down = af6->ip6af_down;
712         af6->ip6af_down->ip6af_up = af6->ip6af_up;
713 }
714
715 void
716 frag6_insque(struct ip6q *new, struct ip6q *old)
717 {
718
719         IP6Q_LOCK_ASSERT();
720
721         new->ip6q_prev = old;
722         new->ip6q_next = old->ip6q_next;
723         old->ip6q_next->ip6q_prev= new;
724         old->ip6q_next = new;
725 }
726
727 void
728 frag6_remque(struct ip6q *p6)
729 {
730
731         IP6Q_LOCK_ASSERT();
732
733         p6->ip6q_prev->ip6q_next = p6->ip6q_next;
734         p6->ip6q_next->ip6q_prev = p6->ip6q_prev;
735 }
736
737 /*
738  * IPv6 reassembling timer processing;
739  * if a timer expires on a reassembly
740  * queue, discard it.
741  */
742 void
743 frag6_slowtimo(void)
744 {
745         VNET_ITERATOR_DECL(vnet_iter);
746         struct ip6q *q6;
747
748         VNET_LIST_RLOCK_NOSLEEP();
749         IP6Q_LOCK();
750         VNET_FOREACH(vnet_iter) {
751                 CURVNET_SET(vnet_iter);
752                 q6 = V_ip6q.ip6q_next;
753                 if (q6)
754                         while (q6 != &V_ip6q) {
755                                 --q6->ip6q_ttl;
756                                 q6 = q6->ip6q_next;
757                                 if (q6->ip6q_prev->ip6q_ttl == 0) {
758                                         IP6STAT_INC(ip6s_fragtimeout);
759                                         /* XXX in6_ifstat_inc(ifp, ifs6_reass_fail) */
760                                         frag6_freef(q6->ip6q_prev);
761                                 }
762                         }
763                 /*
764                  * If we are over the maximum number of fragments
765                  * (due to the limit being lowered), drain off
766                  * enough to get down to the new limit.
767                  */
768                 while (V_frag6_nfragpackets > (u_int)V_ip6_maxfragpackets &&
769                     V_ip6q.ip6q_prev) {
770                         IP6STAT_INC(ip6s_fragoverflow);
771                         /* XXX in6_ifstat_inc(ifp, ifs6_reass_fail) */
772                         frag6_freef(V_ip6q.ip6q_prev);
773                 }
774                 CURVNET_RESTORE();
775         }
776         IP6Q_UNLOCK();
777         VNET_LIST_RUNLOCK_NOSLEEP();
778 }
779
780 /*
781  * Drain off all datagram fragments.
782  */
783 void
784 frag6_drain(void)
785 {
786         VNET_ITERATOR_DECL(vnet_iter);
787
788         VNET_LIST_RLOCK_NOSLEEP();
789         if (IP6Q_TRYLOCK() == 0) {
790                 VNET_LIST_RUNLOCK_NOSLEEP();
791                 return;
792         }
793         VNET_FOREACH(vnet_iter) {
794                 CURVNET_SET(vnet_iter);
795                 while (V_ip6q.ip6q_next != &V_ip6q) {
796                         IP6STAT_INC(ip6s_fragdropped);
797                         /* XXX in6_ifstat_inc(ifp, ifs6_reass_fail) */
798                         frag6_freef(V_ip6q.ip6q_next);
799                 }
800                 CURVNET_RESTORE();
801         }
802         IP6Q_UNLOCK();
803         VNET_LIST_RUNLOCK_NOSLEEP();
804 }
805
806 int
807 ip6_deletefraghdr(struct mbuf *m, int offset, int wait)
808 {
809         struct ip6_hdr *ip6 = mtod(m, struct ip6_hdr *);
810         struct mbuf *t;
811
812         /* Delete frag6 header. */
813         if (m->m_len >= offset + sizeof(struct ip6_frag)) {
814                 /* This is the only possible case with !PULLDOWN_TEST. */
815                 bcopy(ip6, (char *)ip6 + sizeof(struct ip6_frag),
816                     offset);
817                 m->m_data += sizeof(struct ip6_frag);
818                 m->m_len -= sizeof(struct ip6_frag);
819         } else {
820                 /* This comes with no copy if the boundary is on cluster. */
821                 if ((t = m_split(m, offset, wait)) == NULL)
822                         return (ENOMEM);
823                 m_adj(t, sizeof(struct ip6_frag));
824                 m_cat(m, t);
825         }
826
827         m->m_flags |= M_FRAGMENTED;
828         return (0);
829 }