]> git.saurik.com Git - apple/xnu.git/blobdiff - bsd/netinet6/frag6.c
xnu-1504.9.17.tar.gz
[apple/xnu.git] / bsd / netinet6 / frag6.c
index f686d01cd6d6c034029cfafdf3e75b4dbe9138f7..ea75e5acb94665d90a26d726fc00c142f2d3309f 100644 (file)
@@ -1,3 +1,31 @@
+/*
+ * Copyright (c) 2008 Apple Inc. All rights reserved.
+ *
+ * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
+ * 
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. The rights granted to you under the License
+ * may not be used to create, or enable the creation or redistribution of,
+ * unlawful or unlicensed copies of an Apple operating system, or to
+ * circumvent, violate, or enable the circumvention or violation of, any
+ * terms of an Apple operating system software license agreement.
+ * 
+ * Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this file.
+ * 
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ * 
+ * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
+ */
+
 /*     $FreeBSD: src/sys/netinet6/frag6.c,v 1.2.2.5 2001/07/03 11:01:50 ume Exp $      */
 /*     $KAME: frag6.c,v 1.31 2001/05/17 13:45:34 jinmei Exp $  */
 
@@ -42,6 +70,7 @@
 #include <sys/kernel.h>
 #include <sys/syslog.h>
 #include <kern/queue.h>
+#include <kern/locks.h>
 
 #include <net/if.h>
 #include <net/route.h>
  */
 #define IN6_IFSTAT_STRICT
 
-static void frag6_enq __P((struct ip6asfrag *, struct ip6asfrag *));
-static void frag6_deq __P((struct ip6asfrag *));
-static void frag6_insque __P((struct ip6q *, struct ip6q *));
-static void frag6_remque __P((struct ip6q *));
-static void frag6_freef __P((struct ip6q *));
+static void frag6_enq(struct ip6asfrag *, struct ip6asfrag *);
+static void frag6_deq(struct ip6asfrag *);
+static void frag6_insque(struct ip6q *, struct ip6q *);
+static void frag6_remque(struct ip6q *);
+static void frag6_freef(struct ip6q *);
 
 /* XXX we eventually need splreass6, or some real semaphore */
 int frag6_doing_reass;
 u_int frag6_nfragpackets;
+static u_int frag6_nfrags;
 struct ip6q ip6q;      /* ip6 reassemble queue */
 
 #ifndef __APPLE__
 MALLOC_DEFINE(M_FTABLE, "fragment", "fragment reassembly header");
 #endif
 
+extern lck_mtx_t *inet6_domain_mutex;
 /*
  * Initialise reassembly queue and fragment identifier.
  */
@@ -85,6 +116,7 @@ frag6_init()
        struct timeval tv;
 
        ip6_maxfragpackets = nmbclusters / 32;
+       ip6_maxfrags = nmbclusters / 4;
 
        /*
         * in many cases, random() here does NOT return random number
@@ -126,6 +158,8 @@ frag6_init()
  */
 /*
  * Fragment input
+ * NOTE: this function is called with the inet6_domain_mutex held from ip6_input.
+ *      inet6_domain_mutex is protecting he frag6 queue manipulation.
  */
 int
 frag6_input(mp, offp)
@@ -141,14 +175,15 @@ frag6_input(mp, offp)
        int first_frag = 0;
        int fragoff, frgpartlen;        /* must be larger than u_int16_t */
        struct ifnet *dstifp;
+       struct ifaddr *ifa = NULL;
 #ifdef IN6_IFSTAT_STRICT
-       static struct route_in6 ro;
+       struct route_in6 ro;
        struct sockaddr_in6 *dst;
 #endif
 
        ip6 = mtod(m, struct ip6_hdr *);
 #ifndef PULLDOWN_TEST
-       IP6_EXTHDR_CHECK(m, offset, sizeof(struct ip6_frag), IPPROTO_DONE);
+       IP6_EXTHDR_CHECK(m, offset, sizeof(struct ip6_frag), return IPPROTO_DONE);
        ip6f = (struct ip6_frag *)((caddr_t)ip6 + offset);
 #else
        IP6_EXTHDR_GET(ip6f, struct ip6_frag *, m, offset, sizeof(*ip6f));
@@ -159,22 +194,23 @@ frag6_input(mp, offp)
        dstifp = NULL;
 #ifdef IN6_IFSTAT_STRICT
        /* find the destination interface of the packet. */
