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