ng_base.c revision 159373
1173362Sbenjsc/*
2173362Sbenjsc * ng_base.c
3173362Sbenjsc */
4173362Sbenjsc
5173362Sbenjsc/*-
6173362Sbenjsc * Copyright (c) 1996-1999 Whistle Communications, Inc.
7173362Sbenjsc * All rights reserved.
8173362Sbenjsc *
9173362Sbenjsc * Subject to the following obligations and disclaimer of warranty, use and
10173362Sbenjsc * redistribution of this software, in source or object code forms, with or
11173362Sbenjsc * without modifications are expressly permitted by Whistle Communications;
12173362Sbenjsc * provided, however, that:
13173362Sbenjsc * 1. Any and all reproductions of the source or object code must include the
14173362Sbenjsc *    copyright notice above and the following disclaimer of warranties; and
15173362Sbenjsc * 2. No rights are granted, in any manner or form, to use Whistle
16173362Sbenjsc *    Communications, Inc. trademarks, including the mark "WHISTLE
17173362Sbenjsc *    COMMUNICATIONS" on advertising, endorsements, or otherwise except as
18173362Sbenjsc *    such appears in the above copyright notice or in the software.
19173362Sbenjsc *
20173362Sbenjsc * THIS SOFTWARE IS BEING PROVIDED BY WHISTLE COMMUNICATIONS "AS IS", AND
21173362Sbenjsc * TO THE MAXIMUM EXTENT PERMITTED BY LAW, WHISTLE COMMUNICATIONS MAKES NO
22173362Sbenjsc * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING THIS SOFTWARE,
23173362Sbenjsc * INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED WARRANTIES OF
24173362Sbenjsc * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT.
25173362Sbenjsc * WHISTLE COMMUNICATIONS DOES NOT WARRANT, GUARANTEE, OR MAKE ANY
26173362Sbenjsc * REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS OF THE USE OF THIS
27173362Sbenjsc * SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY, RELIABILITY OR OTHERWISE.
28173362Sbenjsc * IN NO EVENT SHALL WHISTLE COMMUNICATIONS BE LIABLE FOR ANY DAMAGES
29345636Savos * RESULTING FROM OR ARISING OUT OF ANY USE OF THIS SOFTWARE, INCLUDING
30173362Sbenjsc * WITHOUT LIMITATION, ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
31173362Sbenjsc * PUNITIVE, OR CONSEQUENTIAL DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR
32173362Sbenjsc * SERVICES, LOSS OF USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY
33173362Sbenjsc * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
34173362Sbenjsc * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
35173362Sbenjsc * THIS SOFTWARE, EVEN IF WHISTLE COMMUNICATIONS IS ADVISED OF THE POSSIBILITY
36173362Sbenjsc * OF SUCH DAMAGE.
37173362Sbenjsc *
38173362Sbenjsc * Authors: Julian Elischer <julian@freebsd.org>
39173362Sbenjsc *          Archie Cobbs <archie@freebsd.org>
40173362Sbenjsc *
41173362Sbenjsc * $FreeBSD: head/sys/netgraph/ng_base.c 159373 2006-06-07 12:42:15Z glebius $
42173362Sbenjsc * $Whistle: ng_base.c,v 1.39 1999/01/28 23:54:53 julian Exp $
43173362Sbenjsc */
44173362Sbenjsc
45173362Sbenjsc/*
46278366Sadrian * This file implements the base netgraph code.
47173362Sbenjsc */
48173362Sbenjsc
49173362Sbenjsc#include <sys/param.h>
50173362Sbenjsc#include <sys/systm.h>
51173362Sbenjsc#include <sys/ctype.h>
52173362Sbenjsc#include <sys/errno.h>
53173362Sbenjsc#include <sys/kdb.h>
54173362Sbenjsc#include <sys/kernel.h>
55280059Sadrian#include <sys/ktr.h>
56278366Sadrian#include <sys/limits.h>
57278366Sadrian#include <sys/malloc.h>
58173362Sbenjsc#include <sys/mbuf.h>
59173362Sbenjsc#include <sys/queue.h>
60173362Sbenjsc#include <sys/sysctl.h>
61173362Sbenjsc#include <sys/syslog.h>
62173362Sbenjsc
63278366Sadrian#include <net/netisr.h>
64173362Sbenjsc
65173362Sbenjsc#include <netgraph/ng_message.h>
66173362Sbenjsc#include <netgraph/netgraph.h>
67173362Sbenjsc#include <netgraph/ng_parse.h>
68173362Sbenjsc
69173362SbenjscMODULE_VERSION(netgraph, NG_ABI_VERSION);
70173362Sbenjsc
71173362Sbenjsc/* List of all active nodes */
72173362Sbenjscstatic LIST_HEAD(, ng_node) ng_nodelist;
73278366Sadrianstatic struct mtx	ng_nodelist_mtx;
74173362Sbenjsc
75289124Sadrian/* Mutex to protect topology events. */
76289124Sadrianstatic struct mtx	ng_topo_mtx;
77289163Sadrian
78289124Sadrian#ifdef	NETGRAPH_DEBUG
79289124Sadrianstatic struct mtx	ngq_mtx;	/* protects the queue item list */
80173362Sbenjsc
81173362Sbenjscstatic SLIST_HEAD(, ng_node) ng_allnodes;
82173362Sbenjscstatic LIST_HEAD(, ng_node) ng_freenodes; /* in debug, we never free() them */
83278366Sadrianstatic SLIST_HEAD(, ng_hook) ng_allhooks;
84278366Sadrianstatic LIST_HEAD(, ng_hook) ng_freehooks; /* in debug, we never free() them */
85173362Sbenjsc
86173362Sbenjscstatic void ng_dumpitems(void);
87173362Sbenjscstatic void ng_dumpnodes(void);
88173362Sbenjscstatic void ng_dumphooks(void);
89173362Sbenjsc
90173362Sbenjsc#endif	/* NETGRAPH_DEBUG */
91177043Sthompsa/*
92289124Sadrian * DEAD versions of the structures.
93278366Sadrian * In order to avoid races, it is sometimes neccesary to point
94173362Sbenjsc * at SOMETHING even though theoretically, the current entity is
95173362Sbenjsc * INVALID. Use these to avoid these races.
96278366Sadrian */
97278366Sadrianstruct ng_type ng_deadtype = {
98278366Sadrian	NG_ABI_VERSION,
99173362Sbenjsc	"dead",
100280063Sadrian	NULL,	/* modevent */
101173362Sbenjsc	NULL,	/* constructor */
102173362Sbenjsc	NULL,	/* rcvmsg */
103173362Sbenjsc	NULL,	/* shutdown */
104173362Sbenjsc	NULL,	/* newhook */
105173362Sbenjsc	NULL,	/* findhook */
106173362Sbenjsc	NULL,	/* connect */
107173362Sbenjsc	NULL,	/* rcvdata */
108173362Sbenjsc	NULL,	/* disconnect */
109178354Ssam	NULL, 	/* cmdlist */
110178354Ssam};
111178354Ssam
112178354Ssamstruct ng_node ng_deadnode = {
113173362Sbenjsc	"dead",
114173362Sbenjsc	&ng_deadtype,
115278366Sadrian	NGF_INVALID,
116280082Sadrian	1,	/* refs */
117278366Sadrian	0,	/* numhooks */
118278366Sadrian	NULL,	/* private */
119278366Sadrian	0,	/* ID */
120289124Sadrian	LIST_HEAD_INITIALIZER(ng_deadnode.hooks),
121289124Sadrian	{},	/* all_nodes list entry */
122278366Sadrian	{},	/* id hashtable list entry */
123278366Sadrian	{},	/* workqueue entry */
124178354Ssam	{	0,
125282401Sadrian		{}, /* should never use! (should hang) */
126178354Ssam		NULL,
127282401Sadrian		&ng_deadnode.nd_input_queue.queue,
128282401Sadrian		&ng_deadnode
129280084Sadrian	},
130289124Sadrian#ifdef	NETGRAPH_DEBUG
131282401Sadrian	ND_MAGIC,
132280088Sadrian	__FILE__,
133282401Sadrian	__LINE__,
134282401Sadrian	{NULL}
135282401Sadrian#endif	/* NETGRAPH_DEBUG */
136283536Sadrian};
137283536Sadrian
138283536Sadrianstruct ng_hook ng_deadhook = {
139178354Ssam	"dead",
140178354Ssam	NULL,		/* private */
141178354Ssam	HK_INVALID | HK_DEAD,
142280086Sadrian	1,		/* refs always >= 1 */
143280086Sadrian	0,		/* undefined data link type */
144280086Sadrian	&ng_deadhook,	/* Peer is self */
145280086Sadrian	&ng_deadnode,	/* attached to deadnode */
146280086Sadrian	{},		/* hooks list */
147280086Sadrian	NULL,		/* override rcvmsg() */
148280086Sadrian	NULL,		/* override rcvdata() */
149280086Sadrian#ifdef	NETGRAPH_DEBUG
150278366Sadrian	HK_MAGIC,
151278366Sadrian	__FILE__,
152278366Sadrian	__LINE__,
153278366Sadrian	{NULL}
154278366Sadrian#endif	/* NETGRAPH_DEBUG */
155278366Sadrian};
156278366Sadrian
157278366Sadrian/*
158278366Sadrian * END DEAD STRUCTURES
159278366Sadrian */
160278366Sadrian/* List nodes with unallocated work */
161278366Sadrianstatic TAILQ_HEAD(, ng_node) ng_worklist = TAILQ_HEAD_INITIALIZER(ng_worklist);
162278366Sadrianstatic struct mtx	ng_worklist_mtx;   /* MUST LOCK NODE FIRST */
163278366Sadrian
164278366Sadrian/* List of installed types */
165173362Sbenjscstatic LIST_HEAD(, ng_type) ng_typelist;
166173362Sbenjscstatic struct mtx	ng_typelist_mtx;
167278366Sadrian
168278366Sadrian/* Hash related definitions */
169287197Sglebius/* XXX Don't need to initialise them because it's a LIST */
170282400Sadrian#define NG_ID_HASH_SIZE 32 /* most systems wont need even this many */
171173362Sbenjscstatic LIST_HEAD(, ng_node) ng_ID_hash[NG_ID_HASH_SIZE];
172287197Sglebiusstatic struct mtx	ng_idhash_mtx;
173287197Sglebius/* Method to find a node.. used twice so do it here */
174280113Sadrian#define NG_IDHASH_FN(ID) ((ID) % (NG_ID_HASH_SIZE))
175173362Sbenjsc#define NG_IDHASH_FIND(ID, node)					\
176278366Sadrian	do { 								\
177173362Sbenjsc		mtx_assert(&ng_idhash_mtx, MA_OWNED);			\
178173362Sbenjsc		LIST_FOREACH(node, &ng_ID_hash[NG_IDHASH_FN(ID)],	\
179173362Sbenjsc						nd_idnodes) {		\
180289126Sadrian			if (NG_NODE_IS_VALID(node)			\
181280071Sadrian			&& (NG_NODE_ID(node) == ID)) {			\
182280112Sadrian				break;					\
183280071Sadrian			}						\
184173362Sbenjsc		}							\
185282401Sadrian	} while (0)
186173362Sbenjsc
187278366Sadrian
188173362Sbenjsc/* Internal functions */
189173362Sbenjscstatic int	ng_add_hook(node_p node, const char *name, hook_p * hookp);
190280106Sadrianstatic int	ng_generic_msg(node_p here, item_p item, hook_p lasthook);
191280109Sadrianstatic ng_ID_t	ng_decodeidname(const char *name);
192280106Sadrianstatic int	ngb_mod_event(module_t mod, int event, void *data);
193280109Sadrianstatic void	ng_worklist_remove(node_p node);
194278366Sadrianstatic void	ngintr(void);
195173362Sbenjscstatic int	ng_apply_item(node_p node, item_p item, int rw);
196278366Sadrianstatic void	ng_flush_input_queue(struct ng_queue * ngq);
197278366Sadrianstatic void	ng_setisr(node_p node);
198278366Sadrianstatic node_p	ng_ID2noderef(ng_ID_t ID);
199278366Sadrianstatic int	ng_con_nodes(node_p node, const char *name, node_p node2,
200173362Sbenjsc							const char *name2);
201173362Sbenjscstatic void	ng_con_part2(node_p node, hook_p hook, void *arg1, int arg2);
202173362Sbenjscstatic void	ng_con_part3(node_p node, hook_p hook, void *arg1, int arg2);
203173362Sbenjscstatic int	ng_mkpeer(node_p node, const char *name,
204173362Sbenjsc						const char *name2, char *type);
205278366Sadrian
206280059Sadrian/* Imported, these used to be externally visible, some may go back. */
207173362Sbenjscvoid	ng_destroy_hook(hook_p hook);
208278366Sadriannode_p	ng_name2noderef(node_p node, const char *name);
209280108Sadrianint	ng_path2noderef(node_p here, const char *path,
210280108Sadrian	node_p *dest, hook_p *lasthook);
211173362Sbenjscint	ng_make_node(const char *type, node_p *nodepp);
212280077Sadrianint	ng_path_parse(char *addr, char **node, char **path, char **hook);
213280069Sadrianvoid	ng_rmnode(node_p node, hook_p dummy1, void *dummy2, int dummy3);
214280077Sadrianvoid	ng_unname(node_p node);
215173362Sbenjsc
216278366Sadrian
217282400Sadrian/* Our own netgraph malloc type */
218282400SadrianMALLOC_DEFINE(M_NETGRAPH, "netgraph", "netgraph structures and ctrl messages");
219282400SadrianMALLOC_DEFINE(M_NETGRAPH_HOOK, "netgraph_hook", "netgraph hook structures");
220173362SbenjscMALLOC_DEFINE(M_NETGRAPH_NODE, "netgraph_node", "netgraph node structures");
221280115SadrianMALLOC_DEFINE(M_NETGRAPH_ITEM, "netgraph_item", "netgraph item structures");
222280115SadrianMALLOC_DEFINE(M_NETGRAPH_MSG, "netgraph_msg", "netgraph name storage");
223173362Sbenjsc
224278366Sadrian/* Should not be visible outside this file */
225173362Sbenjsc
226173362Sbenjsc#define _NG_ALLOC_HOOK(hook) \
227278366Sadrian	MALLOC(hook, hook_p, sizeof(*hook), M_NETGRAPH_HOOK, M_NOWAIT | M_ZERO)
228173362Sbenjsc#define _NG_ALLOC_NODE(node) \
229173362Sbenjsc	MALLOC(node, node_p, sizeof(*node), M_NETGRAPH_NODE, M_NOWAIT | M_ZERO)
230278366Sadrian
231278366Sadrian#ifdef NETGRAPH_DEBUG /*----------------------------------------------*/
232278366Sadrian/*
233173362Sbenjsc * In debug mode:
234278366Sadrian * In an attempt to help track reference count screwups
235173362Sbenjsc * we do not free objects back to the malloc system, but keep them
236173362Sbenjsc * in a local cache where we can examine them and keep information safely
237173362Sbenjsc * after they have been freed.
238278366Sadrian * We use this scheme for nodes and hooks, and to some extent for items.
239278366Sadrian */
240173362Sbenjscstatic __inline hook_p
241173362Sbenjscng_alloc_hook(void)
242278366Sadrian{
243173362Sbenjsc	hook_p hook;
244278366Sadrian	SLIST_ENTRY(ng_hook) temp;
245280112Sadrian	mtx_lock(&ng_nodelist_mtx);
246280112Sadrian	hook = LIST_FIRST(&ng_freehooks);
247280112Sadrian	if (hook) {
248280112Sadrian		LIST_REMOVE(hook, hk_hooks);
249280113Sadrian		bcopy(&hook->hk_all, &temp, sizeof(temp));
250280113Sadrian		bzero(hook, sizeof(struct ng_hook));
251280113Sadrian		bcopy(&temp, &hook->hk_all, sizeof(temp));
252280113Sadrian		mtx_unlock(&ng_nodelist_mtx);
253280112Sadrian		hook->hk_magic = HK_MAGIC;
254280071Sadrian	} else {
255173362Sbenjsc		mtx_unlock(&ng_nodelist_mtx);
256173362Sbenjsc		_NG_ALLOC_HOOK(hook);
257280059Sadrian		if (hook) {
258177043Sthompsa			hook->hk_magic = HK_MAGIC;
259177043Sthompsa			mtx_lock(&ng_nodelist_mtx);
260280115Sadrian			SLIST_INSERT_HEAD(&ng_allhooks, hook, hk_all);
261177043Sthompsa			mtx_unlock(&ng_nodelist_mtx);
262280071Sadrian		}
263280108Sadrian	}
264280108Sadrian	return (hook);
265280108Sadrian}
266280108Sadrian
267280108Sadrianstatic __inline node_p
268280108Sadrianng_alloc_node(void)
269280108Sadrian{
270280113Sadrian	node_p node;
271280113Sadrian	SLIST_ENTRY(ng_node) temp;
272280113Sadrian	mtx_lock(&ng_nodelist_mtx);
273280113Sadrian	node = LIST_FIRST(&ng_freenodes);
274280113Sadrian	if (node) {
275280113Sadrian		LIST_REMOVE(node, nd_nodes);
276280077Sadrian		bcopy(&node->nd_all, &temp, sizeof(temp));
277280077Sadrian		bzero(node, sizeof(struct ng_node));
278280077Sadrian		bcopy(&temp, &node->nd_all, sizeof(temp));
279280077Sadrian		mtx_unlock(&ng_nodelist_mtx);
280280077Sadrian		node->nd_magic = ND_MAGIC;
281280077Sadrian	} else {
282280071Sadrian		mtx_unlock(&ng_nodelist_mtx);
283280071Sadrian		_NG_ALLOC_NODE(node);
284280071Sadrian		if (node) {
285280071Sadrian			node->nd_magic = ND_MAGIC;
286280071Sadrian			mtx_lock(&ng_nodelist_mtx);
287280112Sadrian			SLIST_INSERT_HEAD(&ng_allnodes, node, nd_all);
288280112Sadrian			mtx_unlock(&ng_nodelist_mtx);
289280112Sadrian		}
290280112Sadrian	}
291280112Sadrian	return (node);
292280112Sadrian}
293
294#define NG_ALLOC_HOOK(hook) do { (hook) = ng_alloc_hook(); } while (0)
295#define NG_ALLOC_NODE(node) do { (node) = ng_alloc_node(); } while (0)
296
297
298#define NG_FREE_HOOK(hook)						\
299	do {								\
300		mtx_lock(&ng_nodelist_mtx);			\
301		LIST_INSERT_HEAD(&ng_freehooks, hook, hk_hooks);	\
302		hook->hk_magic = 0;					\
303		mtx_unlock(&ng_nodelist_mtx);			\
304	} while (0)
305
306#define NG_FREE_NODE(node)						\
307	do {								\
308		mtx_lock(&ng_nodelist_mtx);			\
309		LIST_INSERT_HEAD(&ng_freenodes, node, nd_nodes);	\
310		node->nd_magic = 0;					\
311		mtx_unlock(&ng_nodelist_mtx);			\
312	} while (0)
313
314#else /* NETGRAPH_DEBUG */ /*----------------------------------------------*/
315
316#define NG_ALLOC_HOOK(hook) _NG_ALLOC_HOOK(hook)
317#define NG_ALLOC_NODE(node) _NG_ALLOC_NODE(node)
318
319#define NG_FREE_HOOK(hook) do { FREE((hook), M_NETGRAPH_HOOK); } while (0)
320#define NG_FREE_NODE(node) do { FREE((node), M_NETGRAPH_NODE); } while (0)
321
322#endif /* NETGRAPH_DEBUG */ /*----------------------------------------------*/
323
324/* Set this to kdb_enter("X") to catch all errors as they occur */
325#ifndef TRAP_ERROR
326#define TRAP_ERROR()
327#endif
328
329static	ng_ID_t nextID = 1;
330
331#ifdef INVARIANTS
332#define CHECK_DATA_MBUF(m)	do {					\
333		struct mbuf *n;						\
334		int total;						\
335									\
336		M_ASSERTPKTHDR(m);					\
337		for (total = 0, n = (m); n != NULL; n = n->m_next) {	\
338			total += n->m_len;				\
339			if (n->m_nextpkt != NULL)			\
340				panic("%s: m_nextpkt", __func__);	\
341		}							\
342									\
343		if ((m)->m_pkthdr.len != total) {			\
344			panic("%s: %d != %d",				\
345			    __func__, (m)->m_pkthdr.len, total);	\
346		}							\
347	} while (0)
348#else
349#define CHECK_DATA_MBUF(m)
350#endif
351
352
353/************************************************************************
354	Parse type definitions for generic messages
355************************************************************************/
356
357/* Handy structure parse type defining macro */
358#define DEFINE_PARSE_STRUCT_TYPE(lo, up, args)				\
359static const struct ng_parse_struct_field				\
360	ng_ ## lo ## _type_fields[] = NG_GENERIC_ ## up ## _INFO args;	\
361static const struct ng_parse_type ng_generic_ ## lo ## _type = {	\
362	&ng_parse_struct_type,						\
363	&ng_ ## lo ## _type_fields					\
364}
365
366DEFINE_PARSE_STRUCT_TYPE(mkpeer, MKPEER, ());
367DEFINE_PARSE_STRUCT_TYPE(connect, CONNECT, ());
368DEFINE_PARSE_STRUCT_TYPE(name, NAME, ());
369DEFINE_PARSE_STRUCT_TYPE(rmhook, RMHOOK, ());
370DEFINE_PARSE_STRUCT_TYPE(nodeinfo, NODEINFO, ());
371DEFINE_PARSE_STRUCT_TYPE(typeinfo, TYPEINFO, ());
372DEFINE_PARSE_STRUCT_TYPE(linkinfo, LINKINFO, (&ng_generic_nodeinfo_type));
373
374/* Get length of an array when the length is stored as a 32 bit
375   value immediately preceding the array -- as with struct namelist
376   and struct typelist. */
377static int
378ng_generic_list_getLength(const struct ng_parse_type *type,
379	const u_char *start, const u_char *buf)
380{
381	return *((const u_int32_t *)(buf - 4));
382}
383
384/* Get length of the array of struct linkinfo inside a struct hooklist */
385static int
386ng_generic_linkinfo_getLength(const struct ng_parse_type *type,
387	const u_char *start, const u_char *buf)
388{
389	const struct hooklist *hl = (const struct hooklist *)start;
390
391	return hl->nodeinfo.hooks;
392}
393
394/* Array type for a variable length array of struct namelist */
395static const struct ng_parse_array_info ng_nodeinfoarray_type_info = {
396	&ng_generic_nodeinfo_type,
397	&ng_generic_list_getLength
398};
399static const struct ng_parse_type ng_generic_nodeinfoarray_type = {
400	&ng_parse_array_type,
401	&ng_nodeinfoarray_type_info
402};
403
404/* Array type for a variable length array of struct typelist */
405static const struct ng_parse_array_info ng_typeinfoarray_type_info = {
406	&ng_generic_typeinfo_type,
407	&ng_generic_list_getLength
408};
409static const struct ng_parse_type ng_generic_typeinfoarray_type = {
410	&ng_parse_array_type,
411	&ng_typeinfoarray_type_info
412};
413
414/* Array type for array of struct linkinfo in struct hooklist */
415static const struct ng_parse_array_info ng_generic_linkinfo_array_type_info = {
416	&ng_generic_linkinfo_type,
417	&ng_generic_linkinfo_getLength
418};
419static const struct ng_parse_type ng_generic_linkinfo_array_type = {
420	&ng_parse_array_type,
421	&ng_generic_linkinfo_array_type_info
422};
423
424DEFINE_PARSE_STRUCT_TYPE(typelist, TYPELIST, (&ng_generic_nodeinfoarray_type));
425DEFINE_PARSE_STRUCT_TYPE(hooklist, HOOKLIST,
426	(&ng_generic_nodeinfo_type, &ng_generic_linkinfo_array_type));
427DEFINE_PARSE_STRUCT_TYPE(listnodes, LISTNODES,
428	(&ng_generic_nodeinfoarray_type));
429
430/* List of commands and how to convert arguments to/from ASCII */
431static const struct ng_cmdlist ng_generic_cmds[] = {
432	{
433	  NGM_GENERIC_COOKIE,
434	  NGM_SHUTDOWN,
435	  "shutdown",
436	  NULL,
437	  NULL
438	},
439	{
440	  NGM_GENERIC_COOKIE,
441	  NGM_MKPEER,
442	  "mkpeer",
443	  &ng_generic_mkpeer_type,
444	  NULL
445	},
446	{
447	  NGM_GENERIC_COOKIE,
448	  NGM_CONNECT,
449	  "connect",
450	  &ng_generic_connect_type,
451	  NULL
452	},
453	{
454	  NGM_GENERIC_COOKIE,
455	  NGM_NAME,
456	  "name",
457	  &ng_generic_name_type,
458	  NULL
459	},
460	{
461	  NGM_GENERIC_COOKIE,
462	  NGM_RMHOOK,
463	  "rmhook",
464	  &ng_generic_rmhook_type,
465	  NULL
466	},
467	{
468	  NGM_GENERIC_COOKIE,
469	  NGM_NODEINFO,
470	  "nodeinfo",
471	  NULL,
472	  &ng_generic_nodeinfo_type
473	},
474	{
475	  NGM_GENERIC_COOKIE,
476	  NGM_LISTHOOKS,
477	  "listhooks",
478	  NULL,
479	  &ng_generic_hooklist_type
480	},
481	{
482	  NGM_GENERIC_COOKIE,
483	  NGM_LISTNAMES,
484	  "listnames",
485	  NULL,
486	  &ng_generic_listnodes_type	/* same as NGM_LISTNODES */
487	},
488	{
489	  NGM_GENERIC_COOKIE,
490	  NGM_LISTNODES,
491	  "listnodes",
492	  NULL,
493	  &ng_generic_listnodes_type
494	},
495	{
496	  NGM_GENERIC_COOKIE,
497	  NGM_LISTTYPES,
498	  "listtypes",
499	  NULL,
500	  &ng_generic_typeinfo_type
501	},
502	{
503	  NGM_GENERIC_COOKIE,
504	  NGM_TEXT_CONFIG,
505	  "textconfig",
506	  NULL,
507	  &ng_parse_string_type
508	},
509	{
510	  NGM_GENERIC_COOKIE,
511	  NGM_TEXT_STATUS,
512	  "textstatus",
513	  NULL,
514	  &ng_parse_string_type
515	},
516	{
517	  NGM_GENERIC_COOKIE,
518	  NGM_ASCII2BINARY,
519	  "ascii2binary",
520	  &ng_parse_ng_mesg_type,
521	  &ng_parse_ng_mesg_type
522	},
523	{
524	  NGM_GENERIC_COOKIE,
525	  NGM_BINARY2ASCII,
526	  "binary2ascii",
527	  &ng_parse_ng_mesg_type,
528	  &ng_parse_ng_mesg_type
529	},
530	{ 0 }
531};
532
533/************************************************************************
534			Node routines
535************************************************************************/
536
537/*
538 * Instantiate a node of the requested type
539 */
540int
541ng_make_node(const char *typename, node_p *nodepp)
542{
543	struct ng_type *type;
544	int	error;
545
546	/* Check that the type makes sense */
547	if (typename == NULL) {
548		TRAP_ERROR();
549		return (EINVAL);
550	}
551
552	/* Locate the node type. If we fail we return. Do not try to load
553	 * module.
554	 */
555	if ((type = ng_findtype(typename)) == NULL)
556		return (ENXIO);
557
558	/*
559	 * If we have a constructor, then make the node and
560	 * call the constructor to do type specific initialisation.
561	 */
562	if (type->constructor != NULL) {
563		if ((error = ng_make_node_common(type, nodepp)) == 0) {
564			if ((error = ((*type->constructor)(*nodepp)) != 0)) {
565				NG_NODE_UNREF(*nodepp);
566			}
567		}
568	} else {
569		/*
570		 * Node has no constructor. We cannot ask for one
571		 * to be made. It must be brought into existance by
572		 * some external agency. The external agency should
573		 * call ng_make_node_common() directly to get the
574		 * netgraph part initialised.
575		 */
576		TRAP_ERROR();
577		error = EINVAL;
578	}
579	return (error);
580}
581
582/*
583 * Generic node creation. Called by node initialisation for externally
584 * instantiated nodes (e.g. hardware, sockets, etc ).
585 * The returned node has a reference count of 1.
586 */
587int
588ng_make_node_common(struct ng_type *type, node_p *nodepp)
589{
590	node_p node;
591
592	/* Require the node type to have been already installed */
593	if (ng_findtype(type->name) == NULL) {
594		TRAP_ERROR();
595		return (EINVAL);
596	}
597
598	/* Make a node and try attach it to the type */
599	NG_ALLOC_NODE(node);
600	if (node == NULL) {
601		TRAP_ERROR();
602		return (ENOMEM);
603	}
604	node->nd_type = type;
605	NG_NODE_REF(node);				/* note reference */
606	type->refs++;
607
608	mtx_init(&node->nd_input_queue.q_mtx, "ng_node", NULL, MTX_SPIN);
609	node->nd_input_queue.queue = NULL;
610	node->nd_input_queue.last = &node->nd_input_queue.queue;
611	node->nd_input_queue.q_flags = 0;
612	node->nd_input_queue.q_node = node;
613
614	/* Initialize hook list for new node */
615	LIST_INIT(&node->nd_hooks);
616
617	/* Link us into the node linked list */
618	mtx_lock(&ng_nodelist_mtx);
619	LIST_INSERT_HEAD(&ng_nodelist, node, nd_nodes);
620	mtx_unlock(&ng_nodelist_mtx);
621
622
623	/* get an ID and put us in the hash chain */
624	mtx_lock(&ng_idhash_mtx);
625	for (;;) { /* wrap protection, even if silly */
626		node_p node2 = NULL;
627		node->nd_ID = nextID++; /* 137/second for 1 year before wrap */
628
629		/* Is there a problem with the new number? */
630		NG_IDHASH_FIND(node->nd_ID, node2); /* already taken? */
631		if ((node->nd_ID != 0) && (node2 == NULL)) {
632			break;
633		}
634	}
635	LIST_INSERT_HEAD(&ng_ID_hash[NG_IDHASH_FN(node->nd_ID)],
636							node, nd_idnodes);
637	mtx_unlock(&ng_idhash_mtx);
638
639	/* Done */
640	*nodepp = node;
641	return (0);
642}
643
644/*
645 * Forceably start the shutdown process on a node. Either call
646 * it's shutdown method, or do the default shutdown if there is
647 * no type-specific method.
648 *
649 * We can only be called form a shutdown message, so we know we have
650 * a writer lock, and therefore exclusive access. It also means
651 * that we should not be on the work queue, but we check anyhow.
652 *
653 * Persistent node types must have a type-specific method which
654 * Allocates a new node in which case, this one is irretrievably going away,
655 * or cleans up anything it needs, and just makes the node valid again,
656 * in which case we allow the node to survive.
657 *
658 * XXX We need to think of how to tell a persistant node that we
659 * REALLY need to go away because the hardware has gone or we
660 * are rebooting.... etc.
661 */
662void
663ng_rmnode(node_p node, hook_p dummy1, void *dummy2, int dummy3)
664{
665	hook_p hook;
666
667	/* Check if it's already shutting down */
668	if ((node->nd_flags & NGF_CLOSING) != 0)
669		return;
670
671	if (node == &ng_deadnode) {
672		printf ("shutdown called on deadnode\n");
673		return;
674	}
675
676	/* Add an extra reference so it doesn't go away during this */
677	NG_NODE_REF(node);
678
679	/*
680	 * Mark it invalid so any newcomers know not to try use it
681	 * Also add our own mark so we can't recurse
682	 * note that NGF_INVALID does not do this as it's also set during
683	 * creation
684	 */
685	node->nd_flags |= NGF_INVALID|NGF_CLOSING;
686
687	/* If node has its pre-shutdown method, then call it first*/
688	if (node->nd_type && node->nd_type->close)
689		(*node->nd_type->close)(node);
690
691	/* Notify all remaining connected nodes to disconnect */
692	while ((hook = LIST_FIRST(&node->nd_hooks)) != NULL)
693		ng_destroy_hook(hook);
694
695	/*
696	 * Drain the input queue forceably.
697	 * it has no hooks so what's it going to do, bleed on someone?
698	 * Theoretically we came here from a queue entry that was added
699	 * Just before the queue was closed, so it should be empty anyway.
700	 * Also removes us from worklist if needed.
701	 */
702	ng_flush_input_queue(&node->nd_input_queue);
703
704	/* Ask the type if it has anything to do in this case */
705	if (node->nd_type && node->nd_type->shutdown) {
706		(*node->nd_type->shutdown)(node);
707		if (NG_NODE_IS_VALID(node)) {
708			/*
709			 * Well, blow me down if the node code hasn't declared
710			 * that it doesn't want to die.
711			 * Presumably it is a persistant node.
712			 * If we REALLY want it to go away,
713			 *  e.g. hardware going away,
714			 * Our caller should set NGF_REALLY_DIE in nd_flags.
715			 */
716			node->nd_flags &= ~(NGF_INVALID|NGF_CLOSING);
717			NG_NODE_UNREF(node); /* Assume they still have theirs */
718			return;
719		}
720	} else {				/* do the default thing */
721		NG_NODE_UNREF(node);
722	}
723
724	ng_unname(node); /* basically a NOP these days */
725
726	/*
727	 * Remove extra reference, possibly the last
728	 * Possible other holders of references may include
729	 * timeout callouts, but theoretically the node's supposed to
730	 * have cancelled them. Possibly hardware dependencies may
731	 * force a driver to 'linger' with a reference.
732	 */
733	NG_NODE_UNREF(node);
734}
735
736/*
737 * Remove a reference to the node, possibly the last.
738 * deadnode always acts as it it were the last.
739 */
740int
741ng_unref_node(node_p node)
742{
743	int v;
744
745	if (node == &ng_deadnode) {
746		return (0);
747	}
748
749	do {
750		v = node->nd_refs - 1;
751	} while (! atomic_cmpset_int(&node->nd_refs, v + 1, v));
752
753	if (v == 0) { /* we were the last */
754
755		mtx_lock(&ng_nodelist_mtx);
756		node->nd_type->refs--; /* XXX maybe should get types lock? */
757		LIST_REMOVE(node, nd_nodes);
758		mtx_unlock(&ng_nodelist_mtx);
759
760		mtx_lock(&ng_idhash_mtx);
761		LIST_REMOVE(node, nd_idnodes);
762		mtx_unlock(&ng_idhash_mtx);
763
764		mtx_destroy(&node->nd_input_queue.q_mtx);
765		NG_FREE_NODE(node);
766	}
767	return (v);
768}
769
770/************************************************************************
771			Node ID handling
772************************************************************************/
773static node_p
774ng_ID2noderef(ng_ID_t ID)
775{
776	node_p node;
777	mtx_lock(&ng_idhash_mtx);
778	NG_IDHASH_FIND(ID, node);
779	if(node)
780		NG_NODE_REF(node);
781	mtx_unlock(&ng_idhash_mtx);
782	return(node);
783}
784
785ng_ID_t
786ng_node2ID(node_p node)
787{
788	return (node ? NG_NODE_ID(node) : 0);
789}
790
791/************************************************************************
792			Node name handling
793************************************************************************/
794
795/*
796 * Assign a node a name. Once assigned, the name cannot be changed.
797 */
798int
799ng_name_node(node_p node, const char *name)
800{
801	int i;
802	node_p node2;
803
804	/* Check the name is valid */
805	for (i = 0; i < NG_NODESIZ; i++) {
806		if (name[i] == '\0' || name[i] == '.' || name[i] == ':')
807			break;
808	}
809	if (i == 0 || name[i] != '\0') {
810		TRAP_ERROR();
811		return (EINVAL);
812	}
813	if (ng_decodeidname(name) != 0) { /* valid IDs not allowed here */
814		TRAP_ERROR();
815		return (EINVAL);
816	}
817
818	/* Check the name isn't already being used */
819	if ((node2 = ng_name2noderef(node, name)) != NULL) {
820		NG_NODE_UNREF(node2);
821		TRAP_ERROR();
822		return (EADDRINUSE);
823	}
824
825	/* copy it */
826	strlcpy(NG_NODE_NAME(node), name, NG_NODESIZ);
827
828	return (0);
829}
830
831/*
832 * Find a node by absolute name. The name should NOT end with ':'
833 * The name "." means "this node" and "[xxx]" means "the node
834 * with ID (ie, at address) xxx".
835 *
836 * Returns the node if found, else NULL.
837 * Eventually should add something faster than a sequential search.
838 * Note it aquires a reference on the node so you can be sure it's still there.
839 */
840node_p
841ng_name2noderef(node_p here, const char *name)
842{
843	node_p node;
844	ng_ID_t temp;
845
846	/* "." means "this node" */
847	if (strcmp(name, ".") == 0) {
848		NG_NODE_REF(here);
849		return(here);
850	}
851
852	/* Check for name-by-ID */
853	if ((temp = ng_decodeidname(name)) != 0) {
854		return (ng_ID2noderef(temp));
855	}
856
857	/* Find node by name */
858	mtx_lock(&ng_nodelist_mtx);
859	LIST_FOREACH(node, &ng_nodelist, nd_nodes) {
860		if (NG_NODE_IS_VALID(node)
861		&& NG_NODE_HAS_NAME(node)
862		&& (strcmp(NG_NODE_NAME(node), name) == 0)) {
863			break;
864		}
865	}
866	if (node)
867		NG_NODE_REF(node);
868	mtx_unlock(&ng_nodelist_mtx);
869	return (node);
870}
871
872/*
873 * Decode an ID name, eg. "[f03034de]". Returns 0 if the
874 * string is not valid, otherwise returns the value.
875 */
876static ng_ID_t
877ng_decodeidname(const char *name)
878{
879	const int len = strlen(name);
880	char *eptr;
881	u_long val;
882
883	/* Check for proper length, brackets, no leading junk */
884	if ((len < 3)
885	|| (name[0] != '[')
886	|| (name[len - 1] != ']')
887	|| (!isxdigit(name[1]))) {
888		return ((ng_ID_t)0);
889	}
890
891	/* Decode number */
892	val = strtoul(name + 1, &eptr, 16);
893	if ((eptr - name != len - 1)
894	|| (val == ULONG_MAX)
895	|| (val == 0)) {
896		return ((ng_ID_t)0);
897	}
898	return (ng_ID_t)val;
899}
900
901/*
902 * Remove a name from a node. This should only be called
903 * when shutting down and removing the node.
904 * IF we allow name changing this may be more resurected.
905 */
906void
907ng_unname(node_p node)
908{
909}
910
911/************************************************************************
912			Hook routines
913 Names are not optional. Hooks are always connected, except for a
914 brief moment within these routines. On invalidation or during creation
915 they are connected to the 'dead' hook.
916************************************************************************/
917
918/*
919 * Remove a hook reference
920 */
921void
922ng_unref_hook(hook_p hook)
923{
924	int v;
925
926	if (hook == &ng_deadhook) {
927		return;
928	}
929	do {
930		v = hook->hk_refs;
931	} while (! atomic_cmpset_int(&hook->hk_refs, v, v - 1));
932
933	if (v == 1) { /* we were the last */
934		if (_NG_HOOK_NODE(hook)) { /* it'll probably be ng_deadnode */
935			_NG_NODE_UNREF((_NG_HOOK_NODE(hook)));
936			hook->hk_node = NULL;
937		}
938		NG_FREE_HOOK(hook);
939	}
940}
941
942/*
943 * Add an unconnected hook to a node. Only used internally.
944 * Assumes node is locked. (XXX not yet true )
945 */
946static int
947ng_add_hook(node_p node, const char *name, hook_p *hookp)
948{
949	hook_p hook;
950	int error = 0;
951
952	/* Check that the given name is good */
953	if (name == NULL) {
954		TRAP_ERROR();
955		return (EINVAL);
956	}
957	if (ng_findhook(node, name) != NULL) {
958		TRAP_ERROR();
959		return (EEXIST);
960	}
961
962	/* Allocate the hook and link it up */
963	NG_ALLOC_HOOK(hook);
964	if (hook == NULL) {
965		TRAP_ERROR();
966		return (ENOMEM);
967	}
968	hook->hk_refs = 1;		/* add a reference for us to return */
969	hook->hk_flags = HK_INVALID;
970	hook->hk_peer = &ng_deadhook;	/* start off this way */
971	hook->hk_node = node;
972	NG_NODE_REF(node);		/* each hook counts as a reference */
973
974	/* Set hook name */
975	strlcpy(NG_HOOK_NAME(hook), name, NG_HOOKSIZ);
976
977	/*
978	 * Check if the node type code has something to say about it
979	 * If it fails, the unref of the hook will also unref the node.
980	 */
981	if (node->nd_type->newhook != NULL) {
982		if ((error = (*node->nd_type->newhook)(node, hook, name))) {
983			NG_HOOK_UNREF(hook);	/* this frees the hook */
984			return (error);
985		}
986	}
987	/*
988	 * The 'type' agrees so far, so go ahead and link it in.
989	 * We'll ask again later when we actually connect the hooks.
990	 */
991	LIST_INSERT_HEAD(&node->nd_hooks, hook, hk_hooks);
992	node->nd_numhooks++;
993	NG_HOOK_REF(hook);	/* one for the node */
994
995	if (hookp)
996		*hookp = hook;
997	return (0);
998}
999
1000/*
1001 * Find a hook
1002 *
1003 * Node types may supply their own optimized routines for finding
1004 * hooks.  If none is supplied, we just do a linear search.
1005 * XXX Possibly we should add a reference to the hook?
1006 */
1007hook_p
1008ng_findhook(node_p node, const char *name)
1009{
1010	hook_p hook;
1011
1012	if (node->nd_type->findhook != NULL)
1013		return (*node->nd_type->findhook)(node, name);
1014	LIST_FOREACH(hook, &node->nd_hooks, hk_hooks) {
1015		if (NG_HOOK_IS_VALID(hook)
1016		&& (strcmp(NG_HOOK_NAME(hook), name) == 0))
1017			return (hook);
1018	}
1019	return (NULL);
1020}
1021
1022/*
1023 * Destroy a hook
1024 *
1025 * As hooks are always attached, this really destroys two hooks.
1026 * The one given, and the one attached to it. Disconnect the hooks
1027 * from each other first. We reconnect the peer hook to the 'dead'
1028 * hook so that it can still exist after we depart. We then
1029 * send the peer its own destroy message. This ensures that we only
1030 * interact with the peer's structures when it is locked processing that
1031 * message. We hold a reference to the peer hook so we are guaranteed that
1032 * the peer hook and node are still going to exist until
1033 * we are finished there as the hook holds a ref on the node.
1034 * We run this same code again on the peer hook, but that time it is already
1035 * attached to the 'dead' hook.
1036 *
1037 * This routine is called at all stages of hook creation
1038 * on error detection and must be able to handle any such stage.
1039 */
1040void
1041ng_destroy_hook(hook_p hook)
1042{
1043	hook_p peer;
1044	node_p node;
1045
1046	if (hook == &ng_deadhook) {	/* better safe than sorry */
1047		printf("ng_destroy_hook called on deadhook\n");
1048		return;
1049	}
1050
1051	/*
1052	 * Protect divorce process with mutex, to avoid races on
1053	 * simultaneous disconnect.
1054	 */
1055	mtx_lock(&ng_topo_mtx);
1056
1057	hook->hk_flags |= HK_INVALID;
1058
1059	peer = NG_HOOK_PEER(hook);
1060	node = NG_HOOK_NODE(hook);
1061
1062	if (peer && (peer != &ng_deadhook)) {
1063		/*
1064		 * Set the peer to point to ng_deadhook
1065		 * from this moment on we are effectively independent it.
1066		 * send it an rmhook message of it's own.
1067		 */
1068		peer->hk_peer = &ng_deadhook;	/* They no longer know us */
1069		hook->hk_peer = &ng_deadhook;	/* Nor us, them */
1070		if (NG_HOOK_NODE(peer) == &ng_deadnode) {
1071			/*
1072			 * If it's already divorced from a node,
1073			 * just free it.
1074			 */
1075			mtx_unlock(&ng_topo_mtx);
1076		} else {
1077			mtx_unlock(&ng_topo_mtx);
1078			ng_rmhook_self(peer); 	/* Send it a surprise */
1079		}
1080		NG_HOOK_UNREF(peer);		/* account for peer link */
1081		NG_HOOK_UNREF(hook);		/* account for peer link */
1082	} else
1083		mtx_unlock(&ng_topo_mtx);
1084
1085	mtx_assert(&ng_topo_mtx, MA_NOTOWNED);
1086
1087	/*
1088	 * Remove the hook from the node's list to avoid possible recursion
1089	 * in case the disconnection results in node shutdown.
1090	 */
1091	if (node == &ng_deadnode) { /* happens if called from ng_con_nodes() */
1092		return;
1093	}
1094	LIST_REMOVE(hook, hk_hooks);
1095	node->nd_numhooks--;
1096	if (node->nd_type->disconnect) {
1097		/*
1098		 * The type handler may elect to destroy the node so don't
1099		 * trust its existance after this point. (except
1100		 * that we still hold a reference on it. (which we
1101		 * inherrited from the hook we are destroying)
1102		 */
1103		(*node->nd_type->disconnect) (hook);
1104	}
1105
1106	/*
1107	 * Note that because we will point to ng_deadnode, the original node
1108	 * is not decremented automatically so we do that manually.
1109	 */
1110	_NG_HOOK_NODE(hook) = &ng_deadnode;
1111	NG_NODE_UNREF(node);	/* We no longer point to it so adjust count */
1112	NG_HOOK_UNREF(hook);	/* Account for linkage (in list) to node */
1113}
1114
1115/*
1116 * Take two hooks on a node and merge the connection so that the given node
1117 * is effectively bypassed.
1118 */
1119int
1120ng_bypass(hook_p hook1, hook_p hook2)
1121{
1122	if (hook1->hk_node != hook2->hk_node) {
1123		TRAP_ERROR();
1124		return (EINVAL);
1125	}
1126	hook1->hk_peer->hk_peer = hook2->hk_peer;
1127	hook2->hk_peer->hk_peer = hook1->hk_peer;
1128
1129	hook1->hk_peer = &ng_deadhook;
1130	hook2->hk_peer = &ng_deadhook;
1131
1132	/* XXX If we ever cache methods on hooks update them as well */
1133	ng_destroy_hook(hook1);
1134	ng_destroy_hook(hook2);
1135	return (0);
1136}
1137
1138/*
1139 * Install a new netgraph type
1140 */
1141int
1142ng_newtype(struct ng_type *tp)
1143{
1144	const size_t namelen = strlen(tp->name);
1145
1146	/* Check version and type name fields */
1147	if ((tp->version != NG_ABI_VERSION)
1148	|| (namelen == 0)
1149	|| (namelen >= NG_TYPESIZ)) {
1150		TRAP_ERROR();
1151		if (tp->version != NG_ABI_VERSION) {
1152			printf("Netgraph: Node type rejected. ABI mismatch. Suggest recompile\n");
1153		}
1154		return (EINVAL);
1155	}
1156
1157	/* Check for name collision */
1158	if (ng_findtype(tp->name) != NULL) {
1159		TRAP_ERROR();
1160		return (EEXIST);
1161	}
1162
1163
1164	/* Link in new type */
1165	mtx_lock(&ng_typelist_mtx);
1166	LIST_INSERT_HEAD(&ng_typelist, tp, types);
1167	tp->refs = 1;	/* first ref is linked list */
1168	mtx_unlock(&ng_typelist_mtx);
1169	return (0);
1170}
1171
1172/*
1173 * unlink a netgraph type
1174 * If no examples exist
1175 */
1176int
1177ng_rmtype(struct ng_type *tp)
1178{
1179	/* Check for name collision */
1180	if (tp->refs != 1) {
1181		TRAP_ERROR();
1182		return (EBUSY);
1183	}
1184
1185	/* Unlink type */
1186	mtx_lock(&ng_typelist_mtx);
1187	LIST_REMOVE(tp, types);
1188	mtx_unlock(&ng_typelist_mtx);
1189	return (0);
1190}
1191
1192/*
1193 * Look for a type of the name given
1194 */
1195struct ng_type *
1196ng_findtype(const char *typename)
1197{
1198	struct ng_type *type;
1199
1200	mtx_lock(&ng_typelist_mtx);
1201	LIST_FOREACH(type, &ng_typelist, types) {
1202		if (strcmp(type->name, typename) == 0)
1203			break;
1204	}
1205	mtx_unlock(&ng_typelist_mtx);
1206	return (type);
1207}
1208
1209/************************************************************************
1210			Composite routines
1211************************************************************************/
1212/*
1213 * Connect two nodes using the specified hooks, using queued functions.
1214 */
1215static void
1216ng_con_part3(node_p node, hook_p hook, void *arg1, int arg2)
1217{
1218
1219	/*
1220	 * When we run, we know that the node 'node' is locked for us.
1221	 * Our caller has a reference on the hook.
1222	 * Our caller has a reference on the node.
1223	 * (In this case our caller is ng_apply_item() ).
1224	 * The peer hook has a reference on the hook.
1225	 * We are all set up except for the final call to the node, and
1226	 * the clearing of the INVALID flag.
1227	 */
1228	if (NG_HOOK_NODE(hook) == &ng_deadnode) {
1229		/*
1230		 * The node must have been freed again since we last visited
1231		 * here. ng_destry_hook() has this effect but nothing else does.
1232		 * We should just release our references and
1233		 * free anything we can think of.
1234		 * Since we know it's been destroyed, and it's our caller
1235		 * that holds the references, just return.
1236		 */
1237		return ;
1238	}
1239	if (hook->hk_node->nd_type->connect) {
1240		if ((*hook->hk_node->nd_type->connect) (hook)) {
1241			ng_destroy_hook(hook);	/* also zaps peer */
1242			printf("failed in ng_con_part3()\n");
1243			return ;
1244		}
1245	}
1246	/*
1247	 *  XXX this is wrong for SMP. Possibly we need
1248	 * to separate out 'create' and 'invalid' flags.
1249	 * should only set flags on hooks we have locked under our node.
1250	 */
1251	hook->hk_flags &= ~HK_INVALID;
1252	return ;
1253}
1254
1255static void
1256ng_con_part2(node_p node, hook_p hook, void *arg1, int arg2)
1257{
1258	hook_p peer;
1259
1260	/*
1261	 * When we run, we know that the node 'node' is locked for us.
1262	 * Our caller has a reference on the hook.
1263	 * Our caller has a reference on the node.
1264	 * (In this case our caller is ng_apply_item() ).
1265	 * The peer hook has a reference on the hook.
1266	 * our node pointer points to the 'dead' node.
1267	 * First check the hook name is unique.
1268	 * Should not happen because we checked before queueing this.
1269	 */
1270	if (ng_findhook(node, NG_HOOK_NAME(hook)) != NULL) {
1271		TRAP_ERROR();
1272		ng_destroy_hook(hook); /* should destroy peer too */
1273		printf("failed in ng_con_part2()\n");
1274		return ;
1275	}
1276	/*
1277	 * Check if the node type code has something to say about it
1278	 * If it fails, the unref of the hook will also unref the attached node,
1279	 * however since that node is 'ng_deadnode' this will do nothing.
1280	 * The peer hook will also be destroyed.
1281	 */
1282	if (node->nd_type->newhook != NULL) {
1283		if ((*node->nd_type->newhook)(node, hook, hook->hk_name)) {
1284			ng_destroy_hook(hook); /* should destroy peer too */
1285			printf("failed in ng_con_part2()\n");
1286			return ;
1287		}
1288	}
1289
1290	/*
1291	 * The 'type' agrees so far, so go ahead and link it in.
1292	 * We'll ask again later when we actually connect the hooks.
1293	 */
1294	hook->hk_node = node;		/* just overwrite ng_deadnode */
1295	NG_NODE_REF(node);		/* each hook counts as a reference */
1296	LIST_INSERT_HEAD(&node->nd_hooks, hook, hk_hooks);
1297	node->nd_numhooks++;
1298	NG_HOOK_REF(hook);	/* one for the node */
1299
1300	/*
1301	 * We now have a symetrical situation, where both hooks have been
1302	 * linked to their nodes, the newhook methods have been called
1303	 * And the references are all correct. The hooks are still marked
1304	 * as invalid, as we have not called the 'connect' methods
1305	 * yet.
1306	 * We can call the local one immediatly as we have the
1307	 * node locked, but we need to queue the remote one.
1308	 */
1309	if (hook->hk_node->nd_type->connect) {
1310		if ((*hook->hk_node->nd_type->connect) (hook)) {
1311			ng_destroy_hook(hook);	/* also zaps peer */
1312			printf("failed in ng_con_part2(A)\n");
1313			return ;
1314		}
1315	}
1316
1317	/*
1318	 * Acquire topo mutex to avoid race with ng_destroy_hook().
1319	 */
1320	mtx_lock(&ng_topo_mtx);
1321	peer = hook->hk_peer;
1322	if (peer == &ng_deadhook) {
1323		mtx_unlock(&ng_topo_mtx);
1324		printf("failed in ng_con_part2(B)\n");
1325		ng_destroy_hook(hook);
1326		return ;
1327	}
1328	mtx_unlock(&ng_topo_mtx);
1329
1330	if (ng_send_fn(peer->hk_node, peer, &ng_con_part3, arg1, arg2)) {
1331		printf("failed in ng_con_part2(C)\n");
1332		ng_destroy_hook(hook);	/* also zaps peer */
1333		return ;
1334	}
1335	hook->hk_flags &= ~HK_INVALID; /* need both to be able to work */
1336	return ;
1337}
1338
1339/*
1340 * Connect this node with another node. We assume that this node is
1341 * currently locked, as we are only called from an NGM_CONNECT message.
1342 */
1343static int
1344ng_con_nodes(node_p node, const char *name, node_p node2, const char *name2)
1345{
1346	int	error;
1347	hook_p	hook;
1348	hook_p	hook2;
1349
1350	if (ng_findhook(node2, name2) != NULL) {
1351		return(EEXIST);
1352	}
1353	if ((error = ng_add_hook(node, name, &hook)))  /* gives us a ref */
1354		return (error);
1355	/* Allocate the other hook and link it up */
1356	NG_ALLOC_HOOK(hook2);
1357	if (hook2 == NULL) {
1358		TRAP_ERROR();
1359		ng_destroy_hook(hook);	/* XXX check ref counts so far */
1360		NG_HOOK_UNREF(hook);	/* including our ref */
1361		return (ENOMEM);
1362	}
1363	hook2->hk_refs = 1;		/* start with a reference for us. */
1364	hook2->hk_flags = HK_INVALID;
1365	hook2->hk_peer = hook;		/* Link the two together */
1366	hook->hk_peer = hook2;
1367	NG_HOOK_REF(hook);		/* Add a ref for the peer to each*/
1368	NG_HOOK_REF(hook2);
1369	hook2->hk_node = &ng_deadnode;
1370	strlcpy(NG_HOOK_NAME(hook2), name2, NG_HOOKSIZ);
1371
1372	/*
1373	 * Queue the function above.
1374	 * Procesing continues in that function in the lock context of
1375	 * the other node.
1376	 */
1377	ng_send_fn(node2, hook2, &ng_con_part2, NULL, 0);
1378
1379	NG_HOOK_UNREF(hook);		/* Let each hook go if it wants to */
1380	NG_HOOK_UNREF(hook2);
1381	return (0);
1382}
1383
1384/*
1385 * Make a peer and connect.
1386 * We assume that the local node is locked.
1387 * The new node probably doesn't need a lock until
1388 * it has a hook, because it cannot really have any work until then,
1389 * but we should think about it a bit more.
1390 *
1391 * The problem may come if the other node also fires up
1392 * some hardware or a timer or some other source of activation,
1393 * also it may already get a command msg via it's ID.
1394 *
1395 * We could use the same method as ng_con_nodes() but we'd have
1396 * to add ability to remove the node when failing. (Not hard, just
1397 * make arg1 point to the node to remove).
1398 * Unless of course we just ignore failure to connect and leave
1399 * an unconnected node?
1400 */
1401static int
1402ng_mkpeer(node_p node, const char *name, const char *name2, char *type)
1403{
1404	node_p	node2;
1405	hook_p	hook1, hook2;
1406	int	error;
1407
1408	if ((error = ng_make_node(type, &node2))) {
1409		return (error);
1410	}
1411
1412	if ((error = ng_add_hook(node, name, &hook1))) { /* gives us a ref */
1413		ng_rmnode(node2, NULL, NULL, 0);
1414		return (error);
1415	}
1416
1417	if ((error = ng_add_hook(node2, name2, &hook2))) {
1418		ng_rmnode(node2, NULL, NULL, 0);
1419		ng_destroy_hook(hook1);
1420		NG_HOOK_UNREF(hook1);
1421		return (error);
1422	}
1423
1424	/*
1425	 * Actually link the two hooks together.
1426	 */
1427	hook1->hk_peer = hook2;
1428	hook2->hk_peer = hook1;
1429
1430	/* Each hook is referenced by the other */
1431	NG_HOOK_REF(hook1);
1432	NG_HOOK_REF(hook2);
1433
1434	/* Give each node the opportunity to veto the pending connection */
1435	if (hook1->hk_node->nd_type->connect) {
1436		error = (*hook1->hk_node->nd_type->connect) (hook1);
1437	}
1438
1439	if ((error == 0) && hook2->hk_node->nd_type->connect) {
1440		error = (*hook2->hk_node->nd_type->connect) (hook2);
1441
1442	}
1443
1444	/*
1445	 * drop the references we were holding on the two hooks.
1446	 */
1447	if (error) {
1448		ng_destroy_hook(hook2);	/* also zaps hook1 */
1449		ng_rmnode(node2, NULL, NULL, 0);
1450	} else {
1451		/* As a last act, allow the hooks to be used */
1452		hook1->hk_flags &= ~HK_INVALID;
1453		hook2->hk_flags &= ~HK_INVALID;
1454	}
1455	NG_HOOK_UNREF(hook1);
1456	NG_HOOK_UNREF(hook2);
1457	return (error);
1458}
1459
1460/************************************************************************
1461		Utility routines to send self messages
1462************************************************************************/
1463
1464/* Shut this node down as soon as everyone is clear of it */
1465/* Should add arg "immediatly" to jump the queue */
1466int
1467ng_rmnode_self(node_p node)
1468{
1469	int		error;
1470
1471	if (node == &ng_deadnode)
1472		return (0);
1473	node->nd_flags |= NGF_INVALID;
1474	if (node->nd_flags & NGF_CLOSING)
1475		return (0);
1476
1477	error = ng_send_fn(node, NULL, &ng_rmnode, NULL, 0);
1478	return (error);
1479}
1480
1481static void
1482ng_rmhook_part2(node_p node, hook_p hook, void *arg1, int arg2)
1483{
1484	ng_destroy_hook(hook);
1485	return ;
1486}
1487
1488int
1489ng_rmhook_self(hook_p hook)
1490{
1491	int		error;
1492	node_p node = NG_HOOK_NODE(hook);
1493
1494	if (node == &ng_deadnode)
1495		return (0);
1496
1497	error = ng_send_fn(node, hook, &ng_rmhook_part2, NULL, 0);
1498	return (error);
1499}
1500
1501/***********************************************************************
1502 * Parse and verify a string of the form:  <NODE:><PATH>
1503 *
1504 * Such a string can refer to a specific node or a specific hook
1505 * on a specific node, depending on how you look at it. In the
1506 * latter case, the PATH component must not end in a dot.
1507 *
1508 * Both <NODE:> and <PATH> are optional. The <PATH> is a string
1509 * of hook names separated by dots. This breaks out the original
1510 * string, setting *nodep to "NODE" (or NULL if none) and *pathp
1511 * to "PATH" (or NULL if degenerate). Also, *hookp will point to
1512 * the final hook component of <PATH>, if any, otherwise NULL.
1513 *
1514 * This returns -1 if the path is malformed. The char ** are optional.
1515 ***********************************************************************/
1516int
1517ng_path_parse(char *addr, char **nodep, char **pathp, char **hookp)
1518{
1519	char	*node, *path, *hook;
1520	int	k;
1521
1522	/*
1523	 * Extract absolute NODE, if any
1524	 */
1525	for (path = addr; *path && *path != ':'; path++);
1526	if (*path) {
1527		node = addr;	/* Here's the NODE */
1528		*path++ = '\0';	/* Here's the PATH */
1529
1530		/* Node name must not be empty */
1531		if (!*node)
1532			return -1;
1533
1534		/* A name of "." is OK; otherwise '.' not allowed */
1535		if (strcmp(node, ".") != 0) {
1536			for (k = 0; node[k]; k++)
1537				if (node[k] == '.')
1538					return -1;
1539		}
1540	} else {
1541		node = NULL;	/* No absolute NODE */
1542		path = addr;	/* Here's the PATH */
1543	}
1544
1545	/* Snoop for illegal characters in PATH */
1546	for (k = 0; path[k]; k++)
1547		if (path[k] == ':')
1548			return -1;
1549
1550	/* Check for no repeated dots in PATH */
1551	for (k = 0; path[k]; k++)
1552		if (path[k] == '.' && path[k + 1] == '.')
1553			return -1;
1554
1555	/* Remove extra (degenerate) dots from beginning or end of PATH */
1556	if (path[0] == '.')
1557		path++;
1558	if (*path && path[strlen(path) - 1] == '.')
1559		path[strlen(path) - 1] = 0;
1560
1561	/* If PATH has a dot, then we're not talking about a hook */
1562	if (*path) {
1563		for (hook = path, k = 0; path[k]; k++)
1564			if (path[k] == '.') {
1565				hook = NULL;
1566				break;
1567			}
1568	} else
1569		path = hook = NULL;
1570
1571	/* Done */
1572	if (nodep)
1573		*nodep = node;
1574	if (pathp)
1575		*pathp = path;
1576	if (hookp)
1577		*hookp = hook;
1578	return (0);
1579}
1580
1581/*
1582 * Given a path, which may be absolute or relative, and a starting node,
1583 * return the destination node.
1584 */
1585int
1586ng_path2noderef(node_p here, const char *address,
1587				node_p *destp, hook_p *lasthook)
1588{
1589	char    fullpath[NG_PATHSIZ];
1590	char   *nodename, *path, pbuf[2];
1591	node_p  node, oldnode;
1592	char   *cp;
1593	hook_p hook = NULL;
1594
1595	/* Initialize */
1596	if (destp == NULL) {
1597		TRAP_ERROR();
1598		return EINVAL;
1599	}
1600	*destp = NULL;
1601
1602	/* Make a writable copy of address for ng_path_parse() */
1603	strncpy(fullpath, address, sizeof(fullpath) - 1);
1604	fullpath[sizeof(fullpath) - 1] = '\0';
1605
1606	/* Parse out node and sequence of hooks */
1607	if (ng_path_parse(fullpath, &nodename, &path, NULL) < 0) {
1608		TRAP_ERROR();
1609		return EINVAL;
1610	}
1611	if (path == NULL) {
1612		pbuf[0] = '.';	/* Needs to be writable */
1613		pbuf[1] = '\0';
1614		path = pbuf;
1615	}
1616
1617	/*
1618	 * For an absolute address, jump to the starting node.
1619	 * Note that this holds a reference on the node for us.
1620	 * Don't forget to drop the reference if we don't need it.
1621	 */
1622	if (nodename) {
1623		node = ng_name2noderef(here, nodename);
1624		if (node == NULL) {
1625			TRAP_ERROR();
1626			return (ENOENT);
1627		}
1628	} else {
1629		if (here == NULL) {
1630			TRAP_ERROR();
1631			return (EINVAL);
1632		}
1633		node = here;
1634		NG_NODE_REF(node);
1635	}
1636
1637	/*
1638	 * Now follow the sequence of hooks
1639	 * XXX
1640	 * We actually cannot guarantee that the sequence
1641	 * is not being demolished as we crawl along it
1642	 * without extra-ordinary locking etc.
1643	 * So this is a bit dodgy to say the least.
1644	 * We can probably hold up some things by holding
1645	 * the nodelist mutex for the time of this
1646	 * crawl if we wanted.. At least that way we wouldn't have to
1647	 * worry about the nodes dissappearing, but the hooks would still
1648	 * be a problem.
1649	 */
1650	for (cp = path; node != NULL && *cp != '\0'; ) {
1651		char *segment;
1652
1653		/*
1654		 * Break out the next path segment. Replace the dot we just
1655		 * found with a NUL; "cp" points to the next segment (or the
1656		 * NUL at the end).
1657		 */
1658		for (segment = cp; *cp != '\0'; cp++) {
1659			if (*cp == '.') {
1660				*cp++ = '\0';
1661				break;
1662			}
1663		}
1664
1665		/* Empty segment */
1666		if (*segment == '\0')
1667			continue;
1668
1669		/* We have a segment, so look for a hook by that name */
1670		hook = ng_findhook(node, segment);
1671
1672		/* Can't get there from here... */
1673		if (hook == NULL
1674		    || NG_HOOK_PEER(hook) == NULL
1675		    || NG_HOOK_NOT_VALID(hook)
1676		    || NG_HOOK_NOT_VALID(NG_HOOK_PEER(hook))) {
1677			TRAP_ERROR();
1678			NG_NODE_UNREF(node);
1679#if 0
1680			printf("hooknotvalid %s %s %d %d %d %d ",
1681					path,
1682					segment,
1683					hook == NULL,
1684					NG_HOOK_PEER(hook) == NULL,
1685					NG_HOOK_NOT_VALID(hook),
1686					NG_HOOK_NOT_VALID(NG_HOOK_PEER(hook)));
1687#endif
1688			return (ENOENT);
1689		}
1690
1691		/*
1692		 * Hop on over to the next node
1693		 * XXX
1694		 * Big race conditions here as hooks and nodes go away
1695		 * *** Idea.. store an ng_ID_t in each hook and use that
1696		 * instead of the direct hook in this crawl?
1697		 */
1698		oldnode = node;
1699		if ((node = NG_PEER_NODE(hook)))
1700			NG_NODE_REF(node);	/* XXX RACE */
1701		NG_NODE_UNREF(oldnode);	/* XXX another race */
1702		if (NG_NODE_NOT_VALID(node)) {
1703			NG_NODE_UNREF(node);	/* XXX more races */
1704			node = NULL;
1705		}
1706	}
1707
1708	/* If node somehow missing, fail here (probably this is not needed) */
1709	if (node == NULL) {
1710		TRAP_ERROR();
1711		return (ENXIO);
1712	}
1713
1714	/* Done */
1715	*destp = node;
1716	if (lasthook != NULL)
1717		*lasthook = (hook ? NG_HOOK_PEER(hook) : NULL);
1718	return (0);
1719}
1720
1721/***************************************************************\
1722* Input queue handling.
1723* All activities are submitted to the node via the input queue
1724* which implements a multiple-reader/single-writer gate.
1725* Items which cannot be handled immeditly are queued.
1726*
1727* read-write queue locking inline functions			*
1728\***************************************************************/
1729
1730static __inline item_p ng_dequeue(struct ng_queue * ngq, int *rw);
1731static __inline item_p ng_acquire_read(struct ng_queue * ngq,
1732					item_p  item);
1733static __inline item_p ng_acquire_write(struct ng_queue * ngq,
1734					item_p  item);
1735static __inline void	ng_leave_read(struct ng_queue * ngq);
1736static __inline void	ng_leave_write(struct ng_queue * ngq);
1737static __inline void	ng_queue_rw(struct ng_queue * ngq,
1738					item_p  item, int rw);
1739
1740/*
1741 * Definition of the bits fields in the ng_queue flag word.
1742 * Defined here rather than in netgraph.h because no-one should fiddle
1743 * with them.
1744 *
1745 * The ordering here may be important! don't shuffle these.
1746 */
1747/*-
1748 Safety Barrier--------+ (adjustable to suit taste) (not used yet)
1749                       |
1750                       V
1751+-------+-------+-------+-------+-------+-------+-------+-------+
1752  | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | |
1753  | |A|c|t|i|v|e| |R|e|a|d|e|r| |C|o|u|n|t| | | | | | | | | |P|A|
1754  | | | | | | | | | | | | | | | | | | | | | | | | | | | | | |O|W|
1755+-------+-------+-------+-------+-------+-------+-------+-------+
1756  \___________________________ ____________________________/ | |
1757                            V                                | |
1758                  [active reader count]                      | |
1759                                                             | |
1760            Operation Pending -------------------------------+ |
1761                                                               |
1762          Active Writer ---------------------------------------+
1763
1764
1765*/
1766#define WRITER_ACTIVE	0x00000001
1767#define OP_PENDING	0x00000002
1768#define READER_INCREMENT 0x00000004
1769#define READER_MASK	0xfffffffc	/* Not valid if WRITER_ACTIVE is set */
1770#define SAFETY_BARRIER	0x00100000	/* 128K items queued should be enough */
1771
1772/* Defines of more elaborate states on the queue */
1773/* Mask of bits a new read cares about */
1774#define NGQ_RMASK	(WRITER_ACTIVE|OP_PENDING)
1775
1776/* Mask of bits a new write cares about */
1777#define NGQ_WMASK	(NGQ_RMASK|READER_MASK)
1778
1779/* Test to decide if there is something on the queue. */
1780#define QUEUE_ACTIVE(QP) ((QP)->q_flags & OP_PENDING)
1781
1782/* How to decide what the next queued item is. */
1783#define HEAD_IS_READER(QP)  NGI_QUEUED_READER((QP)->queue)
1784#define HEAD_IS_WRITER(QP)  NGI_QUEUED_WRITER((QP)->queue) /* notused */
1785
1786/* Read the status to decide if the next item on the queue can now run. */
1787#define QUEUED_READER_CAN_PROCEED(QP)			\
1788		(((QP)->q_flags & (NGQ_RMASK & ~OP_PENDING)) == 0)
1789#define QUEUED_WRITER_CAN_PROCEED(QP)			\
1790		(((QP)->q_flags & (NGQ_WMASK & ~OP_PENDING)) == 0)
1791
1792/* Is there a chance of getting ANY work off the queue? */
1793#define NEXT_QUEUED_ITEM_CAN_PROCEED(QP)				\
1794	(QUEUE_ACTIVE(QP) && 						\
1795	((HEAD_IS_READER(QP)) ? QUEUED_READER_CAN_PROCEED(QP) :		\
1796				QUEUED_WRITER_CAN_PROCEED(QP)))
1797
1798
1799#define NGQRW_R 0
1800#define NGQRW_W 1
1801
1802/*
1803 * Taking into account the current state of the queue and node, possibly take
1804 * the next entry off the queue and return it. Return NULL if there was
1805 * nothing we could return, either because there really was nothing there, or
1806 * because the node was in a state where it cannot yet process the next item
1807 * on the queue.
1808 *
1809 * This MUST MUST MUST be called with the mutex held.
1810 */
1811static __inline item_p
1812ng_dequeue(struct ng_queue *ngq, int *rw)
1813{
1814	item_p item;
1815	u_int		add_arg;
1816
1817	mtx_assert(&ngq->q_mtx, MA_OWNED);
1818	/*
1819	 * If there is nothing queued, then just return.
1820	 * No point in continuing.
1821	 * XXXGL: assert this?
1822	 */
1823	if (!QUEUE_ACTIVE(ngq)) {
1824		CTR4(KTR_NET, "%20s: node [%x] (%p) queue empty; "
1825		    "queue flags 0x%lx", __func__,
1826		    ngq->q_node->nd_ID, ngq->q_node, ngq->q_flags);
1827		return (NULL);
1828	}
1829
1830	/*
1831	 * From here, we can assume there is a head item.
1832	 * We need to find out what it is and if it can be dequeued, given
1833	 * the current state of the node.
1834	 */
1835	if (HEAD_IS_READER(ngq)) {
1836		if (!QUEUED_READER_CAN_PROCEED(ngq)) {
1837			/*
1838			 * It's a reader but we can't use it.
1839			 * We are stalled so make sure we don't
1840			 * get called again until something changes.
1841			 */
1842			ng_worklist_remove(ngq->q_node);
1843			CTR4(KTR_NET, "%20s: node [%x] (%p) queued reader "
1844			    "can't proceed; queue flags 0x%lx", __func__,
1845			    ngq->q_node->nd_ID, ngq->q_node, ngq->q_flags);
1846			return (NULL);
1847		}
1848		/*
1849		 * Head of queue is a reader and we have no write active.
1850		 * We don't care how many readers are already active.
1851		 * Add the correct increment for the reader count.
1852		 */
1853		add_arg = READER_INCREMENT;
1854		*rw = NGQRW_R;
1855	} else if (QUEUED_WRITER_CAN_PROCEED(ngq)) {
1856		/*
1857		 * There is a pending write, no readers and no active writer.
1858		 * This means we can go ahead with the pending writer. Note
1859		 * the fact that we now have a writer, ready for when we take
1860		 * it off the queue.
1861		 *
1862		 * We don't need to worry about a possible collision with the
1863		 * fasttrack reader.
1864		 *
1865		 * The fasttrack thread may take a long time to discover that we
1866		 * are running so we would have an inconsistent state in the
1867		 * flags for a while. Since we ignore the reader count
1868		 * entirely when the WRITER_ACTIVE flag is set, this should
1869		 * not matter (in fact it is defined that way). If it tests
1870		 * the flag before this operation, the OP_PENDING flag
1871		 * will make it fail, and if it tests it later, the
1872		 * WRITER_ACTIVE flag will do the same. If it is SO slow that
1873		 * we have actually completed the operation, and neither flag
1874		 * is set by the time that it tests the flags, then it is
1875		 * actually ok for it to continue. If it completes and we've
1876		 * finished and the read pending is set it still fails.
1877		 *
1878		 * So we can just ignore it,  as long as we can ensure that the
1879		 * transition from WRITE_PENDING state to the WRITER_ACTIVE
1880		 * state is atomic.
1881		 *
1882		 * After failing, first it will be held back by the mutex, then
1883		 * when it can proceed, it will queue its request, then it
1884		 * would arrive at this function. Usually it will have to
1885		 * leave empty handed because the ACTIVE WRITER bit will be
1886		 * set.
1887		 *
1888		 * Adjust the flags for the new active writer.
1889		 */
1890		add_arg = WRITER_ACTIVE;
1891		*rw = NGQRW_W;
1892		/*
1893		 * We want to write "active writer, no readers " Now go make
1894		 * it true. In fact there may be a number in the readers
1895		 * count but we know it is not true and will be fixed soon.
1896		 * We will fix the flags for the next pending entry in a
1897		 * moment.
1898		 */
1899	} else {
1900		/*
1901		 * We can't dequeue anything.. return and say so. Probably we
1902		 * have a write pending and the readers count is non zero. If
1903		 * we got here because a reader hit us just at the wrong
1904		 * moment with the fasttrack code, and put us in a strange
1905		 * state, then it will be coming through in just a moment,
1906		 * (just as soon as we release the mutex) and keep things
1907		 * moving.
1908		 * Make sure we remove ourselves from the work queue. It
1909		 * would be a waste of effort to do all this again.
1910		 */
1911		ng_worklist_remove(ngq->q_node);
1912		CTR4(KTR_NET, "%20s: node [%x] (%p) can't dequeue anything; "
1913		    "queue flags 0x%lx", __func__,
1914		    ngq->q_node->nd_ID, ngq->q_node, ngq->q_flags);
1915		return (NULL);
1916	}
1917
1918	/*
1919	 * Now we dequeue the request (whatever it may be) and correct the
1920	 * pending flags and the next and last pointers.
1921	 */
1922	item = ngq->queue;
1923	ngq->queue = item->el_next;
1924	CTR6(KTR_NET, "%20s: node [%x] (%p) dequeued item %p with flags 0x%lx; "
1925	    "queue flags 0x%lx", __func__,
1926	    ngq->q_node->nd_ID,ngq->q_node, item, item->el_flags, ngq->q_flags);
1927	if (ngq->last == &(item->el_next)) {
1928		/*
1929		 * that was the last entry in the queue so set the 'last
1930		 * pointer up correctly and make sure the pending flag is
1931		 * clear.
1932		 */
1933		add_arg += -OP_PENDING;
1934		ngq->last = &(ngq->queue);
1935		/*
1936		 * Whatever flag was set will be cleared and
1937		 * the new acive field will be set by the add as well,
1938		 * so we don't need to change add_arg.
1939		 * But we know we don't need to be on the work list.
1940		 */
1941		atomic_add_long(&ngq->q_flags, add_arg);
1942		ng_worklist_remove(ngq->q_node);
1943	} else {
1944		/*
1945		 * Since there is still something on the queue
1946		 * we don't need to change the PENDING flag.
1947		 */
1948		atomic_add_long(&ngq->q_flags, add_arg);
1949		/*
1950		 * If we see more doable work, make sure we are
1951		 * on the work queue.
1952		 */
1953		if (NEXT_QUEUED_ITEM_CAN_PROCEED(ngq)) {
1954			ng_setisr(ngq->q_node);
1955		}
1956	}
1957	CTR6(KTR_NET, "%20s: node [%x] (%p) returning item %p as %s; "
1958	    "queue flags 0x%lx", __func__,
1959	    ngq->q_node->nd_ID, ngq->q_node, item, *rw ? "WRITER" : "READER" ,
1960	    ngq->q_flags);
1961	return (item);
1962}
1963
1964/*
1965 * Queue a packet to be picked up by someone else.
1966 * We really don't care who, but we can't or don't want to hang around
1967 * to process it ourselves. We are probably an interrupt routine..
1968 * If the queue could be run, flag the netisr handler to start.
1969 */
1970static __inline void
1971ng_queue_rw(struct ng_queue * ngq, item_p  item, int rw)
1972{
1973	mtx_assert(&ngq->q_mtx, MA_OWNED);
1974
1975	if (rw == NGQRW_W)
1976		NGI_SET_WRITER(item);
1977	else
1978		NGI_SET_READER(item);
1979	item->el_next = NULL;	/* maybe not needed */
1980	*ngq->last = item;
1981	CTR5(KTR_NET, "%20s: node [%x] (%p) queued item %p as %s", __func__,
1982	    ngq->q_node->nd_ID, ngq->q_node, item, rw ? "WRITER" : "READER" );
1983	/*
1984	 * If it was the first item in the queue then we need to
1985	 * set the last pointer and the type flags.
1986	 */
1987	if (ngq->last == &(ngq->queue)) {
1988		atomic_add_long(&ngq->q_flags, OP_PENDING);
1989		CTR3(KTR_NET, "%20s: node [%x] (%p) set OP_PENDING", __func__,
1990		    ngq->q_node->nd_ID, ngq->q_node);
1991	}
1992
1993	ngq->last = &(item->el_next);
1994	/*
1995	 * We can take the worklist lock with the node locked
1996	 * BUT NOT THE REVERSE!
1997	 */
1998	if (NEXT_QUEUED_ITEM_CAN_PROCEED(ngq))
1999		ng_setisr(ngq->q_node);
2000}
2001
2002
2003/*
2004 * This function 'cheats' in that it first tries to 'grab' the use of the
2005 * node, without going through the mutex. We can do this becasue of the
2006 * semantics of the lock. The semantics include a clause that says that the
2007 * value of the readers count is invalid if the WRITER_ACTIVE flag is set. It
2008 * also says that the WRITER_ACTIVE flag cannot be set if the readers count
2009 * is not zero. Note that this talks about what is valid to SET the
2010 * WRITER_ACTIVE flag, because from the moment it is set, the value if the
2011 * reader count is immaterial, and not valid. The two 'pending' flags have a
2012 * similar effect, in that If they are orthogonal to the two active fields in
2013 * how they are set, but if either is set, the attempted 'grab' need to be
2014 * backed out because there is earlier work, and we maintain ordering in the
2015 * queue. The result of this is that the reader request can try obtain use of
2016 * the node with only a single atomic addition, and without any of the mutex
2017 * overhead. If this fails the operation degenerates to the same as for other
2018 * cases.
2019 *
2020 */
2021static __inline item_p
2022ng_acquire_read(struct ng_queue *ngq, item_p item)
2023{
2024	KASSERT(ngq != &ng_deadnode.nd_input_queue,
2025	    ("%s: working on deadnode", __func__));
2026
2027	/* ######### Hack alert ######### */
2028	atomic_add_long(&ngq->q_flags, READER_INCREMENT);
2029	if ((ngq->q_flags & NGQ_RMASK) == 0) {
2030		/* Successfully grabbed node */
2031		CTR4(KTR_NET, "%20s: node [%x] (%p) fast acquired item %p",
2032		    __func__, ngq->q_node->nd_ID, ngq->q_node, item);
2033		return (item);
2034	}
2035	/* undo the damage if we didn't succeed */
2036	atomic_subtract_long(&ngq->q_flags, READER_INCREMENT);
2037
2038	/* ######### End Hack alert ######### */
2039	mtx_lock_spin((&ngq->q_mtx));
2040	/*
2041	 * Try again. Another processor (or interrupt for that matter) may
2042	 * have removed the last queued item that was stopping us from
2043	 * running, between the previous test, and the moment that we took
2044	 * the mutex. (Or maybe a writer completed.)
2045	 * Even if another fast-track reader hits during this period
2046	 * we don't care as multiple readers is OK.
2047	 */
2048	if ((ngq->q_flags & NGQ_RMASK) == 0) {
2049		atomic_add_long(&ngq->q_flags, READER_INCREMENT);
2050		mtx_unlock_spin((&ngq->q_mtx));
2051		CTR4(KTR_NET, "%20s: node [%x] (%p) slow acquired item %p",
2052		    __func__, ngq->q_node->nd_ID, ngq->q_node, item);
2053		return (item);
2054	}
2055
2056	/*
2057	 * and queue the request for later.
2058	 */
2059	ng_queue_rw(ngq, item, NGQRW_R);
2060	mtx_unlock_spin(&(ngq->q_mtx));
2061
2062	return (NULL);
2063}
2064
2065static __inline item_p
2066ng_acquire_write(struct ng_queue *ngq, item_p item)
2067{
2068	KASSERT(ngq != &ng_deadnode.nd_input_queue,
2069	    ("%s: working on deadnode", __func__));
2070
2071restart:
2072	mtx_lock_spin(&(ngq->q_mtx));
2073	/*
2074	 * If there are no readers, no writer, and no pending packets, then
2075	 * we can just go ahead. In all other situations we need to queue the
2076	 * request
2077	 */
2078	if ((ngq->q_flags & NGQ_WMASK) == 0) {
2079		/* collision could happen *HERE* */
2080		atomic_add_long(&ngq->q_flags, WRITER_ACTIVE);
2081		mtx_unlock_spin((&ngq->q_mtx));
2082		if (ngq->q_flags & READER_MASK) {
2083			/* Collision with fast-track reader */
2084			atomic_subtract_long(&ngq->q_flags, WRITER_ACTIVE);
2085			goto restart;
2086		}
2087		CTR4(KTR_NET, "%20s: node [%x] (%p) acquired item %p",
2088		    __func__, ngq->q_node->nd_ID, ngq->q_node, item);
2089		return (item);
2090	}
2091
2092	/*
2093	 * and queue the request for later.
2094	 */
2095	ng_queue_rw(ngq, item, NGQRW_W);
2096	mtx_unlock_spin(&(ngq->q_mtx));
2097
2098	return (NULL);
2099}
2100
2101static __inline void
2102ng_leave_read(struct ng_queue *ngq)
2103{
2104	atomic_subtract_long(&ngq->q_flags, READER_INCREMENT);
2105}
2106
2107static __inline void
2108ng_leave_write(struct ng_queue *ngq)
2109{
2110	atomic_subtract_long(&ngq->q_flags, WRITER_ACTIVE);
2111}
2112
2113static void
2114ng_flush_input_queue(struct ng_queue * ngq)
2115{
2116	item_p item;
2117
2118	mtx_lock_spin(&ngq->q_mtx);
2119	while (ngq->queue) {
2120		item = ngq->queue;
2121		ngq->queue = item->el_next;
2122		if (ngq->last == &(item->el_next)) {
2123			ngq->last = &(ngq->queue);
2124			atomic_add_long(&ngq->q_flags, -OP_PENDING);
2125		}
2126		mtx_unlock_spin(&ngq->q_mtx);
2127
2128		/* If the item is supplying a callback, call it with an error */
2129		if (item->apply != NULL) {
2130			(item->apply)(item->context, ENOENT);
2131			item->apply = NULL;
2132		}
2133		NG_FREE_ITEM(item);
2134		mtx_lock_spin(&ngq->q_mtx);
2135	}
2136	/*
2137	 * Take us off the work queue if we are there.
2138	 * We definatly have no work to be done.
2139	 */
2140	ng_worklist_remove(ngq->q_node);
2141	mtx_unlock_spin(&ngq->q_mtx);
2142}
2143
2144/***********************************************************************
2145* Externally visible method for sending or queueing messages or data.
2146***********************************************************************/
2147
2148/*
2149 * The module code should have filled out the item correctly by this stage:
2150 * Common:
2151 *    reference to destination node.
2152 *    Reference to destination rcv hook if relevant.
2153 * Data:
2154 *    pointer to mbuf
2155 * Control_Message:
2156 *    pointer to msg.
2157 *    ID of original sender node. (return address)
2158 * Function:
2159 *    Function pointer
2160 *    void * argument
2161 *    integer argument
2162 *
2163 * The nodes have several routines and macros to help with this task:
2164 */
2165
2166int
2167ng_snd_item(item_p item, int flags)
2168{
2169	hook_p hook = NGI_HOOK(item);
2170	node_p node = NGI_NODE(item);
2171	int queue, rw;
2172	struct ng_queue * ngq = &node->nd_input_queue;
2173	int error = 0;
2174
2175#ifdef	NETGRAPH_DEBUG
2176	_ngi_check(item, __FILE__, __LINE__);
2177#endif
2178
2179	queue = (flags & NG_QUEUE) ? 1 : 0;
2180
2181	if (item == NULL) {
2182		TRAP_ERROR();
2183		return (EINVAL);	/* failed to get queue element */
2184	}
2185	if (node == NULL) {
2186		NG_FREE_ITEM(item);
2187		TRAP_ERROR();
2188		return (EINVAL);	/* No address */
2189	}
2190	switch(item->el_flags & NGQF_TYPE) {
2191	case NGQF_DATA:
2192		/*
2193		 * DATA MESSAGE
2194		 * Delivered to a node via a non-optional hook.
2195		 * Both should be present in the item even though
2196		 * the node is derivable from the hook.
2197		 * References are held on both by the item.
2198		 */
2199
2200		/* Protect nodes from sending NULL pointers
2201		 * to each other
2202		 */
2203		if (NGI_M(item) == NULL)
2204			return (EINVAL);
2205
2206		CHECK_DATA_MBUF(NGI_M(item));
2207		if (hook == NULL) {
2208			NG_FREE_ITEM(item);
2209			TRAP_ERROR();
2210			return(EINVAL);
2211		}
2212		if ((NG_HOOK_NOT_VALID(hook))
2213		|| (NG_NODE_NOT_VALID(NG_HOOK_NODE(hook)))) {
2214			NG_FREE_ITEM(item);
2215			return (ENOTCONN);
2216		}
2217		if ((hook->hk_flags & HK_QUEUE)) {
2218			queue = 1;
2219		}
2220		break;
2221	case NGQF_MESG:
2222		/*
2223		 * CONTROL MESSAGE
2224		 * Delivered to a node.
2225		 * Hook is optional.
2226		 * References are held by the item on the node and
2227		 * the hook if it is present.
2228		 */
2229		if (hook && (hook->hk_flags & HK_QUEUE)) {
2230			queue = 1;
2231		}
2232		break;
2233	case NGQF_FN:
2234		break;
2235	default:
2236		NG_FREE_ITEM(item);
2237		TRAP_ERROR();
2238		return (EINVAL);
2239	}
2240	switch(item->el_flags & NGQF_RW) {
2241	case NGQF_READER:
2242		rw = NGQRW_R;
2243		break;
2244	case NGQF_WRITER:
2245		rw = NGQRW_W;
2246		break;
2247	default:
2248		panic("%s: invalid item flags %lx", __func__, item->el_flags);
2249	}
2250
2251	/*
2252	 * If the node specifies single threading, force writer semantics.
2253	 * Similarly, the node may say one hook always produces writers.
2254	 * These are overrides.
2255	 */
2256	if ((node->nd_flags & NGF_FORCE_WRITER)
2257	    || (hook && (hook->hk_flags & HK_FORCE_WRITER)))
2258			rw = NGQRW_W;
2259
2260	if (queue) {
2261		/* Put it on the queue for that node*/
2262#ifdef	NETGRAPH_DEBUG
2263		_ngi_check(item, __FILE__, __LINE__);
2264#endif
2265		mtx_lock_spin(&(ngq->q_mtx));
2266		ng_queue_rw(ngq, item, rw);
2267		mtx_unlock_spin(&(ngq->q_mtx));
2268
2269		if (flags & NG_PROGRESS)
2270			return (EINPROGRESS);
2271		else
2272			return (0);
2273	}
2274
2275	/*
2276	 * We already decided how we will be queueud or treated.
2277	 * Try get the appropriate operating permission.
2278	 */
2279 	if (rw == NGQRW_R)
2280		item = ng_acquire_read(ngq, item);
2281	else
2282		item = ng_acquire_write(ngq, item);
2283
2284
2285	if (item == NULL) {
2286		if (flags & NG_PROGRESS)
2287			return (EINPROGRESS);
2288		else
2289			return (0);
2290	}
2291
2292#ifdef	NETGRAPH_DEBUG
2293	_ngi_check(item, __FILE__, __LINE__);
2294#endif
2295
2296	NGI_GET_NODE(item, node); /* zaps stored node */
2297
2298	error = ng_apply_item(node, item, rw); /* drops r/w lock when done */
2299
2300	/*
2301	 * If the node goes away when we remove the reference,
2302	 * whatever we just did caused it.. whatever we do, DO NOT
2303	 * access the node again!
2304	 */
2305	if (NG_NODE_UNREF(node) == 0) {
2306		return (error);
2307	}
2308
2309	mtx_lock_spin(&(ngq->q_mtx));
2310	if (NEXT_QUEUED_ITEM_CAN_PROCEED(ngq))
2311		ng_setisr(ngq->q_node);
2312	mtx_unlock_spin(&(ngq->q_mtx));
2313
2314	return (error);
2315}
2316
2317/*
2318 * We have an item that was possibly queued somewhere.
2319 * It should contain all the information needed
2320 * to run it on the appropriate node/hook.
2321 */
2322static int
2323ng_apply_item(node_p node, item_p item, int rw)
2324{
2325	hook_p  hook;
2326	int	error = 0;
2327	ng_rcvdata_t *rcvdata;
2328	ng_rcvmsg_t *rcvmsg;
2329	ng_apply_t *apply = NULL;
2330	void	*context = NULL;
2331
2332	NGI_GET_HOOK(item, hook); /* clears stored hook */
2333#ifdef	NETGRAPH_DEBUG
2334	_ngi_check(item, __FILE__, __LINE__);
2335#endif
2336
2337	/*
2338	 * If the item has an "apply" callback, store it.
2339	 * Clear item's callback immediately, to avoid an extra call if
2340	 * the item is reused by the destination node.
2341	 */
2342	if (item->apply != NULL) {
2343		apply = item->apply;
2344		context = item->context;
2345		item->apply = NULL;
2346	}
2347
2348	switch (item->el_flags & NGQF_TYPE) {
2349	case NGQF_DATA:
2350		/*
2351		 * Check things are still ok as when we were queued.
2352		 */
2353		if ((hook == NULL)
2354		|| NG_HOOK_NOT_VALID(hook)
2355		|| NG_NODE_NOT_VALID(node) ) {
2356			error = EIO;
2357			NG_FREE_ITEM(item);
2358			break;
2359		}
2360		/*
2361		 * If no receive method, just silently drop it.
2362		 * Give preference to the hook over-ride method
2363		 */
2364		if ((!(rcvdata = hook->hk_rcvdata))
2365		&& (!(rcvdata = NG_HOOK_NODE(hook)->nd_type->rcvdata))) {
2366			error = 0;
2367			NG_FREE_ITEM(item);
2368			break;
2369		}
2370		error = (*rcvdata)(hook, item);
2371		break;
2372	case NGQF_MESG:
2373		if (hook) {
2374			if (NG_HOOK_NOT_VALID(hook)) {
2375				/*
2376				 * The hook has been zapped then we can't
2377				 * use it. Immediatly drop its reference.
2378				 * The message may not need it.
2379				 */
2380				NG_HOOK_UNREF(hook);
2381				hook = NULL;
2382			}
2383		}
2384		/*
2385		 * Similarly, if the node is a zombie there is
2386		 * nothing we can do with it, drop everything.
2387		 */
2388		if (NG_NODE_NOT_VALID(node)) {
2389			TRAP_ERROR();
2390			error = EINVAL;
2391			NG_FREE_ITEM(item);
2392		} else {
2393			/*
2394			 * Call the appropriate message handler for the object.
2395			 * It is up to the message handler to free the message.
2396			 * If it's a generic message, handle it generically,
2397			 * otherwise call the type's message handler
2398			 * (if it exists)
2399			 * XXX (race). Remember that a queued message may
2400			 * reference a node or hook that has just been
2401			 * invalidated. It will exist as the queue code
2402			 * is holding a reference, but..
2403			 */
2404
2405			struct ng_mesg *msg = NGI_MSG(item);
2406
2407			/*
2408			 * check if the generic handler owns it.
2409			 */
2410			if ((msg->header.typecookie == NGM_GENERIC_COOKIE)
2411			&& ((msg->header.flags & NGF_RESP) == 0)) {
2412				error = ng_generic_msg(node, item, hook);
2413				break;
2414			}
2415			/*
2416			 * Now see if there is a handler (hook or node specific)
2417			 * in the target node. If none, silently discard.
2418			 */
2419			if (((!hook) || (!(rcvmsg = hook->hk_rcvmsg)))
2420			&& (!(rcvmsg = node->nd_type->rcvmsg))) {
2421				TRAP_ERROR();
2422				error = 0;
2423				NG_FREE_ITEM(item);
2424				break;
2425			}
2426			error = (*rcvmsg)(node, item, hook);
2427		}
2428		break;
2429	case NGQF_FN:
2430		/*
2431		 *  We have to implicitly trust the hook,
2432		 * as some of these are used for system purposes
2433		 * where the hook is invalid. In the case of
2434		 * the shutdown message we allow it to hit
2435		 * even if the node is invalid.
2436		 */
2437		if ((NG_NODE_NOT_VALID(node))
2438		&& (NGI_FN(item) != &ng_rmnode)) {
2439			TRAP_ERROR();
2440			error = EINVAL;
2441			NG_FREE_ITEM(item);
2442			break;
2443		}
2444		(*NGI_FN(item))(node, hook, NGI_ARG1(item), NGI_ARG2(item));
2445		NG_FREE_ITEM(item);
2446		break;
2447
2448	}
2449	/*
2450	 * We held references on some of the resources
2451	 * that we took from the item. Now that we have
2452	 * finished doing everything, drop those references.
2453	 */
2454	if (hook) {
2455		NG_HOOK_UNREF(hook);
2456	}
2457
2458 	if (rw == NGQRW_R) {
2459		ng_leave_read(&node->nd_input_queue);
2460	} else {
2461		ng_leave_write(&node->nd_input_queue);
2462	}
2463
2464	/* Apply callback. */
2465	if (apply != NULL)
2466		(*apply)(context, error);
2467
2468	return (error);
2469}
2470
2471/***********************************************************************
2472 * Implement the 'generic' control messages
2473 ***********************************************************************/
2474static int
2475ng_generic_msg(node_p here, item_p item, hook_p lasthook)
2476{
2477	int error = 0;
2478	struct ng_mesg *msg;
2479	struct ng_mesg *resp = NULL;
2480
2481	NGI_GET_MSG(item, msg);
2482	if (msg->header.typecookie != NGM_GENERIC_COOKIE) {
2483		TRAP_ERROR();
2484		error = EINVAL;
2485		goto out;
2486	}
2487	switch (msg->header.cmd) {
2488	case NGM_SHUTDOWN:
2489		ng_rmnode(here, NULL, NULL, 0);
2490		break;
2491	case NGM_MKPEER:
2492	    {
2493		struct ngm_mkpeer *const mkp = (struct ngm_mkpeer *) msg->data;
2494
2495		if (msg->header.arglen != sizeof(*mkp)) {
2496			TRAP_ERROR();
2497			error = EINVAL;
2498			break;
2499		}
2500		mkp->type[sizeof(mkp->type) - 1] = '\0';
2501		mkp->ourhook[sizeof(mkp->ourhook) - 1] = '\0';
2502		mkp->peerhook[sizeof(mkp->peerhook) - 1] = '\0';
2503		error = ng_mkpeer(here, mkp->ourhook, mkp->peerhook, mkp->type);
2504		break;
2505	    }
2506	case NGM_CONNECT:
2507	    {
2508		struct ngm_connect *const con =
2509			(struct ngm_connect *) msg->data;
2510		node_p node2;
2511
2512		if (msg->header.arglen != sizeof(*con)) {
2513			TRAP_ERROR();
2514			error = EINVAL;
2515			break;
2516		}
2517		con->path[sizeof(con->path) - 1] = '\0';
2518		con->ourhook[sizeof(con->ourhook) - 1] = '\0';
2519		con->peerhook[sizeof(con->peerhook) - 1] = '\0';
2520		/* Don't forget we get a reference.. */
2521		error = ng_path2noderef(here, con->path, &node2, NULL);
2522		if (error)
2523			break;
2524		error = ng_con_nodes(here, con->ourhook, node2, con->peerhook);
2525		NG_NODE_UNREF(node2);
2526		break;
2527	    }
2528	case NGM_NAME:
2529	    {
2530		struct ngm_name *const nam = (struct ngm_name *) msg->data;
2531
2532		if (msg->header.arglen != sizeof(*nam)) {
2533			TRAP_ERROR();
2534			error = EINVAL;
2535			break;
2536		}
2537		nam->name[sizeof(nam->name) - 1] = '\0';
2538		error = ng_name_node(here, nam->name);
2539		break;
2540	    }
2541	case NGM_RMHOOK:
2542	    {
2543		struct ngm_rmhook *const rmh = (struct ngm_rmhook *) msg->data;
2544		hook_p hook;
2545
2546		if (msg->header.arglen != sizeof(*rmh)) {
2547			TRAP_ERROR();
2548			error = EINVAL;
2549			break;
2550		}
2551		rmh->ourhook[sizeof(rmh->ourhook) - 1] = '\0';
2552		if ((hook = ng_findhook(here, rmh->ourhook)) != NULL)
2553			ng_destroy_hook(hook);
2554		break;
2555	    }
2556	case NGM_NODEINFO:
2557	    {
2558		struct nodeinfo *ni;
2559
2560		NG_MKRESPONSE(resp, msg, sizeof(*ni), M_NOWAIT);
2561		if (resp == NULL) {
2562			error = ENOMEM;
2563			break;
2564		}
2565
2566		/* Fill in node info */
2567		ni = (struct nodeinfo *) resp->data;
2568		if (NG_NODE_HAS_NAME(here))
2569			strcpy(ni->name, NG_NODE_NAME(here));
2570		strcpy(ni->type, here->nd_type->name);
2571		ni->id = ng_node2ID(here);
2572		ni->hooks = here->nd_numhooks;
2573		break;
2574	    }
2575	case NGM_LISTHOOKS:
2576	    {
2577		const int nhooks = here->nd_numhooks;
2578		struct hooklist *hl;
2579		struct nodeinfo *ni;
2580		hook_p hook;
2581
2582		/* Get response struct */
2583		NG_MKRESPONSE(resp, msg, sizeof(*hl)
2584		    + (nhooks * sizeof(struct linkinfo)), M_NOWAIT);
2585		if (resp == NULL) {
2586			error = ENOMEM;
2587			break;
2588		}
2589		hl = (struct hooklist *) resp->data;
2590		ni = &hl->nodeinfo;
2591
2592		/* Fill in node info */
2593		if (NG_NODE_HAS_NAME(here))
2594			strcpy(ni->name, NG_NODE_NAME(here));
2595		strcpy(ni->type, here->nd_type->name);
2596		ni->id = ng_node2ID(here);
2597
2598		/* Cycle through the linked list of hooks */
2599		ni->hooks = 0;
2600		LIST_FOREACH(hook, &here->nd_hooks, hk_hooks) {
2601			struct linkinfo *const link = &hl->link[ni->hooks];
2602
2603			if (ni->hooks >= nhooks) {
2604				log(LOG_ERR, "%s: number of %s changed\n",
2605				    __func__, "hooks");
2606				break;
2607			}
2608			if (NG_HOOK_NOT_VALID(hook))
2609				continue;
2610			strcpy(link->ourhook, NG_HOOK_NAME(hook));
2611			strcpy(link->peerhook, NG_PEER_HOOK_NAME(hook));
2612			if (NG_PEER_NODE_NAME(hook)[0] != '\0')
2613				strcpy(link->nodeinfo.name,
2614				    NG_PEER_NODE_NAME(hook));
2615			strcpy(link->nodeinfo.type,
2616			   NG_PEER_NODE(hook)->nd_type->name);
2617			link->nodeinfo.id = ng_node2ID(NG_PEER_NODE(hook));
2618			link->nodeinfo.hooks = NG_PEER_NODE(hook)->nd_numhooks;
2619			ni->hooks++;
2620		}
2621		break;
2622	    }
2623
2624	case NGM_LISTNAMES:
2625	case NGM_LISTNODES:
2626	    {
2627		const int unnamed = (msg->header.cmd == NGM_LISTNODES);
2628		struct namelist *nl;
2629		node_p node;
2630		int num = 0;
2631
2632		mtx_lock(&ng_nodelist_mtx);
2633		/* Count number of nodes */
2634		LIST_FOREACH(node, &ng_nodelist, nd_nodes) {
2635			if (NG_NODE_IS_VALID(node)
2636			&& (unnamed || NG_NODE_HAS_NAME(node))) {
2637				num++;
2638			}
2639		}
2640		mtx_unlock(&ng_nodelist_mtx);
2641
2642		/* Get response struct */
2643		NG_MKRESPONSE(resp, msg, sizeof(*nl)
2644		    + (num * sizeof(struct nodeinfo)), M_NOWAIT);
2645		if (resp == NULL) {
2646			error = ENOMEM;
2647			break;
2648		}
2649		nl = (struct namelist *) resp->data;
2650
2651		/* Cycle through the linked list of nodes */
2652		nl->numnames = 0;
2653		mtx_lock(&ng_nodelist_mtx);
2654		LIST_FOREACH(node, &ng_nodelist, nd_nodes) {
2655			struct nodeinfo *const np = &nl->nodeinfo[nl->numnames];
2656
2657			if (NG_NODE_NOT_VALID(node))
2658				continue;
2659			if (!unnamed && (! NG_NODE_HAS_NAME(node)))
2660				continue;
2661			if (nl->numnames >= num) {
2662				log(LOG_ERR, "%s: number of %s changed\n",
2663				    __func__, "nodes");
2664				break;
2665			}
2666			if (NG_NODE_HAS_NAME(node))
2667				strcpy(np->name, NG_NODE_NAME(node));
2668			strcpy(np->type, node->nd_type->name);
2669			np->id = ng_node2ID(node);
2670			np->hooks = node->nd_numhooks;
2671			nl->numnames++;
2672		}
2673		mtx_unlock(&ng_nodelist_mtx);
2674		break;
2675	    }
2676
2677	case NGM_LISTTYPES:
2678	    {
2679		struct typelist *tl;
2680		struct ng_type *type;
2681		int num = 0;
2682
2683		mtx_lock(&ng_typelist_mtx);
2684		/* Count number of types */
2685		LIST_FOREACH(type, &ng_typelist, types) {
2686			num++;
2687		}
2688		mtx_unlock(&ng_typelist_mtx);
2689
2690		/* Get response struct */
2691		NG_MKRESPONSE(resp, msg, sizeof(*tl)
2692		    + (num * sizeof(struct typeinfo)), M_NOWAIT);
2693		if (resp == NULL) {
2694			error = ENOMEM;
2695			break;
2696		}
2697		tl = (struct typelist *) resp->data;
2698
2699		/* Cycle through the linked list of types */
2700		tl->numtypes = 0;
2701		mtx_lock(&ng_typelist_mtx);
2702		LIST_FOREACH(type, &ng_typelist, types) {
2703			struct typeinfo *const tp = &tl->typeinfo[tl->numtypes];
2704
2705			if (tl->numtypes >= num) {
2706				log(LOG_ERR, "%s: number of %s changed\n",
2707				    __func__, "types");
2708				break;
2709			}
2710			strcpy(tp->type_name, type->name);
2711			tp->numnodes = type->refs - 1; /* don't count list */
2712			tl->numtypes++;
2713		}
2714		mtx_unlock(&ng_typelist_mtx);
2715		break;
2716	    }
2717
2718	case NGM_BINARY2ASCII:
2719	    {
2720		int bufSize = 20 * 1024;	/* XXX hard coded constant */
2721		const struct ng_parse_type *argstype;
2722		const struct ng_cmdlist *c;
2723		struct ng_mesg *binary, *ascii;
2724
2725		/* Data area must contain a valid netgraph message */
2726		binary = (struct ng_mesg *)msg->data;
2727		if (msg->header.arglen < sizeof(struct ng_mesg) ||
2728		    (msg->header.arglen - sizeof(struct ng_mesg) <
2729		    binary->header.arglen)) {
2730			TRAP_ERROR();
2731			error = EINVAL;
2732			break;
2733		}
2734
2735		/* Get a response message with lots of room */
2736		NG_MKRESPONSE(resp, msg, sizeof(*ascii) + bufSize, M_NOWAIT);
2737		if (resp == NULL) {
2738			error = ENOMEM;
2739			break;
2740		}
2741		ascii = (struct ng_mesg *)resp->data;
2742
2743		/* Copy binary message header to response message payload */
2744		bcopy(binary, ascii, sizeof(*binary));
2745
2746		/* Find command by matching typecookie and command number */
2747		for (c = here->nd_type->cmdlist;
2748		    c != NULL && c->name != NULL; c++) {
2749			if (binary->header.typecookie == c->cookie
2750			    && binary->header.cmd == c->cmd)
2751				break;
2752		}
2753		if (c == NULL || c->name == NULL) {
2754			for (c = ng_generic_cmds; c->name != NULL; c++) {
2755				if (binary->header.typecookie == c->cookie
2756				    && binary->header.cmd == c->cmd)
2757					break;
2758			}
2759			if (c->name == NULL) {
2760				NG_FREE_MSG(resp);
2761				error = ENOSYS;
2762				break;
2763			}
2764		}
2765
2766		/* Convert command name to ASCII */
2767		snprintf(ascii->header.cmdstr, sizeof(ascii->header.cmdstr),
2768		    "%s", c->name);
2769
2770		/* Convert command arguments to ASCII */
2771		argstype = (binary->header.flags & NGF_RESP) ?
2772		    c->respType : c->mesgType;
2773		if (argstype == NULL) {
2774			*ascii->data = '\0';
2775		} else {
2776			if ((error = ng_unparse(argstype,
2777			    (u_char *)binary->data,
2778			    ascii->data, bufSize)) != 0) {
2779				NG_FREE_MSG(resp);
2780				break;
2781			}
2782		}
2783
2784		/* Return the result as struct ng_mesg plus ASCII string */
2785		bufSize = strlen(ascii->data) + 1;
2786		ascii->header.arglen = bufSize;
2787		resp->header.arglen = sizeof(*ascii) + bufSize;
2788		break;
2789	    }
2790
2791	case NGM_ASCII2BINARY:
2792	    {
2793		int bufSize = 2000;	/* XXX hard coded constant */
2794		const struct ng_cmdlist *c;
2795		const struct ng_parse_type *argstype;
2796		struct ng_mesg *ascii, *binary;
2797		int off = 0;
2798
2799		/* Data area must contain at least a struct ng_mesg + '\0' */
2800		ascii = (struct ng_mesg *)msg->data;
2801		if ((msg->header.arglen < sizeof(*ascii) + 1) ||
2802		    (ascii->header.arglen < 1) ||
2803		    (msg->header.arglen < sizeof(*ascii) +
2804		    ascii->header.arglen)) {
2805			TRAP_ERROR();
2806			error = EINVAL;
2807			break;
2808		}
2809		ascii->data[ascii->header.arglen - 1] = '\0';
2810
2811		/* Get a response message with lots of room */
2812		NG_MKRESPONSE(resp, msg, sizeof(*binary) + bufSize, M_NOWAIT);
2813		if (resp == NULL) {
2814			error = ENOMEM;
2815			break;
2816		}
2817		binary = (struct ng_mesg *)resp->data;
2818
2819		/* Copy ASCII message header to response message payload */
2820		bcopy(ascii, binary, sizeof(*ascii));
2821
2822		/* Find command by matching ASCII command string */
2823		for (c = here->nd_type->cmdlist;
2824		    c != NULL && c->name != NULL; c++) {
2825			if (strcmp(ascii->header.cmdstr, c->name) == 0)
2826				break;
2827		}
2828		if (c == NULL || c->name == NULL) {
2829			for (c = ng_generic_cmds; c->name != NULL; c++) {
2830				if (strcmp(ascii->header.cmdstr, c->name) == 0)
2831					break;
2832			}
2833			if (c->name == NULL) {
2834				NG_FREE_MSG(resp);
2835				error = ENOSYS;
2836				break;
2837			}
2838		}
2839
2840		/* Convert command name to binary */
2841		binary->header.cmd = c->cmd;
2842		binary->header.typecookie = c->cookie;
2843
2844		/* Convert command arguments to binary */
2845		argstype = (binary->header.flags & NGF_RESP) ?
2846		    c->respType : c->mesgType;
2847		if (argstype == NULL) {
2848			bufSize = 0;
2849		} else {
2850			if ((error = ng_parse(argstype, ascii->data,
2851			    &off, (u_char *)binary->data, &bufSize)) != 0) {
2852				NG_FREE_MSG(resp);
2853				break;
2854			}
2855		}
2856
2857		/* Return the result */
2858		binary->header.arglen = bufSize;
2859		resp->header.arglen = sizeof(*binary) + bufSize;
2860		break;
2861	    }
2862
2863	case NGM_TEXT_CONFIG:
2864	case NGM_TEXT_STATUS:
2865		/*
2866		 * This one is tricky as it passes the command down to the
2867		 * actual node, even though it is a generic type command.
2868		 * This means we must assume that the item/msg is already freed
2869		 * when control passes back to us.
2870		 */
2871		if (here->nd_type->rcvmsg != NULL) {
2872			NGI_MSG(item) = msg; /* put it back as we found it */
2873			return((*here->nd_type->rcvmsg)(here, item, lasthook));
2874		}
2875		/* Fall through if rcvmsg not supported */
2876	default:
2877		TRAP_ERROR();
2878		error = EINVAL;
2879	}
2880	/*
2881	 * Sometimes a generic message may be statically allocated
2882	 * to avoid problems with allocating when in tight memeory situations.
2883	 * Don't free it if it is so.
2884	 * I break them appart here, because erros may cause a free if the item
2885	 * in which case we'd be doing it twice.
2886	 * they are kept together above, to simplify freeing.
2887	 */
2888out:
2889	NG_RESPOND_MSG(error, here, item, resp);
2890	if (msg)
2891		NG_FREE_MSG(msg);
2892	return (error);
2893}
2894
2895/************************************************************************
2896			Queue element get/free routines
2897************************************************************************/
2898
2899uma_zone_t			ng_qzone;
2900static int			maxalloc = 512;	/* limit the damage of a leak */
2901
2902TUNABLE_INT("net.graph.maxalloc", &maxalloc);
2903SYSCTL_INT(_net_graph, OID_AUTO, maxalloc, CTLFLAG_RDTUN, &maxalloc,
2904    0, "Maximum number of queue items to allocate");
2905
2906#ifdef	NETGRAPH_DEBUG
2907static TAILQ_HEAD(, ng_item) ng_itemlist = TAILQ_HEAD_INITIALIZER(ng_itemlist);
2908static int			allocated;	/* number of items malloc'd */
2909#endif
2910
2911/*
2912 * Get a queue entry.
2913 * This is usually called when a packet first enters netgraph.
2914 * By definition, this is usually from an interrupt, or from a user.
2915 * Users are not so important, but try be quick for the times that it's
2916 * an interrupt.
2917 */
2918static __inline item_p
2919ng_getqblk(int flags)
2920{
2921	item_p item = NULL;
2922	int wait;
2923
2924	wait = (flags & NG_WAITOK) ? M_WAITOK : M_NOWAIT;
2925
2926	item = uma_zalloc(ng_qzone, wait | M_ZERO);
2927
2928#ifdef	NETGRAPH_DEBUG
2929	if (item) {
2930			mtx_lock(&ngq_mtx);
2931			TAILQ_INSERT_TAIL(&ng_itemlist, item, all);
2932			allocated++;
2933			mtx_unlock(&ngq_mtx);
2934	}
2935#endif
2936
2937	return (item);
2938}
2939
2940/*
2941 * Release a queue entry
2942 */
2943void
2944ng_free_item(item_p item)
2945{
2946	KASSERT(item->apply == NULL, ("%s: leaking apply callback", __func__));
2947
2948	/*
2949	 * The item may hold resources on it's own. We need to free
2950	 * these before we can free the item. What they are depends upon
2951	 * what kind of item it is. it is important that nodes zero
2952	 * out pointers to resources that they remove from the item
2953	 * or we release them again here.
2954	 */
2955	switch (item->el_flags & NGQF_TYPE) {
2956	case NGQF_DATA:
2957		/* If we have an mbuf still attached.. */
2958		NG_FREE_M(_NGI_M(item));
2959		break;
2960	case NGQF_MESG:
2961		_NGI_RETADDR(item) = 0;
2962		NG_FREE_MSG(_NGI_MSG(item));
2963		break;
2964	case NGQF_FN:
2965		/* nothing to free really, */
2966		_NGI_FN(item) = NULL;
2967		_NGI_ARG1(item) = NULL;
2968		_NGI_ARG2(item) = 0;
2969	case NGQF_UNDEF:
2970		break;
2971	}
2972	/* If we still have a node or hook referenced... */
2973	_NGI_CLR_NODE(item);
2974	_NGI_CLR_HOOK(item);
2975
2976#ifdef	NETGRAPH_DEBUG
2977	mtx_lock(&ngq_mtx);
2978	TAILQ_REMOVE(&ng_itemlist, item, all);
2979	allocated--;
2980	mtx_unlock(&ngq_mtx);
2981#endif
2982	uma_zfree(ng_qzone, item);
2983}
2984
2985/************************************************************************
2986			Module routines
2987************************************************************************/
2988
2989/*
2990 * Handle the loading/unloading of a netgraph node type module
2991 */
2992int
2993ng_mod_event(module_t mod, int event, void *data)
2994{
2995	struct ng_type *const type = data;
2996	int s, error = 0;
2997
2998	switch (event) {
2999	case MOD_LOAD:
3000
3001		/* Register new netgraph node type */
3002		s = splnet();
3003		if ((error = ng_newtype(type)) != 0) {
3004			splx(s);
3005			break;
3006		}
3007
3008		/* Call type specific code */
3009		if (type->mod_event != NULL)
3010			if ((error = (*type->mod_event)(mod, event, data))) {
3011				mtx_lock(&ng_typelist_mtx);
3012				type->refs--;	/* undo it */
3013				LIST_REMOVE(type, types);
3014				mtx_unlock(&ng_typelist_mtx);
3015			}
3016		splx(s);
3017		break;
3018
3019	case MOD_UNLOAD:
3020		s = splnet();
3021		if (type->refs > 1) {		/* make sure no nodes exist! */
3022			error = EBUSY;
3023		} else {
3024			if (type->refs == 0) {
3025				/* failed load, nothing to undo */
3026				splx(s);
3027				break;
3028			}
3029			if (type->mod_event != NULL) {	/* check with type */
3030				error = (*type->mod_event)(mod, event, data);
3031				if (error != 0) {	/* type refuses.. */
3032					splx(s);
3033					break;
3034				}
3035			}
3036			mtx_lock(&ng_typelist_mtx);
3037			LIST_REMOVE(type, types);
3038			mtx_unlock(&ng_typelist_mtx);
3039		}
3040		splx(s);
3041		break;
3042
3043	default:
3044		if (type->mod_event != NULL)
3045			error = (*type->mod_event)(mod, event, data);
3046		else
3047			error = EOPNOTSUPP;		/* XXX ? */
3048		break;
3049	}
3050	return (error);
3051}
3052
3053/*
3054 * Handle loading and unloading for this code.
3055 * The only thing we need to link into is the NETISR strucure.
3056 */
3057static int
3058ngb_mod_event(module_t mod, int event, void *data)
3059{
3060	int error = 0;
3061
3062	switch (event) {
3063	case MOD_LOAD:
3064		/* Initialize everything. */
3065		mtx_init(&ng_worklist_mtx, "ng_worklist", NULL, MTX_SPIN);
3066		mtx_init(&ng_typelist_mtx, "netgraph types mutex", NULL,
3067		    MTX_DEF);
3068		mtx_init(&ng_nodelist_mtx, "netgraph nodelist mutex", NULL,
3069		    MTX_DEF);
3070		mtx_init(&ng_idhash_mtx, "netgraph idhash mutex", NULL,
3071		    MTX_DEF);
3072		mtx_init(&ng_topo_mtx, "netgraph topology mutex", NULL,
3073		    MTX_DEF);
3074#ifdef	NETGRAPH_DEBUG
3075		mtx_init(&ngq_mtx, "netgraph item list mutex", NULL,
3076		    MTX_DEF);
3077#endif
3078		ng_qzone = uma_zcreate("NetGraph items", sizeof(struct ng_item),
3079		    NULL, NULL, NULL, NULL, UMA_ALIGN_CACHE, 0);
3080		uma_zone_set_max(ng_qzone, maxalloc);
3081		netisr_register(NETISR_NETGRAPH, (netisr_t *)ngintr, NULL,
3082		    NETISR_MPSAFE);
3083		break;
3084	case MOD_UNLOAD:
3085		/* You cant unload it because an interface may be using it.  */
3086		error = EBUSY;
3087		break;
3088	default:
3089		error = EOPNOTSUPP;
3090		break;
3091	}
3092	return (error);
3093}
3094
3095static moduledata_t netgraph_mod = {
3096	"netgraph",
3097	ngb_mod_event,
3098	(NULL)
3099};
3100DECLARE_MODULE(netgraph, netgraph_mod, SI_SUB_NETGRAPH, SI_ORDER_MIDDLE);
3101SYSCTL_NODE(_net, OID_AUTO, graph, CTLFLAG_RW, 0, "netgraph Family");
3102SYSCTL_INT(_net_graph, OID_AUTO, abi_version, CTLFLAG_RD, 0, NG_ABI_VERSION,"");
3103SYSCTL_INT(_net_graph, OID_AUTO, msg_version, CTLFLAG_RD, 0, NG_VERSION, "");
3104
3105#ifdef	NETGRAPH_DEBUG
3106void
3107dumphook (hook_p hook, char *file, int line)
3108{
3109	printf("hook: name %s, %d refs, Last touched:\n",
3110		_NG_HOOK_NAME(hook), hook->hk_refs);
3111	printf("	Last active @ %s, line %d\n",
3112		hook->lastfile, hook->lastline);
3113	if (line) {
3114		printf(" problem discovered at file %s, line %d\n", file, line);
3115	}
3116}
3117
3118void
3119dumpnode(node_p node, char *file, int line)
3120{
3121	printf("node: ID [%x]: type '%s', %d hooks, flags 0x%x, %d refs, %s:\n",
3122		_NG_NODE_ID(node), node->nd_type->name,
3123		node->nd_numhooks, node->nd_flags,
3124		node->nd_refs, node->nd_name);
3125	printf("	Last active @ %s, line %d\n",
3126		node->lastfile, node->lastline);
3127	if (line) {
3128		printf(" problem discovered at file %s, line %d\n", file, line);
3129	}
3130}
3131
3132void
3133dumpitem(item_p item, char *file, int line)
3134{
3135	printf(" ACTIVE item, last used at %s, line %d",
3136		item->lastfile, item->lastline);
3137	switch(item->el_flags & NGQF_TYPE) {
3138	case NGQF_DATA:
3139		printf(" - [data]\n");
3140		break;
3141	case NGQF_MESG:
3142		printf(" - retaddr[%d]:\n", _NGI_RETADDR(item));
3143		break;
3144	case NGQF_FN:
3145		printf(" - fn@%p (%p, %p, %p, %d (%x))\n",
3146			item->body.fn.fn_fn,
3147			_NGI_NODE(item),
3148			_NGI_HOOK(item),
3149			item->body.fn.fn_arg1,
3150			item->body.fn.fn_arg2,
3151			item->body.fn.fn_arg2);
3152		break;
3153	case NGQF_UNDEF:
3154		printf(" - UNDEFINED!\n");
3155	}
3156	if (line) {
3157		printf(" problem discovered at file %s, line %d\n", file, line);
3158		if (_NGI_NODE(item)) {
3159			printf("node %p ([%x])\n",
3160				_NGI_NODE(item), ng_node2ID(_NGI_NODE(item)));
3161		}
3162	}
3163}
3164
3165static void
3166ng_dumpitems(void)
3167{
3168	item_p item;
3169	int i = 1;
3170	TAILQ_FOREACH(item, &ng_itemlist, all) {
3171		printf("[%d] ", i++);
3172		dumpitem(item, NULL, 0);
3173	}
3174}
3175
3176static void
3177ng_dumpnodes(void)
3178{
3179	node_p node;
3180	int i = 1;
3181	mtx_lock(&ng_nodelist_mtx);
3182	SLIST_FOREACH(node, &ng_allnodes, nd_all) {
3183		printf("[%d] ", i++);
3184		dumpnode(node, NULL, 0);
3185	}
3186	mtx_unlock(&ng_nodelist_mtx);
3187}
3188
3189static void
3190ng_dumphooks(void)
3191{
3192	hook_p hook;
3193	int i = 1;
3194	mtx_lock(&ng_nodelist_mtx);
3195	SLIST_FOREACH(hook, &ng_allhooks, hk_all) {
3196		printf("[%d] ", i++);
3197		dumphook(hook, NULL, 0);
3198	}
3199	mtx_unlock(&ng_nodelist_mtx);
3200}
3201
3202static int
3203sysctl_debug_ng_dump_items(SYSCTL_HANDLER_ARGS)
3204{
3205	int error;
3206	int val;
3207	int i;
3208
3209	val = allocated;
3210	i = 1;
3211	error = sysctl_handle_int(oidp, &val, sizeof(int), req);
3212	if (error != 0 || req->newptr == NULL)
3213		return (error);
3214	if (val == 42) {
3215		ng_dumpitems();
3216		ng_dumpnodes();
3217		ng_dumphooks();
3218	}
3219	return (0);
3220}
3221
3222SYSCTL_PROC(_debug, OID_AUTO, ng_dump_items, CTLTYPE_INT | CTLFLAG_RW,
3223    0, sizeof(int), sysctl_debug_ng_dump_items, "I", "Number of allocated items");
3224#endif	/* NETGRAPH_DEBUG */
3225
3226
3227/***********************************************************************
3228* Worklist routines
3229**********************************************************************/
3230/* NETISR thread enters here */
3231/*
3232 * Pick a node off the list of nodes with work,
3233 * try get an item to process off it.
3234 * If there are no more, remove the node from the list.
3235 */
3236static void
3237ngintr(void)
3238{
3239	item_p item;
3240	node_p  node = NULL;
3241
3242	for (;;) {
3243		mtx_lock_spin(&ng_worklist_mtx);
3244		node = TAILQ_FIRST(&ng_worklist);
3245		if (!node) {
3246			mtx_unlock_spin(&ng_worklist_mtx);
3247			break;
3248		}
3249		node->nd_flags &= ~NGF_WORKQ;
3250		TAILQ_REMOVE(&ng_worklist, node, nd_work);
3251		mtx_unlock_spin(&ng_worklist_mtx);
3252		CTR3(KTR_NET, "%20s: node [%x] (%p) taken off worklist",
3253		    __func__, node->nd_ID, node);
3254		/*
3255		 * We have the node. We also take over the reference
3256		 * that the list had on it.
3257		 * Now process as much as you can, until it won't
3258		 * let you have another item off the queue.
3259		 * All this time, keep the reference
3260		 * that lets us be sure that the node still exists.
3261		 * Let the reference go at the last minute.
3262		 * ng_dequeue will put us back on the worklist
3263		 * if there is more too do. This may be of use if there
3264		 * are Multiple Processors and multiple Net threads in the
3265		 * future.
3266		 */
3267		for (;;) {
3268			int rw;
3269
3270			mtx_lock_spin(&node->nd_input_queue.q_mtx);
3271			item = ng_dequeue(&node->nd_input_queue, &rw);
3272			if (item == NULL) {
3273				mtx_unlock_spin(&node->nd_input_queue.q_mtx);
3274				break; /* go look for another node */
3275			} else {
3276				mtx_unlock_spin(&node->nd_input_queue.q_mtx);
3277				NGI_GET_NODE(item, node); /* zaps stored node */
3278				ng_apply_item(node, item, rw);
3279				NG_NODE_UNREF(node);
3280			}
3281		}
3282		NG_NODE_UNREF(node);
3283	}
3284}
3285
3286static void
3287ng_worklist_remove(node_p node)
3288{
3289	mtx_assert(&node->nd_input_queue.q_mtx, MA_OWNED);
3290
3291	mtx_lock_spin(&ng_worklist_mtx);
3292	if (node->nd_flags & NGF_WORKQ) {
3293		node->nd_flags &= ~NGF_WORKQ;
3294		TAILQ_REMOVE(&ng_worklist, node, nd_work);
3295		mtx_unlock_spin(&ng_worklist_mtx);
3296		NG_NODE_UNREF(node);
3297		CTR3(KTR_NET, "%20s: node [%x] (%p) removed from worklist",
3298		    __func__, node->nd_ID, node);
3299	} else {
3300		mtx_unlock_spin(&ng_worklist_mtx);
3301	}
3302}
3303
3304/*
3305 * XXX
3306 * It's posible that a debugging NG_NODE_REF may need
3307 * to be outside the mutex zone
3308 */
3309static void
3310ng_setisr(node_p node)
3311{
3312
3313	mtx_assert(&node->nd_input_queue.q_mtx, MA_OWNED);
3314
3315	if ((node->nd_flags & NGF_WORKQ) == 0) {
3316		/*
3317		 * If we are not already on the work queue,
3318		 * then put us on.
3319		 */
3320		node->nd_flags |= NGF_WORKQ;
3321		mtx_lock_spin(&ng_worklist_mtx);
3322		TAILQ_INSERT_TAIL(&ng_worklist, node, nd_work);
3323		mtx_unlock_spin(&ng_worklist_mtx);
3324		NG_NODE_REF(node); /* XXX fafe in mutex? */
3325		CTR3(KTR_NET, "%20s: node [%x] (%p) put on worklist", __func__,
3326		    node->nd_ID, node);
3327	} else
3328		CTR3(KTR_NET, "%20s: node [%x] (%p) already on worklist",
3329		    __func__, node->nd_ID, node);
3330	schednetisr(NETISR_NETGRAPH);
3331}
3332
3333
3334/***********************************************************************
3335* Externally useable functions to set up a queue item ready for sending
3336***********************************************************************/
3337
3338#ifdef	NETGRAPH_DEBUG
3339#define	ITEM_DEBUG_CHECKS						\
3340	do {								\
3341		if (NGI_NODE(item) ) {					\
3342			printf("item already has node");		\
3343			kdb_enter("has node");				\
3344			NGI_CLR_NODE(item);				\
3345		}							\
3346		if (NGI_HOOK(item) ) {					\
3347			printf("item already has hook");		\
3348			kdb_enter("has hook");				\
3349			NGI_CLR_HOOK(item);				\
3350		}							\
3351	} while (0)
3352#else
3353#define ITEM_DEBUG_CHECKS
3354#endif
3355
3356/*
3357 * Put mbuf into the item.
3358 * Hook and node references will be removed when the item is dequeued.
3359 * (or equivalent)
3360 * (XXX) Unsafe because no reference held by peer on remote node.
3361 * remote node might go away in this timescale.
3362 * We know the hooks can't go away because that would require getting
3363 * a writer item on both nodes and we must have at least a  reader
3364 * here to be able to do this.
3365 * Note that the hook loaded is the REMOTE hook.
3366 *
3367 * This is possibly in the critical path for new data.
3368 */
3369item_p
3370ng_package_data(struct mbuf *m, int flags)
3371{
3372	item_p item;
3373
3374	if ((item = ng_getqblk(flags)) == NULL) {
3375		NG_FREE_M(m);
3376		return (NULL);
3377	}
3378	ITEM_DEBUG_CHECKS;
3379	item->el_flags = NGQF_DATA | NGQF_READER;
3380	item->el_next = NULL;
3381	NGI_M(item) = m;
3382	return (item);
3383}
3384
3385/*
3386 * Allocate a queue item and put items into it..
3387 * Evaluate the address as this will be needed to queue it and
3388 * to work out what some of the fields should be.
3389 * Hook and node references will be removed when the item is dequeued.
3390 * (or equivalent)
3391 */
3392item_p
3393ng_package_msg(struct ng_mesg *msg, int flags)
3394{
3395	item_p item;
3396
3397	if ((item = ng_getqblk(flags)) == NULL) {
3398		NG_FREE_MSG(msg);
3399		return (NULL);
3400	}
3401	ITEM_DEBUG_CHECKS;
3402	/* Messages items count as writers unless explicitly exempted. */
3403	if (msg->header.cmd & NGM_READONLY)
3404		item->el_flags = NGQF_MESG | NGQF_READER;
3405	else
3406		item->el_flags = NGQF_MESG | NGQF_WRITER;
3407	item->el_next = NULL;
3408	/*
3409	 * Set the current lasthook into the queue item
3410	 */
3411	NGI_MSG(item) = msg;
3412	NGI_RETADDR(item) = 0;
3413	return (item);
3414}
3415
3416
3417
3418#define SET_RETADDR(item, here, retaddr)				\
3419	do {	/* Data or fn items don't have retaddrs */		\
3420		if ((item->el_flags & NGQF_TYPE) == NGQF_MESG) {	\
3421			if (retaddr) {					\
3422				NGI_RETADDR(item) = retaddr;		\
3423			} else {					\
3424				/*					\
3425				 * The old return address should be ok.	\
3426				 * If there isn't one, use the address	\
3427				 * here.				\
3428				 */					\
3429				if (NGI_RETADDR(item) == 0) {		\
3430					NGI_RETADDR(item)		\
3431						= ng_node2ID(here);	\
3432				}					\
3433			}						\
3434		}							\
3435	} while (0)
3436
3437int
3438ng_address_hook(node_p here, item_p item, hook_p hook, ng_ID_t retaddr)
3439{
3440	hook_p peer;
3441	node_p peernode;
3442	ITEM_DEBUG_CHECKS;
3443	/*
3444	 * Quick sanity check..
3445	 * Since a hook holds a reference on it's node, once we know
3446	 * that the peer is still connected (even if invalid,) we know
3447	 * that the peer node is present, though maybe invalid.
3448	 */
3449	if ((hook == NULL)
3450	|| NG_HOOK_NOT_VALID(hook)
3451	|| (NG_HOOK_PEER(hook) == NULL)
3452	|| NG_HOOK_NOT_VALID(NG_HOOK_PEER(hook))
3453	|| NG_NODE_NOT_VALID(NG_PEER_NODE(hook))) {
3454		NG_FREE_ITEM(item);
3455		TRAP_ERROR();
3456		return (ENETDOWN);
3457	}
3458
3459	/*
3460	 * Transfer our interest to the other (peer) end.
3461	 */
3462	peer = NG_HOOK_PEER(hook);
3463	NG_HOOK_REF(peer);
3464	NGI_SET_HOOK(item, peer);
3465	peernode = NG_PEER_NODE(hook);
3466	NG_NODE_REF(peernode);
3467	NGI_SET_NODE(item, peernode);
3468	SET_RETADDR(item, here, retaddr);
3469	return (0);
3470}
3471
3472int
3473ng_address_path(node_p here, item_p item, char *address, ng_ID_t retaddr)
3474{
3475	node_p	dest = NULL;
3476	hook_p	hook = NULL;
3477	int	error;
3478
3479	ITEM_DEBUG_CHECKS;
3480	/*
3481	 * Note that ng_path2noderef increments the reference count
3482	 * on the node for us if it finds one. So we don't have to.
3483	 */
3484	error = ng_path2noderef(here, address, &dest, &hook);
3485	if (error) {
3486		NG_FREE_ITEM(item);
3487		return (error);
3488	}
3489	NGI_SET_NODE(item, dest);
3490	if ( hook) {
3491		NG_HOOK_REF(hook);	/* don't let it go while on the queue */
3492		NGI_SET_HOOK(item, hook);
3493	}
3494	SET_RETADDR(item, here, retaddr);
3495	return (0);
3496}
3497
3498int
3499ng_address_ID(node_p here, item_p item, ng_ID_t ID, ng_ID_t retaddr)
3500{
3501	node_p dest;
3502
3503	ITEM_DEBUG_CHECKS;
3504	/*
3505	 * Find the target node.
3506	 */
3507	dest = ng_ID2noderef(ID); /* GETS REFERENCE! */
3508	if (dest == NULL) {
3509		NG_FREE_ITEM(item);
3510		TRAP_ERROR();
3511		return(EINVAL);
3512	}
3513	/* Fill out the contents */
3514	NGI_SET_NODE(item, dest);
3515	NGI_CLR_HOOK(item);
3516	SET_RETADDR(item, here, retaddr);
3517	return (0);
3518}
3519
3520/*
3521 * special case to send a message to self (e.g. destroy node)
3522 * Possibly indicate an arrival hook too.
3523 * Useful for removing that hook :-)
3524 */
3525item_p
3526ng_package_msg_self(node_p here, hook_p hook, struct ng_mesg *msg)
3527{
3528	item_p item;
3529
3530	/*
3531	 * Find the target node.
3532	 * If there is a HOOK argument, then use that in preference
3533	 * to the address.
3534	 */
3535	if ((item = ng_getqblk(NG_NOFLAGS)) == NULL) {
3536		NG_FREE_MSG(msg);
3537		return (NULL);
3538	}
3539
3540	/* Fill out the contents */
3541	item->el_flags = NGQF_MESG | NGQF_WRITER;
3542	item->el_next = NULL;
3543	NG_NODE_REF(here);
3544	NGI_SET_NODE(item, here);
3545	if (hook) {
3546		NG_HOOK_REF(hook);
3547		NGI_SET_HOOK(item, hook);
3548	}
3549	NGI_MSG(item) = msg;
3550	NGI_RETADDR(item) = ng_node2ID(here);
3551	return (item);
3552}
3553
3554int
3555ng_send_fn1(node_p node, hook_p hook, ng_item_fn *fn, void * arg1, int arg2,
3556	int flags)
3557{
3558	item_p item;
3559
3560	if ((item = ng_getqblk(flags)) == NULL) {
3561		return (ENOMEM);
3562	}
3563	item->el_flags = NGQF_FN | NGQF_WRITER;
3564	NG_NODE_REF(node); /* and one for the item */
3565	NGI_SET_NODE(item, node);
3566	if (hook) {
3567		NG_HOOK_REF(hook);
3568		NGI_SET_HOOK(item, hook);
3569	}
3570	NGI_FN(item) = fn;
3571	NGI_ARG1(item) = arg1;
3572	NGI_ARG2(item) = arg2;
3573	return(ng_snd_item(item, flags));
3574}
3575
3576/*
3577 * Official timeout routines for Netgraph nodes.
3578 */
3579static void
3580ng_callout_trampoline(void *arg)
3581{
3582	item_p item = arg;
3583
3584	ng_snd_item(item, 0);
3585}
3586
3587
3588int
3589ng_callout(struct callout *c, node_p node, hook_p hook, int ticks,
3590    ng_item_fn *fn, void * arg1, int arg2)
3591{
3592	item_p item, oitem;
3593
3594	if ((item = ng_getqblk(NG_NOFLAGS)) == NULL)
3595		return (ENOMEM);
3596
3597	item->el_flags = NGQF_FN | NGQF_WRITER;
3598	NG_NODE_REF(node);		/* and one for the item */
3599	NGI_SET_NODE(item, node);
3600	if (hook) {
3601		NG_HOOK_REF(hook);
3602		NGI_SET_HOOK(item, hook);
3603	}
3604	NGI_FN(item) = fn;
3605	NGI_ARG1(item) = arg1;
3606	NGI_ARG2(item) = arg2;
3607	oitem = c->c_arg;
3608	if (callout_reset(c, ticks, &ng_callout_trampoline, item) == 1 &&
3609	    oitem != NULL)
3610		NG_FREE_ITEM(oitem);
3611	return (0);
3612}
3613
3614/* A special modified version of untimeout() */
3615int
3616ng_uncallout(struct callout *c, node_p node)
3617{
3618	item_p item;
3619	int rval;
3620
3621	KASSERT(c != NULL, ("ng_uncallout: NULL callout"));
3622	KASSERT(node != NULL, ("ng_uncallout: NULL node"));
3623
3624	rval = callout_stop(c);
3625	item = c->c_arg;
3626	/* Do an extra check */
3627	if ((rval > 0) && (c->c_func == &ng_callout_trampoline) &&
3628	    (NGI_NODE(item) == node)) {
3629		/*
3630		 * We successfully removed it from the queue before it ran
3631		 * So now we need to unreference everything that was
3632		 * given extra references. (NG_FREE_ITEM does this).
3633		 */
3634		NG_FREE_ITEM(item);
3635	}
3636	c->c_arg = NULL;
3637
3638	return (rval);
3639}
3640
3641/*
3642 * Set the address, if none given, give the node here.
3643 */
3644void
3645ng_replace_retaddr(node_p here, item_p item, ng_ID_t retaddr)
3646{
3647	if (retaddr) {
3648		NGI_RETADDR(item) = retaddr;
3649	} else {
3650		/*
3651		 * The old return address should be ok.
3652		 * If there isn't one, use the address here.
3653		 */
3654		NGI_RETADDR(item) = ng_node2ID(here);
3655	}
3656}
3657
3658#define TESTING
3659#ifdef TESTING
3660/* just test all the macros */
3661void
3662ng_macro_test(item_p item);
3663void
3664ng_macro_test(item_p item)
3665{
3666	node_p node = NULL;
3667	hook_p hook = NULL;
3668	struct mbuf *m;
3669	struct ng_mesg *msg;
3670	ng_ID_t retaddr;
3671	int	error;
3672
3673	NGI_GET_M(item, m);
3674	NGI_GET_MSG(item, msg);
3675	retaddr = NGI_RETADDR(item);
3676	NG_SEND_DATA(error, hook, m, NULL);
3677	NG_SEND_DATA_ONLY(error, hook, m);
3678	NG_FWD_NEW_DATA(error, item, hook, m);
3679	NG_FWD_ITEM_HOOK(error, item, hook);
3680	NG_SEND_MSG_HOOK(error, node, msg, hook, retaddr);
3681	NG_SEND_MSG_ID(error, node, msg, retaddr, retaddr);
3682	NG_SEND_MSG_PATH(error, node, msg, ".:", retaddr);
3683	NG_FWD_MSG_HOOK(error, node, item, hook, retaddr);
3684}
3685#endif /* TESTING */
3686
3687