xref: /freebsd/sys/net/netisr.c (revision fdf95c0b8147bc7d44c8c045b45b44080df15760)
11cafed39SJonathan Lemon /*-
2d4b5cae4SRobert Watson  * Copyright (c) 2007-2009 Robert N. M. Watson
3f2d2d694SRobert Watson  * Copyright (c) 2010-2011 Juniper Networks, Inc.
4e3b6e33cSJake Burkholder  * All rights reserved.
5e3b6e33cSJake Burkholder  *
62d22f334SRobert Watson  * This software was developed by Robert N. M. Watson under contract
72d22f334SRobert Watson  * to Juniper Networks, Inc.
82d22f334SRobert Watson  *
9e3b6e33cSJake Burkholder  * Redistribution and use in source and binary forms, with or without
10e3b6e33cSJake Burkholder  * modification, are permitted provided that the following conditions
11e3b6e33cSJake Burkholder  * are met:
12e3b6e33cSJake Burkholder  * 1. Redistributions of source code must retain the above copyright
131cafed39SJonathan Lemon  *    notice, this list of conditions and the following disclaimer.
14e3b6e33cSJake Burkholder  * 2. Redistributions in binary form must reproduce the above copyright
15e3b6e33cSJake Burkholder  *    notice, this list of conditions and the following disclaimer in the
16e3b6e33cSJake Burkholder  *    documentation and/or other materials provided with the distribution.
17e3b6e33cSJake Burkholder  *
181cafed39SJonathan Lemon  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
191cafed39SJonathan Lemon  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
201cafed39SJonathan Lemon  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
211cafed39SJonathan Lemon  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
221cafed39SJonathan Lemon  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
231cafed39SJonathan Lemon  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
241cafed39SJonathan Lemon  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
251cafed39SJonathan Lemon  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
261cafed39SJonathan Lemon  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
271cafed39SJonathan Lemon  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
281cafed39SJonathan Lemon  * SUCH DAMAGE.
29e3b6e33cSJake Burkholder  */
30e3b6e33cSJake Burkholder 
31d4b5cae4SRobert Watson #include <sys/cdefs.h>
32d4b5cae4SRobert Watson __FBSDID("$FreeBSD$");
33d4b5cae4SRobert Watson 
34d4b5cae4SRobert Watson /*
35d4b5cae4SRobert Watson  * netisr is a packet dispatch service, allowing synchronous (directly
36d4b5cae4SRobert Watson  * dispatched) and asynchronous (deferred dispatch) processing of packets by
37d4b5cae4SRobert Watson  * registered protocol handlers.  Callers pass a protocol identifier and
38d4b5cae4SRobert Watson  * packet to netisr, along with a direct dispatch hint, and work will either
390a32e29fSRobert Watson  * be immediately processed by the registered handler, or passed to a
400a32e29fSRobert Watson  * software interrupt (SWI) thread for deferred dispatch.  Callers will
410a32e29fSRobert Watson  * generally select one or the other based on:
42d4b5cae4SRobert Watson  *
430a32e29fSRobert Watson  * - Whether directly dispatching a netisr handler lead to code reentrance or
44d4b5cae4SRobert Watson  *   lock recursion, such as entering the socket code from the socket code.
450a32e29fSRobert Watson  * - Whether directly dispatching a netisr handler lead to recursive
46d4b5cae4SRobert Watson  *   processing, such as when decapsulating several wrapped layers of tunnel
47d4b5cae4SRobert Watson  *   information (IPSEC within IPSEC within ...).
48d4b5cae4SRobert Watson  *
49d4b5cae4SRobert Watson  * Maintaining ordering for protocol streams is a critical design concern.
50d4b5cae4SRobert Watson  * Enforcing ordering limits the opportunity for concurrency, but maintains
51d4b5cae4SRobert Watson  * the strong ordering requirements found in some protocols, such as TCP.  Of
52d4b5cae4SRobert Watson  * related concern is CPU affinity--it is desirable to process all data
53d4b5cae4SRobert Watson  * associated with a particular stream on the same CPU over time in order to
54d4b5cae4SRobert Watson  * avoid acquiring locks associated with the connection on different CPUs,
55d4b5cae4SRobert Watson  * keep connection data in one cache, and to generally encourage associated
56d4b5cae4SRobert Watson  * user threads to live on the same CPU as the stream.  It's also desirable
57d4b5cae4SRobert Watson  * to avoid lock migration and contention where locks are associated with
58d4b5cae4SRobert Watson  * more than one flow.
59d4b5cae4SRobert Watson  *
60d4b5cae4SRobert Watson  * netisr supports several policy variations, represented by the
610a32e29fSRobert Watson  * NETISR_POLICY_* constants, allowing protocols to play various roles in
62d4b5cae4SRobert Watson  * identifying flows, assigning work to CPUs, etc.  These are described in
630a32e29fSRobert Watson  * netisr.h.
64d4b5cae4SRobert Watson  */
65d4b5cae4SRobert Watson 
66d4b5cae4SRobert Watson #include "opt_ddb.h"
67f0796cd2SGleb Smirnoff #include "opt_device_polling.h"
681d8cd39eSRobert Watson 
69e3b6e33cSJake Burkholder #include <sys/param.h>
70e3b6e33cSJake Burkholder #include <sys/bus.h>
71e3b6e33cSJake Burkholder #include <sys/kernel.h>
721cafed39SJonathan Lemon #include <sys/kthread.h>
738ec07310SGleb Smirnoff #include <sys/malloc.h>
74d4b5cae4SRobert Watson #include <sys/interrupt.h>
751cafed39SJonathan Lemon #include <sys/lock.h>
761cafed39SJonathan Lemon #include <sys/mbuf.h>
77d4b5cae4SRobert Watson #include <sys/mutex.h>
7853402767SRobert Watson #include <sys/pcpu.h>
79d4b5cae4SRobert Watson #include <sys/proc.h>
80d4b5cae4SRobert Watson #include <sys/rmlock.h>
81d4b5cae4SRobert Watson #include <sys/sched.h>
82d4b5cae4SRobert Watson #include <sys/smp.h>
831cafed39SJonathan Lemon #include <sys/socket.h>
84d4b5cae4SRobert Watson #include <sys/sysctl.h>
85d4b5cae4SRobert Watson #include <sys/systm.h>
86d4b5cae4SRobert Watson 
87d4b5cae4SRobert Watson #ifdef DDB
88d4b5cae4SRobert Watson #include <ddb/ddb.h>
89d4b5cae4SRobert Watson #endif
901cafed39SJonathan Lemon 
91938448cdSRobert Watson #define	_WANT_NETISR_INTERNAL	/* Enable definitions from netisr_internal.h */
921cafed39SJonathan Lemon #include <net/if.h>
931cafed39SJonathan Lemon #include <net/if_var.h>
94e3b6e33cSJake Burkholder #include <net/netisr.h>
95938448cdSRobert Watson #include <net/netisr_internal.h>
96530c0060SRobert Watson #include <net/vnet.h>
97e3b6e33cSJake Burkholder 
98d4b5cae4SRobert Watson /*-
99d4b5cae4SRobert Watson  * Synchronize use and modification of the registered netisr data structures;
100d4b5cae4SRobert Watson  * acquire a read lock while modifying the set of registered protocols to
101d4b5cae4SRobert Watson  * prevent partially registered or unregistered protocols from being run.
102d4b5cae4SRobert Watson  *
103d4b5cae4SRobert Watson  * The following data structures and fields are protected by this lock:
104d4b5cae4SRobert Watson  *
105938448cdSRobert Watson  * - The netisr_proto array, including all fields of struct netisr_proto.
106d4b5cae4SRobert Watson  * - The nws array, including all fields of struct netisr_worker.
107d4b5cae4SRobert Watson  * - The nws_array array.
108d4b5cae4SRobert Watson  *
109d4b5cae4SRobert Watson  * Note: the NETISR_LOCKING define controls whether read locks are acquired
110d4b5cae4SRobert Watson  * in packet processing paths requiring netisr registration stability.  This
1110a32e29fSRobert Watson  * is disabled by default as it can lead to measurable performance
112d4b5cae4SRobert Watson  * degradation even with rmlocks (3%-6% for loopback ping-pong traffic), and
113d4b5cae4SRobert Watson  * because netisr registration and unregistration is extremely rare at
114d4b5cae4SRobert Watson  * runtime.  If it becomes more common, this decision should be revisited.
115d4b5cae4SRobert Watson  *
116d4b5cae4SRobert Watson  * XXXRW: rmlocks don't support assertions.
117d4b5cae4SRobert Watson  */
118d4b5cae4SRobert Watson static struct rmlock	netisr_rmlock;
119d4b5cae4SRobert Watson #define	NETISR_LOCK_INIT()	rm_init_flags(&netisr_rmlock, "netisr", \
120d4b5cae4SRobert Watson 				    RM_NOWITNESS)
121d4b5cae4SRobert Watson #define	NETISR_LOCK_ASSERT()
122d4b5cae4SRobert Watson #define	NETISR_RLOCK(tracker)	rm_rlock(&netisr_rmlock, (tracker))
123d4b5cae4SRobert Watson #define	NETISR_RUNLOCK(tracker)	rm_runlock(&netisr_rmlock, (tracker))
124d4b5cae4SRobert Watson #define	NETISR_WLOCK()		rm_wlock(&netisr_rmlock)
125d4b5cae4SRobert Watson #define	NETISR_WUNLOCK()	rm_wunlock(&netisr_rmlock)
126d4b5cae4SRobert Watson /* #define	NETISR_LOCKING */
127e3b6e33cSJake Burkholder 
1286472ac3dSEd Schouten static SYSCTL_NODE(_net, OID_AUTO, isr, CTLFLAG_RW, 0, "netisr");
1291cafed39SJonathan Lemon 
130d4b5cae4SRobert Watson /*-
131f2d2d694SRobert Watson  * Three global direct dispatch policies are supported:
132d4b5cae4SRobert Watson  *
133da162ca8SSergey Kandaurov  * NETISR_DISPATCH_DEFERRED: All work is deferred for a netisr, regardless of
134f2d2d694SRobert Watson  * context (may be overriden by protocols).
135d4b5cae4SRobert Watson  *
136f2d2d694SRobert Watson  * NETISR_DISPATCH_HYBRID: If the executing context allows direct dispatch,
137f2d2d694SRobert Watson  * and we're running on the CPU the work would be performed on, then direct
138f2d2d694SRobert Watson  * dispatch it if it wouldn't violate ordering constraints on the workstream.
139d4b5cae4SRobert Watson  *
140f2d2d694SRobert Watson  * NETISR_DISPATCH_DIRECT: If the executing context allows direct dispatch,
141f2d2d694SRobert Watson  * always direct dispatch.  (The default.)
142d4b5cae4SRobert Watson  *
143d4b5cae4SRobert Watson  * Notice that changing the global policy could lead to short periods of
144d4b5cae4SRobert Watson  * misordered processing, but this is considered acceptable as compared to
145f2d2d694SRobert Watson  * the complexity of enforcing ordering during policy changes.  Protocols can
146f2d2d694SRobert Watson  * override the global policy (when they're not doing that, they select
147f2d2d694SRobert Watson  * NETISR_DISPATCH_DEFAULT).
148d4b5cae4SRobert Watson  */
149f2d2d694SRobert Watson #define	NETISR_DISPATCH_POLICY_DEFAULT	NETISR_DISPATCH_DIRECT
150f2d2d694SRobert Watson #define	NETISR_DISPATCH_POLICY_MAXSTR	20 /* Used for temporary buffers. */
151f2d2d694SRobert Watson static u_int	netisr_dispatch_policy = NETISR_DISPATCH_POLICY_DEFAULT;
152f2d2d694SRobert Watson static int	sysctl_netisr_dispatch_policy(SYSCTL_HANDLER_ARGS);
153af3b2549SHans Petter Selasky SYSCTL_PROC(_net_isr, OID_AUTO, dispatch, CTLTYPE_STRING | CTLFLAG_RWTUN,
154af3b2549SHans Petter Selasky     0, 0, sysctl_netisr_dispatch_policy, "A",
155f2d2d694SRobert Watson     "netisr dispatch policy");
156e3b6e33cSJake Burkholder 
157f2d2d694SRobert Watson /*
158d4b5cae4SRobert Watson  * Allow the administrator to limit the number of threads (CPUs) to use for
159d4b5cae4SRobert Watson  * netisr.  We don't check netisr_maxthreads before creating the thread for
160a9467c3cSHiren Panchasara  * CPU 0. This must be set at boot. We will create at most one thread per CPU.
161a9467c3cSHiren Panchasara  * By default we initialize this to 1 which would assign just 1 cpu (cpu0) and
162a9467c3cSHiren Panchasara  * therefore only 1 workstream. If set to -1, netisr would use all cpus
163a9467c3cSHiren Panchasara  * (mp_ncpus) and therefore would have those many workstreams. One workstream
164a9467c3cSHiren Panchasara  * per thread (CPU).
1655fd04e38SRobert Watson  */
166a9467c3cSHiren Panchasara static int	netisr_maxthreads = 1;		/* Max number of threads. */
16778494902SPawel Jakub Dawidek SYSCTL_INT(_net_isr, OID_AUTO, maxthreads, CTLFLAG_RDTUN,
168d4b5cae4SRobert Watson     &netisr_maxthreads, 0,
169d4b5cae4SRobert Watson     "Use at most this many CPUs for netisr processing");
1705fd04e38SRobert Watson 
171d4b5cae4SRobert Watson static int	netisr_bindthreads = 0;		/* Bind threads to CPUs. */
17278494902SPawel Jakub Dawidek SYSCTL_INT(_net_isr, OID_AUTO, bindthreads, CTLFLAG_RDTUN,
173d4b5cae4SRobert Watson     &netisr_bindthreads, 0, "Bind netisr threads to CPUs.");
174d4b5cae4SRobert Watson 
175d4b5cae4SRobert Watson /*
1760a32e29fSRobert Watson  * Limit per-workstream mbuf queue limits s to at most net.isr.maxqlimit,
1770a32e29fSRobert Watson  * both for initial configuration and later modification using
1780a32e29fSRobert Watson  * netisr_setqlimit().
179d4b5cae4SRobert Watson  */
180d4b5cae4SRobert Watson #define	NETISR_DEFAULT_MAXQLIMIT	10240
181d4b5cae4SRobert Watson static u_int	netisr_maxqlimit = NETISR_DEFAULT_MAXQLIMIT;
182f88910cdSMatthew D Fleming SYSCTL_UINT(_net_isr, OID_AUTO, maxqlimit, CTLFLAG_RDTUN,
183d4b5cae4SRobert Watson     &netisr_maxqlimit, 0,
184d4b5cae4SRobert Watson     "Maximum netisr per-protocol, per-CPU queue depth.");
185d4b5cae4SRobert Watson 
186d4b5cae4SRobert Watson /*
1870a32e29fSRobert Watson  * The default per-workstream mbuf queue limit for protocols that don't
1880a32e29fSRobert Watson  * initialize the nh_qlimit field of their struct netisr_handler.  If this is
1890a32e29fSRobert Watson  * set above netisr_maxqlimit, we truncate it to the maximum during boot.
190d4b5cae4SRobert Watson  */
191d4b5cae4SRobert Watson #define	NETISR_DEFAULT_DEFAULTQLIMIT	256
192d4b5cae4SRobert Watson static u_int	netisr_defaultqlimit = NETISR_DEFAULT_DEFAULTQLIMIT;
193f88910cdSMatthew D Fleming SYSCTL_UINT(_net_isr, OID_AUTO, defaultqlimit, CTLFLAG_RDTUN,
194d4b5cae4SRobert Watson     &netisr_defaultqlimit, 0,
195d4b5cae4SRobert Watson     "Default netisr per-protocol, per-CPU queue limit if not set by protocol");
196d4b5cae4SRobert Watson 
197d4b5cae4SRobert Watson /*
198938448cdSRobert Watson  * Store and export the compile-time constant NETISR_MAXPROT limit on the
199938448cdSRobert Watson  * number of protocols that can register with netisr at a time.  This is
200938448cdSRobert Watson  * required for crashdump analysis, as it sizes netisr_proto[].
201d4b5cae4SRobert Watson  */
202938448cdSRobert Watson static u_int	netisr_maxprot = NETISR_MAXPROT;
203f88910cdSMatthew D Fleming SYSCTL_UINT(_net_isr, OID_AUTO, maxprot, CTLFLAG_RD,
204938448cdSRobert Watson     &netisr_maxprot, 0,
205938448cdSRobert Watson     "Compile-time limit on the number of protocols supported by netisr.");
206d4b5cae4SRobert Watson 
207d4b5cae4SRobert Watson /*
208938448cdSRobert Watson  * The netisr_proto array describes all registered protocols, indexed by
209938448cdSRobert Watson  * protocol number.  See netisr_internal.h for more details.
210d4b5cae4SRobert Watson  */
211938448cdSRobert Watson static struct netisr_proto	netisr_proto[NETISR_MAXPROT];
212d4b5cae4SRobert Watson 
213484149deSBjoern A. Zeeb #ifdef VIMAGE
214484149deSBjoern A. Zeeb /*
215484149deSBjoern A. Zeeb  * The netisr_enable array describes a per-VNET flag for registered
216484149deSBjoern A. Zeeb  * protocols on whether this netisr is active in this VNET or not.
217484149deSBjoern A. Zeeb  * netisr_register() will automatically enable the netisr for the
218484149deSBjoern A. Zeeb  * default VNET and all currently active instances.
219484149deSBjoern A. Zeeb  * netisr_unregister() will disable all active VNETs, including vnet0.
220484149deSBjoern A. Zeeb  * Individual network stack instances can be enabled/disabled by the
221484149deSBjoern A. Zeeb  * netisr_(un)register _vnet() functions.
222484149deSBjoern A. Zeeb  * With this we keep the one netisr_proto per protocol but add a
223484149deSBjoern A. Zeeb  * mechanism to stop netisr processing for vnet teardown.
224484149deSBjoern A. Zeeb  * Apart from that we expect a VNET to always be enabled.
225484149deSBjoern A. Zeeb  */
226484149deSBjoern A. Zeeb static VNET_DEFINE(u_int,	netisr_enable[NETISR_MAXPROT]);
227484149deSBjoern A. Zeeb #define	V_netisr_enable		VNET(netisr_enable)
228484149deSBjoern A. Zeeb #endif
229484149deSBjoern A. Zeeb 
230d4b5cae4SRobert Watson /*
231938448cdSRobert Watson  * Per-CPU workstream data.  See netisr_internal.h for more details.
232d4b5cae4SRobert Watson  */
23353402767SRobert Watson DPCPU_DEFINE(struct netisr_workstream, nws);
234d4b5cae4SRobert Watson 
235d4b5cae4SRobert Watson /*
236d4b5cae4SRobert Watson  * Map contiguous values between 0 and nws_count into CPU IDs appropriate for
23753402767SRobert Watson  * accessing workstreams.  This allows constructions of the form
23853402767SRobert Watson  * DPCPU_ID_GET(nws_array[arbitraryvalue % nws_count], nws).
239d4b5cae4SRobert Watson  */
240d4b5cae4SRobert Watson static u_int				 nws_array[MAXCPU];
241d4b5cae4SRobert Watson 
242d4b5cae4SRobert Watson /*
243d4b5cae4SRobert Watson  * Number of registered workstreams.  Will be at most the number of running
244d4b5cae4SRobert Watson  * CPUs once fully started.
245d4b5cae4SRobert Watson  */
246d4b5cae4SRobert Watson static u_int				 nws_count;
247f88910cdSMatthew D Fleming SYSCTL_UINT(_net_isr, OID_AUTO, numthreads, CTLFLAG_RD,
248d4b5cae4SRobert Watson     &nws_count, 0, "Number of extant netisr threads.");
249d4b5cae4SRobert Watson 
250d4b5cae4SRobert Watson /*
251d4b5cae4SRobert Watson  * Synchronization for each workstream: a mutex protects all mutable fields
252d4b5cae4SRobert Watson  * in each stream, including per-protocol state (mbuf queues).  The SWI is
253d4b5cae4SRobert Watson  * woken up if asynchronous dispatch is required.
254d4b5cae4SRobert Watson  */
255d4b5cae4SRobert Watson #define	NWS_LOCK(s)		mtx_lock(&(s)->nws_mtx)
256d4b5cae4SRobert Watson #define	NWS_LOCK_ASSERT(s)	mtx_assert(&(s)->nws_mtx, MA_OWNED)
257d4b5cae4SRobert Watson #define	NWS_UNLOCK(s)		mtx_unlock(&(s)->nws_mtx)
258d4b5cae4SRobert Watson #define	NWS_SIGNAL(s)		swi_sched((s)->nws_swi_cookie, 0)
259d4b5cae4SRobert Watson 
260d4b5cae4SRobert Watson /*
261d4b5cae4SRobert Watson  * Utility routines for protocols that implement their own mapping of flows
262d4b5cae4SRobert Watson  * to CPUs.
263d4b5cae4SRobert Watson  */
264d4b5cae4SRobert Watson u_int
265d4b5cae4SRobert Watson netisr_get_cpucount(void)
266d4b5cae4SRobert Watson {
267d4b5cae4SRobert Watson 
268d4b5cae4SRobert Watson 	return (nws_count);
2695fd04e38SRobert Watson }
270d4b5cae4SRobert Watson 
271d4b5cae4SRobert Watson u_int
272d4b5cae4SRobert Watson netisr_get_cpuid(u_int cpunumber)
273d4b5cae4SRobert Watson {
274d4b5cae4SRobert Watson 
275*fdf95c0bSAndrey V. Elsukov 	return (nws_array[cpunumber % nws_count]);
2765fd04e38SRobert Watson }
2775fd04e38SRobert Watson 
2785fd04e38SRobert Watson /*
2790a32e29fSRobert Watson  * The default implementation of flow -> CPU ID mapping.
280d4b5cae4SRobert Watson  *
281d4b5cae4SRobert Watson  * Non-static so that protocols can use it to map their own work to specific
282d4b5cae4SRobert Watson  * CPUs in a manner consistent to netisr for affinity purposes.
283d4b5cae4SRobert Watson  */
284d4b5cae4SRobert Watson u_int
285d4b5cae4SRobert Watson netisr_default_flow2cpu(u_int flowid)
286d4b5cae4SRobert Watson {
287d4b5cae4SRobert Watson 
288d4b5cae4SRobert Watson 	return (nws_array[flowid % nws_count]);
289d4b5cae4SRobert Watson }
290d4b5cae4SRobert Watson 
291d4b5cae4SRobert Watson /*
292f2d2d694SRobert Watson  * Dispatch tunable and sysctl configuration.
293f2d2d694SRobert Watson  */
294f2d2d694SRobert Watson struct netisr_dispatch_table_entry {
295f2d2d694SRobert Watson 	u_int		 ndte_policy;
296f2d2d694SRobert Watson 	const char	*ndte_policy_str;
297f2d2d694SRobert Watson };
298f2d2d694SRobert Watson static const struct netisr_dispatch_table_entry netisr_dispatch_table[] = {
299f2d2d694SRobert Watson 	{ NETISR_DISPATCH_DEFAULT, "default" },
300f2d2d694SRobert Watson 	{ NETISR_DISPATCH_DEFERRED, "deferred" },
301f2d2d694SRobert Watson 	{ NETISR_DISPATCH_HYBRID, "hybrid" },
302f2d2d694SRobert Watson 	{ NETISR_DISPATCH_DIRECT, "direct" },
303f2d2d694SRobert Watson };
304f2d2d694SRobert Watson 
305f2d2d694SRobert Watson static void
306f2d2d694SRobert Watson netisr_dispatch_policy_to_str(u_int dispatch_policy, char *buffer,
307f2d2d694SRobert Watson     u_int buflen)
308f2d2d694SRobert Watson {
309f2d2d694SRobert Watson 	const struct netisr_dispatch_table_entry *ndtep;
310f2d2d694SRobert Watson 	const char *str;
311f2d2d694SRobert Watson 	u_int i;
312f2d2d694SRobert Watson 
313f2d2d694SRobert Watson 	str = "unknown";
3148dfea464SPedro F. Giffuni 	for (i = 0; i < nitems(netisr_dispatch_table); i++) {
315f2d2d694SRobert Watson 		ndtep = &netisr_dispatch_table[i];
316f2d2d694SRobert Watson 		if (ndtep->ndte_policy == dispatch_policy) {
317f2d2d694SRobert Watson 			str = ndtep->ndte_policy_str;
318f2d2d694SRobert Watson 			break;
319f2d2d694SRobert Watson 		}
320f2d2d694SRobert Watson 	}
321f2d2d694SRobert Watson 	snprintf(buffer, buflen, "%s", str);
322f2d2d694SRobert Watson }
323f2d2d694SRobert Watson 
324f2d2d694SRobert Watson static int
325f2d2d694SRobert Watson netisr_dispatch_policy_from_str(const char *str, u_int *dispatch_policyp)
326f2d2d694SRobert Watson {
327f2d2d694SRobert Watson 	const struct netisr_dispatch_table_entry *ndtep;
328f2d2d694SRobert Watson 	u_int i;
329f2d2d694SRobert Watson 
3308dfea464SPedro F. Giffuni 	for (i = 0; i < nitems(netisr_dispatch_table); i++) {
331f2d2d694SRobert Watson 		ndtep = &netisr_dispatch_table[i];
332f2d2d694SRobert Watson 		if (strcmp(ndtep->ndte_policy_str, str) == 0) {
333f2d2d694SRobert Watson 			*dispatch_policyp = ndtep->ndte_policy;
334f2d2d694SRobert Watson 			return (0);
335f2d2d694SRobert Watson 		}
336f2d2d694SRobert Watson 	}
337f2d2d694SRobert Watson 	return (EINVAL);
338f2d2d694SRobert Watson }
339f2d2d694SRobert Watson 
340f2d2d694SRobert Watson static int
341f2d2d694SRobert Watson sysctl_netisr_dispatch_policy(SYSCTL_HANDLER_ARGS)
342f2d2d694SRobert Watson {
343f2d2d694SRobert Watson 	char tmp[NETISR_DISPATCH_POLICY_MAXSTR];
344f2d2d694SRobert Watson 	u_int dispatch_policy;
345f2d2d694SRobert Watson 	int error;
346f2d2d694SRobert Watson 
347f2d2d694SRobert Watson 	netisr_dispatch_policy_to_str(netisr_dispatch_policy, tmp,
348f2d2d694SRobert Watson 	    sizeof(tmp));
349f2d2d694SRobert Watson 	error = sysctl_handle_string(oidp, tmp, sizeof(tmp), req);
350f2d2d694SRobert Watson 	if (error == 0 && req->newptr != NULL) {
351f2d2d694SRobert Watson 		error = netisr_dispatch_policy_from_str(tmp,
352f2d2d694SRobert Watson 		    &dispatch_policy);
353f2d2d694SRobert Watson 		if (error == 0 && dispatch_policy == NETISR_DISPATCH_DEFAULT)
354f2d2d694SRobert Watson 			error = EINVAL;
355933e681dSDavide Italiano 		if (error == 0)
356f2d2d694SRobert Watson 			netisr_dispatch_policy = dispatch_policy;
357f2d2d694SRobert Watson 	}
358f2d2d694SRobert Watson 	return (error);
359f2d2d694SRobert Watson }
360f2d2d694SRobert Watson 
361f2d2d694SRobert Watson /*
362d4b5cae4SRobert Watson  * Register a new netisr handler, which requires initializing per-protocol
363d4b5cae4SRobert Watson  * fields for each workstream.  All netisr work is briefly suspended while
364d4b5cae4SRobert Watson  * the protocol is installed.
3651cafed39SJonathan Lemon  */
3661cafed39SJonathan Lemon void
367d4b5cae4SRobert Watson netisr_register(const struct netisr_handler *nhp)
3681cafed39SJonathan Lemon {
369484149deSBjoern A. Zeeb 	VNET_ITERATOR_DECL(vnet_iter);
370d4b5cae4SRobert Watson 	struct netisr_work *npwp;
371d4b5cae4SRobert Watson 	const char *name;
372d4b5cae4SRobert Watson 	u_int i, proto;
3731cafed39SJonathan Lemon 
374d4b5cae4SRobert Watson 	proto = nhp->nh_proto;
375d4b5cae4SRobert Watson 	name = nhp->nh_name;
37659dd72d0SRobert Watson 
3777902224cSSam Leffler 	/*
378d4b5cae4SRobert Watson 	 * Test that the requested registration is valid.
3797902224cSSam Leffler 	 */
380d4b5cae4SRobert Watson 	KASSERT(nhp->nh_name != NULL,
381d4b5cae4SRobert Watson 	    ("%s: nh_name NULL for %u", __func__, proto));
382d4b5cae4SRobert Watson 	KASSERT(nhp->nh_handler != NULL,
383d4b5cae4SRobert Watson 	    ("%s: nh_handler NULL for %s", __func__, name));
384d4b5cae4SRobert Watson 	KASSERT(nhp->nh_policy == NETISR_POLICY_SOURCE ||
385d4b5cae4SRobert Watson 	    nhp->nh_policy == NETISR_POLICY_FLOW ||
386d4b5cae4SRobert Watson 	    nhp->nh_policy == NETISR_POLICY_CPU,
387d4b5cae4SRobert Watson 	    ("%s: unsupported nh_policy %u for %s", __func__,
388d4b5cae4SRobert Watson 	    nhp->nh_policy, name));
389d4b5cae4SRobert Watson 	KASSERT(nhp->nh_policy == NETISR_POLICY_FLOW ||
390d4b5cae4SRobert Watson 	    nhp->nh_m2flow == NULL,
391d4b5cae4SRobert Watson 	    ("%s: nh_policy != FLOW but m2flow defined for %s", __func__,
392d4b5cae4SRobert Watson 	    name));
393d4b5cae4SRobert Watson 	KASSERT(nhp->nh_policy == NETISR_POLICY_CPU || nhp->nh_m2cpuid == NULL,
394d4b5cae4SRobert Watson 	    ("%s: nh_policy != CPU but m2cpuid defined for %s", __func__,
395d4b5cae4SRobert Watson 	    name));
396d4b5cae4SRobert Watson 	KASSERT(nhp->nh_policy != NETISR_POLICY_CPU || nhp->nh_m2cpuid != NULL,
397d4b5cae4SRobert Watson 	    ("%s: nh_policy == CPU but m2cpuid not defined for %s", __func__,
398d4b5cae4SRobert Watson 	    name));
399f2d2d694SRobert Watson 	KASSERT(nhp->nh_dispatch == NETISR_DISPATCH_DEFAULT ||
400f2d2d694SRobert Watson 	    nhp->nh_dispatch == NETISR_DISPATCH_DEFERRED ||
401f2d2d694SRobert Watson 	    nhp->nh_dispatch == NETISR_DISPATCH_HYBRID ||
402f2d2d694SRobert Watson 	    nhp->nh_dispatch == NETISR_DISPATCH_DIRECT,
403f2d2d694SRobert Watson 	    ("%s: invalid nh_dispatch (%u)", __func__, nhp->nh_dispatch));
404f2d2d694SRobert Watson 
405d4b5cae4SRobert Watson 	KASSERT(proto < NETISR_MAXPROT,
406d4b5cae4SRobert Watson 	    ("%s(%u, %s): protocol too big", __func__, proto, name));
407d4b5cae4SRobert Watson 
408d4b5cae4SRobert Watson 	/*
409d4b5cae4SRobert Watson 	 * Test that no existing registration exists for this protocol.
410d4b5cae4SRobert Watson 	 */
411d4b5cae4SRobert Watson 	NETISR_WLOCK();
412938448cdSRobert Watson 	KASSERT(netisr_proto[proto].np_name == NULL,
413d4b5cae4SRobert Watson 	    ("%s(%u, %s): name present", __func__, proto, name));
414938448cdSRobert Watson 	KASSERT(netisr_proto[proto].np_handler == NULL,
415d4b5cae4SRobert Watson 	    ("%s(%u, %s): handler present", __func__, proto, name));
416d4b5cae4SRobert Watson 
417938448cdSRobert Watson 	netisr_proto[proto].np_name = name;
418938448cdSRobert Watson 	netisr_proto[proto].np_handler = nhp->nh_handler;
419938448cdSRobert Watson 	netisr_proto[proto].np_m2flow = nhp->nh_m2flow;
420938448cdSRobert Watson 	netisr_proto[proto].np_m2cpuid = nhp->nh_m2cpuid;
421938448cdSRobert Watson 	netisr_proto[proto].np_drainedcpu = nhp->nh_drainedcpu;
422d4b5cae4SRobert Watson 	if (nhp->nh_qlimit == 0)
423938448cdSRobert Watson 		netisr_proto[proto].np_qlimit = netisr_defaultqlimit;
424d4b5cae4SRobert Watson 	else if (nhp->nh_qlimit > netisr_maxqlimit) {
425d4b5cae4SRobert Watson 		printf("%s: %s requested queue limit %u capped to "
426d4b5cae4SRobert Watson 		    "net.isr.maxqlimit %u\n", __func__, name, nhp->nh_qlimit,
427d4b5cae4SRobert Watson 		    netisr_maxqlimit);
428938448cdSRobert Watson 		netisr_proto[proto].np_qlimit = netisr_maxqlimit;
429d4b5cae4SRobert Watson 	} else
430938448cdSRobert Watson 		netisr_proto[proto].np_qlimit = nhp->nh_qlimit;
431938448cdSRobert Watson 	netisr_proto[proto].np_policy = nhp->nh_policy;
432f2d2d694SRobert Watson 	netisr_proto[proto].np_dispatch = nhp->nh_dispatch;
4333aa6d94eSJohn Baldwin 	CPU_FOREACH(i) {
43453402767SRobert Watson 		npwp = &(DPCPU_ID_PTR(i, nws))->nws_work[proto];
435d4b5cae4SRobert Watson 		bzero(npwp, sizeof(*npwp));
436938448cdSRobert Watson 		npwp->nw_qlimit = netisr_proto[proto].np_qlimit;
4371cafed39SJonathan Lemon 	}
438484149deSBjoern A. Zeeb 
439484149deSBjoern A. Zeeb #ifdef VIMAGE
440484149deSBjoern A. Zeeb 	/*
441484149deSBjoern A. Zeeb 	 * Test that we are in vnet0 and have a curvnet set.
442484149deSBjoern A. Zeeb 	 */
443484149deSBjoern A. Zeeb 	KASSERT(curvnet != NULL, ("%s: curvnet is NULL", __func__));
444484149deSBjoern A. Zeeb 	KASSERT(IS_DEFAULT_VNET(curvnet), ("%s: curvnet %p is not vnet0 %p",
445484149deSBjoern A. Zeeb 	    __func__, curvnet, vnet0));
446484149deSBjoern A. Zeeb 	VNET_LIST_RLOCK_NOSLEEP();
447484149deSBjoern A. Zeeb 	VNET_FOREACH(vnet_iter) {
448484149deSBjoern A. Zeeb 		CURVNET_SET(vnet_iter);
449484149deSBjoern A. Zeeb 		V_netisr_enable[proto] = 1;
450484149deSBjoern A. Zeeb 		CURVNET_RESTORE();
451484149deSBjoern A. Zeeb 	}
452484149deSBjoern A. Zeeb 	VNET_LIST_RUNLOCK_NOSLEEP();
453484149deSBjoern A. Zeeb #endif
454d4b5cae4SRobert Watson 	NETISR_WUNLOCK();
4551cafed39SJonathan Lemon }
4561cafed39SJonathan Lemon 
4571cafed39SJonathan Lemon /*
458d4b5cae4SRobert Watson  * Clear drop counters across all workstreams for a protocol.
459d4b5cae4SRobert Watson  */
460d4b5cae4SRobert Watson void
461d4b5cae4SRobert Watson netisr_clearqdrops(const struct netisr_handler *nhp)
462d4b5cae4SRobert Watson {
463d4b5cae4SRobert Watson 	struct netisr_work *npwp;
464d4b5cae4SRobert Watson #ifdef INVARIANTS
465d4b5cae4SRobert Watson 	const char *name;
466d4b5cae4SRobert Watson #endif
467d4b5cae4SRobert Watson 	u_int i, proto;
468d4b5cae4SRobert Watson 
469d4b5cae4SRobert Watson 	proto = nhp->nh_proto;
470d4b5cae4SRobert Watson #ifdef INVARIANTS
471d4b5cae4SRobert Watson 	name = nhp->nh_name;
472d4b5cae4SRobert Watson #endif
473d4b5cae4SRobert Watson 	KASSERT(proto < NETISR_MAXPROT,
474d4b5cae4SRobert Watson 	    ("%s(%u): protocol too big for %s", __func__, proto, name));
475d4b5cae4SRobert Watson 
476d4b5cae4SRobert Watson 	NETISR_WLOCK();
477938448cdSRobert Watson 	KASSERT(netisr_proto[proto].np_handler != NULL,
478d4b5cae4SRobert Watson 	    ("%s(%u): protocol not registered for %s", __func__, proto,
479d4b5cae4SRobert Watson 	    name));
480d4b5cae4SRobert Watson 
4813aa6d94eSJohn Baldwin 	CPU_FOREACH(i) {
48253402767SRobert Watson 		npwp = &(DPCPU_ID_PTR(i, nws))->nws_work[proto];
483d4b5cae4SRobert Watson 		npwp->nw_qdrops = 0;
484d4b5cae4SRobert Watson 	}
485d4b5cae4SRobert Watson 	NETISR_WUNLOCK();
486d4b5cae4SRobert Watson }
487d4b5cae4SRobert Watson 
488d4b5cae4SRobert Watson /*
4890a32e29fSRobert Watson  * Query current drop counters across all workstreams for a protocol.
490d4b5cae4SRobert Watson  */
491d4b5cae4SRobert Watson void
492d4b5cae4SRobert Watson netisr_getqdrops(const struct netisr_handler *nhp, u_int64_t *qdropp)
493d4b5cae4SRobert Watson {
494d4b5cae4SRobert Watson 	struct netisr_work *npwp;
495d4b5cae4SRobert Watson 	struct rm_priotracker tracker;
496d4b5cae4SRobert Watson #ifdef INVARIANTS
497d4b5cae4SRobert Watson 	const char *name;
498d4b5cae4SRobert Watson #endif
499d4b5cae4SRobert Watson 	u_int i, proto;
500d4b5cae4SRobert Watson 
501d4b5cae4SRobert Watson 	*qdropp = 0;
502d4b5cae4SRobert Watson 	proto = nhp->nh_proto;
503d4b5cae4SRobert Watson #ifdef INVARIANTS
504d4b5cae4SRobert Watson 	name = nhp->nh_name;
505d4b5cae4SRobert Watson #endif
506d4b5cae4SRobert Watson 	KASSERT(proto < NETISR_MAXPROT,
507d4b5cae4SRobert Watson 	    ("%s(%u): protocol too big for %s", __func__, proto, name));
508d4b5cae4SRobert Watson 
509d4b5cae4SRobert Watson 	NETISR_RLOCK(&tracker);
510938448cdSRobert Watson 	KASSERT(netisr_proto[proto].np_handler != NULL,
511d4b5cae4SRobert Watson 	    ("%s(%u): protocol not registered for %s", __func__, proto,
512d4b5cae4SRobert Watson 	    name));
513d4b5cae4SRobert Watson 
5143aa6d94eSJohn Baldwin 	CPU_FOREACH(i) {
51553402767SRobert Watson 		npwp = &(DPCPU_ID_PTR(i, nws))->nws_work[proto];
516d4b5cae4SRobert Watson 		*qdropp += npwp->nw_qdrops;
517d4b5cae4SRobert Watson 	}
518d4b5cae4SRobert Watson 	NETISR_RUNLOCK(&tracker);
519d4b5cae4SRobert Watson }
520d4b5cae4SRobert Watson 
521d4b5cae4SRobert Watson /*
5220a32e29fSRobert Watson  * Query current per-workstream queue limit for a protocol.
523d4b5cae4SRobert Watson  */
524d4b5cae4SRobert Watson void
525d4b5cae4SRobert Watson netisr_getqlimit(const struct netisr_handler *nhp, u_int *qlimitp)
526d4b5cae4SRobert Watson {
527d4b5cae4SRobert Watson 	struct rm_priotracker tracker;
528d4b5cae4SRobert Watson #ifdef INVARIANTS
529d4b5cae4SRobert Watson 	const char *name;
530d4b5cae4SRobert Watson #endif
531d4b5cae4SRobert Watson 	u_int proto;
532d4b5cae4SRobert Watson 
533d4b5cae4SRobert Watson 	proto = nhp->nh_proto;
534d4b5cae4SRobert Watson #ifdef INVARIANTS
535d4b5cae4SRobert Watson 	name = nhp->nh_name;
536d4b5cae4SRobert Watson #endif
537d4b5cae4SRobert Watson 	KASSERT(proto < NETISR_MAXPROT,
538d4b5cae4SRobert Watson 	    ("%s(%u): protocol too big for %s", __func__, proto, name));
539d4b5cae4SRobert Watson 
540d4b5cae4SRobert Watson 	NETISR_RLOCK(&tracker);
541938448cdSRobert Watson 	KASSERT(netisr_proto[proto].np_handler != NULL,
542d4b5cae4SRobert Watson 	    ("%s(%u): protocol not registered for %s", __func__, proto,
543d4b5cae4SRobert Watson 	    name));
544938448cdSRobert Watson 	*qlimitp = netisr_proto[proto].np_qlimit;
545d4b5cae4SRobert Watson 	NETISR_RUNLOCK(&tracker);
546d4b5cae4SRobert Watson }
547d4b5cae4SRobert Watson 
548d4b5cae4SRobert Watson /*
549d4b5cae4SRobert Watson  * Update the queue limit across per-workstream queues for a protocol.  We
550d4b5cae4SRobert Watson  * simply change the limits, and don't drain overflowed packets as they will
551d4b5cae4SRobert Watson  * (hopefully) take care of themselves shortly.
5521cafed39SJonathan Lemon  */
5531cafed39SJonathan Lemon int
554d4b5cae4SRobert Watson netisr_setqlimit(const struct netisr_handler *nhp, u_int qlimit)
5551cafed39SJonathan Lemon {
556d4b5cae4SRobert Watson 	struct netisr_work *npwp;
557d4b5cae4SRobert Watson #ifdef INVARIANTS
558d4b5cae4SRobert Watson 	const char *name;
559d4b5cae4SRobert Watson #endif
560d4b5cae4SRobert Watson 	u_int i, proto;
5611cafed39SJonathan Lemon 
562d4b5cae4SRobert Watson 	if (qlimit > netisr_maxqlimit)
563d4b5cae4SRobert Watson 		return (EINVAL);
564d4b5cae4SRobert Watson 
565d4b5cae4SRobert Watson 	proto = nhp->nh_proto;
566d4b5cae4SRobert Watson #ifdef INVARIANTS
567d4b5cae4SRobert Watson 	name = nhp->nh_name;
568d4b5cae4SRobert Watson #endif
569d4b5cae4SRobert Watson 	KASSERT(proto < NETISR_MAXPROT,
570d4b5cae4SRobert Watson 	    ("%s(%u): protocol too big for %s", __func__, proto, name));
571d4b5cae4SRobert Watson 
572d4b5cae4SRobert Watson 	NETISR_WLOCK();
573938448cdSRobert Watson 	KASSERT(netisr_proto[proto].np_handler != NULL,
574d4b5cae4SRobert Watson 	    ("%s(%u): protocol not registered for %s", __func__, proto,
575d4b5cae4SRobert Watson 	    name));
576d4b5cae4SRobert Watson 
577938448cdSRobert Watson 	netisr_proto[proto].np_qlimit = qlimit;
5783aa6d94eSJohn Baldwin 	CPU_FOREACH(i) {
57953402767SRobert Watson 		npwp = &(DPCPU_ID_PTR(i, nws))->nws_work[proto];
580d4b5cae4SRobert Watson 		npwp->nw_qlimit = qlimit;
581fb68148fSJonathan Lemon 	}
582d4b5cae4SRobert Watson 	NETISR_WUNLOCK();
5833161f583SAndre Oppermann 	return (0);
584e3b6e33cSJake Burkholder }
585e3b6e33cSJake Burkholder 
586d4b5cae4SRobert Watson /*
587d4b5cae4SRobert Watson  * Drain all packets currently held in a particular protocol work queue.
588d4b5cae4SRobert Watson  */
589e3b6e33cSJake Burkholder static void
590d4b5cae4SRobert Watson netisr_drain_proto(struct netisr_work *npwp)
591e3b6e33cSJake Burkholder {
592d4b5cae4SRobert Watson 	struct mbuf *m;
593d4b5cae4SRobert Watson 
594d4b5cae4SRobert Watson 	/*
595d4b5cae4SRobert Watson 	 * We would assert the lock on the workstream but it's not passed in.
596d4b5cae4SRobert Watson 	 */
597d4b5cae4SRobert Watson 	while ((m = npwp->nw_head) != NULL) {
598d4b5cae4SRobert Watson 		npwp->nw_head = m->m_nextpkt;
599d4b5cae4SRobert Watson 		m->m_nextpkt = NULL;
600d4b5cae4SRobert Watson 		if (npwp->nw_head == NULL)
601d4b5cae4SRobert Watson 			npwp->nw_tail = NULL;
602d4b5cae4SRobert Watson 		npwp->nw_len--;
603d4b5cae4SRobert Watson 		m_freem(m);
604d4b5cae4SRobert Watson 	}
605d4b5cae4SRobert Watson 	KASSERT(npwp->nw_tail == NULL, ("%s: tail", __func__));
606d4b5cae4SRobert Watson 	KASSERT(npwp->nw_len == 0, ("%s: len", __func__));
607d4b5cae4SRobert Watson }
608d4b5cae4SRobert Watson 
609d4b5cae4SRobert Watson /*
610d4b5cae4SRobert Watson  * Remove the registration of a network protocol, which requires clearing
611d4b5cae4SRobert Watson  * per-protocol fields across all workstreams, including freeing all mbufs in
612d4b5cae4SRobert Watson  * the queues at time of unregister.  All work in netisr is briefly suspended
613d4b5cae4SRobert Watson  * while this takes place.
614d4b5cae4SRobert Watson  */
615d4b5cae4SRobert Watson void
616d4b5cae4SRobert Watson netisr_unregister(const struct netisr_handler *nhp)
617d4b5cae4SRobert Watson {
618484149deSBjoern A. Zeeb 	VNET_ITERATOR_DECL(vnet_iter);
619d4b5cae4SRobert Watson 	struct netisr_work *npwp;
620d4b5cae4SRobert Watson #ifdef INVARIANTS
621d4b5cae4SRobert Watson 	const char *name;
622d4b5cae4SRobert Watson #endif
623d4b5cae4SRobert Watson 	u_int i, proto;
624d4b5cae4SRobert Watson 
625d4b5cae4SRobert Watson 	proto = nhp->nh_proto;
626d4b5cae4SRobert Watson #ifdef INVARIANTS
627d4b5cae4SRobert Watson 	name = nhp->nh_name;
628d4b5cae4SRobert Watson #endif
629d4b5cae4SRobert Watson 	KASSERT(proto < NETISR_MAXPROT,
630d4b5cae4SRobert Watson 	    ("%s(%u): protocol too big for %s", __func__, proto, name));
631d4b5cae4SRobert Watson 
632d4b5cae4SRobert Watson 	NETISR_WLOCK();
633938448cdSRobert Watson 	KASSERT(netisr_proto[proto].np_handler != NULL,
634d4b5cae4SRobert Watson 	    ("%s(%u): protocol not registered for %s", __func__, proto,
635d4b5cae4SRobert Watson 	    name));
636d4b5cae4SRobert Watson 
637484149deSBjoern A. Zeeb #ifdef VIMAGE
638484149deSBjoern A. Zeeb 	VNET_LIST_RLOCK_NOSLEEP();
639484149deSBjoern A. Zeeb 	VNET_FOREACH(vnet_iter) {
640484149deSBjoern A. Zeeb 		CURVNET_SET(vnet_iter);
641484149deSBjoern A. Zeeb 		V_netisr_enable[proto] = 0;
642484149deSBjoern A. Zeeb 		CURVNET_RESTORE();
643484149deSBjoern A. Zeeb 	}
644484149deSBjoern A. Zeeb 	VNET_LIST_RUNLOCK_NOSLEEP();
645484149deSBjoern A. Zeeb #endif
646484149deSBjoern A. Zeeb 
647938448cdSRobert Watson 	netisr_proto[proto].np_name = NULL;
648938448cdSRobert Watson 	netisr_proto[proto].np_handler = NULL;
649938448cdSRobert Watson 	netisr_proto[proto].np_m2flow = NULL;
650938448cdSRobert Watson 	netisr_proto[proto].np_m2cpuid = NULL;
651938448cdSRobert Watson 	netisr_proto[proto].np_qlimit = 0;
652938448cdSRobert Watson 	netisr_proto[proto].np_policy = 0;
6533aa6d94eSJohn Baldwin 	CPU_FOREACH(i) {
65453402767SRobert Watson 		npwp = &(DPCPU_ID_PTR(i, nws))->nws_work[proto];
655d4b5cae4SRobert Watson 		netisr_drain_proto(npwp);
656d4b5cae4SRobert Watson 		bzero(npwp, sizeof(*npwp));
657d4b5cae4SRobert Watson 	}
658d4b5cae4SRobert Watson 	NETISR_WUNLOCK();
659d4b5cae4SRobert Watson }
660d4b5cae4SRobert Watson 
661484149deSBjoern A. Zeeb #ifdef VIMAGE
662484149deSBjoern A. Zeeb void
663484149deSBjoern A. Zeeb netisr_register_vnet(const struct netisr_handler *nhp)
664484149deSBjoern A. Zeeb {
665484149deSBjoern A. Zeeb 	u_int proto;
666484149deSBjoern A. Zeeb 
667484149deSBjoern A. Zeeb 	proto = nhp->nh_proto;
668484149deSBjoern A. Zeeb 
669484149deSBjoern A. Zeeb 	KASSERT(curvnet != NULL, ("%s: curvnet is NULL", __func__));
670484149deSBjoern A. Zeeb 	KASSERT(proto < NETISR_MAXPROT,
671484149deSBjoern A. Zeeb 	    ("%s(%u): protocol too big for %s", __func__, proto, nhp->nh_name));
672484149deSBjoern A. Zeeb 	NETISR_WLOCK();
673484149deSBjoern A. Zeeb 	KASSERT(netisr_proto[proto].np_handler != NULL,
674484149deSBjoern A. Zeeb 	    ("%s(%u): protocol not registered for %s", __func__, proto,
675484149deSBjoern A. Zeeb 	    nhp->nh_name));
676484149deSBjoern A. Zeeb 
677484149deSBjoern A. Zeeb 	V_netisr_enable[proto] = 1;
678484149deSBjoern A. Zeeb 	NETISR_WUNLOCK();
679484149deSBjoern A. Zeeb }
680484149deSBjoern A. Zeeb 
681484149deSBjoern A. Zeeb static void
682484149deSBjoern A. Zeeb netisr_drain_proto_vnet(struct vnet *vnet, u_int proto)
683484149deSBjoern A. Zeeb {
684484149deSBjoern A. Zeeb 	struct netisr_workstream *nwsp;
685484149deSBjoern A. Zeeb 	struct netisr_work *npwp;
686484149deSBjoern A. Zeeb 	struct mbuf *m, *mp, *n, *ne;
687484149deSBjoern A. Zeeb 	u_int i;
688484149deSBjoern A. Zeeb 
689484149deSBjoern A. Zeeb 	KASSERT(vnet != NULL, ("%s: vnet is NULL", __func__));
690484149deSBjoern A. Zeeb 	NETISR_LOCK_ASSERT();
691484149deSBjoern A. Zeeb 
692484149deSBjoern A. Zeeb 	CPU_FOREACH(i) {
693484149deSBjoern A. Zeeb 		nwsp = DPCPU_ID_PTR(i, nws);
694484149deSBjoern A. Zeeb 		if (nwsp->nws_intr_event == NULL)
695484149deSBjoern A. Zeeb 			continue;
696484149deSBjoern A. Zeeb 		npwp = &nwsp->nws_work[proto];
697484149deSBjoern A. Zeeb 		NWS_LOCK(nwsp);
698484149deSBjoern A. Zeeb 
699484149deSBjoern A. Zeeb 		/*
700484149deSBjoern A. Zeeb 		 * Rather than dissecting and removing mbufs from the middle
701484149deSBjoern A. Zeeb 		 * of the chain, we build a new chain if the packet stays and
702484149deSBjoern A. Zeeb 		 * update the head and tail pointers at the end.  All packets
703484149deSBjoern A. Zeeb 		 * matching the given vnet are freed.
704484149deSBjoern A. Zeeb 		 */
705484149deSBjoern A. Zeeb 		m = npwp->nw_head;
706484149deSBjoern A. Zeeb 		n = ne = NULL;
707484149deSBjoern A. Zeeb 		while (m != NULL) {
708484149deSBjoern A. Zeeb 			mp = m;
709484149deSBjoern A. Zeeb 			m = m->m_nextpkt;
710484149deSBjoern A. Zeeb 			mp->m_nextpkt = NULL;
711484149deSBjoern A. Zeeb 			if (mp->m_pkthdr.rcvif->if_vnet != vnet) {
712484149deSBjoern A. Zeeb 				if (n == NULL) {
713484149deSBjoern A. Zeeb 					n = ne = mp;
714484149deSBjoern A. Zeeb 				} else {
715484149deSBjoern A. Zeeb 					ne->m_nextpkt = mp;
716484149deSBjoern A. Zeeb 					ne = mp;
717484149deSBjoern A. Zeeb 				}
718484149deSBjoern A. Zeeb 				continue;
719484149deSBjoern A. Zeeb 			}
720484149deSBjoern A. Zeeb 			/* This is a packet in the selected vnet. Free it. */
721484149deSBjoern A. Zeeb 			npwp->nw_len--;
722484149deSBjoern A. Zeeb 			m_freem(mp);
723484149deSBjoern A. Zeeb 		}
724484149deSBjoern A. Zeeb 		npwp->nw_head = n;
725484149deSBjoern A. Zeeb 		npwp->nw_tail = ne;
726484149deSBjoern A. Zeeb 		NWS_UNLOCK(nwsp);
727484149deSBjoern A. Zeeb 	}
728484149deSBjoern A. Zeeb }
729484149deSBjoern A. Zeeb 
730484149deSBjoern A. Zeeb void
731484149deSBjoern A. Zeeb netisr_unregister_vnet(const struct netisr_handler *nhp)
732484149deSBjoern A. Zeeb {
733484149deSBjoern A. Zeeb 	u_int proto;
734484149deSBjoern A. Zeeb 
735484149deSBjoern A. Zeeb 	proto = nhp->nh_proto;
736484149deSBjoern A. Zeeb 
737484149deSBjoern A. Zeeb 	KASSERT(curvnet != NULL, ("%s: curvnet is NULL", __func__));
738484149deSBjoern A. Zeeb 	KASSERT(proto < NETISR_MAXPROT,
739484149deSBjoern A. Zeeb 	    ("%s(%u): protocol too big for %s", __func__, proto, nhp->nh_name));
740484149deSBjoern A. Zeeb 	NETISR_WLOCK();
741484149deSBjoern A. Zeeb 	KASSERT(netisr_proto[proto].np_handler != NULL,
742484149deSBjoern A. Zeeb 	    ("%s(%u): protocol not registered for %s", __func__, proto,
743484149deSBjoern A. Zeeb 	    nhp->nh_name));
744484149deSBjoern A. Zeeb 
745484149deSBjoern A. Zeeb 	V_netisr_enable[proto] = 0;
746484149deSBjoern A. Zeeb 
747484149deSBjoern A. Zeeb 	netisr_drain_proto_vnet(curvnet, proto);
748484149deSBjoern A. Zeeb 	NETISR_WUNLOCK();
749484149deSBjoern A. Zeeb }
750484149deSBjoern A. Zeeb #endif
751484149deSBjoern A. Zeeb 
752d4b5cae4SRobert Watson /*
753f2d2d694SRobert Watson  * Compose the global and per-protocol policies on dispatch, and return the
754f2d2d694SRobert Watson  * dispatch policy to use.
755f2d2d694SRobert Watson  */
756f2d2d694SRobert Watson static u_int
757f2d2d694SRobert Watson netisr_get_dispatch(struct netisr_proto *npp)
758f2d2d694SRobert Watson {
759f2d2d694SRobert Watson 
760f2d2d694SRobert Watson 	/*
761f2d2d694SRobert Watson 	 * Protocol-specific configuration overrides the global default.
762f2d2d694SRobert Watson 	 */
763f2d2d694SRobert Watson 	if (npp->np_dispatch != NETISR_DISPATCH_DEFAULT)
764f2d2d694SRobert Watson 		return (npp->np_dispatch);
765f2d2d694SRobert Watson 	return (netisr_dispatch_policy);
766f2d2d694SRobert Watson }
767f2d2d694SRobert Watson 
768f2d2d694SRobert Watson /*
769d4b5cae4SRobert Watson  * Look up the workstream given a packet and source identifier.  Do this by
770d4b5cae4SRobert Watson  * checking the protocol's policy, and optionally call out to the protocol
771d4b5cae4SRobert Watson  * for assistance if required.
772d4b5cae4SRobert Watson  */
773d4b5cae4SRobert Watson static struct mbuf *
774f2d2d694SRobert Watson netisr_select_cpuid(struct netisr_proto *npp, u_int dispatch_policy,
775f2d2d694SRobert Watson     uintptr_t source, struct mbuf *m, u_int *cpuidp)
776d4b5cae4SRobert Watson {
777d4b5cae4SRobert Watson 	struct ifnet *ifp;
778f2d2d694SRobert Watson 	u_int policy;
779d4b5cae4SRobert Watson 
780d4b5cae4SRobert Watson 	NETISR_LOCK_ASSERT();
781d4b5cae4SRobert Watson 
782d4b5cae4SRobert Watson 	/*
783d4b5cae4SRobert Watson 	 * In the event we have only one worker, shortcut and deliver to it
784d4b5cae4SRobert Watson 	 * without further ado.
785d4b5cae4SRobert Watson 	 */
786d4b5cae4SRobert Watson 	if (nws_count == 1) {
787d4b5cae4SRobert Watson 		*cpuidp = nws_array[0];
788d4b5cae4SRobert Watson 		return (m);
789d4b5cae4SRobert Watson 	}
790d4b5cae4SRobert Watson 
791d4b5cae4SRobert Watson 	/*
792d4b5cae4SRobert Watson 	 * What happens next depends on the policy selected by the protocol.
793d4b5cae4SRobert Watson 	 * If we want to support per-interface policies, we should do that
794d4b5cae4SRobert Watson 	 * here first.
795d4b5cae4SRobert Watson 	 */
796f2d2d694SRobert Watson 	policy = npp->np_policy;
797f2d2d694SRobert Watson 	if (policy == NETISR_POLICY_CPU) {
798f2d2d694SRobert Watson 		m = npp->np_m2cpuid(m, source, cpuidp);
799f2d2d694SRobert Watson 		if (m == NULL)
800f2d2d694SRobert Watson 			return (NULL);
801d4b5cae4SRobert Watson 
802f2d2d694SRobert Watson 		/*
803f2d2d694SRobert Watson 		 * It's possible for a protocol not to have a good idea about
804f2d2d694SRobert Watson 		 * where to process a packet, in which case we fall back on
805f2d2d694SRobert Watson 		 * the netisr code to decide.  In the hybrid case, return the
806f2d2d694SRobert Watson 		 * current CPU ID, which will force an immediate direct
807f2d2d694SRobert Watson 		 * dispatch.  In the queued case, fall back on the SOURCE
808f2d2d694SRobert Watson 		 * policy.
809f2d2d694SRobert Watson 		 */
810*fdf95c0bSAndrey V. Elsukov 		if (*cpuidp != NETISR_CPUID_NONE) {
811*fdf95c0bSAndrey V. Elsukov 			*cpuidp = netisr_get_cpuid(*cpuidp);
812f2d2d694SRobert Watson 			return (m);
813*fdf95c0bSAndrey V. Elsukov 		}
814f2d2d694SRobert Watson 		if (dispatch_policy == NETISR_DISPATCH_HYBRID) {
815*fdf95c0bSAndrey V. Elsukov 			*cpuidp = netisr_get_cpuid(curcpu);
816f2d2d694SRobert Watson 			return (m);
817f2d2d694SRobert Watson 		}
818f2d2d694SRobert Watson 		policy = NETISR_POLICY_SOURCE;
819f2d2d694SRobert Watson 	}
820f2d2d694SRobert Watson 
821f2d2d694SRobert Watson 	if (policy == NETISR_POLICY_FLOW) {
822c2529042SHans Petter Selasky 		if (M_HASHTYPE_GET(m) == M_HASHTYPE_NONE &&
823c2529042SHans Petter Selasky 		    npp->np_m2flow != NULL) {
824d4b5cae4SRobert Watson 			m = npp->np_m2flow(m, source);
825d4b5cae4SRobert Watson 			if (m == NULL)
826d4b5cae4SRobert Watson 				return (NULL);
827d4b5cae4SRobert Watson 		}
828c2529042SHans Petter Selasky 		if (M_HASHTYPE_GET(m) != M_HASHTYPE_NONE) {
829d4b5cae4SRobert Watson 			*cpuidp =
830d4b5cae4SRobert Watson 			    netisr_default_flow2cpu(m->m_pkthdr.flowid);
831d4b5cae4SRobert Watson 			return (m);
832d4b5cae4SRobert Watson 		}
833f2d2d694SRobert Watson 		policy = NETISR_POLICY_SOURCE;
834f2d2d694SRobert Watson 	}
835d4b5cae4SRobert Watson 
836f2d2d694SRobert Watson 	KASSERT(policy == NETISR_POLICY_SOURCE,
837f2d2d694SRobert Watson 	    ("%s: invalid policy %u for %s", __func__, npp->np_policy,
838f2d2d694SRobert Watson 	    npp->np_name));
839f2d2d694SRobert Watson 
840d4b5cae4SRobert Watson 	ifp = m->m_pkthdr.rcvif;
841d4b5cae4SRobert Watson 	if (ifp != NULL)
842f2d2d694SRobert Watson 		*cpuidp = nws_array[(ifp->if_index + source) % nws_count];
843d4b5cae4SRobert Watson 	else
844d4b5cae4SRobert Watson 		*cpuidp = nws_array[source % nws_count];
845d4b5cae4SRobert Watson 	return (m);
846d4b5cae4SRobert Watson }
847d4b5cae4SRobert Watson 
848d4b5cae4SRobert Watson /*
849d4b5cae4SRobert Watson  * Process packets associated with a workstream and protocol.  For reasons of
850d4b5cae4SRobert Watson  * fairness, we process up to one complete netisr queue at a time, moving the
851d4b5cae4SRobert Watson  * queue to a stack-local queue for processing, but do not loop refreshing
852d4b5cae4SRobert Watson  * from the global queue.  The caller is responsible for deciding whether to
853d4b5cae4SRobert Watson  * loop, and for setting the NWS_RUNNING flag.  The passed workstream will be
854d4b5cae4SRobert Watson  * locked on entry and relocked before return, but will be released while
855d4b5cae4SRobert Watson  * processing.  The number of packets processed is returned.
856d4b5cae4SRobert Watson  */
857d4b5cae4SRobert Watson static u_int
858d4b5cae4SRobert Watson netisr_process_workstream_proto(struct netisr_workstream *nwsp, u_int proto)
859d4b5cae4SRobert Watson {
860d4b5cae4SRobert Watson 	struct netisr_work local_npw, *npwp;
861d4b5cae4SRobert Watson 	u_int handled;
862d4b5cae4SRobert Watson 	struct mbuf *m;
863d4b5cae4SRobert Watson 
864d4b5cae4SRobert Watson 	NETISR_LOCK_ASSERT();
865d4b5cae4SRobert Watson 	NWS_LOCK_ASSERT(nwsp);
866d4b5cae4SRobert Watson 
867d4b5cae4SRobert Watson 	KASSERT(nwsp->nws_flags & NWS_RUNNING,
868d4b5cae4SRobert Watson 	    ("%s(%u): not running", __func__, proto));
869d4b5cae4SRobert Watson 	KASSERT(proto >= 0 && proto < NETISR_MAXPROT,
870d4b5cae4SRobert Watson 	    ("%s(%u): invalid proto\n", __func__, proto));
871d4b5cae4SRobert Watson 
872d4b5cae4SRobert Watson 	npwp = &nwsp->nws_work[proto];
873d4b5cae4SRobert Watson 	if (npwp->nw_len == 0)
874d4b5cae4SRobert Watson 		return (0);
875d4b5cae4SRobert Watson 
876d4b5cae4SRobert Watson 	/*
877d4b5cae4SRobert Watson 	 * Move the global work queue to a thread-local work queue.
878d4b5cae4SRobert Watson 	 *
879d4b5cae4SRobert Watson 	 * Notice that this means the effective maximum length of the queue
880d4b5cae4SRobert Watson 	 * is actually twice that of the maximum queue length specified in
881d4b5cae4SRobert Watson 	 * the protocol registration call.
882d4b5cae4SRobert Watson 	 */
883d4b5cae4SRobert Watson 	handled = npwp->nw_len;
884d4b5cae4SRobert Watson 	local_npw = *npwp;
885d4b5cae4SRobert Watson 	npwp->nw_head = NULL;
886d4b5cae4SRobert Watson 	npwp->nw_tail = NULL;
887d4b5cae4SRobert Watson 	npwp->nw_len = 0;
888d4b5cae4SRobert Watson 	nwsp->nws_pendingbits &= ~(1 << proto);
889d4b5cae4SRobert Watson 	NWS_UNLOCK(nwsp);
890d4b5cae4SRobert Watson 	while ((m = local_npw.nw_head) != NULL) {
891d4b5cae4SRobert Watson 		local_npw.nw_head = m->m_nextpkt;
892d4b5cae4SRobert Watson 		m->m_nextpkt = NULL;
893d4b5cae4SRobert Watson 		if (local_npw.nw_head == NULL)
894d4b5cae4SRobert Watson 			local_npw.nw_tail = NULL;
895d4b5cae4SRobert Watson 		local_npw.nw_len--;
8960028e524SBjoern A. Zeeb 		VNET_ASSERT(m->m_pkthdr.rcvif != NULL,
8970028e524SBjoern A. Zeeb 		    ("%s:%d rcvif == NULL: m=%p", __func__, __LINE__, m));
898d4b5cae4SRobert Watson 		CURVNET_SET(m->m_pkthdr.rcvif->if_vnet);
899938448cdSRobert Watson 		netisr_proto[proto].np_handler(m);
900d4b5cae4SRobert Watson 		CURVNET_RESTORE();
901d4b5cae4SRobert Watson 	}
902d4b5cae4SRobert Watson 	KASSERT(local_npw.nw_len == 0,
903d4b5cae4SRobert Watson 	    ("%s(%u): len %u", __func__, proto, local_npw.nw_len));
904938448cdSRobert Watson 	if (netisr_proto[proto].np_drainedcpu)
905938448cdSRobert Watson 		netisr_proto[proto].np_drainedcpu(nwsp->nws_cpu);
906d4b5cae4SRobert Watson 	NWS_LOCK(nwsp);
907d4b5cae4SRobert Watson 	npwp->nw_handled += handled;
908d4b5cae4SRobert Watson 	return (handled);
909d4b5cae4SRobert Watson }
910d4b5cae4SRobert Watson 
911d4b5cae4SRobert Watson /*
9120a32e29fSRobert Watson  * SWI handler for netisr -- processes packets in a set of workstreams that
913d4b5cae4SRobert Watson  * it owns, woken up by calls to NWS_SIGNAL().  If this workstream is already
914d4b5cae4SRobert Watson  * being direct dispatched, go back to sleep and wait for the dispatching
915d4b5cae4SRobert Watson  * thread to wake us up again.
916d4b5cae4SRobert Watson  */
917d4b5cae4SRobert Watson static void
918d4b5cae4SRobert Watson swi_net(void *arg)
919d4b5cae4SRobert Watson {
920d4b5cae4SRobert Watson #ifdef NETISR_LOCKING
921d4b5cae4SRobert Watson 	struct rm_priotracker tracker;
922d4b5cae4SRobert Watson #endif
923d4b5cae4SRobert Watson 	struct netisr_workstream *nwsp;
924d4b5cae4SRobert Watson 	u_int bits, prot;
925d4b5cae4SRobert Watson 
926d4b5cae4SRobert Watson 	nwsp = arg;
927d4b5cae4SRobert Watson 
9281cafed39SJonathan Lemon #ifdef DEVICE_POLLING
929d4b5cae4SRobert Watson 	KASSERT(nws_count == 1,
930d4b5cae4SRobert Watson 	    ("%s: device_polling but nws_count != 1", __func__));
931d4b5cae4SRobert Watson 	netisr_poll();
932d4b5cae4SRobert Watson #endif
933d4b5cae4SRobert Watson #ifdef NETISR_LOCKING
934d4b5cae4SRobert Watson 	NETISR_RLOCK(&tracker);
935d4b5cae4SRobert Watson #endif
936d4b5cae4SRobert Watson 	NWS_LOCK(nwsp);
937d4b5cae4SRobert Watson 	KASSERT(!(nwsp->nws_flags & NWS_RUNNING), ("swi_net: running"));
938d4b5cae4SRobert Watson 	if (nwsp->nws_flags & NWS_DISPATCHING)
939d4b5cae4SRobert Watson 		goto out;
940d4b5cae4SRobert Watson 	nwsp->nws_flags |= NWS_RUNNING;
941d4b5cae4SRobert Watson 	nwsp->nws_flags &= ~NWS_SCHEDULED;
942d4b5cae4SRobert Watson 	while ((bits = nwsp->nws_pendingbits) != 0) {
943d4b5cae4SRobert Watson 		while ((prot = ffs(bits)) != 0) {
944d4b5cae4SRobert Watson 			prot--;
945d4b5cae4SRobert Watson 			bits &= ~(1 << prot);
946d4b5cae4SRobert Watson 			(void)netisr_process_workstream_proto(nwsp, prot);
947d4b5cae4SRobert Watson 		}
948d4b5cae4SRobert Watson 	}
949d4b5cae4SRobert Watson 	nwsp->nws_flags &= ~NWS_RUNNING;
950d4b5cae4SRobert Watson out:
951d4b5cae4SRobert Watson 	NWS_UNLOCK(nwsp);
952d4b5cae4SRobert Watson #ifdef NETISR_LOCKING
953d4b5cae4SRobert Watson 	NETISR_RUNLOCK(&tracker);
954d4b5cae4SRobert Watson #endif
955d4b5cae4SRobert Watson #ifdef DEVICE_POLLING
956d4b5cae4SRobert Watson 	netisr_pollmore();
957d4b5cae4SRobert Watson #endif
958d4b5cae4SRobert Watson }
959d4b5cae4SRobert Watson 
960d4b5cae4SRobert Watson static int
961d4b5cae4SRobert Watson netisr_queue_workstream(struct netisr_workstream *nwsp, u_int proto,
962d4b5cae4SRobert Watson     struct netisr_work *npwp, struct mbuf *m, int *dosignalp)
963d4b5cae4SRobert Watson {
964d4b5cae4SRobert Watson 
965d4b5cae4SRobert Watson 	NWS_LOCK_ASSERT(nwsp);
966d4b5cae4SRobert Watson 
967d4b5cae4SRobert Watson 	*dosignalp = 0;
968d4b5cae4SRobert Watson 	if (npwp->nw_len < npwp->nw_qlimit) {
969d4b5cae4SRobert Watson 		m->m_nextpkt = NULL;
970d4b5cae4SRobert Watson 		if (npwp->nw_head == NULL) {
971d4b5cae4SRobert Watson 			npwp->nw_head = m;
972d4b5cae4SRobert Watson 			npwp->nw_tail = m;
973d4b5cae4SRobert Watson 		} else {
974d4b5cae4SRobert Watson 			npwp->nw_tail->m_nextpkt = m;
975d4b5cae4SRobert Watson 			npwp->nw_tail = m;
976d4b5cae4SRobert Watson 		}
977d4b5cae4SRobert Watson 		npwp->nw_len++;
978d4b5cae4SRobert Watson 		if (npwp->nw_len > npwp->nw_watermark)
979d4b5cae4SRobert Watson 			npwp->nw_watermark = npwp->nw_len;
9800a32e29fSRobert Watson 
9810a32e29fSRobert Watson 		/*
9820a32e29fSRobert Watson 		 * We must set the bit regardless of NWS_RUNNING, so that
9830a32e29fSRobert Watson 		 * swi_net() keeps calling netisr_process_workstream_proto().
9840a32e29fSRobert Watson 		 */
985d4b5cae4SRobert Watson 		nwsp->nws_pendingbits |= (1 << proto);
986d4b5cae4SRobert Watson 		if (!(nwsp->nws_flags &
987d4b5cae4SRobert Watson 		    (NWS_RUNNING | NWS_DISPATCHING | NWS_SCHEDULED))) {
988d4b5cae4SRobert Watson 			nwsp->nws_flags |= NWS_SCHEDULED;
989d4b5cae4SRobert Watson 			*dosignalp = 1;	/* Defer until unlocked. */
990d4b5cae4SRobert Watson 		}
991d4b5cae4SRobert Watson 		npwp->nw_queued++;
992d4b5cae4SRobert Watson 		return (0);
993d4b5cae4SRobert Watson 	} else {
994ba3b25b3SBjoern A. Zeeb 		m_freem(m);
995d4b5cae4SRobert Watson 		npwp->nw_qdrops++;
996d4b5cae4SRobert Watson 		return (ENOBUFS);
997d4b5cae4SRobert Watson 	}
998d4b5cae4SRobert Watson }
999d4b5cae4SRobert Watson 
1000d4b5cae4SRobert Watson static int
1001d4b5cae4SRobert Watson netisr_queue_internal(u_int proto, struct mbuf *m, u_int cpuid)
1002d4b5cae4SRobert Watson {
1003d4b5cae4SRobert Watson 	struct netisr_workstream *nwsp;
1004d4b5cae4SRobert Watson 	struct netisr_work *npwp;
1005d4b5cae4SRobert Watson 	int dosignal, error;
1006d4b5cae4SRobert Watson 
1007d4b5cae4SRobert Watson #ifdef NETISR_LOCKING
1008d4b5cae4SRobert Watson 	NETISR_LOCK_ASSERT();
1009d4b5cae4SRobert Watson #endif
10109e6e01ebSRobert Watson 	KASSERT(cpuid <= mp_maxid, ("%s: cpuid too big (%u, %u)", __func__,
10119e6e01ebSRobert Watson 	    cpuid, mp_maxid));
101253402767SRobert Watson 	KASSERT(!CPU_ABSENT(cpuid), ("%s: CPU %u absent", __func__, cpuid));
1013d4b5cae4SRobert Watson 
1014d4b5cae4SRobert Watson 	dosignal = 0;
1015d4b5cae4SRobert Watson 	error = 0;
101653402767SRobert Watson 	nwsp = DPCPU_ID_PTR(cpuid, nws);
1017d4b5cae4SRobert Watson 	npwp = &nwsp->nws_work[proto];
1018d4b5cae4SRobert Watson 	NWS_LOCK(nwsp);
1019d4b5cae4SRobert Watson 	error = netisr_queue_workstream(nwsp, proto, npwp, m, &dosignal);
1020d4b5cae4SRobert Watson 	NWS_UNLOCK(nwsp);
1021d4b5cae4SRobert Watson 	if (dosignal)
1022d4b5cae4SRobert Watson 		NWS_SIGNAL(nwsp);
1023d4b5cae4SRobert Watson 	return (error);
1024d4b5cae4SRobert Watson }
1025d4b5cae4SRobert Watson 
1026d4b5cae4SRobert Watson int
1027d4b5cae4SRobert Watson netisr_queue_src(u_int proto, uintptr_t source, struct mbuf *m)
1028d4b5cae4SRobert Watson {
1029d4b5cae4SRobert Watson #ifdef NETISR_LOCKING
1030d4b5cae4SRobert Watson 	struct rm_priotracker tracker;
1031d4b5cae4SRobert Watson #endif
1032d4b5cae4SRobert Watson 	u_int cpuid;
1033d4b5cae4SRobert Watson 	int error;
1034d4b5cae4SRobert Watson 
1035d4b5cae4SRobert Watson 	KASSERT(proto < NETISR_MAXPROT,
1036d4b5cae4SRobert Watson 	    ("%s: invalid proto %u", __func__, proto));
1037d4b5cae4SRobert Watson 
1038d4b5cae4SRobert Watson #ifdef NETISR_LOCKING
1039d4b5cae4SRobert Watson 	NETISR_RLOCK(&tracker);
1040d4b5cae4SRobert Watson #endif
1041938448cdSRobert Watson 	KASSERT(netisr_proto[proto].np_handler != NULL,
1042d4b5cae4SRobert Watson 	    ("%s: invalid proto %u", __func__, proto));
1043d4b5cae4SRobert Watson 
1044484149deSBjoern A. Zeeb #ifdef VIMAGE
1045484149deSBjoern A. Zeeb 	if (V_netisr_enable[proto] == 0) {
1046484149deSBjoern A. Zeeb 		m_freem(m);
1047484149deSBjoern A. Zeeb 		return (ENOPROTOOPT);
1048484149deSBjoern A. Zeeb 	}
1049484149deSBjoern A. Zeeb #endif
1050484149deSBjoern A. Zeeb 
1051f2d2d694SRobert Watson 	m = netisr_select_cpuid(&netisr_proto[proto], NETISR_DISPATCH_DEFERRED,
1052f2d2d694SRobert Watson 	    source, m, &cpuid);
105353402767SRobert Watson 	if (m != NULL) {
105453402767SRobert Watson 		KASSERT(!CPU_ABSENT(cpuid), ("%s: CPU %u absent", __func__,
105553402767SRobert Watson 		    cpuid));
1056d4b5cae4SRobert Watson 		error = netisr_queue_internal(proto, m, cpuid);
105753402767SRobert Watson 	} else
1058d4b5cae4SRobert Watson 		error = ENOBUFS;
1059d4b5cae4SRobert Watson #ifdef NETISR_LOCKING
1060d4b5cae4SRobert Watson 	NETISR_RUNLOCK(&tracker);
1061d4b5cae4SRobert Watson #endif
1062d4b5cae4SRobert Watson 	return (error);
1063d4b5cae4SRobert Watson }
1064d4b5cae4SRobert Watson 
1065d4b5cae4SRobert Watson int
1066d4b5cae4SRobert Watson netisr_queue(u_int proto, struct mbuf *m)
1067d4b5cae4SRobert Watson {
1068d4b5cae4SRobert Watson 
1069d4b5cae4SRobert Watson 	return (netisr_queue_src(proto, 0, m));
1070d4b5cae4SRobert Watson }
1071d4b5cae4SRobert Watson 
1072d4b5cae4SRobert Watson /*
10730a32e29fSRobert Watson  * Dispatch a packet for netisr processing; direct dispatch is permitted by
1074d4b5cae4SRobert Watson  * calling context.
1075d4b5cae4SRobert Watson  */
1076d4b5cae4SRobert Watson int
1077d4b5cae4SRobert Watson netisr_dispatch_src(u_int proto, uintptr_t source, struct mbuf *m)
1078d4b5cae4SRobert Watson {
1079d4b5cae4SRobert Watson #ifdef NETISR_LOCKING
1080d4b5cae4SRobert Watson 	struct rm_priotracker tracker;
1081d4b5cae4SRobert Watson #endif
1082d4b5cae4SRobert Watson 	struct netisr_workstream *nwsp;
1083f2d2d694SRobert Watson 	struct netisr_proto *npp;
1084d4b5cae4SRobert Watson 	struct netisr_work *npwp;
1085d4b5cae4SRobert Watson 	int dosignal, error;
1086f2d2d694SRobert Watson 	u_int cpuid, dispatch_policy;
1087d4b5cae4SRobert Watson 
1088d4b5cae4SRobert Watson 	KASSERT(proto < NETISR_MAXPROT,
1089d4b5cae4SRobert Watson 	    ("%s: invalid proto %u", __func__, proto));
1090d4b5cae4SRobert Watson #ifdef NETISR_LOCKING
1091d4b5cae4SRobert Watson 	NETISR_RLOCK(&tracker);
1092d4b5cae4SRobert Watson #endif
1093f2d2d694SRobert Watson 	npp = &netisr_proto[proto];
1094f2d2d694SRobert Watson 	KASSERT(npp->np_handler != NULL, ("%s: invalid proto %u", __func__,
1095f2d2d694SRobert Watson 	    proto));
1096f2d2d694SRobert Watson 
1097484149deSBjoern A. Zeeb #ifdef VIMAGE
1098484149deSBjoern A. Zeeb 	if (V_netisr_enable[proto] == 0) {
1099484149deSBjoern A. Zeeb 		m_freem(m);
1100484149deSBjoern A. Zeeb 		return (ENOPROTOOPT);
1101484149deSBjoern A. Zeeb 	}
1102484149deSBjoern A. Zeeb #endif
1103484149deSBjoern A. Zeeb 
1104f2d2d694SRobert Watson 	dispatch_policy = netisr_get_dispatch(npp);
1105f2d2d694SRobert Watson 	if (dispatch_policy == NETISR_DISPATCH_DEFERRED)
1106f2d2d694SRobert Watson 		return (netisr_queue_src(proto, source, m));
1107d4b5cae4SRobert Watson 
1108d4b5cae4SRobert Watson 	/*
1109d4b5cae4SRobert Watson 	 * If direct dispatch is forced, then unconditionally dispatch
1110d4b5cae4SRobert Watson 	 * without a formal CPU selection.  Borrow the current CPU's stats,
1111d4b5cae4SRobert Watson 	 * even if there's no worker on it.  In this case we don't update
1112d4b5cae4SRobert Watson 	 * nws_flags because all netisr processing will be source ordered due
1113d4b5cae4SRobert Watson 	 * to always being forced to directly dispatch.
1114d4b5cae4SRobert Watson 	 */
1115f2d2d694SRobert Watson 	if (dispatch_policy == NETISR_DISPATCH_DIRECT) {
111653402767SRobert Watson 		nwsp = DPCPU_PTR(nws);
1117d4b5cae4SRobert Watson 		npwp = &nwsp->nws_work[proto];
1118d4b5cae4SRobert Watson 		npwp->nw_dispatched++;
1119d4b5cae4SRobert Watson 		npwp->nw_handled++;
1120938448cdSRobert Watson 		netisr_proto[proto].np_handler(m);
1121d4b5cae4SRobert Watson 		error = 0;
1122d4b5cae4SRobert Watson 		goto out_unlock;
1123d4b5cae4SRobert Watson 	}
1124d4b5cae4SRobert Watson 
1125f2d2d694SRobert Watson 	KASSERT(dispatch_policy == NETISR_DISPATCH_HYBRID,
1126f2d2d694SRobert Watson 	    ("%s: unknown dispatch policy (%u)", __func__, dispatch_policy));
1127f2d2d694SRobert Watson 
1128d4b5cae4SRobert Watson 	/*
1129d4b5cae4SRobert Watson 	 * Otherwise, we execute in a hybrid mode where we will try to direct
1130d4b5cae4SRobert Watson 	 * dispatch if we're on the right CPU and the netisr worker isn't
1131d4b5cae4SRobert Watson 	 * already running.
1132d4b5cae4SRobert Watson 	 */
1133f2d2d694SRobert Watson 	sched_pin();
1134f2d2d694SRobert Watson 	m = netisr_select_cpuid(&netisr_proto[proto], NETISR_DISPATCH_HYBRID,
1135f2d2d694SRobert Watson 	    source, m, &cpuid);
1136d4b5cae4SRobert Watson 	if (m == NULL) {
1137d4b5cae4SRobert Watson 		error = ENOBUFS;
1138f2d2d694SRobert Watson 		goto out_unpin;
1139d4b5cae4SRobert Watson 	}
114053402767SRobert Watson 	KASSERT(!CPU_ABSENT(cpuid), ("%s: CPU %u absent", __func__, cpuid));
1141d4b5cae4SRobert Watson 	if (cpuid != curcpu)
1142d4b5cae4SRobert Watson 		goto queue_fallback;
114353402767SRobert Watson 	nwsp = DPCPU_PTR(nws);
1144d4b5cae4SRobert Watson 	npwp = &nwsp->nws_work[proto];
1145d4b5cae4SRobert Watson 
1146d4b5cae4SRobert Watson 	/*-
1147d4b5cae4SRobert Watson 	 * We are willing to direct dispatch only if three conditions hold:
1148d4b5cae4SRobert Watson 	 *
1149d4b5cae4SRobert Watson 	 * (1) The netisr worker isn't already running,
1150d4b5cae4SRobert Watson 	 * (2) Another thread isn't already directly dispatching, and
1151d4b5cae4SRobert Watson 	 * (3) The netisr hasn't already been woken up.
1152d4b5cae4SRobert Watson 	 */
1153d4b5cae4SRobert Watson 	NWS_LOCK(nwsp);
1154d4b5cae4SRobert Watson 	if (nwsp->nws_flags & (NWS_RUNNING | NWS_DISPATCHING | NWS_SCHEDULED)) {
1155d4b5cae4SRobert Watson 		error = netisr_queue_workstream(nwsp, proto, npwp, m,
1156d4b5cae4SRobert Watson 		    &dosignal);
115753402767SRobert Watson 		NWS_UNLOCK(nwsp);
1158d4b5cae4SRobert Watson 		if (dosignal)
1159d4b5cae4SRobert Watson 			NWS_SIGNAL(nwsp);
1160d4b5cae4SRobert Watson 		goto out_unpin;
1161d4b5cae4SRobert Watson 	}
1162d4b5cae4SRobert Watson 
1163d4b5cae4SRobert Watson 	/*
1164d4b5cae4SRobert Watson 	 * The current thread is now effectively the netisr worker, so set
1165d4b5cae4SRobert Watson 	 * the dispatching flag to prevent concurrent processing of the
1166d4b5cae4SRobert Watson 	 * stream from another thread (even the netisr worker), which could
1167d4b5cae4SRobert Watson 	 * otherwise lead to effective misordering of the stream.
1168d4b5cae4SRobert Watson 	 */
1169d4b5cae4SRobert Watson 	nwsp->nws_flags |= NWS_DISPATCHING;
1170d4b5cae4SRobert Watson 	NWS_UNLOCK(nwsp);
1171938448cdSRobert Watson 	netisr_proto[proto].np_handler(m);
1172d4b5cae4SRobert Watson 	NWS_LOCK(nwsp);
1173d4b5cae4SRobert Watson 	nwsp->nws_flags &= ~NWS_DISPATCHING;
1174d4b5cae4SRobert Watson 	npwp->nw_handled++;
1175d4b5cae4SRobert Watson 	npwp->nw_hybrid_dispatched++;
1176d4b5cae4SRobert Watson 
1177d4b5cae4SRobert Watson 	/*
1178d4b5cae4SRobert Watson 	 * If other work was enqueued by another thread while we were direct
1179d4b5cae4SRobert Watson 	 * dispatching, we need to signal the netisr worker to do that work.
1180d4b5cae4SRobert Watson 	 * In the future, we might want to do some of that work in the
1181d4b5cae4SRobert Watson 	 * current thread, rather than trigger further context switches.  If
1182d4b5cae4SRobert Watson 	 * so, we'll want to establish a reasonable bound on the work done in
1183d4b5cae4SRobert Watson 	 * the "borrowed" context.
1184d4b5cae4SRobert Watson 	 */
1185d4b5cae4SRobert Watson 	if (nwsp->nws_pendingbits != 0) {
1186d4b5cae4SRobert Watson 		nwsp->nws_flags |= NWS_SCHEDULED;
1187d4b5cae4SRobert Watson 		dosignal = 1;
1188d4b5cae4SRobert Watson 	} else
1189d4b5cae4SRobert Watson 		dosignal = 0;
1190d4b5cae4SRobert Watson 	NWS_UNLOCK(nwsp);
1191d4b5cae4SRobert Watson 	if (dosignal)
1192d4b5cae4SRobert Watson 		NWS_SIGNAL(nwsp);
1193d4b5cae4SRobert Watson 	error = 0;
1194d4b5cae4SRobert Watson 	goto out_unpin;
1195d4b5cae4SRobert Watson 
1196d4b5cae4SRobert Watson queue_fallback:
1197d4b5cae4SRobert Watson 	error = netisr_queue_internal(proto, m, cpuid);
1198d4b5cae4SRobert Watson out_unpin:
1199d4b5cae4SRobert Watson 	sched_unpin();
1200d4b5cae4SRobert Watson out_unlock:
1201d4b5cae4SRobert Watson #ifdef NETISR_LOCKING
1202d4b5cae4SRobert Watson 	NETISR_RUNLOCK(&tracker);
1203d4b5cae4SRobert Watson #endif
1204d4b5cae4SRobert Watson 	return (error);
1205d4b5cae4SRobert Watson }
1206d4b5cae4SRobert Watson 
1207d4b5cae4SRobert Watson int
1208d4b5cae4SRobert Watson netisr_dispatch(u_int proto, struct mbuf *m)
1209d4b5cae4SRobert Watson {
1210d4b5cae4SRobert Watson 
1211d4b5cae4SRobert Watson 	return (netisr_dispatch_src(proto, 0, m));
1212d4b5cae4SRobert Watson }
1213d4b5cae4SRobert Watson 
1214d4b5cae4SRobert Watson #ifdef DEVICE_POLLING
1215d4b5cae4SRobert Watson /*
1216d4b5cae4SRobert Watson  * Kernel polling borrows a netisr thread to run interface polling in; this
1217d4b5cae4SRobert Watson  * function allows kernel polling to request that the netisr thread be
1218d4b5cae4SRobert Watson  * scheduled even if no packets are pending for protocols.
1219d4b5cae4SRobert Watson  */
1220d4b5cae4SRobert Watson void
1221d4b5cae4SRobert Watson netisr_sched_poll(void)
1222d4b5cae4SRobert Watson {
1223d4b5cae4SRobert Watson 	struct netisr_workstream *nwsp;
1224d4b5cae4SRobert Watson 
122553402767SRobert Watson 	nwsp = DPCPU_ID_PTR(nws_array[0], nws);
1226d4b5cae4SRobert Watson 	NWS_SIGNAL(nwsp);
1227d4b5cae4SRobert Watson }
12281cafed39SJonathan Lemon #endif
1229e3b6e33cSJake Burkholder 
1230d4b5cae4SRobert Watson static void
1231d4b5cae4SRobert Watson netisr_start_swi(u_int cpuid, struct pcpu *pc)
1232d4b5cae4SRobert Watson {
1233d4b5cae4SRobert Watson 	char swiname[12];
1234d4b5cae4SRobert Watson 	struct netisr_workstream *nwsp;
1235d4b5cae4SRobert Watson 	int error;
1236d4b5cae4SRobert Watson 
123753402767SRobert Watson 	KASSERT(!CPU_ABSENT(cpuid), ("%s: CPU %u absent", __func__, cpuid));
123853402767SRobert Watson 
123953402767SRobert Watson 	nwsp = DPCPU_ID_PTR(cpuid, nws);
1240d4b5cae4SRobert Watson 	mtx_init(&nwsp->nws_mtx, "netisr_mtx", NULL, MTX_DEF);
1241d4b5cae4SRobert Watson 	nwsp->nws_cpu = cpuid;
1242d4b5cae4SRobert Watson 	snprintf(swiname, sizeof(swiname), "netisr %u", cpuid);
1243d4b5cae4SRobert Watson 	error = swi_add(&nwsp->nws_intr_event, swiname, swi_net, nwsp,
1244d4b5cae4SRobert Watson 	    SWI_NET, INTR_MPSAFE, &nwsp->nws_swi_cookie);
1245d4b5cae4SRobert Watson 	if (error)
1246d4b5cae4SRobert Watson 		panic("%s: swi_add %d", __func__, error);
1247d4b5cae4SRobert Watson 	pc->pc_netisr = nwsp->nws_intr_event;
1248d4b5cae4SRobert Watson 	if (netisr_bindthreads) {
1249d4b5cae4SRobert Watson 		error = intr_event_bind(nwsp->nws_intr_event, cpuid);
1250d4b5cae4SRobert Watson 		if (error != 0)
1251d4b5cae4SRobert Watson 			printf("%s: cpu %u: intr_event_bind: %d", __func__,
1252d4b5cae4SRobert Watson 			    cpuid, error);
1253e3b6e33cSJake Burkholder 	}
1254d4b5cae4SRobert Watson 	NETISR_WLOCK();
1255d4b5cae4SRobert Watson 	nws_array[nws_count] = nwsp->nws_cpu;
1256d4b5cae4SRobert Watson 	nws_count++;
1257d4b5cae4SRobert Watson 	NETISR_WUNLOCK();
1258e3b6e33cSJake Burkholder }
1259e3b6e33cSJake Burkholder 
1260d4b5cae4SRobert Watson /*
1261d4b5cae4SRobert Watson  * Initialize the netisr subsystem.  We rely on BSS and static initialization
1262d4b5cae4SRobert Watson  * of most fields in global data structures.
1263d4b5cae4SRobert Watson  *
1264d4b5cae4SRobert Watson  * Start a worker thread for the boot CPU so that we can support network
1265d4b5cae4SRobert Watson  * traffic immediately in case the network stack is used before additional
1266d4b5cae4SRobert Watson  * CPUs are started (for example, diskless boot).
1267d4b5cae4SRobert Watson  */
1268e3b6e33cSJake Burkholder static void
1269d4b5cae4SRobert Watson netisr_init(void *arg)
1270e3b6e33cSJake Burkholder {
1271fdce57a0SJohn Baldwin #ifdef EARLY_AP_STARTUP
1272fdce57a0SJohn Baldwin 	struct pcpu *pc;
1273fdce57a0SJohn Baldwin #endif
1274fdce57a0SJohn Baldwin 
1275d4b5cae4SRobert Watson 	NETISR_LOCK_INIT();
1276a9467c3cSHiren Panchasara 	if (netisr_maxthreads == 0 || netisr_maxthreads < -1 )
1277a9467c3cSHiren Panchasara 		netisr_maxthreads = 1;		/* default behavior */
1278a9467c3cSHiren Panchasara 	else if (netisr_maxthreads == -1)
1279a9467c3cSHiren Panchasara 		netisr_maxthreads = mp_ncpus;	/* use max cpus */
12809e6e01ebSRobert Watson 	if (netisr_maxthreads > mp_ncpus) {
1281912f6323SRobert Watson 		printf("netisr_init: forcing maxthreads from %d to %d\n",
12829e6e01ebSRobert Watson 		    netisr_maxthreads, mp_ncpus);
12839e6e01ebSRobert Watson 		netisr_maxthreads = mp_ncpus;
1284ed54411cSRobert Watson 	}
1285ed54411cSRobert Watson 	if (netisr_defaultqlimit > netisr_maxqlimit) {
1286912f6323SRobert Watson 		printf("netisr_init: forcing defaultqlimit from %d to %d\n",
12879e6e01ebSRobert Watson 		    netisr_defaultqlimit, netisr_maxqlimit);
1288d4b5cae4SRobert Watson 		netisr_defaultqlimit = netisr_maxqlimit;
1289ed54411cSRobert Watson 	}
1290d4b5cae4SRobert Watson #ifdef DEVICE_POLLING
1291d4b5cae4SRobert Watson 	/*
1292d4b5cae4SRobert Watson 	 * The device polling code is not yet aware of how to deal with
1293d4b5cae4SRobert Watson 	 * multiple netisr threads, so for the time being compiling in device
1294d4b5cae4SRobert Watson 	 * polling disables parallel netisr workers.
1295d4b5cae4SRobert Watson 	 */
129651d4054eSGeorge V. Neville-Neil 	if (netisr_maxthreads != 1 || netisr_bindthreads != 0) {
1297912f6323SRobert Watson 		printf("netisr_init: forcing maxthreads to 1 and "
1298912f6323SRobert Watson 		    "bindthreads to 0 for device polling\n");
1299d4b5cae4SRobert Watson 		netisr_maxthreads = 1;
1300d4b5cae4SRobert Watson 		netisr_bindthreads = 0;
1301ed54411cSRobert Watson 	}
1302d4b5cae4SRobert Watson #endif
1303fdce57a0SJohn Baldwin 
1304fdce57a0SJohn Baldwin #ifdef EARLY_AP_STARTUP
1305fdce57a0SJohn Baldwin 	STAILQ_FOREACH(pc, &cpuhead, pc_allcpu) {
1306fdce57a0SJohn Baldwin 		if (nws_count >= netisr_maxthreads)
1307fdce57a0SJohn Baldwin 			break;
1308fdce57a0SJohn Baldwin 		netisr_start_swi(pc->pc_cpuid, pc);
1309fdce57a0SJohn Baldwin 	}
1310fdce57a0SJohn Baldwin #else
1311d4b5cae4SRobert Watson 	netisr_start_swi(curcpu, pcpu_find(curcpu));
1312fdce57a0SJohn Baldwin #endif
1313e3b6e33cSJake Burkholder }
1314d4b5cae4SRobert Watson SYSINIT(netisr_init, SI_SUB_SOFTINTR, SI_ORDER_FIRST, netisr_init, NULL);
1315d4b5cae4SRobert Watson 
1316fdce57a0SJohn Baldwin #ifndef EARLY_AP_STARTUP
1317d4b5cae4SRobert Watson /*
1318d4b5cae4SRobert Watson  * Start worker threads for additional CPUs.  No attempt to gracefully handle
1319d4b5cae4SRobert Watson  * work reassignment, we don't yet support dynamic reconfiguration.
1320d4b5cae4SRobert Watson  */
1321d4b5cae4SRobert Watson static void
1322d4b5cae4SRobert Watson netisr_start(void *arg)
1323d4b5cae4SRobert Watson {
1324d4b5cae4SRobert Watson 	struct pcpu *pc;
1325d4b5cae4SRobert Watson 
1326d098f930SNathan Whitehorn 	STAILQ_FOREACH(pc, &cpuhead, pc_allcpu) {
1327d4b5cae4SRobert Watson 		if (nws_count >= netisr_maxthreads)
1328d4b5cae4SRobert Watson 			break;
1329d4b5cae4SRobert Watson 		/* Worker will already be present for boot CPU. */
1330d4b5cae4SRobert Watson 		if (pc->pc_netisr != NULL)
1331d4b5cae4SRobert Watson 			continue;
1332d4b5cae4SRobert Watson 		netisr_start_swi(pc->pc_cpuid, pc);
1333d4b5cae4SRobert Watson 	}
1334d4b5cae4SRobert Watson }
1335d4b5cae4SRobert Watson SYSINIT(netisr_start, SI_SUB_SMP, SI_ORDER_MIDDLE, netisr_start, NULL);
1336fdce57a0SJohn Baldwin #endif
1337d4b5cae4SRobert Watson 
13382d22f334SRobert Watson /*
13392d22f334SRobert Watson  * Sysctl monitoring for netisr: query a list of registered protocols.
13402d22f334SRobert Watson  */
13412d22f334SRobert Watson static int
13422d22f334SRobert Watson sysctl_netisr_proto(SYSCTL_HANDLER_ARGS)
13432d22f334SRobert Watson {
13442d22f334SRobert Watson 	struct rm_priotracker tracker;
13452d22f334SRobert Watson 	struct sysctl_netisr_proto *snpp, *snp_array;
13462d22f334SRobert Watson 	struct netisr_proto *npp;
13472d22f334SRobert Watson 	u_int counter, proto;
13482d22f334SRobert Watson 	int error;
13492d22f334SRobert Watson 
13502d22f334SRobert Watson 	if (req->newptr != NULL)
13512d22f334SRobert Watson 		return (EINVAL);
13522d22f334SRobert Watson 	snp_array = malloc(sizeof(*snp_array) * NETISR_MAXPROT, M_TEMP,
13532d22f334SRobert Watson 	    M_ZERO | M_WAITOK);
13542d22f334SRobert Watson 	counter = 0;
13552d22f334SRobert Watson 	NETISR_RLOCK(&tracker);
13562d22f334SRobert Watson 	for (proto = 0; proto < NETISR_MAXPROT; proto++) {
1357938448cdSRobert Watson 		npp = &netisr_proto[proto];
13582d22f334SRobert Watson 		if (npp->np_name == NULL)
13592d22f334SRobert Watson 			continue;
13602d22f334SRobert Watson 		snpp = &snp_array[counter];
13612d22f334SRobert Watson 		snpp->snp_version = sizeof(*snpp);
13622d22f334SRobert Watson 		strlcpy(snpp->snp_name, npp->np_name, NETISR_NAMEMAXLEN);
13632d22f334SRobert Watson 		snpp->snp_proto = proto;
13642d22f334SRobert Watson 		snpp->snp_qlimit = npp->np_qlimit;
13652d22f334SRobert Watson 		snpp->snp_policy = npp->np_policy;
1366f2d2d694SRobert Watson 		snpp->snp_dispatch = npp->np_dispatch;
13672d22f334SRobert Watson 		if (npp->np_m2flow != NULL)
13682d22f334SRobert Watson 			snpp->snp_flags |= NETISR_SNP_FLAGS_M2FLOW;
13692d22f334SRobert Watson 		if (npp->np_m2cpuid != NULL)
13702d22f334SRobert Watson 			snpp->snp_flags |= NETISR_SNP_FLAGS_M2CPUID;
13712d22f334SRobert Watson 		if (npp->np_drainedcpu != NULL)
13722d22f334SRobert Watson 			snpp->snp_flags |= NETISR_SNP_FLAGS_DRAINEDCPU;
13732d22f334SRobert Watson 		counter++;
13742d22f334SRobert Watson 	}
13752d22f334SRobert Watson 	NETISR_RUNLOCK(&tracker);
13767f450febSRobert Watson 	KASSERT(counter <= NETISR_MAXPROT,
13772d22f334SRobert Watson 	    ("sysctl_netisr_proto: counter too big (%d)", counter));
13782d22f334SRobert Watson 	error = SYSCTL_OUT(req, snp_array, sizeof(*snp_array) * counter);
13792d22f334SRobert Watson 	free(snp_array, M_TEMP);
13802d22f334SRobert Watson 	return (error);
13812d22f334SRobert Watson }
13822d22f334SRobert Watson 
13832d22f334SRobert Watson SYSCTL_PROC(_net_isr, OID_AUTO, proto,
13842d22f334SRobert Watson     CTLFLAG_RD|CTLTYPE_STRUCT|CTLFLAG_MPSAFE, 0, 0, sysctl_netisr_proto,
13852d22f334SRobert Watson     "S,sysctl_netisr_proto",
13862d22f334SRobert Watson     "Return list of protocols registered with netisr");
13872d22f334SRobert Watson 
13882d22f334SRobert Watson /*
13892d22f334SRobert Watson  * Sysctl monitoring for netisr: query a list of workstreams.
13902d22f334SRobert Watson  */
13912d22f334SRobert Watson static int
13922d22f334SRobert Watson sysctl_netisr_workstream(SYSCTL_HANDLER_ARGS)
13932d22f334SRobert Watson {
13942d22f334SRobert Watson 	struct rm_priotracker tracker;
13952d22f334SRobert Watson 	struct sysctl_netisr_workstream *snwsp, *snws_array;
13962d22f334SRobert Watson 	struct netisr_workstream *nwsp;
13972d22f334SRobert Watson 	u_int counter, cpuid;
13982d22f334SRobert Watson 	int error;
13992d22f334SRobert Watson 
14002d22f334SRobert Watson 	if (req->newptr != NULL)
14012d22f334SRobert Watson 		return (EINVAL);
14022d22f334SRobert Watson 	snws_array = malloc(sizeof(*snws_array) * MAXCPU, M_TEMP,
14032d22f334SRobert Watson 	    M_ZERO | M_WAITOK);
14042d22f334SRobert Watson 	counter = 0;
14052d22f334SRobert Watson 	NETISR_RLOCK(&tracker);
14063aa6d94eSJohn Baldwin 	CPU_FOREACH(cpuid) {
14072d22f334SRobert Watson 		nwsp = DPCPU_ID_PTR(cpuid, nws);
14082d22f334SRobert Watson 		if (nwsp->nws_intr_event == NULL)
14092d22f334SRobert Watson 			continue;
14102d22f334SRobert Watson 		NWS_LOCK(nwsp);
14112d22f334SRobert Watson 		snwsp = &snws_array[counter];
14122d22f334SRobert Watson 		snwsp->snws_version = sizeof(*snwsp);
14132d22f334SRobert Watson 
14142d22f334SRobert Watson 		/*
14152d22f334SRobert Watson 		 * For now, we equate workstream IDs and CPU IDs in the
14162d22f334SRobert Watson 		 * kernel, but expose them independently to userspace in case
14172d22f334SRobert Watson 		 * that assumption changes in the future.
14182d22f334SRobert Watson 		 */
14192d22f334SRobert Watson 		snwsp->snws_wsid = cpuid;
14202d22f334SRobert Watson 		snwsp->snws_cpu = cpuid;
14212d22f334SRobert Watson 		if (nwsp->nws_intr_event != NULL)
14222d22f334SRobert Watson 			snwsp->snws_flags |= NETISR_SNWS_FLAGS_INTR;
14232d22f334SRobert Watson 		NWS_UNLOCK(nwsp);
14242d22f334SRobert Watson 		counter++;
14252d22f334SRobert Watson 	}
14262d22f334SRobert Watson 	NETISR_RUNLOCK(&tracker);
14277f450febSRobert Watson 	KASSERT(counter <= MAXCPU,
14282d22f334SRobert Watson 	    ("sysctl_netisr_workstream: counter too big (%d)", counter));
14292d22f334SRobert Watson 	error = SYSCTL_OUT(req, snws_array, sizeof(*snws_array) * counter);
14302d22f334SRobert Watson 	free(snws_array, M_TEMP);
14312d22f334SRobert Watson 	return (error);
14322d22f334SRobert Watson }
14332d22f334SRobert Watson 
14342d22f334SRobert Watson SYSCTL_PROC(_net_isr, OID_AUTO, workstream,
14352d22f334SRobert Watson     CTLFLAG_RD|CTLTYPE_STRUCT|CTLFLAG_MPSAFE, 0, 0, sysctl_netisr_workstream,
14362d22f334SRobert Watson     "S,sysctl_netisr_workstream",
14372d22f334SRobert Watson     "Return list of workstreams implemented by netisr");
14382d22f334SRobert Watson 
14392d22f334SRobert Watson /*
14402d22f334SRobert Watson  * Sysctl monitoring for netisr: query per-protocol data across all
14412d22f334SRobert Watson  * workstreams.
14422d22f334SRobert Watson  */
14432d22f334SRobert Watson static int
14442d22f334SRobert Watson sysctl_netisr_work(SYSCTL_HANDLER_ARGS)
14452d22f334SRobert Watson {
14462d22f334SRobert Watson 	struct rm_priotracker tracker;
14472d22f334SRobert Watson 	struct sysctl_netisr_work *snwp, *snw_array;
14482d22f334SRobert Watson 	struct netisr_workstream *nwsp;
14492d22f334SRobert Watson 	struct netisr_proto *npp;
14502d22f334SRobert Watson 	struct netisr_work *nwp;
14512d22f334SRobert Watson 	u_int counter, cpuid, proto;
14522d22f334SRobert Watson 	int error;
14532d22f334SRobert Watson 
14542d22f334SRobert Watson 	if (req->newptr != NULL)
14552d22f334SRobert Watson 		return (EINVAL);
14562d22f334SRobert Watson 	snw_array = malloc(sizeof(*snw_array) * MAXCPU * NETISR_MAXPROT,
14572d22f334SRobert Watson 	    M_TEMP, M_ZERO | M_WAITOK);
14582d22f334SRobert Watson 	counter = 0;
14592d22f334SRobert Watson 	NETISR_RLOCK(&tracker);
14603aa6d94eSJohn Baldwin 	CPU_FOREACH(cpuid) {
14612d22f334SRobert Watson 		nwsp = DPCPU_ID_PTR(cpuid, nws);
14622d22f334SRobert Watson 		if (nwsp->nws_intr_event == NULL)
14632d22f334SRobert Watson 			continue;
14642d22f334SRobert Watson 		NWS_LOCK(nwsp);
14652d22f334SRobert Watson 		for (proto = 0; proto < NETISR_MAXPROT; proto++) {
1466938448cdSRobert Watson 			npp = &netisr_proto[proto];
14672d22f334SRobert Watson 			if (npp->np_name == NULL)
14682d22f334SRobert Watson 				continue;
14692d22f334SRobert Watson 			nwp = &nwsp->nws_work[proto];
14702d22f334SRobert Watson 			snwp = &snw_array[counter];
14712d22f334SRobert Watson 			snwp->snw_version = sizeof(*snwp);
14722d22f334SRobert Watson 			snwp->snw_wsid = cpuid;		/* See comment above. */
14732d22f334SRobert Watson 			snwp->snw_proto = proto;
14742d22f334SRobert Watson 			snwp->snw_len = nwp->nw_len;
14752d22f334SRobert Watson 			snwp->snw_watermark = nwp->nw_watermark;
14762d22f334SRobert Watson 			snwp->snw_dispatched = nwp->nw_dispatched;
14772d22f334SRobert Watson 			snwp->snw_hybrid_dispatched =
14782d22f334SRobert Watson 			    nwp->nw_hybrid_dispatched;
14792d22f334SRobert Watson 			snwp->snw_qdrops = nwp->nw_qdrops;
14802d22f334SRobert Watson 			snwp->snw_queued = nwp->nw_queued;
14812d22f334SRobert Watson 			snwp->snw_handled = nwp->nw_handled;
14822d22f334SRobert Watson 			counter++;
14832d22f334SRobert Watson 		}
14842d22f334SRobert Watson 		NWS_UNLOCK(nwsp);
14852d22f334SRobert Watson 	}
14867f450febSRobert Watson 	KASSERT(counter <= MAXCPU * NETISR_MAXPROT,
14872d22f334SRobert Watson 	    ("sysctl_netisr_work: counter too big (%d)", counter));
14882d22f334SRobert Watson 	NETISR_RUNLOCK(&tracker);
14892d22f334SRobert Watson 	error = SYSCTL_OUT(req, snw_array, sizeof(*snw_array) * counter);
14902d22f334SRobert Watson 	free(snw_array, M_TEMP);
14912d22f334SRobert Watson 	return (error);
14922d22f334SRobert Watson }
14932d22f334SRobert Watson 
14942d22f334SRobert Watson SYSCTL_PROC(_net_isr, OID_AUTO, work,
14952d22f334SRobert Watson     CTLFLAG_RD|CTLTYPE_STRUCT|CTLFLAG_MPSAFE, 0, 0, sysctl_netisr_work,
14962d22f334SRobert Watson     "S,sysctl_netisr_work",
14972d22f334SRobert Watson     "Return list of per-workstream, per-protocol work in netisr");
14982d22f334SRobert Watson 
1499d4b5cae4SRobert Watson #ifdef DDB
1500d4b5cae4SRobert Watson DB_SHOW_COMMAND(netisr, db_show_netisr)
1501d4b5cae4SRobert Watson {
1502d4b5cae4SRobert Watson 	struct netisr_workstream *nwsp;
1503d4b5cae4SRobert Watson 	struct netisr_work *nwp;
1504d4b5cae4SRobert Watson 	int first, proto;
150553402767SRobert Watson 	u_int cpuid;
1506d4b5cae4SRobert Watson 
1507d4b5cae4SRobert Watson 	db_printf("%3s %6s %5s %5s %5s %8s %8s %8s %8s\n", "CPU", "Proto",
1508d4b5cae4SRobert Watson 	    "Len", "WMark", "Max", "Disp", "HDisp", "Drop", "Queue");
15093aa6d94eSJohn Baldwin 	CPU_FOREACH(cpuid) {
151053402767SRobert Watson 		nwsp = DPCPU_ID_PTR(cpuid, nws);
1511d4b5cae4SRobert Watson 		if (nwsp->nws_intr_event == NULL)
1512d4b5cae4SRobert Watson 			continue;
1513d4b5cae4SRobert Watson 		first = 1;
1514d4b5cae4SRobert Watson 		for (proto = 0; proto < NETISR_MAXPROT; proto++) {
1515938448cdSRobert Watson 			if (netisr_proto[proto].np_handler == NULL)
1516d4b5cae4SRobert Watson 				continue;
1517d4b5cae4SRobert Watson 			nwp = &nwsp->nws_work[proto];
1518d4b5cae4SRobert Watson 			if (first) {
151953402767SRobert Watson 				db_printf("%3d ", cpuid);
1520d4b5cae4SRobert Watson 				first = 0;
1521d4b5cae4SRobert Watson 			} else
1522d4b5cae4SRobert Watson 				db_printf("%3s ", "");
1523d4b5cae4SRobert Watson 			db_printf(
1524d4b5cae4SRobert Watson 			    "%6s %5d %5d %5d %8ju %8ju %8ju %8ju\n",
1525938448cdSRobert Watson 			    netisr_proto[proto].np_name, nwp->nw_len,
1526d4b5cae4SRobert Watson 			    nwp->nw_watermark, nwp->nw_qlimit,
1527d4b5cae4SRobert Watson 			    nwp->nw_dispatched, nwp->nw_hybrid_dispatched,
1528d4b5cae4SRobert Watson 			    nwp->nw_qdrops, nwp->nw_queued);
1529d4b5cae4SRobert Watson 		}
1530d4b5cae4SRobert Watson 	}
1531d4b5cae4SRobert Watson }
1532d4b5cae4SRobert Watson #endif
1533