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 51*5e4bc63bSGleb Smirnoff #include <security/mac/mac_framework.h> 52*5e4bc63bSGleb 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 1410cbefd30SAlexander V. Chernikov mb_dupcl(struct mbuf *n, const struct mbuf *m) 14256a4e45aSAndre Oppermann { 14356a4e45aSAndre Oppermann 1441fbe6a82SGleb Smirnoff KASSERT(m->m_flags & M_EXT, ("%s: M_EXT not set on %p", __func__, m)); 1451fbe6a82SGleb Smirnoff KASSERT(!(n->m_flags & M_EXT), ("%s: M_EXT set on %p", __func__, n)); 1461fbe6a82SGleb Smirnoff 1471fbe6a82SGleb Smirnoff switch (m->m_ext.ext_type) { 1481fbe6a82SGleb Smirnoff case EXT_SFBUF: 1492bab0c55SGleb Smirnoff case EXT_SFBUF_NOCACHE: 1501fbe6a82SGleb Smirnoff sf_ext_ref(m->m_ext.ext_arg1, m->m_ext.ext_arg2); 1511fbe6a82SGleb Smirnoff break; 1521fbe6a82SGleb Smirnoff default: 1531fbe6a82SGleb Smirnoff KASSERT(m->m_ext.ext_cnt != NULL, 1541fbe6a82SGleb Smirnoff ("%s: no refcounting pointer on %p", __func__, m)); 155fcc34a23SGleb Smirnoff if (*(m->m_ext.ext_cnt) == 1) 156fcc34a23SGleb Smirnoff *(m->m_ext.ext_cnt) += 1; 15756a4e45aSAndre Oppermann else 158fcc34a23SGleb Smirnoff atomic_add_int(m->m_ext.ext_cnt, 1); 1591fbe6a82SGleb Smirnoff } 1601fbe6a82SGleb Smirnoff 161c71b4037SGleb Smirnoff n->m_ext = m->m_ext; 16256a4e45aSAndre Oppermann n->m_flags |= M_EXT; 16332a8b1d8SColin Percival n->m_flags |= m->m_flags & M_RDONLY; 16456a4e45aSAndre Oppermann } 16556a4e45aSAndre Oppermann 1669523d1bfSNavdeep Parhar void 1679523d1bfSNavdeep Parhar m_demote_pkthdr(struct mbuf *m) 1689523d1bfSNavdeep Parhar { 1699523d1bfSNavdeep Parhar 1709523d1bfSNavdeep Parhar M_ASSERTPKTHDR(m); 1719523d1bfSNavdeep Parhar 1729523d1bfSNavdeep Parhar m_tag_delete_chain(m, NULL); 1739523d1bfSNavdeep Parhar m->m_flags &= ~M_PKTHDR; 1749523d1bfSNavdeep Parhar bzero(&m->m_pkthdr, sizeof(struct pkthdr)); 1759523d1bfSNavdeep Parhar } 1769523d1bfSNavdeep Parhar 17756a4e45aSAndre Oppermann /* 178ed111688SAndre Oppermann * Clean up mbuf (chain) from any tags and packet headers. 179e0068c3aSAndre Oppermann * If "all" is set then the first mbuf in the chain will be 180e0068c3aSAndre Oppermann * cleaned too. 181ed111688SAndre Oppermann */ 182ed111688SAndre Oppermann void 183651e4e6aSGleb Smirnoff m_demote(struct mbuf *m0, int all, int flags) 184ed111688SAndre Oppermann { 185ed111688SAndre Oppermann struct mbuf *m; 186ed111688SAndre Oppermann 187ed111688SAndre Oppermann for (m = all ? m0 : m0->m_next; m != NULL; m = m->m_next) { 1887ee2d058SGleb Smirnoff KASSERT(m->m_nextpkt == NULL, ("%s: m_nextpkt in m %p, m0 %p", 1897ee2d058SGleb Smirnoff __func__, m, m0)); 1909523d1bfSNavdeep Parhar if (m->m_flags & M_PKTHDR) 1919523d1bfSNavdeep Parhar m_demote_pkthdr(m); 192651e4e6aSGleb Smirnoff m->m_flags = m->m_flags & (M_EXT | M_RDONLY | M_NOFREE | flags); 193ed111688SAndre Oppermann } 194ed111688SAndre Oppermann } 195ed111688SAndre Oppermann 196ed111688SAndre Oppermann /* 197fdcc028dSAndre Oppermann * Sanity checks on mbuf (chain) for use in KASSERT() and general 198fdcc028dSAndre Oppermann * debugging. 199fdcc028dSAndre Oppermann * Returns 0 or panics when bad and 1 on all tests passed. 200fdcc028dSAndre Oppermann * Sanitize, 0 to run M_SANITY_ACTION, 1 to garble things so they 201fdcc028dSAndre Oppermann * blow up later. 202a048affbSAndre Oppermann */ 203a048affbSAndre Oppermann int 204a048affbSAndre Oppermann m_sanity(struct mbuf *m0, int sanitize) 205a048affbSAndre Oppermann { 206a048affbSAndre Oppermann struct mbuf *m; 207a048affbSAndre Oppermann caddr_t a, b; 208a048affbSAndre Oppermann int pktlen = 0; 209a048affbSAndre Oppermann 21021ee3e7aSKip Macy #ifdef INVARIANTS 21121ee3e7aSKip Macy #define M_SANITY_ACTION(s) panic("mbuf %p: " s, m) 21221ee3e7aSKip Macy #else 21321ee3e7aSKip Macy #define M_SANITY_ACTION(s) printf("mbuf %p: " s, m) 21421ee3e7aSKip Macy #endif 215a048affbSAndre Oppermann 216fdcc028dSAndre Oppermann for (m = m0; m != NULL; m = m->m_next) { 217a048affbSAndre Oppermann /* 218a048affbSAndre Oppermann * Basic pointer checks. If any of these fails then some 219a048affbSAndre Oppermann * unrelated kernel memory before or after us is trashed. 220a048affbSAndre Oppermann * No way to recover from that. 221a048affbSAndre Oppermann */ 222b66f2a48SRobert Watson a = M_START(m); 223b66f2a48SRobert Watson b = a + M_SIZE(m); 224a048affbSAndre Oppermann if ((caddr_t)m->m_data < a) 225a048affbSAndre Oppermann M_SANITY_ACTION("m_data outside mbuf data range left"); 226a048affbSAndre Oppermann if ((caddr_t)m->m_data > b) 227a048affbSAndre Oppermann M_SANITY_ACTION("m_data outside mbuf data range right"); 228a048affbSAndre Oppermann if ((caddr_t)m->m_data + m->m_len > b) 229a048affbSAndre Oppermann M_SANITY_ACTION("m_data + m_len exeeds mbuf space"); 230a048affbSAndre Oppermann 231a048affbSAndre Oppermann /* m->m_nextpkt may only be set on first mbuf in chain. */ 232fdcc028dSAndre Oppermann if (m != m0 && m->m_nextpkt != NULL) { 233a048affbSAndre Oppermann if (sanitize) { 234a048affbSAndre Oppermann m_freem(m->m_nextpkt); 235a048affbSAndre Oppermann m->m_nextpkt = (struct mbuf *)0xDEADC0DE; 236a048affbSAndre Oppermann } else 237a048affbSAndre Oppermann M_SANITY_ACTION("m->m_nextpkt on in-chain mbuf"); 238a048affbSAndre Oppermann } 239a048affbSAndre Oppermann 240a048affbSAndre Oppermann /* packet length (not mbuf length!) calculation */ 241a048affbSAndre Oppermann if (m0->m_flags & M_PKTHDR) 242a048affbSAndre Oppermann pktlen += m->m_len; 243a048affbSAndre Oppermann 244a048affbSAndre Oppermann /* m_tags may only be attached to first mbuf in chain. */ 245a048affbSAndre Oppermann if (m != m0 && m->m_flags & M_PKTHDR && 246a048affbSAndre Oppermann !SLIST_EMPTY(&m->m_pkthdr.tags)) { 247a048affbSAndre Oppermann if (sanitize) { 248a048affbSAndre Oppermann m_tag_delete_chain(m, NULL); 249a048affbSAndre Oppermann /* put in 0xDEADC0DE perhaps? */ 250fdcc028dSAndre Oppermann } else 251a048affbSAndre Oppermann M_SANITY_ACTION("m_tags on in-chain mbuf"); 252a048affbSAndre Oppermann } 253a048affbSAndre Oppermann 254a048affbSAndre Oppermann /* M_PKTHDR may only be set on first mbuf in chain */ 255a048affbSAndre Oppermann if (m != m0 && m->m_flags & M_PKTHDR) { 256a048affbSAndre Oppermann if (sanitize) { 257a048affbSAndre Oppermann bzero(&m->m_pkthdr, sizeof(m->m_pkthdr)); 258a048affbSAndre Oppermann m->m_flags &= ~M_PKTHDR; 259a048affbSAndre Oppermann /* put in 0xDEADCODE and leave hdr flag in */ 260a048affbSAndre Oppermann } else 261a048affbSAndre Oppermann M_SANITY_ACTION("M_PKTHDR on in-chain mbuf"); 262a048affbSAndre Oppermann } 263a048affbSAndre Oppermann } 264fdcc028dSAndre Oppermann m = m0; 265fdcc028dSAndre Oppermann if (pktlen && pktlen != m->m_pkthdr.len) { 266a048affbSAndre Oppermann if (sanitize) 267fdcc028dSAndre Oppermann m->m_pkthdr.len = 0; 268a048affbSAndre Oppermann else 269a048affbSAndre Oppermann M_SANITY_ACTION("m_pkthdr.len != mbuf chain length"); 270a048affbSAndre Oppermann } 271a048affbSAndre Oppermann return 1; 272fdcc028dSAndre Oppermann 273fdcc028dSAndre Oppermann #undef M_SANITY_ACTION 274a048affbSAndre Oppermann } 275a048affbSAndre Oppermann 276*5e4bc63bSGleb Smirnoff /* 277*5e4bc63bSGleb Smirnoff * Non-inlined part of m_init(). 278*5e4bc63bSGleb Smirnoff */ 279*5e4bc63bSGleb Smirnoff int 280*5e4bc63bSGleb Smirnoff m_pkthdr_init(struct mbuf *m, int how) 281*5e4bc63bSGleb Smirnoff { 282*5e4bc63bSGleb Smirnoff #ifdef MAC 283*5e4bc63bSGleb Smirnoff int error; 284*5e4bc63bSGleb Smirnoff #endif 285*5e4bc63bSGleb Smirnoff m->m_data = m->m_pktdat; 286*5e4bc63bSGleb Smirnoff bzero(&m->m_pkthdr, sizeof(m->m_pkthdr)); 287*5e4bc63bSGleb Smirnoff #ifdef MAC 288*5e4bc63bSGleb Smirnoff /* If the label init fails, fail the alloc */ 289*5e4bc63bSGleb Smirnoff error = mac_mbuf_init(m, how); 290*5e4bc63bSGleb Smirnoff if (error) 291*5e4bc63bSGleb Smirnoff return (error); 292*5e4bc63bSGleb Smirnoff #endif 293*5e4bc63bSGleb Smirnoff 294*5e4bc63bSGleb Smirnoff return (0); 295*5e4bc63bSGleb Smirnoff } 296a048affbSAndre Oppermann 297a048affbSAndre Oppermann /* 2989967cafcSSam Leffler * "Move" mbuf pkthdr from "from" to "to". 299e37b1fcdSRobert Watson * "from" must have M_PKTHDR set, and "to" must be empty. 300e37b1fcdSRobert Watson */ 301e37b1fcdSRobert Watson void 3029967cafcSSam Leffler m_move_pkthdr(struct mbuf *to, struct mbuf *from) 303e37b1fcdSRobert Watson { 304e37b1fcdSRobert Watson 305e37b1fcdSRobert Watson #if 0 3069967cafcSSam Leffler /* see below for why these are not enabled */ 307fe584538SDag-Erling Smørgrav M_ASSERTPKTHDR(to); 308225bff6fSRobert Watson /* Note: with MAC, this may not be a good assertion. */ 3099967cafcSSam Leffler KASSERT(SLIST_EMPTY(&to->m_pkthdr.tags), 3109967cafcSSam Leffler ("m_move_pkthdr: to has tags")); 311e37b1fcdSRobert Watson #endif 312e32a5b94SRobert Watson #ifdef MAC 313225bff6fSRobert Watson /* 314225bff6fSRobert Watson * XXXMAC: It could be this should also occur for non-MAC? 315225bff6fSRobert Watson */ 316e32a5b94SRobert Watson if (to->m_flags & M_PKTHDR) 317225bff6fSRobert Watson m_tag_delete_chain(to, NULL); 318e32a5b94SRobert Watson #endif 319a4e71429SSam Leffler to->m_flags = (from->m_flags & M_COPYFLAGS) | (to->m_flags & M_EXT); 320a4e71429SSam Leffler if ((to->m_flags & M_EXT) == 0) 3219967cafcSSam Leffler to->m_data = to->m_pktdat; 3229967cafcSSam Leffler to->m_pkthdr = from->m_pkthdr; /* especially tags */ 3239967cafcSSam Leffler SLIST_INIT(&from->m_pkthdr.tags); /* purge tags from src */ 3249967cafcSSam Leffler from->m_flags &= ~M_PKTHDR; 3259967cafcSSam Leffler } 3269967cafcSSam Leffler 3279967cafcSSam Leffler /* 3289967cafcSSam Leffler * Duplicate "from"'s mbuf pkthdr in "to". 3299967cafcSSam Leffler * "from" must have M_PKTHDR set, and "to" must be empty. 3309967cafcSSam Leffler * In particular, this does a deep copy of the packet tags. 3319967cafcSSam Leffler */ 3329967cafcSSam Leffler int 3330cbefd30SAlexander V. Chernikov m_dup_pkthdr(struct mbuf *to, const struct mbuf *from, int how) 3349967cafcSSam Leffler { 3359967cafcSSam Leffler 3369967cafcSSam Leffler #if 0 3379967cafcSSam Leffler /* 3389967cafcSSam Leffler * The mbuf allocator only initializes the pkthdr 339c95be8b5SGleb Smirnoff * when the mbuf is allocated with m_gethdr(). Many users 340c95be8b5SGleb Smirnoff * (e.g. m_copy*, m_prepend) use m_get() and then 3419967cafcSSam Leffler * smash the pkthdr as needed causing these 3429967cafcSSam Leffler * assertions to trip. For now just disable them. 3439967cafcSSam Leffler */ 344fe584538SDag-Erling Smørgrav M_ASSERTPKTHDR(to); 345225bff6fSRobert Watson /* Note: with MAC, this may not be a good assertion. */ 3469967cafcSSam Leffler KASSERT(SLIST_EMPTY(&to->m_pkthdr.tags), ("m_dup_pkthdr: to has tags")); 3479967cafcSSam Leffler #endif 348063d8114SAlfred Perlstein MBUF_CHECKSLEEP(how); 3499967cafcSSam Leffler #ifdef MAC 3509967cafcSSam Leffler if (to->m_flags & M_PKTHDR) 351225bff6fSRobert Watson m_tag_delete_chain(to, NULL); 3529967cafcSSam Leffler #endif 353df8c7fc9SMike Silbersack to->m_flags = (from->m_flags & M_COPYFLAGS) | (to->m_flags & M_EXT); 354df8c7fc9SMike Silbersack if ((to->m_flags & M_EXT) == 0) 3559967cafcSSam Leffler to->m_data = to->m_pktdat; 356e37b1fcdSRobert Watson to->m_pkthdr = from->m_pkthdr; 3579967cafcSSam Leffler SLIST_INIT(&to->m_pkthdr.tags); 35894985f74SGleb Smirnoff return (m_tag_copy_chain(to, from, how)); 359e37b1fcdSRobert Watson } 360e37b1fcdSRobert Watson 361e37b1fcdSRobert Watson /* 362df8bae1dSRodney W. Grimes * Lesser-used path for M_PREPEND: 363df8bae1dSRodney W. Grimes * allocate new mbuf to prepend to chain, 364df8bae1dSRodney W. Grimes * copy junk along. 365df8bae1dSRodney W. Grimes */ 366df8bae1dSRodney W. Grimes struct mbuf * 367122a814aSBosko Milekic m_prepend(struct mbuf *m, int len, int how) 368df8bae1dSRodney W. Grimes { 369df8bae1dSRodney W. Grimes struct mbuf *mn; 370df8bae1dSRodney W. Grimes 371f8bf8e39SMike Silbersack if (m->m_flags & M_PKTHDR) 372c95be8b5SGleb Smirnoff mn = m_gethdr(how, m->m_type); 373f8bf8e39SMike Silbersack else 374c95be8b5SGleb Smirnoff mn = m_get(how, m->m_type); 375122a814aSBosko Milekic if (mn == NULL) { 376df8bae1dSRodney W. Grimes m_freem(m); 377122a814aSBosko Milekic return (NULL); 378df8bae1dSRodney W. Grimes } 379225bff6fSRobert Watson if (m->m_flags & M_PKTHDR) 380c95be8b5SGleb Smirnoff m_move_pkthdr(mn, m); 381df8bae1dSRodney W. Grimes mn->m_next = m; 382df8bae1dSRodney W. Grimes m = mn; 383ed6a66caSRobert Watson if (len < M_SIZE(m)) 3845288989fSRandall Stewart M_ALIGN(m, len); 385df8bae1dSRodney W. Grimes m->m_len = len; 386df8bae1dSRodney W. Grimes return (m); 387df8bae1dSRodney W. Grimes } 388df8bae1dSRodney W. Grimes 389df8bae1dSRodney W. Grimes /* 390df8bae1dSRodney W. Grimes * Make a copy of an mbuf chain starting "off0" bytes from the beginning, 391df8bae1dSRodney W. Grimes * continuing for "len" bytes. If len is M_COPYALL, copy to end of mbuf. 392eb1b1807SGleb Smirnoff * The wait parameter is a choice of M_WAITOK/M_NOWAIT from caller. 3931c38f2eaSArchie Cobbs * Note that the copy is read-only, because clusters are not copied, 3941c38f2eaSArchie Cobbs * only their reference counts are incremented. 395df8bae1dSRodney W. Grimes */ 396df8bae1dSRodney W. Grimes struct mbuf * 3970cbefd30SAlexander V. Chernikov m_copym(const struct mbuf *m, int off0, int len, int wait) 398df8bae1dSRodney W. Grimes { 399122a814aSBosko Milekic struct mbuf *n, **np; 400122a814aSBosko Milekic int off = off0; 401df8bae1dSRodney W. Grimes struct mbuf *top; 402df8bae1dSRodney W. Grimes int copyhdr = 0; 403df8bae1dSRodney W. Grimes 404e0a653ddSAlfred Perlstein KASSERT(off >= 0, ("m_copym, negative off %d", off)); 405e0a653ddSAlfred Perlstein KASSERT(len >= 0, ("m_copym, negative len %d", len)); 406063d8114SAlfred Perlstein MBUF_CHECKSLEEP(wait); 407df8bae1dSRodney W. Grimes if (off == 0 && m->m_flags & M_PKTHDR) 408df8bae1dSRodney W. Grimes copyhdr = 1; 409df8bae1dSRodney W. Grimes while (off > 0) { 410e0a653ddSAlfred Perlstein KASSERT(m != NULL, ("m_copym, offset > size of mbuf chain")); 411df8bae1dSRodney W. Grimes if (off < m->m_len) 412df8bae1dSRodney W. Grimes break; 413df8bae1dSRodney W. Grimes off -= m->m_len; 414df8bae1dSRodney W. Grimes m = m->m_next; 415df8bae1dSRodney W. Grimes } 416df8bae1dSRodney W. Grimes np = ⊤ 417df8bae1dSRodney W. Grimes top = 0; 418df8bae1dSRodney W. Grimes while (len > 0) { 419122a814aSBosko Milekic if (m == NULL) { 420e0a653ddSAlfred Perlstein KASSERT(len == M_COPYALL, 421e0a653ddSAlfred Perlstein ("m_copym, length > size of mbuf chain")); 422df8bae1dSRodney W. Grimes break; 423df8bae1dSRodney W. Grimes } 424f8bf8e39SMike Silbersack if (copyhdr) 425c95be8b5SGleb Smirnoff n = m_gethdr(wait, m->m_type); 426f8bf8e39SMike Silbersack else 427c95be8b5SGleb Smirnoff n = m_get(wait, m->m_type); 428df8bae1dSRodney W. Grimes *np = n; 429122a814aSBosko Milekic if (n == NULL) 430df8bae1dSRodney W. Grimes goto nospace; 431df8bae1dSRodney W. Grimes if (copyhdr) { 4329967cafcSSam Leffler if (!m_dup_pkthdr(n, m, wait)) 4339967cafcSSam Leffler goto nospace; 434df8bae1dSRodney W. Grimes if (len == M_COPYALL) 435df8bae1dSRodney W. Grimes n->m_pkthdr.len -= off0; 436df8bae1dSRodney W. Grimes else 437df8bae1dSRodney W. Grimes n->m_pkthdr.len = len; 438df8bae1dSRodney W. Grimes copyhdr = 0; 439df8bae1dSRodney W. Grimes } 440df8bae1dSRodney W. Grimes n->m_len = min(len, m->m_len - off); 441df8bae1dSRodney W. Grimes if (m->m_flags & M_EXT) { 442df8bae1dSRodney W. Grimes n->m_data = m->m_data + off; 44356a4e45aSAndre Oppermann mb_dupcl(n, m); 444df8bae1dSRodney W. Grimes } else 445df8bae1dSRodney W. Grimes bcopy(mtod(m, caddr_t)+off, mtod(n, caddr_t), 446bd395ae8SBosko Milekic (u_int)n->m_len); 447df8bae1dSRodney W. Grimes if (len != M_COPYALL) 448df8bae1dSRodney W. Grimes len -= n->m_len; 449df8bae1dSRodney W. Grimes off = 0; 450df8bae1dSRodney W. Grimes m = m->m_next; 451df8bae1dSRodney W. Grimes np = &n->m_next; 452df8bae1dSRodney W. Grimes } 45308442f8aSBosko Milekic 454df8bae1dSRodney W. Grimes return (top); 455df8bae1dSRodney W. Grimes nospace: 456df8bae1dSRodney W. Grimes m_freem(top); 457122a814aSBosko Milekic return (NULL); 458df8bae1dSRodney W. Grimes } 459df8bae1dSRodney W. Grimes 460df8bae1dSRodney W. Grimes /* 4616a06dea0SGarrett Wollman * Copy an entire packet, including header (which must be present). 4626a06dea0SGarrett Wollman * An optimization of the common case `m_copym(m, 0, M_COPYALL, how)'. 4631c38f2eaSArchie Cobbs * Note that the copy is read-only, because clusters are not copied, 4641c38f2eaSArchie Cobbs * only their reference counts are incremented. 4655fe86675SLuigi Rizzo * Preserve alignment of the first mbuf so if the creator has left 4665fe86675SLuigi Rizzo * some room at the beginning (e.g. for inserting protocol headers) 4675fe86675SLuigi Rizzo * the copies still have the room available. 4686a06dea0SGarrett Wollman */ 4696a06dea0SGarrett Wollman struct mbuf * 470122a814aSBosko Milekic m_copypacket(struct mbuf *m, int how) 4716a06dea0SGarrett Wollman { 4726a06dea0SGarrett Wollman struct mbuf *top, *n, *o; 4736a06dea0SGarrett Wollman 474063d8114SAlfred Perlstein MBUF_CHECKSLEEP(how); 475c95be8b5SGleb Smirnoff n = m_get(how, m->m_type); 4766a06dea0SGarrett Wollman top = n; 477122a814aSBosko Milekic if (n == NULL) 4786a06dea0SGarrett Wollman goto nospace; 4796a06dea0SGarrett Wollman 4809967cafcSSam Leffler if (!m_dup_pkthdr(n, m, how)) 4819967cafcSSam Leffler goto nospace; 4826a06dea0SGarrett Wollman n->m_len = m->m_len; 4836a06dea0SGarrett Wollman if (m->m_flags & M_EXT) { 4846a06dea0SGarrett Wollman n->m_data = m->m_data; 48556a4e45aSAndre Oppermann mb_dupcl(n, m); 4866a06dea0SGarrett Wollman } else { 4875fe86675SLuigi Rizzo n->m_data = n->m_pktdat + (m->m_data - m->m_pktdat ); 4886a06dea0SGarrett Wollman bcopy(mtod(m, char *), mtod(n, char *), n->m_len); 4896a06dea0SGarrett Wollman } 4906a06dea0SGarrett Wollman 4916a06dea0SGarrett Wollman m = m->m_next; 4926a06dea0SGarrett Wollman while (m) { 493c95be8b5SGleb Smirnoff o = m_get(how, m->m_type); 494122a814aSBosko Milekic if (o == NULL) 4956a06dea0SGarrett Wollman goto nospace; 4966a06dea0SGarrett Wollman 4976a06dea0SGarrett Wollman n->m_next = o; 4986a06dea0SGarrett Wollman n = n->m_next; 4996a06dea0SGarrett Wollman 5006a06dea0SGarrett Wollman n->m_len = m->m_len; 5016a06dea0SGarrett Wollman if (m->m_flags & M_EXT) { 5026a06dea0SGarrett Wollman n->m_data = m->m_data; 50356a4e45aSAndre Oppermann mb_dupcl(n, m); 5046a06dea0SGarrett Wollman } else { 5056a06dea0SGarrett Wollman bcopy(mtod(m, char *), mtod(n, char *), n->m_len); 5066a06dea0SGarrett Wollman } 5076a06dea0SGarrett Wollman 5086a06dea0SGarrett Wollman m = m->m_next; 5096a06dea0SGarrett Wollman } 5106a06dea0SGarrett Wollman return top; 5116a06dea0SGarrett Wollman nospace: 5126a06dea0SGarrett Wollman m_freem(top); 513122a814aSBosko Milekic return (NULL); 5146a06dea0SGarrett Wollman } 5156a06dea0SGarrett Wollman 5166a06dea0SGarrett Wollman /* 517df8bae1dSRodney W. Grimes * Copy data from an mbuf chain starting "off" bytes from the beginning, 518df8bae1dSRodney W. Grimes * continuing for "len" bytes, into the indicated buffer. 519df8bae1dSRodney W. Grimes */ 52026f9a767SRodney W. Grimes void 521a8cfc0eeSJulian Elischer m_copydata(const struct mbuf *m, int off, int len, caddr_t cp) 522df8bae1dSRodney W. Grimes { 523bd395ae8SBosko Milekic u_int count; 524df8bae1dSRodney W. Grimes 525e0a653ddSAlfred Perlstein KASSERT(off >= 0, ("m_copydata, negative off %d", off)); 526e0a653ddSAlfred Perlstein KASSERT(len >= 0, ("m_copydata, negative len %d", len)); 527df8bae1dSRodney W. Grimes while (off > 0) { 528e0a653ddSAlfred Perlstein KASSERT(m != NULL, ("m_copydata, offset > size of mbuf chain")); 529df8bae1dSRodney W. Grimes if (off < m->m_len) 530df8bae1dSRodney W. Grimes break; 531df8bae1dSRodney W. Grimes off -= m->m_len; 532df8bae1dSRodney W. Grimes m = m->m_next; 533df8bae1dSRodney W. Grimes } 534df8bae1dSRodney W. Grimes while (len > 0) { 535e0a653ddSAlfred Perlstein KASSERT(m != NULL, ("m_copydata, length > size of mbuf chain")); 536df8bae1dSRodney W. Grimes count = min(m->m_len - off, len); 537df8bae1dSRodney W. Grimes bcopy(mtod(m, caddr_t) + off, cp, count); 538df8bae1dSRodney W. Grimes len -= count; 539df8bae1dSRodney W. Grimes cp += count; 540df8bae1dSRodney W. Grimes off = 0; 541df8bae1dSRodney W. Grimes m = m->m_next; 542df8bae1dSRodney W. Grimes } 543df8bae1dSRodney W. Grimes } 544df8bae1dSRodney W. Grimes 545df8bae1dSRodney W. Grimes /* 5461c38f2eaSArchie Cobbs * Copy a packet header mbuf chain into a completely new chain, including 5471c38f2eaSArchie Cobbs * copying any mbuf clusters. Use this instead of m_copypacket() when 5481c38f2eaSArchie Cobbs * you need a writable copy of an mbuf chain. 5491c38f2eaSArchie Cobbs */ 5501c38f2eaSArchie Cobbs struct mbuf * 5510cbefd30SAlexander V. Chernikov m_dup(const struct mbuf *m, int how) 5521c38f2eaSArchie Cobbs { 5531c38f2eaSArchie Cobbs struct mbuf **p, *top = NULL; 5541c38f2eaSArchie Cobbs int remain, moff, nsize; 5551c38f2eaSArchie Cobbs 556063d8114SAlfred Perlstein MBUF_CHECKSLEEP(how); 5571c38f2eaSArchie Cobbs /* Sanity check */ 5581c38f2eaSArchie Cobbs if (m == NULL) 559122a814aSBosko Milekic return (NULL); 560fe584538SDag-Erling Smørgrav M_ASSERTPKTHDR(m); 5611c38f2eaSArchie Cobbs 5621c38f2eaSArchie Cobbs /* While there's more data, get a new mbuf, tack it on, and fill it */ 5631c38f2eaSArchie Cobbs remain = m->m_pkthdr.len; 5641c38f2eaSArchie Cobbs moff = 0; 5651c38f2eaSArchie Cobbs p = ⊤ 5661c38f2eaSArchie Cobbs while (remain > 0 || top == NULL) { /* allow m->m_pkthdr.len == 0 */ 5671c38f2eaSArchie Cobbs struct mbuf *n; 5681c38f2eaSArchie Cobbs 5691c38f2eaSArchie Cobbs /* Get the next new mbuf */ 570099a0e58SBosko Milekic if (remain >= MINCLSIZE) { 571099a0e58SBosko Milekic n = m_getcl(how, m->m_type, 0); 572099a0e58SBosko Milekic nsize = MCLBYTES; 573099a0e58SBosko Milekic } else { 574099a0e58SBosko Milekic n = m_get(how, m->m_type); 575099a0e58SBosko Milekic nsize = MLEN; 576099a0e58SBosko Milekic } 5771c38f2eaSArchie Cobbs if (n == NULL) 5781c38f2eaSArchie Cobbs goto nospace; 579099a0e58SBosko Milekic 580099a0e58SBosko Milekic if (top == NULL) { /* First one, must be PKTHDR */ 581099a0e58SBosko Milekic if (!m_dup_pkthdr(n, m, how)) { 582099a0e58SBosko Milekic m_free(n); 5831c38f2eaSArchie Cobbs goto nospace; 5841c38f2eaSArchie Cobbs } 58563e6f390SEd Maste if ((n->m_flags & M_EXT) == 0) 586099a0e58SBosko Milekic nsize = MHLEN; 587089bb672SAndrey V. Elsukov n->m_flags &= ~M_RDONLY; 5881c38f2eaSArchie Cobbs } 5891c38f2eaSArchie Cobbs n->m_len = 0; 5901c38f2eaSArchie Cobbs 5911c38f2eaSArchie Cobbs /* Link it into the new chain */ 5921c38f2eaSArchie Cobbs *p = n; 5931c38f2eaSArchie Cobbs p = &n->m_next; 5941c38f2eaSArchie Cobbs 5951c38f2eaSArchie Cobbs /* Copy data from original mbuf(s) into new mbuf */ 5961c38f2eaSArchie Cobbs while (n->m_len < nsize && m != NULL) { 5971c38f2eaSArchie Cobbs int chunk = min(nsize - n->m_len, m->m_len - moff); 5981c38f2eaSArchie Cobbs 5991c38f2eaSArchie Cobbs bcopy(m->m_data + moff, n->m_data + n->m_len, chunk); 6001c38f2eaSArchie Cobbs moff += chunk; 6011c38f2eaSArchie Cobbs n->m_len += chunk; 6021c38f2eaSArchie Cobbs remain -= chunk; 6031c38f2eaSArchie Cobbs if (moff == m->m_len) { 6041c38f2eaSArchie Cobbs m = m->m_next; 6051c38f2eaSArchie Cobbs moff = 0; 6061c38f2eaSArchie Cobbs } 6071c38f2eaSArchie Cobbs } 6081c38f2eaSArchie Cobbs 6091c38f2eaSArchie Cobbs /* Check correct total mbuf length */ 6101c38f2eaSArchie Cobbs KASSERT((remain > 0 && m != NULL) || (remain == 0 && m == NULL), 611a48740b6SDavid E. O'Brien ("%s: bogus m_pkthdr.len", __func__)); 6121c38f2eaSArchie Cobbs } 6131c38f2eaSArchie Cobbs return (top); 6141c38f2eaSArchie Cobbs 6151c38f2eaSArchie Cobbs nospace: 6161c38f2eaSArchie Cobbs m_freem(top); 617122a814aSBosko Milekic return (NULL); 6181c38f2eaSArchie Cobbs } 6191c38f2eaSArchie Cobbs 6201c38f2eaSArchie Cobbs /* 621df8bae1dSRodney W. Grimes * Concatenate mbuf chain n to m. 622df8bae1dSRodney W. Grimes * Both chains must be of the same type (e.g. MT_DATA). 623df8bae1dSRodney W. Grimes * Any m_pkthdr is not updated. 624df8bae1dSRodney W. Grimes */ 62526f9a767SRodney W. Grimes void 626122a814aSBosko Milekic m_cat(struct mbuf *m, struct mbuf *n) 627df8bae1dSRodney W. Grimes { 628df8bae1dSRodney W. Grimes while (m->m_next) 629df8bae1dSRodney W. Grimes m = m->m_next; 630df8bae1dSRodney W. Grimes while (n) { 63114d7c5b1SAndre Oppermann if (!M_WRITABLE(m) || 63214d7c5b1SAndre Oppermann M_TRAILINGSPACE(m) < n->m_len) { 633df8bae1dSRodney W. Grimes /* just join the two chains */ 634df8bae1dSRodney W. Grimes m->m_next = n; 635df8bae1dSRodney W. Grimes return; 636df8bae1dSRodney W. Grimes } 637df8bae1dSRodney W. Grimes /* splat the data from one into the other */ 638df8bae1dSRodney W. Grimes bcopy(mtod(n, caddr_t), mtod(m, caddr_t) + m->m_len, 639df8bae1dSRodney W. Grimes (u_int)n->m_len); 640df8bae1dSRodney W. Grimes m->m_len += n->m_len; 641df8bae1dSRodney W. Grimes n = m_free(n); 642df8bae1dSRodney W. Grimes } 643df8bae1dSRodney W. Grimes } 644df8bae1dSRodney W. Grimes 6451967edbaSGleb Smirnoff /* 6461967edbaSGleb Smirnoff * Concatenate two pkthdr mbuf chains. 6471967edbaSGleb Smirnoff */ 6481967edbaSGleb Smirnoff void 6491967edbaSGleb Smirnoff m_catpkt(struct mbuf *m, struct mbuf *n) 6501967edbaSGleb Smirnoff { 6511967edbaSGleb Smirnoff 6521967edbaSGleb Smirnoff M_ASSERTPKTHDR(m); 6531967edbaSGleb Smirnoff M_ASSERTPKTHDR(n); 6541967edbaSGleb Smirnoff 6551967edbaSGleb Smirnoff m->m_pkthdr.len += n->m_pkthdr.len; 656651e4e6aSGleb Smirnoff m_demote(n, 1, 0); 6571967edbaSGleb Smirnoff 6581967edbaSGleb Smirnoff m_cat(m, n); 6591967edbaSGleb Smirnoff } 6601967edbaSGleb Smirnoff 66126f9a767SRodney W. Grimes void 662122a814aSBosko Milekic m_adj(struct mbuf *mp, int req_len) 663df8bae1dSRodney W. Grimes { 664122a814aSBosko Milekic int len = req_len; 665122a814aSBosko Milekic struct mbuf *m; 666122a814aSBosko Milekic int count; 667df8bae1dSRodney W. Grimes 668df8bae1dSRodney W. Grimes if ((m = mp) == NULL) 669df8bae1dSRodney W. Grimes return; 670df8bae1dSRodney W. Grimes if (len >= 0) { 671df8bae1dSRodney W. Grimes /* 672df8bae1dSRodney W. Grimes * Trim from head. 673df8bae1dSRodney W. Grimes */ 674df8bae1dSRodney W. Grimes while (m != NULL && len > 0) { 675df8bae1dSRodney W. Grimes if (m->m_len <= len) { 676df8bae1dSRodney W. Grimes len -= m->m_len; 677df8bae1dSRodney W. Grimes m->m_len = 0; 678df8bae1dSRodney W. Grimes m = m->m_next; 679df8bae1dSRodney W. Grimes } else { 680df8bae1dSRodney W. Grimes m->m_len -= len; 681df8bae1dSRodney W. Grimes m->m_data += len; 682df8bae1dSRodney W. Grimes len = 0; 683df8bae1dSRodney W. Grimes } 684df8bae1dSRodney W. Grimes } 685df8bae1dSRodney W. Grimes if (mp->m_flags & M_PKTHDR) 686a83baab6SMarko Zec mp->m_pkthdr.len -= (req_len - len); 687df8bae1dSRodney W. Grimes } else { 688df8bae1dSRodney W. Grimes /* 689df8bae1dSRodney W. Grimes * Trim from tail. Scan the mbuf chain, 690df8bae1dSRodney W. Grimes * calculating its length and finding the last mbuf. 691df8bae1dSRodney W. Grimes * If the adjustment only affects this mbuf, then just 692df8bae1dSRodney W. Grimes * adjust and return. Otherwise, rescan and truncate 693df8bae1dSRodney W. Grimes * after the remaining size. 694df8bae1dSRodney W. Grimes */ 695df8bae1dSRodney W. Grimes len = -len; 696df8bae1dSRodney W. Grimes count = 0; 697df8bae1dSRodney W. Grimes for (;;) { 698df8bae1dSRodney W. Grimes count += m->m_len; 699df8bae1dSRodney W. Grimes if (m->m_next == (struct mbuf *)0) 700df8bae1dSRodney W. Grimes break; 701df8bae1dSRodney W. Grimes m = m->m_next; 702df8bae1dSRodney W. Grimes } 703df8bae1dSRodney W. Grimes if (m->m_len >= len) { 704df8bae1dSRodney W. Grimes m->m_len -= len; 705df8bae1dSRodney W. Grimes if (mp->m_flags & M_PKTHDR) 706df8bae1dSRodney W. Grimes mp->m_pkthdr.len -= len; 707df8bae1dSRodney W. Grimes return; 708df8bae1dSRodney W. Grimes } 709df8bae1dSRodney W. Grimes count -= len; 710df8bae1dSRodney W. Grimes if (count < 0) 711df8bae1dSRodney W. Grimes count = 0; 712df8bae1dSRodney W. Grimes /* 713df8bae1dSRodney W. Grimes * Correct length for chain is "count". 714df8bae1dSRodney W. Grimes * Find the mbuf with last data, adjust its length, 715df8bae1dSRodney W. Grimes * and toss data from remaining mbufs on chain. 716df8bae1dSRodney W. Grimes */ 717df8bae1dSRodney W. Grimes m = mp; 718df8bae1dSRodney W. Grimes if (m->m_flags & M_PKTHDR) 719df8bae1dSRodney W. Grimes m->m_pkthdr.len = count; 720df8bae1dSRodney W. Grimes for (; m; m = m->m_next) { 721df8bae1dSRodney W. Grimes if (m->m_len >= count) { 722df8bae1dSRodney W. Grimes m->m_len = count; 72359d8b310SSam Leffler if (m->m_next != NULL) { 72459d8b310SSam Leffler m_freem(m->m_next); 72559d8b310SSam Leffler m->m_next = NULL; 72659d8b310SSam Leffler } 727df8bae1dSRodney W. Grimes break; 728df8bae1dSRodney W. Grimes } 729df8bae1dSRodney W. Grimes count -= m->m_len; 730df8bae1dSRodney W. Grimes } 731df8bae1dSRodney W. Grimes } 732df8bae1dSRodney W. Grimes } 733df8bae1dSRodney W. Grimes 734df8bae1dSRodney W. Grimes /* 735df8bae1dSRodney W. Grimes * Rearange an mbuf chain so that len bytes are contiguous 736a2c36a02SKevin Lo * and in the data area of an mbuf (so that mtod will work 737a2c36a02SKevin Lo * for a structure of size len). Returns the resulting 738df8bae1dSRodney W. Grimes * mbuf chain on success, frees it and returns null on failure. 739df8bae1dSRodney W. Grimes * If there is room, it will add up to max_protohdr-len extra bytes to the 740df8bae1dSRodney W. Grimes * contiguous region in an attempt to avoid being called next time. 741df8bae1dSRodney W. Grimes */ 742df8bae1dSRodney W. Grimes struct mbuf * 743122a814aSBosko Milekic m_pullup(struct mbuf *n, int len) 744df8bae1dSRodney W. Grimes { 745122a814aSBosko Milekic struct mbuf *m; 746122a814aSBosko Milekic int count; 747df8bae1dSRodney W. Grimes int space; 748df8bae1dSRodney W. Grimes 749df8bae1dSRodney W. Grimes /* 750df8bae1dSRodney W. Grimes * If first mbuf has no cluster, and has room for len bytes 751df8bae1dSRodney W. Grimes * without shifting current data, pullup into it, 752df8bae1dSRodney W. Grimes * otherwise allocate a new mbuf to prepend to the chain. 753df8bae1dSRodney W. Grimes */ 754df8bae1dSRodney W. Grimes if ((n->m_flags & M_EXT) == 0 && 755df8bae1dSRodney W. Grimes n->m_data + len < &n->m_dat[MLEN] && n->m_next) { 756df8bae1dSRodney W. Grimes if (n->m_len >= len) 757df8bae1dSRodney W. Grimes return (n); 758df8bae1dSRodney W. Grimes m = n; 759df8bae1dSRodney W. Grimes n = n->m_next; 760df8bae1dSRodney W. Grimes len -= m->m_len; 761df8bae1dSRodney W. Grimes } else { 762df8bae1dSRodney W. Grimes if (len > MHLEN) 763df8bae1dSRodney W. Grimes goto bad; 764c95be8b5SGleb Smirnoff m = m_get(M_NOWAIT, n->m_type); 765122a814aSBosko Milekic if (m == NULL) 766df8bae1dSRodney W. Grimes goto bad; 7679967cafcSSam Leffler if (n->m_flags & M_PKTHDR) 768c95be8b5SGleb Smirnoff m_move_pkthdr(m, n); 769df8bae1dSRodney W. Grimes } 770df8bae1dSRodney W. Grimes space = &m->m_dat[MLEN] - (m->m_data + m->m_len); 771df8bae1dSRodney W. Grimes do { 772df8bae1dSRodney W. Grimes count = min(min(max(len, max_protohdr), space), n->m_len); 773df8bae1dSRodney W. Grimes bcopy(mtod(n, caddr_t), mtod(m, caddr_t) + m->m_len, 774bd395ae8SBosko Milekic (u_int)count); 775df8bae1dSRodney W. Grimes len -= count; 776df8bae1dSRodney W. Grimes m->m_len += count; 777df8bae1dSRodney W. Grimes n->m_len -= count; 778df8bae1dSRodney W. Grimes space -= count; 779df8bae1dSRodney W. Grimes if (n->m_len) 780df8bae1dSRodney W. Grimes n->m_data += count; 781df8bae1dSRodney W. Grimes else 782df8bae1dSRodney W. Grimes n = m_free(n); 783df8bae1dSRodney W. Grimes } while (len > 0 && n); 784df8bae1dSRodney W. Grimes if (len > 0) { 785df8bae1dSRodney W. Grimes (void) m_free(m); 786df8bae1dSRodney W. Grimes goto bad; 787df8bae1dSRodney W. Grimes } 788df8bae1dSRodney W. Grimes m->m_next = n; 789df8bae1dSRodney W. Grimes return (m); 790df8bae1dSRodney W. Grimes bad: 791df8bae1dSRodney W. Grimes m_freem(n); 792122a814aSBosko Milekic return (NULL); 793df8bae1dSRodney W. Grimes } 794df8bae1dSRodney W. Grimes 795df8bae1dSRodney W. Grimes /* 7967ac139a9SJohn-Mark Gurney * Like m_pullup(), except a new mbuf is always allocated, and we allow 7977ac139a9SJohn-Mark Gurney * the amount of empty space before the data in the new mbuf to be specified 7987ac139a9SJohn-Mark Gurney * (in the event that the caller expects to prepend later). 7997ac139a9SJohn-Mark Gurney */ 8007ac139a9SJohn-Mark Gurney struct mbuf * 8017ac139a9SJohn-Mark Gurney m_copyup(struct mbuf *n, int len, int dstoff) 8027ac139a9SJohn-Mark Gurney { 8037ac139a9SJohn-Mark Gurney struct mbuf *m; 8047ac139a9SJohn-Mark Gurney int count, space; 8057ac139a9SJohn-Mark Gurney 8067ac139a9SJohn-Mark Gurney if (len > (MHLEN - dstoff)) 8077ac139a9SJohn-Mark Gurney goto bad; 808c95be8b5SGleb Smirnoff m = m_get(M_NOWAIT, n->m_type); 8097ac139a9SJohn-Mark Gurney if (m == NULL) 8107ac139a9SJohn-Mark Gurney goto bad; 8117ac139a9SJohn-Mark Gurney if (n->m_flags & M_PKTHDR) 812c95be8b5SGleb Smirnoff m_move_pkthdr(m, n); 8137ac139a9SJohn-Mark Gurney m->m_data += dstoff; 8147ac139a9SJohn-Mark Gurney space = &m->m_dat[MLEN] - (m->m_data + m->m_len); 8157ac139a9SJohn-Mark Gurney do { 8167ac139a9SJohn-Mark Gurney count = min(min(max(len, max_protohdr), space), n->m_len); 8177ac139a9SJohn-Mark Gurney memcpy(mtod(m, caddr_t) + m->m_len, mtod(n, caddr_t), 8187ac139a9SJohn-Mark Gurney (unsigned)count); 8197ac139a9SJohn-Mark Gurney len -= count; 8207ac139a9SJohn-Mark Gurney m->m_len += count; 8217ac139a9SJohn-Mark Gurney n->m_len -= count; 8227ac139a9SJohn-Mark Gurney space -= count; 8237ac139a9SJohn-Mark Gurney if (n->m_len) 8247ac139a9SJohn-Mark Gurney n->m_data += count; 8257ac139a9SJohn-Mark Gurney else 8267ac139a9SJohn-Mark Gurney n = m_free(n); 8277ac139a9SJohn-Mark Gurney } while (len > 0 && n); 8287ac139a9SJohn-Mark Gurney if (len > 0) { 8297ac139a9SJohn-Mark Gurney (void) m_free(m); 8307ac139a9SJohn-Mark Gurney goto bad; 8317ac139a9SJohn-Mark Gurney } 8327ac139a9SJohn-Mark Gurney m->m_next = n; 8337ac139a9SJohn-Mark Gurney return (m); 8347ac139a9SJohn-Mark Gurney bad: 8357ac139a9SJohn-Mark Gurney m_freem(n); 8367ac139a9SJohn-Mark Gurney return (NULL); 8377ac139a9SJohn-Mark Gurney } 8387ac139a9SJohn-Mark Gurney 8397ac139a9SJohn-Mark Gurney /* 840df8bae1dSRodney W. Grimes * Partition an mbuf chain in two pieces, returning the tail -- 841df8bae1dSRodney W. Grimes * all but the first len0 bytes. In case of failure, it returns NULL and 842df8bae1dSRodney W. Grimes * attempts to restore the chain to its original state. 84348d183faSArchie Cobbs * 84448d183faSArchie Cobbs * Note that the resulting mbufs might be read-only, because the new 84548d183faSArchie Cobbs * mbuf can end up sharing an mbuf cluster with the original mbuf if 84648d183faSArchie Cobbs * the "breaking point" happens to lie within a cluster mbuf. Use the 84748d183faSArchie Cobbs * M_WRITABLE() macro to check for this case. 848df8bae1dSRodney W. Grimes */ 849df8bae1dSRodney W. Grimes struct mbuf * 850122a814aSBosko Milekic m_split(struct mbuf *m0, int len0, int wait) 851df8bae1dSRodney W. Grimes { 852122a814aSBosko Milekic struct mbuf *m, *n; 853bd395ae8SBosko Milekic u_int len = len0, remain; 854df8bae1dSRodney W. Grimes 855063d8114SAlfred Perlstein MBUF_CHECKSLEEP(wait); 856df8bae1dSRodney W. Grimes for (m = m0; m && len > m->m_len; m = m->m_next) 857df8bae1dSRodney W. Grimes len -= m->m_len; 858122a814aSBosko Milekic if (m == NULL) 859122a814aSBosko Milekic return (NULL); 860df8bae1dSRodney W. Grimes remain = m->m_len - len; 86121f39848SGleb Smirnoff if (m0->m_flags & M_PKTHDR && remain == 0) { 86221f39848SGleb Smirnoff n = m_gethdr(wait, m0->m_type); 86377badb18SGleb Smirnoff if (n == NULL) 86421f39848SGleb Smirnoff return (NULL); 86521f39848SGleb Smirnoff n->m_next = m->m_next; 86621f39848SGleb Smirnoff m->m_next = NULL; 86721f39848SGleb Smirnoff n->m_pkthdr.rcvif = m0->m_pkthdr.rcvif; 86821f39848SGleb Smirnoff n->m_pkthdr.len = m0->m_pkthdr.len - len0; 86921f39848SGleb Smirnoff m0->m_pkthdr.len = len0; 87021f39848SGleb Smirnoff return (n); 87121f39848SGleb Smirnoff } else if (m0->m_flags & M_PKTHDR) { 872c95be8b5SGleb Smirnoff n = m_gethdr(wait, m0->m_type); 873122a814aSBosko Milekic if (n == NULL) 874122a814aSBosko Milekic return (NULL); 875df8bae1dSRodney W. Grimes n->m_pkthdr.rcvif = m0->m_pkthdr.rcvif; 876df8bae1dSRodney W. Grimes n->m_pkthdr.len = m0->m_pkthdr.len - len0; 877df8bae1dSRodney W. Grimes m0->m_pkthdr.len = len0; 878df8bae1dSRodney W. Grimes if (m->m_flags & M_EXT) 879df8bae1dSRodney W. Grimes goto extpacket; 880df8bae1dSRodney W. Grimes if (remain > MHLEN) { 881df8bae1dSRodney W. Grimes /* m can't be the lead packet */ 882ed6a66caSRobert Watson M_ALIGN(n, 0); 883df8bae1dSRodney W. Grimes n->m_next = m_split(m, len, wait); 884122a814aSBosko Milekic if (n->m_next == NULL) { 885df8bae1dSRodney W. Grimes (void) m_free(n); 886122a814aSBosko Milekic return (NULL); 88740376987SJeffrey Hsu } else { 88840376987SJeffrey Hsu n->m_len = 0; 889df8bae1dSRodney W. Grimes return (n); 89040376987SJeffrey Hsu } 891df8bae1dSRodney W. Grimes } else 892ed6a66caSRobert Watson M_ALIGN(n, remain); 893df8bae1dSRodney W. Grimes } else if (remain == 0) { 894df8bae1dSRodney W. Grimes n = m->m_next; 895122a814aSBosko Milekic m->m_next = NULL; 896df8bae1dSRodney W. Grimes return (n); 897df8bae1dSRodney W. Grimes } else { 898c95be8b5SGleb Smirnoff n = m_get(wait, m->m_type); 899122a814aSBosko Milekic if (n == NULL) 900122a814aSBosko Milekic return (NULL); 901df8bae1dSRodney W. Grimes M_ALIGN(n, remain); 902df8bae1dSRodney W. Grimes } 903df8bae1dSRodney W. Grimes extpacket: 904df8bae1dSRodney W. Grimes if (m->m_flags & M_EXT) { 905df8bae1dSRodney W. Grimes n->m_data = m->m_data + len; 90656a4e45aSAndre Oppermann mb_dupcl(n, m); 907df8bae1dSRodney W. Grimes } else { 908df8bae1dSRodney W. Grimes bcopy(mtod(m, caddr_t) + len, mtod(n, caddr_t), remain); 909df8bae1dSRodney W. Grimes } 910df8bae1dSRodney W. Grimes n->m_len = remain; 911df8bae1dSRodney W. Grimes m->m_len = len; 912df8bae1dSRodney W. Grimes n->m_next = m->m_next; 913122a814aSBosko Milekic m->m_next = NULL; 914df8bae1dSRodney W. Grimes return (n); 915df8bae1dSRodney W. Grimes } 916df8bae1dSRodney W. Grimes /* 917df8bae1dSRodney W. Grimes * Routine to copy from device local memory into mbufs. 918f5eece3fSBosko Milekic * Note that `off' argument is offset into first mbuf of target chain from 919f5eece3fSBosko Milekic * which to begin copying the data to. 920df8bae1dSRodney W. Grimes */ 921df8bae1dSRodney W. Grimes struct mbuf * 922f5eece3fSBosko Milekic m_devget(char *buf, int totlen, int off, struct ifnet *ifp, 923122a814aSBosko Milekic void (*copy)(char *from, caddr_t to, u_int len)) 924df8bae1dSRodney W. Grimes { 925122a814aSBosko Milekic struct mbuf *m; 926099a0e58SBosko Milekic struct mbuf *top = NULL, **mp = ⊤ 927f5eece3fSBosko Milekic int len; 928df8bae1dSRodney W. Grimes 929f5eece3fSBosko Milekic if (off < 0 || off > MHLEN) 930f5eece3fSBosko Milekic return (NULL); 931f5eece3fSBosko Milekic 932df8bae1dSRodney W. Grimes while (totlen > 0) { 933099a0e58SBosko Milekic if (top == NULL) { /* First one, must be PKTHDR */ 934f5eece3fSBosko Milekic if (totlen + off >= MINCLSIZE) { 935eb1b1807SGleb Smirnoff m = m_getcl(M_NOWAIT, MT_DATA, M_PKTHDR); 936f5eece3fSBosko Milekic len = MCLBYTES; 937df8bae1dSRodney W. Grimes } else { 938eb1b1807SGleb Smirnoff m = m_gethdr(M_NOWAIT, MT_DATA); 939099a0e58SBosko Milekic len = MHLEN; 940099a0e58SBosko Milekic 941099a0e58SBosko Milekic /* Place initial small packet/header at end of mbuf */ 942099a0e58SBosko Milekic if (m && totlen + off + max_linkhdr <= MLEN) { 943df8bae1dSRodney W. Grimes m->m_data += max_linkhdr; 944f5eece3fSBosko Milekic len -= max_linkhdr; 945df8bae1dSRodney W. Grimes } 946f5eece3fSBosko Milekic } 947099a0e58SBosko Milekic if (m == NULL) 948099a0e58SBosko Milekic return NULL; 949099a0e58SBosko Milekic m->m_pkthdr.rcvif = ifp; 950099a0e58SBosko Milekic m->m_pkthdr.len = totlen; 951099a0e58SBosko Milekic } else { 952099a0e58SBosko Milekic if (totlen + off >= MINCLSIZE) { 953eb1b1807SGleb Smirnoff m = m_getcl(M_NOWAIT, MT_DATA, 0); 954099a0e58SBosko Milekic len = MCLBYTES; 955099a0e58SBosko Milekic } else { 956eb1b1807SGleb Smirnoff m = m_get(M_NOWAIT, MT_DATA); 957099a0e58SBosko Milekic len = MLEN; 958099a0e58SBosko Milekic } 959099a0e58SBosko Milekic if (m == NULL) { 960099a0e58SBosko Milekic m_freem(top); 961099a0e58SBosko Milekic return NULL; 962099a0e58SBosko Milekic } 963099a0e58SBosko Milekic } 964f5eece3fSBosko Milekic if (off) { 965f5eece3fSBosko Milekic m->m_data += off; 966f5eece3fSBosko Milekic len -= off; 967f5eece3fSBosko Milekic off = 0; 968f5eece3fSBosko Milekic } 969f5eece3fSBosko Milekic m->m_len = len = min(totlen, len); 970df8bae1dSRodney W. Grimes if (copy) 971bd395ae8SBosko Milekic copy(buf, mtod(m, caddr_t), (u_int)len); 972df8bae1dSRodney W. Grimes else 973bd395ae8SBosko Milekic bcopy(buf, mtod(m, caddr_t), (u_int)len); 974f5eece3fSBosko Milekic buf += len; 975df8bae1dSRodney W. Grimes *mp = m; 976df8bae1dSRodney W. Grimes mp = &m->m_next; 977df8bae1dSRodney W. Grimes totlen -= len; 978df8bae1dSRodney W. Grimes } 979df8bae1dSRodney W. Grimes return (top); 980df8bae1dSRodney W. Grimes } 981c5789ba3SPoul-Henning Kamp 982c5789ba3SPoul-Henning Kamp /* 983c5789ba3SPoul-Henning Kamp * Copy data from a buffer back into the indicated mbuf chain, 984c5789ba3SPoul-Henning Kamp * starting "off" bytes from the beginning, extending the mbuf 985c5789ba3SPoul-Henning Kamp * chain if necessary. 986c5789ba3SPoul-Henning Kamp */ 987c5789ba3SPoul-Henning Kamp void 98824665342SLuigi Rizzo m_copyback(struct mbuf *m0, int off, int len, c_caddr_t cp) 989c5789ba3SPoul-Henning Kamp { 990122a814aSBosko Milekic int mlen; 991122a814aSBosko Milekic struct mbuf *m = m0, *n; 992c5789ba3SPoul-Henning Kamp int totlen = 0; 993c5789ba3SPoul-Henning Kamp 994122a814aSBosko Milekic if (m0 == NULL) 995c5789ba3SPoul-Henning Kamp return; 996c5789ba3SPoul-Henning Kamp while (off > (mlen = m->m_len)) { 997c5789ba3SPoul-Henning Kamp off -= mlen; 998c5789ba3SPoul-Henning Kamp totlen += mlen; 999122a814aSBosko Milekic if (m->m_next == NULL) { 1000eb1b1807SGleb Smirnoff n = m_get(M_NOWAIT, m->m_type); 1001122a814aSBosko Milekic if (n == NULL) 1002c5789ba3SPoul-Henning Kamp goto out; 1003099a0e58SBosko Milekic bzero(mtod(n, caddr_t), MLEN); 1004c5789ba3SPoul-Henning Kamp n->m_len = min(MLEN, len + off); 1005c5789ba3SPoul-Henning Kamp m->m_next = n; 1006c5789ba3SPoul-Henning Kamp } 1007c5789ba3SPoul-Henning Kamp m = m->m_next; 1008c5789ba3SPoul-Henning Kamp } 1009c5789ba3SPoul-Henning Kamp while (len > 0) { 1010129c5c81SAlexander Motin if (m->m_next == NULL && (len > m->m_len - off)) { 1011129c5c81SAlexander Motin m->m_len += min(len - (m->m_len - off), 1012129c5c81SAlexander Motin M_TRAILINGSPACE(m)); 1013129c5c81SAlexander Motin } 1014c5789ba3SPoul-Henning Kamp mlen = min (m->m_len - off, len); 1015bd395ae8SBosko Milekic bcopy(cp, off + mtod(m, caddr_t), (u_int)mlen); 1016c5789ba3SPoul-Henning Kamp cp += mlen; 1017c5789ba3SPoul-Henning Kamp len -= mlen; 1018c5789ba3SPoul-Henning Kamp mlen += off; 1019c5789ba3SPoul-Henning Kamp off = 0; 1020c5789ba3SPoul-Henning Kamp totlen += mlen; 1021c5789ba3SPoul-Henning Kamp if (len == 0) 1022c5789ba3SPoul-Henning Kamp break; 1023122a814aSBosko Milekic if (m->m_next == NULL) { 1024eb1b1807SGleb Smirnoff n = m_get(M_NOWAIT, m->m_type); 1025122a814aSBosko Milekic if (n == NULL) 1026c5789ba3SPoul-Henning Kamp break; 1027c5789ba3SPoul-Henning Kamp n->m_len = min(MLEN, len); 1028c5789ba3SPoul-Henning Kamp m->m_next = n; 1029c5789ba3SPoul-Henning Kamp } 1030c5789ba3SPoul-Henning Kamp m = m->m_next; 1031c5789ba3SPoul-Henning Kamp } 1032c5789ba3SPoul-Henning Kamp out: if (((m = m0)->m_flags & M_PKTHDR) && (m->m_pkthdr.len < totlen)) 1033c5789ba3SPoul-Henning Kamp m->m_pkthdr.len = totlen; 1034c5789ba3SPoul-Henning Kamp } 1035ce4a64f7SPoul-Henning Kamp 103637621fd5SBruce M Simpson /* 10374873d175SSam Leffler * Append the specified data to the indicated mbuf chain, 10384873d175SSam Leffler * Extend the mbuf chain if the new data does not fit in 10394873d175SSam Leffler * existing space. 10404873d175SSam Leffler * 10414873d175SSam Leffler * Return 1 if able to complete the job; otherwise 0. 10424873d175SSam Leffler */ 10434873d175SSam Leffler int 10444873d175SSam Leffler m_append(struct mbuf *m0, int len, c_caddr_t cp) 10454873d175SSam Leffler { 10464873d175SSam Leffler struct mbuf *m, *n; 10474873d175SSam Leffler int remainder, space; 10484873d175SSam Leffler 10494873d175SSam Leffler for (m = m0; m->m_next != NULL; m = m->m_next) 10504873d175SSam Leffler ; 10514873d175SSam Leffler remainder = len; 10524873d175SSam Leffler space = M_TRAILINGSPACE(m); 10534873d175SSam Leffler if (space > 0) { 10544873d175SSam Leffler /* 10554873d175SSam Leffler * Copy into available space. 10564873d175SSam Leffler */ 10574873d175SSam Leffler if (space > remainder) 10584873d175SSam Leffler space = remainder; 10594873d175SSam Leffler bcopy(cp, mtod(m, caddr_t) + m->m_len, space); 10604873d175SSam Leffler m->m_len += space; 10614873d175SSam Leffler cp += space, remainder -= space; 10624873d175SSam Leffler } 10634873d175SSam Leffler while (remainder > 0) { 10644873d175SSam Leffler /* 10654873d175SSam Leffler * Allocate a new mbuf; could check space 10664873d175SSam Leffler * and allocate a cluster instead. 10674873d175SSam Leffler */ 1068eb1b1807SGleb Smirnoff n = m_get(M_NOWAIT, m->m_type); 10694873d175SSam Leffler if (n == NULL) 10704873d175SSam Leffler break; 10714873d175SSam Leffler n->m_len = min(MLEN, remainder); 1072a37c415eSSam Leffler bcopy(cp, mtod(n, caddr_t), n->m_len); 1073a37c415eSSam Leffler cp += n->m_len, remainder -= n->m_len; 10744873d175SSam Leffler m->m_next = n; 10754873d175SSam Leffler m = n; 10764873d175SSam Leffler } 10774873d175SSam Leffler if (m0->m_flags & M_PKTHDR) 10784873d175SSam Leffler m0->m_pkthdr.len += len - remainder; 10794873d175SSam Leffler return (remainder == 0); 10804873d175SSam Leffler } 10814873d175SSam Leffler 10824873d175SSam Leffler /* 108337621fd5SBruce M Simpson * Apply function f to the data in an mbuf chain starting "off" bytes from 108437621fd5SBruce M Simpson * the beginning, continuing for "len" bytes. 108537621fd5SBruce M Simpson */ 108637621fd5SBruce M Simpson int 108737621fd5SBruce M Simpson m_apply(struct mbuf *m, int off, int len, 108854065297SBruce M Simpson int (*f)(void *, void *, u_int), void *arg) 108937621fd5SBruce M Simpson { 109054065297SBruce M Simpson u_int count; 109137621fd5SBruce M Simpson int rval; 109237621fd5SBruce M Simpson 109337621fd5SBruce M Simpson KASSERT(off >= 0, ("m_apply, negative off %d", off)); 109437621fd5SBruce M Simpson KASSERT(len >= 0, ("m_apply, negative len %d", len)); 109537621fd5SBruce M Simpson while (off > 0) { 109637621fd5SBruce M Simpson KASSERT(m != NULL, ("m_apply, offset > size of mbuf chain")); 109737621fd5SBruce M Simpson if (off < m->m_len) 109837621fd5SBruce M Simpson break; 109937621fd5SBruce M Simpson off -= m->m_len; 110037621fd5SBruce M Simpson m = m->m_next; 110137621fd5SBruce M Simpson } 110237621fd5SBruce M Simpson while (len > 0) { 110337621fd5SBruce M Simpson KASSERT(m != NULL, ("m_apply, offset > size of mbuf chain")); 110437621fd5SBruce M Simpson count = min(m->m_len - off, len); 110537621fd5SBruce M Simpson rval = (*f)(arg, mtod(m, caddr_t) + off, count); 110637621fd5SBruce M Simpson if (rval) 110737621fd5SBruce M Simpson return (rval); 110837621fd5SBruce M Simpson len -= count; 110937621fd5SBruce M Simpson off = 0; 111037621fd5SBruce M Simpson m = m->m_next; 111137621fd5SBruce M Simpson } 111237621fd5SBruce M Simpson return (0); 111337621fd5SBruce M Simpson } 111437621fd5SBruce M Simpson 111537621fd5SBruce M Simpson /* 111637621fd5SBruce M Simpson * Return a pointer to mbuf/offset of location in mbuf chain. 111737621fd5SBruce M Simpson */ 111837621fd5SBruce M Simpson struct mbuf * 111937621fd5SBruce M Simpson m_getptr(struct mbuf *m, int loc, int *off) 112037621fd5SBruce M Simpson { 112137621fd5SBruce M Simpson 112237621fd5SBruce M Simpson while (loc >= 0) { 112354065297SBruce M Simpson /* Normal end of search. */ 112437621fd5SBruce M Simpson if (m->m_len > loc) { 112537621fd5SBruce M Simpson *off = loc; 112637621fd5SBruce M Simpson return (m); 112737621fd5SBruce M Simpson } else { 112837621fd5SBruce M Simpson loc -= m->m_len; 112937621fd5SBruce M Simpson if (m->m_next == NULL) { 113037621fd5SBruce M Simpson if (loc == 0) { 113154065297SBruce M Simpson /* Point at the end of valid data. */ 113237621fd5SBruce M Simpson *off = m->m_len; 113337621fd5SBruce M Simpson return (m); 113454065297SBruce M Simpson } 113537621fd5SBruce M Simpson return (NULL); 113654065297SBruce M Simpson } 113737621fd5SBruce M Simpson m = m->m_next; 113837621fd5SBruce M Simpson } 113937621fd5SBruce M Simpson } 114037621fd5SBruce M Simpson return (NULL); 114137621fd5SBruce M Simpson } 114237621fd5SBruce M Simpson 1143ce4a64f7SPoul-Henning Kamp void 11447b125090SJohn-Mark Gurney m_print(const struct mbuf *m, int maxlen) 1145ce4a64f7SPoul-Henning Kamp { 1146ce4a64f7SPoul-Henning Kamp int len; 11477b125090SJohn-Mark Gurney int pdata; 11486357e7b5SEivind Eklund const struct mbuf *m2; 1149ce4a64f7SPoul-Henning Kamp 11507e949c46SKenneth D. Merry if (m == NULL) { 11517e949c46SKenneth D. Merry printf("mbuf: %p\n", m); 11527e949c46SKenneth D. Merry return; 11537e949c46SKenneth D. Merry } 11547e949c46SKenneth D. Merry 11557b125090SJohn-Mark Gurney if (m->m_flags & M_PKTHDR) 1156ce4a64f7SPoul-Henning Kamp len = m->m_pkthdr.len; 11577b125090SJohn-Mark Gurney else 11587b125090SJohn-Mark Gurney len = -1; 1159ce4a64f7SPoul-Henning Kamp m2 = m; 11607b125090SJohn-Mark Gurney while (m2 != NULL && (len == -1 || len)) { 11617b125090SJohn-Mark Gurney pdata = m2->m_len; 11627b125090SJohn-Mark Gurney if (maxlen != -1 && pdata > maxlen) 11637b125090SJohn-Mark Gurney pdata = maxlen; 11647b125090SJohn-Mark Gurney printf("mbuf: %p len: %d, next: %p, %b%s", m2, m2->m_len, 11657b125090SJohn-Mark Gurney m2->m_next, m2->m_flags, "\20\20freelist\17skipfw" 11667b125090SJohn-Mark Gurney "\11proto5\10proto4\7proto3\6proto2\5proto1\4rdonly" 11677b125090SJohn-Mark Gurney "\3eor\2pkthdr\1ext", pdata ? "" : "\n"); 11687b125090SJohn-Mark Gurney if (pdata) 116945e0d0aaSJohn-Mark Gurney printf(", %*D\n", pdata, (u_char *)m2->m_data, "-"); 11707b125090SJohn-Mark Gurney if (len != -1) 1171ce4a64f7SPoul-Henning Kamp len -= m2->m_len; 1172ce4a64f7SPoul-Henning Kamp m2 = m2->m_next; 1173ce4a64f7SPoul-Henning Kamp } 11747b125090SJohn-Mark Gurney if (len > 0) 11757b125090SJohn-Mark Gurney printf("%d bytes unaccounted for.\n", len); 1176ce4a64f7SPoul-Henning Kamp return; 1177ce4a64f7SPoul-Henning Kamp } 11783f2e06c5SPoul-Henning Kamp 1179bd395ae8SBosko Milekic u_int 11803f2e06c5SPoul-Henning Kamp m_fixhdr(struct mbuf *m0) 11813f2e06c5SPoul-Henning Kamp { 1182bd395ae8SBosko Milekic u_int len; 11833f2e06c5SPoul-Henning Kamp 1184ac6e585dSPoul-Henning Kamp len = m_length(m0, NULL); 11853f2e06c5SPoul-Henning Kamp m0->m_pkthdr.len = len; 1186ac6e585dSPoul-Henning Kamp return (len); 1187ac6e585dSPoul-Henning Kamp } 1188ac6e585dSPoul-Henning Kamp 1189bd395ae8SBosko Milekic u_int 1190ac6e585dSPoul-Henning Kamp m_length(struct mbuf *m0, struct mbuf **last) 1191ac6e585dSPoul-Henning Kamp { 1192ac6e585dSPoul-Henning Kamp struct mbuf *m; 1193bd395ae8SBosko Milekic u_int len; 1194ac6e585dSPoul-Henning Kamp 1195ac6e585dSPoul-Henning Kamp len = 0; 1196ac6e585dSPoul-Henning Kamp for (m = m0; m != NULL; m = m->m_next) { 1197ac6e585dSPoul-Henning Kamp len += m->m_len; 1198ac6e585dSPoul-Henning Kamp if (m->m_next == NULL) 1199ac6e585dSPoul-Henning Kamp break; 1200ac6e585dSPoul-Henning Kamp } 1201ac6e585dSPoul-Henning Kamp if (last != NULL) 1202ac6e585dSPoul-Henning Kamp *last = m; 1203ac6e585dSPoul-Henning Kamp return (len); 12043f2e06c5SPoul-Henning Kamp } 120555e9f80dSMike Silbersack 120655e9f80dSMike Silbersack /* 120755e9f80dSMike Silbersack * Defragment a mbuf chain, returning the shortest possible 120855e9f80dSMike Silbersack * chain of mbufs and clusters. If allocation fails and 120955e9f80dSMike Silbersack * this cannot be completed, NULL will be returned, but 121055e9f80dSMike Silbersack * the passed in chain will be unchanged. Upon success, 121155e9f80dSMike Silbersack * the original chain will be freed, and the new chain 121255e9f80dSMike Silbersack * will be returned. 121355e9f80dSMike Silbersack * 121455e9f80dSMike Silbersack * If a non-packet header is passed in, the original 121555e9f80dSMike Silbersack * mbuf (chain?) will be returned unharmed. 121655e9f80dSMike Silbersack */ 121755e9f80dSMike Silbersack struct mbuf * 121855e9f80dSMike Silbersack m_defrag(struct mbuf *m0, int how) 121955e9f80dSMike Silbersack { 122055e9f80dSMike Silbersack struct mbuf *m_new = NULL, *m_final = NULL; 122155e9f80dSMike Silbersack int progress = 0, length; 122255e9f80dSMike Silbersack 1223063d8114SAlfred Perlstein MBUF_CHECKSLEEP(how); 122455e9f80dSMike Silbersack if (!(m0->m_flags & M_PKTHDR)) 122555e9f80dSMike Silbersack return (m0); 122655e9f80dSMike Silbersack 1227f8bf8e39SMike Silbersack m_fixhdr(m0); /* Needed sanity check */ 1228f8bf8e39SMike Silbersack 1229352d050eSMike Silbersack #ifdef MBUF_STRESS_TEST 1230352d050eSMike Silbersack if (m_defragrandomfailures) { 1231352d050eSMike Silbersack int temp = arc4random() & 0xff; 1232352d050eSMike Silbersack if (temp == 0xba) 1233352d050eSMike Silbersack goto nospace; 1234352d050eSMike Silbersack } 1235352d050eSMike Silbersack #endif 123655e9f80dSMike Silbersack 123755e9f80dSMike Silbersack if (m0->m_pkthdr.len > MHLEN) 123855e9f80dSMike Silbersack m_final = m_getcl(how, MT_DATA, M_PKTHDR); 123955e9f80dSMike Silbersack else 124055e9f80dSMike Silbersack m_final = m_gethdr(how, MT_DATA); 124155e9f80dSMike Silbersack 124255e9f80dSMike Silbersack if (m_final == NULL) 124355e9f80dSMike Silbersack goto nospace; 124455e9f80dSMike Silbersack 1245a89ec05eSPeter Wemm if (m_dup_pkthdr(m_final, m0, how) == 0) 124655e9f80dSMike Silbersack goto nospace; 124755e9f80dSMike Silbersack 124855e9f80dSMike Silbersack m_new = m_final; 124955e9f80dSMike Silbersack 125055e9f80dSMike Silbersack while (progress < m0->m_pkthdr.len) { 125155e9f80dSMike Silbersack length = m0->m_pkthdr.len - progress; 125255e9f80dSMike Silbersack if (length > MCLBYTES) 125355e9f80dSMike Silbersack length = MCLBYTES; 125455e9f80dSMike Silbersack 125555e9f80dSMike Silbersack if (m_new == NULL) { 125655e9f80dSMike Silbersack if (length > MLEN) 125755e9f80dSMike Silbersack m_new = m_getcl(how, MT_DATA, 0); 125855e9f80dSMike Silbersack else 125955e9f80dSMike Silbersack m_new = m_get(how, MT_DATA); 126055e9f80dSMike Silbersack if (m_new == NULL) 126155e9f80dSMike Silbersack goto nospace; 126255e9f80dSMike Silbersack } 126355e9f80dSMike Silbersack 126455e9f80dSMike Silbersack m_copydata(m0, progress, length, mtod(m_new, caddr_t)); 126555e9f80dSMike Silbersack progress += length; 126655e9f80dSMike Silbersack m_new->m_len = length; 126755e9f80dSMike Silbersack if (m_new != m_final) 126855e9f80dSMike Silbersack m_cat(m_final, m_new); 126955e9f80dSMike Silbersack m_new = NULL; 127055e9f80dSMike Silbersack } 127151710a45SMike Silbersack #ifdef MBUF_STRESS_TEST 127255e9f80dSMike Silbersack if (m0->m_next == NULL) 127355e9f80dSMike Silbersack m_defraguseless++; 127451710a45SMike Silbersack #endif 127555e9f80dSMike Silbersack m_freem(m0); 127655e9f80dSMike Silbersack m0 = m_final; 127751710a45SMike Silbersack #ifdef MBUF_STRESS_TEST 127855e9f80dSMike Silbersack m_defragpackets++; 127955e9f80dSMike Silbersack m_defragbytes += m0->m_pkthdr.len; 128051710a45SMike Silbersack #endif 128155e9f80dSMike Silbersack return (m0); 128255e9f80dSMike Silbersack nospace: 128351710a45SMike Silbersack #ifdef MBUF_STRESS_TEST 128455e9f80dSMike Silbersack m_defragfailure++; 128551710a45SMike Silbersack #endif 128655e9f80dSMike Silbersack if (m_final) 128755e9f80dSMike Silbersack m_freem(m_final); 128855e9f80dSMike Silbersack return (NULL); 128955e9f80dSMike Silbersack } 12903390d476SMike Silbersack 1291eeb76a18SSam Leffler /* 1292eeb76a18SSam Leffler * Defragment an mbuf chain, returning at most maxfrags separate 1293eeb76a18SSam Leffler * mbufs+clusters. If this is not possible NULL is returned and 1294eeb76a18SSam Leffler * the original mbuf chain is left in it's present (potentially 1295eeb76a18SSam Leffler * modified) state. We use two techniques: collapsing consecutive 1296eeb76a18SSam Leffler * mbufs and replacing consecutive mbufs by a cluster. 1297eeb76a18SSam Leffler * 1298eeb76a18SSam Leffler * NB: this should really be named m_defrag but that name is taken 1299eeb76a18SSam Leffler */ 1300eeb76a18SSam Leffler struct mbuf * 1301eeb76a18SSam Leffler m_collapse(struct mbuf *m0, int how, int maxfrags) 1302eeb76a18SSam Leffler { 1303eeb76a18SSam Leffler struct mbuf *m, *n, *n2, **prev; 1304eeb76a18SSam Leffler u_int curfrags; 1305eeb76a18SSam Leffler 1306eeb76a18SSam Leffler /* 1307eeb76a18SSam Leffler * Calculate the current number of frags. 1308eeb76a18SSam Leffler */ 1309eeb76a18SSam Leffler curfrags = 0; 1310eeb76a18SSam Leffler for (m = m0; m != NULL; m = m->m_next) 1311eeb76a18SSam Leffler curfrags++; 1312eeb76a18SSam Leffler /* 1313eeb76a18SSam Leffler * First, try to collapse mbufs. Note that we always collapse 1314eeb76a18SSam Leffler * towards the front so we don't need to deal with moving the 1315eeb76a18SSam Leffler * pkthdr. This may be suboptimal if the first mbuf has much 1316eeb76a18SSam Leffler * less data than the following. 1317eeb76a18SSam Leffler */ 1318eeb76a18SSam Leffler m = m0; 1319eeb76a18SSam Leffler again: 1320eeb76a18SSam Leffler for (;;) { 1321eeb76a18SSam Leffler n = m->m_next; 1322eeb76a18SSam Leffler if (n == NULL) 1323eeb76a18SSam Leffler break; 132414d7c5b1SAndre Oppermann if (M_WRITABLE(m) && 1325eeb76a18SSam Leffler n->m_len < M_TRAILINGSPACE(m)) { 1326eeb76a18SSam Leffler bcopy(mtod(n, void *), mtod(m, char *) + m->m_len, 1327eeb76a18SSam Leffler n->m_len); 1328eeb76a18SSam Leffler m->m_len += n->m_len; 1329eeb76a18SSam Leffler m->m_next = n->m_next; 1330eeb76a18SSam Leffler m_free(n); 1331eeb76a18SSam Leffler if (--curfrags <= maxfrags) 1332eeb76a18SSam Leffler return m0; 1333eeb76a18SSam Leffler } else 1334eeb76a18SSam Leffler m = n; 1335eeb76a18SSam Leffler } 1336eeb76a18SSam Leffler KASSERT(maxfrags > 1, 1337eeb76a18SSam Leffler ("maxfrags %u, but normal collapse failed", maxfrags)); 1338eeb76a18SSam Leffler /* 1339eeb76a18SSam Leffler * Collapse consecutive mbufs to a cluster. 1340eeb76a18SSam Leffler */ 1341eeb76a18SSam Leffler prev = &m0->m_next; /* NB: not the first mbuf */ 1342eeb76a18SSam Leffler while ((n = *prev) != NULL) { 1343eeb76a18SSam Leffler if ((n2 = n->m_next) != NULL && 1344eeb76a18SSam Leffler n->m_len + n2->m_len < MCLBYTES) { 1345eeb76a18SSam Leffler m = m_getcl(how, MT_DATA, 0); 1346eeb76a18SSam Leffler if (m == NULL) 1347eeb76a18SSam Leffler goto bad; 1348eeb76a18SSam Leffler bcopy(mtod(n, void *), mtod(m, void *), n->m_len); 1349eeb76a18SSam Leffler bcopy(mtod(n2, void *), mtod(m, char *) + n->m_len, 1350eeb76a18SSam Leffler n2->m_len); 1351eeb76a18SSam Leffler m->m_len = n->m_len + n2->m_len; 1352eeb76a18SSam Leffler m->m_next = n2->m_next; 1353eeb76a18SSam Leffler *prev = m; 1354eeb76a18SSam Leffler m_free(n); 1355eeb76a18SSam Leffler m_free(n2); 1356eeb76a18SSam Leffler if (--curfrags <= maxfrags) /* +1 cl -2 mbufs */ 1357eeb76a18SSam Leffler return m0; 1358eeb76a18SSam Leffler /* 1359eeb76a18SSam Leffler * Still not there, try the normal collapse 1360eeb76a18SSam Leffler * again before we allocate another cluster. 1361eeb76a18SSam Leffler */ 1362eeb76a18SSam Leffler goto again; 1363eeb76a18SSam Leffler } 1364eeb76a18SSam Leffler prev = &n->m_next; 1365eeb76a18SSam Leffler } 1366eeb76a18SSam Leffler /* 1367eeb76a18SSam Leffler * No place where we can collapse to a cluster; punt. 1368eeb76a18SSam Leffler * This can occur if, for example, you request 2 frags 1369eeb76a18SSam Leffler * but the packet requires that both be clusters (we 1370eeb76a18SSam Leffler * never reallocate the first mbuf to avoid moving the 1371eeb76a18SSam Leffler * packet header). 1372eeb76a18SSam Leffler */ 1373eeb76a18SSam Leffler bad: 1374eeb76a18SSam Leffler return NULL; 1375eeb76a18SSam Leffler } 1376eeb76a18SSam Leffler 13773390d476SMike Silbersack #ifdef MBUF_STRESS_TEST 13783390d476SMike Silbersack 13793390d476SMike Silbersack /* 13803390d476SMike Silbersack * Fragment an mbuf chain. There's no reason you'd ever want to do 13813390d476SMike Silbersack * this in normal usage, but it's great for stress testing various 13823390d476SMike Silbersack * mbuf consumers. 13833390d476SMike Silbersack * 13843390d476SMike Silbersack * If fragmentation is not possible, the original chain will be 13853390d476SMike Silbersack * returned. 13863390d476SMike Silbersack * 13873390d476SMike Silbersack * Possible length values: 13883390d476SMike Silbersack * 0 no fragmentation will occur 13893390d476SMike Silbersack * > 0 each fragment will be of the specified length 13903390d476SMike Silbersack * -1 each fragment will be the same random value in length 13913390d476SMike Silbersack * -2 each fragment's length will be entirely random 13923390d476SMike Silbersack * (Random values range from 1 to 256) 13933390d476SMike Silbersack */ 13943390d476SMike Silbersack struct mbuf * 13953390d476SMike Silbersack m_fragment(struct mbuf *m0, int how, int length) 13963390d476SMike Silbersack { 13973390d476SMike Silbersack struct mbuf *m_new = NULL, *m_final = NULL; 13983390d476SMike Silbersack int progress = 0; 13993390d476SMike Silbersack 14003390d476SMike Silbersack if (!(m0->m_flags & M_PKTHDR)) 14013390d476SMike Silbersack return (m0); 14023390d476SMike Silbersack 14033390d476SMike Silbersack if ((length == 0) || (length < -2)) 14043390d476SMike Silbersack return (m0); 14053390d476SMike Silbersack 14063390d476SMike Silbersack m_fixhdr(m0); /* Needed sanity check */ 14073390d476SMike Silbersack 14083390d476SMike Silbersack m_final = m_getcl(how, MT_DATA, M_PKTHDR); 14093390d476SMike Silbersack 14103390d476SMike Silbersack if (m_final == NULL) 14113390d476SMike Silbersack goto nospace; 14123390d476SMike Silbersack 14138dee2f67SMike Silbersack if (m_dup_pkthdr(m_final, m0, how) == 0) 14143390d476SMike Silbersack goto nospace; 14153390d476SMike Silbersack 14163390d476SMike Silbersack m_new = m_final; 14173390d476SMike Silbersack 14183390d476SMike Silbersack if (length == -1) 14193390d476SMike Silbersack length = 1 + (arc4random() & 255); 14203390d476SMike Silbersack 14213390d476SMike Silbersack while (progress < m0->m_pkthdr.len) { 14223390d476SMike Silbersack int fraglen; 14233390d476SMike Silbersack 14243390d476SMike Silbersack if (length > 0) 14253390d476SMike Silbersack fraglen = length; 14263390d476SMike Silbersack else 14273390d476SMike Silbersack fraglen = 1 + (arc4random() & 255); 14283390d476SMike Silbersack if (fraglen > m0->m_pkthdr.len - progress) 14293390d476SMike Silbersack fraglen = m0->m_pkthdr.len - progress; 14303390d476SMike Silbersack 14313390d476SMike Silbersack if (fraglen > MCLBYTES) 14323390d476SMike Silbersack fraglen = MCLBYTES; 14333390d476SMike Silbersack 14343390d476SMike Silbersack if (m_new == NULL) { 14353390d476SMike Silbersack m_new = m_getcl(how, MT_DATA, 0); 14363390d476SMike Silbersack if (m_new == NULL) 14373390d476SMike Silbersack goto nospace; 14383390d476SMike Silbersack } 14393390d476SMike Silbersack 14403390d476SMike Silbersack m_copydata(m0, progress, fraglen, mtod(m_new, caddr_t)); 14413390d476SMike Silbersack progress += fraglen; 14423390d476SMike Silbersack m_new->m_len = fraglen; 14433390d476SMike Silbersack if (m_new != m_final) 14443390d476SMike Silbersack m_cat(m_final, m_new); 14453390d476SMike Silbersack m_new = NULL; 14463390d476SMike Silbersack } 14473390d476SMike Silbersack m_freem(m0); 14483390d476SMike Silbersack m0 = m_final; 14493390d476SMike Silbersack return (m0); 14503390d476SMike Silbersack nospace: 14513390d476SMike Silbersack if (m_final) 14523390d476SMike Silbersack m_freem(m_final); 14533390d476SMike Silbersack /* Return the original chain on failure */ 14543390d476SMike Silbersack return (m0); 14553390d476SMike Silbersack } 14563390d476SMike Silbersack 14573390d476SMike Silbersack #endif 1458beb699c7SMike Silbersack 14595e20f43dSAndre Oppermann /* 14605e20f43dSAndre Oppermann * Copy the contents of uio into a properly sized mbuf chain. 14615e20f43dSAndre Oppermann */ 1462beb699c7SMike Silbersack struct mbuf * 14635e20f43dSAndre Oppermann m_uiotombuf(struct uio *uio, int how, int len, int align, int flags) 1464beb699c7SMike Silbersack { 14655e20f43dSAndre Oppermann struct mbuf *m, *mb; 1466526d0bd5SKonstantin Belousov int error, length; 1467526d0bd5SKonstantin Belousov ssize_t total; 14685e20f43dSAndre Oppermann int progress = 0; 1469beb699c7SMike Silbersack 14705e20f43dSAndre Oppermann /* 14715e20f43dSAndre Oppermann * len can be zero or an arbitrary large value bound by 14725e20f43dSAndre Oppermann * the total data supplied by the uio. 14735e20f43dSAndre Oppermann */ 1474beb699c7SMike Silbersack if (len > 0) 1475beb699c7SMike Silbersack total = min(uio->uio_resid, len); 1476beb699c7SMike Silbersack else 1477beb699c7SMike Silbersack total = uio->uio_resid; 14785e20f43dSAndre Oppermann 14795e20f43dSAndre Oppermann /* 14805e20f43dSAndre Oppermann * The smallest unit returned by m_getm2() is a single mbuf 14819128ec21SAndrew Thompson * with pkthdr. We can't align past it. 14825e20f43dSAndre Oppermann */ 148375ae2570SMaksim Yevmenkin if (align >= MHLEN) 1484beb699c7SMike Silbersack return (NULL); 14855e20f43dSAndre Oppermann 14867c32173bSAndre Oppermann /* 14877c32173bSAndre Oppermann * Give us the full allocation or nothing. 14887c32173bSAndre Oppermann * If len is zero return the smallest empty mbuf. 14897c32173bSAndre Oppermann */ 14907c32173bSAndre Oppermann m = m_getm2(NULL, max(total + align, 1), how, MT_DATA, flags); 14915e20f43dSAndre Oppermann if (m == NULL) 14925e20f43dSAndre Oppermann return (NULL); 14935e20f43dSAndre Oppermann m->m_data += align; 14945e20f43dSAndre Oppermann 14955e20f43dSAndre Oppermann /* Fill all mbufs with uio data and update header information. */ 14965e20f43dSAndre Oppermann for (mb = m; mb != NULL; mb = mb->m_next) { 14975e20f43dSAndre Oppermann length = min(M_TRAILINGSPACE(mb), total - progress); 14985e20f43dSAndre Oppermann 14995e20f43dSAndre Oppermann error = uiomove(mtod(mb, void *), length, uio); 15005e20f43dSAndre Oppermann if (error) { 15015e20f43dSAndre Oppermann m_freem(m); 15025e20f43dSAndre Oppermann return (NULL); 15035e20f43dSAndre Oppermann } 15045e20f43dSAndre Oppermann 15055e20f43dSAndre Oppermann mb->m_len = length; 15065e20f43dSAndre Oppermann progress += length; 15075e20f43dSAndre Oppermann if (flags & M_PKTHDR) 15085e20f43dSAndre Oppermann m->m_pkthdr.len += length; 15095e20f43dSAndre Oppermann } 15105e20f43dSAndre Oppermann KASSERT(progress == total, ("%s: progress != total", __func__)); 15115e20f43dSAndre Oppermann 15125e20f43dSAndre Oppermann return (m); 1513beb699c7SMike Silbersack } 1514ab8ab90cSSam Leffler 1515ab8ab90cSSam Leffler /* 1516bc05b2f6SAndre Oppermann * Copy an mbuf chain into a uio limited by len if set. 1517bc05b2f6SAndre Oppermann */ 1518bc05b2f6SAndre Oppermann int 1519bc05b2f6SAndre Oppermann m_mbuftouio(struct uio *uio, struct mbuf *m, int len) 1520bc05b2f6SAndre Oppermann { 1521bc05b2f6SAndre Oppermann int error, length, total; 1522bc05b2f6SAndre Oppermann int progress = 0; 1523bc05b2f6SAndre Oppermann 1524bc05b2f6SAndre Oppermann if (len > 0) 1525bc05b2f6SAndre Oppermann total = min(uio->uio_resid, len); 1526bc05b2f6SAndre Oppermann else 1527bc05b2f6SAndre Oppermann total = uio->uio_resid; 1528bc05b2f6SAndre Oppermann 1529bc05b2f6SAndre Oppermann /* Fill the uio with data from the mbufs. */ 1530bc05b2f6SAndre Oppermann for (; m != NULL; m = m->m_next) { 1531bc05b2f6SAndre Oppermann length = min(m->m_len, total - progress); 1532bc05b2f6SAndre Oppermann 1533bc05b2f6SAndre Oppermann error = uiomove(mtod(m, void *), length, uio); 1534bc05b2f6SAndre Oppermann if (error) 1535bc05b2f6SAndre Oppermann return (error); 1536bc05b2f6SAndre Oppermann 1537bc05b2f6SAndre Oppermann progress += length; 1538bc05b2f6SAndre Oppermann } 1539bc05b2f6SAndre Oppermann 1540bc05b2f6SAndre Oppermann return (0); 1541bc05b2f6SAndre Oppermann } 1542bc05b2f6SAndre Oppermann 1543bc05b2f6SAndre Oppermann /* 154447e2996eSSam Leffler * Create a writable copy of the mbuf chain. While doing this 154547e2996eSSam Leffler * we compact the chain with a goal of producing a chain with 154647e2996eSSam Leffler * at most two mbufs. The second mbuf in this chain is likely 154747e2996eSSam Leffler * to be a cluster. The primary purpose of this work is to create 154847e2996eSSam Leffler * a writable packet for encryption, compression, etc. The 154947e2996eSSam Leffler * secondary goal is to linearize the data so the data can be 155047e2996eSSam Leffler * passed to crypto hardware in the most efficient manner possible. 155147e2996eSSam Leffler */ 155247e2996eSSam Leffler struct mbuf * 155347e2996eSSam Leffler m_unshare(struct mbuf *m0, int how) 155447e2996eSSam Leffler { 155547e2996eSSam Leffler struct mbuf *m, *mprev; 155647e2996eSSam Leffler struct mbuf *n, *mfirst, *mlast; 155747e2996eSSam Leffler int len, off; 155847e2996eSSam Leffler 155947e2996eSSam Leffler mprev = NULL; 156047e2996eSSam Leffler for (m = m0; m != NULL; m = mprev->m_next) { 156147e2996eSSam Leffler /* 156247e2996eSSam Leffler * Regular mbufs are ignored unless there's a cluster 156347e2996eSSam Leffler * in front of it that we can use to coalesce. We do 156447e2996eSSam Leffler * the latter mainly so later clusters can be coalesced 156547e2996eSSam Leffler * also w/o having to handle them specially (i.e. convert 156647e2996eSSam Leffler * mbuf+cluster -> cluster). This optimization is heavily 156747e2996eSSam Leffler * influenced by the assumption that we're running over 156847e2996eSSam Leffler * Ethernet where MCLBYTES is large enough that the max 156947e2996eSSam Leffler * packet size will permit lots of coalescing into a 157047e2996eSSam Leffler * single cluster. This in turn permits efficient 157147e2996eSSam Leffler * crypto operations, especially when using hardware. 157247e2996eSSam Leffler */ 157347e2996eSSam Leffler if ((m->m_flags & M_EXT) == 0) { 157447e2996eSSam Leffler if (mprev && (mprev->m_flags & M_EXT) && 157547e2996eSSam Leffler m->m_len <= M_TRAILINGSPACE(mprev)) { 157647e2996eSSam Leffler /* XXX: this ignores mbuf types */ 157747e2996eSSam Leffler memcpy(mtod(mprev, caddr_t) + mprev->m_len, 157847e2996eSSam Leffler mtod(m, caddr_t), m->m_len); 157947e2996eSSam Leffler mprev->m_len += m->m_len; 158047e2996eSSam Leffler mprev->m_next = m->m_next; /* unlink from chain */ 158147e2996eSSam Leffler m_free(m); /* reclaim mbuf */ 158247e2996eSSam Leffler #if 0 158347e2996eSSam Leffler newipsecstat.ips_mbcoalesced++; 158447e2996eSSam Leffler #endif 158547e2996eSSam Leffler } else { 158647e2996eSSam Leffler mprev = m; 158747e2996eSSam Leffler } 158847e2996eSSam Leffler continue; 158947e2996eSSam Leffler } 159047e2996eSSam Leffler /* 159147e2996eSSam Leffler * Writable mbufs are left alone (for now). 159247e2996eSSam Leffler */ 159347e2996eSSam Leffler if (M_WRITABLE(m)) { 159447e2996eSSam Leffler mprev = m; 159547e2996eSSam Leffler continue; 159647e2996eSSam Leffler } 159747e2996eSSam Leffler 159847e2996eSSam Leffler /* 159947e2996eSSam Leffler * Not writable, replace with a copy or coalesce with 160047e2996eSSam Leffler * the previous mbuf if possible (since we have to copy 160147e2996eSSam Leffler * it anyway, we try to reduce the number of mbufs and 160247e2996eSSam Leffler * clusters so that future work is easier). 160347e2996eSSam Leffler */ 160447e2996eSSam Leffler KASSERT(m->m_flags & M_EXT, ("m_flags 0x%x", m->m_flags)); 160547e2996eSSam Leffler /* NB: we only coalesce into a cluster or larger */ 160647e2996eSSam Leffler if (mprev != NULL && (mprev->m_flags & M_EXT) && 160747e2996eSSam Leffler m->m_len <= M_TRAILINGSPACE(mprev)) { 160847e2996eSSam Leffler /* XXX: this ignores mbuf types */ 160947e2996eSSam Leffler memcpy(mtod(mprev, caddr_t) + mprev->m_len, 161047e2996eSSam Leffler mtod(m, caddr_t), m->m_len); 161147e2996eSSam Leffler mprev->m_len += m->m_len; 161247e2996eSSam Leffler mprev->m_next = m->m_next; /* unlink from chain */ 161347e2996eSSam Leffler m_free(m); /* reclaim mbuf */ 161447e2996eSSam Leffler #if 0 161547e2996eSSam Leffler newipsecstat.ips_clcoalesced++; 161647e2996eSSam Leffler #endif 161747e2996eSSam Leffler continue; 161847e2996eSSam Leffler } 161947e2996eSSam Leffler 162047e2996eSSam Leffler /* 16215368b81eSGleb Smirnoff * Allocate new space to hold the copy and copy the data. 16225368b81eSGleb Smirnoff * We deal with jumbo mbufs (i.e. m_len > MCLBYTES) by 16235368b81eSGleb Smirnoff * splitting them into clusters. We could just malloc a 16245368b81eSGleb Smirnoff * buffer and make it external but too many device drivers 16255368b81eSGleb Smirnoff * don't know how to break up the non-contiguous memory when 16265368b81eSGleb Smirnoff * doing DMA. 162747e2996eSSam Leffler */ 162847e2996eSSam Leffler n = m_getcl(how, m->m_type, m->m_flags); 162947e2996eSSam Leffler if (n == NULL) { 163047e2996eSSam Leffler m_freem(m0); 163147e2996eSSam Leffler return (NULL); 163247e2996eSSam Leffler } 1633e40e8705SGleb Smirnoff if (m->m_flags & M_PKTHDR) { 1634e40e8705SGleb Smirnoff KASSERT(mprev == NULL, ("%s: m0 %p, m %p has M_PKTHDR", 1635e40e8705SGleb Smirnoff __func__, m0, m)); 1636e40e8705SGleb Smirnoff m_move_pkthdr(n, m); 1637e40e8705SGleb Smirnoff } 163847e2996eSSam Leffler len = m->m_len; 163947e2996eSSam Leffler off = 0; 164047e2996eSSam Leffler mfirst = n; 164147e2996eSSam Leffler mlast = NULL; 164247e2996eSSam Leffler for (;;) { 164347e2996eSSam Leffler int cc = min(len, MCLBYTES); 164447e2996eSSam Leffler memcpy(mtod(n, caddr_t), mtod(m, caddr_t) + off, cc); 164547e2996eSSam Leffler n->m_len = cc; 164647e2996eSSam Leffler if (mlast != NULL) 164747e2996eSSam Leffler mlast->m_next = n; 164847e2996eSSam Leffler mlast = n; 164947e2996eSSam Leffler #if 0 165047e2996eSSam Leffler newipsecstat.ips_clcopied++; 165147e2996eSSam Leffler #endif 165247e2996eSSam Leffler 165347e2996eSSam Leffler len -= cc; 165447e2996eSSam Leffler if (len <= 0) 165547e2996eSSam Leffler break; 165647e2996eSSam Leffler off += cc; 165747e2996eSSam Leffler 165847e2996eSSam Leffler n = m_getcl(how, m->m_type, m->m_flags); 165947e2996eSSam Leffler if (n == NULL) { 166047e2996eSSam Leffler m_freem(mfirst); 166147e2996eSSam Leffler m_freem(m0); 166247e2996eSSam Leffler return (NULL); 166347e2996eSSam Leffler } 166447e2996eSSam Leffler } 166547e2996eSSam Leffler n->m_next = m->m_next; 166647e2996eSSam Leffler if (mprev == NULL) 166747e2996eSSam Leffler m0 = mfirst; /* new head of chain */ 166847e2996eSSam Leffler else 166947e2996eSSam Leffler mprev->m_next = mfirst; /* replace old mbuf */ 167047e2996eSSam Leffler m_free(m); /* release old mbuf */ 167147e2996eSSam Leffler mprev = mfirst; 167247e2996eSSam Leffler } 167347e2996eSSam Leffler return (m0); 167447e2996eSSam Leffler } 16756eeac1d9SJulian Elischer 16766eeac1d9SJulian Elischer #ifdef MBUF_PROFILING 16776eeac1d9SJulian Elischer 16786eeac1d9SJulian Elischer #define MP_BUCKETS 32 /* don't just change this as things may overflow.*/ 16796eeac1d9SJulian Elischer struct mbufprofile { 16802182c0cfSJulian Elischer uintmax_t wasted[MP_BUCKETS]; 16812182c0cfSJulian Elischer uintmax_t used[MP_BUCKETS]; 16822182c0cfSJulian Elischer uintmax_t segments[MP_BUCKETS]; 16836eeac1d9SJulian Elischer } mbprof; 16846eeac1d9SJulian Elischer 16856eeac1d9SJulian Elischer #define MP_MAXDIGITS 21 /* strlen("16,000,000,000,000,000,000") == 21 */ 16866eeac1d9SJulian Elischer #define MP_NUMLINES 6 16876eeac1d9SJulian Elischer #define MP_NUMSPERLINE 16 16886eeac1d9SJulian Elischer #define MP_EXTRABYTES 64 /* > strlen("used:\nwasted:\nsegments:\n") */ 16896eeac1d9SJulian Elischer /* work out max space needed and add a bit of spare space too */ 16906eeac1d9SJulian Elischer #define MP_MAXLINE ((MP_MAXDIGITS+1) * MP_NUMSPERLINE) 16916eeac1d9SJulian Elischer #define MP_BUFSIZE ((MP_MAXLINE * MP_NUMLINES) + 1 + MP_EXTRABYTES) 16926eeac1d9SJulian Elischer 16936eeac1d9SJulian Elischer char mbprofbuf[MP_BUFSIZE]; 16946eeac1d9SJulian Elischer 16956eeac1d9SJulian Elischer void 16966eeac1d9SJulian Elischer m_profile(struct mbuf *m) 16976eeac1d9SJulian Elischer { 16986eeac1d9SJulian Elischer int segments = 0; 16996eeac1d9SJulian Elischer int used = 0; 17006eeac1d9SJulian Elischer int wasted = 0; 17016eeac1d9SJulian Elischer 17026eeac1d9SJulian Elischer while (m) { 17036eeac1d9SJulian Elischer segments++; 17046eeac1d9SJulian Elischer used += m->m_len; 17056eeac1d9SJulian Elischer if (m->m_flags & M_EXT) { 17066eeac1d9SJulian Elischer wasted += MHLEN - sizeof(m->m_ext) + 17076eeac1d9SJulian Elischer m->m_ext.ext_size - m->m_len; 17086eeac1d9SJulian Elischer } else { 17096eeac1d9SJulian Elischer if (m->m_flags & M_PKTHDR) 17106eeac1d9SJulian Elischer wasted += MHLEN - m->m_len; 17116eeac1d9SJulian Elischer else 17126eeac1d9SJulian Elischer wasted += MLEN - m->m_len; 17136eeac1d9SJulian Elischer } 17146eeac1d9SJulian Elischer m = m->m_next; 17156eeac1d9SJulian Elischer } 17166eeac1d9SJulian Elischer /* be paranoid.. it helps */ 17176eeac1d9SJulian Elischer if (segments > MP_BUCKETS - 1) 17186eeac1d9SJulian Elischer segments = MP_BUCKETS - 1; 17196eeac1d9SJulian Elischer if (used > 100000) 17206eeac1d9SJulian Elischer used = 100000; 17216eeac1d9SJulian Elischer if (wasted > 100000) 17226eeac1d9SJulian Elischer wasted = 100000; 17236eeac1d9SJulian Elischer /* store in the appropriate bucket */ 17246eeac1d9SJulian Elischer /* don't bother locking. if it's slightly off, so what? */ 17256eeac1d9SJulian Elischer mbprof.segments[segments]++; 17266eeac1d9SJulian Elischer mbprof.used[fls(used)]++; 17276eeac1d9SJulian Elischer mbprof.wasted[fls(wasted)]++; 17286eeac1d9SJulian Elischer } 17296eeac1d9SJulian Elischer 17306eeac1d9SJulian Elischer static void 17316eeac1d9SJulian Elischer mbprof_textify(void) 17326eeac1d9SJulian Elischer { 17336eeac1d9SJulian Elischer int offset; 17346eeac1d9SJulian Elischer char *c; 173560ae52f7SEd Schouten uint64_t *p; 17366eeac1d9SJulian Elischer 17376eeac1d9SJulian Elischer p = &mbprof.wasted[0]; 17386eeac1d9SJulian Elischer c = mbprofbuf; 17396eeac1d9SJulian Elischer offset = snprintf(c, MP_MAXLINE + 10, 17406eeac1d9SJulian Elischer "wasted:\n" 17412182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju " 17422182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju\n", 17436eeac1d9SJulian Elischer p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 17446eeac1d9SJulian Elischer p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]); 17456eeac1d9SJulian Elischer #ifdef BIG_ARRAY 17466eeac1d9SJulian Elischer p = &mbprof.wasted[16]; 17476eeac1d9SJulian Elischer c += offset; 17486eeac1d9SJulian Elischer offset = snprintf(c, MP_MAXLINE, 17492182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju " 17502182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju\n", 17516eeac1d9SJulian Elischer p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 17526eeac1d9SJulian Elischer p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]); 17536eeac1d9SJulian Elischer #endif 17546eeac1d9SJulian Elischer p = &mbprof.used[0]; 17556eeac1d9SJulian Elischer c += offset; 17566eeac1d9SJulian Elischer offset = snprintf(c, MP_MAXLINE + 10, 17576eeac1d9SJulian Elischer "used:\n" 17582182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju " 17592182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju\n", 17606eeac1d9SJulian Elischer p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 17616eeac1d9SJulian Elischer p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]); 17626eeac1d9SJulian Elischer #ifdef BIG_ARRAY 17636eeac1d9SJulian Elischer p = &mbprof.used[16]; 17646eeac1d9SJulian Elischer c += offset; 17656eeac1d9SJulian Elischer offset = snprintf(c, MP_MAXLINE, 17662182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju " 17672182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju\n", 17686eeac1d9SJulian Elischer p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 17696eeac1d9SJulian Elischer p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]); 17706eeac1d9SJulian Elischer #endif 17716eeac1d9SJulian Elischer p = &mbprof.segments[0]; 17726eeac1d9SJulian Elischer c += offset; 17736eeac1d9SJulian Elischer offset = snprintf(c, MP_MAXLINE + 10, 17746eeac1d9SJulian Elischer "segments:\n" 17752182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju " 17762182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju\n", 17776eeac1d9SJulian Elischer p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 17786eeac1d9SJulian Elischer p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]); 17796eeac1d9SJulian Elischer #ifdef BIG_ARRAY 17806eeac1d9SJulian Elischer p = &mbprof.segments[16]; 17816eeac1d9SJulian Elischer c += offset; 17826eeac1d9SJulian Elischer offset = snprintf(c, MP_MAXLINE, 17832182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %ju " 17842182c0cfSJulian Elischer "%ju %ju %ju %ju %ju %ju %ju %jju", 17856eeac1d9SJulian Elischer p[0], p[1], p[2], p[3], p[4], p[5], p[6], p[7], 17866eeac1d9SJulian Elischer p[8], p[9], p[10], p[11], p[12], p[13], p[14], p[15]); 17876eeac1d9SJulian Elischer #endif 17886eeac1d9SJulian Elischer } 17896eeac1d9SJulian Elischer 17906eeac1d9SJulian Elischer static int 17916eeac1d9SJulian Elischer mbprof_handler(SYSCTL_HANDLER_ARGS) 17926eeac1d9SJulian Elischer { 17936eeac1d9SJulian Elischer int error; 17946eeac1d9SJulian Elischer 17956eeac1d9SJulian Elischer mbprof_textify(); 17966eeac1d9SJulian Elischer error = SYSCTL_OUT(req, mbprofbuf, strlen(mbprofbuf) + 1); 17976eeac1d9SJulian Elischer return (error); 17986eeac1d9SJulian Elischer } 17996eeac1d9SJulian Elischer 18006eeac1d9SJulian Elischer static int 18016eeac1d9SJulian Elischer mbprof_clr_handler(SYSCTL_HANDLER_ARGS) 18026eeac1d9SJulian Elischer { 18036eeac1d9SJulian Elischer int clear, error; 18046eeac1d9SJulian Elischer 18056eeac1d9SJulian Elischer clear = 0; 18066eeac1d9SJulian Elischer error = sysctl_handle_int(oidp, &clear, 0, req); 18076eeac1d9SJulian Elischer if (error || !req->newptr) 18086eeac1d9SJulian Elischer return (error); 18096eeac1d9SJulian Elischer 18106eeac1d9SJulian Elischer if (clear) { 18116eeac1d9SJulian Elischer bzero(&mbprof, sizeof(mbprof)); 18126eeac1d9SJulian Elischer } 18136eeac1d9SJulian Elischer 18146eeac1d9SJulian Elischer return (error); 18156eeac1d9SJulian Elischer } 18166eeac1d9SJulian Elischer 18176eeac1d9SJulian Elischer 18186eeac1d9SJulian Elischer SYSCTL_PROC(_kern_ipc, OID_AUTO, mbufprofile, CTLTYPE_STRING|CTLFLAG_RD, 18196eeac1d9SJulian Elischer NULL, 0, mbprof_handler, "A", "mbuf profiling statistics"); 18206eeac1d9SJulian Elischer 18216eeac1d9SJulian Elischer SYSCTL_PROC(_kern_ipc, OID_AUTO, mbufprofileclr, CTLTYPE_INT|CTLFLAG_RW, 18226eeac1d9SJulian Elischer NULL, 0, mbprof_clr_handler, "I", "clear mbuf profiling statistics"); 18236eeac1d9SJulian Elischer #endif 18246eeac1d9SJulian Elischer 1825