30 31#include "opt_mac.h" 32#include "opt_param.h" 33 34#include <sys/param.h> 35#include <sys/mac.h> 36#include <sys/malloc.h> 37#include <sys/systm.h> 38#include <sys/mbuf.h> 39#include <sys/domain.h> 40#include <sys/eventhandler.h> 41#include <sys/kernel.h> 42#include <sys/protosw.h> 43#include <sys/smp.h> 44#include <sys/sysctl.h> 45 46#include <vm/vm.h> 47#include <vm/vm_page.h> 48#include <vm/uma.h> 49#include <vm/uma_int.h> 50#include <vm/uma_dbg.h> 51 52/* 53 * In FreeBSD, Mbufs and Mbuf Clusters are allocated from UMA 54 * Zones. 55 * 56 * Mbuf Clusters (2K, contiguous) are allocated from the Cluster 57 * Zone. The Zone can be capped at kern.ipc.nmbclusters, if the 58 * administrator so desires. 59 * 60 * Mbufs are allocated from a UMA Master Zone called the Mbuf 61 * Zone. 62 * 63 * Additionally, FreeBSD provides a Packet Zone, which it 64 * configures as a Secondary Zone to the Mbuf Master Zone, 65 * thus sharing backend Slab kegs with the Mbuf Master Zone. 66 * 67 * Thus common-case allocations and locking are simplified: 68 * 69 * m_clget() m_getcl() 70 * | | 71 * | .------------>[(Packet Cache)] m_get(), m_gethdr() 72 * | | [ Packet ] | 73 * [(Cluster Cache)] [ Secondary ] [ (Mbuf Cache) ] 74 * [ Cluster Zone ] [ Zone ] [ Mbuf Master Zone ] 75 * | \________ | 76 * [ Cluster Keg ] \ / 77 * | [ Mbuf Keg ] 78 * [ Cluster Slabs ] | 79 * | [ Mbuf Slabs ] 80 * \____________(VM)_________________/ 81 * 82 * 83 * Whenever a object is allocated with uma_zalloc() out of the 84 * one of the Zones its _ctor_ function is executed. The same 85 * for any deallocation through uma_zfree() the _dror_ function 86 * is executed. 87 * 88 * Caches are per-CPU and are filled from the Master Zone. 89 * 90 * Whenever a object is allocated from the underlying global 91 * memory pool it gets pre-initialized with the _zinit_ functions. 92 * When the Keg's are overfull objects get decomissioned with 93 * _zfini_ functions and free'd back to the global memory pool. 94 * 95 */ 96 97int nmbclusters; /* limits number of mbuf clusters */ 98int nmbjumbo9; /* limits number of 9k jumbo clusters */ 99int nmbjumbo16; /* limits number of 16k jumbo clusters */ 100struct mbstat mbstat; 101 102static void 103tunable_mbinit(void *dummy) 104{ 105 106 /* This has to be done before VM init. */ 107 nmbclusters = 1024 + maxusers * 64; 108 TUNABLE_INT_FETCH("kern.ipc.nmbclusters", &nmbclusters); 109} 110SYSINIT(tunable_mbinit, SI_SUB_TUNABLES, SI_ORDER_ANY, tunable_mbinit, NULL); 111 112SYSCTL_DECL(_kern_ipc); 113/* XXX: These should be tuneables. Can't change UMA limits on the fly. */ 114SYSCTL_INT(_kern_ipc, OID_AUTO, nmbclusters, CTLFLAG_RW, &nmbclusters, 0, 115 "Maximum number of mbuf clusters allowed"); 116SYSCTL_INT(_kern_ipc, OID_AUTO, nmbjumbo9, CTLFLAG_RW, &nmbjumbo9, 0, 117 "Maximum number of mbuf 9k jumbo clusters allowed"); 118SYSCTL_INT(_kern_ipc, OID_AUTO, nmbjumbo16, CTLFLAG_RW, &nmbjumbo16, 0, 119 "Maximum number of mbuf 16k jumbo clusters allowed"); 120SYSCTL_STRUCT(_kern_ipc, OID_AUTO, mbstat, CTLFLAG_RD, &mbstat, mbstat, 121 "Mbuf general information and statistics"); 122 123/* 124 * Zones from which we allocate. 125 */ 126uma_zone_t zone_mbuf; 127uma_zone_t zone_clust; 128uma_zone_t zone_pack; 129uma_zone_t zone_jumbo9; 130uma_zone_t zone_jumbo16; 131uma_zone_t zone_ext_refcnt; 132 133/* 134 * Local prototypes. 135 */ 136static int mb_ctor_mbuf(void *, int, void *, int); 137static int mb_ctor_clust(void *, int, void *, int); 138static int mb_ctor_pack(void *, int, void *, int); 139static void mb_dtor_mbuf(void *, int, void *); 140static void mb_dtor_clust(void *, int, void *); 141static void mb_dtor_pack(void *, int, void *); 142static int mb_zinit_pack(void *, int, int); 143static void mb_zfini_pack(void *, int); 144 145static void mb_reclaim(void *); 146static void mbuf_init(void *); 147 148/* Ensure that MSIZE doesn't break dtom() - it must be a power of 2 */ 149CTASSERT((((MSIZE - 1) ^ MSIZE) + 1) >> 1 == MSIZE); 150 151/* 152 * Initialize FreeBSD Network buffer allocation. 153 */ 154SYSINIT(mbuf, SI_SUB_MBUF, SI_ORDER_FIRST, mbuf_init, NULL) 155static void 156mbuf_init(void *dummy) 157{ 158 159 /* 160 * Configure UMA zones for Mbufs, Clusters, and Packets. 161 */ 162 zone_mbuf = uma_zcreate(MBUF_MEM_NAME, MSIZE, 163 mb_ctor_mbuf, mb_dtor_mbuf, 164#ifdef INVARIANTS 165 trash_init, trash_fini, 166#else 167 NULL, NULL, 168#endif 169 MSIZE - 1, UMA_ZONE_MAXBUCKET); 170 171 zone_clust = uma_zcreate(MBUF_CLUSTER_MEM_NAME, MCLBYTES, 172 mb_ctor_clust, mb_dtor_clust, 173#ifdef INVARIANTS 174 trash_init, trash_fini, 175#else 176 NULL, NULL, 177#endif 178 UMA_ALIGN_PTR, UMA_ZONE_REFCNT); 179 if (nmbclusters > 0) 180 uma_zone_set_max(zone_clust, nmbclusters); 181 182 zone_pack = uma_zsecond_create(MBUF_PACKET_MEM_NAME, mb_ctor_pack, 183 mb_dtor_pack, mb_zinit_pack, mb_zfini_pack, zone_mbuf); 184 185 /* Make jumbo frame zone too. 9k and 16k. */ 186 zone_jumbo9 = uma_zcreate(MBUF_JUMBO9_MEM_NAME, MJUM9BYTES, 187 mb_ctor_clust, mb_dtor_clust, 188#ifdef INVARIANTS 189 trash_init, trash_fini, 190#else 191 NULL, NULL, 192#endif 193 UMA_ALIGN_PTR, UMA_ZONE_REFCNT); 194 if (nmbjumbo9 > 0) 195 uma_zone_set_max(zone_jumbo9, nmbjumbo9); 196 197 zone_jumbo16 = uma_zcreate(MBUF_JUMBO16_MEM_NAME, MJUM16BYTES, 198 mb_ctor_clust, mb_dtor_clust, 199#ifdef INVARIANTS 200 trash_init, trash_fini, 201#else 202 NULL, NULL, 203#endif 204 UMA_ALIGN_PTR, UMA_ZONE_REFCNT); 205 if (nmbjumbo16 > 0) 206 uma_zone_set_max(zone_jumbo16, nmbjumbo16); 207 208 zone_ext_refcnt = uma_zcreate(MBUF_EXTREFCNT_MEM_NAME, sizeof(u_int), 209 NULL, NULL, 210 NULL, NULL, 211 UMA_ALIGN_PTR, UMA_ZONE_ZINIT); 212 213 /* uma_prealloc() goes here... */ 214 215 /* 216 * Hook event handler for low-memory situation, used to 217 * drain protocols and push data back to the caches (UMA 218 * later pushes it back to VM). 219 */ 220 EVENTHANDLER_REGISTER(vm_lowmem, mb_reclaim, NULL, 221 EVENTHANDLER_PRI_FIRST); 222 223 /* 224 * [Re]set counters and local statistics knobs. 225 * XXX Some of these should go and be replaced, but UMA stat 226 * gathering needs to be revised. 227 */ 228 mbstat.m_mbufs = 0; 229 mbstat.m_mclusts = 0; 230 mbstat.m_drain = 0; 231 mbstat.m_msize = MSIZE; 232 mbstat.m_mclbytes = MCLBYTES; 233 mbstat.m_minclsize = MINCLSIZE; 234 mbstat.m_mlen = MLEN; 235 mbstat.m_mhlen = MHLEN; 236 mbstat.m_numtypes = MT_NTYPES; 237 238 mbstat.m_mcfail = mbstat.m_mpfail = 0; 239 mbstat.sf_iocnt = 0; 240 mbstat.sf_allocwait = mbstat.sf_allocfail = 0; 241} 242 243/* 244 * Constructor for Mbuf master zone. 245 * 246 * The 'arg' pointer points to a mb_args structure which 247 * contains call-specific information required to support the 248 * mbuf allocation API. See mbuf.h. 249 */ 250static int 251mb_ctor_mbuf(void *mem, int size, void *arg, int how) 252{ 253 struct mbuf *m; 254 struct mb_args *args; 255#ifdef MAC 256 int error; 257#endif 258 int flags; 259 short type; 260 261#ifdef INVARIANTS 262 trash_ctor(mem, size, arg, how); 263#endif 264 m = (struct mbuf *)mem; 265 args = (struct mb_args *)arg; 266 flags = args->flags; 267 type = args->type; 268 269 /* 270 * The mbuf is initialized later. The caller has the 271 * responseability to setup any MAC labels too. 272 */ 273 if (type == MT_NOINIT) 274 return (0); 275 276 m->m_next = NULL; 277 m->m_nextpkt = NULL; 278 m->m_len = 0; 279 m->m_flags = flags; 280 m->m_type = type; 281 if (flags & M_PKTHDR) { 282 m->m_data = m->m_pktdat; 283 m->m_pkthdr.rcvif = NULL; 284 m->m_pkthdr.len = 0; 285 m->m_pkthdr.header = NULL; 286 m->m_pkthdr.csum_flags = 0; 287 m->m_pkthdr.csum_data = 0; 288 SLIST_INIT(&m->m_pkthdr.tags); 289#ifdef MAC 290 /* If the label init fails, fail the alloc */ 291 error = mac_init_mbuf(m, how); 292 if (error) 293 return (error); 294#endif 295 } else 296 m->m_data = m->m_dat; 297 return (0); 298} 299 300/* 301 * The Mbuf master zone destructor. 302 */ 303static void 304mb_dtor_mbuf(void *mem, int size, void *arg) 305{ 306 struct mbuf *m; 307 308 m = (struct mbuf *)mem; 309 if ((m->m_flags & M_PKTHDR) != 0) 310 m_tag_delete_chain(m, NULL); 311 KASSERT((m->m_flags & M_EXT) == 0, ("%s: M_EXT set", __func__)); 312#ifdef INVARIANTS 313 trash_dtor(mem, size, arg); 314#endif 315} 316 317/* 318 * The Mbuf Packet zone destructor. 319 */ 320static void 321mb_dtor_pack(void *mem, int size, void *arg) 322{ 323 struct mbuf *m; 324 325 m = (struct mbuf *)mem; 326 if ((m->m_flags & M_PKTHDR) != 0) 327 m_tag_delete_chain(m, NULL); 328 329 /* Make sure we've got a clean cluster back. */ 330 KASSERT((m->m_flags & M_EXT) == M_EXT, ("%s: M_EXT not set", __func__)); 331 KASSERT(m->m_ext.ext_buf != NULL, ("%s: ext_buf == NULL", __func__)); 332 KASSERT(m->m_ext.ext_free == NULL, ("%s: ext_free != NULL", __func__)); 333 KASSERT(m->m_ext.ext_args == NULL, ("%s: ext_args != NULL", __func__)); 334 KASSERT(m->m_ext.ext_size == MCLBYTES, ("%s: ext_size != MCLBYTES", __func__));
|
336 KASSERT(*m->m_ext.ref_cnt == 1, ("%s: ref_cnt != 1", __func__)); 337#ifdef INVARIANTS 338 trash_dtor(m->m_ext.ext_buf, MCLBYTES, arg); 339#endif 340} 341 342/* 343 * The Cluster and Jumbo[9|16] zone constructor. 344 * 345 * Here the 'arg' pointer points to the Mbuf which we 346 * are configuring cluster storage for. If 'arg' is 347 * empty we allocate just the cluster without setting 348 * the mbuf to it. See mbuf.h. 349 */ 350static int 351mb_ctor_clust(void *mem, int size, void *arg, int how) 352{ 353 struct mbuf *m; 354 u_int *refcnt; 355 int type = 0; 356 357#ifdef INVARIANTS 358 trash_ctor(mem, size, arg, how); 359#endif 360 m = (struct mbuf *)arg; 361 if (m != NULL) { 362 switch (size) { 363 case MCLBYTES: 364 type = EXT_CLUSTER; 365 break; 366 case MJUM9BYTES: 367 type = EXT_JUMBO9; 368 break; 369 case MJUM16BYTES: 370 type = EXT_JUMBO16; 371 break; 372 default: 373 panic("unknown cluster size"); 374 break; 375 } 376 m->m_ext.ext_buf = (caddr_t)mem; 377 m->m_data = m->m_ext.ext_buf; 378 m->m_flags |= M_EXT; 379 m->m_ext.ext_free = NULL; 380 m->m_ext.ext_args = NULL; 381 m->m_ext.ext_size = size; 382 m->m_ext.ext_type = type; 383 m->m_ext.ref_cnt = uma_find_refcnt(zone_clust, mem); 384 *m->m_ext.ref_cnt = 1; 385 } else { 386 refcnt = uma_find_refcnt(zone_clust, mem); 387 *refcnt = 1; 388 } 389 return (0); 390} 391 392/* 393 * The Mbuf Cluster zone destructor. 394 */ 395static void 396mb_dtor_clust(void *mem, int size, void *arg) 397{ 398 399 KASSERT(*(uma_find_refcnt(zone_clust, mem)) <= 1, 400 ("%s: refcnt incorrect %u", __func__, 401 *(uma_find_refcnt(zone_clust, mem))) ); 402#ifdef INVARIANTS 403 trash_dtor(mem, size, arg); 404#endif 405} 406 407/* 408 * The Packet secondary zone's init routine, executed on the 409 * object's transition from mbuf keg slab to zone cache. 410 */ 411static int 412mb_zinit_pack(void *mem, int size, int how) 413{ 414 struct mbuf *m; 415 416 m = (struct mbuf *)mem; /* m is virgin. */ 417 uma_zalloc_arg(zone_clust, m, how); 418 if (m->m_ext.ext_buf == NULL) 419 return (ENOMEM); 420 m->m_ext.ext_type = EXT_PACKET; /* Override. */ 421#ifdef INVARIANTS 422 trash_init(m->m_ext.ext_buf, MCLBYTES, how); 423#endif 424 return (0); 425} 426 427/* 428 * The Packet secondary zone's fini routine, executed on the 429 * object's transition from zone cache to keg slab. 430 */ 431static void 432mb_zfini_pack(void *mem, int size) 433{ 434 struct mbuf *m; 435 436 m = (struct mbuf *)mem; 437#ifdef INVARIANTS 438 trash_fini(m->m_ext.ext_buf, MCLBYTES); 439#endif 440 uma_zfree_arg(zone_clust, m->m_ext.ext_buf, NULL); 441#ifdef INVARIANTS 442 trash_dtor(mem, size, NULL); 443#endif 444} 445 446/* 447 * The "packet" keg constructor. 448 */ 449static int 450mb_ctor_pack(void *mem, int size, void *arg, int how) 451{ 452 struct mbuf *m; 453 struct mb_args *args; 454#ifdef MAC 455 int error; 456#endif 457 int flags; 458 short type; 459 460 m = (struct mbuf *)mem; 461 args = (struct mb_args *)arg; 462 flags = args->flags; 463 type = args->type; 464 465#ifdef INVARIANTS 466 trash_ctor(m->m_ext.ext_buf, MCLBYTES, arg, how); 467#endif 468 m->m_next = NULL; 469 m->m_nextpkt = NULL; 470 m->m_data = m->m_ext.ext_buf; 471 m->m_len = 0; 472 m->m_flags = (flags | M_EXT); 473 m->m_type = type; 474 475 if (flags & M_PKTHDR) { 476 m->m_pkthdr.rcvif = NULL; 477 m->m_pkthdr.len = 0; 478 m->m_pkthdr.header = NULL; 479 m->m_pkthdr.csum_flags = 0; 480 m->m_pkthdr.csum_data = 0; 481 SLIST_INIT(&m->m_pkthdr.tags); 482#ifdef MAC 483 /* If the label init fails, fail the alloc */ 484 error = mac_init_mbuf(m, how); 485 if (error) 486 return (error); 487#endif 488 } 489 /* m_ext is already initialized. */ 490 491 return (0); 492} 493 494/* 495 * This is the protocol drain routine. 496 * 497 * No locks should be held when this is called. The drain routines have to 498 * presently acquire some locks which raises the possibility of lock order 499 * reversal. 500 */ 501static void 502mb_reclaim(void *junk) 503{ 504 struct domain *dp; 505 struct protosw *pr; 506 507 WITNESS_WARN(WARN_GIANTOK | WARN_SLEEPOK | WARN_PANIC, NULL, 508 "mb_reclaim()"); 509 510 for (dp = domains; dp != NULL; dp = dp->dom_next) 511 for (pr = dp->dom_protosw; pr < dp->dom_protoswNPROTOSW; pr++) 512 if (pr->pr_drain != NULL) 513 (*pr->pr_drain)(); 514}
|