+       bzero(&ro, sizeof (ro));
        dst = (struct sockaddr_in6 *)&ro.ro_dst;
-       if (ro.ro_rt
-        && ((ro.ro_rt->rt_flags & RTF_UP) == 0
-         || !IN6_ARE_ADDR_EQUAL(&dst->sin6_addr, &ip6->ip6_dst))) {
+       dst->sin6_family = AF_INET6;
+       dst->sin6_len = sizeof (struct sockaddr_in6);
+       dst->sin6_addr = ip6->ip6_dst;
+
+       rtalloc((struct route *)&ro);
+       if (ro.ro_rt != NULL) {
+               RT_LOCK(ro.ro_rt);
+               if ((ifa = ro.ro_rt->rt_ifa) != NULL) {
+                       ifaref(ifa);
+                       dstifp = ((struct in6_ifaddr *)ro.ro_rt->rt_ifa)->ia_ifp;
+               }
+               RT_UNLOCK(ro.ro_rt);
                rtfree(ro.ro_rt);
-               ro.ro_rt = (struct rtentry *)0;
-       }
-       if (ro.ro_rt == NULL) {
-               bzero(dst, sizeof(*dst));
-               dst->sin6_family = AF_INET6;
-               dst->sin6_len = sizeof(struct sockaddr_in6);
-               dst->sin6_addr = ip6->ip6_dst;
+               ro.ro_rt = NULL;
        }
-       rtalloc((struct route *)&ro);
-       if (ro.ro_rt != NULL && ro.ro_rt->rt_ifa != NULL)
-               dstifp = ((struct in6_ifaddr *)ro.ro_rt->rt_ifa)->ia_ifp;
 #else
        /* we are violating the spec, this is not the destination interface */
        if ((m->m_flags & M_PKTHDR) != 0)
@@ -185,6 +221,8 @@ frag6_input(mp, offp)
        if (ip6->ip6_plen == 0) {
                icmp6_error(m, ICMP6_PARAM_PROB, ICMP6_PARAMPROB_HEADER, offset);
                in6_ifstat_inc(dstifp, ifs6_reass_fail);
+               if (ifa != NULL)
+                       ifafree(ifa);
                return IPPROTO_DONE;
        }
 
@@ -200,6 +238,8 @@ frag6_input(mp, offp)
                            ICMP6_PARAMPROB_HEADER,
                            offsetof(struct ip6_hdr, ip6_plen));
                in6_ifstat_inc(dstifp, ifs6_reass_fail);
+               if (ifa != NULL)
+                       ifafree(ifa);
                return IPPROTO_DONE;
        }
 
@@ -211,6 +251,16 @@ frag6_input(mp, offp)
 
        frag6_doing_reass = 1;
 
+       /*
+        * Enforce upper bound on number of fragments.
+        * If maxfrag is 0, never accept fragments.
+        * If maxfrag is -1, accept all fragments without limitation.
+        */
+       if (ip6_maxfrags < 0)
+               ;
+       else if (frag6_nfrags >= (u_int)ip6_maxfrags)
+               goto dropfrag;
+
        for (q6 = ip6q.ip6q_next; q6 != &ip6q; q6 = q6->ip6q_next)
                if (ip6f->ip6f_ident == q6->ip6q_ident &&
                    IN6_ARE_ADDR_EQUAL(&ip6->ip6_src, &q6->ip6q_src) &&
@@ -253,6 +303,8 @@ frag6_input(mp, offp)
                q6->ip6q_src    = ip6->ip6_src;
                q6->ip6q_dst    = ip6->ip6_dst;
                q6->ip6q_unfrglen = -1; /* The 1st fragment has not arrived. */
+
+               q6->ip6q_nfrag  = 0;
        }
 
        /*
@@ -279,6 +331,8 @@ frag6_input(mp, offp)
                                    offset - sizeof(struct ip6_frag) +
                                        offsetof(struct ip6_frag, ip6f_offlg));
                        frag6_doing_reass = 0;
+                       if (ifa != NULL)
+                               ifafree(ifa);
                        return(IPPROTO_DONE);
                }
        }
@@ -287,6 +341,8 @@ frag6_input(mp, offp)
                            offset - sizeof(struct ip6_frag) +
                                offsetof(struct ip6_frag, ip6f_offlg));
                frag6_doing_reass = 0;
+               if (ifa != NULL)
+                       ifafree(ifa);
                return(IPPROTO_DONE);
        }
        /*
@@ -431,6 +487,8 @@ insert:
         * the most recently active fragmented packet.
         */
        frag6_enq(ip6af, af6->ip6af_up);
