]> git.saurik.com Git - apple/xnu.git/blobdiff - bsd/kern/kpi_mbuf.c
xnu-4570.71.2.tar.gz
[apple/xnu.git] / bsd / kern / kpi_mbuf.c
index 9ea0ae567ba6bbb20d7913bccf00aee0ac48b6ca..6cb79b10e8bcc66f8b7a159f709ea7e32f4b82a0 100644 (file)
@@ -1,8 +1,8 @@
 /*
- * Copyright (c) 2004 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 2004-2017 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
  * 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,
  * 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@
  */
 
-#define __KPI__
-//#include <sys/kpi_interface.h>
+#define        __KPI__
 
 #include <sys/param.h>
 #include <sys/mbuf.h>
+#include <sys/mcache.h>
 #include <sys/socket.h>
 #include <kern/debug.h>
 #include <libkern/OSAtomic.h>
 #include <kern/kalloc.h>
 #include <string.h>
+#include <net/dlil.h>
+#include <netinet/in.h>
+#include <netinet/ip_var.h>
+
+#include "net/net_str_id.h"
+
+/* mbuf flags visible to KPI clients; do not add private flags here */
+static const mbuf_flags_t mbuf_flags_mask = (MBUF_EXT | MBUF_PKTHDR | MBUF_EOR |
+    MBUF_LOOP | MBUF_BCAST | MBUF_MCAST | MBUF_FRAG | MBUF_FIRSTFRAG |
+    MBUF_LASTFRAG | MBUF_PROMISC | MBUF_HASFCS);
+
+/* Unalterable mbuf flags */
+static const mbuf_flags_t mbuf_cflags_mask = (MBUF_EXT);
+
+#define        MAX_MBUF_TX_COMPL_FUNC 32
+mbuf_tx_compl_func
+mbuf_tx_compl_table[MAX_MBUF_TX_COMPL_FUNC];
+extern lck_rw_t *mbuf_tx_compl_tbl_lock;
+u_int32_t mbuf_tx_compl_index = 0;
 
-void mbuf_tag_id_first_last(u_long *first, u_long *last);
-errno_t mbuf_tag_id_find_internal(const char *string, u_long *out_id, int create);
+#if (DEVELOPMENT || DEBUG)
+int mbuf_tx_compl_debug = 0;
+SInt64 mbuf_tx_compl_outstanding __attribute__((aligned(8))) = 0;
+u_int64_t mbuf_tx_compl_aborted __attribute__((aligned(8))) = 0;
 
-static const mbuf_flags_t mbuf_flags_mask = MBUF_EXT | MBUF_PKTHDR | MBUF_EOR |
-                               MBUF_BCAST | MBUF_MCAST | MBUF_FRAG | MBUF_FIRSTFRAG |
-                               MBUF_LASTFRAG | MBUF_PROMISC;
+SYSCTL_DECL(_kern_ipc);
+SYSCTL_NODE(_kern_ipc, OID_AUTO, mbtxcf,
+    CTLFLAG_RW | CTLFLAG_LOCKED, 0, "");
+SYSCTL_INT(_kern_ipc_mbtxcf, OID_AUTO, debug,
+    CTLFLAG_RW | CTLFLAG_LOCKED, &mbuf_tx_compl_debug, 0, "");
+SYSCTL_INT(_kern_ipc_mbtxcf, OID_AUTO, index,
+    CTLFLAG_RD | CTLFLAG_LOCKED, &mbuf_tx_compl_index, 0, "");
+SYSCTL_QUAD(_kern_ipc_mbtxcf, OID_AUTO, oustanding,
+    CTLFLAG_RD | CTLFLAG_LOCKED, &mbuf_tx_compl_outstanding, "");
+SYSCTL_QUAD(_kern_ipc_mbtxcf, OID_AUTO, aborted,
+    CTLFLAG_RD | CTLFLAG_LOCKED, &mbuf_tx_compl_aborted, "");
+#endif /* (DEBUG || DEVELOPMENT) */
 
-void* mbuf_data(mbuf_t mbuf)
+void *
+mbuf_data(mbuf_t mbuf)
 {
-       return m_mtod(mbuf);
+       return (mbuf->m_data);
 }
 
-void* mbuf_datastart(mbuf_t mbuf)
+void *
+mbuf_datastart(mbuf_t mbuf)
 {
        if (mbuf->m_flags & M_EXT)
-               return mbuf->m_ext.ext_buf;
+               return (mbuf->m_ext.ext_buf);
        if (mbuf->m_flags & M_PKTHDR)
-               return mbuf->m_pktdat;
-       return mbuf->m_dat;
+               return (mbuf->m_pktdat);
+       return (mbuf->m_dat);
 }
 
-errno_t mbuf_setdata(mbuf_t mbuf, void* data, size_t len)
+errno_t
+mbuf_setdata(mbuf_t mbuf, void *data, size_t len)
 {
-       size_t  start = (size_t)((char*)mbuf_datastart(mbuf));
+       size_t  start = (size_t)((char *)mbuf_datastart(mbuf));
        size_t  maxlen = mbuf_maxlen(mbuf);
-       
+
        if ((size_t)data < start || ((size_t)data) + len > start + maxlen)
-               return EINVAL;
+               return (EINVAL);
        mbuf->m_data = data;
        mbuf->m_len = len;
-       
-       return 0;
+
+       return (0);
 }
 
-errno_t mbuf_align_32(mbuf_t mbuf, size_t len)
+errno_t
+mbuf_align_32(mbuf_t mbuf, size_t len)
 {
        if ((mbuf->m_flags & M_EXT) != 0 && m_mclhasreference(mbuf))
-               return ENOTSUP;
+               return (ENOTSUP);
        mbuf->m_data = mbuf_datastart(mbuf);
-       mbuf->m_data += ((mbuf_trailingspace(mbuf) - len) &~ (sizeof(u_int32_t) - 1));
-       
-       return 0;
+       mbuf->m_data +=
+           ((mbuf_trailingspace(mbuf) - len) &~ (sizeof(u_int32_t) - 1));
+
+       return (0);
 }
 
-addr64_t mbuf_data_to_physical(void* ptr)
+/*
+ * This function is used to provide mcl_to_paddr via symbol indirection,
+ * please avoid any change in behavior or remove the indirection in
+ * config/Unsupported*
+ */
+addr64_t
+mbuf_data_to_physical(void *ptr)
 {
-       return (addr64_t)mcl_to_paddr(ptr);
+       return ((addr64_t)mcl_to_paddr(ptr));
 }
 
-errno_t mbuf_get(mbuf_how_t how, mbuf_type_t type, mbuf_t *mbuf)
+errno_t
+mbuf_get(mbuf_how_t how, mbuf_type_t type, mbuf_t *mbuf)
 {
        /* Must set *mbuf to NULL in failure case */
        *mbuf = m_get(how, type);
-       
-       return (*mbuf == NULL) ? ENOMEM : 0;
+
+       return (*mbuf == NULL ? ENOMEM : 0);
 }
 
-errno_t mbuf_gethdr(mbuf_how_t how, mbuf_type_t type, mbuf_t *mbuf)
+errno_t
+mbuf_gethdr(mbuf_how_t how, mbuf_type_t type, mbuf_t *mbuf)
 {
        /* Must set *mbuf to NULL in failure case */
        *mbuf = m_gethdr(how, type);
-       
-       return (*mbuf == NULL) ? ENOMEM : 0;
+
+       return (*mbuf == NULL ? ENOMEM : 0);
+}
+
+errno_t
+mbuf_attachcluster(mbuf_how_t how, mbuf_type_t type, mbuf_t *mbuf,
+    caddr_t extbuf, void (*extfree)(caddr_t, u_int, caddr_t),
+    size_t extsize, caddr_t extarg)
+{
+       if (mbuf == NULL || extbuf == NULL || extfree == NULL || extsize == 0)
+               return (EINVAL);
+
+       if ((*mbuf = m_clattach(*mbuf, type, extbuf,
+           extfree, extsize, extarg, how, 0)) == NULL)
+               return (ENOMEM);
+
+       return (0);
+}
+
+errno_t
+mbuf_ring_cluster_alloc(mbuf_how_t how, mbuf_type_t type, mbuf_t *mbuf,
+    void (*extfree)(caddr_t, u_int, caddr_t), size_t *size)
+{
+       caddr_t extbuf = NULL;
+       errno_t err;
+
+       if (mbuf == NULL || extfree == NULL || size == NULL || *size == 0)
+               return (EINVAL);
+
+       if ((err = mbuf_alloccluster(how, size, &extbuf)) != 0)
+               return (err);
+
+       if ((*mbuf = m_clattach(*mbuf, type, extbuf,
+           extfree, *size, NULL, how, 1)) == NULL) {
+               mbuf_freecluster(extbuf, *size);
+               return (ENOMEM);
+       }
+
+       return (0);
+}
+
+int
+mbuf_ring_cluster_is_active(mbuf_t mbuf)
+{
+       return (m_ext_paired_is_active(mbuf));
+}
+
+errno_t
+mbuf_ring_cluster_activate(mbuf_t mbuf)
+{
+       if (mbuf_ring_cluster_is_active(mbuf))
+               return (EBUSY);
+
+       m_ext_paired_activate(mbuf);
+       return (0);
+}
+
+errno_t
+mbuf_cluster_set_prop(mbuf_t mbuf, u_int32_t oldprop, u_int32_t newprop)
+{
+       if (mbuf == NULL || !(mbuf->m_flags & M_EXT))
+               return (EINVAL);
+
+       return (m_ext_set_prop(mbuf, oldprop, newprop) ? 0 : EBUSY);
+}
+
+errno_t
+mbuf_cluster_get_prop(mbuf_t mbuf, u_int32_t *prop)
+{
+       if (mbuf == NULL || prop == NULL || !(mbuf->m_flags & M_EXT))
+               return (EINVAL);
+
+       *prop = m_ext_get_prop(mbuf);
+       return (0);
+}
+
+errno_t
+mbuf_alloccluster(mbuf_how_t how, size_t *size, caddr_t *addr)
+{
+       if (size == NULL || *size == 0 || addr == NULL)
+               return (EINVAL);
+
+       *addr = NULL;
+
+       /* Jumbo cluster pool not available? */
+       if (*size > MBIGCLBYTES && njcl == 0)
+               return (ENOTSUP);
+
+       if (*size <= MCLBYTES && (*addr = m_mclalloc(how)) != NULL)
+               *size = MCLBYTES;
+       else if (*size > MCLBYTES && *size <= MBIGCLBYTES &&
+           (*addr = m_bigalloc(how)) != NULL)
+               *size = MBIGCLBYTES;
+       else if (*size > MBIGCLBYTES && *size <= M16KCLBYTES &&
+           (*addr = m_16kalloc(how)) != NULL)
+               *size = M16KCLBYTES;
+       else
+               *size = 0;
+
+       if (*addr == NULL)
+               return (ENOMEM);
+
+       return (0);
 }
 
