2 * Copyright (c) 2004 Ruslan Ermilov
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
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.
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
29 #include <sys/param.h>
30 #include <sys/errno.h>
31 #include <sys/kernel.h>
33 #include <sys/systm.h>
35 #include <netgraph/ng_message.h>
36 #include <netgraph/ng_hub.h>
37 #include <netgraph/netgraph.h>
39 #ifdef NG_SEPARATE_MALLOC
40 static MALLOC_DEFINE(M_NETGRAPH_HUB
, "netgraph_hub", "netgraph hub node");
42 #define M_NETGRAPH_HUB M_NETGRAPH
45 /* Per-node private data */
46 struct ng_hub_private
{
47 int persistent
; /* can exist w/o hooks */
49 typedef struct ng_hub_private
*priv_p
;
51 /* Netgraph node methods */
52 static ng_constructor_t ng_hub_constructor
;
53 static ng_rcvmsg_t ng_hub_rcvmsg
;
54 static ng_shutdown_t ng_hub_shutdown
;
55 static ng_rcvdata_t ng_hub_rcvdata
;
56 static ng_disconnect_t ng_hub_disconnect
;
58 /* List of commands and how to convert arguments to/from ASCII */
59 static const struct ng_cmdlist ng_hub_cmdlist
[] = {
62 NGM_HUB_SET_PERSISTENT
,
70 static struct ng_type ng_hub_typestruct
= {
71 .version
= NG_ABI_VERSION
,
72 .name
= NG_HUB_NODE_TYPE
,
73 .constructor
= ng_hub_constructor
,
74 .rcvmsg
= ng_hub_rcvmsg
,
75 .shutdown
= ng_hub_shutdown
,
76 .rcvdata
= ng_hub_rcvdata
,
77 .disconnect
= ng_hub_disconnect
,
78 .cmdlist
= ng_hub_cmdlist
,
80 NETGRAPH_INIT(hub
, &ng_hub_typestruct
);
84 ng_hub_constructor(node_p node
)
88 /* Allocate and initialize private info */
89 priv
= malloc(sizeof(*priv
), M_NETGRAPH_HUB
, M_WAITOK
| M_ZERO
);
91 NG_NODE_SET_PRIVATE(node
, priv
);
96 * Receive a control message
99 ng_hub_rcvmsg(node_p node
, item_p item
, hook_p lasthook
)
101 const priv_p priv
= NG_NODE_PRIVATE(node
);
105 NGI_GET_MSG(item
, msg
);
106 if (msg
->header
.typecookie
== NGM_HUB_COOKIE
&&
107 msg
->header
.cmd
== NGM_HUB_SET_PERSISTENT
) {
108 priv
->persistent
= 1;
118 ng_hub_rcvdata(hook_p hook
, item_p item
)
120 const node_p node
= NG_HOOK_NODE(hook
);
123 struct mbuf
* const m
= NGI_M(item
), *m2
;
126 if ((nhooks
= NG_NODE_NUMHOOKS(node
)) == 1) {
130 LIST_FOREACH(hook2
, &node
->nd_hooks
, hk_hooks
) {
134 NG_FWD_ITEM_HOOK(error
, item
, hook2
);
136 if ((m2
= m_dup(m
, M_NOWAIT
)) == NULL
) {
140 NG_SEND_DATA_ONLY(error
, hook2
, m2
);
142 continue; /* don't give up */
153 ng_hub_shutdown(node_p node
)
155 const priv_p priv
= NG_NODE_PRIVATE(node
);
157 free(priv
, M_NETGRAPH_HUB
);
158 NG_NODE_SET_PRIVATE(node
, NULL
);
164 ng_hub_disconnect(hook_p hook
)
166 const priv_p priv
= NG_NODE_PRIVATE(NG_HOOK_NODE(hook
));
168 if (NG_NODE_NUMHOOKS(NG_HOOK_NODE(hook
)) == 0 &&
169 NG_NODE_IS_VALID(NG_HOOK_NODE(hook
)) && !priv
->persistent
)
170 ng_rmnode_self(NG_HOOK_NODE(hook
));