+       frag6_nfrags++;
+       q6->ip6q_nfrag++;
 #if 0 /* xxx */
        if (q6 != ip6q.ip6q_next) {
                frag6_remque(q6);
@@ -442,12 +500,16 @@ insert:
             af6 = af6->ip6af_down) {
                if (af6->ip6af_off != next) {
                        frag6_doing_reass = 0;
+                       if (ifa != NULL)
+                               ifafree(ifa);
                        return IPPROTO_DONE;
                }
                next += af6->ip6af_frglen;
        }
        if (af6->ip6af_up->ip6af_mff) {
                frag6_doing_reass = 0;
+               if (ifa != NULL)
+                       ifafree(ifa);
                return IPPROTO_DONE;
        }
 
@@ -493,6 +555,7 @@ insert:
                /* this comes with no copy if the boundary is on cluster */
                if ((t = m_split(m, offset, M_DONTWAIT)) == NULL) {
                        frag6_remque(q6);
+                       frag6_nfrags -= q6->ip6q_nfrag;
                        FREE(q6, M_FTABLE);
                        frag6_nfragpackets--;
                        goto dropfrag;
@@ -510,6 +573,7 @@ insert:
        }
 
        frag6_remque(q6);
+       frag6_nfrags -= q6->ip6q_nfrag;
        FREE(q6, M_FTABLE);
        frag6_nfragpackets--;
 
@@ -531,6 +595,8 @@ insert:
        *offp = offset;
 
        frag6_doing_reass = 0;
+       if (ifa != NULL)
+               ifafree(ifa);
        return nxt;
 
  dropfrag:
@@ -538,6 +604,8 @@ insert:
        ip6stat.ip6s_fragdropped++;
        m_freem(m);
        frag6_doing_reass = 0;
+       if (ifa != NULL)
+               ifafree(ifa);
        return IPPROTO_DONE;
 }
 
@@ -571,7 +639,6 @@ frag6_freef(q6)
                        /* restoure source and destination addresses */
                        ip6->ip6_src = q6->ip6q_src;
                        ip6->ip6_dst = q6->ip6q_dst;
-
                        icmp6_error(m, ICMP6_TIME_EXCEEDED,
                                    ICMP6_TIME_EXCEED_REASSEMBLY, 0);
                } else
@@ -580,6 +647,7 @@ frag6_freef(q6)
 
        }
        frag6_remque(q6);
+       frag6_nfrags -= q6->ip6q_nfrag;
        FREE(q6, M_FTABLE);
        frag6_nfragpackets--;
 }
@@ -636,7 +704,7 @@ void
 frag6_slowtimo()
 {
        struct ip6q *q6;
-       int s = splnet();
+       lck_mtx_lock(inet6_domain_mutex);
 
        frag6_doing_reass = 1;
        q6 = ip6q.ip6q_next;
@@ -662,24 +730,7 @@ frag6_slowtimo()
                frag6_freef(ip6q.ip6q_prev);
        }
        frag6_doing_reass = 0;
-
-#if 0
-       /*
-        * Routing changes might produce a better route than we last used;
-        * make sure we notice eventually, even if forwarding only for one
-        * destination and the cache is never replaced.
-        */
-       if (ip6_forward_rt.ro_rt) {
-               rtfree(ip6_forward_rt.ro_rt);
-               ip6_forward_rt.ro_rt = 0;
-       }
-       if (ipsrcchk_rt.ro_rt) {
-               rtfree(ipsrcchk_rt.ro_rt);
-               ipsrcchk_rt.ro_rt = 0;
-       }
-#endif
-
-       splx(s);
+       lck_mtx_unlock(inet6_domain_mutex);
 }
 
 /*
@@ -690,9 +741,11 @@ frag6_drain()
 {
        if (frag6_doing_reass)
                return;
+       lck_mtx_lock(inet6_domain_mutex);
        while (ip6q.ip6q_next != &ip6q) {
                ip6stat.ip6s_fragdropped++;
                /* XXX in6_ifstat_inc(ifp, ifs6_reass_fail) */
                frag6_freef(ip6q.ip6q_next);
        }
+       lck_mtx_unlock(inet6_domain_mutex);
 }