19454b2d8SWarner Losh /*- 2df8bae1dSRodney W. Grimes * Copyright (c) 1982, 1986, 1988, 1991, 1993 3df8bae1dSRodney W. Grimes * The Regents of the University of California. All rights reserved. 4df8bae1dSRodney W. Grimes * 5df8bae1dSRodney W. Grimes * Redistribution and use in source and binary forms, with or without 6df8bae1dSRodney W. Grimes * modification, are permitted provided that the following conditions 7df8bae1dSRodney W. Grimes * are met: 8df8bae1dSRodney W. Grimes * 1. Redistributions of source code must retain the above copyright 9df8bae1dSRodney W. Grimes * notice, this list of conditions and the following disclaimer. 10df8bae1dSRodney W. Grimes * 2. Redistributions in binary form must reproduce the above copyright 11df8bae1dSRodney W. Grimes * notice, this list of conditions and the following disclaimer in the 12df8bae1dSRodney W. Grimes * documentation and/or other materials provided with the distribution. 13df8bae1dSRodney W. Grimes * 4. Neither the name of the University nor the names of its contributors 14df8bae1dSRodney W. Grimes * may be used to endorse or promote products derived from this software 15df8bae1dSRodney W. Grimes * without specific prior written permission. 16df8bae1dSRodney W. Grimes * 17df8bae1dSRodney W. Grimes * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 18df8bae1dSRodney W. Grimes * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19df8bae1dSRodney W. Grimes * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20df8bae1dSRodney W. Grimes * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 21df8bae1dSRodney W. Grimes * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22df8bae1dSRodney W. Grimes * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23df8bae1dSRodney W. Grimes * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24df8bae1dSRodney W. Grimes * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25df8bae1dSRodney W. Grimes * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26df8bae1dSRodney W. Grimes * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27df8bae1dSRodney W. Grimes * SUCH DAMAGE. 28df8bae1dSRodney W. Grimes * 29df8bae1dSRodney W. Grimes * @(#)uipc_mbuf.c 8.2 (Berkeley) 1/4/94 30df8bae1dSRodney W. Grimes */ 31df8bae1dSRodney W. Grimes 32677b542eSDavid E. O'Brien #include <sys/cdefs.h> 33677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$"); 34677b542eSDavid E. O'Brien 35240ef842SDavid E. O'Brien #include "opt_param.h" 36352d050eSMike Silbersack #include "opt_mbuf_stress_test.h" 376eeac1d9SJulian Elischer #include "opt_mbuf_profiling.h" 38e32a5b94SRobert Watson 39df8bae1dSRodney W. Grimes #include <sys/param.h> 40df8bae1dSRodney W. Grimes #include <sys/systm.h> 41fb919e4dSMark Murray #include <sys/kernel.h> 42beb699c7SMike Silbersack #include <sys/limits.h> 43fb919e4dSMark Murray #include <sys/lock.h> 44f9d0d524SRobert Watson #include <sys/malloc.h> 45df8bae1dSRodney W. Grimes #include <sys/mbuf.h> 46639acc13SGarrett Wollman #include <sys/sysctl.h> 47df8bae1dSRodney W. Grimes #include <sys/domain.h> 48df8bae1dSRodney W. Grimes #include <sys/protosw.h> 49beb699c7SMike Silbersack #include <sys/uio.h> 50fb919e4dSMark Murray 515e4bc63bSGleb Smirnoff #include <security/mac/mac_framework.h> 525e4bc63bSGleb Smirnoff 5328f8db14SBruce Evans int max_linkhdr; 5428f8db14SBruce Evans int max_protohdr; 5528f8db14SBruce Evans int max_hdr; 5628f8db14SBruce Evans int max_datalen; 5751710a45SMike Silbersack #ifdef MBUF_STRESS_TEST 5855e9f80dSMike Silbersack int m_defragpackets; 5955e9f80dSMike Silbersack int m_defragbytes; 6055e9f80dSMike Silbersack int m_defraguseless; 6155e9f80dSMike Silbersack int m_defragfailure; 62352d050eSMike Silbersack int m_defragrandomfailures; 63352d050eSMike Silbersack #endif 647d032714SBosko Milekic 657d032714SBosko Milekic /* 667d032714SBosko Milekic * sysctl(8) exported objects 677d032714SBosko Milekic */ 6880444f88SAndre Oppermann SYSCTL_INT(_kern_ipc, KIPC_MAX_LINKHDR, max_linkhdr, CTLFLAG_RD, 6980444f88SAndre Oppermann &max_linkhdr, 0, "Size of largest link layer header"); 7080444f88SAndre Oppermann SYSCTL_INT(_kern_ipc, KIPC_MAX_PROTOHDR, max_protohdr, CTLFLAG_RD, 7180444f88SAndre Oppermann &max_protohdr, 0, "Size of largest protocol layer header"); 7280444f88SAndre Oppermann SYSCTL_INT(_kern_ipc, KIPC_MAX_HDR, max_hdr, CTLFLAG_RD, 7380444f88SAndre Oppermann &max_hdr, 0, "Size of largest link plus protocol header"); 7480444f88SAndre Oppermann SYSCTL_INT(_kern_ipc, KIPC_MAX_DATALEN, max_datalen, CTLFLAG_RD, 7580444f88SAndre Oppermann &max_datalen, 0, "Minimum space left in mbuf after max_hdr"); 7651710a45SMike Silbersack #ifdef MBUF_STRESS_TEST 7755e9f80dSMike Silbersack SYSCTL_INT(_kern_ipc, OID_AUTO, m_defragpackets, CTLFLAG_RD, 7855e9f80dSMike Silbersack &m_defragpackets, 0, ""); 7955e9f80dSMike Silbersack SYSCTL_INT(_kern_ipc, OID_AUTO, m_defragbytes, CTLFLAG_RD, 8055e9f80dSMike Silbersack &m_defragbytes, 0, ""); 8155e9f80dSMike Silbersack SYSCTL_INT(_kern_ipc, OID_AUTO, m_defraguseless, CTLFLAG_RD, 8255e9f80dSMike Silbersack &m_defraguseless, 0, ""); 8355e9f80dSMike Silbersack SYSCTL_INT(_kern_ipc, OID_AUTO, m_defragfailure, CTLFLAG_RD, 8455e9f80dSMike Silbersack &m_defragfailure, 0, ""); 85352d050eSMike Silbersack SYSCTL_INT(_kern_ipc, OID_AUTO, m_defragrandomfailures, CTLFLAG_RW, 86352d050eSMike Silbersack &m_defragrandomfailures, 0, ""); 87352d050eSMike Silbersack #endif 88df8bae1dSRodney W. Grimes 89df8bae1dSRodney W. Grimes /* 90f729ede6SAndre Oppermann * Ensure the correct size of various mbuf parameters. It could be off due 91f729ede6SAndre Oppermann * to compiler-induced padding and alignment artifacts. 92f729ede6SAndre Oppermann */ 93f729ede6SAndre Oppermann CTASSERT(MSIZE - offsetof(struct mbuf, m_dat) == MLEN); 94f729ede6SAndre Oppermann CTASSERT(MSIZE - offsetof(struct mbuf, m_pktdat) == MHLEN); 95f729ede6SAndre Oppermann 96f729ede6SAndre Oppermann /* 973d1a9ed3SRobert Watson * mbuf data storage should be 64-bit aligned regardless of architectural 983d1a9ed3SRobert Watson * pointer size; check this is the case with and without a packet header. 993d1a9ed3SRobert Watson */ 1003d1a9ed3SRobert Watson CTASSERT(offsetof(struct mbuf, m_dat) % 8 == 0); 1013d1a9ed3SRobert Watson CTASSERT(offsetof(struct mbuf, m_pktdat) % 8 == 0); 1023d1a9ed3SRobert Watson 1033d1a9ed3SRobert Watson /* 1043d1a9ed3SRobert Watson * While the specific values here don't matter too much (i.e., +/- a few 1053d1a9ed3SRobert Watson * words), we do want to ensure that changes to these values are carefully 1063d1a9ed3SRobert Watson * reasoned about and properly documented. This is especially the case as 1073d1a9ed3SRobert Watson * network-protocol and device-driver modules encode these layouts, and must 1083d1a9ed3SRobert Watson * be recompiled if the structures change. Check these values at compile time 1093d1a9ed3SRobert Watson * against the ones documented in comments in mbuf.h. 1103d1a9ed3SRobert Watson * 1113d1a9ed3SRobert Watson * NB: Possibly they should be documented there via #define's and not just 1123d1a9ed3SRobert Watson * comments. 1133d1a9ed3SRobert Watson */ 1143d1a9ed3SRobert Watson #if defined(__LP64__) 1153d1a9ed3SRobert Watson CTASSERT(offsetof(struct mbuf, m_dat) == 32); 1163d1a9ed3SRobert Watson CTASSERT(sizeof(struct pkthdr) == 56); 117ee52391eSGleb Smirnoff CTASSERT(sizeof(struct m_ext) == 48); 1183d1a9ed3SRobert Watson #else 1193d1a9ed3SRobert Watson CTASSERT(offsetof(struct mbuf, m_dat) == 24); 1203d1a9ed3SRobert Watson CTASSERT(sizeof(struct pkthdr) == 48); 121ee52391eSGleb Smirnoff CTASSERT(sizeof(struct m_ext) == 28); 1223d1a9ed3SRobert Watson #endif 1233d1a9ed3SRobert Watson 1243d1a9ed3SRobert Watson /* 125ec9d83ddSGleb Smirnoff * Assert that the queue(3) macros produce code of the same size as an old 126ec9d83ddSGleb Smirnoff * plain pointer does. 127ec9d83ddSGleb Smirnoff */ 128ec9d83ddSGleb Smirnoff #ifdef INVARIANTS 129ec9d83ddSGleb Smirnoff static struct mbuf m_assertbuf; 130ec9d83ddSGleb Smirnoff CTASSERT(sizeof(m_assertbuf.m_slist) == sizeof(m_assertbuf.m_next)); 131ec9d83ddSGleb Smirnoff CTASSERT(sizeof(m_assertbuf.m_stailq) == sizeof(m_assertbuf.m_next)); 132ec9d83ddSGleb Smirnoff CTASSERT(sizeof(m_assertbuf.m_slistpkt) == sizeof(m_assertbuf.m_nextpkt)); 133ec9d83ddSGleb Smirnoff CTASSERT(sizeof(m_assertbuf.m_stailqpkt) == sizeof(m_assertbuf.m_nextpkt)); 134ec9d83ddSGleb Smirnoff #endif 135ec9d83ddSGleb Smirnoff 136ec9d83ddSGleb Smirnoff /* 1376bccea7cSRebecca Cran * Attach the cluster from *m to *n, set up m_ext in *n 13856a4e45aSAndre Oppermann * and bump the refcount of the cluster. 13956a4e45aSAndre Oppermann */ 14086a996e6SHiren Panchasara void 141*56a5f52eSGleb Smirnoff mb_dupcl(struct mbuf *n, struct mbuf *m) 14256a4e45aSAndre Oppermann { 143*56a5f52eSGleb Smirnoff volatile u_int *refcnt; 14456a4e45aSAndre Oppermann 1451fbe6a82SGleb Smirnoff KASSERT(m->m_flags & M_EXT, ("%s: M_EXT not set on %p", __func__, m)); 1461fbe6a82SGleb Smirnoff KASSERT(!(n->m_flags & M_EXT), ("%s: M_EXT set on %p", __func__, n)); 1471fbe6a82SGleb Smirnoff 148c71b4037SGleb Smirnoff n->m_ext = m->m_ext; 14956a4e45aSAndre Oppermann n->m_flags |= M_EXT; 15032a8b1d8SColin Percival n->m_flags |= m->m_flags & M_RDONLY; 151*56a5f52eSGleb Smirnoff 152*56a5f52eSGleb Smirnoff /* See if this is the mbuf that holds the embedded refcount. */ 153*56a5f52eSGleb Smirnoff if (m->m_ext.ext_flags & EXT_FLAG_EMBREF) { 154*56a5f52eSGleb Smirnoff refcnt = n->m_ext.ext_cnt = &m->m_ext.ext_count; 155*56a5f52eSGleb Smirnoff n->m_ext.ext_flags &= ~EXT_FLAG_EMBREF; 156*56a5f52eSGleb Smirnoff } else { 157*56a5f52eSGleb Smirnoff KASSERT(m->m_ext.ext_cnt != NULL, 158*56a5f52eSGleb Smirnoff ("%s: no refcounting pointer on %p", __func__, m)); 159*56a5f52eSGleb Smirnoff refcnt = m->m_ext.ext_cnt; 160*56a5f52eSGleb Smirnoff } 161*56a5f52eSGleb Smirnoff 162*56a5f52eSGleb Smirnoff if (*refcnt == 1) 163*56a5f52eSGleb Smirnoff *refcnt += 1; 164*56a5f52eSGleb Smirnoff else 165*56a5f52eSGleb Smirnoff atomic_add_int(refcnt, 1); 16656a4e45aSAndre Oppermann } 16756a4e45aSAndre Oppermann 1689523d1bfSNavdeep Parhar void 1699523d1bfSNavdeep Parhar m_demote_pkthdr(struct mbuf *m) 1709523d1bfSNavdeep Parhar { 1719523d1bfSNavdeep Parhar 1729523d1bfSNavdeep Parhar M_ASSERTPKTHDR(m); 1739523d1bfSNavdeep Parhar 1749523d1bfSNavdeep Parhar m_tag_delete_chain(m, NULL); 1759523d1bfSNavdeep Parhar m->m_flags &= ~M_PKTHDR; 1769523d1bfSNavdeep Parhar bzero(&m->m_pkthdr, sizeof(struct pkthdr)); 1779523d1bfSNavdeep Parhar } 1789523d1bfSNavdeep Parhar 17956a4e45aSAndre Oppermann /* 180ed111688SAndre Oppermann * Clean up mbuf (chain) from any tags and packet headers. 181e0068c3aSAndre Oppermann * If "all" is set then the first mbuf in the chain will be 182e0068c3aSAndre Oppermann * cleaned too. 183ed111688SAndre Oppermann */ 184ed111688SAndre Oppermann void 185651e4e6aSGleb Smirnoff m_demote(struct mbuf *m0, int all, int flags) 186ed111688SAndre Oppermann { 187ed111688SAndre Oppermann struct mbuf *m; 188ed111688SAndre Oppermann 189ed111688SAndre Oppermann for (m = all ? m0 : m0->m_next; m != NULL; m = m->m_next) { 1907ee2d058SGleb Smirnoff KASSERT(m->m_nextpkt == NULL, ("%s: m_nextpkt in m %p, m0 %p", 1917ee2d058SGleb Smirnoff __func__, m, m0)); 1929523d1bfSNavdeep Parhar if (m->m_flags & M_PKTHDR) 1939523d1bfSNavdeep Parhar m_demote_pkthdr(m); 194651e4e6aSGleb Smirnoff m->m_flags = m->m_flags & (M_EXT | M_RDONLY | M_NOFREE | flags); 195ed111688SAndre Oppermann } 196ed111688SAndre Oppermann } 197ed111688SAndre Oppermann 198ed111688SAndre Oppermann /* 199fdcc028dSAndre Oppermann * Sanity checks on mbuf (chain) for use in KASSERT() and general 200fdcc028dSAndre Oppermann * debugging. 201fdcc028dSAndre Oppermann * Returns 0 or panics when bad and 1 on all tests passed. 202fdcc028dSAndre Oppermann * Sanitize, 0 to run M_SANITY_ACTION, 1 to garble things so they 203fdcc028dSAndre Oppermann * blow up later. 204a048affbSAndre Oppermann */ 205a048affbSAndre Oppermann int 206a048affbSAndre Oppermann m_sanity(struct mbuf *m0, int sanitize) 207a048affbSAndre Oppermann { 208a048affbSAndre Oppermann struct mbuf *m; 209a048affbSAndre Oppermann caddr_t a, b; 210a048affbSAndre Oppermann int pktlen = 0; 211a048affbSAndre Oppermann 21221ee3e7aSKip Macy #ifdef INVARIANTS 21321ee3e7aSKip Macy #define M_SANITY_ACTION(s) panic("mbuf %p: " s, m) 21421ee3e7aSKip Macy #else 21521ee3e7aSKip Macy #define M_SANITY_ACTION(s) printf("mbuf %p: " s, m) 21621ee3e7aSKip Macy #endif 217a048affbSAndre Oppermann 218fdcc028dSAndre Oppermann for (m = m0; m != NULL; m = m->m_next) { 219a048affbSAndre Oppermann /* 220a048affbSAndre Oppermann * Basic pointer checks. If any of these fails then some 221a048affbSAndre Oppermann * unrelated kernel memory before or after us is trashed. 222a048affbSAndre Oppermann * No way to recover from that. 223a048affbSAndre Oppermann */ 224b66f2a48SRobert Watson a = M_START(m); 225b66f2a48SRobert Watson b = a + M_SIZE(m); 226a048affbSAndre Oppermann if ((caddr_t)m->m_data < a) 227a048affbSAndre Oppermann M_SANITY_ACTION("m_data outside mbuf data range left"); 228a048affbSAndre Oppermann if ((caddr_t)m->m_data > b) 229a048affbSAndre Oppermann M_SANITY_ACTION("m_data outside mbuf data range right"); 230a048affbSAndre Oppermann if ((caddr_t)m->m_data + m->m_len > b) 231a048affbSAndre Oppermann M_SANITY_ACTION("m_data + m_len exeeds mbuf space"); 232a048affbSAndre Oppermann 233a048affbSAndre Oppermann /* m->m_nextpkt may only be set on first mbuf in chain. */ 234fdcc028dSAndre Oppermann if (m != m0 && m->m_nextpkt != NULL) { 235a048affbSAndre Oppermann if (sanitize) { 236a048affbSAndre Oppermann m_freem(m->m_nextpkt); 237a048affbSAndre Oppermann m->m_nextpkt = (struct mbuf *)0xDEADC0DE; 238a048affbSAndre Oppermann } else 239a048affbSAndre Oppermann M_SANITY_ACTION("m->m_nextpkt on in-chain mbuf"); 240a048affbSAndre Oppermann } 241a048affbSAndre Oppermann 242a048affbSAndre Oppermann /* packet length (not mbuf length!) calculation */ 243a048affbSAndre Oppermann if (m0->m_flags & M_PKTHDR) 244a048affbSAndre Oppermann pktlen += m->m_len; 245a048affbSAndre Oppermann 246a048affbSAndre Oppermann /* m_tags may only be attached to first mbuf in chain. */ 247a048affbSAndre Oppermann if (m != m0 && m->m_flags & M_PKTHDR && 248a048affbSAndre Oppermann !SLIST_EMPTY(&m->m_pkthdr.tags)) { 249a048affbSAndre Oppermann if (sanitize) { 250a048affbSAndre Oppermann m_tag_delete_chain(m, NULL); 251a048affbSAndre Oppermann /* put in 0xDEADC0DE perhaps? */ 252fdcc028dSAndre Oppermann } else 253a048affbSAndre Oppermann M_SANITY_ACTION("m_tags on in-chain mbuf"); 254a048affbSAndre Oppermann } 255a048affbSAndre Oppermann 256a048affbSAndre Oppermann /* M_PKTHDR may only be set on first mbuf in chain */ 257a048affbSAndre Oppermann if (m != m0 && m->m_flags & M_PKTHDR) { 258a048affbSAndre Oppermann if (sanitize) { 259a048affbSAndre Oppermann bzero(&m->m_pkthdr, sizeof(m->m_pkthdr)); 260a048affbSAndre Oppermann m->m_flags &= ~M_PKTHDR; 261a048affbSAndre Oppermann /* put in 0xDEADCODE and leave hdr flag in */ 262a048affbSAndre Oppermann } else 263a048affbSAndre Oppermann M_SANITY_ACTION("M_PKTHDR on in-chain mbuf"); 264a048affbSAndre Oppermann } 265a048affbSAndre Oppermann } 266fdcc028dSAndre Oppermann m = m0; 267fdcc028dSAndre Oppermann if (pktlen && pktlen != m->m_pkthdr.len) { 268a048affbSAndre Oppermann if (sanitize) 269fdcc028dSAndre Oppermann m->m_pkthdr.len = 0; 270a048affbSAndre Oppermann else 271a048affbSAndre Oppermann M_SANITY_ACTION("m_pkthdr.len != mbuf chain length"); 272a048affbSAndre Oppermann } 273a048affbSAndre Oppermann return 1; 274fdcc028dSAndre Oppermann 275fdcc028dSAndre Oppermann #undef M_SANITY_ACTION 276a048affbSAndre Oppermann } 277a048affbSAndre Oppermann 2785e4bc63bSGleb Smirnoff /* 2795e4bc63bSGleb Smirnoff * Non-inlined part of m_init(). 2805e4bc63bSGleb Smirnoff */ 2815e4bc63bSGleb Smirnoff int 2825e4bc63bSGleb Smirnoff m_pkthdr_init(struct mbuf *m, int how) 2835e4bc63bSGleb Smirnoff { 2845e4bc63bSGleb Smirnoff #ifdef MAC 2855e4bc63bSGleb Smirnoff int error; 2865e4bc63bSGleb Smirnoff #endif 2875e4bc63bSGleb Smirnoff m->m_data = m->m_pktdat; 2885e4bc63bSGleb Smirnoff bzero(&m->m_pkthdr, sizeof(m->m_pkthdr)); 2895e4bc63bSGleb Smirnoff #ifdef MAC 2905e4bc63bSGleb Smirnoff /* If the label init fails, fail the alloc */ 2915e4bc63bSGleb Smirnoff error = mac_mbuf_init(m, how); 2925e4bc63bSGleb Smirnoff if (error) 2935e4bc63bSGleb Smirnoff return (error); 2945e4bc63bSGleb Smirnoff #endif 2955e4bc63bSGleb Smirnoff 2965e4bc63bSGleb Smirnoff return (0); 2975e4bc63bSGleb Smirnoff } 298a048affbSAndre Oppermann 299a048affbSAndre Oppermann /* 3009967cafcSSam Leffler * "Move" mbuf pkthdr from "from" to "to". 301e37b1fcdSRobert Watson * "from" must have M_PKTHDR set, and "to" must be empty. 302e37b1fcdSRobert Watson */ 303e37b1fcdSRobert Watson void 3049967cafcSSam Leffler m_move_pkthdr(struct mbuf *to, struct mbuf *from) 305e37b1fcdSRobert Watson { 306e37b1fcdSRobert Watson 307e37b1fcdSRobert Watson #if 0 3089967cafcSSam Leffler /* see below for why these are not enabled */ 309fe584538SDag-Erling Smørgrav M_ASSERTPKTHDR(to); 310225bff6fSRobert Watson /* Note: with MAC, this may not be a good assertion. */ 3119967cafcSSam Leffler KASSERT(SLIST_EMPTY(&to->m_pkthdr.tags), 3129967cafcSSam Leffler ("m_move_pkthdr: to has tags")); 313e37b1fcdSRobert Watson #endif 314e32a5b94SRobert Watson #ifdef MAC 315225bff6fSRobert Watson /* 316225bff6fSRobert Watson * XXXMAC: It could be this should also occur for non-MAC? 317225bff6fSRobert Watson */ 318e32a5b94SRobert Watson if (to->m_flags & M_PKTHDR) 319225bff6fSRobert Watson m_tag_delete_chain(to, NULL); 320e32a5b94SRobert Watson #endif 321a4e71429SSam Leffler to->m_flags = (from->m_flags & M_COPYFLAGS) | (to->m_flags & M_EXT); 322a4e71429SSam Leffler if ((to->m_flags & M_EXT) == 0) 3239967cafcSSam Leffler to->m_data = to->m_pktdat; 3249967cafcSSam Leffler to->m_pkthdr = from->m_pkthdr; /* especially tags */ 3259967cafcSSam Leffler SLIST_INIT(&from->m_pkthdr.tags); /* purge tags from src */ 3269967cafcSSam Leffler from->m_flags &= ~M_PKTHDR; 3279967cafcSSam Leffler } 3289967cafcSSam Leffler 3299967cafcSSam Leffler /* 3309967cafcSSam Leffler * Duplicate "from"'s mbuf pkthdr in "to". 3319967cafcSSam Leffler * "from" must have M_PKTHDR set, and "to" must be empty. 3329967cafcSSam Leffler * In particular, this does a deep copy of the packet tags. 3339967cafcSSam Leffler */ 3349967cafcSSam Leffler int 3350cbefd30SAlexander V. Chernikov m_dup_pkthdr(struct mbuf *to, const struct mbuf *from, int how) 3369967cafcSSam Leffler { 3379967cafcSSam Leffler 3389967cafcSSam Leffler #if 0 3399967cafcSSam Leffler /* 3409967cafcSSam Leffler * The mbuf allocator only initializes the pkthdr 341c95be8b5SGleb Smirnoff * when the mbuf is allocated with m_gethdr(). Many users 342c95be8b5SGleb Smirnoff * (e.g. m_copy*, m_prepend) use m_get() and then 3439967cafcSSam Leffler * smash the pkthdr as needed causing these 3449967cafcSSam Leffler * assertions to trip. For now just disable them. 3459967cafcSSam Leffler */ 346fe584538SDag-Erling Smørgrav M_ASSERTPKTHDR(to); 347225bff6fSRobert Watson /* Note: with MAC, this may not be a good assertion. */ 3489967cafcSSam Leffler KASSERT(SLIST_EMPTY(&to->m_pkthdr.tags), ("m_dup_pkthdr: to has tags")); 3499967cafcSSam Leffler #endif 350063d8114SAlfred Perlstein MBUF_CHECKSLEEP(how); 3519967cafcSSam Leffler #ifdef MAC 3529967cafcSSam Leffler if (to->m_flags & M_PKTHDR) 353225bff6fSRobert Watson m_tag_delete_chain(to, NULL); 3549967cafcSSam Leffler #endif 355df8c7fc9SMike Silbersack to->m_flags = (from->m_flags & M_COPYFLAGS) | (to->m_flags & M_EXT); 356df8c7fc9SMike Silbersack if ((to->m_flags & M_EXT) == 0) 3579967cafcSSam Leffler to->m_data = to->m_pktdat; 358e37b1fcdSRobert Watson to->m_pkthdr = from->m_pkthdr; 3599967cafcSSam Leffler SLIST_INIT(&to->m_pkthdr.tags); 36094985f74SGleb Smirnoff return (m_tag_copy_chain(to, from, how)); 361e37b1fcdSRobert Watson } 362e37b1fcdSRobert Watson 363e37b1fcdSRobert Watson /* 364df8bae1dSRodney W. Grimes * Lesser-used path for M_PREPEND: 365df8bae1dSRodney W. Grimes * allocate new mbuf to prepend to chain, 366df8bae1dSRodney W. Grimes * copy junk along. 367df8bae1dSRodney W. Grimes */ 368df8bae1dSRodney W. Grimes struct mbuf * 369122a814aSBosko Milekic m_prepend(struct mbuf *m, int len, int how) 370df8bae1dSRodney W. Grimes { 371df8bae1dSRodney W. Grimes struct mbuf *mn; 372df8bae1dSRodney W. Grimes 373f8bf8e39SMike Silbersack if (m->m_flags & M_PKTHDR) 374c95be8b5SGleb Smirnoff mn = m_gethdr(how, m->m_type); 375f8bf8e39SMike Silbersack else 376c95be8b5SGleb Smirnoff mn = m_get(how, m->m_type); 377122a814aSBosko Milekic if (mn == NULL) { 378df8bae1dSRodney W. Grimes m_freem(m); 379122a814aSBosko Milekic return (NULL); 380df8bae1dSRodney W. Grimes } 381225bff6fSRobert Watson if (m->m_flags & M_PKTHDR) 382c95be8b5SGleb Smirnoff m_move_pkthdr(mn, m); 383df8bae1dSRodney W. Grimes mn->m_next = m; 384df8bae1dSRodney W. Grimes m = mn; 385ed6a66caSRobert Watson if (len < M_SIZE(m)) 3865288989fSRandall Stewart M_ALIGN(m, len); 387df8bae1dSRodney W. Grimes m->m_len = len; 388df8bae1dSRodney W. Grimes return (m); 389df8bae1dSRodney W. Grimes } 390df8bae1dSRodney W. Grimes 391df8bae1dSRodney W. Grimes /* 392df8bae1dSRodney W. Grimes * Make a copy of an mbuf chain starting "off0" bytes from the beginning, 393df8bae1dSRodney W. Grimes * continuing for "len" bytes. If len is M_COPYALL, copy to end of mbuf. 394eb1b1807SGleb Smirnoff * The wait parameter is a choice of M_WAITOK/M_NOWAIT from caller. 3951c38f2eaSArchie Cobbs * Note that the copy is read-only, because clusters are not copied, 3961c38f2eaSArchie Cobbs * only their reference counts are incremented. 397df8bae1dSRodney W. Grimes */ 398df8bae1dSRodney W. Grimes struct mbuf * 399*56a5f52eSGleb Smirnoff m_copym(struct mbuf *m, int off0, int len, int wait) 400df8bae1dSRodney W. Grimes { 401122a814aSBosko Milekic struct mbuf *n, **np; 402122a814aSBosko Milekic int off = off0; 403df8bae1dSRodney W. Grimes struct mbuf *top; 404df8bae1dSRodney W. Grimes int copyhdr = 0; 405df8bae1dSRodney W. Grimes 406e0a653ddSAlfred Perlstein KASSERT(off >= 0, ("m_copym, negative off %d", off)); 407e0a653ddSAlfred Perlstein KASSERT(len >= 0, ("m_copym, negative len %d", len)); 408063d8114SAlfred Perlstein MBUF_CHECKSLEEP(wait); 409df8bae1dSRodney W. Grimes if (off == 0 && m->m_flags & M_PKTHDR) 410df8bae1dSRodney W. Grimes copyhdr = 1; 411df8bae1dSRodney W. Grimes while (off > 0) { 412e0a653ddSAlfred Perlstein KASSERT(m != NULL, ("m_copym, offset > size of mbuf chain")); 413df8bae1dSRodney W. Grimes if (off < m->m_len) 414df8bae1dSRodney W. Grimes break; 415df8bae1dSRodney W. Grimes off -= m->m_len; 416df8bae1dSRodney W. Grimes m = m->m_next; 417df8bae1dSRodney W. Grimes } 418df8bae1dSRodney W. Grimes np = ⊤ 419df8bae1dSRodney W. Grimes top = 0; 420df8bae1dSRodney W. Grimes while (len > 0) { 421122a814aSBosko Milekic if (m == NULL) { 422e0a653ddSAlfred Perlstein KASSERT(len == M_COPYALL, 423e0a653ddSAlfred Perlstein ("m_copym, length > size of mbuf chain")); 424df8bae1dSRodney W. Grimes break; 425df8bae1dSRodney W. Grimes } 426f8bf8e39SMike Silbersack if (copyhdr) 427c95be8b5SGleb Smirnoff n = m_gethdr(wait, m->m_type); 428f8bf8e39SMike Silbersack else 429c95be8b5SGleb Smirnoff n = m_get(wait, m->m_type); 430df8bae1dSRodney W. Grimes *np = n; 431122a814aSBosko Milekic if (n == NULL) 432df8bae1dSRodney W. Grimes goto nospace; 433df8bae1dSRodney W. Grimes if (copyhdr) { 4349967cafcSSam Leffler if (!m_dup_pkthdr(n, m, wait)) 4359967cafcSSam Leffler goto nospace; 436df8bae1dSRodney W. Grimes if (len == M_COPYALL) 437df8bae1dSRodney W. Grimes n->m_pkthdr.len -= off0; 438df8bae1dSRodney W. Grimes else 439df8bae1dSRodney W. Grimes n->m_pkthdr.len = len; 440df8bae1dSRodney W. Grimes copyhdr = 0; 441df8bae1dSRodney W. Grimes } 442df8bae1dSRodney W. Grimes n->m_len = min(len, m->m_len - off); 443df8bae1dSRodney W. Grimes if (m->m_flags & M_EXT) { 444df8bae1dSRodney W. Grimes n->m_data = m->m_data + off; 44556a4e45aSAndre Oppermann mb_dupcl(n, m); 446df8bae1dSRodney W. Grimes } else 447df8bae1dSRodney W. Grimes bcopy(mtod(m, caddr_t)+off, mtod(n, caddr_t), 448bd395ae8SBosko Milekic (u_int)n->m_len); 449df8bae1dSRodney W. Grimes if (len != M_COPYALL) 450df8bae1dSRodney W. Grimes len -= n->m_len; 451df8bae1dSRodney W. Grimes off = 0; 452df8bae1dSRodney W. Grimes m = m->m_next; 453df8bae1dSRodney W. Grimes np = &n->m_next; 454df8bae1dSRodney W. Grimes } 45508442f8aSBosko Milekic 456df8bae1dSRodney W. Grimes return (top); 457df8bae1dSRodney W. Grimes nospace: 458df8bae1dSRodney W. Grimes m_freem(top); 459122a814aSBosko Milekic return (NULL); 460df8bae1dSRodney W. Grimes } 461df8bae1dSRodney W. Grimes 462df8bae1dSRodney W. Grimes /* 4636a06dea0SGarrett Wollman * Copy an entire packet, including header (which must be present). 4646a06dea0SGarrett Wollman * An optimization of the common case `m_copym(m, 0, M_COPYALL, how)'. 4651c38f2eaSArchie Cobbs * Note that the copy is read-only, because clusters are not copied, 4661c38f2eaSArchie Cobbs * only their reference counts are incremented. 4675fe86675SLuigi Rizzo * Preserve alignment of the first mbuf so if the creator has left 4685fe86675SLuigi Rizzo * some room at the beginning (e.g. for inserting protocol headers) 4695fe86675SLuigi Rizzo * the copies still have the room available. 4706a06dea0SGarrett Wollman */ 4716a06dea0SGarrett Wollman struct mbuf * 472122a814aSBosko Milekic m_copypacket(struct mbuf *m, int how) 4736a06dea0SGarrett Wollman { 4746a06dea0SGarrett Wollman struct mbuf *top, *n, *o; 4756a06dea0SGarrett Wollman 476063d8114SAlfred Perlstein MBUF_CHECKSLEEP(how); 477c95be8b5SGleb Smirnoff n = m_get(how, m->m_type); 4786a06dea0SGarrett Wollman top = n; 479122a814aSBosko Milekic if (n == NULL) 4806a06dea0SGarrett Wollman goto nospace; 4816a06dea0SGarrett Wollman 4829967cafcSSam Leffler if (!m_dup_pkthdr(n, m, how)) 4839967cafcSSam Leffler goto nospace; 4846a06dea0SGarrett Wollman n->m_len = m->m_len; 4856a06dea0SGarrett Wollman if (m->m_flags & M_EXT) { 4866a06dea0SGarrett Wollman n->m_data = m->m_data; 48756a4e45aSAndre Oppermann mb_dupcl(n, m); 4886a06dea0SGarrett Wollman } else { 4895fe86675SLuigi Rizzo n->m_data = n->m_pktdat + (m->m_data - m->m_pktdat ); 4906a06dea0SGarrett Wollman bcopy(mtod(m, char *), mtod(n, char *), n->m_len); 4916a06dea0SGarrett Wollman } 4926a06dea0SGarrett Wollman 4936a06dea0SGarrett Wollman m = m->m_next; 4946a06dea0SGarrett Wollman while (m) { 495c95be8b5SGleb Smirnoff o = m_get(how, m->m_type); 496122a814aSBosko Milekic if (o == NULL) 4976a06dea0SGarrett Wollman goto nospace; 4986a06dea0SGarrett Wollman 4996a06dea0SGarrett Wollman n->m_next = o; 5006a06dea0SGarrett Wollman n = n->m_next; 5016a06dea0SGarrett Wollman 5026a06dea0SGarrett Wollman n->m_len = m->m_len; 5036a06dea0SGarrett Wollman if (m->m_flags & M_EXT) { 5046a06dea0SGarrett Wollman n->m_data = m->m_data; 50556a4e45aSAndre Oppermann mb_dupcl(n, m); 5066a06dea0SGarrett Wollman } else { 5076a06dea0SGarrett Wollman bcopy(mtod(m, char *), mtod(n, char *), n->m_len); 5086a06dea0SGarrett Wollman } 5096a06dea0SGarrett Wollman 5106a06dea0SGarrett Wollman m = m->m_next; 5116a06dea0SGarrett Wollman } 5126a06dea0SGarrett Wollman return top; 5136a06dea0SGarrett Wollman nospace: 5146a06dea0SGarrett Wollman m_freem(top); 515122a814aSBosko Milekic return (NULL); 5166a06dea0SGarrett Wollman } 5176a06dea0SGarrett Wollman 5186a06dea0SGarrett Wollman /* 519df8bae1dSRodney W. Grimes * Copy data from an mbuf chain starting "off" bytes from the beginning, 520df8bae1dSRodney W. Grimes * continuing for "len" bytes, into the indicated buffer. 521df8bae1dSRodney W. Grimes */ 52226f9a767SRodney W. Grimes void 523a8cfc0eeSJulian Elischer m_copydata(const struct mbuf *m, int off, int len, caddr_t cp) 524df8bae1dSRodney W. Grimes { 525bd395ae8SBosko Milekic u_int count; 526df8bae1dSRodney W. Grimes 527e0a653ddSAlfred Perlstein KASSERT(off >= 0, ("m_copydata, negative off %d", off)); 528e0a653ddSAlfred Perlstein KASSERT(len >= 0, ("m_copydata, negative len %d", len)); 529df8bae1dSRodney W. Grimes while (off > 0) { 530e0a653ddSAlfred Perlstein KASSERT(m != NULL, ("m_copydata, offset > size of mbuf chain")); 531df8bae1dSRodney W. Grimes if (off < m->m_len) 532df8bae1dSRodney W. Grimes break; 533df8bae1dSRodney W. Grimes off -= m->m_len; 534df8bae1dSRodney W. Grimes m = m->m_next; 535df8bae1dSRodney W. Grimes } 536df8bae1dSRodney W. Grimes while (len > 0) { 537e0a653ddSAlfred Perlstein KASSERT(m != NULL, ("m_copydata, length > size of mbuf chain")); 538df8bae1dSRodney W. Grimes count = min(m->m_len - off, len); 539df8bae1dSRodney W. Grimes bcopy(mtod(m, caddr_t) + off, cp, count); 540df8bae1dSRodney W. Grimes len -= count; 541df8bae1dSRodney W. Grimes cp += count; 542df8bae1dSRodney W. Grimes off = 0; 543df8bae1dSRodney W. Grimes m = m->m_next; 544df8bae1dSRodney W. Grimes } 545df8bae1dSRodney W. Grimes } 546df8bae1dSRodney W. Grimes 547df8bae1dSRodney W. Grimes /* 5481c38f2eaSArchie Cobbs * Copy a packet header mbuf chain into a completely new chain, including 5491c38f2eaSArchie Cobbs * copying any mbuf clusters. Use this instead of m_copypacket() when 5501c38f2eaSArchie Cobbs * you need a writable copy of an mbuf chain. 5511c38f2eaSArchie Cobbs */ 5521c38f2eaSArchie Cobbs struct mbuf * 5530cbefd30SAlexander V. Chernikov m_dup(const struct mbuf *m, int how) 5541c38f2eaSArchie Cobbs { 5551c38f2eaSArchie Cobbs struct mbuf **p, *top = NULL; 5561c38f2eaSArchie Cobbs int remain, moff, nsize; 5571c38f2eaSArchie Cobbs 558063d8114SAlfred Perlstein MBUF_CHECKSLEEP(how); 5591c38f2eaSArchie Cobbs /* Sanity check */ 5601c38f2eaSArchie Cobbs if (m == NULL) 561122a814aSBosko Milekic return (NULL); 562fe584538SDag-Erling Smørgrav M_ASSERTPKTHDR(m); 5631c38f2eaSArchie Cobbs 5641c38f2eaSArchie Cobbs /* While there's more data, get a new mbuf, tack it on, and fill it */ 5651c38f2eaSArchie Cobbs remain = m->m_pkthdr.len; 5661c38f2eaSArchie Cobbs moff = 0; 5671c38f2eaSArchie Cobbs p = ⊤ 5681c38f2eaSArchie Cobbs while (remain > 0 || top == NULL) { /* allow m->m_pkthdr.len == 0 */ 5691c38f2eaSArchie Cobbs struct mbuf *n; 5701c38f2eaSArchie Cobbs 5711c38f2eaSArchie Cobbs /* Get the next new mbuf */ 572099a0e58SBosko Milekic if (remain >= MINCLSIZE) { 573099a0e58SBosko Milekic n = m_getcl(how, m->m_type, 0); 574099a0e58SBosko Milekic nsize = MCLBYTES; 575099a0e58SBosko Milekic } else { 576099a0e58SBosko Milekic n = m_get(how, m->m_type); 577099a0e58SBosko Milekic nsize = MLEN; 578099a0e58SBosko Milekic } 5791c38f2eaSArchie Cobbs if (n == NULL) 5801c38f2eaSArchie Cobbs goto nospace; 581099a0e58SBosko Milekic 582099a0e58SBosko Milekic if (top == NULL) { /* First one, must be PKTHDR */ 583099a0e58SBosko Milekic if (!m_dup_pkthdr(n, m, how)) { 584099a0e58SBosko Milekic m_free(n); 5851c38f2eaSArchie Cobbs goto nospace; 5861c38f2eaSArchie Cobbs } 58763e6f390SEd Maste if ((n->m_flags & M_EXT) == 0) 588099a0e58SBosko Milekic nsize = MHLEN; 589089bb672SAndrey V. Elsukov n->m_flags &= ~M_RDONLY; 5901c38f2eaSArchie Cobbs } 5911c38f2eaSArchie Cobbs n->m_len = 0; 5921c38f2eaSArchie Cobbs 5931c38f2eaSArchie Cobbs /* Link it into the new chain */ 5941c38f2eaSArchie Cobbs *p = n; 5951c38f2eaSArchie Cobbs p = &n->m_next; 5961c38f2eaSArchie Cobbs 5971c38f2eaSArchie Cobbs /* Copy data from original mbuf(s) into new mbuf */ 5981c38f2eaSArchie Cobbs while (n->m_len < nsize && m != NULL) { 5991c38f2eaSArchie Cobbs int chunk = min(nsize - n->m_len, m->m_len - moff); 6001c38f2eaSArchie Cobbs 6011c38f2eaSArchie Cobbs bcopy(m->m_data + moff, n->m_data + n->m_len, chunk); 6021c38f2eaSArchie Cobbs moff += chunk; 6031c38f2eaSArchie Cobbs n->m_len += chunk; 6041c38f2eaSArchie Cobbs remain -= chunk; 6051c38f2eaSArchie Cobbs if (moff == m->m_len) { 6061c38f2eaSArchie Cobbs m = m->m_next; 6071c38f2eaSArchie Cobbs moff = 0; 6081c38f2eaSArchie Cobbs } 6091c38f2eaSArchie Cobbs } 6101c38f2eaSArchie Cobbs 6111c38f2eaSArchie Cobbs /* Check correct total mbuf length */ 6121c38f2eaSArchie Cobbs KASSERT((remain > 0 && m != NULL) || (remain == 0 && m == NULL), 613a48740b6SDavid E. O'Brien ("%s: bogus m_pkthdr.len", __func__)); 6141c38f2eaSArchie Cobbs } 6151c38f2eaSArchie Cobbs return (top); 6161c38f2eaSArchie Cobbs 6171c38f2eaSArchie Cobbs nospace: 6181c38f2eaSArchie Cobbs m_freem(top); 619122a814aSBosko Milekic return (NULL); 6201c38f2eaSArchie Cobbs } 6211c38f2eaSArchie Cobbs 6221c38f2eaSArchie Cobbs /* 623df8bae1dSRodney W. Grimes * Concatenate mbuf chain n to m. 624df8bae1dSRodney W. Grimes * Both chains must be of the same type (e.g. MT_DATA). 625df8bae1dSRodney W. Grimes * Any m_pkthdr is not updated. 626df8bae1dSRodney W. Grimes */ 62726f9a767SRodney W. Grimes void 628122a814aSBosko Milekic m_cat(struct mbuf *m, struct mbuf *n) 629df8bae1dSRodney W. Grimes { 630df8bae1dSRodney W. Grimes while (m->m_next) 631df8bae1dSRodney W. Grimes m = m->m_next; 632df8bae1dSRodney W. Grimes while (n) { 63314d7c5b1SAndre Oppermann if (!M_WRITABLE(m) || 63414d7c5b1SAndre Oppermann M_TRAILINGSPACE(m) < n->m_len) { 635df8bae1dSRodney W. Grimes /* just join the two chains */ 636df8bae1dSRodney W. Grimes m->m_next = n; 637df8bae1dSRodney W. Grimes return; 638df8bae1dSRodney W. Grimes } 639df8bae1dSRodney W. Grimes /* splat the data from one into the other */ 640df8bae1dSRodney W. Grimes bcopy(mtod(n, caddr_t), mtod(m, caddr_t) + m->m_len, 641df8bae1dSRodney W. Grimes (u_int)n->m_len); 642df8bae1dSRodney W. Grimes m->m_len += n->m_len; 643df8bae1dSRodney W. Grimes n = m_free(n); 644df8bae1dSRodney W. Grimes } 645df8bae1dSRodney W. Grimes } 646df8bae1dSRodney W. Grimes 6471967edbaSGleb Smirnoff /* 6481967edbaSGleb Smirnoff * Concatenate two pkthdr mbuf chains. 6491967edbaSGleb Smirnoff */ 6501967edbaSGleb Smirnoff void 6511967edbaSGleb Smirnoff m_catpkt(struct mbuf *m, struct mbuf *n) 6521967edbaSGleb Smirnoff { 6531967edbaSGleb Smirnoff 6541967edbaSGleb Smirnoff M_ASSERTPKTHDR(m); 6551967edbaSGleb Smirnoff M_ASSERTPKTHDR(n); 6561967edbaSGleb Smirnoff 6571967edbaSGleb Smirnoff m->m_pkthdr.len += n->m_pkthdr.len; 658651e4e6aSGleb Smirnoff m_demote(n, 1, 0); 6591967edbaSGleb Smirnoff 6601967edbaSGleb Smirnoff m_cat(m, n); 6611967edbaSGleb Smirnoff } 6621967edbaSGleb Smirnoff 66326f9a767SRodney W. Grimes void 664122a814aSBosko Milekic m_adj(struct mbuf *mp, int req_len) 665df8bae1dSRodney W. Grimes { 666122a814aSBosko Milekic int len = req_len; 667122a814aSBosko Milekic struct mbuf *m; 668122a814aSBosko Milekic int count; 669df8bae1dSRodney W. Grimes 670df8bae1dSRodney W. Grimes if ((m = mp) == NULL) 671df8bae1dSRodney W. Grimes return; 672df8bae1dSRodney W. Grimes if (len >= 0) { 673df8bae1dSRodney W. Grimes /* 674df8bae1dSRodney W. Grimes * Trim from head. 675df8bae1dSRodney W. Grimes */ 676df8bae1dSRodney W. Grimes while (m != NULL && len > 0) { 677df8bae1dSRodney W. Grimes if (m->m_len <= len) { 678df8bae1dSRodney W. Grimes len -= m->m_len; 679df8bae1dSRodney W. Grimes m->m_len = 0; 680df8bae1dSRodney W. Grimes m = m->m_next; 681df8bae1dSRodney W. Grimes } else { 682df8bae1dSRodney W. Grimes m->m_len -= len; 683df8bae1dSRodney W. Grimes m->m_data += len; 684df8bae1dSRodney W. Grimes len = 0; 685df8bae1dSRodney W. Grimes } 686df8bae1dSRodney W. Grimes } 687df8bae1dSRodney W. Grimes if (mp->m_flags & M_PKTHDR) 688a83baab6SMarko Zec mp->m_pkthdr.len -= (req_len - len); 689df8bae1dSRodney W. Grimes } else { 690df8bae1dSRodney W. Grimes /* 691df8bae1dSRodney W. Grimes * Trim from tail. Scan the mbuf chain, 692df8bae1dSRodney W. Grimes * calculating its length and finding the last mbuf. 693df8bae1dSRodney W. Grimes * If the adjustment only affects this mbuf, then just 694df8bae1dSRodney W. Grimes * adjust and return. Otherwise, rescan and truncate 695df8bae1dSRodney W. Grimes * after the remaining size. 696df8bae1dSRodney W. Grimes */ 697df8bae1dSRodney W. Grimes len = -len; 698df8bae1dSRodney W. Grimes count = 0; 699df8bae1dSRodney W. Grimes for (;;) { 700df8bae1dSRodney W. Grimes count += m->m_len; 701df8bae1dSRodney W. Grimes if (m->m_next == (struct mbuf *)0) 702df8bae1dSRodney W. Grimes break; 703df8bae1dSRodney W. Grimes m = m->m_next; 704df8bae1dSRodney W. Grimes } 705df8bae1dSRodney W. Grimes if (m->m_len >= len) { 706df8bae1dSRodney W. Grimes m->m_len -= len; 707df8bae1dSRodney W. Grimes if (mp->m_flags & M_PKTHDR) 708df8bae1dSRodney W. Grimes mp->m_pkthdr.len -= len; 709df8bae1dSRodney W. Grimes return; 710df8bae1dSRodney W. Grimes } 711df8bae1dSRodney W. Grimes count -= len; 712df8bae1dSRodney W. Grimes if (count < 0) 713df8bae1dSRodney W. Grimes count = 0; 714df8bae1dSRodney W. Grimes /* 715df8bae1dSRodney W. Grimes * Correct length for chain is "count". 716df8bae1dSRodney W. Grimes * Find the mbuf with last data, adjust its length, 717df8bae1dSRodney W. Grimes * and toss data from remaining mbufs on chain. 718df8bae1dSRodney W. Grimes */ 719df8bae1dSRodney W. Grimes m = mp; 720df8bae1dSRodney W. Grimes if (m->m_flags & M_PKTHDR) 721df8bae1dSRodney W. Grimes m->m_pkthdr.len = count; 722df8bae1dSRodney W. Grimes for (; m; m = m->m_next) { 723df8bae1dSRodney W. Grimes if (m->m_len >= count) { 724df8bae1dSRodney W. Grimes m->m_len = count; 72559d8b310SSam Leffler if (m->m_next != NULL) { 72659d8b310SSam Leffler m_freem(m->m_next); 72759d8b310SSam Leffler m->m_next = NULL; 72859d8b310SSam Leffler } 729df8bae1dSRodney W. Grimes break; 730df8bae1dSRodney W. Grimes } 731df8bae1dSRodney W. Grimes count -= m->m_len; 732df8bae1dSRodney W. Grimes } 733df8bae1dSRodney W. Grimes } 734df8bae1dSRodney W. Grimes } 735df8bae1dSRodney W. Grimes 736df8bae1dSRodney W. Grimes /* 737df8bae1dSRodney W. Grimes * Rearange an mbuf chain so that len bytes are contiguous 738a2c36a02SKevin Lo * and in the data area of an mbuf (so that mtod will work 739a2c36a02SKevin Lo * for a structure of size len). Returns the resulting 740df8bae1dSRodney W. Grimes * mbuf chain on success, frees it and returns null on failure. 741df8bae1dSRodney W. Grimes * If there is room, it will add up to max_protohdr-len extra bytes to the 742df8bae1dSRodney W. Grimes * contiguous region in an attempt to avoid being called next time. 743df8bae1dSRodney W. Grimes */ 744df8bae1dSRodney W. Grimes struct mbuf * 745122a814aSBosko Milekic m_pullup(struct mbuf *n, int len) 746df8bae1dSRodney W. Grimes { 747122a814aSBosko Milekic struct mbuf *m; 748122a814aSBosko Milekic int count; 749df8bae1dSRodney W. Grimes int space; 750df8bae1dSRodney W. Grimes 751df8bae1dSRodney W. Grimes /* 752df8bae1dSRodney W. Grimes * If first mbuf has no cluster, and has room for len bytes 753df8bae1dSRodney W. Grimes * without shifting current data, pullup into it, 754df8bae1dSRodney W. Grimes * otherwise allocate a new mbuf to prepend to the chain. 755df8bae1dSRodney W. Grimes */ 756df8bae1dSRodney W. Grimes if ((n->m_flags & M_EXT) == 0 && 757df8bae1dSRodney W. Grimes n->m_data + len < &n->m_dat[MLEN] && n->m_next) { 758df8bae1dSRodney W. Grimes if (n->m_len >= len) 759df8bae1dSRodney W. Grimes return (n); 760df8bae1dSRodney W. Grimes m = n; 761df8bae1dSRodney W. Grimes n = n->m_next; 762df8bae1dSRodney W. Grimes len -= m->m_len; 763df8bae1dSRodney W. Grimes } else { 764df8bae1dSRodney W. Grimes if (len > MHLEN) 765df8bae1dSRodney W. Grimes goto bad; 766c95be8b5SGleb Smirnoff m = m_get(M_NOWAIT, n->m_type); 767122a814aSBosko Milekic if (m == NULL) 768df8bae1dSRodney W. Grimes goto bad; 7699967cafcSSam Leffler if (n->m_flags & M_PKTHDR) 770c95be8b5SGleb Smirnoff m_move_pkthdr(m, n); 771df8bae1dSRodney W. Grimes } 772df8bae1dSRodney W. Grimes space = &m->m_dat[MLEN] - (m->m_data + m->m_len); 773df8bae1dSRodney W. Grimes do { 774df8bae1dSRodney W. Grimes count = min(min(max(len, max_protohdr), space), n->m_len); 775df8bae1dSRodney W. Grimes bcopy(mtod(n, caddr_t), mtod(m, caddr_t) + m->m_len, 776bd395ae8SBosko Milekic (u_int)count); 777df8bae1dSRodney W. Grimes len -= count; 778df8bae1dSRodney W. Grimes m->m_len += count; 779df8bae1dSRodney W. Grimes n->m_len -= count; 780df8bae1dSRodney W. Grimes space -= count; 781df8bae1dSRodney W. Grimes if (n->m_len) 782df8bae1dSRodney W. Grimes n->m_data += count; 783df8bae1dSRodney W. Grimes else 784df8bae1dSRodney W. Grimes n = m_free(n); 785df8bae1dSRodney W. Grimes } while (len > 0 && n); 786df8bae1dSRodney W. Grimes if (len > 0) { 787df8bae1dSRodney W. Grimes (void) m_free(m); 788df8bae1dSRodney W. Grimes goto bad; 789df8bae1dSRodney W. Grimes } 790df8bae1dSRodney W. Grimes m->m_next = n; 791df8bae1dSRodney W. Grimes return (m); 792df8bae1dSRodney W. Grimes bad: 793df8bae1dSRodney W. Grimes m_freem(n); 794122a814aSBosko Milekic return (NULL); 795df8bae1dSRodney W. Grimes } 796df8bae1dSRodney W. Grimes 797df8bae1dSRodney W. Grimes /* 7987ac139a9SJohn-Mark Gurney * Like m_pullup(), except a new mbuf is always allocated, and we allow 7997ac139a9SJohn-Mark Gurney * the amount of empty space before the data in the new mbuf to be specified 8007ac139a9SJohn-Mark Gurney * (in the event that the caller expects to prepend later). 8017ac139a9SJohn-Mark Gurney */ 8027ac139a9SJohn-Mark Gurney struct mbuf * 8037ac139a9SJohn-Mark Gurney m_copyup(struct mbuf *n, int len, int dstoff) 8047ac139a9SJohn-Mark Gurney { 8057ac139a9SJohn-Mark Gurney struct mbuf *m; 8067ac139a9SJohn-Mark Gurney int count, space; 8077ac139a9SJohn-Mark Gurney 8087ac139a9SJohn-Mark Gurney if (len > (MHLEN - dstoff)) 8097ac139a9SJohn-Mark Gurney goto bad; 810c95be8b5SGleb Smirnoff m = m_get(M_NOWAIT, n->m_type); 8117ac139a9SJohn-Mark Gurney if (m == NULL) 8127ac139a9SJohn-Mark Gurney goto bad; 8137ac139a9SJohn-Mark Gurney if (n->m_flags & M_PKTHDR) 814c95be8b5SGleb Smirnoff m_move_pkthdr(m, n); 8157ac139a9SJohn-Mark Gurney m->m_data += dstoff; 8167ac139a9SJohn-Mark Gurney space = &m->m_dat[MLEN] - (m->m_data + m->m_len); 8177ac139a9SJohn-Mark Gurney do { 8187ac139a9SJohn-Mark Gurney count = min(min(max(len, max_protohdr), space), n->m_len); 8197ac139a9SJohn-Mark Gurney memcpy(mtod(m, caddr_t) + m->m_len, mtod(n, caddr_t), 8207ac139a9SJohn-Mark Gurney (unsigned)count); 8217ac139a9SJohn-Mark Gurney len -= count; 8227ac139a9SJohn-Mark Gurney m->m_len += count; 8237ac139a9SJohn-Mark Gurney n->m_len -= count; 8247ac139a9SJohn-Mark Gurney space -= count; 8257ac139a9SJohn-Mark Gurney if (n->m_len) 8267ac139a9SJohn-Mark Gurney n->m_data += count; 8277ac139a9SJohn-Mark Gurney else 8287ac139a9SJohn-Mark Gurney n = m_free(n); 8297ac139a9SJohn-Mark Gurney } while (len > 0 && n); 8307ac139a9SJohn-Mark Gurney if (len > 0) { 8317ac139a9SJohn-Mark Gurney (void) m_free(m); 8327ac139a9SJohn-Mark Gurney goto bad; 8337ac139a9SJohn-Mark Gurney } 8347ac139a9SJohn-Mark Gurney m->m_next = n; 8357ac139a9SJohn-Mark Gurney return (m); 8367ac139a9SJohn-Mark Gurney bad: 8377ac139a9SJohn-Mark Gurney m_freem(n); 8387ac139a9SJohn-Mark Gurney return (NULL); 8397ac139a9SJohn-Mark Gurney } 8407ac139a9SJohn-Mark Gurney 8417ac139a9SJohn-Mark Gurney /* 842df8bae1dSRodney W. Grimes * Partition an mbuf chain in two pieces, returning the tail -- 843df8bae1dSRodney W. Grimes * all but the first len0 bytes. In case of failure, it returns NULL and 844df8bae1dSRodney W. Grimes * attempts to restore the chain to its original state. 84548d183faSArchie Cobbs * 84648d183faSArchie Cobbs * Note that the resulting mbufs might be read-only, because the new 84748d183faSArchie Cobbs * mbuf can end up sharing an mbuf cluster with the original mbuf if 84848d183faSArchie Cobbs * the "breaking point" happens to lie within a cluster mbuf. Use the 84948d183faSArchie Cobbs * M_WRITABLE() macro to check for this case. 850df8bae1dSRodney W. Grimes */ 851df8bae1dSRodney W. Grimes struct mbuf * 852122a814aSBosko Milekic m_split(struct mbuf *m0, int len0, int wait) 853df8bae1dSRodney W. Grimes { 854122a814aSBosko Milekic struct mbuf *m, *n; 855bd395ae8SBosko Milekic u_int len = len0, remain; 856df8bae1dSRodney W. Grimes 857063d8114SAlfred Perlstein MBUF_CHECKSLEEP(wait); 858df8bae1dSRodney W. Grimes for (m = m0; m && len > m->m_len; m = m->m_next) 859df8bae1dSRodney W. Grimes len -= m->m_len; 860122a814aSBosko Milekic if (m == NULL) 861122a814aSBosko Milekic return (NULL); 862df8bae1dSRodney W. Grimes remain = m->m_len - len; 86321f39848SGleb Smirnoff if (m0->m_flags & M_PKTHDR && remain == 0) { 86421f39848SGleb Smirnoff n = m_gethdr(wait, m0->m_type); 86577badb18SGleb Smirnoff if (n == NULL) 86621f39848SGleb Smirnoff return (NULL); 86721f39848SGleb Smirnoff n->m_next = m->m_next; 86821f39848SGleb Smirnoff m->m_next = NULL; 86921f39848SGleb Smirnoff n->m_pkthdr.rcvif = m0->m_pkthdr.rcvif; 87021f39848SGleb Smirnoff n->m_pkthdr.len = m0->m_pkthdr.len - len0; 87121f39848SGleb Smirnoff m0->m_pkthdr.len = len0; 87221f39848SGleb Smirnoff return (n); 87321f39848SGleb Smirnoff } else if (m0->m_flags & M_PKTHDR) { 874c95be8b5SGleb Smirnoff n = m_gethdr(wait, m0->m_type); 875122a814aSBosko Milekic if (n == NULL) 876122a814aSBosko Milekic return (NULL); 877df8bae1dSRodney W. Grimes n->m_pkthdr.rcvif = m0->m_pkthdr.rcvif; 878df8bae1dSRodney W. Grimes n->m_pkthdr.len = m0->m_pkthdr.len - len0; 879df8bae1dSRodney W. Grimes m0->m_pkthdr.len = len0; 880df8bae1dSRodney W. Grimes if (m->m_flags & M_EXT) 881df8bae1dSRodney W. Grimes goto extpacket; 882df8bae1dSRodney W. Grimes if (remain > MHLEN) { 883df8bae1dSRodney W. Grimes /* m can't be the lead packet */ 884ed6a66caSRobert Watson M_ALIGN(n, 0); 885df8bae1dSRodney W. Grimes n->m_next = m_split(m, len, wait); 886122a814aSBosko Milekic if (n->m_next == NULL) { 887df8bae1dSRodney W. Grimes (void) m_free(n); 888122a814aSBosko Milekic return (NULL); 88940376987SJeffrey Hsu } else { 89040376987SJeffrey Hsu n->m_len = 0; 891df8bae1dSRodney W. Grimes return (n); 89240376987SJeffrey Hsu } 893df8bae1dSRodney W. Grimes } else 894ed6a66caSRobert Watson M_ALIGN(n, remain); 895df8bae1dSRodney W. Grimes } else if (remain == 0) { 896df8bae1dSRodney W. Grimes n = m->m_next; 897122a814aSBosko Milekic m->m_next = NULL; 898df8bae1dSRodney W. Grimes return (n); 899df8bae1dSRodney W. Grimes } else { 900c95be8b5SGleb Smirnoff n = m_get(wait, m->m_type); 901122a814aSBosko Milekic if (n == NULL) 902122a814aSBosko Milekic return (NULL); 903df8bae1dSRodney W. Grimes M_ALIGN(n, remain); 904df8bae1dSRodney W. Grimes } 905df8bae1dSRodney W. Grimes extpacket: 906df8bae1dSRodney W. Grimes if (m->m_flags & M_EXT) { 907df8bae1dSRodney W. Grimes n->m_data = m->m_data + len; 90856a4e45aSAndre Oppermann mb_dupcl(n, m); 909df8bae1dSRodney W. Grimes } else { 910df8bae1dSRodney W. Grimes bcopy(mtod(m, caddr_t) + len, mtod(n, caddr_t), remain); 911df8bae1dSRodney W. Grimes } 912df8bae1dSRodney W. Grimes n->m_len = remain; 913df8bae1dSRodney W. Grimes m->m_len = len; 914df8bae1dSRodney W. Grimes n->m_next = m->m_next; 915122a814aSBosko Milekic m->m_next = NULL; 916df8bae1dSRodney W. Grimes return (n); 917df8bae1dSRodney W. Grimes } 918df8bae1dSRodney W. Grimes /* 919df8bae1dSRodney W. Grimes * Routine to copy from device local memory into mbufs. 920f5eece3fSBosko Milekic * Note that `off' argument is offset into first mbuf of target chain from 921f5eece3fSBosko Milekic * which to begin copying the data to. 922df8bae1dSRodney W. Grimes */ 923df8bae1dSRodney W. Grimes struct mbuf * 924f5eece3fSBosko Milekic m_devget(char *buf, int totlen, int off, struct ifnet *ifp, 925122a814aSBosko Milekic void (*copy)(char *from, caddr_t to, u_int len)) 926df8bae1dSRodney W. Grimes { 927122a814aSBosko Milekic struct mbuf *m; 928099a0e58SBosko Milekic struct mbuf *top = NULL, **mp = ⊤ 929f5eece3fSBosko Milekic int len; 930df8bae1dSRodney W. Grimes 931f5eece3fSBosko Milekic if (off < 0 || off > MHLEN) 932f5eece3fSBosko Milekic return (NULL); 933f5eece3fSBosko Milekic 934df8bae1dSRodney W. Grimes while (totlen > 0) { 935099a0e58SBosko Milekic if (top == NULL) { /* First one, must be PKTHDR */ 936f5eece3fSBosko Milekic if (totlen + off >= MINCLSIZE) { 937eb1b1807SGleb Smirnoff m = m_getcl(M_NOWAIT, MT_DATA, M_PKTHDR); 938f5eece3fSBosko Milekic len = MCLBYTES; 939df8bae1dSRodney W. Grimes } else { 940eb1b1807SGleb Smirnoff m = m_gethdr(M_NOWAIT, MT_DATA); 941099a0e58SBosko Milekic len = MHLEN; 942099a0e58SBosko Milekic 943099a0e58SBosko Milekic /* Place initial small packet/header at end of mbuf */ 944099a0e58SBosko Milekic if (m && totlen + off + max_linkhdr <= MLEN) { 945df8bae1dSRodney W. Grimes m->m_data += max_linkhdr; 946f5eece3fSBosko Milekic len -= max_linkhdr; 947df8bae1dSRodney W. Grimes } 948f5eece3fSBosko Milekic } 949099a0e58SBosko Milekic if (m == NULL) 950099a0e58SBosko Milekic return NULL; 951099a0e58SBosko Milekic m->m_pkthdr.rcvif = ifp; 952099a0e58SBosko Milekic m->m_pkthdr.len = totlen; 953099a0e58SBosko Milekic } else { 954099a0e58SBosko Milekic if (totlen + off >= MINCLSIZE) { 955eb1b1807SGleb Smirnoff m = m_getcl(M_NOWAIT, MT_DATA, 0); 956099a0e58SBosko Milekic len = MCLBYTES; 957099a0e58SBosko Milekic } else { 958eb1b1807SGleb Smirnoff m = m_get(M_NOWAIT, MT_DATA); 959099a0e58SBosko Milekic len = MLEN; 960099a0e58SBosko Milekic } 961099a0e58SBosko Milekic if (m == NULL) { 962099a0e58SBosko Milekic m_freem(top); 963099a0e58SBosko Milekic return NULL; 964099a0e58SBosko Milekic } 965099a0e58SBosko Milekic } 966f5eece3fSBosko Milekic if (off) { 967f5eece3fSBosko Milekic m->m_data += off; 968f5eece3fSBosko Milekic len -= off; 969f5eece3fSBosko Milekic off = 0; 970f5eece3fSBosko Milekic } 971f5eece3fSBosko Milekic m->m_len = len = min(totlen, len); 972df8bae1dSRodney W. Grimes if (copy) 973bd395ae8SBosko Milekic copy(buf, mtod(m, caddr_t), (u_int)len); 974df8bae1dSRodney W. Grimes else 975bd395ae8SBosko Milekic bcopy(buf, mtod(m, caddr_t), (u_int)len); 976f5eece3fSBosko Milekic buf += len; 977df8bae1dSRodney W. Grimes *mp = m; 978df8bae1dSRodney W. Grimes mp = &m->m_next; 979df8bae1dSRodney W. Grimes totlen -= len; 980df8bae1dSRodney W. Grimes } 981df8bae1dSRodney W. Grimes return (top); 982df8bae1dSRodney W. Grimes } 983c5789ba3SPoul-Henning Kamp 984c5789ba3SPoul-Henning Kamp /* 985c5789ba3SPoul-Henning Kamp * Copy data from a buffer back into the indicated mbuf chain, 986c5789ba3SPoul-Henning Kamp * starting "off" bytes from the beginning, extending the mbuf 987c5789ba3SPoul-Henning Kamp * chain if necessary. 988c5789ba3SPoul-Henning Kamp */ 989c5789ba3SPoul-Henning Kamp void 99024665342SLuigi Rizzo m_copyback(struct mbuf *m0, int off, int len, c_caddr_t cp) 991c5789ba3SPoul-Henning Kamp { 992122a814aSBosko Milekic int mlen; 993122a814aSBosko Milekic struct mbuf *m = m0, *n; 994c5789ba3SPoul-Henning Kamp int totlen = 0; 995c5789ba3SPoul-Henning Kamp 996122a814aSBosko Milekic if (m0 == NULL) 997c5789ba3SPoul-Henning Kamp return; 998c5789ba3SPoul-Henning Kamp while (off > (mlen = m->m_len)) { 999c5789ba3SPoul-Henning Kamp off -= mlen; 1000c5789ba3SPoul-Henning Kamp totlen += mlen; 1001122a814aSBosko Milekic if (m->m_next == NULL) { 1002eb1b1807SGleb Smirnoff n = m_get(M_NOWAIT, m->m_type); 1003122a814aSBosko Milekic if (n == NULL) 1004c5789ba3SPoul-Henning Kamp goto out; 1005099a0e58SBosko Milekic bzero(mtod(n, caddr_t), MLEN); 1006c5789ba3SPoul-Henning Kamp n->m_len = min(MLEN, len + off); 1007c5789ba3SPoul-Henning Kamp m->m_next = n; 1008c5789ba3SPoul-Henning Kamp } 1009c5789ba3SPoul-Henning Kamp m = m->m_next; 1010c5789ba3SPoul-Henning Kamp } 1011c5789ba3SPoul-Henning Kamp while (len > 0) { 1012129c5c81SAlexander Motin if (m->m_next == NULL && (len > m->m_len - off)) { 1013129c5c81SAlexander Motin m->m_len += min(len - (m->m_len - off), 1014129c5c81SAlexander Motin M_TRAILINGSPACE(m)); 1015129c5c81SAlexander Motin } 1016c5789ba3SPoul-Henning Kamp mlen = min (m->m_len - off, len); 1017bd395ae8SBosko Milekic bcopy(cp, off + mtod(m, caddr_t), (u_int)mlen); 1018c5789ba3SPoul-Henning Kamp cp += mlen; 1019c5789ba3SPoul-Henning Kamp len -= mlen; 1020c5789ba3SPoul-Henning Kamp mlen += off; 1021c5789ba3SPoul-Henning Kamp off = 0; 1022c5789ba3SPoul-Henning Kamp totlen += mlen; 1023c5789ba3SPoul-Henning Kamp if (len == 0) 1024c5789ba3SPoul-Henning Kamp break; 1025122a814aSBosko Milekic if (m->m_next == NULL) { 1026eb1b1807SGleb Smirnoff n = m_get(M_NOWAIT, m->m_type); 1027122a814aSBosko Milekic if (n == NULL) 1028c5789ba3SPoul-Henning Kamp break; 1029c5789ba3SPoul-Henning Kamp n->m_len = min(MLEN, len); 1030c5789ba3SPoul-Henning Kamp m->m_next = n; 1031c5789ba3SPoul-Henning Kamp } 1032c5789ba3SPoul-Henning Kamp m = m->m_next; 1033c5789ba3SPoul-Henning Kamp } 1034c5789ba3SPoul-Henning Kamp out: if (((m = m0)->m_flags & M_PKTHDR) && (m->m_pkthdr.len < totlen)) 1035c5789ba3SPoul-Henning Kamp m->m_pkthdr.len = totlen; 1036c5789ba3SPoul-Henning Kamp } 1037ce4a64f7SPoul-Henning Kamp 103837621fd5SBruce M Simpson /* 10394873d175SSam Leffler * Append the specified data to the indicated mbuf chain, 10404873d175SSam Leffler * Extend the mbuf chain if the new data does not fit in 10414873d175SSam Leffler * existing space. 10424873d175SSam Leffler * 10434873d175SSam Leffler * Return 1 if able to complete the job; otherwise 0. 10444873d175SSam Leffler */ 10454873d175SSam Leffler int 10464873d175SSam Leffler m_append(struct mbuf *m0, int len, c_caddr_t cp) 10474873d175SSam Leffler { 10484873d175SSam Leffler struct mbuf *m, *n; 10494873d175SSam Leffler int remainder, space; 10504873d175SSam Leffler 10514873d175SSam Leffler for (m = m0; m->m_next != NULL; m = m->m_next) 10524873d175SSam Leffler ; 10534873d175SSam Leffler remainder = len; 10544873d175SSam Leffler space = M_TRAILINGSPACE(m); 10554873d175SSam Leffler if (space > 0) { 10564873d175SSam Leffler /* 10574873d175SSam Leffler * Copy into available space. 10584873d175SSam Leffler */ 10594873d175SSam Leffler if (space > remainder) 10604873d175SSam Leffler space = remainder; 10614873d175SSam Leffler bcopy(cp, mtod(m, caddr_t) + m->m_len, space); 10624873d175SSam Leffler m->m_len += space; 10634873d175SSam Leffler cp += space, remainder -= space; 10644873d175SSam Leffler } 10654873d175SSam Leffler while (remainder > 0) { 10664873d175SSam Leffler /* 10674873d175SSam Leffler * Allocate a new mbuf; could check space 10684873d175SSam Leffler * and allocate a cluster instead. 10694873d175SSam Leffler */ 1070eb1b1807SGleb Smirnoff n = m_get(M_NOWAIT, m->m_type); 10714873d175SSam Leffler if (n == NULL) 10724873d175SSam Leffler break; 10734873d175SSam Leffler n->m_len = min(MLEN, remainder); 1074a37c415eSSam Leffler bcopy(cp, mtod(n, caddr_t), n->m_len); 1075a37c415eSSam Leffler cp += n->m_len, remainder -= n->m_len; 10764873d175SSam Leffler m->m_next = n; 10774873d175SSam Leffler m = n; 10784873d175SSam Leffler } 10794873d175SSam Leffler if (m0->m_flags & M_PKTHDR) 10804873d175SSam Leffler m0->m_pkthdr.len += len - remainder; 10814873d175SSam Leffler return (remainder == 0); 10824873d175SSam Leffler } 10834873d175SSam Leffler 10844873d175SSam Leffler /* 108537621fd5SBruce M Simpson * Apply function f to the data in an mbuf chain starting "off" bytes from 108637621fd5SBruce M Simpson * the beginning, continuing for "len" bytes. 108737621fd5SBruce M Simpson */ 108837621fd5SBruce M Simpson int 108937621fd5SBruce M Simpson m_apply(struct mbuf *m, int off, int len, 109054065297SBruce M Simpson int (*f)(void *, void *, u_int), void *arg) 109137621fd5SBruce M Simpson { 109254065297SBruce M Simpson u_int count; 109337621fd5SBruce M Simpson int rval; 109437621fd5SBruce M Simpson 109537621fd5SBruce M Simpson KASSERT(off >= 0, ("m_apply, negative off %d", off)); 109637621fd5SBruce M Simpson KASSERT(len >= 0, ("m_apply, negative len %d", len)); 109737621fd5SBruce M Simpson while (off > 0) { 109837621fd5SBruce M Simpson KASSERT(m != NULL, ("m_apply, offset > size of mbuf chain")); 109937621fd5SBruce M Simpson if (off < m->m_len) 110037621fd5SBruce M Simpson break; 110137621fd5SBruce M Simpson off -= m->m_len; 110237621fd5SBruce M Simpson m = m->m_next; 110337621fd5SBruce M Simpson } 110437621fd5SBruce M Simpson while (len > 0) { 110537621fd5SBruce M Simpson KASSERT(m != NULL, ("m_apply, offset > size of mbuf chain")); 110637621fd5SBruce M Simpson count = min(m->m_len - off, len); 110737621fd5SBruce M Simpson rval = (*f)(arg, mtod(m, caddr_t) + off, count); 110837621fd5SBruce M Simpson if (rval) 110937621fd5SBruce M Simpson return (rval); 111037621fd5SBruce M Simpson len -= count; 111137621fd5SBruce M Simpson off = 0; 111237621fd5SBruce M Simpson m = m->m_next; 111337621fd5SBruce M Simpson } 111437621fd5SBruce M Simpson return (0); 111537621fd5SBruce M Simpson } 111637621fd5SBruce M Simpson 111737621fd5SBruce M Simpson /* 111837621fd5SBruce M Simpson * Return a pointer to mbuf/offset of location in mbuf chain. 111937621fd5SBruce M Simpson */ 112037621fd5SBruce M Simpson struct mbuf * 112137621fd5SBruce M Simpson m_getptr(struct mbuf *m, int loc, int *off) 112237621fd5SBruce M Simpson { 112337621fd5SBruce M Simpson 112437621fd5SBruce M Simpson while (loc >= 0) { 112554065297SBruce M Simpson /* Normal end of search. */ 112637621fd5SBruce M Simpson if (m->m_len > loc) { 112737621fd5SBruce M Simpson *off = loc; 112837621fd5SBruce M Simpson return (m); 112937621fd5SBruce M Simpson } else { 113037621fd5SBruce M Simpson loc -= m->m_len; 113137621fd5SBruce M Simpson if (m->m_next == NULL) { 113237621fd5SBruce M Simpson if (loc == 0) { 113354065297SBruce M Simpson /* Point at the end of valid data. */ 113437621fd5SBruce M Simpson *off = m->m_len; 113537621fd5SBruce M Simpson return (m); 113654065297SBruce M Simpson } 113737621fd5SBruce M Simpson return (NULL); 113854065297SBruce M Simpson } 113937621fd5SBruce M Simpson m = m->m_next; 114037621fd5SBruce M Simpson } 114137621fd5SBruce M Simpson } 114237621fd5SBruce M Simpson return (NULL); 114337621fd5SBruce M Simpson } 114437621fd5SBruce M Simpson 1145ce4a64f7SPoul-Henning Kamp void 11467b125090SJohn-Mark Gurney m_print(const struct mbuf *m, int maxlen) 1147ce4a64f7SPoul-Henning Kamp { 1148ce4a64f7SPoul-Henning Kamp int len; 11497b125090SJohn-Mark Gurney int pdata; 11506357e7b5SEivind Eklund const struct mbuf *m2; 1151ce4a64f7SPoul-Henning Kamp 11527e949c46SKenneth D. Merry if (m == NULL) { 11537e949c46SKenneth D. Merry printf("mbuf: %p\n", m); 11547e949c46SKenneth D. Merry return; 11557e949c46SKenneth D. Merry } 11567e949c46SKenneth D. Merry 11577b125090SJohn-Mark Gurney if (m->m_flags & M_PKTHDR) 1158ce4a64f7SPoul-Henning Kamp len = m->m_pkthdr.len; 11597b125090SJohn-Mark Gurney else 11607b125090SJohn-Mark Gurney len = -1; 1161ce4a64f7SPoul-Henning Kamp m2 = m; 11627b125090SJohn-Mark Gurney while (m2 != NULL && (len == -1 || len)) { 11637b125090SJohn-Mark Gurney pdata = m2->m_len; 11647b125090SJohn-Mark Gurney if (maxlen != -1 && pdata > maxlen) 11657b125090SJohn-Mark Gurney pdata = maxlen; 11667b125090SJohn-Mark Gurney printf("mbuf: %p len: %d, next: %p, %b%s", m2, m2->m_len, 11677b125090SJohn-Mark Gurney m2->m_next, m2->m_flags, "\20\20freelist\17skipfw" 11687b125090SJohn-Mark Gurney "\11proto5\10proto4\7proto3\6proto2\5proto1\4rdonly" 11697b125090SJohn-Mark Gurney "\3eor\2pkthdr\1ext", pdata ? "" : "\n"); 11707b125090SJohn-Mark Gurney if (pdata) 117145e0d0aaSJohn-Mark Gurney printf(", %*D\n", pdata, (u_char *)m2->m_data, "-"); 11727b125090SJohn-Mark Gurney if (len != -1) 1173ce4a64f7SPoul-Henning Kamp len -= m2->m_len; 1174ce4a64f7SPoul-Henning Kamp m2 = m2->m_next; 1175ce4a64f7SPoul-Henning Kamp } 11767b125090SJohn-Mark Gurney if (len > 0) 11777b125090SJohn-Mark Gurney printf("%d bytes unaccounted for.\n", len); 1178ce4a64f7SPoul-Henning Kamp return; 1179ce4a64f7SPoul-Henning Kamp } 11803f2e06c5SPoul-Henning Kamp 1181bd395ae8SBosko Milekic u_int 11823f2e06c5SPoul-Henning Kamp m_fixhdr(struct mbuf *m0) 11833f2e06c5SPoul-Henning Kamp { 1184bd395ae8SBosko Milekic u_int len; 11853f2e06c5SPoul-Henning Kamp 1186ac6e585dSPoul-Henning Kamp len = m_length(m0, NULL); 11873f2e06c5SPoul-Henning Kamp m0->m_pkthdr.len = len; 1188ac6e585dSPoul-Henning Kamp return (len); 1189ac6e585dSPoul-Henning Kamp } 1190ac6e585dSPoul-Henning Kamp 1191bd395ae8SBosko Milekic u_int 1192ac6e585dSPoul-Henning Kamp m_length(struct mbuf *m0, struct mbuf **last) 1193ac6e585dSPoul-Henning Kamp { 1194ac6e585dSPoul-Henning Kamp struct mbuf *m; 1195bd395ae8SBosko Milekic u_int len; 1196ac6e585dSPoul-Henning Kamp 1197ac6e585dSPoul-Henning Kamp len = 0; 1198ac6e585dSPoul-Henning Kamp for (m = m0; m != NULL; m = m->m_next) { 1199ac6e585dSPoul-Henning Kamp len += m->m_len; 1200ac6e585dSPoul-Henning Kamp if (m->m_next == NULL) 1201ac6e585dSPoul-Henning Kamp break; 1202ac6e585dSPoul-Henning Kamp } 1203ac6e585dSPoul-Henning Kamp if (last != NULL) 1204ac6e585dSPoul-Henning Kamp *last = m; 1205ac6e585dSPoul-Henning Kamp return (len); 12063f2e06c5SPoul-Henning Kamp } 120755e9f80dSMike Silbersack 120855e9f80dSMike Silbersack /* 120955e9f80dSMike Silbersack * Defragment a mbuf chain, returning the shortest possible 121055e9f80dSMike Silbersack * chain of mbufs and clusters. If allocation fails and 121155e9f80dSMike Silbersack * this cannot be completed, NULL will be returned, but 121255e9f80dSMike Silbersack * the passed in chain will be unchanged. Upon success, 121355e9f80dSMike Silbersack * the original chain will be freed, and the new chain 121455e9f80dSMike Silbersack * will be returned. 121555e9f80dSMike Silbersack * 121655e9f80dSMike Silbersack * If a non-packet header is passed in, the original 121755e9f80dSMike Silbersack * mbuf (chain?) will be returned unharmed. 121855e9f80dSMike Silbersack */ 121955e9f80dSMike Silbersack struct mbuf * 122055e9f80dSMike Silbersack m_defrag(struct mbuf *m0, int how) 122155e9f80dSMike Silbersack { 122255e9f80dSMike Silbersack struct mbuf *m_new = NULL, *m_final = NULL; 122355e9f80dSMike Silbersack int progress = 0, length; 122455e9f80dSMike Silbersack 1225063d8114SAlfred Perlstein MBUF_CHECKSLEEP(how); 122655e9f80dSMike Silbersack if (!(m0->m_flags & M_PKTHDR)) 122755e9f80dSMike Silbersack return (m0); 122855e9f80dSMike Silbersack 1229f8bf8e39SMike Silbersack m_fixhdr(m0); /* Needed sanity check */ 1230f8bf8e39SMike Silbersack 1231352d050eSMike Silbersack #ifdef MBUF_STRESS_TEST 1232352d050eSMike Silbersack if (m_defragrandomfailures) { 1233352d050eSMike Silbersack int temp = arc4random() & 0xff; 1234352d050eSMike Silbersack if (temp == 0xba) 1235352d050eSMike Silbersack goto nospace; 1236352d050eSMike Silbersack } 1237352d050eSMike Silbersack #endif 123855e9f80dSMike Silbersack 123955e9f80dSMike Silbersack if (m0->m_pkthdr.len > MHLEN) 124055e9f80dSMike Silbersack m_final = m_getcl(how, MT_DATA, M_PKTHDR); 124155e9f80dSMike Silbersack else 124255e9f80dSMike Silbersack m_final = m_gethdr(how, MT_DATA); 124355e9f80dSMike Silbersack 124455e9f80dSMike Silbersack if (m_final == NULL) 124555e9f80dSMike Silbersack goto nospace; 124655e9f80dSMike Silbersack 1247a89ec05eSPeter Wemm if (m_dup_pkthdr(m_final, m0, how) == 0) 124855e9f80dSMike Silbersack goto nospace; 124955e9f80dSMike Silbersack 125055e9f80dSMike Silbersack m_new = m_final; 125155e9f80dSMike Silbersack 125255e9f80dSMike Silbersack while (progress < m0->m_pkthdr.len) { 125355e9f80dSMike Silbersack length = m0->m_pkthdr.len - progress; 125455e9f80dSMike Silbersack if (length > MCLBYTES) 125555e9f80dSMike Silbersack length = MCLBYTES; 125655e9f80dSMike Silbersack 125755e9f80dSMike Silbersack if (m_new == NULL) { 125855e9f80dSMike Silbersack if (length > MLEN) 125955e9f80dSMike Silbersack m_new = m_getcl(how, MT_DATA, 0); 126055e9f80dSMike Silbersack else 126155e9f80dSMike Silbersack m_new = m_get(how, MT_DATA); 126255e9f80dSMike Silbersack if (m_new == NULL) 126355e9f80dSMike Silbersack goto nospace; 126455e9f80dSMike Silbersack } 126555e9f80dSMike Silbersack 126655e9f80dSMike Silbersack m_copydata(m0, progress, length, mtod(m_new, caddr_t)); 126755e9f80dSMike Silbersack progress += length; 126855e9f80dSMike Silbersack m_new->m_len = length; 126955e9f80dSMike Silbersack if (m_new != m_final) 127055e9f80dSMike Silbersack m_cat(m_final, m_new); 127155e9f80dSMike Silbersack m_new = NULL; 127255e9f80dSMike Silbersack } 127351710a45SMike Silbersack #ifdef MBUF_STRESS_TEST 127455e9f80dSMike Silbersack if (m0->m_next == NULL) 127555e9f80dSMike Silbersack m_defraguseless++; 127651710a45SMike Silbersack #endif 127755e9f80dSMike Silbersack m_freem(m0); 127855e9f80dSMike Silbersack m0 = m_final; 127951710a45SMike Silbersack #ifdef MBUF_STRESS_TEST 128055e9f80dSMike Silbersack m_defragpackets++; 128155e9f80dSMike Silbersack m_defragbytes += m0->m_pkthdr.len; 128251710a45SMike Silbersack #endif 128355e9f80dSMike Silbersack return (m0); 128455e9f80dSMike Silbersack nospace: 128551710a45SMike Silbersack #ifdef MBUF_STRESS_TEST 128655e9f80dSMike Silbersack m_defragfailure++; 128751710a45SMike Silbersack #endif 128855e9f80dSMike Silbersack if (m_final) 128955e9f80dSMike Silbersack m_freem(m_final); 129055e9f80dSMike Silbersack return (NULL); 129155e9f80dSMike Silbersack } 12923390d476SMike Silbersack 1293eeb76a18SSam Leffler /* 1294eeb76a18SSam Leffler * Defragment an mbuf chain, returning at most maxfrags separate 1295eeb76a18SSam Leffler * mbufs+clusters. If this is not possible NULL is returned and 1296eeb76a18SSam Leffler * the original mbuf chain is left in it's present (potentially 1297eeb76a18SSam Leffler * modified) state. We use two techniques: collapsing consecutive 1298eeb76a18SSam Leffler * mbufs and replacing consecutive mbufs by a cluster. 1299eeb76a18SSam Leffler * 1300eeb76a18SSam Leffler * NB: this should really be named m_defrag but that name is taken 1301eeb76a18SSam Leffler */ 1302eeb76a18SSam Leffler struct mbuf * 1303eeb76a18SSam Leffler m_collapse(struct mbuf *m0, int how, int maxfrags) 1304eeb76a18SSam Leffler { 1305eeb76a18SSam Leffler struct mbuf *m, *n, *n2, **prev; 1306eeb76a18SSam Leffler u_int curfrags; 1307eeb76a18SSam Leffler 1308eeb76a18SSam Leffler /* 1309eeb76a18SSam Leffler * Calculate the current number of frags. 1310eeb76a18SSam Leffler */ 1311eeb76a18SSam Leffler curfrags = 0; 1312eeb76a18SSam Leffler for (m = m0; m != NULL; m = m->m_next) 1313eeb76a18SSam Leffler curfrags++; 1314eeb76a18SSam Leffler /* 1315eeb76a18SSam Leffler * First, try to collapse mbufs. Note that we always collapse 1316eeb76a18SSam Leffler * towards the front so we don't need to deal with moving the 1317eeb76a18SSam Leffler * pkthdr. This may be suboptimal if the first mbuf has much 1318eeb76a18SSam Leffler * less data than the following. 1319eeb76a18SSam Leffler */ 1320eeb76a18SSam Leffler m = m0; 1321eeb76a18SSam Leffler again: 1322eeb76a18SSam Leffler for (;;) { 1323eeb76a18SSam Leffler n = m->m_next; 1324eeb76a18SSam Leffler if (n == NULL) 1325eeb76a18SSam Leffler break; 132614d7c5b1SAndre Oppermann if (M_WRITABLE(m) && 1327eeb76a18SSam Leffler n->m_len < M_TRAILINGSPACE(m)) { 1328eeb76a18SSam Leffler bcopy(mtod(n, void *), mtod(m, char *) + m->m_len, 1329eeb76a18SSam Leffler n->m_len); 1330eeb76a18SSam Leffler m->m_len += n->m_len; 1331eeb76a18SSam Leffler m->m_next = n->m_next; 1332eeb76a18SSam Leffler m_free(n); 1333eeb76a18SSam Leffler if (--curfrags <= maxfrags) 1334eeb76a18SSam Leffler return m0; 1335eeb76a18SSam Leffler } else 1336eeb76a18SSam Leffler m = n; 1337eeb76a18SSam Leffler } 1338eeb76a18SSam Leffler KASSERT(maxfrags > 1, 1339eeb76a18SSam Leffler ("maxfrags %u, but normal collapse failed", maxfrags)); 1340eeb76a18SSam Leffler /* 1341eeb76a18SSam Leffler * Collapse consecutive mbufs to a cluster. 1342eeb76a18SSam Leffler */ 1343eeb76a18SSam Leffler prev = &m0->m_next; /* NB: not the first mbuf */ 1344eeb76a18SSam Leffler while ((n = *prev) != NULL) { 1345eeb76a18SSam Leffler if ((n2 = n->m_next) != NULL && 1346eeb76a18SSam Leffler n->m_len + n2->m_len < MCLBYTES) { 1347eeb76a18SSam Leffler m = m_getcl(how, MT_DATA, 0); 1348eeb76a18SSam Leffler if (m == NULL) 1349eeb76a18SSam Leffler goto bad; 1350eeb76a18SSam Leffler bcopy(mtod(n, void *), mtod(m, void *), n->m_len); 1351eeb76a18SSam Leffler bcopy(mtod(n2, void *), mtod(m, char *) + n->m_len, 1352eeb76a18SSam Leffler n2->m_len); 1353eeb76a18SSam Leffler m->m_len = n->m_len + n2->m_len; 1354eeb76a18SSam Leffler m->m_next = n2->m_next; 1355eeb76a18SSam Leffler *prev = m; 1356eeb76a18SSam Leffler m_free(n); 1357eeb76a18SSam Leffler m_free(n2); 1358eeb76a18SSam Leffler if (--curfrags <= maxfrags) /* +1 cl -2 mbufs */ 1359eeb76a18SSam Leffler return m0; 1360eeb76a18SSam Leffler /* 1361eeb76a18SSam Leffler * Still not there, try the normal collapse 1362eeb76a18SSam Leffler * again before we allocate another cluster. 1363eeb76a18SSam Leffler */ 1364eeb76a18SSam Leffler goto again; 1365eeb76a18SSam Leffler } 1366eeb76a18SSam Leffler prev = &n->m_next; 1367eeb76a18SSam Leffler } 1368eeb76a18SSam Leffler /* 1369eeb76a18SSam Leffler * No place where we can collapse to a cluster; punt. 1370eeb76a18SSam Leffler * This can occur if, for example, you request 2 frags 1371eeb76a18SSam Leffler * but the packet requires that both be clusters (we 1372eeb76a18SSam Leffler * never reallocate the first mbuf to avoid moving the 1373eeb76a18SSam Leffler * packet header). 1374eeb76a18SSam Leffler */ 1375eeb76a18SSam Leffler bad: 1376eeb76a18SSam Leffler return NULL; 1377eeb76a18SSam Leffler } 1378eeb76a18SSam Leffler 13793390d476SMike Silbersack #ifdef MBUF_STRESS_TEST 13803390d476SMike Silbersack 13813390d476SMike Silbersack /* 13823390d476SMike Silbersack * Fragment an mbuf chain. There's no reason you'd ever want to do 13833390d476SMike Silbersack * this in normal usage, but it's great for stress testing various 13843390d476SMike Silbersack * mbuf consumers. 13853390d476SMike Silbersack * 13863390d476SMike Silbersack * If fragmentation is not possible, the original chain will be 13873390d476SMike Silbersack * returned. 13883390d476SMike Silbersack * 13893390d476SMike Silbersack * Possible length values: 13903390d476SMike Silbersack * 0 no fragmentation will occur 13913390d476SMike Silbersack * > 0 each fragment will be of the specified length 13923390d476SMike Silbersack * -1 each fragment will be the same random value in length 13933390d476SMike Silbersack * -2 each fragment's length will be entirely random 13943390d476SMike Silbersack * (Random values range from 1 to 256) 13953390d476SMike Silbersack */ 13963390d476SMike Silbersack struct mbuf * 13973390d476SMike Silbersack m_fragment(struct mbuf *m0, int how, int length) 13983390d476SMike Silbersack { 13993390d476SMike Silbersack struct mbuf *m_new = NULL, *m_final = NULL; 14003390d476SMike Silbersack int progress = 0; 14013390d476SMike Silbersack 14023390d476SMike Silbersack if (!(m0->m_flags & M_PKTHDR)) 14033390d476SMike Silbersack return (m0); 14043390d476SMike Silbersack 14053390d476SMike Silbersack if ((length == 0) || (length < -2)) 14063390d476SMike Silbersack return (m0); 14073390d476SMike Silbersack 14083390d476SMike Silbersack m_fixhdr(m0); /* Needed sanity check */ 14093390d476SMike Silbersack 14103390d476SMike Silbersack m_final = m_getcl(how, MT_DATA, M_PKTHDR); 14113390d476SMike Silbersack 14123390d476SMike Silbersack if (m_final == NULL) 14133390d476SMike Silbersack goto nospace; 14143390d476SMike Silbersack 14158dee2f67SMike Silbersack if (m_dup_pkthdr(m_final, m0, how) == 0) 14163390d476SMike Silbersack goto nospace; 14173390d476SMike Silbersack 14183390d476SMike Silbersack m_new = m_final; 14193390d476SMike Silbersack 14203390d476SMike Silbersack if (length == -1) 14213390d476SMike Silbersack length = 1 + (arc4random() & 255); 14223390d476SMike Silbersack 14233390d476SMike Silbersack while (progress < m0->m_pkthdr.len) { 14243390d476SMike Silbersack int fraglen; 14253390d476SMike Silbersack 14263390d476SMike Silbersack if (length > 0) 14273390d476SMike Silbersack fraglen = length; 14283390d476SMike Silbersack else 14293390d476SMike Silbersack fraglen = 1 + (arc4random() & 255); 14303390d476SMike Silbersack if (fraglen > m0->m_pkthdr.len - progress) 14313390d476SMike Silbersack fraglen = m0->m_pkthdr.len - progress; 14323390d476SMike Silbersack 14333390d476SMike Silbersack if (fraglen > MCLBYTES) 14343390d476SMike Silbersack fraglen = MCLBYTES; 14353390d476SMike Silbersack 14363390d476SMike Silbersack if (m_new == NULL) { 14373390d476SMike Silbersack m_new = m_getcl(how, MT_DATA, 0); 14383390d476SMike Silbersack if (m_new == NULL) 14393390d476SMike Silbersack goto nospace; 14403390d476SMike Silbersack } 14413390d476SMike Silbersack 14423390d476SMike Silbersack m_copydata(m0, progress, fraglen, mtod(m_new, caddr_t)); 14433390d476SMike Silbersack progress += fraglen; 14443390d476SMike Silbersack m_new->m_len = fraglen; 14453390d476SMike Silbersack if (m_new != m_final) 14463390d476SMike Silbersack m_cat(m_final, m_new); 14473390d476SMike Silbersack m_new = NULL; 14483390d476SMike Silbersack } 14493390d476SMike Silbersack m_freem(m0); 14503390d476SMike Silbersack m0 = m_final; 14513390d476SMike Silbersack return (m0); 14523390d476SMike Silbersack nospace: 14533390d476SMike Silbersack if (m_final) 14543390d476SMike Silbersack m_freem(m_final); 14553390d476SMike Silbersack /* Return the original chain on failure */ 14563390d476SMike Silbersack return (m0); 14573390d476SMike Silbersack } 14583390d476SMike Silbersack 14593390d476SMike Silbersack #endif 1460beb699c7SMike Silbersack 14615e20f43dSAndre Oppermann /* 14625e20f43dSAndre Oppermann * Copy the contents of uio into a properly sized mbuf chain. 14635e20f43dSAndre Oppermann */ 1464beb699c7SMike Silbersack struct mbuf * 14655e20f43dSAndre Oppermann m_uiotombuf(struct uio *uio, int how, int len, int align, int flags) 1466beb699c7SMike Silbersack { 14675e20f43dSAndre Oppermann struct mbuf *m, *mb; 1468526d0bd5SKonstantin Belousov int error, length; 1469526d0bd5SKonstantin Belousov ssize_t total; 14705e20f43dSAndre Oppermann int progress = 0; 1471beb699c7SMike Silbersack 14725e20f43dSAndre Oppermann /* 14735e20f43dSAndre Oppermann * len can be zero or an arbitrary large value bound by 14745e20f43dSAndre Oppermann * the total data supplied by the uio. 14755e20f43dSAndre Oppermann */ 1476beb699c7SMike Silbersack if (len > 0) 1477beb699c7SMike Silbersack total = min(uio->uio_resid, len); 1478beb699c7SMike Silbersack else 1479beb699c7SMike Silbersack total = uio->uio_resid; 14805e20f43dSAndre Oppermann 14815e20f43dSAndre Oppermann /* 14825e20f43dSAndre Oppermann * The smallest unit returned by m_getm2() is a single mbuf 14839128ec21SAndrew Thompson * with pkthdr. We can't align past it. 14845e20f43dSAndre Oppermann */ 148575ae2570SMaksim Yevmenkin if (align >= MHLEN) 1486beb699c7SMike Silbersack return (NULL); 14875e20f43dSAndre Oppermann 14887c32173bSAndre Oppermann /* 14897c32173bSAndre Oppermann * Give us the full allocation or nothing. 14907c32173bSAndre Oppermann * If len is zero return the smallest empty mbuf. 14917c32173bSAndre Oppermann */ 14927c32173bSAndre Oppermann m = m_getm2(NULL, max(total + align, 1), how, MT_DATA, flags); 14935e20f43dSAndre Oppermann if (m == NULL) 14945e20f43dSAndre Oppermann return (NULL); 14955e20f43dSAndre Oppermann m->m_data += align; 14965e20f43dSAndre Oppermann 14975e20f43dSAndre Oppermann /* Fill all mbufs with uio data and update header information. */ 14985e20f43dSAndre Oppermann for (mb = m; mb != NULL; mb = mb->m_next) { 14995e20f43dSAndre Oppermann length = min(M_TRAILINGSPACE(mb), total - progress); 15005e20f43dSAndre Oppermann 15015e20f43dSAndre Oppermann error = uiomove(mtod(mb, void *), length, uio); 15025e20f43dSAndre Oppermann if (error) { 15035e20f43dSAndre Oppermann m_freem(m); 15045e20f43dSAndre Oppermann return (NULL); 15055e20f43dSAndre Oppermann } 15065e20f43dSAndre Oppermann 15075e20f43dSAndre Oppermann mb->m_len = length; 15085e20f43dSAndre Oppermann progress += length; 15095e20f43dSAndre Oppermann if (flags & M_PKTHDR) 15105e20f43dSAndre Oppermann m->m_pkthdr.len += length; 15115e20f43dSAndre Oppermann } 15125e20f43dSAndre Oppermann KASSERT(progress == total, ("%s: progress != total", __func__)); 15135e20f43dSAndre Oppermann 15145e20f43dSAndre Oppermann return (m); 1515beb699c7SMike Silbersack } 1516ab8ab90cSSam Leffler 1517ab8ab90cSSam Leffler /* 1518bc05b2f6SAndre Oppermann * Copy an mbuf chain into a uio limited by len if set. 1519bc05b2f6SAndre Oppermann */ 1520bc05b2f6SAndre Oppermann int 1521bc05b2f6SAndre Oppermann m_mbuftouio(struct uio *uio, struct mbuf *m, int len) 1522bc05b2f6SAndre Oppermann { 1523bc05b2f6SAndre Oppermann int error, length, total; 1524bc05b2f6SAndre Oppermann int progress = 0; 1525bc05b2f6SAndre Oppermann 1526bc05b2f6SAndre Oppermann if (len > 0) 1527bc05b2f6SAndre Oppermann total = min(uio->uio_resid, len); 1528bc05b2f6SAndre Oppermann else 1529bc05b2f6SAndre Oppermann total = uio->uio_resid; 1530bc05b2f6SAndre Oppermann 1531bc05b2f6SAndre Oppermann /* Fill the uio with data from the mbufs. */ 1532bc05b2f6SAndre Oppermann for (; m != NULL; m = m->m_next) { 1533bc05b2f6SAndre Oppermann length = min(m->m_len, total - progress); 1534bc05b2f6SAndre Oppermann 1535bc05b2f6SAndre Oppermann error = uiomove(mtod(m, void *), length, uio); 1536bc05b2f6SAndre Oppermann if (error) 1537bc05b2f6SAndre Oppermann return (error); 1538bc05b2f6SAndre Oppermann 1539bc05b2f6SAndre Oppermann progress += length; 1540bc05b2f6SAndre Oppermann } 1541bc05b2f6SAndre Oppermann 1542bc05b2f6SAndre Oppermann return (0); 1543bc05b2f6SAndre Oppermann } 1544bc05b2f6SAndre Oppermann 1545bc05b2f6SAndre Oppermann /* 154647e2996eSSam Leffler * Create a writable copy of the mbuf chain. While doing this 154747e2996eSSam Leffler * we compact the chain with a goal of producing a chain with 154847e2996eSSam Leffler * at most two mbufs. The second mbuf in this chain is likely 154947e2996eSSam Leffler * to be a cluster. The primary purpose of this work is to create 155047e2996eSSam Leffler * a writable packet for encryption, compression, etc. The 155147e2996eSSam Leffler * secondary goal is to linearize the data so the data can be 155247e2996eSSam Leffler * passed to crypto hardware in the most efficient manner possible. 155347e2996eSSam Leffler */ 155447e2996eSSam Leffler struct mbuf * 155547e2996eSSam Leffler m_unshare(struct mbuf *m0, int how) 155647e2996eSSam Leffler { 155747e2996eSSam Leffler struct mbuf *m, *mprev; 155847e2996eSSam Leffler struct mbuf *n, *mfirst, *mlast; 155947e2996eSSam Leffler int len, off; 156047e2996eSSam Leffler 156147e2996eSSam Leffler mprev = NULL; 156247e2996eSSam Leffler for (m = m0; m != NULL; m = mprev->m_next) { 156347e2996eSSam Leffler /* 156447e2996eSSam Leffler * Regular mbufs are ignored unless there's a cluster 156547e2996eSSam Leffler * in front of it that we can use to coalesce. We do 156647e2996eSSam Leffler * the latter mainly so later clusters can be coalesced 156747e2996eSSam Leffler * also w/o having to handle them specially (i.e. convert 156847e2996eSSam Leffler * mbuf+cluster -> cluster). This optimization is heavily 156947e2996eSSam Leffler * influenced by the assumption that we're running over 157047e2996eSSam Leffler * Ethernet where MCLBYTES is large enough that the max 157147e2996eSSam Leffler * packet size will permit lots of coalescing into a 157247e2996eSSam Leffler * single cluster. This in turn permits efficient 157347e2996eSSam Leffler * crypto operations, especially when using hardware. 157447e2996eSSam Leffler */ 157547e2996eSSam Leffler if ((m->m_flags & M_EXT) == 0) { 157647e2996eSSam Leffler if (mprev && (mprev->m_flags & M_EXT) && 157747e2996eSSam Leffler m->m_len <= M_TRAILINGSPACE(mprev)) { 157847e2996eSSam Leffler /* XXX: this ignores mbuf types */ 157947e2996eSSam Leffler memcpy(mtod(mprev, caddr_t) + mprev->m_len, 158047e2996eSSam Leffler mtod(m, caddr_t), m->m_len); 158147e2996eSSam Leffler mprev->m_len += m->m_len; 158247e2996eSSam Leffler mprev->m_next = m->m_next; /* unlink from chain */ 158347e2996eSSam Leffler m_free(m); /* reclaim mbuf */ 158447e2996eSSam Leffler #if 0 158547e2996eSSam Leffler newipsecstat.ips_mbcoalesced++; 158647e2996eSSam Leffler #endif 158747e2996eSSam Leffler } else { 158847e2996eSSam Leffler mprev = m; 158947e2996eSSam Leffler } 159047e2996eSSam Leffler continue; 159147e2996eSSam Leffler } 159247e2996eSSam Leffler /* 159347e2996eSSam Leffler * Writable mbufs are left alone (for now). 159447e2996eSSam Leffler */ 159547e2996eSSam Leffler if (M_WRITABLE(m)) { 159647e2996eSSam Leffler mprev = m; 159747e2996eSSam Leffler continue; 159847e2996eSSam Leffler } 159947e2996eSSam Leffler 160047e2996eSSam Leffler /* 160147e2996eSSam Leffler * Not writable, replace with a copy or coalesce with 160247e2996eSSam Leffler * the previous mbuf if possible (since we have to copy 160347e2996eSSam Leffler * it anyway, we try to reduce the number of mbufs and 160447e2996eSSam Leffler * clusters so that future work is easier). 160547e2996eSSam Leffler */ 160647e2996eSSam Leffler KASSERT(m->m_flags & M_EXT, ("m_flags 0x%x", m->m_flags)); 160747e2996eSSam Leffler /* NB: we only coalesce into a cluster or larger */ 160847e2996eSSam Leffler if (mprev != NULL && (mprev->m_flags & M_EXT) && 160947e2996eSSam Leffler m->m_len <= M_TRAILINGSPACE(mprev)) { 161047e2996eSSam Leffler /* XXX: this ignores mbuf types */ 161147e2996eSSam Leffler memcpy(mtod(mprev, caddr_t) + mprev->m_len, 161247e2996eSSam Leffler mtod(m, caddr_t), m->m_len); 161347e2996eSSam Leffler mprev->m_len += m->m_len; 161447e2996eSSam Leffler mprev->m_next = m->m_next; /* unlink from chain */ 161547e2996eSSam Leffler m_free(m); /* reclaim mbuf */ 161647e2996eSSam Leffler #if 0 161747e2996eSSam Leffler newipsecstat.ips_clcoalesced++; 161847e2996eSSam Leffler #endif 161947e2996eSSam Leffler continue; 162047e2996eSSam Leffler } 162147e2996eSSam Leffler 162247e2996eSSam Leffler /* 16235368b81eSGleb Smirnoff * Allocate new space to hold the copy and copy the data. 16245368b81eSGleb Smirnoff * We deal with jumbo mbufs (i.e. m_len > MCLBYTES) by 16255368b81eSGleb Smirnoff * splitting them into clusters. We could just malloc a 16265368b81eSGleb Smirnoff * buffer and make it external but too many device drivers 16275368b81eSGleb Smirnoff * don't know how to break up the non-contiguous memory when 16285368b81eSGleb Smirnoff * doing DMA. 162947e2996eSSam Leffler */ 163047e2996eSSam Leffler n = m_getcl(how, m->m_type, m->m_flags); 163147e2996eSSam Leffler if (n == NULL) { 163247e2996eSSam Leffler m_freem(m0); 163347e2996eSSam Leffler return (NULL); 163447e2996eSSam Leffler } 1635e40e8705SGleb Smirnoff if (m->m_flags & M_PKTHDR) { 1636e40e8705SGleb Smirnoff KASSERT(mprev == NULL, ("%s: m0 %p, m %p has M_PKTHDR", 1637e40e8705SGleb Smirnoff __func__, m0, m)); 1638e40e8705SGleb Smirnoff m_move_pkthdr(n, m); 1639e40e8705SGleb Smirnoff } 164047e2996eSSam Leffler len = m->m_len; 164147e2996eSSam Leffler off = 0; 164247e2996eSSam Leffler mfirst = n; 164347e2996eSSam Leffler mlast = NULL; 164447e2996eSSam Leffler for (;;) { 164547e2996eSSam Leffler int cc = min(len, MCLBYTES); 164647e2996eSSam Leffler memcpy(mtod(n, caddr_t), mtod(m, caddr_t) + off, cc); 164747e2996eSSam Leffler n->m_len = cc; 164847e2996eSSam Leffler if (mlast != NULL) 164947e2996eSSam Leffler mlast->m_next = n; 165047e2996eSSam Leffler mlast = n; 165147e2996eSSam Leffler #if 0 165247e2996eSSam Leffler newipsecstat.ips_clcopied++; 165347e2996eSSam Leffler #endif 165447e2996eSSam Leffler 165547e2996eSSam Leffler len -= cc; 165647e2996eSSam Leffler if (len <= 0) 165747e2996eSSam Leffler break; 165847e2996eSSam Leffler off += cc; 165947e2996eSSam Leffler 166047e2996eSSam Leffler n = m_getcl(how, m->m_type, m->m_flags); 166147e2996eSSam Leffler if (n == NULL) { 166247e2996eSSam Leffler m_freem(mfirst); 166347e2996eSSam Leffler m_freem(m0); 166447e2996eSSam Leffler return (NULL); 166547e2996eSSam Leffler } 166647e2996eSSam Leffler } 166747e2996eSSam Leffler n->m_next = m->m_next; 166847e2996eSSam Leffler if (mprev == NULL) 166947e2996eSSam Leffler m0 = mfirst; /* new head of chain */ 167047e2996eSSam Leffler else 167147e2996eSSam Leffler mprev->m_next = mfirst; /* replace old mbuf */ 167247e2996eSSam Leffler m_free(m); /* release old mbuf */ 167347e2996eSSam Leffler mprev = mfirst; 167447e2996eSSam Leffler } 167547e2996eSSam Leffler return (m0); 167647e2996eSSam Leffler } 16776eeac1d9SJulian Elischer 16786eeac1d9SJulian Elischer #ifdef MBUF_PROFILING 16796eeac1d9SJulian Elischer 16806eeac1d9SJulian Elischer #define MP_BUCKETS 32 /* don't just change this as things may overflow.*/ 16816eeac1d9SJulian Elischer struct mbufprofile { 16822182c0cfSJulian Elischer uintmax_t wasted[MP_BUCKETS]; 16832182c0cfSJulian Elischer uintmax_t used[MP_BUCKETS]; 16842182c0cfSJulian Elischer uintmax_t segments[MP_BUCKETS]; 16856eeac1d9SJulian Elischer } mbprof; 16866eeac1d9SJulian Elischer 16876eeac1d9SJulian Elischer #define MP_MAXDIGITS 21 /* strlen("16,000,000,000,000,000,000") == 21 */ 16886eeac1d9SJulian Elischer #define MP_NUMLINES 6 16896eeac1d9SJulian Elischer #define MP_NUMSPERLINE 16 16906eeac1d9SJulian Elischer #define MP_EXTRABYTES 64 /* > strlen("used:\nwasted:\nsegments:\n") */ 16916eeac1d9SJulian Elischer /* work out max space needed and add a bit of spare space too */ 16926eeac1d9SJulian Elischer #define MP_MAXLINE ((MP_MAXDIGITS+1) * MP_NUMSPERLINE) 16936eeac1d9SJulian Elischer #define MP_BUFSIZE ((MP_MAXLINE * MP_NUMLINES) + 1 + MP_EXTRABYTES) 16946eeac1d9SJulian Elischer 16956eeac1d9SJulian Elischer char mbprofbuf[MP_BUFSIZE]; 16966eeac1d9SJulian Elischer 16976eeac1d9SJulian Elischer void 16986eeac1d9SJulian Elischer m_profile(struct mbuf *m) 16996eeac1d9SJulian Elischer { 17006eeac1d9SJulian Elischer int segments = 0; 17016eeac1d9SJulian Elischer int used = 0; 17026eeac1d9SJulian Elischer int wasted = 0; 17036eeac1d9SJulian Elischer 17046eeac1d9SJulian Elischer while (m) { 17056eeac1d9SJulian Elischer segments++; 17066eeac1d9SJulian Elischer used += m->m_len; 17076eeac1d9SJulian Elischer if (m->m_flags & M_EXT) { 17086eeac1d9SJulian Elischer wasted += MHLEN - sizeof(m->m_ext) + 17096eeac1d9SJulian Elischer m->m_ext.ext_size - m->m_len; 17106eeac1d9SJulian Elischer } else { 17116eeac1d9SJulian Elischer if (m->m_flags & M_PKTHDR) 17126eeac1d9SJulian Elischer wasted += MHLEN - m->m_len; 17136eeac1d9SJulian Elischer else 17146eeac1d9SJulian Elischer wasted += MLEN - m->m_len; 17156eeac1d9SJulian Elischer } 17166eeac1d9SJulian Elischer m = m->m_next; 17176eeac1d9SJulian Elischer } 17186eeac1d9SJulian Elischer /* be paranoid.. it helps */ 17196eeac1d9SJulian Elischer if (segments > MP_BUCKETS - 1) 17206eeac1d9SJulian Elischer segments = MP_BUCKETS - 1; 17216eeac1d9SJulian Elischer if (used > 100000) 17226eeac1d9SJulian Elischer used = 100000; 17236eeac1d9SJulian Elischer if (wasted > 100000) 17246eeac1d9SJulian Elischer wasted = 100000; 17256eeac1d9SJulian Elischer /* store in the appropriate bucket */ 17266eeac1d9SJulian Elischer /* don't bother locking. if it's slightly off, so what? */ 17276eeac1d9SJulian Elischer mbprof.segments[segments]++; 17286eeac1d9SJulian Elischer mbprof.used[fls(used)]++; 17296eeac1d9SJulian Elischer mbprof.wasted[fls(wasted)]++; 17306eeac1d9SJulian Elischer } 17316eeac1d9SJulian Elischer 17326eeac1d9SJulian Elischer static void 17336eeac1d9SJulian Elischer mbprof_textify(void) 17346eeac1d9SJulian Elischer { 17356eeac1d9SJulian Elischer int offset; 17366eeac1d9SJulian Elischer char *c; 173760ae52f7SEd Schouten uint64_t *p; 17386eeac1d9SJulian Elischer 17396eeac1d9SJulian Elischer p = &mbprof.wasted[0]; 17406eeac1d9SJulian Elischer c = mbprofbuf; 17416eeac1d9SJulian Elischer offset = snprintf(c, MP_MAXLINE + 10, 17426eeac1d9SJulian Elischer "wasted:\n" 17432182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju " 17442182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju\n", 17456eeac1d9SJulian Elischer p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 17466eeac1d9SJulian Elischer p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]); 17476eeac1d9SJulian Elischer #ifdef BIG_ARRAY 17486eeac1d9SJulian Elischer p = &mbprof.wasted[16]; 17496eeac1d9SJulian Elischer c += offset; 17506eeac1d9SJulian Elischer offset = snprintf(c, MP_MAXLINE, 17512182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju " 17522182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju\n", 17536eeac1d9SJulian Elischer p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 17546eeac1d9SJulian Elischer p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]); 17556eeac1d9SJulian Elischer #endif 17566eeac1d9SJulian Elischer p = &mbprof.used[0]; 17576eeac1d9SJulian Elischer c += offset; 17586eeac1d9SJulian Elischer offset = snprintf(c, MP_MAXLINE + 10, 17596eeac1d9SJulian Elischer "used:\n" 17602182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju " 17612182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju\n", 17626eeac1d9SJulian Elischer p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 17636eeac1d9SJulian Elischer p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]); 17646eeac1d9SJulian Elischer #ifdef BIG_ARRAY 17656eeac1d9SJulian Elischer p = &mbprof.used[16]; 17666eeac1d9SJulian Elischer c += offset; 17676eeac1d9SJulian Elischer offset = snprintf(c, MP_MAXLINE, 17682182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju " 17692182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju\n", 17706eeac1d9SJulian Elischer p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 17716eeac1d9SJulian Elischer p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]); 17726eeac1d9SJulian Elischer #endif 17736eeac1d9SJulian Elischer p = &mbprof.segments[0]; 17746eeac1d9SJulian Elischer c += offset; 17756eeac1d9SJulian Elischer offset = snprintf(c, MP_MAXLINE + 10, 17766eeac1d9SJulian Elischer "segments:\n" 17772182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju " 17782182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju\n", 17796eeac1d9SJulian Elischer p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 17806eeac1d9SJulian Elischer p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]); 17816eeac1d9SJulian Elischer #ifdef BIG_ARRAY 17826eeac1d9SJulian Elischer p = &mbprof.segments[16]; 17836eeac1d9SJulian Elischer c += offset; 17846eeac1d9SJulian Elischer offset = snprintf(c, MP_MAXLINE, 17852182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju " 17862182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %jju", 17876eeac1d9SJulian Elischer p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 17886eeac1d9SJulian Elischer p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]); 17896eeac1d9SJulian Elischer #endif 17906eeac1d9SJulian Elischer } 17916eeac1d9SJulian Elischer 17926eeac1d9SJulian Elischer static int 17936eeac1d9SJulian Elischer mbprof_handler(SYSCTL_HANDLER_ARGS) 17946eeac1d9SJulian Elischer { 17956eeac1d9SJulian Elischer int error; 17966eeac1d9SJulian Elischer 17976eeac1d9SJulian Elischer mbprof_textify(); 17986eeac1d9SJulian Elischer error = SYSCTL_OUT(req, mbprofbuf, strlen(mbprofbuf) + 1); 17996eeac1d9SJulian Elischer return (error); 18006eeac1d9SJulian Elischer } 18016eeac1d9SJulian Elischer 18026eeac1d9SJulian Elischer static int 18036eeac1d9SJulian Elischer mbprof_clr_handler(SYSCTL_HANDLER_ARGS) 18046eeac1d9SJulian Elischer { 18056eeac1d9SJulian Elischer int clear, error; 18066eeac1d9SJulian Elischer 18076eeac1d9SJulian Elischer clear = 0; 18086eeac1d9SJulian Elischer error = sysctl_handle_int(oidp, &clear, 0, req); 18096eeac1d9SJulian Elischer if (error || !req->newptr) 18106eeac1d9SJulian Elischer return (error); 18116eeac1d9SJulian Elischer 18126eeac1d9SJulian Elischer if (clear) { 18136eeac1d9SJulian Elischer bzero(&mbprof, sizeof(mbprof)); 18146eeac1d9SJulian Elischer } 18156eeac1d9SJulian Elischer 18166eeac1d9SJulian Elischer return (error); 18176eeac1d9SJulian Elischer } 18186eeac1d9SJulian Elischer 18196eeac1d9SJulian Elischer 18206eeac1d9SJulian Elischer SYSCTL_PROC(_kern_ipc, OID_AUTO, mbufprofile, CTLTYPE_STRING|CTLFLAG_RD, 18216eeac1d9SJulian Elischer NULL, 0, mbprof_handler, "A", "mbuf profiling statistics"); 18226eeac1d9SJulian Elischer 18236eeac1d9SJulian Elischer SYSCTL_PROC(_kern_ipc, OID_AUTO, mbufprofileclr, CTLTYPE_INT|CTLFLAG_RW, 18246eeac1d9SJulian Elischer NULL, 0, mbprof_clr_handler, "I", "clear mbuf profiling statistics"); 18256eeac1d9SJulian Elischer #endif 18266eeac1d9SJulian Elischer 1827