1 /*- 2 * Copyright (c) 2007-2009 Robert N. M. Watson 3 * Copyright (c) 2010-2011 Juniper Networks, Inc. 4 * All rights reserved. 5 * 6 * This software was developed by Robert N. M. Watson under contract 7 * to Juniper Networks, Inc. 8 * 9 * Redistribution and use in source and binary forms, with or without 10 * modification, are permitted provided that the following conditions 11 * are met: 12 * 1. Redistributions of source code must retain the above copyright 13 * notice, this list of conditions and the following disclaimer. 14 * 2. Redistributions in binary form must reproduce the above copyright 15 * notice, this list of conditions and the following disclaimer in the 16 * documentation and/or other materials provided with the distribution. 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 19 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 21 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 22 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 23 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 24 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 25 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 26 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 27 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 28 * SUCH DAMAGE. 29 */ 30 31 #include <sys/cdefs.h> 32 __FBSDID("$FreeBSD$"); 33 34 /* 35 * netisr is a packet dispatch service, allowing synchronous (directly 36 * dispatched) and asynchronous (deferred dispatch) processing of packets by 37 * registered protocol handlers. Callers pass a protocol identifier and 38 * packet to netisr, along with a direct dispatch hint, and work will either 39 * be immediately processed by the registered handler, or passed to a 40 * software interrupt (SWI) thread for deferred dispatch. Callers will 41 * generally select one or the other based on: 42 * 43 * - Whether directly dispatching a netisr handler lead to code reentrance or 44 * lock recursion, such as entering the socket code from the socket code. 45 * - Whether directly dispatching a netisr handler lead to recursive 46 * processing, such as when decapsulating several wrapped layers of tunnel 47 * information (IPSEC within IPSEC within ...). 48 * 49 * Maintaining ordering for protocol streams is a critical design concern. 50 * Enforcing ordering limits the opportunity for concurrency, but maintains 51 * the strong ordering requirements found in some protocols, such as TCP. Of 52 * related concern is CPU affinity--it is desirable to process all data 53 * associated with a particular stream on the same CPU over time in order to 54 * avoid acquiring locks associated with the connection on different CPUs, 55 * keep connection data in one cache, and to generally encourage associated 56 * user threads to live on the same CPU as the stream. It's also desirable 57 * to avoid lock migration and contention where locks are associated with 58 * more than one flow. 59 * 60 * netisr supports several policy variations, represented by the 61 * NETISR_POLICY_* constants, allowing protocols to play various roles in 62 * identifying flows, assigning work to CPUs, etc. These are described in 63 * netisr.h. 64 */ 65 66 #include "opt_ddb.h" 67 #include "opt_device_polling.h" 68 69 #include <sys/param.h> 70 #include <sys/bus.h> 71 #include <sys/kernel.h> 72 #include <sys/kthread.h> 73 #include <sys/malloc.h> 74 #include <sys/interrupt.h> 75 #include <sys/lock.h> 76 #include <sys/mbuf.h> 77 #include <sys/mutex.h> 78 #include <sys/pcpu.h> 79 #include <sys/proc.h> 80 #include <sys/rmlock.h> 81 #include <sys/sched.h> 82 #include <sys/smp.h> 83 #include <sys/socket.h> 84 #include <sys/sysctl.h> 85 #include <sys/systm.h> 86 87 #ifdef DDB 88 #include <ddb/ddb.h> 89 #endif 90 91 #define _WANT_NETISR_INTERNAL /* Enable definitions from netisr_internal.h */ 92 #include <net/if.h> 93 #include <net/if_var.h> 94 #include <net/netisr.h> 95 #include <net/netisr_internal.h> 96 #include <net/vnet.h> 97 98 /*- 99 * Synchronize use and modification of the registered netisr data structures; 100 * acquire a read lock while modifying the set of registered protocols to 101 * prevent partially registered or unregistered protocols from being run. 102 * 103 * The following data structures and fields are protected by this lock: 104 * 105 * - The netisr_proto array, including all fields of struct netisr_proto. 106 * - The nws array, including all fields of struct netisr_worker. 107 * - The nws_array array. 108 * 109 * Note: the NETISR_LOCKING define controls whether read locks are acquired 110 * in packet processing paths requiring netisr registration stability. This 111 * is disabled by default as it can lead to measurable performance 112 * degradation even with rmlocks (3%-6% for loopback ping-pong traffic), and 113 * because netisr registration and unregistration is extremely rare at 114 * runtime. If it becomes more common, this decision should be revisited. 115 * 116 * XXXRW: rmlocks don't support assertions. 117 */ 118 static struct rmlock netisr_rmlock; 119 #define NETISR_LOCK_INIT() rm_init_flags(&netisr_rmlock, "netisr", \ 120 RM_NOWITNESS) 121 #define NETISR_LOCK_ASSERT() 122 #define NETISR_RLOCK(tracker) rm_rlock(&netisr_rmlock, (tracker)) 123 #define NETISR_RUNLOCK(tracker) rm_runlock(&netisr_rmlock, (tracker)) 124 #define NETISR_WLOCK() rm_wlock(&netisr_rmlock) 125 #define NETISR_WUNLOCK() rm_wunlock(&netisr_rmlock) 126 /* #define NETISR_LOCKING */ 127 128 static SYSCTL_NODE(_net, OID_AUTO, isr, CTLFLAG_RW, 0, "netisr"); 129 130 /*- 131 * Three global direct dispatch policies are supported: 132 * 133 * NETISR_DISPATCH_DEFERRED: All work is deferred for a netisr, regardless of 134 * context (may be overriden by protocols). 135 * 136 * NETISR_DISPATCH_HYBRID: If the executing context allows direct dispatch, 137 * and we're running on the CPU the work would be performed on, then direct 138 * dispatch it if it wouldn't violate ordering constraints on the workstream. 139 * 140 * NETISR_DISPATCH_DIRECT: If the executing context allows direct dispatch, 141 * always direct dispatch. (The default.) 142 * 143 * Notice that changing the global policy could lead to short periods of 144 * misordered processing, but this is considered acceptable as compared to 145 * the complexity of enforcing ordering during policy changes. Protocols can 146 * override the global policy (when they're not doing that, they select 147 * NETISR_DISPATCH_DEFAULT). 148 */ 149 #define NETISR_DISPATCH_POLICY_DEFAULT NETISR_DISPATCH_DIRECT 150 #define NETISR_DISPATCH_POLICY_MAXSTR 20 /* Used for temporary buffers. */ 151 static u_int netisr_dispatch_policy = NETISR_DISPATCH_POLICY_DEFAULT; 152 static int sysctl_netisr_dispatch_policy(SYSCTL_HANDLER_ARGS); 153 SYSCTL_PROC(_net_isr, OID_AUTO, dispatch, CTLTYPE_STRING | CTLFLAG_RWTUN, 154 0, 0, sysctl_netisr_dispatch_policy, "A", 155 "netisr dispatch policy"); 156 157 /* 158 * Allow the administrator to limit the number of threads (CPUs) to use for 159 * netisr. We don't check netisr_maxthreads before creating the thread for 160 * CPU 0. This must be set at boot. We will create at most one thread per CPU. 161 * By default we initialize this to 1 which would assign just 1 cpu (cpu0) and 162 * therefore only 1 workstream. If set to -1, netisr would use all cpus 163 * (mp_ncpus) and therefore would have those many workstreams. One workstream 164 * per thread (CPU). 165 */ 166 static int netisr_maxthreads = 1; /* Max number of threads. */ 167 SYSCTL_INT(_net_isr, OID_AUTO, maxthreads, CTLFLAG_RDTUN, 168 &netisr_maxthreads, 0, 169 "Use at most this many CPUs for netisr processing"); 170 171 static int netisr_bindthreads = 0; /* Bind threads to CPUs. */ 172 SYSCTL_INT(_net_isr, OID_AUTO, bindthreads, CTLFLAG_RDTUN, 173 &netisr_bindthreads, 0, "Bind netisr threads to CPUs."); 174 175 /* 176 * Limit per-workstream mbuf queue limits s to at most net.isr.maxqlimit, 177 * both for initial configuration and later modification using 178 * netisr_setqlimit(). 179 */ 180 #define NETISR_DEFAULT_MAXQLIMIT 10240 181 static u_int netisr_maxqlimit = NETISR_DEFAULT_MAXQLIMIT; 182 SYSCTL_UINT(_net_isr, OID_AUTO, maxqlimit, CTLFLAG_RDTUN, 183 &netisr_maxqlimit, 0, 184 "Maximum netisr per-protocol, per-CPU queue depth."); 185 186 /* 187 * The default per-workstream mbuf queue limit for protocols that don't 188 * initialize the nh_qlimit field of their struct netisr_handler. If this is 189 * set above netisr_maxqlimit, we truncate it to the maximum during boot. 190 */ 191 #define NETISR_DEFAULT_DEFAULTQLIMIT 256 192 static u_int netisr_defaultqlimit = NETISR_DEFAULT_DEFAULTQLIMIT; 193 SYSCTL_UINT(_net_isr, OID_AUTO, defaultqlimit, CTLFLAG_RDTUN, 194 &netisr_defaultqlimit, 0, 195 "Default netisr per-protocol, per-CPU queue limit if not set by protocol"); 196 197 /* 198 * Store and export the compile-time constant NETISR_MAXPROT limit on the 199 * number of protocols that can register with netisr at a time. This is 200 * required for crashdump analysis, as it sizes netisr_proto[]. 201 */ 202 static u_int netisr_maxprot = NETISR_MAXPROT; 203 SYSCTL_UINT(_net_isr, OID_AUTO, maxprot, CTLFLAG_RD, 204 &netisr_maxprot, 0, 205 "Compile-time limit on the number of protocols supported by netisr."); 206 207 /* 208 * The netisr_proto array describes all registered protocols, indexed by 209 * protocol number. See netisr_internal.h for more details. 210 */ 211 static struct netisr_proto netisr_proto[NETISR_MAXPROT]; 212 213 /* 214 * Per-CPU workstream data. See netisr_internal.h for more details. 215 */ 216 DPCPU_DEFINE(struct netisr_workstream, nws); 217 218 /* 219 * Map contiguous values between 0 and nws_count into CPU IDs appropriate for 220 * accessing workstreams. This allows constructions of the form 221 * DPCPU_ID_GET(nws_array[arbitraryvalue % nws_count], nws). 222 */ 223 static u_int nws_array[MAXCPU]; 224 225 /* 226 * Number of registered workstreams. Will be at most the number of running 227 * CPUs once fully started. 228 */ 229 static u_int nws_count; 230 SYSCTL_UINT(_net_isr, OID_AUTO, numthreads, CTLFLAG_RD, 231 &nws_count, 0, "Number of extant netisr threads."); 232 233 /* 234 * Synchronization for each workstream: a mutex protects all mutable fields 235 * in each stream, including per-protocol state (mbuf queues). The SWI is 236 * woken up if asynchronous dispatch is required. 237 */ 238 #define NWS_LOCK(s) mtx_lock(&(s)->nws_mtx) 239 #define NWS_LOCK_ASSERT(s) mtx_assert(&(s)->nws_mtx, MA_OWNED) 240 #define NWS_UNLOCK(s) mtx_unlock(&(s)->nws_mtx) 241 #define NWS_SIGNAL(s) swi_sched((s)->nws_swi_cookie, 0) 242 243 /* 244 * Utility routines for protocols that implement their own mapping of flows 245 * to CPUs. 246 */ 247 u_int 248 netisr_get_cpucount(void) 249 { 250 251 return (nws_count); 252 } 253 254 u_int 255 netisr_get_cpuid(u_int cpunumber) 256 { 257 258 KASSERT(cpunumber < nws_count, ("%s: %u > %u", __func__, cpunumber, 259 nws_count)); 260 261 return (nws_array[cpunumber]); 262 } 263 264 /* 265 * The default implementation of flow -> CPU ID mapping. 266 * 267 * Non-static so that protocols can use it to map their own work to specific 268 * CPUs in a manner consistent to netisr for affinity purposes. 269 */ 270 u_int 271 netisr_default_flow2cpu(u_int flowid) 272 { 273 274 return (nws_array[flowid % nws_count]); 275 } 276 277 /* 278 * Dispatch tunable and sysctl configuration. 279 */ 280 struct netisr_dispatch_table_entry { 281 u_int ndte_policy; 282 const char *ndte_policy_str; 283 }; 284 static const struct netisr_dispatch_table_entry netisr_dispatch_table[] = { 285 { NETISR_DISPATCH_DEFAULT, "default" }, 286 { NETISR_DISPATCH_DEFERRED, "deferred" }, 287 { NETISR_DISPATCH_HYBRID, "hybrid" }, 288 { NETISR_DISPATCH_DIRECT, "direct" }, 289 }; 290 static const u_int netisr_dispatch_table_len = 291 (sizeof(netisr_dispatch_table) / sizeof(netisr_dispatch_table[0])); 292 293 static void 294 netisr_dispatch_policy_to_str(u_int dispatch_policy, char *buffer, 295 u_int buflen) 296 { 297 const struct netisr_dispatch_table_entry *ndtep; 298 const char *str; 299 u_int i; 300 301 str = "unknown"; 302 for (i = 0; i < netisr_dispatch_table_len; i++) { 303 ndtep = &netisr_dispatch_table[i]; 304 if (ndtep->ndte_policy == dispatch_policy) { 305 str = ndtep->ndte_policy_str; 306 break; 307 } 308 } 309 snprintf(buffer, buflen, "%s", str); 310 } 311 312 static int 313 netisr_dispatch_policy_from_str(const char *str, u_int *dispatch_policyp) 314 { 315 const struct netisr_dispatch_table_entry *ndtep; 316 u_int i; 317 318 for (i = 0; i < netisr_dispatch_table_len; i++) { 319 ndtep = &netisr_dispatch_table[i]; 320 if (strcmp(ndtep->ndte_policy_str, str) == 0) { 321 *dispatch_policyp = ndtep->ndte_policy; 322 return (0); 323 } 324 } 325 return (EINVAL); 326 } 327 328 static int 329 sysctl_netisr_dispatch_policy(SYSCTL_HANDLER_ARGS) 330 { 331 char tmp[NETISR_DISPATCH_POLICY_MAXSTR]; 332 u_int dispatch_policy; 333 int error; 334 335 netisr_dispatch_policy_to_str(netisr_dispatch_policy, tmp, 336 sizeof(tmp)); 337 error = sysctl_handle_string(oidp, tmp, sizeof(tmp), req); 338 if (error == 0 && req->newptr != NULL) { 339 error = netisr_dispatch_policy_from_str(tmp, 340 &dispatch_policy); 341 if (error == 0 && dispatch_policy == NETISR_DISPATCH_DEFAULT) 342 error = EINVAL; 343 if (error == 0) 344 netisr_dispatch_policy = dispatch_policy; 345 } 346 return (error); 347 } 348 349 /* 350 * Register a new netisr handler, which requires initializing per-protocol 351 * fields for each workstream. All netisr work is briefly suspended while 352 * the protocol is installed. 353 */ 354 void 355 netisr_register(const struct netisr_handler *nhp) 356 { 357 struct netisr_work *npwp; 358 const char *name; 359 u_int i, proto; 360 361 proto = nhp->nh_proto; 362 name = nhp->nh_name; 363 364 /* 365 * Test that the requested registration is valid. 366 */ 367 KASSERT(nhp->nh_name != NULL, 368 ("%s: nh_name NULL for %u", __func__, proto)); 369 KASSERT(nhp->nh_handler != NULL, 370 ("%s: nh_handler NULL for %s", __func__, name)); 371 KASSERT(nhp->nh_policy == NETISR_POLICY_SOURCE || 372 nhp->nh_policy == NETISR_POLICY_FLOW || 373 nhp->nh_policy == NETISR_POLICY_CPU, 374 ("%s: unsupported nh_policy %u for %s", __func__, 375 nhp->nh_policy, name)); 376 KASSERT(nhp->nh_policy == NETISR_POLICY_FLOW || 377 nhp->nh_m2flow == NULL, 378 ("%s: nh_policy != FLOW but m2flow defined for %s", __func__, 379 name)); 380 KASSERT(nhp->nh_policy == NETISR_POLICY_CPU || nhp->nh_m2cpuid == NULL, 381 ("%s: nh_policy != CPU but m2cpuid defined for %s", __func__, 382 name)); 383 KASSERT(nhp->nh_policy != NETISR_POLICY_CPU || nhp->nh_m2cpuid != NULL, 384 ("%s: nh_policy == CPU but m2cpuid not defined for %s", __func__, 385 name)); 386 KASSERT(nhp->nh_dispatch == NETISR_DISPATCH_DEFAULT || 387 nhp->nh_dispatch == NETISR_DISPATCH_DEFERRED || 388 nhp->nh_dispatch == NETISR_DISPATCH_HYBRID || 389 nhp->nh_dispatch == NETISR_DISPATCH_DIRECT, 390 ("%s: invalid nh_dispatch (%u)", __func__, nhp->nh_dispatch)); 391 392 KASSERT(proto < NETISR_MAXPROT, 393 ("%s(%u, %s): protocol too big", __func__, proto, name)); 394 395 /* 396 * Test that no existing registration exists for this protocol. 397 */ 398 NETISR_WLOCK(); 399 KASSERT(netisr_proto[proto].np_name == NULL, 400 ("%s(%u, %s): name present", __func__, proto, name)); 401 KASSERT(netisr_proto[proto].np_handler == NULL, 402 ("%s(%u, %s): handler present", __func__, proto, name)); 403 404 netisr_proto[proto].np_name = name; 405 netisr_proto[proto].np_handler = nhp->nh_handler; 406 netisr_proto[proto].np_m2flow = nhp->nh_m2flow; 407 netisr_proto[proto].np_m2cpuid = nhp->nh_m2cpuid; 408 netisr_proto[proto].np_drainedcpu = nhp->nh_drainedcpu; 409 if (nhp->nh_qlimit == 0) 410 netisr_proto[proto].np_qlimit = netisr_defaultqlimit; 411 else if (nhp->nh_qlimit > netisr_maxqlimit) { 412 printf("%s: %s requested queue limit %u capped to " 413 "net.isr.maxqlimit %u\n", __func__, name, nhp->nh_qlimit, 414 netisr_maxqlimit); 415 netisr_proto[proto].np_qlimit = netisr_maxqlimit; 416 } else 417 netisr_proto[proto].np_qlimit = nhp->nh_qlimit; 418 netisr_proto[proto].np_policy = nhp->nh_policy; 419 netisr_proto[proto].np_dispatch = nhp->nh_dispatch; 420 CPU_FOREACH(i) { 421 npwp = &(DPCPU_ID_PTR(i, nws))->nws_work[proto]; 422 bzero(npwp, sizeof(*npwp)); 423 npwp->nw_qlimit = netisr_proto[proto].np_qlimit; 424 } 425 NETISR_WUNLOCK(); 426 } 427 428 /* 429 * Clear drop counters across all workstreams for a protocol. 430 */ 431 void 432 netisr_clearqdrops(const struct netisr_handler *nhp) 433 { 434 struct netisr_work *npwp; 435 #ifdef INVARIANTS 436 const char *name; 437 #endif 438 u_int i, proto; 439 440 proto = nhp->nh_proto; 441 #ifdef INVARIANTS 442 name = nhp->nh_name; 443 #endif 444 KASSERT(proto < NETISR_MAXPROT, 445 ("%s(%u): protocol too big for %s", __func__, proto, name)); 446 447 NETISR_WLOCK(); 448 KASSERT(netisr_proto[proto].np_handler != NULL, 449 ("%s(%u): protocol not registered for %s", __func__, proto, 450 name)); 451 452 CPU_FOREACH(i) { 453 npwp = &(DPCPU_ID_PTR(i, nws))->nws_work[proto]; 454 npwp->nw_qdrops = 0; 455 } 456 NETISR_WUNLOCK(); 457 } 458 459 /* 460 * Query current drop counters across all workstreams for a protocol. 461 */ 462 void 463 netisr_getqdrops(const struct netisr_handler *nhp, u_int64_t *qdropp) 464 { 465 struct netisr_work *npwp; 466 struct rm_priotracker tracker; 467 #ifdef INVARIANTS 468 const char *name; 469 #endif 470 u_int i, proto; 471 472 *qdropp = 0; 473 proto = nhp->nh_proto; 474 #ifdef INVARIANTS 475 name = nhp->nh_name; 476 #endif 477 KASSERT(proto < NETISR_MAXPROT, 478 ("%s(%u): protocol too big for %s", __func__, proto, name)); 479 480 NETISR_RLOCK(&tracker); 481 KASSERT(netisr_proto[proto].np_handler != NULL, 482 ("%s(%u): protocol not registered for %s", __func__, proto, 483 name)); 484 485 CPU_FOREACH(i) { 486 npwp = &(DPCPU_ID_PTR(i, nws))->nws_work[proto]; 487 *qdropp += npwp->nw_qdrops; 488 } 489 NETISR_RUNLOCK(&tracker); 490 } 491 492 /* 493 * Query current per-workstream queue limit for a protocol. 494 */ 495 void 496 netisr_getqlimit(const struct netisr_handler *nhp, u_int *qlimitp) 497 { 498 struct rm_priotracker tracker; 499 #ifdef INVARIANTS 500 const char *name; 501 #endif 502 u_int proto; 503 504 proto = nhp->nh_proto; 505 #ifdef INVARIANTS 506 name = nhp->nh_name; 507 #endif 508 KASSERT(proto < NETISR_MAXPROT, 509 ("%s(%u): protocol too big for %s", __func__, proto, name)); 510 511 NETISR_RLOCK(&tracker); 512 KASSERT(netisr_proto[proto].np_handler != NULL, 513 ("%s(%u): protocol not registered for %s", __func__, proto, 514 name)); 515 *qlimitp = netisr_proto[proto].np_qlimit; 516 NETISR_RUNLOCK(&tracker); 517 } 518 519 /* 520 * Update the queue limit across per-workstream queues for a protocol. We 521 * simply change the limits, and don't drain overflowed packets as they will 522 * (hopefully) take care of themselves shortly. 523 */ 524 int 525 netisr_setqlimit(const struct netisr_handler *nhp, u_int qlimit) 526 { 527 struct netisr_work *npwp; 528 #ifdef INVARIANTS 529 const char *name; 530 #endif 531 u_int i, proto; 532 533 if (qlimit > netisr_maxqlimit) 534 return (EINVAL); 535 536 proto = nhp->nh_proto; 537 #ifdef INVARIANTS 538 name = nhp->nh_name; 539 #endif 540 KASSERT(proto < NETISR_MAXPROT, 541 ("%s(%u): protocol too big for %s", __func__, proto, name)); 542 543 NETISR_WLOCK(); 544 KASSERT(netisr_proto[proto].np_handler != NULL, 545 ("%s(%u): protocol not registered for %s", __func__, proto, 546 name)); 547 548 netisr_proto[proto].np_qlimit = qlimit; 549 CPU_FOREACH(i) { 550 npwp = &(DPCPU_ID_PTR(i, nws))->nws_work[proto]; 551 npwp->nw_qlimit = qlimit; 552 } 553 NETISR_WUNLOCK(); 554 return (0); 555 } 556 557 /* 558 * Drain all packets currently held in a particular protocol work queue. 559 */ 560 static void 561 netisr_drain_proto(struct netisr_work *npwp) 562 { 563 struct mbuf *m; 564 565 /* 566 * We would assert the lock on the workstream but it's not passed in. 567 */ 568 while ((m = npwp->nw_head) != NULL) { 569 npwp->nw_head = m->m_nextpkt; 570 m->m_nextpkt = NULL; 571 if (npwp->nw_head == NULL) 572 npwp->nw_tail = NULL; 573 npwp->nw_len--; 574 m_freem(m); 575 } 576 KASSERT(npwp->nw_tail == NULL, ("%s: tail", __func__)); 577 KASSERT(npwp->nw_len == 0, ("%s: len", __func__)); 578 } 579 580 /* 581 * Remove the registration of a network protocol, which requires clearing 582 * per-protocol fields across all workstreams, including freeing all mbufs in 583 * the queues at time of unregister. All work in netisr is briefly suspended 584 * while this takes place. 585 */ 586 void 587 netisr_unregister(const struct netisr_handler *nhp) 588 { 589 struct netisr_work *npwp; 590 #ifdef INVARIANTS 591 const char *name; 592 #endif 593 u_int i, proto; 594 595 proto = nhp->nh_proto; 596 #ifdef INVARIANTS 597 name = nhp->nh_name; 598 #endif 599 KASSERT(proto < NETISR_MAXPROT, 600 ("%s(%u): protocol too big for %s", __func__, proto, name)); 601 602 NETISR_WLOCK(); 603 KASSERT(netisr_proto[proto].np_handler != NULL, 604 ("%s(%u): protocol not registered for %s", __func__, proto, 605 name)); 606 607 netisr_proto[proto].np_name = NULL; 608 netisr_proto[proto].np_handler = NULL; 609 netisr_proto[proto].np_m2flow = NULL; 610 netisr_proto[proto].np_m2cpuid = NULL; 611 netisr_proto[proto].np_qlimit = 0; 612 netisr_proto[proto].np_policy = 0; 613 CPU_FOREACH(i) { 614 npwp = &(DPCPU_ID_PTR(i, nws))->nws_work[proto]; 615 netisr_drain_proto(npwp); 616 bzero(npwp, sizeof(*npwp)); 617 } 618 NETISR_WUNLOCK(); 619 } 620 621 /* 622 * Compose the global and per-protocol policies on dispatch, and return the 623 * dispatch policy to use. 624 */ 625 static u_int 626 netisr_get_dispatch(struct netisr_proto *npp) 627 { 628 629 /* 630 * Protocol-specific configuration overrides the global default. 631 */ 632 if (npp->np_dispatch != NETISR_DISPATCH_DEFAULT) 633 return (npp->np_dispatch); 634 return (netisr_dispatch_policy); 635 } 636 637 /* 638 * Look up the workstream given a packet and source identifier. Do this by 639 * checking the protocol's policy, and optionally call out to the protocol 640 * for assistance if required. 641 */ 642 static struct mbuf * 643 netisr_select_cpuid(struct netisr_proto *npp, u_int dispatch_policy, 644 uintptr_t source, struct mbuf *m, u_int *cpuidp) 645 { 646 struct ifnet *ifp; 647 u_int policy; 648 649 NETISR_LOCK_ASSERT(); 650 651 /* 652 * In the event we have only one worker, shortcut and deliver to it 653 * without further ado. 654 */ 655 if (nws_count == 1) { 656 *cpuidp = nws_array[0]; 657 return (m); 658 } 659 660 /* 661 * What happens next depends on the policy selected by the protocol. 662 * If we want to support per-interface policies, we should do that 663 * here first. 664 */ 665 policy = npp->np_policy; 666 if (policy == NETISR_POLICY_CPU) { 667 m = npp->np_m2cpuid(m, source, cpuidp); 668 if (m == NULL) 669 return (NULL); 670 671 /* 672 * It's possible for a protocol not to have a good idea about 673 * where to process a packet, in which case we fall back on 674 * the netisr code to decide. In the hybrid case, return the 675 * current CPU ID, which will force an immediate direct 676 * dispatch. In the queued case, fall back on the SOURCE 677 * policy. 678 */ 679 if (*cpuidp != NETISR_CPUID_NONE) 680 return (m); 681 if (dispatch_policy == NETISR_DISPATCH_HYBRID) { 682 *cpuidp = curcpu; 683 return (m); 684 } 685 policy = NETISR_POLICY_SOURCE; 686 } 687 688 if (policy == NETISR_POLICY_FLOW) { 689 if (M_HASHTYPE_GET(m) == M_HASHTYPE_NONE && 690 npp->np_m2flow != NULL) { 691 m = npp->np_m2flow(m, source); 692 if (m == NULL) 693 return (NULL); 694 } 695 if (M_HASHTYPE_GET(m) != M_HASHTYPE_NONE) { 696 *cpuidp = 697 netisr_default_flow2cpu(m->m_pkthdr.flowid); 698 return (m); 699 } 700 policy = NETISR_POLICY_SOURCE; 701 } 702 703 KASSERT(policy == NETISR_POLICY_SOURCE, 704 ("%s: invalid policy %u for %s", __func__, npp->np_policy, 705 npp->np_name)); 706 707 ifp = m->m_pkthdr.rcvif; 708 if (ifp != NULL) 709 *cpuidp = nws_array[(ifp->if_index + source) % nws_count]; 710 else 711 *cpuidp = nws_array[source % nws_count]; 712 return (m); 713 } 714 715 /* 716 * Process packets associated with a workstream and protocol. For reasons of 717 * fairness, we process up to one complete netisr queue at a time, moving the 718 * queue to a stack-local queue for processing, but do not loop refreshing 719 * from the global queue. The caller is responsible for deciding whether to 720 * loop, and for setting the NWS_RUNNING flag. The passed workstream will be 721 * locked on entry and relocked before return, but will be released while 722 * processing. The number of packets processed is returned. 723 */ 724 static u_int 725 netisr_process_workstream_proto(struct netisr_workstream *nwsp, u_int proto) 726 { 727 struct netisr_work local_npw, *npwp; 728 u_int handled; 729 struct mbuf *m; 730 731 NETISR_LOCK_ASSERT(); 732 NWS_LOCK_ASSERT(nwsp); 733 734 KASSERT(nwsp->nws_flags & NWS_RUNNING, 735 ("%s(%u): not running", __func__, proto)); 736 KASSERT(proto >= 0 && proto < NETISR_MAXPROT, 737 ("%s(%u): invalid proto\n", __func__, proto)); 738 739 npwp = &nwsp->nws_work[proto]; 740 if (npwp->nw_len == 0) 741 return (0); 742 743 /* 744 * Move the global work queue to a thread-local work queue. 745 * 746 * Notice that this means the effective maximum length of the queue 747 * is actually twice that of the maximum queue length specified in 748 * the protocol registration call. 749 */ 750 handled = npwp->nw_len; 751 local_npw = *npwp; 752 npwp->nw_head = NULL; 753 npwp->nw_tail = NULL; 754 npwp->nw_len = 0; 755 nwsp->nws_pendingbits &= ~(1 << proto); 756 NWS_UNLOCK(nwsp); 757 while ((m = local_npw.nw_head) != NULL) { 758 local_npw.nw_head = m->m_nextpkt; 759 m->m_nextpkt = NULL; 760 if (local_npw.nw_head == NULL) 761 local_npw.nw_tail = NULL; 762 local_npw.nw_len--; 763 VNET_ASSERT(m->m_pkthdr.rcvif != NULL, 764 ("%s:%d rcvif == NULL: m=%p", __func__, __LINE__, m)); 765 CURVNET_SET(m->m_pkthdr.rcvif->if_vnet); 766 netisr_proto[proto].np_handler(m); 767 CURVNET_RESTORE(); 768 } 769 KASSERT(local_npw.nw_len == 0, 770 ("%s(%u): len %u", __func__, proto, local_npw.nw_len)); 771 if (netisr_proto[proto].np_drainedcpu) 772 netisr_proto[proto].np_drainedcpu(nwsp->nws_cpu); 773 NWS_LOCK(nwsp); 774 npwp->nw_handled += handled; 775 return (handled); 776 } 777 778 /* 779 * SWI handler for netisr -- processes packets in a set of workstreams that 780 * it owns, woken up by calls to NWS_SIGNAL(). If this workstream is already 781 * being direct dispatched, go back to sleep and wait for the dispatching 782 * thread to wake us up again. 783 */ 784 static void 785 swi_net(void *arg) 786 { 787 #ifdef NETISR_LOCKING 788 struct rm_priotracker tracker; 789 #endif 790 struct netisr_workstream *nwsp; 791 u_int bits, prot; 792 793 nwsp = arg; 794 795 #ifdef DEVICE_POLLING 796 KASSERT(nws_count == 1, 797 ("%s: device_polling but nws_count != 1", __func__)); 798 netisr_poll(); 799 #endif 800 #ifdef NETISR_LOCKING 801 NETISR_RLOCK(&tracker); 802 #endif 803 NWS_LOCK(nwsp); 804 KASSERT(!(nwsp->nws_flags & NWS_RUNNING), ("swi_net: running")); 805 if (nwsp->nws_flags & NWS_DISPATCHING) 806 goto out; 807 nwsp->nws_flags |= NWS_RUNNING; 808 nwsp->nws_flags &= ~NWS_SCHEDULED; 809 while ((bits = nwsp->nws_pendingbits) != 0) { 810 while ((prot = ffs(bits)) != 0) { 811 prot--; 812 bits &= ~(1 << prot); 813 (void)netisr_process_workstream_proto(nwsp, prot); 814 } 815 } 816 nwsp->nws_flags &= ~NWS_RUNNING; 817 out: 818 NWS_UNLOCK(nwsp); 819 #ifdef NETISR_LOCKING 820 NETISR_RUNLOCK(&tracker); 821 #endif 822 #ifdef DEVICE_POLLING 823 netisr_pollmore(); 824 #endif 825 } 826 827 static int 828 netisr_queue_workstream(struct netisr_workstream *nwsp, u_int proto, 829 struct netisr_work *npwp, struct mbuf *m, int *dosignalp) 830 { 831 832 NWS_LOCK_ASSERT(nwsp); 833 834 *dosignalp = 0; 835 if (npwp->nw_len < npwp->nw_qlimit) { 836 m->m_nextpkt = NULL; 837 if (npwp->nw_head == NULL) { 838 npwp->nw_head = m; 839 npwp->nw_tail = m; 840 } else { 841 npwp->nw_tail->m_nextpkt = m; 842 npwp->nw_tail = m; 843 } 844 npwp->nw_len++; 845 if (npwp->nw_len > npwp->nw_watermark) 846 npwp->nw_watermark = npwp->nw_len; 847 848 /* 849 * We must set the bit regardless of NWS_RUNNING, so that 850 * swi_net() keeps calling netisr_process_workstream_proto(). 851 */ 852 nwsp->nws_pendingbits |= (1 << proto); 853 if (!(nwsp->nws_flags & 854 (NWS_RUNNING | NWS_DISPATCHING | NWS_SCHEDULED))) { 855 nwsp->nws_flags |= NWS_SCHEDULED; 856 *dosignalp = 1; /* Defer until unlocked. */ 857 } 858 npwp->nw_queued++; 859 return (0); 860 } else { 861 m_freem(m); 862 npwp->nw_qdrops++; 863 return (ENOBUFS); 864 } 865 } 866 867 static int 868 netisr_queue_internal(u_int proto, struct mbuf *m, u_int cpuid) 869 { 870 struct netisr_workstream *nwsp; 871 struct netisr_work *npwp; 872 int dosignal, error; 873 874 #ifdef NETISR_LOCKING 875 NETISR_LOCK_ASSERT(); 876 #endif 877 KASSERT(cpuid <= mp_maxid, ("%s: cpuid too big (%u, %u)", __func__, 878 cpuid, mp_maxid)); 879 KASSERT(!CPU_ABSENT(cpuid), ("%s: CPU %u absent", __func__, cpuid)); 880 881 dosignal = 0; 882 error = 0; 883 nwsp = DPCPU_ID_PTR(cpuid, nws); 884 npwp = &nwsp->nws_work[proto]; 885 NWS_LOCK(nwsp); 886 error = netisr_queue_workstream(nwsp, proto, npwp, m, &dosignal); 887 NWS_UNLOCK(nwsp); 888 if (dosignal) 889 NWS_SIGNAL(nwsp); 890 return (error); 891 } 892 893 int 894 netisr_queue_src(u_int proto, uintptr_t source, struct mbuf *m) 895 { 896 #ifdef NETISR_LOCKING 897 struct rm_priotracker tracker; 898 #endif 899 u_int cpuid; 900 int error; 901 902 KASSERT(proto < NETISR_MAXPROT, 903 ("%s: invalid proto %u", __func__, proto)); 904 905 #ifdef NETISR_LOCKING 906 NETISR_RLOCK(&tracker); 907 #endif 908 KASSERT(netisr_proto[proto].np_handler != NULL, 909 ("%s: invalid proto %u", __func__, proto)); 910 911 m = netisr_select_cpuid(&netisr_proto[proto], NETISR_DISPATCH_DEFERRED, 912 source, m, &cpuid); 913 if (m != NULL) { 914 KASSERT(!CPU_ABSENT(cpuid), ("%s: CPU %u absent", __func__, 915 cpuid)); 916 error = netisr_queue_internal(proto, m, cpuid); 917 } else 918 error = ENOBUFS; 919 #ifdef NETISR_LOCKING 920 NETISR_RUNLOCK(&tracker); 921 #endif 922 return (error); 923 } 924 925 int 926 netisr_queue(u_int proto, struct mbuf *m) 927 { 928 929 return (netisr_queue_src(proto, 0, m)); 930 } 931 932 /* 933 * Dispatch a packet for netisr processing; direct dispatch is permitted by 934 * calling context. 935 */ 936 int 937 netisr_dispatch_src(u_int proto, uintptr_t source, struct mbuf *m) 938 { 939 #ifdef NETISR_LOCKING 940 struct rm_priotracker tracker; 941 #endif 942 struct netisr_workstream *nwsp; 943 struct netisr_proto *npp; 944 struct netisr_work *npwp; 945 int dosignal, error; 946 u_int cpuid, dispatch_policy; 947 948 KASSERT(proto < NETISR_MAXPROT, 949 ("%s: invalid proto %u", __func__, proto)); 950 #ifdef NETISR_LOCKING 951 NETISR_RLOCK(&tracker); 952 #endif 953 npp = &netisr_proto[proto]; 954 KASSERT(npp->np_handler != NULL, ("%s: invalid proto %u", __func__, 955 proto)); 956 957 dispatch_policy = netisr_get_dispatch(npp); 958 if (dispatch_policy == NETISR_DISPATCH_DEFERRED) 959 return (netisr_queue_src(proto, source, m)); 960 961 /* 962 * If direct dispatch is forced, then unconditionally dispatch 963 * without a formal CPU selection. Borrow the current CPU's stats, 964 * even if there's no worker on it. In this case we don't update 965 * nws_flags because all netisr processing will be source ordered due 966 * to always being forced to directly dispatch. 967 */ 968 if (dispatch_policy == NETISR_DISPATCH_DIRECT) { 969 nwsp = DPCPU_PTR(nws); 970 npwp = &nwsp->nws_work[proto]; 971 npwp->nw_dispatched++; 972 npwp->nw_handled++; 973 netisr_proto[proto].np_handler(m); 974 error = 0; 975 goto out_unlock; 976 } 977 978 KASSERT(dispatch_policy == NETISR_DISPATCH_HYBRID, 979 ("%s: unknown dispatch policy (%u)", __func__, dispatch_policy)); 980 981 /* 982 * Otherwise, we execute in a hybrid mode where we will try to direct 983 * dispatch if we're on the right CPU and the netisr worker isn't 984 * already running. 985 */ 986 sched_pin(); 987 m = netisr_select_cpuid(&netisr_proto[proto], NETISR_DISPATCH_HYBRID, 988 source, m, &cpuid); 989 if (m == NULL) { 990 error = ENOBUFS; 991 goto out_unpin; 992 } 993 KASSERT(!CPU_ABSENT(cpuid), ("%s: CPU %u absent", __func__, cpuid)); 994 if (cpuid != curcpu) 995 goto queue_fallback; 996 nwsp = DPCPU_PTR(nws); 997 npwp = &nwsp->nws_work[proto]; 998 999 /*- 1000 * We are willing to direct dispatch only if three conditions hold: 1001 * 1002 * (1) The netisr worker isn't already running, 1003 * (2) Another thread isn't already directly dispatching, and 1004 * (3) The netisr hasn't already been woken up. 1005 */ 1006 NWS_LOCK(nwsp); 1007 if (nwsp->nws_flags & (NWS_RUNNING | NWS_DISPATCHING | NWS_SCHEDULED)) { 1008 error = netisr_queue_workstream(nwsp, proto, npwp, m, 1009 &dosignal); 1010 NWS_UNLOCK(nwsp); 1011 if (dosignal) 1012 NWS_SIGNAL(nwsp); 1013 goto out_unpin; 1014 } 1015 1016 /* 1017 * The current thread is now effectively the netisr worker, so set 1018 * the dispatching flag to prevent concurrent processing of the 1019 * stream from another thread (even the netisr worker), which could 1020 * otherwise lead to effective misordering of the stream. 1021 */ 1022 nwsp->nws_flags |= NWS_DISPATCHING; 1023 NWS_UNLOCK(nwsp); 1024 netisr_proto[proto].np_handler(m); 1025 NWS_LOCK(nwsp); 1026 nwsp->nws_flags &= ~NWS_DISPATCHING; 1027 npwp->nw_handled++; 1028 npwp->nw_hybrid_dispatched++; 1029 1030 /* 1031 * If other work was enqueued by another thread while we were direct 1032 * dispatching, we need to signal the netisr worker to do that work. 1033 * In the future, we might want to do some of that work in the 1034 * current thread, rather than trigger further context switches. If 1035 * so, we'll want to establish a reasonable bound on the work done in 1036 * the "borrowed" context. 1037 */ 1038 if (nwsp->nws_pendingbits != 0) { 1039 nwsp->nws_flags |= NWS_SCHEDULED; 1040 dosignal = 1; 1041 } else 1042 dosignal = 0; 1043 NWS_UNLOCK(nwsp); 1044 if (dosignal) 1045 NWS_SIGNAL(nwsp); 1046 error = 0; 1047 goto out_unpin; 1048 1049 queue_fallback: 1050 error = netisr_queue_internal(proto, m, cpuid); 1051 out_unpin: 1052 sched_unpin(); 1053 out_unlock: 1054 #ifdef NETISR_LOCKING 1055 NETISR_RUNLOCK(&tracker); 1056 #endif 1057 return (error); 1058 } 1059 1060 int 1061 netisr_dispatch(u_int proto, struct mbuf *m) 1062 { 1063 1064 return (netisr_dispatch_src(proto, 0, m)); 1065 } 1066 1067 #ifdef DEVICE_POLLING 1068 /* 1069 * Kernel polling borrows a netisr thread to run interface polling in; this 1070 * function allows kernel polling to request that the netisr thread be 1071 * scheduled even if no packets are pending for protocols. 1072 */ 1073 void 1074 netisr_sched_poll(void) 1075 { 1076 struct netisr_workstream *nwsp; 1077 1078 nwsp = DPCPU_ID_PTR(nws_array[0], nws); 1079 NWS_SIGNAL(nwsp); 1080 } 1081 #endif 1082 1083 static void 1084 netisr_start_swi(u_int cpuid, struct pcpu *pc) 1085 { 1086 char swiname[12]; 1087 struct netisr_workstream *nwsp; 1088 int error; 1089 1090 KASSERT(!CPU_ABSENT(cpuid), ("%s: CPU %u absent", __func__, cpuid)); 1091 1092 nwsp = DPCPU_ID_PTR(cpuid, nws); 1093 mtx_init(&nwsp->nws_mtx, "netisr_mtx", NULL, MTX_DEF); 1094 nwsp->nws_cpu = cpuid; 1095 snprintf(swiname, sizeof(swiname), "netisr %u", cpuid); 1096 error = swi_add(&nwsp->nws_intr_event, swiname, swi_net, nwsp, 1097 SWI_NET, INTR_MPSAFE, &nwsp->nws_swi_cookie); 1098 if (error) 1099 panic("%s: swi_add %d", __func__, error); 1100 pc->pc_netisr = nwsp->nws_intr_event; 1101 if (netisr_bindthreads) { 1102 error = intr_event_bind(nwsp->nws_intr_event, cpuid); 1103 if (error != 0) 1104 printf("%s: cpu %u: intr_event_bind: %d", __func__, 1105 cpuid, error); 1106 } 1107 NETISR_WLOCK(); 1108 nws_array[nws_count] = nwsp->nws_cpu; 1109 nws_count++; 1110 NETISR_WUNLOCK(); 1111 } 1112 1113 /* 1114 * Initialize the netisr subsystem. We rely on BSS and static initialization 1115 * of most fields in global data structures. 1116 * 1117 * Start a worker thread for the boot CPU so that we can support network 1118 * traffic immediately in case the network stack is used before additional 1119 * CPUs are started (for example, diskless boot). 1120 */ 1121 static void 1122 netisr_init(void *arg) 1123 { 1124 KASSERT(curcpu == 0, ("%s: not on CPU 0", __func__)); 1125 1126 NETISR_LOCK_INIT(); 1127 if (netisr_maxthreads == 0 || netisr_maxthreads < -1 ) 1128 netisr_maxthreads = 1; /* default behavior */ 1129 else if (netisr_maxthreads == -1) 1130 netisr_maxthreads = mp_ncpus; /* use max cpus */ 1131 if (netisr_maxthreads > mp_ncpus) { 1132 printf("netisr_init: forcing maxthreads from %d to %d\n", 1133 netisr_maxthreads, mp_ncpus); 1134 netisr_maxthreads = mp_ncpus; 1135 } 1136 if (netisr_defaultqlimit > netisr_maxqlimit) { 1137 printf("netisr_init: forcing defaultqlimit from %d to %d\n", 1138 netisr_defaultqlimit, netisr_maxqlimit); 1139 netisr_defaultqlimit = netisr_maxqlimit; 1140 } 1141 #ifdef DEVICE_POLLING 1142 /* 1143 * The device polling code is not yet aware of how to deal with 1144 * multiple netisr threads, so for the time being compiling in device 1145 * polling disables parallel netisr workers. 1146 */ 1147 if (netisr_maxthreads != 1 || netisr_bindthreads != 0) { 1148 printf("netisr_init: forcing maxthreads to 1 and " 1149 "bindthreads to 0 for device polling\n"); 1150 netisr_maxthreads = 1; 1151 netisr_bindthreads = 0; 1152 } 1153 #endif 1154 netisr_start_swi(curcpu, pcpu_find(curcpu)); 1155 } 1156 SYSINIT(netisr_init, SI_SUB_SOFTINTR, SI_ORDER_FIRST, netisr_init, NULL); 1157 1158 /* 1159 * Start worker threads for additional CPUs. No attempt to gracefully handle 1160 * work reassignment, we don't yet support dynamic reconfiguration. 1161 */ 1162 static void 1163 netisr_start(void *arg) 1164 { 1165 struct pcpu *pc; 1166 1167 STAILQ_FOREACH(pc, &cpuhead, pc_allcpu) { 1168 if (nws_count >= netisr_maxthreads) 1169 break; 1170 /* Worker will already be present for boot CPU. */ 1171 if (pc->pc_netisr != NULL) 1172 continue; 1173 netisr_start_swi(pc->pc_cpuid, pc); 1174 } 1175 } 1176 SYSINIT(netisr_start, SI_SUB_SMP, SI_ORDER_MIDDLE, netisr_start, NULL); 1177 1178 /* 1179 * Sysctl monitoring for netisr: query a list of registered protocols. 1180 */ 1181 static int 1182 sysctl_netisr_proto(SYSCTL_HANDLER_ARGS) 1183 { 1184 struct rm_priotracker tracker; 1185 struct sysctl_netisr_proto *snpp, *snp_array; 1186 struct netisr_proto *npp; 1187 u_int counter, proto; 1188 int error; 1189 1190 if (req->newptr != NULL) 1191 return (EINVAL); 1192 snp_array = malloc(sizeof(*snp_array) * NETISR_MAXPROT, M_TEMP, 1193 M_ZERO | M_WAITOK); 1194 counter = 0; 1195 NETISR_RLOCK(&tracker); 1196 for (proto = 0; proto < NETISR_MAXPROT; proto++) { 1197 npp = &netisr_proto[proto]; 1198 if (npp->np_name == NULL) 1199 continue; 1200 snpp = &snp_array[counter]; 1201 snpp->snp_version = sizeof(*snpp); 1202 strlcpy(snpp->snp_name, npp->np_name, NETISR_NAMEMAXLEN); 1203 snpp->snp_proto = proto; 1204 snpp->snp_qlimit = npp->np_qlimit; 1205 snpp->snp_policy = npp->np_policy; 1206 snpp->snp_dispatch = npp->np_dispatch; 1207 if (npp->np_m2flow != NULL) 1208 snpp->snp_flags |= NETISR_SNP_FLAGS_M2FLOW; 1209 if (npp->np_m2cpuid != NULL) 1210 snpp->snp_flags |= NETISR_SNP_FLAGS_M2CPUID; 1211 if (npp->np_drainedcpu != NULL) 1212 snpp->snp_flags |= NETISR_SNP_FLAGS_DRAINEDCPU; 1213 counter++; 1214 } 1215 NETISR_RUNLOCK(&tracker); 1216 KASSERT(counter <= NETISR_MAXPROT, 1217 ("sysctl_netisr_proto: counter too big (%d)", counter)); 1218 error = SYSCTL_OUT(req, snp_array, sizeof(*snp_array) * counter); 1219 free(snp_array, M_TEMP); 1220 return (error); 1221 } 1222 1223 SYSCTL_PROC(_net_isr, OID_AUTO, proto, 1224 CTLFLAG_RD|CTLTYPE_STRUCT|CTLFLAG_MPSAFE, 0, 0, sysctl_netisr_proto, 1225 "S,sysctl_netisr_proto", 1226 "Return list of protocols registered with netisr"); 1227 1228 /* 1229 * Sysctl monitoring for netisr: query a list of workstreams. 1230 */ 1231 static int 1232 sysctl_netisr_workstream(SYSCTL_HANDLER_ARGS) 1233 { 1234 struct rm_priotracker tracker; 1235 struct sysctl_netisr_workstream *snwsp, *snws_array; 1236 struct netisr_workstream *nwsp; 1237 u_int counter, cpuid; 1238 int error; 1239 1240 if (req->newptr != NULL) 1241 return (EINVAL); 1242 snws_array = malloc(sizeof(*snws_array) * MAXCPU, M_TEMP, 1243 M_ZERO | M_WAITOK); 1244 counter = 0; 1245 NETISR_RLOCK(&tracker); 1246 CPU_FOREACH(cpuid) { 1247 nwsp = DPCPU_ID_PTR(cpuid, nws); 1248 if (nwsp->nws_intr_event == NULL) 1249 continue; 1250 NWS_LOCK(nwsp); 1251 snwsp = &snws_array[counter]; 1252 snwsp->snws_version = sizeof(*snwsp); 1253 1254 /* 1255 * For now, we equate workstream IDs and CPU IDs in the 1256 * kernel, but expose them independently to userspace in case 1257 * that assumption changes in the future. 1258 */ 1259 snwsp->snws_wsid = cpuid; 1260 snwsp->snws_cpu = cpuid; 1261 if (nwsp->nws_intr_event != NULL) 1262 snwsp->snws_flags |= NETISR_SNWS_FLAGS_INTR; 1263 NWS_UNLOCK(nwsp); 1264 counter++; 1265 } 1266 NETISR_RUNLOCK(&tracker); 1267 KASSERT(counter <= MAXCPU, 1268 ("sysctl_netisr_workstream: counter too big (%d)", counter)); 1269 error = SYSCTL_OUT(req, snws_array, sizeof(*snws_array) * counter); 1270 free(snws_array, M_TEMP); 1271 return (error); 1272 } 1273 1274 SYSCTL_PROC(_net_isr, OID_AUTO, workstream, 1275 CTLFLAG_RD|CTLTYPE_STRUCT|CTLFLAG_MPSAFE, 0, 0, sysctl_netisr_workstream, 1276 "S,sysctl_netisr_workstream", 1277 "Return list of workstreams implemented by netisr"); 1278 1279 /* 1280 * Sysctl monitoring for netisr: query per-protocol data across all 1281 * workstreams. 1282 */ 1283 static int 1284 sysctl_netisr_work(SYSCTL_HANDLER_ARGS) 1285 { 1286 struct rm_priotracker tracker; 1287 struct sysctl_netisr_work *snwp, *snw_array; 1288 struct netisr_workstream *nwsp; 1289 struct netisr_proto *npp; 1290 struct netisr_work *nwp; 1291 u_int counter, cpuid, proto; 1292 int error; 1293 1294 if (req->newptr != NULL) 1295 return (EINVAL); 1296 snw_array = malloc(sizeof(*snw_array) * MAXCPU * NETISR_MAXPROT, 1297 M_TEMP, M_ZERO | M_WAITOK); 1298 counter = 0; 1299 NETISR_RLOCK(&tracker); 1300 CPU_FOREACH(cpuid) { 1301 nwsp = DPCPU_ID_PTR(cpuid, nws); 1302 if (nwsp->nws_intr_event == NULL) 1303 continue; 1304 NWS_LOCK(nwsp); 1305 for (proto = 0; proto < NETISR_MAXPROT; proto++) { 1306 npp = &netisr_proto[proto]; 1307 if (npp->np_name == NULL) 1308 continue; 1309 nwp = &nwsp->nws_work[proto]; 1310 snwp = &snw_array[counter]; 1311 snwp->snw_version = sizeof(*snwp); 1312 snwp->snw_wsid = cpuid; /* See comment above. */ 1313 snwp->snw_proto = proto; 1314 snwp->snw_len = nwp->nw_len; 1315 snwp->snw_watermark = nwp->nw_watermark; 1316 snwp->snw_dispatched = nwp->nw_dispatched; 1317 snwp->snw_hybrid_dispatched = 1318 nwp->nw_hybrid_dispatched; 1319 snwp->snw_qdrops = nwp->nw_qdrops; 1320 snwp->snw_queued = nwp->nw_queued; 1321 snwp->snw_handled = nwp->nw_handled; 1322 counter++; 1323 } 1324 NWS_UNLOCK(nwsp); 1325 } 1326 KASSERT(counter <= MAXCPU * NETISR_MAXPROT, 1327 ("sysctl_netisr_work: counter too big (%d)", counter)); 1328 NETISR_RUNLOCK(&tracker); 1329 error = SYSCTL_OUT(req, snw_array, sizeof(*snw_array) * counter); 1330 free(snw_array, M_TEMP); 1331 return (error); 1332 } 1333 1334 SYSCTL_PROC(_net_isr, OID_AUTO, work, 1335 CTLFLAG_RD|CTLTYPE_STRUCT|CTLFLAG_MPSAFE, 0, 0, sysctl_netisr_work, 1336 "S,sysctl_netisr_work", 1337 "Return list of per-workstream, per-protocol work in netisr"); 1338 1339 #ifdef DDB 1340 DB_SHOW_COMMAND(netisr, db_show_netisr) 1341 { 1342 struct netisr_workstream *nwsp; 1343 struct netisr_work *nwp; 1344 int first, proto; 1345 u_int cpuid; 1346 1347 db_printf("%3s %6s %5s %5s %5s %8s %8s %8s %8s\n", "CPU", "Proto", 1348 "Len", "WMark", "Max", "Disp", "HDisp", "Drop", "Queue"); 1349 CPU_FOREACH(cpuid) { 1350 nwsp = DPCPU_ID_PTR(cpuid, nws); 1351 if (nwsp->nws_intr_event == NULL) 1352 continue; 1353 first = 1; 1354 for (proto = 0; proto < NETISR_MAXPROT; proto++) { 1355 if (netisr_proto[proto].np_handler == NULL) 1356 continue; 1357 nwp = &nwsp->nws_work[proto]; 1358 if (first) { 1359 db_printf("%3d ", cpuid); 1360 first = 0; 1361 } else 1362 db_printf("%3s ", ""); 1363 db_printf( 1364 "%6s %5d %5d %5d %8ju %8ju %8ju %8ju\n", 1365 netisr_proto[proto].np_name, nwp->nw_len, 1366 nwp->nw_watermark, nwp->nw_qlimit, 1367 nwp->nw_dispatched, nwp->nw_hybrid_dispatched, 1368 nwp->nw_qdrops, nwp->nw_queued); 1369 } 1370 } 1371 } 1372 #endif 1373