xref: /freebsd/sys/net/netisr.c (revision acc1a9ef8333c798c210fa94be6af4d5fe2dd794)
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 		/* XXXRW: Is skipping absent CPUs still required here? */
1171 		if (CPU_ABSENT(pc->pc_cpuid))
1172 			continue;
1173 		/* Worker will already be present for boot CPU. */
1174 		if (pc->pc_netisr != NULL)
1175 			continue;
1176 		netisr_start_swi(pc->pc_cpuid, pc);
1177 	}
1178 }
1179 SYSINIT(netisr_start, SI_SUB_SMP, SI_ORDER_MIDDLE, netisr_start, NULL);
1180 
1181 /*
1182  * Sysctl monitoring for netisr: query a list of registered protocols.
1183  */
1184 static int
1185 sysctl_netisr_proto(SYSCTL_HANDLER_ARGS)
1186 {
1187 	struct rm_priotracker tracker;
1188 	struct sysctl_netisr_proto *snpp, *snp_array;
1189 	struct netisr_proto *npp;
1190 	u_int counter, proto;
1191 	int error;
1192 
1193 	if (req->newptr != NULL)
1194 		return (EINVAL);
1195 	snp_array = malloc(sizeof(*snp_array) * NETISR_MAXPROT, M_TEMP,
1196 	    M_ZERO | M_WAITOK);
1197 	counter = 0;
1198 	NETISR_RLOCK(&tracker);
1199 	for (proto = 0; proto < NETISR_MAXPROT; proto++) {
1200 		npp = &netisr_proto[proto];
1201 		if (npp->np_name == NULL)
1202 			continue;
1203 		snpp = &snp_array[counter];
1204 		snpp->snp_version = sizeof(*snpp);
1205 		strlcpy(snpp->snp_name, npp->np_name, NETISR_NAMEMAXLEN);
1206 		snpp->snp_proto = proto;
1207 		snpp->snp_qlimit = npp->np_qlimit;
1208 		snpp->snp_policy = npp->np_policy;
1209 		snpp->snp_dispatch = npp->np_dispatch;
1210 		if (npp->np_m2flow != NULL)
1211 			snpp->snp_flags |= NETISR_SNP_FLAGS_M2FLOW;
1212 		if (npp->np_m2cpuid != NULL)
1213 			snpp->snp_flags |= NETISR_SNP_FLAGS_M2CPUID;
1214 		if (npp->np_drainedcpu != NULL)
1215 			snpp->snp_flags |= NETISR_SNP_FLAGS_DRAINEDCPU;
1216 		counter++;
1217 	}
1218 	NETISR_RUNLOCK(&tracker);
1219 	KASSERT(counter <= NETISR_MAXPROT,
1220 	    ("sysctl_netisr_proto: counter too big (%d)", counter));
1221 	error = SYSCTL_OUT(req, snp_array, sizeof(*snp_array) * counter);
1222 	free(snp_array, M_TEMP);
1223 	return (error);
1224 }
1225 
1226 SYSCTL_PROC(_net_isr, OID_AUTO, proto,
1227     CTLFLAG_RD|CTLTYPE_STRUCT|CTLFLAG_MPSAFE, 0, 0, sysctl_netisr_proto,
1228     "S,sysctl_netisr_proto",
1229     "Return list of protocols registered with netisr");
1230 
1231 /*
1232  * Sysctl monitoring for netisr: query a list of workstreams.
1233  */
1234 static int
1235 sysctl_netisr_workstream(SYSCTL_HANDLER_ARGS)
1236 {
1237 	struct rm_priotracker tracker;
1238 	struct sysctl_netisr_workstream *snwsp, *snws_array;
1239 	struct netisr_workstream *nwsp;
1240 	u_int counter, cpuid;
1241 	int error;
1242 
1243 	if (req->newptr != NULL)
1244 		return (EINVAL);
1245 	snws_array = malloc(sizeof(*snws_array) * MAXCPU, M_TEMP,
1246 	    M_ZERO | M_WAITOK);
1247 	counter = 0;
1248 	NETISR_RLOCK(&tracker);
1249 	CPU_FOREACH(cpuid) {
1250 		nwsp = DPCPU_ID_PTR(cpuid, nws);
1251 		if (nwsp->nws_intr_event == NULL)
1252 			continue;
1253 		NWS_LOCK(nwsp);
1254 		snwsp = &snws_array[counter];
1255 		snwsp->snws_version = sizeof(*snwsp);
1256 
1257 		/*
1258 		 * For now, we equate workstream IDs and CPU IDs in the
1259 		 * kernel, but expose them independently to userspace in case
1260 		 * that assumption changes in the future.
1261 		 */
1262 		snwsp->snws_wsid = cpuid;
1263 		snwsp->snws_cpu = cpuid;
1264 		if (nwsp->nws_intr_event != NULL)
1265 			snwsp->snws_flags |= NETISR_SNWS_FLAGS_INTR;
1266 		NWS_UNLOCK(nwsp);
1267 		counter++;
1268 	}
1269 	NETISR_RUNLOCK(&tracker);
1270 	KASSERT(counter <= MAXCPU,
1271 	    ("sysctl_netisr_workstream: counter too big (%d)", counter));
1272 	error = SYSCTL_OUT(req, snws_array, sizeof(*snws_array) * counter);
1273 	free(snws_array, M_TEMP);
1274 	return (error);
1275 }
1276 
1277 SYSCTL_PROC(_net_isr, OID_AUTO, workstream,
1278     CTLFLAG_RD|CTLTYPE_STRUCT|CTLFLAG_MPSAFE, 0, 0, sysctl_netisr_workstream,
1279     "S,sysctl_netisr_workstream",
1280     "Return list of workstreams implemented by netisr");
1281 
1282 /*
1283  * Sysctl monitoring for netisr: query per-protocol data across all
1284  * workstreams.
1285  */
1286 static int
1287 sysctl_netisr_work(SYSCTL_HANDLER_ARGS)
1288 {
1289 	struct rm_priotracker tracker;
1290 	struct sysctl_netisr_work *snwp, *snw_array;
1291 	struct netisr_workstream *nwsp;
1292 	struct netisr_proto *npp;
1293 	struct netisr_work *nwp;
1294 	u_int counter, cpuid, proto;
1295 	int error;
1296 
1297 	if (req->newptr != NULL)
1298 		return (EINVAL);
1299 	snw_array = malloc(sizeof(*snw_array) * MAXCPU * NETISR_MAXPROT,
1300 	    M_TEMP, M_ZERO | M_WAITOK);
1301 	counter = 0;
1302 	NETISR_RLOCK(&tracker);
1303 	CPU_FOREACH(cpuid) {
1304 		nwsp = DPCPU_ID_PTR(cpuid, nws);
1305 		if (nwsp->nws_intr_event == NULL)
1306 			continue;
1307 		NWS_LOCK(nwsp);
1308 		for (proto = 0; proto < NETISR_MAXPROT; proto++) {
1309 			npp = &netisr_proto[proto];
1310 			if (npp->np_name == NULL)
1311 				continue;
1312 			nwp = &nwsp->nws_work[proto];
1313 			snwp = &snw_array[counter];
1314 			snwp->snw_version = sizeof(*snwp);
1315 			snwp->snw_wsid = cpuid;		/* See comment above. */
1316 			snwp->snw_proto = proto;
1317 			snwp->snw_len = nwp->nw_len;
1318 			snwp->snw_watermark = nwp->nw_watermark;
1319 			snwp->snw_dispatched = nwp->nw_dispatched;
1320 			snwp->snw_hybrid_dispatched =
1321 			    nwp->nw_hybrid_dispatched;
1322 			snwp->snw_qdrops = nwp->nw_qdrops;
1323 			snwp->snw_queued = nwp->nw_queued;
1324 			snwp->snw_handled = nwp->nw_handled;
1325 			counter++;
1326 		}
1327 		NWS_UNLOCK(nwsp);
1328 	}
1329 	KASSERT(counter <= MAXCPU * NETISR_MAXPROT,
1330 	    ("sysctl_netisr_work: counter too big (%d)", counter));
1331 	NETISR_RUNLOCK(&tracker);
1332 	error = SYSCTL_OUT(req, snw_array, sizeof(*snw_array) * counter);
1333 	free(snw_array, M_TEMP);
1334 	return (error);
1335 }
1336 
1337 SYSCTL_PROC(_net_isr, OID_AUTO, work,
1338     CTLFLAG_RD|CTLTYPE_STRUCT|CTLFLAG_MPSAFE, 0, 0, sysctl_netisr_work,
1339     "S,sysctl_netisr_work",
1340     "Return list of per-workstream, per-protocol work in netisr");
1341 
1342 #ifdef DDB
1343 DB_SHOW_COMMAND(netisr, db_show_netisr)
1344 {
1345 	struct netisr_workstream *nwsp;
1346 	struct netisr_work *nwp;
1347 	int first, proto;
1348 	u_int cpuid;
1349 
1350 	db_printf("%3s %6s %5s %5s %5s %8s %8s %8s %8s\n", "CPU", "Proto",
1351 	    "Len", "WMark", "Max", "Disp", "HDisp", "Drop", "Queue");
1352 	CPU_FOREACH(cpuid) {
1353 		nwsp = DPCPU_ID_PTR(cpuid, nws);
1354 		if (nwsp->nws_intr_event == NULL)
1355 			continue;
1356 		first = 1;
1357 		for (proto = 0; proto < NETISR_MAXPROT; proto++) {
1358 			if (netisr_proto[proto].np_handler == NULL)
1359 				continue;
1360 			nwp = &nwsp->nws_work[proto];
1361 			if (first) {
1362 				db_printf("%3d ", cpuid);
1363 				first = 0;
1364 			} else
1365 				db_printf("%3s ", "");
1366 			db_printf(
1367 			    "%6s %5d %5d %5d %8ju %8ju %8ju %8ju\n",
1368 			    netisr_proto[proto].np_name, nwp->nw_len,
1369 			    nwp->nw_watermark, nwp->nw_qlimit,
1370 			    nwp->nw_dispatched, nwp->nw_hybrid_dispatched,
1371 			    nwp->nw_qdrops, nwp->nw_queued);
1372 		}
1373 	}
1374 }
1375 #endif
1376