xref: /freebsd/sys/netgraph/ng_hub.c (revision 1e413cf93298b5b97441a21d9a50fdcd0ee9945e)
1 /*-
2  * Copyright (c) 2004 Ruslan Ermilov
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  * $FreeBSD$
27  */
28 
29 #include <sys/param.h>
30 #include <sys/errno.h>
31 #include <sys/kernel.h>
32 #include <sys/mbuf.h>
33 #include <sys/systm.h>
34 
35 #include <netgraph/ng_message.h>
36 #include <netgraph/ng_hub.h>
37 #include <netgraph/netgraph.h>
38 
39 static ng_constructor_t	ng_hub_constructor;
40 static ng_rcvdata_t	ng_hub_rcvdata;
41 static ng_disconnect_t	ng_hub_disconnect;
42 
43 static struct ng_type ng_hub_typestruct = {
44 	.version =	NG_ABI_VERSION,
45 	.name =		NG_HUB_NODE_TYPE,
46 	.constructor =	ng_hub_constructor,
47 	.rcvdata =	ng_hub_rcvdata,
48 	.disconnect =	ng_hub_disconnect,
49 };
50 NETGRAPH_INIT(hub, &ng_hub_typestruct);
51 
52 
53 static int
54 ng_hub_constructor(node_p node)
55 {
56 
57 	return (0);
58 }
59 
60 static int
61 ng_hub_rcvdata(hook_p hook, item_p item)
62 {
63 	const node_p node = NG_HOOK_NODE(hook);
64 	int error = 0;
65 	hook_p hook2;
66 	struct mbuf * const m = NGI_M(item), *m2;
67 	int nhooks;
68 
69 	if ((nhooks = NG_NODE_NUMHOOKS(node)) == 1) {
70 		NG_FREE_ITEM(item);
71 		return (0);
72 	}
73 	LIST_FOREACH(hook2, &node->nd_hooks, hk_hooks) {
74 		if (hook2 == hook)
75 			continue;
76 		if (--nhooks == 1)
77 			NG_FWD_ITEM_HOOK(error, item, hook2);
78 		else {
79 			if ((m2 = m_dup(m, M_DONTWAIT)) == NULL) {
80 				NG_FREE_ITEM(item);
81 				return (ENOBUFS);
82 			}
83 			NG_SEND_DATA_ONLY(error, hook2, m2);
84 			if (error)
85 				continue;	/* don't give up */
86 		}
87 	}
88 
89 	return (error);
90 }
91 
92 static int
93 ng_hub_disconnect(hook_p hook)
94 {
95 
96 	if (NG_NODE_NUMHOOKS(NG_HOOK_NODE(hook)) == 0 &&
97 	    NG_NODE_IS_VALID(NG_HOOK_NODE(hook)))
98 		ng_rmnode_self(NG_HOOK_NODE(hook));
99 	return (0);
100 }
101