1b462702fSRuslan Ermilov /*- 2b462702fSRuslan Ermilov * Copyright (c) 2004 Ruslan Ermilov 3b462702fSRuslan Ermilov * All rights reserved. 4b462702fSRuslan Ermilov * 5b462702fSRuslan Ermilov * Redistribution and use in source and binary forms, with or without 6b462702fSRuslan Ermilov * modification, are permitted provided that the following conditions 7b462702fSRuslan Ermilov * are met: 8b462702fSRuslan Ermilov * 1. Redistributions of source code must retain the above copyright 9b462702fSRuslan Ermilov * notice, this list of conditions and the following disclaimer. 10b462702fSRuslan Ermilov * 2. Redistributions in binary form must reproduce the above copyright 11b462702fSRuslan Ermilov * notice, this list of conditions and the following disclaimer in the 12b462702fSRuslan Ermilov * documentation and/or other materials provided with the distribution. 13b462702fSRuslan Ermilov * 14b462702fSRuslan Ermilov * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 15b462702fSRuslan Ermilov * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16b462702fSRuslan Ermilov * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17b462702fSRuslan Ermilov * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18b462702fSRuslan Ermilov * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19b462702fSRuslan Ermilov * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20b462702fSRuslan Ermilov * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21b462702fSRuslan Ermilov * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22b462702fSRuslan Ermilov * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23b462702fSRuslan Ermilov * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24b462702fSRuslan Ermilov * SUCH DAMAGE. 25b462702fSRuslan Ermilov * 26b462702fSRuslan Ermilov * $FreeBSD$ 27b462702fSRuslan Ermilov */ 28b462702fSRuslan Ermilov 29b462702fSRuslan Ermilov #include <sys/param.h> 30b462702fSRuslan Ermilov #include <sys/errno.h> 31b462702fSRuslan Ermilov #include <sys/kernel.h> 32b462702fSRuslan Ermilov #include <sys/mbuf.h> 33b462702fSRuslan Ermilov #include <sys/systm.h> 34b462702fSRuslan Ermilov 35b462702fSRuslan Ermilov #include <netgraph/ng_message.h> 36b462702fSRuslan Ermilov #include <netgraph/ng_hub.h> 37b462702fSRuslan Ermilov #include <netgraph/netgraph.h> 38b462702fSRuslan Ermilov 39f8aab721SMarko Zec #ifdef NG_SEPARATE_MALLOC 40f8aab721SMarko Zec MALLOC_DEFINE(M_NETGRAPH_HUB, "netgraph_hub", "netgraph hub node"); 41f8aab721SMarko Zec #else 42f8aab721SMarko Zec #define M_NETGRAPH_HUB M_NETGRAPH 43f8aab721SMarko Zec #endif 44f8aab721SMarko Zec 45f8aab721SMarko Zec /* Per-node private data */ 46f8aab721SMarko Zec struct ng_hub_private { 47f8aab721SMarko Zec int persistent; /* can exist w/o hooks */ 48f8aab721SMarko Zec }; 49f8aab721SMarko Zec typedef struct ng_hub_private *priv_p; 50f8aab721SMarko Zec 51f8aab721SMarko Zec /* Netgraph node methods */ 52b462702fSRuslan Ermilov static ng_constructor_t ng_hub_constructor; 53f8aab721SMarko Zec static ng_rcvmsg_t ng_hub_rcvmsg; 54f8aab721SMarko Zec static ng_shutdown_t ng_hub_shutdown; 55b462702fSRuslan Ermilov static ng_rcvdata_t ng_hub_rcvdata; 56b462702fSRuslan Ermilov static ng_disconnect_t ng_hub_disconnect; 57b462702fSRuslan Ermilov 58f8aab721SMarko Zec /* List of commands and how to convert arguments to/from ASCII */ 59f8aab721SMarko Zec static const struct ng_cmdlist ng_hub_cmdlist[] = { 60f8aab721SMarko Zec { 61f8aab721SMarko Zec NGM_HUB_COOKIE, 62f8aab721SMarko Zec NGM_HUB_SET_PERSISTENT, 63f8aab721SMarko Zec "setpersistent", 64f8aab721SMarko Zec NULL, 65f8aab721SMarko Zec NULL 66f8aab721SMarko Zec }, 67f8aab721SMarko Zec { 0 } 68f8aab721SMarko Zec }; 69f8aab721SMarko Zec 70b462702fSRuslan Ermilov static struct ng_type ng_hub_typestruct = { 71b462702fSRuslan Ermilov .version = NG_ABI_VERSION, 72b462702fSRuslan Ermilov .name = NG_HUB_NODE_TYPE, 73b462702fSRuslan Ermilov .constructor = ng_hub_constructor, 74f8aab721SMarko Zec .rcvmsg = ng_hub_rcvmsg, 75f8aab721SMarko Zec .shutdown = ng_hub_shutdown, 76b462702fSRuslan Ermilov .rcvdata = ng_hub_rcvdata, 77b462702fSRuslan Ermilov .disconnect = ng_hub_disconnect, 78f8aab721SMarko Zec .cmdlist = ng_hub_cmdlist, 79b462702fSRuslan Ermilov }; 80b462702fSRuslan Ermilov NETGRAPH_INIT(hub, &ng_hub_typestruct); 81b462702fSRuslan Ermilov 82b462702fSRuslan Ermilov 83b462702fSRuslan Ermilov static int 84b462702fSRuslan Ermilov ng_hub_constructor(node_p node) 85b462702fSRuslan Ermilov { 86f8aab721SMarko Zec priv_p priv; 87b462702fSRuslan Ermilov 88f8aab721SMarko Zec /* Allocate and initialize private info */ 89f8aab721SMarko Zec priv = malloc(sizeof(*priv), M_NETGRAPH_HUB, M_NOWAIT | M_ZERO); 90f8aab721SMarko Zec if (priv == NULL) 91f8aab721SMarko Zec return (ENOMEM); 92f8aab721SMarko Zec 93f8aab721SMarko Zec NG_NODE_SET_PRIVATE(node, priv); 94b462702fSRuslan Ermilov return (0); 95b462702fSRuslan Ermilov } 96b462702fSRuslan Ermilov 97f8aab721SMarko Zec /* 98f8aab721SMarko Zec * Receive a control message 99f8aab721SMarko Zec */ 100f8aab721SMarko Zec static int 101f8aab721SMarko Zec ng_hub_rcvmsg(node_p node, item_p item, hook_p lasthook) 102f8aab721SMarko Zec { 103f8aab721SMarko Zec const priv_p priv = NG_NODE_PRIVATE(node); 104f8aab721SMarko Zec int error = 0; 105f8aab721SMarko Zec struct ng_mesg *msg; 106f8aab721SMarko Zec 107f8aab721SMarko Zec NGI_GET_MSG(item, msg); 108f8aab721SMarko Zec if (msg->header.typecookie == NGM_HUB_COOKIE && 109f8aab721SMarko Zec msg->header.cmd == NGM_HUB_SET_PERSISTENT) { 110f8aab721SMarko Zec priv->persistent = 1; 111f8aab721SMarko Zec } else { 112f8aab721SMarko Zec error = EINVAL; 113f8aab721SMarko Zec } 114f8aab721SMarko Zec 115f8aab721SMarko Zec NG_FREE_MSG(msg); 116f8aab721SMarko Zec return (error); 117f8aab721SMarko Zec } 118f8aab721SMarko Zec 119b462702fSRuslan Ermilov static int 120b462702fSRuslan Ermilov ng_hub_rcvdata(hook_p hook, item_p item) 121b462702fSRuslan Ermilov { 122b462702fSRuslan Ermilov const node_p node = NG_HOOK_NODE(hook); 123b462702fSRuslan Ermilov int error = 0; 124b462702fSRuslan Ermilov hook_p hook2; 125b462702fSRuslan Ermilov struct mbuf * const m = NGI_M(item), *m2; 126b462702fSRuslan Ermilov int nhooks; 127b462702fSRuslan Ermilov 128b462702fSRuslan Ermilov if ((nhooks = NG_NODE_NUMHOOKS(node)) == 1) { 129b462702fSRuslan Ermilov NG_FREE_ITEM(item); 130b462702fSRuslan Ermilov return (0); 131b462702fSRuslan Ermilov } 132b462702fSRuslan Ermilov LIST_FOREACH(hook2, &node->nd_hooks, hk_hooks) { 133b462702fSRuslan Ermilov if (hook2 == hook) 134b462702fSRuslan Ermilov continue; 135b462702fSRuslan Ermilov if (--nhooks == 1) 136b462702fSRuslan Ermilov NG_FWD_ITEM_HOOK(error, item, hook2); 137b462702fSRuslan Ermilov else { 138b462702fSRuslan Ermilov if ((m2 = m_dup(m, M_DONTWAIT)) == NULL) { 139b462702fSRuslan Ermilov NG_FREE_ITEM(item); 140b462702fSRuslan Ermilov return (ENOBUFS); 141b462702fSRuslan Ermilov } 1423ca24c28SJulian Elischer NG_SEND_DATA_ONLY(error, hook2, m2); 143b462702fSRuslan Ermilov if (error) 144dfa515f2SRuslan Ermilov continue; /* don't give up */ 145b462702fSRuslan Ermilov } 146b462702fSRuslan Ermilov } 147b462702fSRuslan Ermilov 148b462702fSRuslan Ermilov return (error); 149b462702fSRuslan Ermilov } 150b462702fSRuslan Ermilov 151f8aab721SMarko Zec /* 152f8aab721SMarko Zec * Shutdown node 153f8aab721SMarko Zec */ 154f8aab721SMarko Zec static int 155f8aab721SMarko Zec ng_hub_shutdown(node_p node) 156f8aab721SMarko Zec { 157f8aab721SMarko Zec const priv_p priv = NG_NODE_PRIVATE(node); 158f8aab721SMarko Zec 159f8aab721SMarko Zec free(priv, M_NETGRAPH_HUB); 160*fae147aaSMarko Zec NG_NODE_SET_PRIVATE(node, NULL); 161*fae147aaSMarko Zec NG_NODE_UNREF(node); 162f8aab721SMarko Zec return (0); 163f8aab721SMarko Zec } 164f8aab721SMarko Zec 165b462702fSRuslan Ermilov static int 166b462702fSRuslan Ermilov ng_hub_disconnect(hook_p hook) 167b462702fSRuslan Ermilov { 168f8aab721SMarko Zec const priv_p priv = NG_NODE_PRIVATE(NG_HOOK_NODE(hook)); 169b462702fSRuslan Ermilov 170b462702fSRuslan Ermilov if (NG_NODE_NUMHOOKS(NG_HOOK_NODE(hook)) == 0 && 171f8aab721SMarko Zec NG_NODE_IS_VALID(NG_HOOK_NODE(hook)) && !priv->persistent) 172b462702fSRuslan Ermilov ng_rmnode_self(NG_HOOK_NODE(hook)); 173b462702fSRuslan Ermilov return (0); 174b462702fSRuslan Ermilov } 175