uipc_mbuf.c revision 1.153 1 1.153 christos /* $NetBSD: uipc_mbuf.c,v 1.153 2013/10/09 20:15:20 christos Exp $ */
2 1.42 thorpej
3 1.42 thorpej /*-
4 1.53 thorpej * Copyright (c) 1999, 2001 The NetBSD Foundation, Inc.
5 1.42 thorpej * All rights reserved.
6 1.42 thorpej *
7 1.42 thorpej * This code is derived from software contributed to The NetBSD Foundation
8 1.42 thorpej * by Jason R. Thorpe of the Numerical Aerospace Simulation Facility,
9 1.42 thorpej * NASA Ames Research Center.
10 1.42 thorpej *
11 1.42 thorpej * Redistribution and use in source and binary forms, with or without
12 1.42 thorpej * modification, are permitted provided that the following conditions
13 1.42 thorpej * are met:
14 1.42 thorpej * 1. Redistributions of source code must retain the above copyright
15 1.42 thorpej * notice, this list of conditions and the following disclaimer.
16 1.42 thorpej * 2. Redistributions in binary form must reproduce the above copyright
17 1.42 thorpej * notice, this list of conditions and the following disclaimer in the
18 1.42 thorpej * documentation and/or other materials provided with the distribution.
19 1.42 thorpej *
20 1.42 thorpej * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
21 1.42 thorpej * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
22 1.42 thorpej * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
23 1.42 thorpej * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
24 1.42 thorpej * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
25 1.42 thorpej * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26 1.42 thorpej * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27 1.42 thorpej * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28 1.42 thorpej * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29 1.42 thorpej * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30 1.42 thorpej * POSSIBILITY OF SUCH DAMAGE.
31 1.42 thorpej */
32 1.10 cgd
33 1.1 cgd /*
34 1.9 mycroft * Copyright (c) 1982, 1986, 1988, 1991, 1993
35 1.9 mycroft * The Regents of the University of California. All rights reserved.
36 1.1 cgd *
37 1.1 cgd * Redistribution and use in source and binary forms, with or without
38 1.1 cgd * modification, are permitted provided that the following conditions
39 1.1 cgd * are met:
40 1.1 cgd * 1. Redistributions of source code must retain the above copyright
41 1.1 cgd * notice, this list of conditions and the following disclaimer.
42 1.1 cgd * 2. Redistributions in binary form must reproduce the above copyright
43 1.1 cgd * notice, this list of conditions and the following disclaimer in the
44 1.1 cgd * documentation and/or other materials provided with the distribution.
45 1.70 agc * 3. Neither the name of the University nor the names of its contributors
46 1.1 cgd * may be used to endorse or promote products derived from this software
47 1.1 cgd * without specific prior written permission.
48 1.1 cgd *
49 1.1 cgd * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
50 1.1 cgd * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
51 1.1 cgd * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
52 1.1 cgd * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
53 1.1 cgd * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
54 1.1 cgd * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
55 1.1 cgd * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
56 1.1 cgd * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
57 1.1 cgd * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
58 1.1 cgd * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
59 1.1 cgd * SUCH DAMAGE.
60 1.1 cgd *
61 1.26 fvdl * @(#)uipc_mbuf.c 8.4 (Berkeley) 2/14/95
62 1.1 cgd */
63 1.56 lukem
64 1.56 lukem #include <sys/cdefs.h>
65 1.153 christos __KERNEL_RCSID(0, "$NetBSD: uipc_mbuf.c,v 1.153 2013/10/09 20:15:20 christos Exp $");
66 1.69 martin
67 1.69 martin #include "opt_mbuftrace.h"
68 1.133 joerg #include "opt_nmbclusters.h"
69 1.105 yamt #include "opt_ddb.h"
70 1.24 mrg
71 1.6 mycroft #include <sys/param.h>
72 1.6 mycroft #include <sys/systm.h>
73 1.125 yamt #include <sys/atomic.h>
74 1.124 yamt #include <sys/cpu.h>
75 1.6 mycroft #include <sys/proc.h>
76 1.6 mycroft #include <sys/mbuf.h>
77 1.6 mycroft #include <sys/kernel.h>
78 1.6 mycroft #include <sys/syslog.h>
79 1.6 mycroft #include <sys/domain.h>
80 1.6 mycroft #include <sys/protosw.h>
81 1.124 yamt #include <sys/percpu.h>
82 1.28 thorpej #include <sys/pool.h>
83 1.27 matt #include <sys/socket.h>
84 1.55 simonb #include <sys/sysctl.h>
85 1.55 simonb
86 1.27 matt #include <net/if.h>
87 1.14 christos
88 1.122 ad pool_cache_t mb_cache; /* mbuf cache */
89 1.122 ad pool_cache_t mcl_cache; /* mbuf cluster cache */
90 1.53 thorpej
91 1.18 thorpej struct mbstat mbstat;
92 1.18 thorpej int max_linkhdr;
93 1.18 thorpej int max_protohdr;
94 1.18 thorpej int max_hdr;
95 1.18 thorpej int max_datalen;
96 1.18 thorpej
97 1.65 thorpej static int mb_ctor(void *, void *, int);
98 1.65 thorpej
99 1.129 pooka static void sysctl_kern_mbuf_setup(void);
100 1.129 pooka
101 1.129 pooka static struct sysctllog *mbuf_sysctllog;
102 1.129 pooka
103 1.79 junyoung static struct mbuf *m_copym0(struct mbuf *, int, int, int, int);
104 1.85 yamt static struct mbuf *m_split0(struct mbuf *, int, int, int);
105 1.86 yamt static int m_copyback0(struct mbuf **, int, int, const void *, int, int);
106 1.85 yamt
107 1.85 yamt /* flags for m_copyback0 */
108 1.85 yamt #define M_COPYBACK0_COPYBACK 0x0001 /* copyback from cp */
109 1.85 yamt #define M_COPYBACK0_PRESERVE 0x0002 /* preserve original data */
110 1.85 yamt #define M_COPYBACK0_COW 0x0004 /* do copy-on-write */
111 1.85 yamt #define M_COPYBACK0_EXTEND 0x0008 /* extend chain */
112 1.28 thorpej
113 1.103 thorpej static const char mclpool_warnmsg[] =
114 1.133 joerg "WARNING: mclpool limit reached; increase kern.mbuf.nmbclusters";
115 1.63 thorpej
116 1.63 thorpej MALLOC_DEFINE(M_MBUF, "mbuf", "mbuf");
117 1.42 thorpej
118 1.124 yamt static percpu_t *mbstat_percpu;
119 1.124 yamt
120 1.64 matt #ifdef MBUFTRACE
121 1.64 matt struct mownerhead mowners = LIST_HEAD_INITIALIZER(mowners);
122 1.64 matt struct mowner unknown_mowners[] = {
123 1.114 dogcow MOWNER_INIT("unknown", "free"),
124 1.114 dogcow MOWNER_INIT("unknown", "data"),
125 1.114 dogcow MOWNER_INIT("unknown", "header"),
126 1.114 dogcow MOWNER_INIT("unknown", "soname"),
127 1.114 dogcow MOWNER_INIT("unknown", "soopts"),
128 1.114 dogcow MOWNER_INIT("unknown", "ftable"),
129 1.114 dogcow MOWNER_INIT("unknown", "control"),
130 1.114 dogcow MOWNER_INIT("unknown", "oobdata"),
131 1.64 matt };
132 1.114 dogcow struct mowner revoked_mowner = MOWNER_INIT("revoked", "");
133 1.64 matt #endif
134 1.64 matt
135 1.125 yamt #define MEXT_ISEMBEDDED(m) ((m)->m_ext_ref == (m))
136 1.125 yamt
137 1.125 yamt #define MCLADDREFERENCE(o, n) \
138 1.125 yamt do { \
139 1.125 yamt KASSERT(((o)->m_flags & M_EXT) != 0); \
140 1.125 yamt KASSERT(((n)->m_flags & M_EXT) == 0); \
141 1.125 yamt KASSERT((o)->m_ext.ext_refcnt >= 1); \
142 1.125 yamt (n)->m_flags |= ((o)->m_flags & M_EXTCOPYFLAGS); \
143 1.125 yamt atomic_inc_uint(&(o)->m_ext.ext_refcnt); \
144 1.125 yamt (n)->m_ext_ref = (o)->m_ext_ref; \
145 1.125 yamt mowner_ref((n), (n)->m_flags); \
146 1.125 yamt MCLREFDEBUGN((n), __FILE__, __LINE__); \
147 1.125 yamt } while (/* CONSTCOND */ 0)
148 1.125 yamt
149 1.133 joerg static int
150 1.133 joerg nmbclusters_limit(void)
151 1.133 joerg {
152 1.136 pooka #if defined(PMAP_MAP_POOLPAGE)
153 1.147 para /* direct mapping, doesn't use space in kmem_arena */
154 1.133 joerg vsize_t max_size = physmem / 4;
155 1.133 joerg #else
156 1.145 para vsize_t max_size = MIN(physmem / 4, nkmempages / 4);
157 1.133 joerg #endif
158 1.133 joerg
159 1.133 joerg max_size = max_size * PAGE_SIZE / MCLBYTES;
160 1.133 joerg #ifdef NMBCLUSTERS_MAX
161 1.133 joerg max_size = MIN(max_size, NMBCLUSTERS_MAX);
162 1.133 joerg #endif
163 1.133 joerg
164 1.133 joerg #ifdef NMBCLUSTERS
165 1.133 joerg return MIN(max_size, NMBCLUSTERS);
166 1.133 joerg #else
167 1.133 joerg return max_size;
168 1.133 joerg #endif
169 1.133 joerg }
170 1.133 joerg
171 1.28 thorpej /*
172 1.68 simonb * Initialize the mbuf allocator.
173 1.28 thorpej */
174 1.4 jtc void
175 1.62 thorpej mbinit(void)
176 1.1 cgd {
177 1.65 thorpej
178 1.128 matt CTASSERT(sizeof(struct _m_ext) <= MHLEN);
179 1.128 matt CTASSERT(sizeof(struct mbuf) == MSIZE);
180 1.65 thorpej
181 1.129 pooka sysctl_kern_mbuf_setup();
182 1.129 pooka
183 1.122 ad mb_cache = pool_cache_init(msize, 0, 0, 0, "mbpl",
184 1.122 ad NULL, IPL_VM, mb_ctor, NULL, NULL);
185 1.122 ad KASSERT(mb_cache != NULL);
186 1.122 ad
187 1.135 rmind mcl_cache = pool_cache_init(mclbytes, 0, 0, 0, "mclpl", NULL,
188 1.135 rmind IPL_VM, NULL, NULL, NULL);
189 1.122 ad KASSERT(mcl_cache != NULL);
190 1.59 thorpej
191 1.122 ad pool_cache_set_drain_hook(mb_cache, m_reclaim, NULL);
192 1.122 ad pool_cache_set_drain_hook(mcl_cache, m_reclaim, NULL);
193 1.37 thorpej
194 1.37 thorpej /*
195 1.133 joerg * Set an arbitrary default limit on the number of mbuf clusters.
196 1.133 joerg */
197 1.133 joerg #ifdef NMBCLUSTERS
198 1.133 joerg nmbclusters = nmbclusters_limit();
199 1.133 joerg #else
200 1.133 joerg nmbclusters = MAX(1024,
201 1.133 joerg (vsize_t)physmem * PAGE_SIZE / MCLBYTES / 16);
202 1.133 joerg nmbclusters = MIN(nmbclusters, nmbclusters_limit());
203 1.133 joerg #endif
204 1.133 joerg
205 1.133 joerg /*
206 1.39 thorpej * Set the hard limit on the mclpool to the number of
207 1.39 thorpej * mbuf clusters the kernel is to support. Log the limit
208 1.39 thorpej * reached message max once a minute.
209 1.39 thorpej */
210 1.122 ad pool_cache_sethardlimit(mcl_cache, nmbclusters, mclpool_warnmsg, 60);
211 1.42 thorpej
212 1.124 yamt mbstat_percpu = percpu_alloc(sizeof(struct mbstat_cpu));
213 1.124 yamt
214 1.39 thorpej /*
215 1.42 thorpej * Set a low water mark for both mbufs and clusters. This should
216 1.42 thorpej * help ensure that they can be allocated in a memory starvation
217 1.42 thorpej * situation. This is important for e.g. diskless systems which
218 1.42 thorpej * must allocate mbufs in order for the pagedaemon to clean pages.
219 1.37 thorpej */
220 1.122 ad pool_cache_setlowat(mb_cache, mblowat);
221 1.122 ad pool_cache_setlowat(mcl_cache, mcllowat);
222 1.64 matt
223 1.64 matt #ifdef MBUFTRACE
224 1.64 matt {
225 1.64 matt /*
226 1.64 matt * Attach the unknown mowners.
227 1.64 matt */
228 1.64 matt int i;
229 1.64 matt MOWNER_ATTACH(&revoked_mowner);
230 1.64 matt for (i = sizeof(unknown_mowners)/sizeof(unknown_mowners[0]);
231 1.64 matt i-- > 0; )
232 1.64 matt MOWNER_ATTACH(&unknown_mowners[i]);
233 1.64 matt }
234 1.64 matt #endif
235 1.42 thorpej }
236 1.42 thorpej
237 1.75 atatat /*
238 1.133 joerg * sysctl helper routine for the kern.mbuf subtree.
239 1.133 joerg * nmbclusters, mblowat and mcllowat need range
240 1.75 atatat * checking and pool tweaking after being reset.
241 1.75 atatat */
242 1.75 atatat static int
243 1.75 atatat sysctl_kern_mbuf(SYSCTLFN_ARGS)
244 1.42 thorpej {
245 1.42 thorpej int error, newval;
246 1.75 atatat struct sysctlnode node;
247 1.42 thorpej
248 1.75 atatat node = *rnode;
249 1.75 atatat node.sysctl_data = &newval;
250 1.75 atatat switch (rnode->sysctl_num) {
251 1.42 thorpej case MBUF_NMBCLUSTERS:
252 1.42 thorpej case MBUF_MBLOWAT:
253 1.42 thorpej case MBUF_MCLLOWAT:
254 1.75 atatat newval = *(int*)rnode->sysctl_data;
255 1.75 atatat break;
256 1.75 atatat default:
257 1.75 atatat return (EOPNOTSUPP);
258 1.75 atatat }
259 1.75 atatat
260 1.75 atatat error = sysctl_lookup(SYSCTLFN_CALL(&node));
261 1.75 atatat if (error || newp == NULL)
262 1.42 thorpej return (error);
263 1.75 atatat if (newval < 0)
264 1.75 atatat return (EINVAL);
265 1.75 atatat
266 1.75 atatat switch (node.sysctl_num) {
267 1.75 atatat case MBUF_NMBCLUSTERS:
268 1.75 atatat if (newval < nmbclusters)
269 1.75 atatat return (EINVAL);
270 1.133 joerg if (newval > nmbclusters_limit())
271 1.133 joerg return (EINVAL);
272 1.75 atatat nmbclusters = newval;
273 1.122 ad pool_cache_sethardlimit(mcl_cache, nmbclusters,
274 1.122 ad mclpool_warnmsg, 60);
275 1.75 atatat break;
276 1.75 atatat case MBUF_MBLOWAT:
277 1.75 atatat mblowat = newval;
278 1.122 ad pool_cache_setlowat(mb_cache, mblowat);
279 1.75 atatat break;
280 1.75 atatat case MBUF_MCLLOWAT:
281 1.76 atatat mcllowat = newval;
282 1.122 ad pool_cache_setlowat(mcl_cache, mcllowat);
283 1.75 atatat break;
284 1.75 atatat }
285 1.75 atatat
286 1.75 atatat return (0);
287 1.75 atatat }
288 1.75 atatat
289 1.64 matt #ifdef MBUFTRACE
290 1.124 yamt static void
291 1.124 yamt mowner_conver_to_user_cb(void *v1, void *v2, struct cpu_info *ci)
292 1.124 yamt {
293 1.124 yamt struct mowner_counter *mc = v1;
294 1.124 yamt struct mowner_user *mo_user = v2;
295 1.124 yamt int i;
296 1.124 yamt
297 1.124 yamt for (i = 0; i < MOWNER_COUNTER_NCOUNTERS; i++) {
298 1.124 yamt mo_user->mo_counter[i] += mc->mc_counter[i];
299 1.124 yamt }
300 1.124 yamt }
301 1.124 yamt
302 1.124 yamt static void
303 1.124 yamt mowner_convert_to_user(struct mowner *mo, struct mowner_user *mo_user)
304 1.124 yamt {
305 1.124 yamt
306 1.124 yamt memset(mo_user, 0, sizeof(*mo_user));
307 1.128 matt CTASSERT(sizeof(mo_user->mo_name) == sizeof(mo->mo_name));
308 1.128 matt CTASSERT(sizeof(mo_user->mo_descr) == sizeof(mo->mo_descr));
309 1.124 yamt memcpy(mo_user->mo_name, mo->mo_name, sizeof(mo->mo_name));
310 1.124 yamt memcpy(mo_user->mo_descr, mo->mo_descr, sizeof(mo->mo_descr));
311 1.124 yamt percpu_foreach(mo->mo_counters, mowner_conver_to_user_cb, mo_user);
312 1.124 yamt }
313 1.124 yamt
314 1.75 atatat static int
315 1.75 atatat sysctl_kern_mbuf_mowners(SYSCTLFN_ARGS)
316 1.75 atatat {
317 1.75 atatat struct mowner *mo;
318 1.75 atatat size_t len = 0;
319 1.75 atatat int error = 0;
320 1.75 atatat
321 1.75 atatat if (namelen != 0)
322 1.75 atatat return (EINVAL);
323 1.75 atatat if (newp != NULL)
324 1.75 atatat return (EPERM);
325 1.75 atatat
326 1.75 atatat LIST_FOREACH(mo, &mowners, mo_link) {
327 1.124 yamt struct mowner_user mo_user;
328 1.124 yamt
329 1.124 yamt mowner_convert_to_user(mo, &mo_user);
330 1.124 yamt
331 1.75 atatat if (oldp != NULL) {
332 1.124 yamt if (*oldlenp - len < sizeof(mo_user)) {
333 1.75 atatat error = ENOMEM;
334 1.75 atatat break;
335 1.75 atatat }
336 1.124 yamt error = copyout(&mo_user, (char *)oldp + len,
337 1.124 yamt sizeof(mo_user));
338 1.75 atatat if (error)
339 1.75 atatat break;
340 1.64 matt }
341 1.124 yamt len += sizeof(mo_user);
342 1.75 atatat }
343 1.75 atatat
344 1.75 atatat if (error == 0)
345 1.64 matt *oldlenp = len;
346 1.75 atatat
347 1.75 atatat return (error);
348 1.75 atatat }
349 1.75 atatat #endif /* MBUFTRACE */
350 1.75 atatat
351 1.124 yamt static void
352 1.124 yamt mbstat_conver_to_user_cb(void *v1, void *v2, struct cpu_info *ci)
353 1.124 yamt {
354 1.124 yamt struct mbstat_cpu *mbsc = v1;
355 1.124 yamt struct mbstat *mbs = v2;
356 1.124 yamt int i;
357 1.124 yamt
358 1.124 yamt for (i = 0; i < __arraycount(mbs->m_mtypes); i++) {
359 1.124 yamt mbs->m_mtypes[i] += mbsc->m_mtypes[i];
360 1.124 yamt }
361 1.124 yamt }
362 1.124 yamt
363 1.124 yamt static void
364 1.124 yamt mbstat_convert_to_user(struct mbstat *mbs)
365 1.124 yamt {
366 1.124 yamt
367 1.124 yamt memset(mbs, 0, sizeof(*mbs));
368 1.124 yamt mbs->m_drain = mbstat.m_drain;
369 1.124 yamt percpu_foreach(mbstat_percpu, mbstat_conver_to_user_cb, mbs);
370 1.124 yamt }
371 1.124 yamt
372 1.124 yamt static int
373 1.124 yamt sysctl_kern_mbuf_stats(SYSCTLFN_ARGS)
374 1.124 yamt {
375 1.124 yamt struct sysctlnode node;
376 1.124 yamt struct mbstat mbs;
377 1.124 yamt
378 1.124 yamt mbstat_convert_to_user(&mbs);
379 1.124 yamt node = *rnode;
380 1.124 yamt node.sysctl_data = &mbs;
381 1.124 yamt node.sysctl_size = sizeof(mbs);
382 1.124 yamt return sysctl_lookup(SYSCTLFN_CALL(&node));
383 1.124 yamt }
384 1.124 yamt
385 1.129 pooka static void
386 1.131 cegger sysctl_kern_mbuf_setup(void)
387 1.75 atatat {
388 1.75 atatat
389 1.129 pooka KASSERT(mbuf_sysctllog == NULL);
390 1.129 pooka sysctl_createv(&mbuf_sysctllog, 0, NULL, NULL,
391 1.80 atatat CTLFLAG_PERMANENT,
392 1.75 atatat CTLTYPE_NODE, "kern", NULL,
393 1.75 atatat NULL, 0, NULL, 0,
394 1.75 atatat CTL_KERN, CTL_EOL);
395 1.129 pooka sysctl_createv(&mbuf_sysctllog, 0, NULL, NULL,
396 1.80 atatat CTLFLAG_PERMANENT,
397 1.82 atatat CTLTYPE_NODE, "mbuf",
398 1.82 atatat SYSCTL_DESCR("mbuf control variables"),
399 1.75 atatat NULL, 0, NULL, 0,
400 1.75 atatat CTL_KERN, KERN_MBUF, CTL_EOL);
401 1.75 atatat
402 1.129 pooka sysctl_createv(&mbuf_sysctllog, 0, NULL, NULL,
403 1.80 atatat CTLFLAG_PERMANENT|CTLFLAG_IMMEDIATE,
404 1.82 atatat CTLTYPE_INT, "msize",
405 1.82 atatat SYSCTL_DESCR("mbuf base size"),
406 1.75 atatat NULL, msize, NULL, 0,
407 1.75 atatat CTL_KERN, KERN_MBUF, MBUF_MSIZE, CTL_EOL);
408 1.129 pooka sysctl_createv(&mbuf_sysctllog, 0, NULL, NULL,
409 1.80 atatat CTLFLAG_PERMANENT|CTLFLAG_IMMEDIATE,
410 1.82 atatat CTLTYPE_INT, "mclbytes",
411 1.82 atatat SYSCTL_DESCR("mbuf cluster size"),
412 1.75 atatat NULL, mclbytes, NULL, 0,
413 1.75 atatat CTL_KERN, KERN_MBUF, MBUF_MCLBYTES, CTL_EOL);
414 1.129 pooka sysctl_createv(&mbuf_sysctllog, 0, NULL, NULL,
415 1.80 atatat CTLFLAG_PERMANENT|CTLFLAG_READWRITE,
416 1.82 atatat CTLTYPE_INT, "nmbclusters",
417 1.82 atatat SYSCTL_DESCR("Limit on the number of mbuf clusters"),
418 1.75 atatat sysctl_kern_mbuf, 0, &nmbclusters, 0,
419 1.75 atatat CTL_KERN, KERN_MBUF, MBUF_NMBCLUSTERS, CTL_EOL);
420 1.129 pooka sysctl_createv(&mbuf_sysctllog, 0, NULL, NULL,
421 1.80 atatat CTLFLAG_PERMANENT|CTLFLAG_READWRITE,
422 1.82 atatat CTLTYPE_INT, "mblowat",
423 1.82 atatat SYSCTL_DESCR("mbuf low water mark"),
424 1.75 atatat sysctl_kern_mbuf, 0, &mblowat, 0,
425 1.75 atatat CTL_KERN, KERN_MBUF, MBUF_MBLOWAT, CTL_EOL);
426 1.129 pooka sysctl_createv(&mbuf_sysctllog, 0, NULL, NULL,
427 1.80 atatat CTLFLAG_PERMANENT|CTLFLAG_READWRITE,
428 1.82 atatat CTLTYPE_INT, "mcllowat",
429 1.82 atatat SYSCTL_DESCR("mbuf cluster low water mark"),
430 1.75 atatat sysctl_kern_mbuf, 0, &mcllowat, 0,
431 1.75 atatat CTL_KERN, KERN_MBUF, MBUF_MCLLOWAT, CTL_EOL);
432 1.129 pooka sysctl_createv(&mbuf_sysctllog, 0, NULL, NULL,
433 1.80 atatat CTLFLAG_PERMANENT,
434 1.82 atatat CTLTYPE_STRUCT, "stats",
435 1.82 atatat SYSCTL_DESCR("mbuf allocation statistics"),
436 1.124 yamt sysctl_kern_mbuf_stats, 0, NULL, 0,
437 1.75 atatat CTL_KERN, KERN_MBUF, MBUF_STATS, CTL_EOL);
438 1.75 atatat #ifdef MBUFTRACE
439 1.129 pooka sysctl_createv(&mbuf_sysctllog, 0, NULL, NULL,
440 1.80 atatat CTLFLAG_PERMANENT,
441 1.82 atatat CTLTYPE_STRUCT, "mowners",
442 1.82 atatat SYSCTL_DESCR("Information about mbuf owners"),
443 1.75 atatat sysctl_kern_mbuf_mowners, 0, NULL, 0,
444 1.75 atatat CTL_KERN, KERN_MBUF, MBUF_MOWNERS, CTL_EOL);
445 1.75 atatat #endif /* MBUFTRACE */
446 1.28 thorpej }
447 1.28 thorpej
448 1.65 thorpej static int
449 1.116 yamt mb_ctor(void *arg, void *object, int flags)
450 1.65 thorpej {
451 1.65 thorpej struct mbuf *m = object;
452 1.65 thorpej
453 1.65 thorpej #ifdef POOL_VTOPHYS
454 1.65 thorpej m->m_paddr = POOL_VTOPHYS(m);
455 1.65 thorpej #else
456 1.65 thorpej m->m_paddr = M_PADDR_INVALID;
457 1.65 thorpej #endif
458 1.65 thorpej return (0);
459 1.1 cgd }
460 1.1 cgd
461 1.150 christos /*
462 1.150 christos * Add mbuf to the end of a chain
463 1.150 christos */
464 1.150 christos struct mbuf *
465 1.150 christos m_add(struct mbuf *c, struct mbuf *m) {
466 1.150 christos struct mbuf *n;
467 1.150 christos
468 1.150 christos if (c == NULL)
469 1.150 christos return m;
470 1.150 christos
471 1.150 christos for (n = c; n->m_next != NULL; n = n->m_next)
472 1.150 christos continue;
473 1.150 christos n->m_next = m;
474 1.150 christos return c;
475 1.150 christos }
476 1.150 christos
477 1.150 christos /*
478 1.150 christos * Set the m_data pointer of a newly-allocated mbuf
479 1.150 christos * to place an object of the specified size at the
480 1.150 christos * end of the mbuf, longword aligned.
481 1.150 christos */
482 1.150 christos void
483 1.150 christos m_align(struct mbuf *m, int len)
484 1.150 christos {
485 1.150 christos int adjust;
486 1.150 christos
487 1.150 christos if (m->m_flags & M_EXT)
488 1.150 christos adjust = m->m_ext.ext_size - len;
489 1.150 christos else if (m->m_flags & M_PKTHDR)
490 1.150 christos adjust = MHLEN - len;
491 1.150 christos else
492 1.150 christos adjust = MLEN - len;
493 1.150 christos m->m_data += adjust &~ (sizeof(long)-1);
494 1.150 christos }
495 1.150 christos
496 1.150 christos /*
497 1.150 christos * Append the specified data to the indicated mbuf chain,
498 1.150 christos * Extend the mbuf chain if the new data does not fit in
499 1.150 christos * existing space.
500 1.150 christos *
501 1.150 christos * Return 1 if able to complete the job; otherwise 0.
502 1.150 christos */
503 1.150 christos int
504 1.150 christos m_append(struct mbuf *m0, int len, const void *cpv)
505 1.150 christos {
506 1.150 christos struct mbuf *m, *n;
507 1.150 christos int remainder, space;
508 1.150 christos const char *cp = cpv;
509 1.150 christos
510 1.150 christos for (m = m0; m->m_next != NULL; m = m->m_next)
511 1.150 christos continue;
512 1.150 christos remainder = len;
513 1.150 christos space = M_TRAILINGSPACE(m);
514 1.150 christos if (space > 0) {
515 1.150 christos /*
516 1.150 christos * Copy into available space.
517 1.150 christos */
518 1.150 christos if (space > remainder)
519 1.150 christos space = remainder;
520 1.150 christos memmove(mtod(m, char *) + m->m_len, cp, space);
521 1.150 christos m->m_len += space;
522 1.150 christos cp = cp + space, remainder -= space;
523 1.150 christos }
524 1.150 christos while (remainder > 0) {
525 1.150 christos /*
526 1.150 christos * Allocate a new mbuf; could check space
527 1.150 christos * and allocate a cluster instead.
528 1.150 christos */
529 1.150 christos n = m_get(M_DONTWAIT, m->m_type);
530 1.150 christos if (n == NULL)
531 1.150 christos break;
532 1.150 christos n->m_len = min(MLEN, remainder);
533 1.150 christos memmove(mtod(n, void *), cp, n->m_len);
534 1.150 christos cp += n->m_len, remainder -= n->m_len;
535 1.150 christos m->m_next = n;
536 1.150 christos m = n;
537 1.150 christos }
538 1.150 christos if (m0->m_flags & M_PKTHDR)
539 1.150 christos m0->m_pkthdr.len += len - remainder;
540 1.150 christos return (remainder == 0);
541 1.150 christos }
542 1.150 christos
543 1.14 christos void
544 1.116 yamt m_reclaim(void *arg, int flags)
545 1.1 cgd {
546 1.27 matt struct domain *dp;
547 1.81 matt const struct protosw *pr;
548 1.27 matt struct ifnet *ifp;
549 1.122 ad int s;
550 1.1 cgd
551 1.122 ad KERNEL_LOCK(1, NULL);
552 1.122 ad s = splvm();
553 1.91 matt DOMAIN_FOREACH(dp) {
554 1.33 thorpej for (pr = dp->dom_protosw;
555 1.33 thorpej pr < dp->dom_protoswNPROTOSW; pr++)
556 1.33 thorpej if (pr->pr_drain)
557 1.33 thorpej (*pr->pr_drain)();
558 1.91 matt }
559 1.92 matt IFNET_FOREACH(ifp) {
560 1.27 matt if (ifp->if_drain)
561 1.27 matt (*ifp->if_drain)(ifp);
562 1.92 matt }
563 1.1 cgd splx(s);
564 1.1 cgd mbstat.m_drain++;
565 1.122 ad KERNEL_UNLOCK_ONE(NULL);
566 1.1 cgd }
567 1.1 cgd
568 1.1 cgd /*
569 1.1 cgd * Space allocation routines.
570 1.1 cgd * These are also available as macros
571 1.1 cgd * for critical paths.
572 1.1 cgd */
573 1.1 cgd struct mbuf *
574 1.62 thorpej m_get(int nowait, int type)
575 1.1 cgd {
576 1.27 matt struct mbuf *m;
577 1.1 cgd
578 1.142 dyoung KASSERT(type != MT_FREE);
579 1.142 dyoung
580 1.124 yamt m = pool_cache_get(mb_cache,
581 1.124 yamt nowait == M_WAIT ? PR_WAITOK|PR_LIMITFAIL : 0);
582 1.124 yamt if (m == NULL)
583 1.124 yamt return NULL;
584 1.124 yamt
585 1.124 yamt mbstat_type_add(type, 1);
586 1.124 yamt mowner_init(m, type);
587 1.125 yamt m->m_ext_ref = m;
588 1.124 yamt m->m_type = type;
589 1.153 christos m->m_len = 0;
590 1.124 yamt m->m_next = NULL;
591 1.124 yamt m->m_nextpkt = NULL;
592 1.124 yamt m->m_data = m->m_dat;
593 1.124 yamt m->m_flags = 0;
594 1.124 yamt
595 1.124 yamt return m;
596 1.1 cgd }
597 1.1 cgd
598 1.1 cgd struct mbuf *
599 1.62 thorpej m_gethdr(int nowait, int type)
600 1.1 cgd {
601 1.27 matt struct mbuf *m;
602 1.1 cgd
603 1.124 yamt m = m_get(nowait, type);
604 1.124 yamt if (m == NULL)
605 1.124 yamt return NULL;
606 1.124 yamt
607 1.124 yamt m->m_data = m->m_pktdat;
608 1.124 yamt m->m_flags = M_PKTHDR;
609 1.124 yamt m->m_pkthdr.rcvif = NULL;
610 1.153 christos m->m_pkthdr.len = 0;
611 1.124 yamt m->m_pkthdr.csum_flags = 0;
612 1.124 yamt m->m_pkthdr.csum_data = 0;
613 1.124 yamt SLIST_INIT(&m->m_pkthdr.tags);
614 1.124 yamt
615 1.124 yamt return m;
616 1.1 cgd }
617 1.1 cgd
618 1.1 cgd struct mbuf *
619 1.62 thorpej m_getclr(int nowait, int type)
620 1.1 cgd {
621 1.27 matt struct mbuf *m;
622 1.1 cgd
623 1.153 christos m = m_get(nowait, type);
624 1.1 cgd if (m == 0)
625 1.71 simonb return (NULL);
626 1.119 christos memset(mtod(m, void *), 0, MLEN);
627 1.1 cgd return (m);
628 1.1 cgd }
629 1.1 cgd
630 1.64 matt void
631 1.64 matt m_clget(struct mbuf *m, int nowait)
632 1.64 matt {
633 1.71 simonb
634 1.64 matt MCLGET(m, nowait);
635 1.64 matt }
636 1.64 matt
637 1.1 cgd struct mbuf *
638 1.62 thorpej m_free(struct mbuf *m)
639 1.1 cgd {
640 1.27 matt struct mbuf *n;
641 1.1 cgd
642 1.1 cgd MFREE(m, n);
643 1.1 cgd return (n);
644 1.1 cgd }
645 1.1 cgd
646 1.9 mycroft void
647 1.62 thorpej m_freem(struct mbuf *m)
648 1.1 cgd {
649 1.27 matt struct mbuf *n;
650 1.1 cgd
651 1.1 cgd if (m == NULL)
652 1.1 cgd return;
653 1.1 cgd do {
654 1.1 cgd MFREE(m, n);
655 1.18 thorpej m = n;
656 1.18 thorpej } while (m);
657 1.1 cgd }
658 1.1 cgd
659 1.64 matt #ifdef MBUFTRACE
660 1.83 jonathan /*
661 1.83 jonathan * Walk a chain of mbufs, claiming ownership of each mbuf in the chain.
662 1.83 jonathan */
663 1.64 matt void
664 1.83 jonathan m_claimm(struct mbuf *m, struct mowner *mo)
665 1.64 matt {
666 1.71 simonb
667 1.64 matt for (; m != NULL; m = m->m_next)
668 1.64 matt MCLAIM(m, mo);
669 1.64 matt }
670 1.64 matt #endif
671 1.64 matt
672 1.1 cgd /*
673 1.1 cgd * Mbuffer utility routines.
674 1.1 cgd */
675 1.1 cgd
676 1.1 cgd /*
677 1.1 cgd * Lesser-used path for M_PREPEND:
678 1.1 cgd * allocate new mbuf to prepend to chain,
679 1.1 cgd * copy junk along.
680 1.1 cgd */
681 1.1 cgd struct mbuf *
682 1.62 thorpej m_prepend(struct mbuf *m, int len, int how)
683 1.1 cgd {
684 1.1 cgd struct mbuf *mn;
685 1.1 cgd
686 1.153 christos mn = m_get(how, m->m_type);
687 1.143 plunky if (mn == NULL) {
688 1.1 cgd m_freem(m);
689 1.143 plunky return (NULL);
690 1.1 cgd }
691 1.1 cgd if (m->m_flags & M_PKTHDR) {
692 1.101 yamt M_MOVE_PKTHDR(mn, m);
693 1.64 matt } else {
694 1.64 matt MCLAIM(mn, m->m_owner);
695 1.1 cgd }
696 1.1 cgd mn->m_next = m;
697 1.1 cgd m = mn;
698 1.1 cgd if (len < MHLEN)
699 1.1 cgd MH_ALIGN(m, len);
700 1.1 cgd m->m_len = len;
701 1.1 cgd return (m);
702 1.1 cgd }
703 1.1 cgd
704 1.1 cgd /*
705 1.1 cgd * Make a copy of an mbuf chain starting "off0" bytes from the beginning,
706 1.1 cgd * continuing for "len" bytes. If len is M_COPYALL, copy to end of mbuf.
707 1.1 cgd * The wait parameter is a choice of M_WAIT/M_DONTWAIT from caller.
708 1.1 cgd */
709 1.1 cgd int MCFail;
710 1.1 cgd
711 1.1 cgd struct mbuf *
712 1.62 thorpej m_copym(struct mbuf *m, int off0, int len, int wait)
713 1.1 cgd {
714 1.71 simonb
715 1.44 itojun return m_copym0(m, off0, len, wait, 0); /* shallow copy on M_EXT */
716 1.44 itojun }
717 1.44 itojun
718 1.44 itojun struct mbuf *
719 1.62 thorpej m_dup(struct mbuf *m, int off0, int len, int wait)
720 1.44 itojun {
721 1.71 simonb
722 1.44 itojun return m_copym0(m, off0, len, wait, 1); /* deep copy */
723 1.44 itojun }
724 1.44 itojun
725 1.44 itojun static struct mbuf *
726 1.62 thorpej m_copym0(struct mbuf *m, int off0, int len, int wait, int deep)
727 1.44 itojun {
728 1.27 matt struct mbuf *n, **np;
729 1.27 matt int off = off0;
730 1.1 cgd struct mbuf *top;
731 1.1 cgd int copyhdr = 0;
732 1.1 cgd
733 1.1 cgd if (off < 0 || len < 0)
734 1.43 thorpej panic("m_copym: off %d, len %d", off, len);
735 1.1 cgd if (off == 0 && m->m_flags & M_PKTHDR)
736 1.1 cgd copyhdr = 1;
737 1.1 cgd while (off > 0) {
738 1.1 cgd if (m == 0)
739 1.90 matt panic("m_copym: m == 0, off %d", off);
740 1.1 cgd if (off < m->m_len)
741 1.1 cgd break;
742 1.1 cgd off -= m->m_len;
743 1.1 cgd m = m->m_next;
744 1.1 cgd }
745 1.1 cgd np = ⊤
746 1.1 cgd top = 0;
747 1.1 cgd while (len > 0) {
748 1.1 cgd if (m == 0) {
749 1.1 cgd if (len != M_COPYALL)
750 1.90 matt panic("m_copym: m == 0, len %d [!COPYALL]",
751 1.90 matt len);
752 1.1 cgd break;
753 1.1 cgd }
754 1.153 christos n = m_get(wait, m->m_type);
755 1.1 cgd *np = n;
756 1.1 cgd if (n == 0)
757 1.1 cgd goto nospace;
758 1.64 matt MCLAIM(n, m->m_owner);
759 1.1 cgd if (copyhdr) {
760 1.1 cgd M_COPY_PKTHDR(n, m);
761 1.1 cgd if (len == M_COPYALL)
762 1.1 cgd n->m_pkthdr.len -= off0;
763 1.1 cgd else
764 1.1 cgd n->m_pkthdr.len = len;
765 1.1 cgd copyhdr = 0;
766 1.1 cgd }
767 1.9 mycroft n->m_len = min(len, m->m_len - off);
768 1.1 cgd if (m->m_flags & M_EXT) {
769 1.44 itojun if (!deep) {
770 1.44 itojun n->m_data = m->m_data + off;
771 1.44 itojun MCLADDREFERENCE(m, n);
772 1.44 itojun } else {
773 1.48 itojun /*
774 1.50 itojun * we are unsure about the way m was allocated.
775 1.50 itojun * copy into multiple MCLBYTES cluster mbufs.
776 1.48 itojun */
777 1.44 itojun MCLGET(n, wait);
778 1.50 itojun n->m_len = M_TRAILINGSPACE(n);
779 1.50 itojun n->m_len = min(n->m_len, len);
780 1.50 itojun n->m_len = min(n->m_len, m->m_len - off);
781 1.119 christos memcpy(mtod(n, void *), mtod(m, char *) + off,
782 1.44 itojun (unsigned)n->m_len);
783 1.44 itojun }
784 1.1 cgd } else
785 1.119 christos memcpy(mtod(n, void *), mtod(m, char *) + off,
786 1.1 cgd (unsigned)n->m_len);
787 1.1 cgd if (len != M_COPYALL)
788 1.1 cgd len -= n->m_len;
789 1.50 itojun off += n->m_len;
790 1.50 itojun #ifdef DIAGNOSTIC
791 1.50 itojun if (off > m->m_len)
792 1.149 pooka panic("m_copym0 overrun %d %d", off, m->m_len);
793 1.50 itojun #endif
794 1.50 itojun if (off == m->m_len) {
795 1.50 itojun m = m->m_next;
796 1.50 itojun off = 0;
797 1.50 itojun }
798 1.1 cgd np = &n->m_next;
799 1.1 cgd }
800 1.1 cgd if (top == 0)
801 1.1 cgd MCFail++;
802 1.1 cgd return (top);
803 1.1 cgd nospace:
804 1.1 cgd m_freem(top);
805 1.1 cgd MCFail++;
806 1.71 simonb return (NULL);
807 1.1 cgd }
808 1.1 cgd
809 1.1 cgd /*
810 1.18 thorpej * Copy an entire packet, including header (which must be present).
811 1.18 thorpej * An optimization of the common case `m_copym(m, 0, M_COPYALL, how)'.
812 1.18 thorpej */
813 1.18 thorpej struct mbuf *
814 1.62 thorpej m_copypacket(struct mbuf *m, int how)
815 1.18 thorpej {
816 1.18 thorpej struct mbuf *top, *n, *o;
817 1.18 thorpej
818 1.153 christos n = m_get(how, m->m_type);
819 1.18 thorpej top = n;
820 1.18 thorpej if (!n)
821 1.18 thorpej goto nospace;
822 1.18 thorpej
823 1.64 matt MCLAIM(n, m->m_owner);
824 1.18 thorpej M_COPY_PKTHDR(n, m);
825 1.18 thorpej n->m_len = m->m_len;
826 1.18 thorpej if (m->m_flags & M_EXT) {
827 1.18 thorpej n->m_data = m->m_data;
828 1.18 thorpej MCLADDREFERENCE(m, n);
829 1.18 thorpej } else {
830 1.30 perry memcpy(mtod(n, char *), mtod(m, char *), n->m_len);
831 1.18 thorpej }
832 1.18 thorpej
833 1.18 thorpej m = m->m_next;
834 1.18 thorpej while (m) {
835 1.153 christos o = m_get(how, m->m_type);
836 1.18 thorpej if (!o)
837 1.18 thorpej goto nospace;
838 1.18 thorpej
839 1.64 matt MCLAIM(o, m->m_owner);
840 1.18 thorpej n->m_next = o;
841 1.18 thorpej n = n->m_next;
842 1.18 thorpej
843 1.18 thorpej n->m_len = m->m_len;
844 1.18 thorpej if (m->m_flags & M_EXT) {
845 1.18 thorpej n->m_data = m->m_data;
846 1.18 thorpej MCLADDREFERENCE(m, n);
847 1.18 thorpej } else {
848 1.30 perry memcpy(mtod(n, char *), mtod(m, char *), n->m_len);
849 1.18 thorpej }
850 1.18 thorpej
851 1.18 thorpej m = m->m_next;
852 1.18 thorpej }
853 1.18 thorpej return top;
854 1.18 thorpej nospace:
855 1.18 thorpej m_freem(top);
856 1.18 thorpej MCFail++;
857 1.71 simonb return NULL;
858 1.18 thorpej }
859 1.18 thorpej
860 1.18 thorpej /*
861 1.1 cgd * Copy data from an mbuf chain starting "off" bytes from the beginning,
862 1.1 cgd * continuing for "len" bytes, into the indicated buffer.
863 1.1 cgd */
864 1.14 christos void
865 1.100 martin m_copydata(struct mbuf *m, int off, int len, void *vp)
866 1.1 cgd {
867 1.94 tron unsigned count;
868 1.119 christos void * cp = vp;
869 1.151 matt struct mbuf *m0 = m;
870 1.151 matt int len0 = len;
871 1.151 matt int off0 = off;
872 1.151 matt void *vp0 = vp;
873 1.1 cgd
874 1.1 cgd if (off < 0 || len < 0)
875 1.90 matt panic("m_copydata: off %d, len %d", off, len);
876 1.1 cgd while (off > 0) {
877 1.94 tron if (m == NULL)
878 1.151 matt panic("m_copydata(%p,%d,%d,%p): m=NULL, off=%d (%d)",
879 1.151 matt m0, len0, off0, vp0, off, off0 - off);
880 1.1 cgd if (off < m->m_len)
881 1.1 cgd break;
882 1.1 cgd off -= m->m_len;
883 1.1 cgd m = m->m_next;
884 1.1 cgd }
885 1.1 cgd while (len > 0) {
886 1.94 tron if (m == NULL)
887 1.151 matt panic("m_copydata(%p,%d,%d,%p): "
888 1.151 matt "m=NULL, off=%d (%d), len=%d (%d)",
889 1.151 matt m0, len0, off0, vp0,
890 1.151 matt off, off0 - off, len, len0 - len);
891 1.9 mycroft count = min(m->m_len - off, len);
892 1.119 christos memcpy(cp, mtod(m, char *) + off, count);
893 1.1 cgd len -= count;
894 1.119 christos cp = (char *)cp + count;
895 1.1 cgd off = 0;
896 1.1 cgd m = m->m_next;
897 1.1 cgd }
898 1.1 cgd }
899 1.1 cgd
900 1.1 cgd /*
901 1.1 cgd * Concatenate mbuf chain n to m.
902 1.72 itojun * n might be copied into m (when n->m_len is small), therefore data portion of
903 1.72 itojun * n could be copied into an mbuf of different mbuf type.
904 1.1 cgd * Any m_pkthdr is not updated.
905 1.1 cgd */
906 1.14 christos void
907 1.62 thorpej m_cat(struct mbuf *m, struct mbuf *n)
908 1.1 cgd {
909 1.73 yamt
910 1.1 cgd while (m->m_next)
911 1.1 cgd m = m->m_next;
912 1.1 cgd while (n) {
913 1.77 itojun if (M_READONLY(m) || n->m_len > M_TRAILINGSPACE(m)) {
914 1.1 cgd /* just join the two chains */
915 1.1 cgd m->m_next = n;
916 1.1 cgd return;
917 1.1 cgd }
918 1.1 cgd /* splat the data from one into the other */
919 1.119 christos memcpy(mtod(m, char *) + m->m_len, mtod(n, void *),
920 1.1 cgd (u_int)n->m_len);
921 1.1 cgd m->m_len += n->m_len;
922 1.1 cgd n = m_free(n);
923 1.1 cgd }
924 1.1 cgd }
925 1.1 cgd
926 1.11 mycroft void
927 1.62 thorpej m_adj(struct mbuf *mp, int req_len)
928 1.1 cgd {
929 1.27 matt int len = req_len;
930 1.27 matt struct mbuf *m;
931 1.27 matt int count;
932 1.1 cgd
933 1.1 cgd if ((m = mp) == NULL)
934 1.1 cgd return;
935 1.1 cgd if (len >= 0) {
936 1.1 cgd /*
937 1.1 cgd * Trim from head.
938 1.1 cgd */
939 1.1 cgd while (m != NULL && len > 0) {
940 1.1 cgd if (m->m_len <= len) {
941 1.1 cgd len -= m->m_len;
942 1.1 cgd m->m_len = 0;
943 1.1 cgd m = m->m_next;
944 1.1 cgd } else {
945 1.1 cgd m->m_len -= len;
946 1.1 cgd m->m_data += len;
947 1.1 cgd len = 0;
948 1.1 cgd }
949 1.1 cgd }
950 1.1 cgd m = mp;
951 1.1 cgd if (mp->m_flags & M_PKTHDR)
952 1.1 cgd m->m_pkthdr.len -= (req_len - len);
953 1.1 cgd } else {
954 1.1 cgd /*
955 1.1 cgd * Trim from tail. Scan the mbuf chain,
956 1.1 cgd * calculating its length and finding the last mbuf.
957 1.1 cgd * If the adjustment only affects this mbuf, then just
958 1.1 cgd * adjust and return. Otherwise, rescan and truncate
959 1.1 cgd * after the remaining size.
960 1.1 cgd */
961 1.1 cgd len = -len;
962 1.1 cgd count = 0;
963 1.1 cgd for (;;) {
964 1.1 cgd count += m->m_len;
965 1.1 cgd if (m->m_next == (struct mbuf *)0)
966 1.1 cgd break;
967 1.1 cgd m = m->m_next;
968 1.1 cgd }
969 1.1 cgd if (m->m_len >= len) {
970 1.1 cgd m->m_len -= len;
971 1.8 deraadt if (mp->m_flags & M_PKTHDR)
972 1.8 deraadt mp->m_pkthdr.len -= len;
973 1.1 cgd return;
974 1.1 cgd }
975 1.1 cgd count -= len;
976 1.1 cgd if (count < 0)
977 1.1 cgd count = 0;
978 1.1 cgd /*
979 1.1 cgd * Correct length for chain is "count".
980 1.1 cgd * Find the mbuf with last data, adjust its length,
981 1.1 cgd * and toss data from remaining mbufs on chain.
982 1.1 cgd */
983 1.1 cgd m = mp;
984 1.1 cgd if (m->m_flags & M_PKTHDR)
985 1.1 cgd m->m_pkthdr.len = count;
986 1.1 cgd for (; m; m = m->m_next) {
987 1.1 cgd if (m->m_len >= count) {
988 1.1 cgd m->m_len = count;
989 1.1 cgd break;
990 1.1 cgd }
991 1.1 cgd count -= m->m_len;
992 1.1 cgd }
993 1.110 christos if (m)
994 1.110 christos while (m->m_next)
995 1.110 christos (m = m->m_next)->m_len = 0;
996 1.1 cgd }
997 1.1 cgd }
998 1.1 cgd
999 1.1 cgd /*
1000 1.148 rmind * m_ensure_contig: rearrange an mbuf chain that given length of bytes
1001 1.148 rmind * would be contiguous and in the data area of an mbuf (therefore, mtod()
1002 1.148 rmind * would work for a structure of given length).
1003 1.148 rmind *
1004 1.148 rmind * => On success, returns true and the resulting mbuf chain; false otherwise.
1005 1.148 rmind * => The mbuf chain may change, but is always preserved valid.
1006 1.1 cgd */
1007 1.148 rmind bool
1008 1.148 rmind m_ensure_contig(struct mbuf **m0, int len)
1009 1.1 cgd {
1010 1.148 rmind struct mbuf *n = *m0, *m;
1011 1.148 rmind size_t count, space;
1012 1.1 cgd
1013 1.1 cgd /*
1014 1.1 cgd * If first mbuf has no cluster, and has room for len bytes
1015 1.1 cgd * without shifting current data, pullup into it,
1016 1.1 cgd * otherwise allocate a new mbuf to prepend to the chain.
1017 1.1 cgd */
1018 1.1 cgd if ((n->m_flags & M_EXT) == 0 &&
1019 1.1 cgd n->m_data + len < &n->m_dat[MLEN] && n->m_next) {
1020 1.148 rmind if (n->m_len >= len) {
1021 1.148 rmind return true;
1022 1.148 rmind }
1023 1.1 cgd m = n;
1024 1.1 cgd n = n->m_next;
1025 1.1 cgd len -= m->m_len;
1026 1.1 cgd } else {
1027 1.148 rmind if (len > MHLEN) {
1028 1.148 rmind return false;
1029 1.148 rmind }
1030 1.153 christos m = m_get(M_DONTWAIT, n->m_type);
1031 1.148 rmind if (m == NULL) {
1032 1.148 rmind return false;
1033 1.148 rmind }
1034 1.64 matt MCLAIM(m, n->m_owner);
1035 1.1 cgd if (n->m_flags & M_PKTHDR) {
1036 1.101 yamt M_MOVE_PKTHDR(m, n);
1037 1.1 cgd }
1038 1.1 cgd }
1039 1.1 cgd space = &m->m_dat[MLEN] - (m->m_data + m->m_len);
1040 1.1 cgd do {
1041 1.148 rmind count = MIN(MIN(MAX(len, max_protohdr), space), n->m_len);
1042 1.119 christos memcpy(mtod(m, char *) + m->m_len, mtod(n, void *),
1043 1.1 cgd (unsigned)count);
1044 1.1 cgd len -= count;
1045 1.1 cgd m->m_len += count;
1046 1.1 cgd n->m_len -= count;
1047 1.1 cgd space -= count;
1048 1.1 cgd if (n->m_len)
1049 1.1 cgd n->m_data += count;
1050 1.1 cgd else
1051 1.1 cgd n = m_free(n);
1052 1.1 cgd } while (len > 0 && n);
1053 1.148 rmind
1054 1.148 rmind m->m_next = n;
1055 1.148 rmind *m0 = m;
1056 1.148 rmind
1057 1.148 rmind return len <= 0;
1058 1.148 rmind }
1059 1.148 rmind
1060 1.148 rmind /*
1061 1.148 rmind * m_pullup: same as m_ensure_contig(), but destroys mbuf chain on error.
1062 1.148 rmind */
1063 1.148 rmind int MPFail;
1064 1.148 rmind
1065 1.148 rmind struct mbuf *
1066 1.148 rmind m_pullup(struct mbuf *n, int len)
1067 1.148 rmind {
1068 1.148 rmind struct mbuf *m = n;
1069 1.148 rmind
1070 1.148 rmind if (!m_ensure_contig(&m, len)) {
1071 1.148 rmind KASSERT(m != NULL);
1072 1.148 rmind m_freem(m);
1073 1.148 rmind MPFail++;
1074 1.148 rmind m = NULL;
1075 1.1 cgd }
1076 1.148 rmind return m;
1077 1.60 thorpej }
1078 1.60 thorpej
1079 1.60 thorpej /*
1080 1.60 thorpej * Like m_pullup(), except a new mbuf is always allocated, and we allow
1081 1.60 thorpej * the amount of empty space before the data in the new mbuf to be specified
1082 1.60 thorpej * (in the event that the caller expects to prepend later).
1083 1.60 thorpej */
1084 1.60 thorpej int MSFail;
1085 1.60 thorpej
1086 1.60 thorpej struct mbuf *
1087 1.60 thorpej m_copyup(struct mbuf *n, int len, int dstoff)
1088 1.60 thorpej {
1089 1.60 thorpej struct mbuf *m;
1090 1.60 thorpej int count, space;
1091 1.60 thorpej
1092 1.60 thorpej if (len > (MHLEN - dstoff))
1093 1.60 thorpej goto bad;
1094 1.153 christos m = m_get(M_DONTWAIT, n->m_type);
1095 1.60 thorpej if (m == NULL)
1096 1.60 thorpej goto bad;
1097 1.64 matt MCLAIM(m, n->m_owner);
1098 1.60 thorpej if (n->m_flags & M_PKTHDR) {
1099 1.101 yamt M_MOVE_PKTHDR(m, n);
1100 1.60 thorpej }
1101 1.60 thorpej m->m_data += dstoff;
1102 1.60 thorpej space = &m->m_dat[MLEN] - (m->m_data + m->m_len);
1103 1.60 thorpej do {
1104 1.60 thorpej count = min(min(max(len, max_protohdr), space), n->m_len);
1105 1.119 christos memcpy(mtod(m, char *) + m->m_len, mtod(n, void *),
1106 1.60 thorpej (unsigned)count);
1107 1.60 thorpej len -= count;
1108 1.60 thorpej m->m_len += count;
1109 1.60 thorpej n->m_len -= count;
1110 1.60 thorpej space -= count;
1111 1.60 thorpej if (n->m_len)
1112 1.60 thorpej n->m_data += count;
1113 1.60 thorpej else
1114 1.60 thorpej n = m_free(n);
1115 1.60 thorpej } while (len > 0 && n);
1116 1.60 thorpej if (len > 0) {
1117 1.60 thorpej (void) m_free(m);
1118 1.60 thorpej goto bad;
1119 1.60 thorpej }
1120 1.60 thorpej m->m_next = n;
1121 1.60 thorpej return (m);
1122 1.60 thorpej bad:
1123 1.60 thorpej m_freem(n);
1124 1.60 thorpej MSFail++;
1125 1.60 thorpej return (NULL);
1126 1.9 mycroft }
1127 1.9 mycroft
1128 1.9 mycroft /*
1129 1.9 mycroft * Partition an mbuf chain in two pieces, returning the tail --
1130 1.9 mycroft * all but the first len0 bytes. In case of failure, it returns NULL and
1131 1.9 mycroft * attempts to restore the chain to its original state.
1132 1.9 mycroft */
1133 1.9 mycroft struct mbuf *
1134 1.62 thorpej m_split(struct mbuf *m0, int len0, int wait)
1135 1.9 mycroft {
1136 1.85 yamt
1137 1.87 yamt return m_split0(m0, len0, wait, 1);
1138 1.85 yamt }
1139 1.85 yamt
1140 1.85 yamt static struct mbuf *
1141 1.85 yamt m_split0(struct mbuf *m0, int len0, int wait, int copyhdr)
1142 1.85 yamt {
1143 1.27 matt struct mbuf *m, *n;
1144 1.22 thorpej unsigned len = len0, remain, len_save;
1145 1.9 mycroft
1146 1.9 mycroft for (m = m0; m && len > m->m_len; m = m->m_next)
1147 1.9 mycroft len -= m->m_len;
1148 1.9 mycroft if (m == 0)
1149 1.71 simonb return (NULL);
1150 1.9 mycroft remain = m->m_len - len;
1151 1.85 yamt if (copyhdr && (m0->m_flags & M_PKTHDR)) {
1152 1.153 christos n = m_gethdr(wait, m0->m_type);
1153 1.153 christos if (n == NULL)
1154 1.153 christos return NULL;
1155 1.112 pavel MCLAIM(n, m0->m_owner);
1156 1.9 mycroft n->m_pkthdr.rcvif = m0->m_pkthdr.rcvif;
1157 1.9 mycroft n->m_pkthdr.len = m0->m_pkthdr.len - len0;
1158 1.22 thorpej len_save = m0->m_pkthdr.len;
1159 1.9 mycroft m0->m_pkthdr.len = len0;
1160 1.9 mycroft if (m->m_flags & M_EXT)
1161 1.9 mycroft goto extpacket;
1162 1.9 mycroft if (remain > MHLEN) {
1163 1.9 mycroft /* m can't be the lead packet */
1164 1.9 mycroft MH_ALIGN(n, 0);
1165 1.132 bouyer n->m_len = 0;
1166 1.9 mycroft n->m_next = m_split(m, len, wait);
1167 1.9 mycroft if (n->m_next == 0) {
1168 1.9 mycroft (void) m_free(n);
1169 1.22 thorpej m0->m_pkthdr.len = len_save;
1170 1.71 simonb return (NULL);
1171 1.9 mycroft } else
1172 1.9 mycroft return (n);
1173 1.9 mycroft } else
1174 1.9 mycroft MH_ALIGN(n, remain);
1175 1.9 mycroft } else if (remain == 0) {
1176 1.9 mycroft n = m->m_next;
1177 1.9 mycroft m->m_next = 0;
1178 1.9 mycroft return (n);
1179 1.9 mycroft } else {
1180 1.153 christos n = m_get(wait, m->m_type);
1181 1.9 mycroft if (n == 0)
1182 1.71 simonb return (NULL);
1183 1.64 matt MCLAIM(n, m->m_owner);
1184 1.9 mycroft M_ALIGN(n, remain);
1185 1.9 mycroft }
1186 1.9 mycroft extpacket:
1187 1.9 mycroft if (m->m_flags & M_EXT) {
1188 1.125 yamt n->m_data = m->m_data + len;
1189 1.18 thorpej MCLADDREFERENCE(m, n);
1190 1.9 mycroft } else {
1191 1.119 christos memcpy(mtod(n, void *), mtod(m, char *) + len, remain);
1192 1.9 mycroft }
1193 1.9 mycroft n->m_len = remain;
1194 1.9 mycroft m->m_len = len;
1195 1.9 mycroft n->m_next = m->m_next;
1196 1.9 mycroft m->m_next = 0;
1197 1.9 mycroft return (n);
1198 1.9 mycroft }
1199 1.9 mycroft /*
1200 1.9 mycroft * Routine to copy from device local memory into mbufs.
1201 1.9 mycroft */
1202 1.9 mycroft struct mbuf *
1203 1.62 thorpej m_devget(char *buf, int totlen, int off0, struct ifnet *ifp,
1204 1.62 thorpej void (*copy)(const void *from, void *to, size_t len))
1205 1.9 mycroft {
1206 1.27 matt struct mbuf *m;
1207 1.9 mycroft struct mbuf *top = 0, **mp = ⊤
1208 1.27 matt int off = off0, len;
1209 1.27 matt char *cp;
1210 1.9 mycroft char *epkt;
1211 1.9 mycroft
1212 1.9 mycroft cp = buf;
1213 1.9 mycroft epkt = cp + totlen;
1214 1.9 mycroft if (off) {
1215 1.13 cgd /*
1216 1.13 cgd * If 'off' is non-zero, packet is trailer-encapsulated,
1217 1.13 cgd * so we have to skip the type and length fields.
1218 1.13 cgd */
1219 1.104 perry cp += off + 2 * sizeof(uint16_t);
1220 1.104 perry totlen -= 2 * sizeof(uint16_t);
1221 1.9 mycroft }
1222 1.153 christos m = m_gethdr(M_DONTWAIT, MT_DATA);
1223 1.153 christos if (m == NULL)
1224 1.153 christos return NULL;
1225 1.9 mycroft m->m_pkthdr.rcvif = ifp;
1226 1.9 mycroft m->m_pkthdr.len = totlen;
1227 1.9 mycroft m->m_len = MHLEN;
1228 1.9 mycroft
1229 1.9 mycroft while (totlen > 0) {
1230 1.9 mycroft if (top) {
1231 1.153 christos m = m_get(M_DONTWAIT, MT_DATA);
1232 1.9 mycroft if (m == 0) {
1233 1.9 mycroft m_freem(top);
1234 1.71 simonb return (NULL);
1235 1.9 mycroft }
1236 1.9 mycroft m->m_len = MLEN;
1237 1.9 mycroft }
1238 1.9 mycroft len = min(totlen, epkt - cp);
1239 1.9 mycroft if (len >= MINCLSIZE) {
1240 1.9 mycroft MCLGET(m, M_DONTWAIT);
1241 1.19 mycroft if ((m->m_flags & M_EXT) == 0) {
1242 1.20 mycroft m_free(m);
1243 1.19 mycroft m_freem(top);
1244 1.71 simonb return (NULL);
1245 1.19 mycroft }
1246 1.19 mycroft m->m_len = len = min(len, MCLBYTES);
1247 1.9 mycroft } else {
1248 1.9 mycroft /*
1249 1.9 mycroft * Place initial small packet/header at end of mbuf.
1250 1.9 mycroft */
1251 1.9 mycroft if (len < m->m_len) {
1252 1.9 mycroft if (top == 0 && len + max_linkhdr <= m->m_len)
1253 1.9 mycroft m->m_data += max_linkhdr;
1254 1.9 mycroft m->m_len = len;
1255 1.9 mycroft } else
1256 1.9 mycroft len = m->m_len;
1257 1.9 mycroft }
1258 1.9 mycroft if (copy)
1259 1.119 christos copy(cp, mtod(m, void *), (size_t)len);
1260 1.9 mycroft else
1261 1.119 christos memcpy(mtod(m, void *), cp, (size_t)len);
1262 1.9 mycroft cp += len;
1263 1.9 mycroft *mp = m;
1264 1.9 mycroft mp = &m->m_next;
1265 1.9 mycroft totlen -= len;
1266 1.9 mycroft if (cp == epkt)
1267 1.9 mycroft cp = buf;
1268 1.9 mycroft }
1269 1.9 mycroft return (top);
1270 1.18 thorpej }
1271 1.18 thorpej
1272 1.18 thorpej /*
1273 1.18 thorpej * Copy data from a buffer back into the indicated mbuf chain,
1274 1.18 thorpej * starting "off" bytes from the beginning, extending the mbuf
1275 1.18 thorpej * chain if necessary.
1276 1.18 thorpej */
1277 1.18 thorpej void
1278 1.86 yamt m_copyback(struct mbuf *m0, int off, int len, const void *cp)
1279 1.18 thorpej {
1280 1.85 yamt #if defined(DEBUG)
1281 1.85 yamt struct mbuf *origm = m0;
1282 1.85 yamt int error;
1283 1.85 yamt #endif /* defined(DEBUG) */
1284 1.85 yamt
1285 1.85 yamt if (m0 == NULL)
1286 1.85 yamt return;
1287 1.85 yamt
1288 1.85 yamt #if defined(DEBUG)
1289 1.85 yamt error =
1290 1.85 yamt #endif /* defined(DEBUG) */
1291 1.85 yamt m_copyback0(&m0, off, len, cp,
1292 1.85 yamt M_COPYBACK0_COPYBACK|M_COPYBACK0_EXTEND, M_DONTWAIT);
1293 1.85 yamt
1294 1.85 yamt #if defined(DEBUG)
1295 1.85 yamt if (error != 0 || (m0 != NULL && origm != m0))
1296 1.85 yamt panic("m_copyback");
1297 1.85 yamt #endif /* defined(DEBUG) */
1298 1.85 yamt }
1299 1.85 yamt
1300 1.85 yamt struct mbuf *
1301 1.86 yamt m_copyback_cow(struct mbuf *m0, int off, int len, const void *cp, int how)
1302 1.85 yamt {
1303 1.85 yamt int error;
1304 1.85 yamt
1305 1.85 yamt /* don't support chain expansion */
1306 1.85 yamt KDASSERT(off + len <= m_length(m0));
1307 1.85 yamt
1308 1.85 yamt error = m_copyback0(&m0, off, len, cp,
1309 1.85 yamt M_COPYBACK0_COPYBACK|M_COPYBACK0_COW, how);
1310 1.85 yamt if (error) {
1311 1.85 yamt /*
1312 1.85 yamt * no way to recover from partial success.
1313 1.85 yamt * just free the chain.
1314 1.85 yamt */
1315 1.85 yamt m_freem(m0);
1316 1.85 yamt return NULL;
1317 1.85 yamt }
1318 1.85 yamt return m0;
1319 1.85 yamt }
1320 1.85 yamt
1321 1.85 yamt /*
1322 1.85 yamt * m_makewritable: ensure the specified range writable.
1323 1.85 yamt */
1324 1.85 yamt int
1325 1.85 yamt m_makewritable(struct mbuf **mp, int off, int len, int how)
1326 1.85 yamt {
1327 1.85 yamt int error;
1328 1.85 yamt #if defined(DEBUG)
1329 1.85 yamt struct mbuf *n;
1330 1.85 yamt int origlen, reslen;
1331 1.85 yamt
1332 1.85 yamt origlen = m_length(*mp);
1333 1.85 yamt #endif /* defined(DEBUG) */
1334 1.85 yamt
1335 1.85 yamt #if 0 /* M_COPYALL is large enough */
1336 1.85 yamt if (len == M_COPYALL)
1337 1.85 yamt len = m_length(*mp) - off; /* XXX */
1338 1.85 yamt #endif
1339 1.85 yamt
1340 1.85 yamt error = m_copyback0(mp, off, len, NULL,
1341 1.85 yamt M_COPYBACK0_PRESERVE|M_COPYBACK0_COW, how);
1342 1.85 yamt
1343 1.85 yamt #if defined(DEBUG)
1344 1.85 yamt reslen = 0;
1345 1.85 yamt for (n = *mp; n; n = n->m_next)
1346 1.85 yamt reslen += n->m_len;
1347 1.85 yamt if (origlen != reslen)
1348 1.85 yamt panic("m_makewritable: length changed");
1349 1.85 yamt if (((*mp)->m_flags & M_PKTHDR) != 0 && reslen != (*mp)->m_pkthdr.len)
1350 1.85 yamt panic("m_makewritable: inconsist");
1351 1.85 yamt #endif /* defined(DEBUG) */
1352 1.85 yamt
1353 1.85 yamt return error;
1354 1.85 yamt }
1355 1.85 yamt
1356 1.142 dyoung /*
1357 1.142 dyoung * Copy the mbuf chain to a new mbuf chain that is as short as possible.
1358 1.142 dyoung * Return the new mbuf chain on success, NULL on failure. On success,
1359 1.142 dyoung * free the old mbuf chain.
1360 1.142 dyoung */
1361 1.142 dyoung struct mbuf *
1362 1.142 dyoung m_defrag(struct mbuf *mold, int flags)
1363 1.142 dyoung {
1364 1.142 dyoung struct mbuf *m0, *mn, *n;
1365 1.142 dyoung size_t sz = mold->m_pkthdr.len;
1366 1.142 dyoung
1367 1.142 dyoung #ifdef DIAGNOSTIC
1368 1.142 dyoung if ((mold->m_flags & M_PKTHDR) == 0)
1369 1.142 dyoung panic("m_defrag: not a mbuf chain header");
1370 1.142 dyoung #endif
1371 1.142 dyoung
1372 1.153 christos m0 = m_gethdr(flags, MT_DATA);
1373 1.142 dyoung if (m0 == NULL)
1374 1.142 dyoung return NULL;
1375 1.142 dyoung M_COPY_PKTHDR(m0, mold);
1376 1.142 dyoung mn = m0;
1377 1.142 dyoung
1378 1.142 dyoung do {
1379 1.142 dyoung if (sz > MHLEN) {
1380 1.142 dyoung MCLGET(mn, M_DONTWAIT);
1381 1.142 dyoung if ((mn->m_flags & M_EXT) == 0) {
1382 1.142 dyoung m_freem(m0);
1383 1.142 dyoung return NULL;
1384 1.142 dyoung }
1385 1.142 dyoung }
1386 1.142 dyoung
1387 1.142 dyoung mn->m_len = MIN(sz, MCLBYTES);
1388 1.142 dyoung
1389 1.142 dyoung m_copydata(mold, mold->m_pkthdr.len - sz, mn->m_len,
1390 1.142 dyoung mtod(mn, void *));
1391 1.142 dyoung
1392 1.142 dyoung sz -= mn->m_len;
1393 1.142 dyoung
1394 1.142 dyoung if (sz > 0) {
1395 1.142 dyoung /* need more mbufs */
1396 1.153 christos n = m_get(M_NOWAIT, MT_DATA);
1397 1.142 dyoung if (n == NULL) {
1398 1.142 dyoung m_freem(m0);
1399 1.142 dyoung return NULL;
1400 1.142 dyoung }
1401 1.142 dyoung
1402 1.142 dyoung mn->m_next = n;
1403 1.142 dyoung mn = n;
1404 1.142 dyoung }
1405 1.142 dyoung } while (sz > 0);
1406 1.142 dyoung
1407 1.142 dyoung m_freem(mold);
1408 1.142 dyoung
1409 1.142 dyoung return m0;
1410 1.142 dyoung }
1411 1.142 dyoung
1412 1.85 yamt int
1413 1.86 yamt m_copyback0(struct mbuf **mp0, int off, int len, const void *vp, int flags,
1414 1.86 yamt int how)
1415 1.85 yamt {
1416 1.27 matt int mlen;
1417 1.85 yamt struct mbuf *m, *n;
1418 1.85 yamt struct mbuf **mp;
1419 1.18 thorpej int totlen = 0;
1420 1.86 yamt const char *cp = vp;
1421 1.18 thorpej
1422 1.85 yamt KASSERT(mp0 != NULL);
1423 1.85 yamt KASSERT(*mp0 != NULL);
1424 1.85 yamt KASSERT((flags & M_COPYBACK0_PRESERVE) == 0 || cp == NULL);
1425 1.85 yamt KASSERT((flags & M_COPYBACK0_COPYBACK) == 0 || cp != NULL);
1426 1.85 yamt
1427 1.106 yamt /*
1428 1.106 yamt * we don't bother to update "totlen" in the case of M_COPYBACK0_COW,
1429 1.106 yamt * assuming that M_COPYBACK0_EXTEND and M_COPYBACK0_COW are exclusive.
1430 1.106 yamt */
1431 1.106 yamt
1432 1.106 yamt KASSERT((~flags & (M_COPYBACK0_EXTEND|M_COPYBACK0_COW)) != 0);
1433 1.106 yamt
1434 1.85 yamt mp = mp0;
1435 1.85 yamt m = *mp;
1436 1.18 thorpej while (off > (mlen = m->m_len)) {
1437 1.18 thorpej off -= mlen;
1438 1.18 thorpej totlen += mlen;
1439 1.109 yamt if (m->m_next == NULL) {
1440 1.109 yamt int tspace;
1441 1.109 yamt extend:
1442 1.85 yamt if ((flags & M_COPYBACK0_EXTEND) == 0)
1443 1.85 yamt goto out;
1444 1.109 yamt
1445 1.109 yamt /*
1446 1.109 yamt * try to make some space at the end of "m".
1447 1.109 yamt */
1448 1.109 yamt
1449 1.109 yamt mlen = m->m_len;
1450 1.109 yamt if (off + len >= MINCLSIZE &&
1451 1.109 yamt (m->m_flags & M_EXT) == 0 && m->m_len == 0) {
1452 1.109 yamt MCLGET(m, how);
1453 1.109 yamt }
1454 1.109 yamt tspace = M_TRAILINGSPACE(m);
1455 1.109 yamt if (tspace > 0) {
1456 1.109 yamt tspace = min(tspace, off + len);
1457 1.109 yamt KASSERT(tspace > 0);
1458 1.109 yamt memset(mtod(m, char *) + m->m_len, 0,
1459 1.109 yamt min(off, tspace));
1460 1.109 yamt m->m_len += tspace;
1461 1.109 yamt off += mlen;
1462 1.109 yamt totlen -= mlen;
1463 1.109 yamt continue;
1464 1.109 yamt }
1465 1.109 yamt
1466 1.109 yamt /*
1467 1.109 yamt * need to allocate an mbuf.
1468 1.109 yamt */
1469 1.109 yamt
1470 1.109 yamt if (off + len >= MINCLSIZE) {
1471 1.109 yamt n = m_getcl(how, m->m_type, 0);
1472 1.109 yamt } else {
1473 1.109 yamt n = m_get(how, m->m_type);
1474 1.109 yamt }
1475 1.109 yamt if (n == NULL) {
1476 1.18 thorpej goto out;
1477 1.109 yamt }
1478 1.109 yamt n->m_len = min(M_TRAILINGSPACE(n), off + len);
1479 1.109 yamt memset(mtod(n, char *), 0, min(n->m_len, off));
1480 1.18 thorpej m->m_next = n;
1481 1.18 thorpej }
1482 1.85 yamt mp = &m->m_next;
1483 1.18 thorpej m = m->m_next;
1484 1.18 thorpej }
1485 1.18 thorpej while (len > 0) {
1486 1.85 yamt mlen = m->m_len - off;
1487 1.85 yamt if (mlen != 0 && M_READONLY(m)) {
1488 1.85 yamt char *datap;
1489 1.85 yamt int eatlen;
1490 1.85 yamt
1491 1.85 yamt /*
1492 1.85 yamt * this mbuf is read-only.
1493 1.85 yamt * allocate a new writable mbuf and try again.
1494 1.85 yamt */
1495 1.85 yamt
1496 1.85 yamt #if defined(DIAGNOSTIC)
1497 1.85 yamt if ((flags & M_COPYBACK0_COW) == 0)
1498 1.85 yamt panic("m_copyback0: read-only");
1499 1.85 yamt #endif /* defined(DIAGNOSTIC) */
1500 1.85 yamt
1501 1.85 yamt /*
1502 1.85 yamt * if we're going to write into the middle of
1503 1.85 yamt * a mbuf, split it first.
1504 1.85 yamt */
1505 1.137 seanb if (off > 0) {
1506 1.85 yamt n = m_split0(m, off, how, 0);
1507 1.85 yamt if (n == NULL)
1508 1.85 yamt goto enobufs;
1509 1.85 yamt m->m_next = n;
1510 1.85 yamt mp = &m->m_next;
1511 1.85 yamt m = n;
1512 1.85 yamt off = 0;
1513 1.85 yamt continue;
1514 1.85 yamt }
1515 1.85 yamt
1516 1.85 yamt /*
1517 1.85 yamt * XXX TODO coalesce into the trailingspace of
1518 1.85 yamt * the previous mbuf when possible.
1519 1.85 yamt */
1520 1.85 yamt
1521 1.85 yamt /*
1522 1.85 yamt * allocate a new mbuf. copy packet header if needed.
1523 1.85 yamt */
1524 1.153 christos n = m_get(how, m->m_type);
1525 1.85 yamt if (n == NULL)
1526 1.85 yamt goto enobufs;
1527 1.85 yamt MCLAIM(n, m->m_owner);
1528 1.85 yamt if (off == 0 && (m->m_flags & M_PKTHDR) != 0) {
1529 1.101 yamt M_MOVE_PKTHDR(n, m);
1530 1.85 yamt n->m_len = MHLEN;
1531 1.85 yamt } else {
1532 1.85 yamt if (len >= MINCLSIZE)
1533 1.85 yamt MCLGET(n, M_DONTWAIT);
1534 1.85 yamt n->m_len =
1535 1.85 yamt (n->m_flags & M_EXT) ? MCLBYTES : MLEN;
1536 1.85 yamt }
1537 1.85 yamt if (n->m_len > len)
1538 1.85 yamt n->m_len = len;
1539 1.85 yamt
1540 1.85 yamt /*
1541 1.85 yamt * free the region which has been overwritten.
1542 1.85 yamt * copying data from old mbufs if requested.
1543 1.85 yamt */
1544 1.85 yamt if (flags & M_COPYBACK0_PRESERVE)
1545 1.85 yamt datap = mtod(n, char *);
1546 1.85 yamt else
1547 1.85 yamt datap = NULL;
1548 1.85 yamt eatlen = n->m_len;
1549 1.85 yamt while (m != NULL && M_READONLY(m) &&
1550 1.85 yamt n->m_type == m->m_type && eatlen > 0) {
1551 1.85 yamt mlen = min(eatlen, m->m_len);
1552 1.85 yamt if (datap) {
1553 1.85 yamt m_copydata(m, 0, mlen, datap);
1554 1.85 yamt datap += mlen;
1555 1.85 yamt }
1556 1.85 yamt m->m_data += mlen;
1557 1.85 yamt m->m_len -= mlen;
1558 1.85 yamt eatlen -= mlen;
1559 1.85 yamt if (m->m_len == 0)
1560 1.85 yamt *mp = m = m_free(m);
1561 1.85 yamt }
1562 1.85 yamt if (eatlen > 0)
1563 1.85 yamt n->m_len -= eatlen;
1564 1.85 yamt n->m_next = m;
1565 1.85 yamt *mp = m = n;
1566 1.85 yamt continue;
1567 1.85 yamt }
1568 1.85 yamt mlen = min(mlen, len);
1569 1.85 yamt if (flags & M_COPYBACK0_COPYBACK) {
1570 1.119 christos memcpy(mtod(m, char *) + off, cp, (unsigned)mlen);
1571 1.85 yamt cp += mlen;
1572 1.85 yamt }
1573 1.18 thorpej len -= mlen;
1574 1.18 thorpej mlen += off;
1575 1.18 thorpej off = 0;
1576 1.18 thorpej totlen += mlen;
1577 1.18 thorpej if (len == 0)
1578 1.18 thorpej break;
1579 1.109 yamt if (m->m_next == NULL) {
1580 1.109 yamt goto extend;
1581 1.18 thorpej }
1582 1.85 yamt mp = &m->m_next;
1583 1.18 thorpej m = m->m_next;
1584 1.18 thorpej }
1585 1.106 yamt out: if (((m = *mp0)->m_flags & M_PKTHDR) && (m->m_pkthdr.len < totlen)) {
1586 1.106 yamt KASSERT((flags & M_COPYBACK0_EXTEND) != 0);
1587 1.18 thorpej m->m_pkthdr.len = totlen;
1588 1.106 yamt }
1589 1.85 yamt
1590 1.85 yamt return 0;
1591 1.85 yamt
1592 1.85 yamt enobufs:
1593 1.85 yamt return ENOBUFS;
1594 1.66 thorpej }
1595 1.66 thorpej
1596 1.101 yamt void
1597 1.101 yamt m_move_pkthdr(struct mbuf *to, struct mbuf *from)
1598 1.101 yamt {
1599 1.101 yamt
1600 1.101 yamt KASSERT((to->m_flags & M_EXT) == 0);
1601 1.101 yamt KASSERT((to->m_flags & M_PKTHDR) == 0 || m_tag_first(to) == NULL);
1602 1.101 yamt KASSERT((from->m_flags & M_PKTHDR) != 0);
1603 1.101 yamt
1604 1.101 yamt to->m_pkthdr = from->m_pkthdr;
1605 1.101 yamt to->m_flags = from->m_flags & M_COPYFLAGS;
1606 1.101 yamt to->m_data = to->m_pktdat;
1607 1.101 yamt
1608 1.101 yamt from->m_flags &= ~M_PKTHDR;
1609 1.101 yamt }
1610 1.101 yamt
1611 1.66 thorpej /*
1612 1.66 thorpej * Apply function f to the data in an mbuf chain starting "off" bytes from the
1613 1.66 thorpej * beginning, continuing for "len" bytes.
1614 1.66 thorpej */
1615 1.66 thorpej int
1616 1.66 thorpej m_apply(struct mbuf *m, int off, int len,
1617 1.119 christos int (*f)(void *, void *, unsigned int), void *arg)
1618 1.66 thorpej {
1619 1.66 thorpej unsigned int count;
1620 1.66 thorpej int rval;
1621 1.66 thorpej
1622 1.66 thorpej KASSERT(len >= 0);
1623 1.66 thorpej KASSERT(off >= 0);
1624 1.66 thorpej
1625 1.66 thorpej while (off > 0) {
1626 1.66 thorpej KASSERT(m != NULL);
1627 1.66 thorpej if (off < m->m_len)
1628 1.66 thorpej break;
1629 1.66 thorpej off -= m->m_len;
1630 1.66 thorpej m = m->m_next;
1631 1.66 thorpej }
1632 1.66 thorpej while (len > 0) {
1633 1.66 thorpej KASSERT(m != NULL);
1634 1.66 thorpej count = min(m->m_len - off, len);
1635 1.66 thorpej
1636 1.119 christos rval = (*f)(arg, mtod(m, char *) + off, count);
1637 1.66 thorpej if (rval)
1638 1.66 thorpej return (rval);
1639 1.66 thorpej
1640 1.66 thorpej len -= count;
1641 1.66 thorpej off = 0;
1642 1.66 thorpej m = m->m_next;
1643 1.66 thorpej }
1644 1.66 thorpej
1645 1.66 thorpej return (0);
1646 1.66 thorpej }
1647 1.66 thorpej
1648 1.66 thorpej /*
1649 1.66 thorpej * Return a pointer to mbuf/offset of location in mbuf chain.
1650 1.66 thorpej */
1651 1.66 thorpej struct mbuf *
1652 1.66 thorpej m_getptr(struct mbuf *m, int loc, int *off)
1653 1.66 thorpej {
1654 1.66 thorpej
1655 1.66 thorpej while (loc >= 0) {
1656 1.66 thorpej /* Normal end of search */
1657 1.66 thorpej if (m->m_len > loc) {
1658 1.66 thorpej *off = loc;
1659 1.66 thorpej return (m);
1660 1.66 thorpej } else {
1661 1.66 thorpej loc -= m->m_len;
1662 1.66 thorpej
1663 1.66 thorpej if (m->m_next == NULL) {
1664 1.66 thorpej if (loc == 0) {
1665 1.66 thorpej /* Point at the end of valid data */
1666 1.66 thorpej *off = m->m_len;
1667 1.66 thorpej return (m);
1668 1.66 thorpej } else
1669 1.66 thorpej return (NULL);
1670 1.66 thorpej } else
1671 1.66 thorpej m = m->m_next;
1672 1.66 thorpej }
1673 1.66 thorpej }
1674 1.66 thorpej
1675 1.66 thorpej return (NULL);
1676 1.1 cgd }
1677 1.105 yamt
1678 1.125 yamt /*
1679 1.125 yamt * m_ext_free: release a reference to the mbuf external storage.
1680 1.125 yamt *
1681 1.125 yamt * => free the mbuf m itsself as well.
1682 1.125 yamt */
1683 1.125 yamt
1684 1.125 yamt void
1685 1.125 yamt m_ext_free(struct mbuf *m)
1686 1.125 yamt {
1687 1.125 yamt bool embedded = MEXT_ISEMBEDDED(m);
1688 1.125 yamt bool dofree = true;
1689 1.125 yamt u_int refcnt;
1690 1.125 yamt
1691 1.125 yamt KASSERT((m->m_flags & M_EXT) != 0);
1692 1.125 yamt KASSERT(MEXT_ISEMBEDDED(m->m_ext_ref));
1693 1.125 yamt KASSERT((m->m_ext_ref->m_flags & M_EXT) != 0);
1694 1.125 yamt KASSERT((m->m_flags & M_EXT_CLUSTER) ==
1695 1.125 yamt (m->m_ext_ref->m_flags & M_EXT_CLUSTER));
1696 1.125 yamt
1697 1.125 yamt if (__predict_true(m->m_ext.ext_refcnt == 1)) {
1698 1.125 yamt refcnt = m->m_ext.ext_refcnt = 0;
1699 1.125 yamt } else {
1700 1.125 yamt refcnt = atomic_dec_uint_nv(&m->m_ext.ext_refcnt);
1701 1.125 yamt }
1702 1.125 yamt if (refcnt > 0) {
1703 1.125 yamt if (embedded) {
1704 1.125 yamt /*
1705 1.125 yamt * other mbuf's m_ext_ref still points to us.
1706 1.125 yamt */
1707 1.125 yamt dofree = false;
1708 1.125 yamt } else {
1709 1.125 yamt m->m_ext_ref = m;
1710 1.125 yamt }
1711 1.125 yamt } else {
1712 1.125 yamt /*
1713 1.125 yamt * dropping the last reference
1714 1.125 yamt */
1715 1.125 yamt if (!embedded) {
1716 1.125 yamt m->m_ext.ext_refcnt++; /* XXX */
1717 1.125 yamt m_ext_free(m->m_ext_ref);
1718 1.125 yamt m->m_ext_ref = m;
1719 1.125 yamt } else if ((m->m_flags & M_EXT_CLUSTER) != 0) {
1720 1.125 yamt pool_cache_put_paddr((struct pool_cache *)
1721 1.125 yamt m->m_ext.ext_arg,
1722 1.125 yamt m->m_ext.ext_buf, m->m_ext.ext_paddr);
1723 1.125 yamt } else if (m->m_ext.ext_free) {
1724 1.125 yamt (*m->m_ext.ext_free)(m,
1725 1.125 yamt m->m_ext.ext_buf, m->m_ext.ext_size,
1726 1.125 yamt m->m_ext.ext_arg);
1727 1.125 yamt /*
1728 1.125 yamt * 'm' is already freed by the ext_free callback.
1729 1.125 yamt */
1730 1.125 yamt dofree = false;
1731 1.125 yamt } else {
1732 1.125 yamt free(m->m_ext.ext_buf, m->m_ext.ext_type);
1733 1.125 yamt }
1734 1.125 yamt }
1735 1.125 yamt if (dofree) {
1736 1.152 christos m->m_type = MT_FREE;
1737 1.125 yamt pool_cache_put(mb_cache, m);
1738 1.125 yamt }
1739 1.125 yamt }
1740 1.125 yamt
1741 1.105 yamt #if defined(DDB)
1742 1.105 yamt void
1743 1.105 yamt m_print(const struct mbuf *m, const char *modif, void (*pr)(const char *, ...))
1744 1.105 yamt {
1745 1.105 yamt char ch;
1746 1.118 thorpej bool opt_c = false;
1747 1.105 yamt char buf[512];
1748 1.105 yamt
1749 1.105 yamt while ((ch = *(modif++)) != '\0') {
1750 1.105 yamt switch (ch) {
1751 1.105 yamt case 'c':
1752 1.118 thorpej opt_c = true;
1753 1.105 yamt break;
1754 1.105 yamt }
1755 1.105 yamt }
1756 1.105 yamt
1757 1.105 yamt nextchain:
1758 1.105 yamt (*pr)("MBUF %p\n", m);
1759 1.130 christos snprintb(buf, sizeof(buf), M_FLAGS_BITS, (u_int)m->m_flags);
1760 1.138 cegger (*pr)(" data=%p, len=%d, type=%d, flags=%s\n",
1761 1.105 yamt m->m_data, m->m_len, m->m_type, buf);
1762 1.105 yamt (*pr)(" owner=%p, next=%p, nextpkt=%p\n", m->m_owner, m->m_next,
1763 1.105 yamt m->m_nextpkt);
1764 1.105 yamt (*pr)(" leadingspace=%u, trailingspace=%u, readonly=%u\n",
1765 1.105 yamt (int)M_LEADINGSPACE(m), (int)M_TRAILINGSPACE(m),
1766 1.105 yamt (int)M_READONLY(m));
1767 1.105 yamt if ((m->m_flags & M_PKTHDR) != 0) {
1768 1.130 christos snprintb(buf, sizeof(buf), M_CSUM_BITS, m->m_pkthdr.csum_flags);
1769 1.105 yamt (*pr)(" pktlen=%d, rcvif=%p, csum_flags=0x%s, csum_data=0x%"
1770 1.105 yamt PRIx32 ", segsz=%u\n",
1771 1.105 yamt m->m_pkthdr.len, m->m_pkthdr.rcvif,
1772 1.105 yamt buf, m->m_pkthdr.csum_data, m->m_pkthdr.segsz);
1773 1.105 yamt }
1774 1.105 yamt if ((m->m_flags & M_EXT)) {
1775 1.125 yamt (*pr)(" ext_refcnt=%u, ext_buf=%p, ext_size=%zd, "
1776 1.105 yamt "ext_free=%p, ext_arg=%p\n",
1777 1.125 yamt m->m_ext.ext_refcnt,
1778 1.105 yamt m->m_ext.ext_buf, m->m_ext.ext_size,
1779 1.105 yamt m->m_ext.ext_free, m->m_ext.ext_arg);
1780 1.105 yamt }
1781 1.105 yamt if ((~m->m_flags & (M_EXT|M_EXT_PAGES)) == 0) {
1782 1.108 yamt vaddr_t sva = (vaddr_t)m->m_ext.ext_buf;
1783 1.108 yamt vaddr_t eva = sva + m->m_ext.ext_size;
1784 1.108 yamt int n = (round_page(eva) - trunc_page(sva)) >> PAGE_SHIFT;
1785 1.108 yamt int i;
1786 1.105 yamt
1787 1.105 yamt (*pr)(" pages:");
1788 1.108 yamt for (i = 0; i < n; i ++) {
1789 1.108 yamt (*pr)(" %p", m->m_ext.ext_pgs[i]);
1790 1.105 yamt }
1791 1.105 yamt (*pr)("\n");
1792 1.105 yamt }
1793 1.105 yamt
1794 1.105 yamt if (opt_c) {
1795 1.105 yamt m = m->m_next;
1796 1.105 yamt if (m != NULL) {
1797 1.105 yamt goto nextchain;
1798 1.105 yamt }
1799 1.105 yamt }
1800 1.105 yamt }
1801 1.105 yamt #endif /* defined(DDB) */
1802 1.124 yamt
1803 1.124 yamt void
1804 1.124 yamt mbstat_type_add(int type, int diff)
1805 1.124 yamt {
1806 1.124 yamt struct mbstat_cpu *mb;
1807 1.124 yamt int s;
1808 1.124 yamt
1809 1.124 yamt s = splvm();
1810 1.126 thorpej mb = percpu_getref(mbstat_percpu);
1811 1.124 yamt mb->m_mtypes[type] += diff;
1812 1.126 thorpej percpu_putref(mbstat_percpu);
1813 1.124 yamt splx(s);
1814 1.124 yamt }
1815 1.124 yamt
1816 1.124 yamt #if defined(MBUFTRACE)
1817 1.124 yamt void
1818 1.124 yamt mowner_attach(struct mowner *mo)
1819 1.124 yamt {
1820 1.124 yamt
1821 1.124 yamt KASSERT(mo->mo_counters == NULL);
1822 1.124 yamt mo->mo_counters = percpu_alloc(sizeof(struct mowner_counter));
1823 1.124 yamt
1824 1.124 yamt /* XXX lock */
1825 1.124 yamt LIST_INSERT_HEAD(&mowners, mo, mo_link);
1826 1.124 yamt }
1827 1.124 yamt
1828 1.124 yamt void
1829 1.124 yamt mowner_detach(struct mowner *mo)
1830 1.124 yamt {
1831 1.124 yamt
1832 1.124 yamt KASSERT(mo->mo_counters != NULL);
1833 1.124 yamt
1834 1.124 yamt /* XXX lock */
1835 1.124 yamt LIST_REMOVE(mo, mo_link);
1836 1.124 yamt
1837 1.124 yamt percpu_free(mo->mo_counters, sizeof(struct mowner_counter));
1838 1.124 yamt mo->mo_counters = NULL;
1839 1.124 yamt }
1840 1.124 yamt
1841 1.124 yamt void
1842 1.124 yamt mowner_init(struct mbuf *m, int type)
1843 1.124 yamt {
1844 1.124 yamt struct mowner_counter *mc;
1845 1.124 yamt struct mowner *mo;
1846 1.124 yamt int s;
1847 1.124 yamt
1848 1.124 yamt m->m_owner = mo = &unknown_mowners[type];
1849 1.124 yamt s = splvm();
1850 1.126 thorpej mc = percpu_getref(mo->mo_counters);
1851 1.124 yamt mc->mc_counter[MOWNER_COUNTER_CLAIMS]++;
1852 1.126 thorpej percpu_putref(mo->mo_counters);
1853 1.124 yamt splx(s);
1854 1.124 yamt }
1855 1.124 yamt
1856 1.124 yamt void
1857 1.124 yamt mowner_ref(struct mbuf *m, int flags)
1858 1.124 yamt {
1859 1.124 yamt struct mowner *mo = m->m_owner;
1860 1.124 yamt struct mowner_counter *mc;
1861 1.124 yamt int s;
1862 1.124 yamt
1863 1.124 yamt s = splvm();
1864 1.126 thorpej mc = percpu_getref(mo->mo_counters);
1865 1.124 yamt if ((flags & M_EXT) != 0)
1866 1.124 yamt mc->mc_counter[MOWNER_COUNTER_EXT_CLAIMS]++;
1867 1.124 yamt if ((flags & M_CLUSTER) != 0)
1868 1.124 yamt mc->mc_counter[MOWNER_COUNTER_CLUSTER_CLAIMS]++;
1869 1.126 thorpej percpu_putref(mo->mo_counters);
1870 1.124 yamt splx(s);
1871 1.124 yamt }
1872 1.124 yamt
1873 1.124 yamt void
1874 1.124 yamt mowner_revoke(struct mbuf *m, bool all, int flags)
1875 1.124 yamt {
1876 1.124 yamt struct mowner *mo = m->m_owner;
1877 1.124 yamt struct mowner_counter *mc;
1878 1.124 yamt int s;
1879 1.124 yamt
1880 1.124 yamt s = splvm();
1881 1.126 thorpej mc = percpu_getref(mo->mo_counters);
1882 1.124 yamt if ((flags & M_EXT) != 0)
1883 1.124 yamt mc->mc_counter[MOWNER_COUNTER_EXT_RELEASES]++;
1884 1.124 yamt if ((flags & M_CLUSTER) != 0)
1885 1.124 yamt mc->mc_counter[MOWNER_COUNTER_CLUSTER_RELEASES]++;
1886 1.124 yamt if (all)
1887 1.124 yamt mc->mc_counter[MOWNER_COUNTER_RELEASES]++;
1888 1.126 thorpej percpu_putref(mo->mo_counters);
1889 1.124 yamt splx(s);
1890 1.124 yamt if (all)
1891 1.124 yamt m->m_owner = &revoked_mowner;
1892 1.124 yamt }
1893 1.124 yamt
1894 1.124 yamt static void
1895 1.124 yamt mowner_claim(struct mbuf *m, struct mowner *mo)
1896 1.124 yamt {
1897 1.124 yamt struct mowner_counter *mc;
1898 1.124 yamt int flags = m->m_flags;
1899 1.124 yamt int s;
1900 1.124 yamt
1901 1.124 yamt s = splvm();
1902 1.126 thorpej mc = percpu_getref(mo->mo_counters);
1903 1.124 yamt mc->mc_counter[MOWNER_COUNTER_CLAIMS]++;
1904 1.124 yamt if ((flags & M_EXT) != 0)
1905 1.124 yamt mc->mc_counter[MOWNER_COUNTER_EXT_CLAIMS]++;
1906 1.124 yamt if ((flags & M_CLUSTER) != 0)
1907 1.124 yamt mc->mc_counter[MOWNER_COUNTER_CLUSTER_CLAIMS]++;
1908 1.126 thorpej percpu_putref(mo->mo_counters);
1909 1.124 yamt splx(s);
1910 1.124 yamt m->m_owner = mo;
1911 1.124 yamt }
1912 1.124 yamt
1913 1.124 yamt void
1914 1.124 yamt m_claim(struct mbuf *m, struct mowner *mo)
1915 1.124 yamt {
1916 1.124 yamt
1917 1.124 yamt if (m->m_owner == mo || mo == NULL)
1918 1.124 yamt return;
1919 1.124 yamt
1920 1.124 yamt mowner_revoke(m, true, m->m_flags);
1921 1.124 yamt mowner_claim(m, mo);
1922 1.124 yamt }
1923 1.124 yamt #endif /* defined(MBUFTRACE) */
1924