117
118/* XXX: These should be tuneables. Can't change UMA limits on the fly. */
119static int
120sysctl_nmbclusters(SYSCTL_HANDLER_ARGS)
121{
122 int error, newnmbclusters;
123
124 newnmbclusters = nmbclusters;
125 error = sysctl_handle_int(oidp, &newnmbclusters, 0, req);
126 if (error == 0 && req->newptr) {
127 if (newnmbclusters > nmbclusters) {
128 nmbclusters = newnmbclusters;
129 uma_zone_set_max(zone_clust, nmbclusters);
130 EVENTHANDLER_INVOKE(nmbclusters_change);
131 } else
132 error = EINVAL;
133 }
134 return (error);
135}
136SYSCTL_PROC(_kern_ipc, OID_AUTO, nmbclusters, CTLTYPE_INT|CTLFLAG_RW,
137&nmbclusters, 0, sysctl_nmbclusters, "IU",
138 "Maximum number of mbuf clusters allowed");
139
140static int
141sysctl_nmbjumbop(SYSCTL_HANDLER_ARGS)
142{
143 int error, newnmbjumbop;
144
145 newnmbjumbop = nmbjumbop;
146 error = sysctl_handle_int(oidp, &newnmbjumbop, 0, req);
147 if (error == 0 && req->newptr) {
148 if (newnmbjumbop> nmbjumbop) {
149 nmbjumbop = newnmbjumbop;
150 uma_zone_set_max(zone_jumbop, nmbjumbop);
151 } else
152 error = EINVAL;
153 }
154 return (error);
155}
156SYSCTL_PROC(_kern_ipc, OID_AUTO, nmbjumbop, CTLTYPE_INT|CTLFLAG_RW,
157&nmbjumbop, 0, sysctl_nmbjumbop, "IU",
158 "Maximum number of mbuf page size jumbo clusters allowed");
159
160
161static int
162sysctl_nmbjumbo9(SYSCTL_HANDLER_ARGS)
163{
164 int error, newnmbjumbo9;
165
166 newnmbjumbo9 = nmbjumbo9;
167 error = sysctl_handle_int(oidp, &newnmbjumbo9, 0, req);
168 if (error == 0 && req->newptr) {
169 if (newnmbjumbo9> nmbjumbo9) {
170 nmbjumbo9 = newnmbjumbo9;
171 uma_zone_set_max(zone_jumbo9, nmbjumbo9);
172 } else
173 error = EINVAL;
174 }
175 return (error);
176}
177SYSCTL_PROC(_kern_ipc, OID_AUTO, nmbjumbo9, CTLTYPE_INT|CTLFLAG_RW,
178&nmbjumbo9, 0, sysctl_nmbjumbo9, "IU",
179 "Maximum number of mbuf 9k jumbo clusters allowed");
180
181static int
182sysctl_nmbjumbo16(SYSCTL_HANDLER_ARGS)
183{
184 int error, newnmbjumbo16;
185
186 newnmbjumbo16 = nmbjumbo16;
187 error = sysctl_handle_int(oidp, &newnmbjumbo16, 0, req);
188 if (error == 0 && req->newptr) {
189 if (newnmbjumbo16> nmbjumbo16) {
190 nmbjumbo16 = newnmbjumbo16;
191 uma_zone_set_max(zone_jumbo16, nmbjumbo16);
192 } else
193 error = EINVAL;
194 }
195 return (error);
196}
197SYSCTL_PROC(_kern_ipc, OID_AUTO, nmbjumbo16, CTLTYPE_INT|CTLFLAG_RW,
198&nmbjumbo16, 0, sysctl_nmbjumbo16, "IU",
199 "Maximum number of mbuf 16k jumbo clusters allowed");
200
201
202
203SYSCTL_STRUCT(_kern_ipc, OID_AUTO, mbstat, CTLFLAG_RD, &mbstat, mbstat,
204 "Mbuf general information and statistics");
205
206/*
207 * Zones from which we allocate.
208 */
209uma_zone_t zone_mbuf;
210uma_zone_t zone_clust;
211uma_zone_t zone_pack;
212uma_zone_t zone_jumbop;
213uma_zone_t zone_jumbo9;
214uma_zone_t zone_jumbo16;
215uma_zone_t zone_ext_refcnt;
216
217/*
218 * Local prototypes.
219 */
220static int mb_ctor_mbuf(void *, int, void *, int);
221static int mb_ctor_clust(void *, int, void *, int);
222static int mb_ctor_pack(void *, int, void *, int);
223static void mb_dtor_mbuf(void *, int, void *);
224static void mb_dtor_clust(void *, int, void *);
225static void mb_dtor_pack(void *, int, void *);
226static int mb_zinit_pack(void *, int, int);
227static void mb_zfini_pack(void *, int);
228
229static void mb_reclaim(void *);
230static void mbuf_init(void *);
231static void *mbuf_jumbo_alloc(uma_zone_t, int, u_int8_t *, int);
232static void mbuf_jumbo_free(void *, int, u_int8_t);
233
234static MALLOC_DEFINE(M_JUMBOFRAME, "jumboframes", "mbuf jumbo frame buffers");
235
236/* Ensure that MSIZE doesn't break dtom() - it must be a power of 2 */
237CTASSERT((((MSIZE - 1) ^ MSIZE) + 1) >> 1 == MSIZE);
238
239/*
240 * Initialize FreeBSD Network buffer allocation.
241 */
242SYSINIT(mbuf, SI_SUB_MBUF, SI_ORDER_FIRST, mbuf_init, NULL);
243static void
244mbuf_init(void *dummy)
245{
246
247 /*
248 * Configure UMA zones for Mbufs, Clusters, and Packets.
249 */
250 zone_mbuf = uma_zcreate(MBUF_MEM_NAME, MSIZE,
251 mb_ctor_mbuf, mb_dtor_mbuf,
252#ifdef INVARIANTS
253 trash_init, trash_fini,
254#else
255 NULL, NULL,
256#endif
257 MSIZE - 1, UMA_ZONE_MAXBUCKET);
258
259 zone_clust = uma_zcreate(MBUF_CLUSTER_MEM_NAME, MCLBYTES,
260 mb_ctor_clust, mb_dtor_clust,
261#ifdef INVARIANTS
262 trash_init, trash_fini,
263#else
264 NULL, NULL,
265#endif
266 UMA_ALIGN_PTR, UMA_ZONE_REFCNT);
267 if (nmbclusters > 0)
268 uma_zone_set_max(zone_clust, nmbclusters);
269
270 zone_pack = uma_zsecond_create(MBUF_PACKET_MEM_NAME, mb_ctor_pack,
271 mb_dtor_pack, mb_zinit_pack, mb_zfini_pack, zone_mbuf);
272
273 /* Make jumbo frame zone too. Page size, 9k and 16k. */
274 zone_jumbop = uma_zcreate(MBUF_JUMBOP_MEM_NAME, MJUMPAGESIZE,
275 mb_ctor_clust, mb_dtor_clust,
276#ifdef INVARIANTS
277 trash_init, trash_fini,
278#else
279 NULL, NULL,
280#endif
281 UMA_ALIGN_PTR, UMA_ZONE_REFCNT);
282 if (nmbjumbop > 0)
283 uma_zone_set_max(zone_jumbop, nmbjumbop);
284
285 zone_jumbo9 = uma_zcreate(MBUF_JUMBO9_MEM_NAME, MJUM9BYTES,
286 mb_ctor_clust, mb_dtor_clust,
287#ifdef INVARIANTS
288 trash_init, trash_fini,
289#else
290 NULL, NULL,
291#endif
292 UMA_ALIGN_PTR, UMA_ZONE_REFCNT);
293 if (nmbjumbo9 > 0)
294 uma_zone_set_max(zone_jumbo9, nmbjumbo9);
295 uma_zone_set_allocf(zone_jumbo9, mbuf_jumbo_alloc);
296 uma_zone_set_freef(zone_jumbo9, mbuf_jumbo_free);
297
298 zone_jumbo16 = uma_zcreate(MBUF_JUMBO16_MEM_NAME, MJUM16BYTES,
299 mb_ctor_clust, mb_dtor_clust,
300#ifdef INVARIANTS
301 trash_init, trash_fini,
302#else
303 NULL, NULL,
304#endif
305 UMA_ALIGN_PTR, UMA_ZONE_REFCNT);
306 if (nmbjumbo16 > 0)
307 uma_zone_set_max(zone_jumbo16, nmbjumbo16);
308 uma_zone_set_allocf(zone_jumbo16, mbuf_jumbo_alloc);
309 uma_zone_set_freef(zone_jumbo16, mbuf_jumbo_free);
310
311 zone_ext_refcnt = uma_zcreate(MBUF_EXTREFCNT_MEM_NAME, sizeof(u_int),
312 NULL, NULL,
313 NULL, NULL,
314 UMA_ALIGN_PTR, UMA_ZONE_ZINIT);
315
316 /* uma_prealloc() goes here... */
317
318 /*
319 * Hook event handler for low-memory situation, used to
320 * drain protocols and push data back to the caches (UMA
321 * later pushes it back to VM).
322 */
323 EVENTHANDLER_REGISTER(vm_lowmem, mb_reclaim, NULL,
324 EVENTHANDLER_PRI_FIRST);
325
326 /*
327 * [Re]set counters and local statistics knobs.
328 * XXX Some of these should go and be replaced, but UMA stat
329 * gathering needs to be revised.
330 */
331 mbstat.m_mbufs = 0;
332 mbstat.m_mclusts = 0;
333 mbstat.m_drain = 0;
334 mbstat.m_msize = MSIZE;
335 mbstat.m_mclbytes = MCLBYTES;
336 mbstat.m_minclsize = MINCLSIZE;
337 mbstat.m_mlen = MLEN;
338 mbstat.m_mhlen = MHLEN;
339 mbstat.m_numtypes = MT_NTYPES;
340
341 mbstat.m_mcfail = mbstat.m_mpfail = 0;
342 mbstat.sf_iocnt = 0;
343 mbstat.sf_allocwait = mbstat.sf_allocfail = 0;
344}
345
346/*
347 * UMA backend page allocator for the jumbo frame zones.
348 *
349 * Allocates kernel virtual memory that is backed by contiguous physical
350 * pages.
351 */
352static void *
353mbuf_jumbo_alloc(uma_zone_t zone, int bytes, u_int8_t *flags, int wait)
354{
355
356 /* Inform UMA that this allocator uses kernel_map/object. */
357 *flags = UMA_SLAB_KERNEL;
358 return (contigmalloc(bytes, M_JUMBOFRAME, wait, (vm_paddr_t)0,
359 ~(vm_paddr_t)0, 1, 0));
360}
361
362/*
363 * UMA backend page deallocator for the jumbo frame zones.
364 */
365static void
366mbuf_jumbo_free(void *mem, int size, u_int8_t flags)
367{
368
369 contigfree(mem, size, M_JUMBOFRAME);
370}
371
372/*
373 * Constructor for Mbuf master zone.
374 *
375 * The 'arg' pointer points to a mb_args structure which
376 * contains call-specific information required to support the
377 * mbuf allocation API. See mbuf.h.
378 */
379static int
380mb_ctor_mbuf(void *mem, int size, void *arg, int how)
381{
382 struct mbuf *m;
383 struct mb_args *args;
384#ifdef MAC
385 int error;
386#endif
387 int flags;
388 short type;
389
390#ifdef INVARIANTS
391 trash_ctor(mem, size, arg, how);
392#endif
393 m = (struct mbuf *)mem;
394 args = (struct mb_args *)arg;
395 flags = args->flags;
396 type = args->type;
397
398 /*
399 * The mbuf is initialized later. The caller has the
400 * responsibility to set up any MAC labels too.
401 */
402 if (type == MT_NOINIT)
403 return (0);
404
405 m->m_next = NULL;
406 m->m_nextpkt = NULL;
407 m->m_len = 0;
408 m->m_flags = flags;
409 m->m_type = type;
410 if (flags & M_PKTHDR) {
411 m->m_data = m->m_pktdat;
412 m->m_pkthdr.rcvif = NULL;
413 m->m_pkthdr.header = NULL;
414 m->m_pkthdr.len = 0;
415 m->m_pkthdr.csum_flags = 0;
416 m->m_pkthdr.csum_data = 0;
417 m->m_pkthdr.tso_segsz = 0;
418 m->m_pkthdr.ether_vtag = 0;
419 SLIST_INIT(&m->m_pkthdr.tags);
420#ifdef MAC
421 /* If the label init fails, fail the alloc */
422 error = mac_mbuf_init(m, how);
423 if (error)
424 return (error);
425#endif
426 } else
427 m->m_data = m->m_dat;
428 return (0);
429}
430
431/*
432 * The Mbuf master zone destructor.
433 */
434static void
435mb_dtor_mbuf(void *mem, int size, void *arg)
436{
437 struct mbuf *m;
438 unsigned long flags;
439
440 m = (struct mbuf *)mem;
441 flags = (unsigned long)arg;
442
443 if ((flags & MB_NOTAGS) == 0 && (m->m_flags & M_PKTHDR) != 0)
444 m_tag_delete_chain(m, NULL);
445 KASSERT((m->m_flags & M_EXT) == 0, ("%s: M_EXT set", __func__));
446 KASSERT((m->m_flags & M_NOFREE) == 0, ("%s: M_NOFREE set", __func__));
447#ifdef INVARIANTS
448 trash_dtor(mem, size, arg);
449#endif
450}
451
452/*
453 * The Mbuf Packet zone destructor.
454 */
455static void
456mb_dtor_pack(void *mem, int size, void *arg)
457{
458 struct mbuf *m;
459
460 m = (struct mbuf *)mem;
461 if ((m->m_flags & M_PKTHDR) != 0)
462 m_tag_delete_chain(m, NULL);
463
464 /* Make sure we've got a clean cluster back. */
465 KASSERT((m->m_flags & M_EXT) == M_EXT, ("%s: M_EXT not set", __func__));
466 KASSERT(m->m_ext.ext_buf != NULL, ("%s: ext_buf == NULL", __func__));
467 KASSERT(m->m_ext.ext_free == NULL, ("%s: ext_free != NULL", __func__));
468 KASSERT(m->m_ext.ext_arg1 == NULL, ("%s: ext_arg1 != NULL", __func__));
469 KASSERT(m->m_ext.ext_arg2 == NULL, ("%s: ext_arg2 != NULL", __func__));
470 KASSERT(m->m_ext.ext_size == MCLBYTES, ("%s: ext_size != MCLBYTES", __func__));
471 KASSERT(m->m_ext.ext_type == EXT_PACKET, ("%s: ext_type != EXT_PACKET", __func__));
472 KASSERT(*m->m_ext.ref_cnt == 1, ("%s: ref_cnt != 1", __func__));
473#ifdef INVARIANTS
474 trash_dtor(m->m_ext.ext_buf, MCLBYTES, arg);
475#endif
476 /*
477 * If there are processes blocked on zone_clust, waiting for pages
478 * to be freed up, * cause them to be woken up by draining the
479 * packet zone. We are exposed to a race here * (in the check for
480 * the UMA_ZFLAG_FULL) where we might miss the flag set, but that
481 * is deliberate. We don't want to acquire the zone lock for every
482 * mbuf free.
483 */
484 if (uma_zone_exhausted_nolock(zone_clust))
485 zone_drain(zone_pack);
486}
487
488/*
489 * The Cluster and Jumbo[PAGESIZE|9|16] zone constructor.
490 *
491 * Here the 'arg' pointer points to the Mbuf which we
492 * are configuring cluster storage for. If 'arg' is
493 * empty we allocate just the cluster without setting
494 * the mbuf to it. See mbuf.h.
495 */
496static int
497mb_ctor_clust(void *mem, int size, void *arg, int how)
498{
499 struct mbuf *m;
500 u_int *refcnt;
501 int type;
502 uma_zone_t zone;
503
504#ifdef INVARIANTS
505 trash_ctor(mem, size, arg, how);
506#endif
507 switch (size) {
508 case MCLBYTES:
509 type = EXT_CLUSTER;
510 zone = zone_clust;
511 break;
512#if MJUMPAGESIZE != MCLBYTES
513 case MJUMPAGESIZE:
514 type = EXT_JUMBOP;
515 zone = zone_jumbop;
516 break;
517#endif
518 case MJUM9BYTES:
519 type = EXT_JUMBO9;
520 zone = zone_jumbo9;
521 break;
522 case MJUM16BYTES:
523 type = EXT_JUMBO16;
524 zone = zone_jumbo16;
525 break;
526 default:
527 panic("unknown cluster size");
528 break;
529 }
530
531 m = (struct mbuf *)arg;
532 refcnt = uma_find_refcnt(zone, mem);
533 *refcnt = 1;
534 if (m != NULL) {
535 m->m_ext.ext_buf = (caddr_t)mem;
536 m->m_data = m->m_ext.ext_buf;
537 m->m_flags |= M_EXT;
538 m->m_ext.ext_free = NULL;
539 m->m_ext.ext_arg1 = NULL;
540 m->m_ext.ext_arg2 = NULL;
541 m->m_ext.ext_size = size;
542 m->m_ext.ext_type = type;
543 m->m_ext.ref_cnt = refcnt;
544 }
545
546 return (0);
547}
548
549/*
550 * The Mbuf Cluster zone destructor.
551 */
552static void
553mb_dtor_clust(void *mem, int size, void *arg)
554{
555#ifdef INVARIANTS
556 uma_zone_t zone;
557
558 zone = m_getzone(size);
559 KASSERT(*(uma_find_refcnt(zone, mem)) <= 1,
560 ("%s: refcnt incorrect %u", __func__,
561 *(uma_find_refcnt(zone, mem))) );
562
563 trash_dtor(mem, size, arg);
564#endif
565}
566
567/*
568 * The Packet secondary zone's init routine, executed on the
569 * object's transition from mbuf keg slab to zone cache.
570 */
571static int
572mb_zinit_pack(void *mem, int size, int how)
573{
574 struct mbuf *m;
575
576 m = (struct mbuf *)mem; /* m is virgin. */
577 if (uma_zalloc_arg(zone_clust, m, how) == NULL ||
578 m->m_ext.ext_buf == NULL)
579 return (ENOMEM);
580 m->m_ext.ext_type = EXT_PACKET; /* Override. */
581#ifdef INVARIANTS
582 trash_init(m->m_ext.ext_buf, MCLBYTES, how);
583#endif
584 return (0);
585}
586
587/*
588 * The Packet secondary zone's fini routine, executed on the
589 * object's transition from zone cache to keg slab.
590 */
591static void
592mb_zfini_pack(void *mem, int size)
593{
594 struct mbuf *m;
595
596 m = (struct mbuf *)mem;
597#ifdef INVARIANTS
598 trash_fini(m->m_ext.ext_buf, MCLBYTES);
599#endif
600 uma_zfree_arg(zone_clust, m->m_ext.ext_buf, NULL);
601#ifdef INVARIANTS
602 trash_dtor(mem, size, NULL);
603#endif
604}
605
606/*
607 * The "packet" keg constructor.
608 */
609static int
610mb_ctor_pack(void *mem, int size, void *arg, int how)
611{
612 struct mbuf *m;
613 struct mb_args *args;
614#ifdef MAC
615 int error;
616#endif
617 int flags;
618 short type;
619
620 m = (struct mbuf *)mem;
621 args = (struct mb_args *)arg;
622 flags = args->flags;
623 type = args->type;
624
625#ifdef INVARIANTS
626 trash_ctor(m->m_ext.ext_buf, MCLBYTES, arg, how);
627#endif
628 m->m_next = NULL;
629 m->m_nextpkt = NULL;
630 m->m_data = m->m_ext.ext_buf;
631 m->m_len = 0;
632 m->m_flags = (flags | M_EXT);
633 m->m_type = type;
634
635 if (flags & M_PKTHDR) {
636 m->m_pkthdr.rcvif = NULL;
637 m->m_pkthdr.len = 0;
638 m->m_pkthdr.header = NULL;
639 m->m_pkthdr.csum_flags = 0;
640 m->m_pkthdr.csum_data = 0;
641 m->m_pkthdr.tso_segsz = 0;
642 m->m_pkthdr.ether_vtag = 0;
643 SLIST_INIT(&m->m_pkthdr.tags);
644#ifdef MAC
645 /* If the label init fails, fail the alloc */
646 error = mac_mbuf_init(m, how);
647 if (error)
648 return (error);
649#endif
650 }
651 /* m_ext is already initialized. */
652
653 return (0);
654}
655
656/*
657 * This is the protocol drain routine.
658 *
659 * No locks should be held when this is called. The drain routines have to
660 * presently acquire some locks which raises the possibility of lock order
661 * reversal.
662 */
663static void
664mb_reclaim(void *junk)
665{
666 struct domain *dp;
667 struct protosw *pr;
668
669 WITNESS_WARN(WARN_GIANTOK | WARN_SLEEPOK | WARN_PANIC, NULL,
670 "mb_reclaim()");
671
672 for (dp = domains; dp != NULL; dp = dp->dom_next)
673 for (pr = dp->dom_protosw; pr < dp->dom_protoswNPROTOSW; pr++)
674 if (pr->pr_drain != NULL)
675 (*pr->pr_drain)();
676}
| 122
123/* XXX: These should be tuneables. Can't change UMA limits on the fly. */
124static int
125sysctl_nmbclusters(SYSCTL_HANDLER_ARGS)
126{
127 int error, newnmbclusters;
128
129 newnmbclusters = nmbclusters;
130 error = sysctl_handle_int(oidp, &newnmbclusters, 0, req);
131 if (error == 0 && req->newptr) {
132 if (newnmbclusters > nmbclusters) {
133 nmbclusters = newnmbclusters;
134 uma_zone_set_max(zone_clust, nmbclusters);
135 EVENTHANDLER_INVOKE(nmbclusters_change);
136 } else
137 error = EINVAL;
138 }
139 return (error);
140}
141SYSCTL_PROC(_kern_ipc, OID_AUTO, nmbclusters, CTLTYPE_INT|CTLFLAG_RW,
142&nmbclusters, 0, sysctl_nmbclusters, "IU",
143 "Maximum number of mbuf clusters allowed");
144
145static int
146sysctl_nmbjumbop(SYSCTL_HANDLER_ARGS)
147{
148 int error, newnmbjumbop;
149
150 newnmbjumbop = nmbjumbop;
151 error = sysctl_handle_int(oidp, &newnmbjumbop, 0, req);
152 if (error == 0 && req->newptr) {
153 if (newnmbjumbop> nmbjumbop) {
154 nmbjumbop = newnmbjumbop;
155 uma_zone_set_max(zone_jumbop, nmbjumbop);
156 } else
157 error = EINVAL;
158 }
159 return (error);
160}
161SYSCTL_PROC(_kern_ipc, OID_AUTO, nmbjumbop, CTLTYPE_INT|CTLFLAG_RW,
162&nmbjumbop, 0, sysctl_nmbjumbop, "IU",
163 "Maximum number of mbuf page size jumbo clusters allowed");
164
165
166static int
167sysctl_nmbjumbo9(SYSCTL_HANDLER_ARGS)
168{
169 int error, newnmbjumbo9;
170
171 newnmbjumbo9 = nmbjumbo9;
172 error = sysctl_handle_int(oidp, &newnmbjumbo9, 0, req);
173 if (error == 0 && req->newptr) {
174 if (newnmbjumbo9> nmbjumbo9) {
175 nmbjumbo9 = newnmbjumbo9;
176 uma_zone_set_max(zone_jumbo9, nmbjumbo9);
177 } else
178 error = EINVAL;
179 }
180 return (error);
181}
182SYSCTL_PROC(_kern_ipc, OID_AUTO, nmbjumbo9, CTLTYPE_INT|CTLFLAG_RW,
183&nmbjumbo9, 0, sysctl_nmbjumbo9, "IU",
184 "Maximum number of mbuf 9k jumbo clusters allowed");
185
186static int
187sysctl_nmbjumbo16(SYSCTL_HANDLER_ARGS)
188{
189 int error, newnmbjumbo16;
190
191 newnmbjumbo16 = nmbjumbo16;
192 error = sysctl_handle_int(oidp, &newnmbjumbo16, 0, req);
193 if (error == 0 && req->newptr) {
194 if (newnmbjumbo16> nmbjumbo16) {
195 nmbjumbo16 = newnmbjumbo16;
196 uma_zone_set_max(zone_jumbo16, nmbjumbo16);
197 } else
198 error = EINVAL;
199 }
200 return (error);
201}
202SYSCTL_PROC(_kern_ipc, OID_AUTO, nmbjumbo16, CTLTYPE_INT|CTLFLAG_RW,
203&nmbjumbo16, 0, sysctl_nmbjumbo16, "IU",
204 "Maximum number of mbuf 16k jumbo clusters allowed");
205
206
207
208SYSCTL_STRUCT(_kern_ipc, OID_AUTO, mbstat, CTLFLAG_RD, &mbstat, mbstat,
209 "Mbuf general information and statistics");
210
211/*
212 * Zones from which we allocate.
213 */
214uma_zone_t zone_mbuf;
215uma_zone_t zone_clust;
216uma_zone_t zone_pack;
217uma_zone_t zone_jumbop;
218uma_zone_t zone_jumbo9;
219uma_zone_t zone_jumbo16;
220uma_zone_t zone_ext_refcnt;
221
222/*
223 * Local prototypes.
224 */
225static int mb_ctor_mbuf(void *, int, void *, int);
226static int mb_ctor_clust(void *, int, void *, int);
227static int mb_ctor_pack(void *, int, void *, int);
228static void mb_dtor_mbuf(void *, int, void *);
229static void mb_dtor_clust(void *, int, void *);
230static void mb_dtor_pack(void *, int, void *);
231static int mb_zinit_pack(void *, int, int);
232static void mb_zfini_pack(void *, int);
233
234static void mb_reclaim(void *);
235static void mbuf_init(void *);
236static void *mbuf_jumbo_alloc(uma_zone_t, int, u_int8_t *, int);
237static void mbuf_jumbo_free(void *, int, u_int8_t);
238
239static MALLOC_DEFINE(M_JUMBOFRAME, "jumboframes", "mbuf jumbo frame buffers");
240
241/* Ensure that MSIZE doesn't break dtom() - it must be a power of 2 */
242CTASSERT((((MSIZE - 1) ^ MSIZE) + 1) >> 1 == MSIZE);
243
244/*
245 * Initialize FreeBSD Network buffer allocation.
246 */
247SYSINIT(mbuf, SI_SUB_MBUF, SI_ORDER_FIRST, mbuf_init, NULL);
248static void
249mbuf_init(void *dummy)
250{
251
252 /*
253 * Configure UMA zones for Mbufs, Clusters, and Packets.
254 */
255 zone_mbuf = uma_zcreate(MBUF_MEM_NAME, MSIZE,
256 mb_ctor_mbuf, mb_dtor_mbuf,
257#ifdef INVARIANTS
258 trash_init, trash_fini,
259#else
260 NULL, NULL,
261#endif
262 MSIZE - 1, UMA_ZONE_MAXBUCKET);
263
264 zone_clust = uma_zcreate(MBUF_CLUSTER_MEM_NAME, MCLBYTES,
265 mb_ctor_clust, mb_dtor_clust,
266#ifdef INVARIANTS
267 trash_init, trash_fini,
268#else
269 NULL, NULL,
270#endif
271 UMA_ALIGN_PTR, UMA_ZONE_REFCNT);
272 if (nmbclusters > 0)
273 uma_zone_set_max(zone_clust, nmbclusters);
274
275 zone_pack = uma_zsecond_create(MBUF_PACKET_MEM_NAME, mb_ctor_pack,
276 mb_dtor_pack, mb_zinit_pack, mb_zfini_pack, zone_mbuf);
277
278 /* Make jumbo frame zone too. Page size, 9k and 16k. */
279 zone_jumbop = uma_zcreate(MBUF_JUMBOP_MEM_NAME, MJUMPAGESIZE,
280 mb_ctor_clust, mb_dtor_clust,
281#ifdef INVARIANTS
282 trash_init, trash_fini,
283#else
284 NULL, NULL,
285#endif
286 UMA_ALIGN_PTR, UMA_ZONE_REFCNT);
287 if (nmbjumbop > 0)
288 uma_zone_set_max(zone_jumbop, nmbjumbop);
289
290 zone_jumbo9 = uma_zcreate(MBUF_JUMBO9_MEM_NAME, MJUM9BYTES,
291 mb_ctor_clust, mb_dtor_clust,
292#ifdef INVARIANTS
293 trash_init, trash_fini,
294#else
295 NULL, NULL,
296#endif
297 UMA_ALIGN_PTR, UMA_ZONE_REFCNT);
298 if (nmbjumbo9 > 0)
299 uma_zone_set_max(zone_jumbo9, nmbjumbo9);
300 uma_zone_set_allocf(zone_jumbo9, mbuf_jumbo_alloc);
301 uma_zone_set_freef(zone_jumbo9, mbuf_jumbo_free);
302
303 zone_jumbo16 = uma_zcreate(MBUF_JUMBO16_MEM_NAME, MJUM16BYTES,
304 mb_ctor_clust, mb_dtor_clust,
305#ifdef INVARIANTS
306 trash_init, trash_fini,
307#else
308 NULL, NULL,
309#endif
310 UMA_ALIGN_PTR, UMA_ZONE_REFCNT);
311 if (nmbjumbo16 > 0)
312 uma_zone_set_max(zone_jumbo16, nmbjumbo16);
313 uma_zone_set_allocf(zone_jumbo16, mbuf_jumbo_alloc);
314 uma_zone_set_freef(zone_jumbo16, mbuf_jumbo_free);
315
316 zone_ext_refcnt = uma_zcreate(MBUF_EXTREFCNT_MEM_NAME, sizeof(u_int),
317 NULL, NULL,
318 NULL, NULL,
319 UMA_ALIGN_PTR, UMA_ZONE_ZINIT);
320
321 /* uma_prealloc() goes here... */
322
323 /*
324 * Hook event handler for low-memory situation, used to
325 * drain protocols and push data back to the caches (UMA
326 * later pushes it back to VM).
327 */
328 EVENTHANDLER_REGISTER(vm_lowmem, mb_reclaim, NULL,
329 EVENTHANDLER_PRI_FIRST);
330
331 /*
332 * [Re]set counters and local statistics knobs.
333 * XXX Some of these should go and be replaced, but UMA stat
334 * gathering needs to be revised.
335 */
336 mbstat.m_mbufs = 0;
337 mbstat.m_mclusts = 0;
338 mbstat.m_drain = 0;
339 mbstat.m_msize = MSIZE;
340 mbstat.m_mclbytes = MCLBYTES;
341 mbstat.m_minclsize = MINCLSIZE;
342 mbstat.m_mlen = MLEN;
343 mbstat.m_mhlen = MHLEN;
344 mbstat.m_numtypes = MT_NTYPES;
345
346 mbstat.m_mcfail = mbstat.m_mpfail = 0;
347 mbstat.sf_iocnt = 0;
348 mbstat.sf_allocwait = mbstat.sf_allocfail = 0;
349}
350
351/*
352 * UMA backend page allocator for the jumbo frame zones.
353 *
354 * Allocates kernel virtual memory that is backed by contiguous physical
355 * pages.
356 */
357static void *
358mbuf_jumbo_alloc(uma_zone_t zone, int bytes, u_int8_t *flags, int wait)
359{
360
361 /* Inform UMA that this allocator uses kernel_map/object. */
362 *flags = UMA_SLAB_KERNEL;
363 return (contigmalloc(bytes, M_JUMBOFRAME, wait, (vm_paddr_t)0,
364 ~(vm_paddr_t)0, 1, 0));
365}
366
367/*
368 * UMA backend page deallocator for the jumbo frame zones.
369 */
370static void
371mbuf_jumbo_free(void *mem, int size, u_int8_t flags)
372{
373
374 contigfree(mem, size, M_JUMBOFRAME);
375}
376
377/*
378 * Constructor for Mbuf master zone.
379 *
380 * The 'arg' pointer points to a mb_args structure which
381 * contains call-specific information required to support the
382 * mbuf allocation API. See mbuf.h.
383 */
384static int
385mb_ctor_mbuf(void *mem, int size, void *arg, int how)
386{
387 struct mbuf *m;
388 struct mb_args *args;
389#ifdef MAC
390 int error;
391#endif
392 int flags;
393 short type;
394
395#ifdef INVARIANTS
396 trash_ctor(mem, size, arg, how);
397#endif
398 m = (struct mbuf *)mem;
399 args = (struct mb_args *)arg;
400 flags = args->flags;
401 type = args->type;
402
403 /*
404 * The mbuf is initialized later. The caller has the
405 * responsibility to set up any MAC labels too.
406 */
407 if (type == MT_NOINIT)
408 return (0);
409
410 m->m_next = NULL;
411 m->m_nextpkt = NULL;
412 m->m_len = 0;
413 m->m_flags = flags;
414 m->m_type = type;
415 if (flags & M_PKTHDR) {
416 m->m_data = m->m_pktdat;
417 m->m_pkthdr.rcvif = NULL;
418 m->m_pkthdr.header = NULL;
419 m->m_pkthdr.len = 0;
420 m->m_pkthdr.csum_flags = 0;
421 m->m_pkthdr.csum_data = 0;
422 m->m_pkthdr.tso_segsz = 0;
423 m->m_pkthdr.ether_vtag = 0;
424 SLIST_INIT(&m->m_pkthdr.tags);
425#ifdef MAC
426 /* If the label init fails, fail the alloc */
427 error = mac_mbuf_init(m, how);
428 if (error)
429 return (error);
430#endif
431 } else
432 m->m_data = m->m_dat;
433 return (0);
434}
435
436/*
437 * The Mbuf master zone destructor.
438 */
439static void
440mb_dtor_mbuf(void *mem, int size, void *arg)
441{
442 struct mbuf *m;
443 unsigned long flags;
444
445 m = (struct mbuf *)mem;
446 flags = (unsigned long)arg;
447
448 if ((flags & MB_NOTAGS) == 0 && (m->m_flags & M_PKTHDR) != 0)
449 m_tag_delete_chain(m, NULL);
450 KASSERT((m->m_flags & M_EXT) == 0, ("%s: M_EXT set", __func__));
451 KASSERT((m->m_flags & M_NOFREE) == 0, ("%s: M_NOFREE set", __func__));
452#ifdef INVARIANTS
453 trash_dtor(mem, size, arg);
454#endif
455}
456
457/*
458 * The Mbuf Packet zone destructor.
459 */
460static void
461mb_dtor_pack(void *mem, int size, void *arg)
462{
463 struct mbuf *m;
464
465 m = (struct mbuf *)mem;
466 if ((m->m_flags & M_PKTHDR) != 0)
467 m_tag_delete_chain(m, NULL);
468
469 /* Make sure we've got a clean cluster back. */
470 KASSERT((m->m_flags & M_EXT) == M_EXT, ("%s: M_EXT not set", __func__));
471 KASSERT(m->m_ext.ext_buf != NULL, ("%s: ext_buf == NULL", __func__));
472 KASSERT(m->m_ext.ext_free == NULL, ("%s: ext_free != NULL", __func__));
473 KASSERT(m->m_ext.ext_arg1 == NULL, ("%s: ext_arg1 != NULL", __func__));
474 KASSERT(m->m_ext.ext_arg2 == NULL, ("%s: ext_arg2 != NULL", __func__));
475 KASSERT(m->m_ext.ext_size == MCLBYTES, ("%s: ext_size != MCLBYTES", __func__));
476 KASSERT(m->m_ext.ext_type == EXT_PACKET, ("%s: ext_type != EXT_PACKET", __func__));
477 KASSERT(*m->m_ext.ref_cnt == 1, ("%s: ref_cnt != 1", __func__));
478#ifdef INVARIANTS
479 trash_dtor(m->m_ext.ext_buf, MCLBYTES, arg);
480#endif
481 /*
482 * If there are processes blocked on zone_clust, waiting for pages
483 * to be freed up, * cause them to be woken up by draining the
484 * packet zone. We are exposed to a race here * (in the check for
485 * the UMA_ZFLAG_FULL) where we might miss the flag set, but that
486 * is deliberate. We don't want to acquire the zone lock for every
487 * mbuf free.
488 */
489 if (uma_zone_exhausted_nolock(zone_clust))
490 zone_drain(zone_pack);
491}
492
493/*
494 * The Cluster and Jumbo[PAGESIZE|9|16] zone constructor.
495 *
496 * Here the 'arg' pointer points to the Mbuf which we
497 * are configuring cluster storage for. If 'arg' is
498 * empty we allocate just the cluster without setting
499 * the mbuf to it. See mbuf.h.
500 */
501static int
502mb_ctor_clust(void *mem, int size, void *arg, int how)
503{
504 struct mbuf *m;
505 u_int *refcnt;
506 int type;
507 uma_zone_t zone;
508
509#ifdef INVARIANTS
510 trash_ctor(mem, size, arg, how);
511#endif
512 switch (size) {
513 case MCLBYTES:
514 type = EXT_CLUSTER;
515 zone = zone_clust;
516 break;
517#if MJUMPAGESIZE != MCLBYTES
518 case MJUMPAGESIZE:
519 type = EXT_JUMBOP;
520 zone = zone_jumbop;
521 break;
522#endif
523 case MJUM9BYTES:
524 type = EXT_JUMBO9;
525 zone = zone_jumbo9;
526 break;
527 case MJUM16BYTES:
528 type = EXT_JUMBO16;
529 zone = zone_jumbo16;
530 break;
531 default:
532 panic("unknown cluster size");
533 break;
534 }
535
536 m = (struct mbuf *)arg;
537 refcnt = uma_find_refcnt(zone, mem);
538 *refcnt = 1;
539 if (m != NULL) {
540 m->m_ext.ext_buf = (caddr_t)mem;
541 m->m_data = m->m_ext.ext_buf;
542 m->m_flags |= M_EXT;
543 m->m_ext.ext_free = NULL;
544 m->m_ext.ext_arg1 = NULL;
545 m->m_ext.ext_arg2 = NULL;
546 m->m_ext.ext_size = size;
547 m->m_ext.ext_type = type;
548 m->m_ext.ref_cnt = refcnt;
549 }
550
551 return (0);
552}
553
554/*
555 * The Mbuf Cluster zone destructor.
556 */
557static void
558mb_dtor_clust(void *mem, int size, void *arg)
559{
560#ifdef INVARIANTS
561 uma_zone_t zone;
562
563 zone = m_getzone(size);
564 KASSERT(*(uma_find_refcnt(zone, mem)) <= 1,
565 ("%s: refcnt incorrect %u", __func__,
566 *(uma_find_refcnt(zone, mem))) );
567
568 trash_dtor(mem, size, arg);
569#endif
570}
571
572/*
573 * The Packet secondary zone's init routine, executed on the
574 * object's transition from mbuf keg slab to zone cache.
575 */
576static int
577mb_zinit_pack(void *mem, int size, int how)
578{
579 struct mbuf *m;
580
581 m = (struct mbuf *)mem; /* m is virgin. */
582 if (uma_zalloc_arg(zone_clust, m, how) == NULL ||
583 m->m_ext.ext_buf == NULL)
584 return (ENOMEM);
585 m->m_ext.ext_type = EXT_PACKET; /* Override. */
586#ifdef INVARIANTS
587 trash_init(m->m_ext.ext_buf, MCLBYTES, how);
588#endif
589 return (0);
590}
591
592/*
593 * The Packet secondary zone's fini routine, executed on the
594 * object's transition from zone cache to keg slab.
595 */
596static void
597mb_zfini_pack(void *mem, int size)
598{
599 struct mbuf *m;
600
601 m = (struct mbuf *)mem;
602#ifdef INVARIANTS
603 trash_fini(m->m_ext.ext_buf, MCLBYTES);
604#endif
605 uma_zfree_arg(zone_clust, m->m_ext.ext_buf, NULL);
606#ifdef INVARIANTS
607 trash_dtor(mem, size, NULL);
608#endif
609}
610
611/*
612 * The "packet" keg constructor.
613 */
614static int
615mb_ctor_pack(void *mem, int size, void *arg, int how)
616{
617 struct mbuf *m;
618 struct mb_args *args;
619#ifdef MAC
620 int error;
621#endif
622 int flags;
623 short type;
624
625 m = (struct mbuf *)mem;
626 args = (struct mb_args *)arg;
627 flags = args->flags;
628 type = args->type;
629
630#ifdef INVARIANTS
631 trash_ctor(m->m_ext.ext_buf, MCLBYTES, arg, how);
632#endif
633 m->m_next = NULL;
634 m->m_nextpkt = NULL;
635 m->m_data = m->m_ext.ext_buf;
636 m->m_len = 0;
637 m->m_flags = (flags | M_EXT);
638 m->m_type = type;
639
640 if (flags & M_PKTHDR) {
641 m->m_pkthdr.rcvif = NULL;
642 m->m_pkthdr.len = 0;
643 m->m_pkthdr.header = NULL;
644 m->m_pkthdr.csum_flags = 0;
645 m->m_pkthdr.csum_data = 0;
646 m->m_pkthdr.tso_segsz = 0;
647 m->m_pkthdr.ether_vtag = 0;
648 SLIST_INIT(&m->m_pkthdr.tags);
649#ifdef MAC
650 /* If the label init fails, fail the alloc */
651 error = mac_mbuf_init(m, how);
652 if (error)
653 return (error);
654#endif
655 }
656 /* m_ext is already initialized. */
657
658 return (0);
659}
660
661/*
662 * This is the protocol drain routine.
663 *
664 * No locks should be held when this is called. The drain routines have to
665 * presently acquire some locks which raises the possibility of lock order
666 * reversal.
667 */
668static void
669mb_reclaim(void *junk)
670{
671 struct domain *dp;
672 struct protosw *pr;
673
674 WITNESS_WARN(WARN_GIANTOK | WARN_SLEEPOK | WARN_PANIC, NULL,
675 "mb_reclaim()");
676
677 for (dp = domains; dp != NULL; dp = dp->dom_next)
678 for (pr = dp->dom_protosw; pr < dp->dom_protoswNPROTOSW; pr++)
679 if (pr->pr_drain != NULL)
680 (*pr->pr_drain)();
681}
|