-extern struct mbuf * m_mbigget(struct mbuf *m, int nowait);
+void
+mbuf_freecluster(caddr_t addr, size_t size)
+{
+       if (size != MCLBYTES && size != MBIGCLBYTES && size != M16KCLBYTES)
+               panic("%s: invalid size (%ld) for cluster %p", __func__,
+                   size, (void *)addr);
+
+       if (size == MCLBYTES)
+               m_mclfree(addr);
+       else if (size == MBIGCLBYTES)
+               m_bigfree(addr, MBIGCLBYTES, NULL);
+       else if (njcl > 0)
+               m_16kfree(addr, M16KCLBYTES, NULL);
+       else
+               panic("%s: freeing jumbo cluster to an empty pool", __func__);
+}
 
-errno_t mbuf_getcluster(mbuf_how_t how, mbuf_type_t type, size_t size, mbuf_t* mbuf)
+errno_t
+mbuf_getcluster(mbuf_how_t how, mbuf_type_t type, size_t size, mbuf_t *mbuf)
 {
        /* Must set *mbuf to NULL in failure case */
        errno_t error = 0;
-       int             created = 0;
+       int     created = 0;
 
        if (mbuf == NULL)
-               return EINVAL;
+               return (EINVAL);
        if (*mbuf == NULL) {
                *mbuf = m_get(how, type);
                if (*mbuf == NULL)
-                       return ENOMEM;
+                       return (ENOMEM);
                created = 1;
        }
        /*
-        * At the time this code was written, m_mclget and m_mbigget would always
-        * return the same value that was passed in to it.
+        * At the time this code was written, m_{mclget,mbigget,m16kget}
+        * would always return the same value that was passed in to it.
         */
        if (size == MCLBYTES) {
                *mbuf = m_mclget(*mbuf, how);
-       } else if (size == NBPG) {
+       } else if (size == MBIGCLBYTES) {
                *mbuf = m_mbigget(*mbuf, how);
+       } else if (size == M16KCLBYTES) {
+               if (njcl > 0) {
+                       *mbuf = m_m16kget(*mbuf, how);
+               } else {
+                       /* Jumbo cluster pool not available? */
+                       error = ENOTSUP;
+                       goto out;
+               }
        } else {
                error = EINVAL;
                goto out;
@@ -134,150 +302,212 @@ errno_t mbuf_getcluster(mbuf_how_t how, mbuf_type_t type, size_t size, mbuf_t* m
                error = ENOMEM;
 out:
        if (created && error != 0) {
-               error = ENOMEM;
                mbuf_free(*mbuf);
                *mbuf = NULL;
        }
-       return error;   
+       return (error);
 }
 
-errno_t mbuf_mclget(mbuf_how_t how, mbuf_type_t type, mbuf_t *mbuf)
+errno_t
+mbuf_mclget(mbuf_how_t how, mbuf_type_t type, mbuf_t *mbuf)
 {
        /* Must set *mbuf to NULL in failure case */
        errno_t error = 0;
        int             created = 0;
-       if (mbuf == NULL) return EINVAL;
+       if (mbuf == NULL)
+               return (EINVAL);
        if (*mbuf == NULL) {
                error = mbuf_get(how, type, mbuf);
                if (error)
-                       return error;
+                       return (error);
                created = 1;
        }
-       
+
        /*
         * At the time this code was written, m_mclget would always
         * return the same value that was passed in to it.
         */
        *mbuf = m_mclget(*mbuf, how);
-       
+
        if (created && ((*mbuf)->m_flags & M_EXT) == 0) {
                mbuf_free(*mbuf);
                *mbuf = NULL;
        }
        if (*mbuf == NULL || ((*mbuf)->m_flags & M_EXT) == 0)
                error = ENOMEM;
-       return error;   
+       return (error);
 }
 
 
-errno_t mbuf_getpacket(mbuf_how_t how, mbuf_t *mbuf)
+errno_t
+mbuf_getpacket(mbuf_how_t how, mbuf_t *mbuf)
 {
        /* Must set *mbuf to NULL in failure case */
        errno_t error = 0;
-       
+
        *mbuf = m_getpacket_how(how);
-       
+
        if (*mbuf == NULL) {
                if (how == MBUF_WAITOK)
                        error = ENOMEM;
                else
                        error = EWOULDBLOCK;
        }
-       
-       return error;
+
+       return (error);
 }
 
-mbuf_t mbuf_free(mbuf_t mbuf)
+/*
+ * This function is used to provide m_free via symbol indirection, please avoid
+ * any change in behavior or remove the indirection in config/Unsupported*
+ */
+mbuf_t
+mbuf_free(mbuf_t mbuf)
 {
-       return m_free(mbuf);
+       return (m_free(mbuf));
 }
 
-void mbuf_freem(mbuf_t mbuf)
+/*
+ * This function is used to provide m_freem via symbol indirection, please avoid
+ * any change in behavior or remove the indirection in config/Unsupported*
+ */
+void
+mbuf_freem(mbuf_t mbuf)
 {
        m_freem(mbuf);
 }
 
-int    mbuf_freem_list(mbuf_t mbuf)
+int
+mbuf_freem_list(mbuf_t mbuf)
 {
-       return m_freem_list(mbuf);
+       return (m_freem_list(mbuf));
 }
 
-size_t mbuf_leadingspace(mbuf_t mbuf)
+size_t
+mbuf_leadingspace(const mbuf_t mbuf)
 {
-       return m_leadingspace(mbuf);
+       return (m_leadingspace(mbuf));
 }
 
-size_t mbuf_trailingspace(mbuf_t mbuf)
+/*
+ * This function is used to provide m_trailingspace via symbol indirection,
+ * please avoid any change in behavior or remove the indirection in
+ * config/Unsupported*
+ */
+size_t
+mbuf_trailingspace(const mbuf_t mbuf)
 {
-       return m_trailingspace(mbuf);
+       return (m_trailingspace(mbuf));
 }
 
 /* Manipulation */
-errno_t mbuf_copym(mbuf_t src, size_t offset, size_t len,
-                                  mbuf_how_t how, mbuf_t *new_mbuf)
+errno_t
+mbuf_copym(const mbuf_t src, size_t offset, size_t len,
+    mbuf_how_t how, mbuf_t *new_mbuf)
 {
        /* Must set *mbuf to NULL in failure case */
        *new_mbuf = m_copym(src, offset, len, how);
-       
-       return (*new_mbuf == NULL) ? ENOMEM : 0;
+
+       return (*new_mbuf == NULL ? ENOMEM : 0);
 }
 
-errno_t        mbuf_dup(mbuf_t src, mbuf_how_t how, mbuf_t *new_mbuf)
+errno_t
+mbuf_dup(const mbuf_t src, mbuf_how_t how, mbuf_t *new_mbuf)
 {
        /* Must set *new_mbuf to NULL in failure case */
        *new_mbuf = m_dup(src, how);
-       
-       return (*new_mbuf == NULL) ? ENOMEM : 0;
+
+       return (*new_mbuf == NULL ? ENOMEM : 0);
 }
 
-errno_t mbuf_prepend(mbuf_t *orig, size_t len, mbuf_how_t how)
+errno_t
+mbuf_prepend(mbuf_t *orig, size_t len, mbuf_how_t how)
 {
        /* Must set *orig to NULL in failure case */
-       *orig = m_prepend_2(*orig, len, how);
-       
-       return (*orig == NULL) ? ENOMEM : 0;
+       *orig = m_prepend_2(*orig, len, how, 0);
+
+       return (*orig == NULL ? ENOMEM : 0);
 }
 
-errno_t mbuf_split(mbuf_t src, size_t offset,
+errno_t
+mbuf_split(mbuf_t src, size_t offset,
                                        mbuf_how_t how, mbuf_t *new_mbuf)
 {
        /* Must set *new_mbuf to NULL in failure case */
        *new_mbuf = m_split(src, offset, how);
-       
-       return (*new_mbuf == NULL) ? ENOMEM : 0;
+
+       return (*new_mbuf == NULL ? ENOMEM : 0);
 }
 
-errno_t mbuf_pullup(mbuf_t *mbuf, size_t len)
+errno_t
+mbuf_pullup(mbuf_t *mbuf, size_t len)
 {
        /* Must set *mbuf to NULL in failure case */
        *mbuf = m_pullup(*mbuf, len);
-       
-       return (*mbuf == NULL) ? ENOMEM : 0;
+
+       return (*mbuf == NULL ? ENOMEM : 0);
 }
 
-errno_t mbuf_pulldown(mbuf_t src, size_t *offset, size_t len, mbuf_t *location)
+errno_t
+mbuf_pulldown(mbuf_t src, size_t *offset, size_t len, mbuf_t *location)
 {
        /* Must set *location to NULL in failure case */
        int new_offset;
        *location = m_pulldown(src, *offset, len, &new_offset);
        *offset = new_offset;
-       
-       return (*location == NULL) ? ENOMEM : 0;
+
+       return (*location == NULL ? ENOMEM : 0);
 }
 
-void mbuf_adj(mbuf_t mbuf, int len)
+/*
+ * This function is used to provide m_adj via symbol indirection, please avoid
+ * any change in behavior or remove the indirection in config/Unsupported*
+ */
+void
+mbuf_adj(mbuf_t mbuf, int len)
 {
        m_adj(mbuf, len);
 }
 
-errno_t mbuf_copydata(mbuf_t m, size_t off, size_t len, void* out_data)
+errno_t
+mbuf_adjustlen(mbuf_t m, int amount)
+{
+       /* Verify m_len will be valid after adding amount */
+       if (amount > 0) {
+               int used = (size_t)mbuf_data(m) - (size_t)mbuf_datastart(m) +
+                   m->m_len;
+
+               if ((size_t)(amount + used) > mbuf_maxlen(m))
+                       return (EINVAL);
+       } else if (-amount > m->m_len) {
+               return (EINVAL);
+       }
+
+       m->m_len += amount;
+       return (0);
+}
+
+mbuf_t
+mbuf_concatenate(mbuf_t dst, mbuf_t src)
+{
+       if (dst == NULL)
+               return (NULL);
+
+       m_cat(dst, src);
+
+       /* return dst as is in the current implementation */
+       return (dst);
+}
+errno_t
+mbuf_copydata(const mbuf_t m0, size_t off, size_t len, void *out_data)
 {
        /* Copied m_copydata, added error handling (don't just panic) */
        int count;
+       mbuf_t  m = m0;
 
        while (off > 0) {
                if (m == 0)
-                       return EINVAL;
+                       return (EINVAL);
                if (off < (size_t)m->m_len)
                        break;
                off -= m->m_len;
@@ -285,174 +515,232 @@ errno_t mbuf_copydata(mbuf_t m, size_t off, size_t len, void* out_data)
        }
        while (len > 0) {
                if (m == 0)
-                       return EINVAL;
+                       return (EINVAL);
                count = m->m_len - off > len ? len : m->m_len - off;
                bcopy(mtod(m, caddr_t) + off, out_data, count);
                len -= count;
-               out_data = ((char*)out_data) + count;
+               out_data = ((char *)out_data) + count;
                off = 0;
                m = m->m_next;
        }
-       
-       return 0;
-}
-
-int mbuf_mclref(mbuf_t mbuf)
-{
-       return m_mclref(mbuf);
-}
 
-int mbuf_mclunref(mbuf_t mbuf)
-{
-       return m_mclunref(mbuf);
+       return (0);
 }
 
-int mbuf_mclhasreference(mbuf_t mbuf)
+int
+mbuf_mclhasreference(mbuf_t mbuf)
 {
        if ((mbuf->m_flags & M_EXT))
-               return m_mclhasreference(mbuf);
+               return (m_mclhasreference(mbuf));
        else
-               return 0;
+               return (0);
 }
 
 
 /* mbuf header */
-mbuf_t mbuf_next(mbuf_t mbuf)
+mbuf_t
+mbuf_next(const mbuf_t mbuf)
 {
-       return mbuf->m_next;
+       return (mbuf->m_next);
 }
 
-errno_t mbuf_setnext(mbuf_t mbuf, mbuf_t next)
+errno_t
+mbuf_setnext(mbuf_t mbuf, mbuf_t next)
 {
        if (next && ((next)->m_nextpkt != NULL ||
-               (next)->m_type == MT_FREE)) return EINVAL;
+           (next)->m_type == MT_FREE))
+               return (EINVAL);
        mbuf->m_next = next;
-       
-       return 0;
+
+       return (0);
 }
 
-mbuf_t mbuf_nextpkt(mbuf_t mbuf)
+mbuf_t
+mbuf_nextpkt(const mbuf_t mbuf)
 {
-       return mbuf->m_nextpkt;
+       return (mbuf->m_nextpkt);
 }
 
-void mbuf_setnextpkt(mbuf_t mbuf, mbuf_t nextpkt)
+void
+mbuf_setnextpkt(mbuf_t mbuf, mbuf_t nextpkt)
 {
        mbuf->m_nextpkt = nextpkt;
 }
 
-size_t mbuf_len(mbuf_t mbuf)
+size_t
+mbuf_len(const mbuf_t mbuf)
 {
-       return mbuf->m_len;
+       return (mbuf->m_len);
 }
 
-void mbuf_setlen(mbuf_t mbuf, size_t len)
+void
+mbuf_setlen(mbuf_t mbuf, size_t len)
 {
        mbuf->m_len = len;
 }
 
-size_t mbuf_maxlen(mbuf_t mbuf)
+size_t
+mbuf_maxlen(const mbuf_t mbuf)
 {
        if (mbuf->m_flags & M_EXT)
-               return mbuf->m_ext.ext_size;
-       return &mbuf->m_dat[MLEN] - ((char*)mbuf_datastart(mbuf));
+               return (mbuf->m_ext.ext_size);
+       return (&mbuf->m_dat[MLEN] - ((char *)mbuf_datastart(mbuf)));
 }
 
-mbuf_type_t mbuf_type(mbuf_t mbuf)
+mbuf_type_t
+mbuf_type(const mbuf_t mbuf)
 {
-       return mbuf->m_type;
+       return (mbuf->m_type);
 }
 
-errno_t mbuf_settype(mbuf_t mbuf, mbuf_type_t new_type)
+errno_t
+mbuf_settype(mbuf_t mbuf, mbuf_type_t new_type)
 {
-       if (new_type == MBUF_TYPE_FREE) return EINVAL;
-       
+       if (new_type == MBUF_TYPE_FREE)
+               return (EINVAL);
+
        m_mchtype(mbuf, new_type);
-       
-       return 0;
+
+       return (0);
 }
 
-mbuf_flags_t mbuf_flags(mbuf_t mbuf)
+mbuf_flags_t
+mbuf_flags(const mbuf_t mbuf)
 {
-       return mbuf->m_flags & mbuf_flags_mask;
+       return (mbuf->m_flags & mbuf_flags_mask);
 }
 
-errno_t mbuf_setflags(mbuf_t mbuf, mbuf_flags_t flags)
+errno_t
+mbuf_setflags(mbuf_t mbuf, mbuf_flags_t flags)
 {
-       if ((flags & ~mbuf_flags_mask) != 0) return EINVAL;
-       mbuf->m_flags = flags |
-               (mbuf->m_flags & ~mbuf_flags_mask);
-       
-       return 0;
+       errno_t ret = 0;
+       mbuf_flags_t oflags = mbuf->m_flags;
+
+       /*
+        * 1. Return error if public but un-alterable flags are changed
+        *    in flags argument.
+        * 2. Return error if bits other than public flags are set in passed
+        *    flags argument.
+        *    Please note that private flag bits must be passed as reset by
+        *    kexts, as they must use mbuf_flags KPI to get current set of
+        *    mbuf flags and mbuf_flags KPI does not expose private flags.
+        */
+       if ((flags ^ oflags) & mbuf_cflags_mask) {
+               ret = EINVAL;
+       } else if (flags & ~mbuf_flags_mask) {
+               ret = EINVAL;
+       } else {
+               mbuf->m_flags = flags | (mbuf->m_flags & ~mbuf_flags_mask);
+               /*
+                * If M_PKTHDR bit has changed, we have work to do;
+                * m_reinit() will take care of setting/clearing the
+                * bit, as well as the rest of bookkeeping.
+                */
+               if ((oflags ^ mbuf->m_flags) & M_PKTHDR) {
+                       mbuf->m_flags ^= M_PKTHDR;      /* restore */
+                       ret = m_reinit(mbuf,
+                           (mbuf->m_flags & M_PKTHDR) ? 0 : 1);
+               }
+       }
+
+       return (ret);
 }
 
-errno_t mbuf_setflags_mask(mbuf_t mbuf, mbuf_flags_t flags, mbuf_flags_t mask)
+errno_t
+mbuf_setflags_mask(mbuf_t mbuf, mbuf_flags_t flags, mbuf_flags_t mask)
 {
-       if (((flags | mask) & ~mbuf_flags_mask) != 0) return EINVAL;
-       
-       mbuf->m_flags = (flags & mask) | (mbuf->m_flags & ~mask);
-       
-       return 0;
+       errno_t ret = 0;
+
+       if (mask & (~mbuf_flags_mask | mbuf_cflags_mask)) {
+               ret = EINVAL;
+       } else {
+               mbuf_flags_t oflags = mbuf->m_flags;
+               mbuf->m_flags = (flags & mask) | (mbuf->m_flags & ~mask);
+               /*
+                * If M_PKTHDR bit has changed, we have work to do;
+                * m_reinit() will take care of setting/clearing the
+                * bit, as well as the rest of bookkeeping.
+                */
+               if ((oflags ^ mbuf->m_flags) & M_PKTHDR) {
+                       mbuf->m_flags ^= M_PKTHDR;      /* restore */
+                       ret = m_reinit(mbuf,
+                           (mbuf->m_flags & M_PKTHDR) ? 0 : 1);
+               }
+       }
+
+       return (ret);
 }
 
-errno_t mbuf_copy_pkthdr(mbuf_t dest, mbuf_t src)
+errno_t
+mbuf_copy_pkthdr(mbuf_t dest, const mbuf_t src)
 {
        if (((src)->m_flags & M_PKTHDR) == 0)
-               return EINVAL;
-       
+               return (EINVAL);
+
        m_copy_pkthdr(dest, src);
-       
-       return 0;
-}
 
-size_t mbuf_pkthdr_len(mbuf_t mbuf)
-{
-       return mbuf->m_pkthdr.len;
+       return (0);
 }
 
-void mbuf_pkthdr_setlen(mbuf_t mbuf, size_t len)
+size_t
+mbuf_pkthdr_len(const mbuf_t mbuf)
 {
-       mbuf->m_pkthdr.len = len;
+       return (mbuf->m_pkthdr.len);
 }
 
-ifnet_t mbuf_pkthdr_rcvif(mbuf_t mbuf)
+__private_extern__ size_t
+mbuf_pkthdr_maxlen(mbuf_t m)
 {
-       // If we reference count ifnets, we should take a reference here before returning
-       return mbuf->m_pkthdr.rcvif;
+       size_t maxlen = 0;
+       mbuf_t n = m;
+
+       while (n) {
+               maxlen += mbuf_maxlen(n);
+               n = mbuf_next(n);
+       }
+       return (maxlen);
 }
 
-errno_t mbuf_pkthdr_setrcvif(mbuf_t mbuf, ifnet_t ifnet)
+void
+mbuf_pkthdr_setlen(mbuf_t mbuf, size_t len)
 {
-       /* May want to walk ifnet list to determine if interface is valid */
-       mbuf->m_pkthdr.rcvif = (struct ifnet*)ifnet;
-       return 0;
+       mbuf->m_pkthdr.len = len;
 }
 
-void* mbuf_pkthdr_header(mbuf_t mbuf)
+void
+mbuf_pkthdr_adjustlen(mbuf_t mbuf, int amount)
 {
-       return mbuf->m_pkthdr.header;
+       mbuf->m_pkthdr.len += amount;
 }
 
-void mbuf_pkthdr_setheader(mbuf_t mbuf, void *header)
+ifnet_t
+mbuf_pkthdr_rcvif(const mbuf_t mbuf)
 {
-       mbuf->m_pkthdr.header = (void*)header;
+       /*
+        * If we reference count ifnets, we should take a reference here
+        * before returning
+        */
+       return (mbuf->m_pkthdr.rcvif);
 }
 
-/* mbuf aux data */
-errno_t mbuf_aux_add(mbuf_t mbuf, int family, mbuf_type_t type, mbuf_t *aux_mbuf)
+errno_t
+mbuf_pkthdr_setrcvif(mbuf_t mbuf, ifnet_t ifnet)
 {
-       *aux_mbuf = m_aux_add(mbuf, family, type);
-       return (*aux_mbuf == NULL) ? ENOMEM : 0;
+       /* May want to walk ifnet list to determine if interface is valid */
+       mbuf->m_pkthdr.rcvif = (struct ifnet *)ifnet;
+       return (0);
 }
 
-mbuf_t mbuf_aux_find(mbuf_t mbuf, int family, mbuf_type_t type)
+void*
+mbuf_pkthdr_header(const mbuf_t mbuf)
 {
-       return m_aux_find(mbuf, family, type);
+       return (mbuf->m_pkthdr.pkt_hdr);
 }
 
-void mbuf_aux_delete(mbuf_t mbuf, mbuf_t aux)
+void
+mbuf_pkthdr_setheader(mbuf_t mbuf, void *header)
 {
-       m_aux_delete(mbuf, aux);
+       mbuf->m_pkthdr.pkt_hdr = (void*)header;
 }
 
 void
@@ -462,50 +750,28 @@ mbuf_inbound_modified(mbuf_t mbuf)
        mbuf->m_pkthdr.csum_flags = 0;
 }
 
-extern void in_cksum_offset(struct mbuf* m, size_t ip_offset);
-extern void in_delayed_cksum_offset(struct mbuf *m, int ip_offset);
-
 void
-mbuf_outbound_finalize(mbuf_t mbuf, u_long protocol_family, size_t protocol_offset)
+mbuf_outbound_finalize(struct mbuf *m, u_int32_t pf, size_t o)
 {
-       if ((mbuf->m_pkthdr.csum_flags &
-                (CSUM_DELAY_DATA | CSUM_DELAY_IP | CSUM_TCP_SUM16)) == 0)
-               return;
-       
        /* Generate the packet in software, client needs it */
-       switch (protocol_family) {
-               case PF_INET:
-                       if (mbuf->m_pkthdr.csum_flags & CSUM_TCP_SUM16) {
-                               /*
-                                * If you're wondering where this lovely code comes
-                                * from, we're trying to undo what happens in ip_output.
-                                * Look for CSUM_TCP_SUM16 in ip_output.
-                                */
-                               u_int16_t       first, second;
-                               mbuf->m_pkthdr.csum_flags &= ~CSUM_TCP_SUM16;
-                               mbuf->m_pkthdr.csum_flags |= CSUM_TCP;
-                               first = mbuf->m_pkthdr.csum_data >> 16;
-                               second = mbuf->m_pkthdr.csum_data & 0xffff;
-                               mbuf->m_pkthdr.csum_data = first - second;
-                       }
-                       if (mbuf->m_pkthdr.csum_flags & CSUM_DELAY_DATA) {
-                               in_delayed_cksum_offset(mbuf, protocol_offset);
-                       }
-                       
-                       if (mbuf->m_pkthdr.csum_flags & CSUM_DELAY_IP) {
-                               in_cksum_offset(mbuf, protocol_offset);
-                       }
-                       
-                       mbuf->m_pkthdr.csum_flags &= ~(CSUM_DELAY_DATA | CSUM_DELAY_IP);
-                       break;
-       
-               default:
-                       /*
-                        * Not sure what to do here if anything.
-                        * Hardware checksum code looked pretty IPv4 specific.
-                        */
-                       if ((mbuf->m_pkthdr.csum_flags & (CSUM_DELAY_DATA | CSUM_DELAY_IP)) != 0)
-                               panic("mbuf_outbound_finalize - CSUM flags set for non-IPv4 packet (%d)!\n", protocol_family);
+       switch (pf) {
+       case PF_INET:
+               (void) in_finalize_cksum(m, o, m->m_pkthdr.csum_flags);
+               break;
+
+       case PF_INET6:
+#if INET6
+               /*
+                * Checksum offload should not have been enabled when
+                * extension headers exist; indicate that the callee
+                * should skip such case by setting optlen to -1.
+                */
+               (void) in6_finalize_cksum(m, o, -1, -1, m->m_pkthdr.csum_flags);
+#endif /* INET6 */
+               break;
+
+       default:
+               break;
        }
 }
 
@@ -516,8 +782,8 @@ mbuf_set_vlan_tag(
 {
        mbuf->m_pkthdr.csum_flags |= CSUM_VLAN_TAG_VALID;
        mbuf->m_pkthdr.vlan_tag = vlan;
-       
-       return 0;
+
+       return (0);
 }
 
 errno_t
@@ -526,11 +792,11 @@ mbuf_get_vlan_tag(
        u_int16_t *vlan)
 {
        if ((mbuf->m_pkthdr.csum_flags & CSUM_VLAN_TAG_VALID) == 0)
-               return ENXIO; // No vlan tag set
-       
+               return (ENXIO); // No vlan tag set
+
        *vlan = mbuf->m_pkthdr.vlan_tag;
-       
-       return 0;
+
+       return (0);
 }
 
 errno_t
@@ -539,12 +805,13 @@ mbuf_clear_vlan_tag(
 {
        mbuf->m_pkthdr.csum_flags &= ~CSUM_VLAN_TAG_VALID;
        mbuf->m_pkthdr.vlan_tag = 0;
-       
-       return 0;
+
+       return (0);
 }
 
-static const mbuf_csum_request_flags_t mbuf_valid_csum_request_flags = 
-       MBUF_CSUM_REQ_IP | MBUF_CSUM_REQ_TCP | MBUF_CSUM_REQ_UDP | MBUF_CSUM_REQ_SUM16;
+static const mbuf_csum_request_flags_t mbuf_valid_csum_request_flags =
+    MBUF_CSUM_REQ_IP | MBUF_CSUM_REQ_TCP | MBUF_CSUM_REQ_UDP |
+    MBUF_CSUM_PARTIAL | MBUF_CSUM_REQ_TCPIPV6 | MBUF_CSUM_REQ_UDPIPV6;
 
 errno_t
 mbuf_set_csum_requested(
@@ -553,10 +820,32 @@ mbuf_set_csum_requested(
        u_int32_t value)
 {
        request &= mbuf_valid_csum_request_flags;
-       mbuf->m_pkthdr.csum_flags = (mbuf->m_pkthdr.csum_flags & 0xffff0000) | request;
+       mbuf->m_pkthdr.csum_flags =
+           (mbuf->m_pkthdr.csum_flags & 0xffff0000) | request;
        mbuf->m_pkthdr.csum_data = value;
-       
-       return 0;
+
+       return (0);
+}
+
+static const mbuf_tso_request_flags_t mbuf_valid_tso_request_flags =
+       MBUF_TSO_IPV4 | MBUF_TSO_IPV6;
+
+errno_t
+mbuf_get_tso_requested(
+       mbuf_t mbuf,
+       mbuf_tso_request_flags_t *request,
+       u_int32_t *value)
+{
+       if (mbuf == NULL || (mbuf->m_flags & M_PKTHDR) == 0 ||
+                       request == NULL || value == NULL)
+               return (EINVAL);
+
+       *request = mbuf->m_pkthdr.csum_flags;
+       *request &= mbuf_valid_tso_request_flags;
+       if (*request && value != NULL)
+               *value = mbuf->m_pkthdr.tso_segsz;
+
+       return (0);
 }
 
 errno_t
@@ -570,8 +859,8 @@ mbuf_get_csum_requested(
        if (value != NULL) {
                *value = mbuf->m_pkthdr.csum_data;
        }
-       
-       return 0;
+
+       return (0);
 }
 
 errno_t
@@ -580,13 +869,13 @@ mbuf_clear_csum_requested(
 {
        mbuf->m_pkthdr.csum_flags &= 0xffff0000;
        mbuf->m_pkthdr.csum_data = 0;
-       
-       return 0;
+
+       return (0);
 }
 
-static const mbuf_csum_performed_flags_t mbuf_valid_csum_performed_flags = 
+static const mbuf_csum_performed_flags_t mbuf_valid_csum_performed_flags =
        MBUF_CSUM_DID_IP | MBUF_CSUM_IP_GOOD | MBUF_CSUM_DID_DATA |
-       MBUF_CSUM_PSEUDO_HDR | MBUF_CSUM_TCP_SUM16;
+       MBUF_CSUM_PSEUDO_HDR | MBUF_CSUM_PARTIAL;
 
 errno_t
 mbuf_set_csum_performed(
@@ -595,10 +884,11 @@ mbuf_set_csum_performed(
        u_int32_t value)
 {
        performed &= mbuf_valid_csum_performed_flags;
-       mbuf->m_pkthdr.csum_flags = (mbuf->m_pkthdr.csum_flags & 0xffff0000) | performed;
+       mbuf->m_pkthdr.csum_flags =
+           (mbuf->m_pkthdr.csum_flags & 0xffff0000) | performed;
        mbuf->m_pkthdr.csum_data = value;
-       
-       return 0;
+
+       return (0);
 }
 
 errno_t
@@ -607,10 +897,11 @@ mbuf_get_csum_performed(
        mbuf_csum_performed_flags_t *performed,
        u_int32_t *value)
 {
-       *performed = mbuf->m_pkthdr.csum_flags & mbuf_valid_csum_performed_flags;
+       *performed =
+           mbuf->m_pkthdr.csum_flags & mbuf_valid_csum_performed_flags;
        *value = mbuf->m_pkthdr.csum_data;
-       
-       return 0;
+
+       return (0);
 }
 
 errno_t
@@ -619,119 +910,86 @@ mbuf_clear_csum_performed(
 {
        mbuf->m_pkthdr.csum_flags &= 0xffff0000;
        mbuf->m_pkthdr.csum_data = 0;
-       
-       return 0;
+
+       return (0);
+}
+
+errno_t
+mbuf_inet_cksum(mbuf_t mbuf, int protocol, u_int32_t offset, u_int32_t length,
+    u_int16_t *csum)
+{
+       if (mbuf == NULL || length == 0 || csum == NULL ||
+           (u_int32_t)mbuf->m_pkthdr.len < (offset + length))
+               return (EINVAL);
+
+       *csum = inet_cksum(mbuf, protocol, offset, length);
+       return (0);
+}
+
+#if INET6
+errno_t
+mbuf_inet6_cksum(mbuf_t mbuf, int protocol, u_int32_t offset, u_int32_t length,
+    u_int16_t *csum)
+{
+       if (mbuf == NULL || length == 0 || csum == NULL ||
+           (u_int32_t)mbuf->m_pkthdr.len < (offset + length))
+               return (EINVAL);
+
+       *csum = inet6_cksum(mbuf, protocol, offset, length);
+       return (0);
 }
+#else /* INET6 */
+errno_t
+mbuf_inet6_cksum(__unused mbuf_t mbuf, __unused int protocol,
+               __unused u_int32_t offset, __unused u_int32_t length,
+               __unused u_int16_t *csum)
+{
+       panic("mbuf_inet6_cksum() doesn't exist on this platform\n");
+       return (0);
+}
+
+u_int16_t
+inet6_cksum(__unused struct mbuf *m, __unused unsigned int nxt,
+               __unused unsigned int off, __unused unsigned int len)
+{
+       panic("inet6_cksum() doesn't exist on this platform\n");
+       return (0);
+}
+
+void nd6_lookup_ipv6(void);
+void
+nd6_lookup_ipv6(void)
+{
+       panic("nd6_lookup_ipv6() doesn't exist on this platform\n");
+}
+
+int
+in6addr_local(__unused struct in6_addr *a)
+{
+       panic("in6addr_local() doesn't exist on this platform\n");
+       return (0);
+}
+
+void nd6_storelladdr(void);
+void
+nd6_storelladdr(void)
+{
+       panic("nd6_storelladdr() doesn't exist on this platform\n");
+}
+#endif /* INET6 */
 
 /*
  * Mbuf tag KPIs
  */
 
-struct mbuf_tag_id_entry {
-       SLIST_ENTRY(mbuf_tag_id_entry)  next;
-       mbuf_tag_id_t                                   id;
-       char                                                    string[];
-};
-
-#define        MBUF_TAG_ID_ENTRY_SIZE(__str) \
-       ((size_t)&(((struct mbuf_tag_id_entry*)0)->string[0]) + \
-        strlen(__str) + 1)
-
-#define        MTAG_FIRST_ID                                   1000
-static u_long                                                  mtag_id_next = MTAG_FIRST_ID;
-static SLIST_HEAD(,mbuf_tag_id_entry)  mtag_id_list = {NULL};
-static lck_mtx_t                                               *mtag_id_lock = NULL;
-
-__private_extern__ void
-mbuf_tag_id_first_last(
-       u_long  *first,
-       u_long  *last)
-{
-       *first = MTAG_FIRST_ID;
-       *last = mtag_id_next - 1;
-}
-
-__private_extern__ errno_t
-mbuf_tag_id_find_internal(
-       const char      *string,
-       u_long          *out_id,
-       int                     create)
-{
-       struct mbuf_tag_id_entry                                *entry = NULL;
-       
-       
-       *out_id = 0;
-       
-       if (string == NULL || out_id == NULL) {
-               return EINVAL;
-       }
-       
-       /* Don't bother allocating the lock if we're only doing a lookup */
-       if (create == 0 && mtag_id_lock == NULL)
-               return ENOENT;
-       
-       /* Allocate lock if necessary */
-       if (mtag_id_lock == NULL) {
-               lck_grp_attr_t  *grp_attrib = NULL;
-               lck_attr_t              *lck_attrb = NULL;
-               lck_grp_t               *lck_group = NULL;
-               lck_mtx_t               *new_lock = NULL;
-               
-               grp_attrib = lck_grp_attr_alloc_init();
-               lck_group = lck_grp_alloc_init("mbuf_tag_allocate_id", grp_attrib);
-               lck_grp_attr_free(grp_attrib);
-               lck_attrb = lck_attr_alloc_init();
-
-               new_lock = lck_mtx_alloc_init(lck_group, lck_attrb);
-               if (!OSCompareAndSwap((UInt32)0, (UInt32)new_lock, (UInt32*)&mtag_id_lock)) {
-                       /*
-                        * If the atomic swap fails, someone else has already
-                        * done this work. We can free the stuff we allocated.
-                        */
-                       lck_mtx_free(new_lock, lck_group);
-                       lck_grp_free(lck_group);
-               }
-               lck_attr_free(lck_attrb);
-       }
-       
-       /* Look for an existing entry */
-       lck_mtx_lock(mtag_id_lock);
-       SLIST_FOREACH(entry, &mtag_id_list, next) {
-               if (strcmp(string, entry->string) == 0) {
-                       break;
-               }
-       }
-       
-       if (entry == NULL) {
-               if (create == 0) {
-                       lck_mtx_unlock(mtag_id_lock);
-                       return ENOENT;
-               }
-               
-               entry = kalloc(MBUF_TAG_ID_ENTRY_SIZE(string));
-               if (entry == NULL) {
-                       lck_mtx_unlock(mtag_id_lock);
-                       return ENOMEM;
-               }
-               
-               strcpy(entry->string, string);
-               entry->id = mtag_id_next;
-               mtag_id_next++;
-               SLIST_INSERT_HEAD(&mtag_id_list, entry, next);
-       }
-       lck_mtx_unlock(mtag_id_lock);
-       
-       *out_id = entry->id;
-       
-       return 0;
-}
+#define        MTAG_FIRST_ID FIRST_KPI_STR_ID
 
 errno_t
 mbuf_tag_id_find(
        const char              *string,
        mbuf_tag_id_t   *out_id)
 {
-       return mbuf_tag_id_find_internal(string, (u_long*)out_id, 1);
+       return (net_str_id_find_internal(string, out_id, NSI_MBUF_TAG, 1));
 }
 
 errno_t
@@ -744,68 +1002,75 @@ mbuf_tag_allocate(
        void**                  data_p)
 {
        struct m_tag *tag;
-       
+       u_int32_t mtag_id_first, mtag_id_last;
+
        if (data_p != NULL)
                *data_p = NULL;
-       
+
        /* Sanity check parameters */
-       if (mbuf == NULL || (mbuf->m_flags & M_PKTHDR) == 0 || id < MTAG_FIRST_ID ||
-               id >= mtag_id_next || length < 1 || (length & 0xffff0000) != 0 ||
-               data_p == NULL) {
-               return EINVAL;
+       (void) net_str_id_first_last(&mtag_id_first, &mtag_id_last,
+           NSI_MBUF_TAG);
+       if (mbuf == NULL || (mbuf->m_flags & M_PKTHDR) == 0 ||
+           id < mtag_id_first || id > mtag_id_last || length < 1 ||
+           (length & 0xffff0000) != 0 || data_p == NULL) {
+               return (EINVAL);
        }
-       
+
        /* Make sure this mtag hasn't already been allocated */
        tag = m_tag_locate(mbuf, id, type, NULL);
        if (tag != NULL) {
-               return EEXIST;
+               return (EEXIST);
        }
-       
+
        /* Allocate an mtag */
-       tag = m_tag_alloc(id, type, length, how);
+       tag = m_tag_create(id, type, length, how, mbuf);
        if (tag == NULL) {
-               return how == M_WAITOK ? ENOMEM : EWOULDBLOCK;
+               return (how == M_WAITOK ? ENOMEM : EWOULDBLOCK);
        }
-       
+
        /* Attach the mtag and set *data_p */
        m_tag_prepend(mbuf, tag);
        *data_p = tag + 1;
-       
-       return 0;
+
+       return (0);
 }
 
 errno_t
 mbuf_tag_find(
-       mbuf_t                  mbuf,
-       mbuf_tag_id_t   id,
-       mbuf_tag_type_t type,
-       size_t*                 length,
-       void**                  data_p)
+       mbuf_t mbuf,
+       mbuf_tag_id_t id,
+       mbuf_tag_type_t type,
+       size_t *length,
+       void **data_p)
 {
        struct m_tag *tag;
-       
+       u_int32_t mtag_id_first, mtag_id_last;
+
        if (length != NULL)
                *length = 0;
        if (data_p != NULL)
                *data_p = NULL;
-       
+
        /* Sanity check parameters */
-       if (mbuf == NULL || (mbuf->m_flags & M_PKTHDR) == 0 || id < MTAG_FIRST_ID ||
-               id >= mtag_id_next || length == NULL || data_p == NULL) {
-               return EINVAL;
+       (void) net_str_id_first_last(&mtag_id_first, &mtag_id_last,
+           NSI_MBUF_TAG);
+       if (mbuf == NULL || (mbuf->m_flags & M_PKTHDR) == 0 ||
+           id < mtag_id_first || id > mtag_id_last || length == NULL ||
+           data_p == NULL) {
+               return (EINVAL);
        }
-       
+
        /* Locate an mtag */
        tag = m_tag_locate(mbuf, id, type, NULL);
        if (tag == NULL) {
-               return ENOENT;
+               return (ENOENT);
        }
-       
+
        /* Copy out the pointer to the data and the lenght value */
        *length = tag->m_tag_len;
        *data_p = tag + 1;
-       
-       return 0;
+
+       return (0);
 }
 
 void
@@ -815,36 +1080,135 @@ mbuf_tag_free(
        mbuf_tag_type_t type)
 {
        struct m_tag *tag;
-       
-       if (mbuf == NULL || (mbuf->m_flags & M_PKTHDR) == 0 || id < MTAG_FIRST_ID ||
-               id >= mtag_id_next)
+       u_int32_t mtag_id_first, mtag_id_last;
+
+       /* Sanity check parameters */
+       (void) net_str_id_first_last(&mtag_id_first, &mtag_id_last,
+           NSI_MBUF_TAG);
+       if (mbuf == NULL || (mbuf->m_flags & M_PKTHDR) == 0 ||
+           id < mtag_id_first || id > mtag_id_last)
                return;
-       
+
        tag = m_tag_locate(mbuf, id, type, NULL);
        if (tag == NULL) {
                return;
        }
-       
+
        m_tag_delete(mbuf, tag);
-       return;
 }
 
-/* mbuf stats */
-void mbuf_stats(struct mbuf_stat *stats)
+/*
+ * Maximum length of driver auxiliary data; keep this small to
+ * fit in a single mbuf to avoid wasting memory, rounded down to
+ * the nearest 64-bit boundary.  This takes into account mbuf
+ * tag-related (m_taghdr + m_tag) as well m_drvaux_tag structs.
+ */
+#define        MBUF_DRVAUX_MAXLEN                                              \
+       P2ROUNDDOWN(MLEN - sizeof (struct m_taghdr) -                   \
+       M_TAG_ALIGN(sizeof (struct m_drvaux_tag)), sizeof (uint64_t))
+
+errno_t
+mbuf_add_drvaux(mbuf_t mbuf, mbuf_how_t how, u_int32_t family,
+    u_int32_t subfamily, size_t length, void **data_p)
 {
-       stats->mbufs = mbstat.m_mbufs;
-       stats->clusters = mbstat.m_clusters;
-       stats->clfree = mbstat.m_clfree;
-       stats->drops = mbstat.m_drops;
-       stats->wait = mbstat.m_wait;
-       stats->drain = mbstat.m_drain;
-       __builtin_memcpy(stats->mtypes, mbstat.m_mtypes, sizeof(stats->mtypes));
-       stats->mcfail = mbstat.m_mcfail;
-       stats->mpfail = mbstat.m_mpfail;
-       stats->msize = mbstat.m_msize;
-       stats->mclbytes = mbstat.m_mclbytes;
-       stats->minclsize = mbstat.m_minclsize;
-       stats->mlen = mbstat.m_mlen;
+       struct m_drvaux_tag *p;
+       struct m_tag *tag;
+
+       if (mbuf == NULL || !(mbuf->m_flags & M_PKTHDR) ||
+           length == 0 || length > MBUF_DRVAUX_MAXLEN)
+               return (EINVAL);
+
+       if (data_p != NULL)
+               *data_p = NULL;
+
+       /* Check if one is already associated */
+       if ((tag = m_tag_locate(mbuf, KERNEL_MODULE_TAG_ID,
+           KERNEL_TAG_TYPE_DRVAUX, NULL)) != NULL)
+               return (EEXIST);
+
+       /* Tag is (m_drvaux_tag + module specific data) */
+       if ((tag = m_tag_create(KERNEL_MODULE_TAG_ID, KERNEL_TAG_TYPE_DRVAUX,
+           sizeof (*p) + length, how, mbuf)) == NULL)
+               return ((how == MBUF_WAITOK) ? ENOMEM : EWOULDBLOCK);
+
+       p = (struct m_drvaux_tag *)(tag + 1);
+       p->da_family = family;
+       p->da_subfamily = subfamily;
+       p->da_length = length;
+
+       /* Associate the tag */
+       m_tag_prepend(mbuf, tag);
+
+       if (data_p != NULL)
+               *data_p = (p + 1);
+
+       return (0);
+}
+
+errno_t
+mbuf_find_drvaux(mbuf_t mbuf, u_int32_t *family_p, u_int32_t *subfamily_p,
+    u_int32_t *length_p, void **data_p)
+{
+       struct m_drvaux_tag *p;
+       struct m_tag *tag;
+
+       if (mbuf == NULL || !(mbuf->m_flags & M_PKTHDR) || data_p == NULL)
+               return (EINVAL);
+
+       *data_p = NULL;
+
+       if ((tag = m_tag_locate(mbuf, KERNEL_MODULE_TAG_ID,
+           KERNEL_TAG_TYPE_DRVAUX, NULL)) == NULL)
+               return (ENOENT);
+
+       /* Must be at least size of m_drvaux_tag */
+       VERIFY(tag->m_tag_len >= sizeof (*p));
+
+       p = (struct m_drvaux_tag *)(tag + 1);
+       VERIFY(p->da_length > 0 && p->da_length <= MBUF_DRVAUX_MAXLEN);
+
+       if (family_p != NULL)
+               *family_p = p->da_family;
+       if (subfamily_p != NULL)
+               *subfamily_p = p->da_subfamily;
+       if (length_p != NULL)
+               *length_p = p->da_length;
+
+       *data_p = (p + 1);
+
+       return (0);
+}
+
+void
+mbuf_del_drvaux(mbuf_t mbuf)
+{
+       struct m_tag *tag;
+
+       if (mbuf == NULL || !(mbuf->m_flags & M_PKTHDR))
+               return;
+
+       if ((tag = m_tag_locate(mbuf, KERNEL_MODULE_TAG_ID,
+           KERNEL_TAG_TYPE_DRVAUX, NULL)) != NULL)
+               m_tag_delete(mbuf, tag);
+}
+
+/* mbuf stats */
+void
+mbuf_stats(struct mbuf_stat *stats)
+{
+       stats->mbufs = mbstat.m_mbufs;
+       stats->clusters = mbstat.m_clusters;
+       stats->clfree = mbstat.m_clfree;
+       stats->drops = mbstat.m_drops;
+       stats->wait = mbstat.m_wait;
+       stats->drain = mbstat.m_drain;
+       __builtin_memcpy(stats->mtypes, mbstat.m_mtypes, sizeof(stats->mtypes));
+       stats->mcfail = mbstat.m_mcfail;
+       stats->mpfail = mbstat.m_mpfail;
+       stats->msize = mbstat.m_msize;
+       stats->mclbytes = mbstat.m_mclbytes;
+       stats->minclsize = mbstat.m_minclsize;
+       stats->mlen = mbstat.m_mlen;
        stats->mhlen = mbstat.m_mhlen;
        stats->bigclusters = mbstat.m_bigclusters;
        stats->bigclfree = mbstat.m_bigclfree;
@@ -852,7 +1216,8 @@ void mbuf_stats(struct mbuf_stat *stats)
 }
 
 errno_t
-mbuf_allocpacket(mbuf_how_t how, size_t packetlen, unsigned int *maxchunks, mbuf_t *mbuf)
+mbuf_allocpacket(mbuf_how_t how, size_t packetlen, unsigned int *maxchunks,
+    mbuf_t *mbuf)
 {
        errno_t error;
        struct mbuf *m;
@@ -863,20 +1228,81 @@ mbuf_allocpacket(mbuf_how_t how, size_t packetlen, unsigned int *maxchunks, mbuf
                error = EINVAL;
                goto out;
        }
-       m = m_allocpacket_internal(&numpkts, packetlen, maxchunks ? &numchunks : NULL, how, 1, 0);
+       m = m_allocpacket_internal(&numpkts, packetlen,
+           maxchunks ? &numchunks : NULL, how, 1, 0);
+       if (m == 0) {
+               if (maxchunks && *maxchunks && numchunks > *maxchunks)
+                       error = ENOBUFS;
+               else
+                       error = ENOMEM;
+       } else {
+               if (maxchunks)
+                       *maxchunks = numchunks;
+               error = 0;
+               *mbuf = m;
+       }
+out:
+       return (error);
+}
+
+errno_t
+mbuf_allocpacket_list(unsigned int numpkts, mbuf_how_t how, size_t packetlen,
+    unsigned int *maxchunks, mbuf_t *mbuf)
+{
+       errno_t error;
+       struct mbuf *m;
+       unsigned int numchunks = maxchunks ? *maxchunks : 0;
+
+       if (numpkts == 0) {
+               error = EINVAL;
+               goto out;
+       }
+       if (packetlen == 0) {
+               error = EINVAL;
+               goto out;
+       }
+       m = m_allocpacket_internal(&numpkts, packetlen,
+           maxchunks ? &numchunks : NULL, how, 1, 0);
        if (m == 0) {
                if (maxchunks && *maxchunks && numchunks > *maxchunks)
                        error = ENOBUFS;
                else
                        error = ENOMEM;
        } else {
+               if (maxchunks)
+                       *maxchunks = numchunks;
                error = 0;
                *mbuf = m;
        }
 out:
-       return error;
+       return (error);
+}
+
+__private_extern__ size_t
+mbuf_pkt_list_len(mbuf_t m)
+{
+       size_t len = 0;
+       mbuf_t n = m;
+
+       while (n) {
+               len += mbuf_pkthdr_len(n);
+               n = mbuf_nextpkt(n);
+       }
+       return (len);
 }
 
+__private_extern__ size_t
+mbuf_pkt_list_maxlen(mbuf_t m)
+{
+       size_t maxlen = 0;
+       mbuf_t n = m;
+
+       while (n) {
+               maxlen += mbuf_pkthdr_maxlen(n);
+               n = mbuf_nextpkt(n);
+       }
+       return (maxlen);
+}
 
 /*
  * mbuf_copyback differs from m_copyback in a few ways:
@@ -901,8 +1327,8 @@ mbuf_copyback(
        const char      *cp = data;
 
        if (m == NULL || len == 0 || data == NULL)
-               return EINVAL;
-       
+               return (EINVAL);
+
        while (off > (mlen = m->m_len)) {
                off -= mlen;
                totlen += mlen;
@@ -917,15 +1343,16 @@ mbuf_copyback(
                }
                m = m->m_next;
        }
-       
+
        while (len > 0) {
                mlen = MIN(m->m_len - off, len);
-               if (mlen < len && m->m_next == NULL && mbuf_trailingspace(m) > 0) {
+               if (mlen < len && m->m_next == NULL &&
+                   mbuf_trailingspace(m) > 0) {
                        size_t  grow = MIN(mbuf_trailingspace(m), len - mlen);
                        mlen += grow;
                        m->m_len += grow;
                }
-               bcopy(cp, off + (char*)mbuf_data(m), (unsigned)mlen);
+               bcopy(cp, off + (char *)mbuf_data(m), (unsigned)mlen);
                cp += mlen;
                len -= mlen;
                mlen += off;
@@ -940,7 +1367,10 @@ mbuf_copyback(
                                goto out;
                        }
                        if (len > MINCLSIZE) {
-                               /* cluter allocation failure is okay, we can grow chain */
+                               /*
+                                * cluster allocation failure is okay,
+                                * we can grow chain
+                                */
                                mbuf_mclget(how, m->m_type, &n);
                        }
                        n->m_len = MIN(mbuf_maxlen(n), len);
@@ -948,10 +1378,510 @@ mbuf_copyback(
                }
                m = m->m_next;
        }
-       
+
 out:
        if ((m_start->m_flags & M_PKTHDR) && (m_start->m_pkthdr.len < totlen))
                m_start->m_pkthdr.len = totlen;
-       
-       return result;
+
+       return (result);
+}
+
+u_int32_t
+mbuf_get_mlen(void)
+{
+       return (_MLEN);
+}
+
+u_int32_t
+mbuf_get_mhlen(void)
+{
+       return (_MHLEN);
+}
+
+u_int32_t
+mbuf_get_minclsize(void)
+{
+       return (MHLEN + MLEN);
+}
+
+u_int32_t
+mbuf_get_traffic_class_max_count(void)
+{
+       return (MBUF_TC_MAX);
+}
+
+errno_t
+mbuf_get_traffic_class_index(mbuf_traffic_class_t tc, u_int32_t *index)
+{
+       if (index == NULL || (u_int32_t)tc >= MBUF_TC_MAX)
+               return (EINVAL);
+
+       *index = MBUF_SCIDX(m_service_class_from_val(MBUF_TC2SCVAL(tc)));
+       return (0);
+}
+
+mbuf_traffic_class_t
+mbuf_get_traffic_class(mbuf_t m)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR))
+               return (MBUF_TC_BE);
+
+       return (m_get_traffic_class(m));
+}
+
+errno_t
+mbuf_set_traffic_class(mbuf_t m, mbuf_traffic_class_t tc)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR) ||
+           ((u_int32_t)tc >= MBUF_TC_MAX))
+               return (EINVAL);
+
+       return (m_set_traffic_class(m, tc));
+}
+
+int
+mbuf_is_traffic_class_privileged(mbuf_t m)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR) ||
+           !MBUF_VALID_SC(m->m_pkthdr.pkt_svc))
+               return (0);
+
+       return ((m->m_pkthdr.pkt_flags & PKTF_PRIO_PRIVILEGED) ? 1 : 0);
+}
+
+u_int32_t
+mbuf_get_service_class_max_count(void)
+{
+       return (MBUF_SC_MAX_CLASSES);
+}
+
+errno_t
+mbuf_get_service_class_index(mbuf_svc_class_t sc, u_int32_t *index)
+{
+       if (index == NULL || !MBUF_VALID_SC(sc))
+               return (EINVAL);
+
+       *index = MBUF_SCIDX(sc);
+       return (0);
+}
+
+mbuf_svc_class_t
+mbuf_get_service_class(mbuf_t m)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR))
+               return (MBUF_SC_BE);
+
+       return (m_get_service_class(m));
+}
+
+errno_t
+mbuf_set_service_class(mbuf_t m, mbuf_svc_class_t sc)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR))
+               return (EINVAL);
+
+       return (m_set_service_class(m, sc));
+}
+
+errno_t
+mbuf_pkthdr_aux_flags(mbuf_t m, mbuf_pkthdr_aux_flags_t *flagsp)
+{
+       u_int32_t flags;
+
+       if (m == NULL || !(m->m_flags & M_PKTHDR) || flagsp == NULL)
+               return (EINVAL);
+
+       *flagsp = 0;
+       flags = m->m_pkthdr.pkt_flags;
+       if ((flags & (PKTF_INET_RESOLVE|PKTF_RESOLVE_RTR)) ==
+           (PKTF_INET_RESOLVE|PKTF_RESOLVE_RTR))
+               *flagsp |= MBUF_PKTAUXF_INET_RESOLVE_RTR;
+       if ((flags & (PKTF_INET6_RESOLVE|PKTF_RESOLVE_RTR)) ==
+           (PKTF_INET6_RESOLVE|PKTF_RESOLVE_RTR))
+               *flagsp |= MBUF_PKTAUXF_INET6_RESOLVE_RTR;
+
+       /* These 2 flags are mutually exclusive */
+       VERIFY((*flagsp &
+           (MBUF_PKTAUXF_INET_RESOLVE_RTR | MBUF_PKTAUXF_INET6_RESOLVE_RTR)) !=
+           (MBUF_PKTAUXF_INET_RESOLVE_RTR | MBUF_PKTAUXF_INET6_RESOLVE_RTR));
+
+       return (0);
+}
+
+errno_t
+mbuf_get_driver_scratch(mbuf_t m, u_int8_t **area, size_t *area_len)
+{
+       if (m == NULL || area == NULL || area_len == NULL ||
+           !(m->m_flags & M_PKTHDR))
+               return (EINVAL);
+
+       *area_len = m_scratch_get(m, area);
+       return (0);
+}
+
+errno_t
+mbuf_get_unsent_data_bytes(const mbuf_t m, u_int32_t *unsent_data)
+{
+       if (m == NULL || unsent_data == NULL || !(m->m_flags & M_PKTHDR))
+               return (EINVAL);
+
+       if (!(m->m_pkthdr.pkt_flags & PKTF_VALID_UNSENT_DATA))
+               return (EINVAL);
+
+       *unsent_data = m->m_pkthdr.bufstatus_if +
+           m->m_pkthdr.bufstatus_sndbuf;
+       return (0);
+}
+
+errno_t
+mbuf_get_buffer_status(const mbuf_t m, mbuf_buffer_status_t *buf_status)
+{
+       if (m == NULL || buf_status == NULL || !(m->m_flags & M_PKTHDR) ||
+           !(m->m_pkthdr.pkt_flags & PKTF_VALID_UNSENT_DATA))
+               return (EINVAL);
+
+       buf_status->buf_interface = m->m_pkthdr.bufstatus_if;
+       buf_status->buf_sndbuf = m->m_pkthdr.bufstatus_sndbuf;
+       return (0);
+}
+
+errno_t
+mbuf_pkt_new_flow(const mbuf_t m, u_int32_t *retval)
+{
+       if (m == NULL || retval == NULL || !(m->m_flags & M_PKTHDR))
+               return (EINVAL);
+       if (m->m_pkthdr.pkt_flags & PKTF_NEW_FLOW)
+               *retval = 1;
+       else
+               *retval = 0;
+       return (0);
+}
+
+errno_t
+mbuf_last_pkt(const mbuf_t m, u_int32_t *retval)
+{
+       if (m == NULL || retval == NULL || !(m->m_flags & M_PKTHDR))
+               return (EINVAL);
+       if (m->m_pkthdr.pkt_flags & PKTF_LAST_PKT)
+               *retval = 1;
+       else
+               *retval = 0;
+       return (0);
+}
+
+errno_t
+mbuf_get_timestamp(mbuf_t m, u_int64_t *ts, boolean_t *valid)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR) || ts == NULL)
+               return (EINVAL);
+
+       if ((m->m_pkthdr.pkt_flags & PKTF_TS_VALID) == 0) {
+               if (valid != NULL)
+                       *valid = FALSE;
+               *ts = 0;
+       } else {
+               if (valid != NULL)
+                       *valid = TRUE;
+               *ts = m->m_pkthdr.pkt_timestamp;
+       }
+       return (0);
+}
+
+errno_t
+mbuf_set_timestamp(mbuf_t m, u_int64_t ts, boolean_t valid)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR))
+               return (EINVAL);
+
+       if (valid == FALSE) {
+               m->m_pkthdr.pkt_flags &= ~PKTF_TS_VALID;
+               m->m_pkthdr.pkt_timestamp = 0;
+       } else {
+               m->m_pkthdr.pkt_flags |= PKTF_TS_VALID;
+               m->m_pkthdr.pkt_timestamp = ts;
+       }
+       return (0);
+}
+
+errno_t
+mbuf_get_status(mbuf_t m, kern_return_t *status)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR) || status == NULL)
+               return (EINVAL);
+
+       if ((m->m_pkthdr.pkt_flags & PKTF_DRIVER_MTAG) == 0) {
+               *status = 0;
+       } else {
+               *status = m->m_pkthdr.drv_tx_status;
+       }
+       return (0);
+}
+
+static void
+driver_mtag_init(mbuf_t m)
+{
+       if ((m->m_pkthdr.pkt_flags & PKTF_DRIVER_MTAG) == 0) {
+               m->m_pkthdr.pkt_flags |= PKTF_DRIVER_MTAG;
+               bzero(&m->m_pkthdr.driver_mtag,
+                   sizeof(m->m_pkthdr.driver_mtag));
+       }
+}
+
+errno_t
+mbuf_set_status(mbuf_t m, kern_return_t status)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR))
+               return (EINVAL);
+
+       driver_mtag_init(m);
+
+       m->m_pkthdr.drv_tx_status = status;
+
+       return (0);
+}
+
+errno_t
+mbuf_get_flowid(mbuf_t m, u_int16_t *flowid)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR) || flowid == NULL)
+               return (EINVAL);
+
+       if ((m->m_pkthdr.pkt_flags & PKTF_DRIVER_MTAG) == 0) {
+               *flowid = 0;
+       } else {
+               *flowid = m->m_pkthdr.drv_flowid;
+       }
+       return (0);
+}
+
+errno_t
+mbuf_set_flowid(mbuf_t m, u_int16_t flowid)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR))
+               return (EINVAL);
+
+       driver_mtag_init(m);
+
+       m->m_pkthdr.drv_flowid = flowid;
+
+       return (0);
+}
+
+errno_t
+mbuf_get_tx_compl_data(mbuf_t m, uintptr_t *arg, uintptr_t *data)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR) || arg == NULL ||
+           data == NULL)
+               return (EINVAL);
+
+       if ((m->m_pkthdr.pkt_flags & PKTF_DRIVER_MTAG) == 0) {
+               *arg = 0;
+               *data = 0;
+       } else {
+               *arg = m->m_pkthdr.drv_tx_compl_arg;
+               *data = m->m_pkthdr.drv_tx_compl_data;
+       }
+       return (0);
+}
+
+errno_t
+mbuf_set_tx_compl_data(mbuf_t m, uintptr_t arg, uintptr_t data)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR))
+               return (EINVAL);
+
+       driver_mtag_init(m);
+
+       m->m_pkthdr.drv_tx_compl_arg = arg;
+       m->m_pkthdr.drv_tx_compl_data = data;
+
+       return (0);
+}
+
+static u_int32_t
+get_tx_compl_callback_index_locked(mbuf_tx_compl_func callback)
+{
+       u_int32_t i;
+
+       for (i = 0; i < MAX_MBUF_TX_COMPL_FUNC; i++) {
+               if (mbuf_tx_compl_table[i] == callback) {
+                       return (i);
+               }
+       }
+       return (UINT32_MAX);
+}
+
+static u_int32_t
+get_tx_compl_callback_index(mbuf_tx_compl_func callback)
+{
+       u_int32_t i;
+
+       lck_rw_lock_shared(mbuf_tx_compl_tbl_lock);
+
+       i = get_tx_compl_callback_index_locked(callback);
+
+       lck_rw_unlock_shared(mbuf_tx_compl_tbl_lock);
+
+       return (i);
+}
+
+errno_t
+mbuf_register_tx_compl_callback(mbuf_tx_compl_func callback)
+{
+       int i;
+       errno_t error;
+
+       if (callback == NULL)
+               return (EINVAL);
+
+       lck_rw_lock_exclusive(mbuf_tx_compl_tbl_lock);
+
+       i = get_tx_compl_callback_index_locked(callback);
+       if (i != -1) {
+               error = EEXIST;
+               goto unlock;
+       }
+
+       /* assume the worst */
+       error = ENOSPC;
+       for (i = 0; i < MAX_MBUF_TX_COMPL_FUNC; i++) {
+               if (mbuf_tx_compl_table[i] == NULL) {
+                       mbuf_tx_compl_table[i] = callback;
+                       error = 0;
+                       goto unlock;
+               }
+       }
+unlock:
+       lck_rw_unlock_exclusive(mbuf_tx_compl_tbl_lock);
+
+       return (error);
+}
+
+errno_t
+mbuf_unregister_tx_compl_callback(mbuf_tx_compl_func callback)
+{
+       int i;
+       errno_t error;
+
+       if (callback == NULL)
+               return (EINVAL);
+
+       lck_rw_lock_exclusive(mbuf_tx_compl_tbl_lock);
+
+       /* assume the worst */
+       error = ENOENT;
+       for (i = 0; i < MAX_MBUF_TX_COMPL_FUNC; i++) {
+               if (mbuf_tx_compl_table[i] == callback) {
+                       mbuf_tx_compl_table[i] = NULL;
+                       error = 0;
+                       goto unlock;
+               }
+       }
+unlock:
+       lck_rw_unlock_exclusive(mbuf_tx_compl_tbl_lock);
+
+       return (error);
+}
+
+errno_t
+mbuf_get_timestamp_requested(mbuf_t m, boolean_t *requested)
+{
+       if (m == NULL || !(m->m_flags & M_PKTHDR))
+               return (EINVAL);
+
+       if ((m->m_pkthdr.pkt_flags & PKTF_TX_COMPL_TS_REQ) == 0) {
+               *requested = FALSE;
+       } else {
+               *requested = TRUE;
+       }
+       return (0);
+}
+
+errno_t
+mbuf_set_timestamp_requested(mbuf_t m, uintptr_t *pktid,
+    mbuf_tx_compl_func callback)
+{
+       size_t i;
+
+       if (m == NULL || !(m->m_flags & M_PKTHDR) || callback == NULL ||
+           pktid == NULL)
+               return (EINVAL);
+
+       i = get_tx_compl_callback_index(callback);
+       if (i == UINT32_MAX)
+               return (ENOENT);
+
+#if (DEBUG || DEVELOPMENT)
+       VERIFY(i < sizeof(m->m_pkthdr.pkt_compl_callbacks));
+#endif /* (DEBUG || DEVELOPMENT) */
+
+       if ((m->m_pkthdr.pkt_flags & PKTF_TX_COMPL_TS_REQ) == 0) {
+               m->m_pkthdr.pkt_compl_callbacks = 0;
+               m->m_pkthdr.pkt_flags |= PKTF_TX_COMPL_TS_REQ;
+               m->m_pkthdr.pkt_compl_context =
+                   atomic_add_32_ov(&mbuf_tx_compl_index, 1);
+
+#if (DEBUG || DEVELOPMENT)
+               if (mbuf_tx_compl_debug != 0) {
+                       OSIncrementAtomic64(&mbuf_tx_compl_outstanding);
+               }
+#endif /* (DEBUG || DEVELOPMENT) */
+       }
+       m->m_pkthdr.pkt_compl_callbacks |= (1 << i);
+       *pktid = m->m_pkthdr.pkt_compl_context;
+
+       return (0);
+}
+
+void
+m_do_tx_compl_callback(struct mbuf *m, struct ifnet *ifp)
+{
+       int i;
+
+       if (m == NULL)
+               return;
+
+       if ((m->m_pkthdr.pkt_flags & PKTF_TX_COMPL_TS_REQ) == 0)
+               return;
+
+#if (DEBUG || DEVELOPMENT)
+       if (mbuf_tx_compl_debug != 0 && ifp != NULL &&
+           (ifp->if_xflags & IFXF_TIMESTAMP_ENABLED) != 0 &&
+           (m->m_pkthdr.pkt_flags & PKTF_TS_VALID) == 0) {
+               struct timespec now;
+
+               nanouptime(&now);
+               net_timernsec(&now, &m->m_pkthdr.pkt_timestamp);
+       }
+#endif /* (DEBUG || DEVELOPMENT) */
+
+       for (i = 0; i < MAX_MBUF_TX_COMPL_FUNC; i++) {
+               mbuf_tx_compl_func callback;
+
+               if ((m->m_pkthdr.pkt_compl_callbacks & (1 << i)) == 0)
+                       continue;
+
+               lck_rw_lock_shared(mbuf_tx_compl_tbl_lock);
+               callback = mbuf_tx_compl_table[i];
+               lck_rw_unlock_shared(mbuf_tx_compl_tbl_lock);
+
+               if (callback != NULL) {
+                       callback(m->m_pkthdr.pkt_compl_context,
+                           ifp,
+                           (m->m_pkthdr.pkt_flags & PKTF_TS_VALID) ?
+                           m->m_pkthdr.pkt_timestamp: 0,
+                           m->m_pkthdr.drv_tx_compl_arg,
+                           m->m_pkthdr.drv_tx_compl_data,
+                           m->m_pkthdr.drv_tx_status);
+               }
+       }
+       m->m_pkthdr.pkt_compl_callbacks = 0;
+
+#if (DEBUG || DEVELOPMENT)
+       if (mbuf_tx_compl_debug != 0) {
+               OSDecrementAtomic64(&mbuf_tx_compl_outstanding);
+               if (ifp == NULL)
+                       atomic_add_64(&mbuf_tx_compl_aborted, 1);
+       }
+#endif /* (DEBUG || DEVELOPMENT) */
 }