kernel/netgraph7: Replace usage of MALLOC/FREE with kmalloc/kfree here too.
[dragonfly.git] / sys / netgraph7 / ng_one2many.c
CommitLineData
b06ebda0
MD
1/*
2 * ng_one2many.c
3 */
4
5/*-
6 * Copyright (c) 2000 Whistle Communications, Inc.
7 * All rights reserved.
8 *
9 * Subject to the following obligations and disclaimer of warranty, use and
10 * redistribution of this software, in source or object code forms, with or
11 * without modifications are expressly permitted by Whistle Communications;
12 * provided, however, that:
13 * 1. Any and all reproductions of the source or object code must include the
14 * copyright notice above and the following disclaimer of warranties; and
15 * 2. No rights are granted, in any manner or form, to use Whistle
16 * Communications, Inc. trademarks, including the mark "WHISTLE
17 * COMMUNICATIONS" on advertising, endorsements, or otherwise except as
18 * such appears in the above copyright notice or in the software.
19 *
20 * THIS SOFTWARE IS BEING PROVIDED BY WHISTLE COMMUNICATIONS "AS IS", AND
21 * TO THE MAXIMUM EXTENT PERMITTED BY LAW, WHISTLE COMMUNICATIONS MAKES NO
22 * REPRESENTATIONS OR WARRANTIES, EXPRESS OR IMPLIED, REGARDING THIS SOFTWARE,
23 * INCLUDING WITHOUT LIMITATION, ANY AND ALL IMPLIED WARRANTIES OF
24 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR NON-INFRINGEMENT.
25 * WHISTLE COMMUNICATIONS DOES NOT WARRANT, GUARANTEE, OR MAKE ANY
26 * REPRESENTATIONS REGARDING THE USE OF, OR THE RESULTS OF THE USE OF THIS
27 * SOFTWARE IN TERMS OF ITS CORRECTNESS, ACCURACY, RELIABILITY OR OTHERWISE.
28 * IN NO EVENT SHALL WHISTLE COMMUNICATIONS BE LIABLE FOR ANY DAMAGES
29 * RESULTING FROM OR ARISING OUT OF ANY USE OF THIS SOFTWARE, INCLUDING
30 * WITHOUT LIMITATION, ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
31 * PUNITIVE, OR CONSEQUENTIAL DAMAGES, PROCUREMENT OF SUBSTITUTE GOODS OR
32 * SERVICES, LOSS OF USE, DATA OR PROFITS, HOWEVER CAUSED AND UNDER ANY
33 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
34 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
35 * THIS SOFTWARE, EVEN IF WHISTLE COMMUNICATIONS IS ADVISED OF THE POSSIBILITY
36 * OF SUCH DAMAGE.
37 *
38 * Author: Archie Cobbs <archie@freebsd.org>
39 *
40 * $FreeBSD: src/sys/netgraph/ng_one2many.c,v 1.21 2005/03/11 10:29:38 glebius Exp $
5a975a3d 41 * $DragonFly: src/sys/netgraph7/ng_one2many.c,v 1.2 2008/06/26 23:05:35 dillon Exp $
b06ebda0
MD
42 */
43
44/*
45 * ng_one2many(4) netgraph node type
46 *
47 * Packets received on the "one" hook are sent out each of the
48 * "many" hooks accoring to an algorithm. Packets received on any
49 * "many" hook are always delivered to the "one" hook.
50 */
51
52#include <sys/param.h>
53#include <sys/systm.h>
54#include <sys/kernel.h>
55#include <sys/malloc.h>
56#include <sys/ctype.h>
57#include <sys/mbuf.h>
58#include <sys/errno.h>
59
5a975a3d
MD
60#include "ng_message.h"
61#include "netgraph.h"
62#include "ng_parse.h"
63#include "ng_one2many.h"
b06ebda0
MD
64
65/* Per-link private data */
66struct ng_one2many_link {
67 hook_p hook; /* netgraph hook */
68 struct ng_one2many_link_stats stats; /* link stats */
69};
70
71/* Per-node private data */
72struct ng_one2many_private {
73 node_p node; /* link to node */
74 struct ng_one2many_config conf; /* node configuration */
75 struct ng_one2many_link one; /* "one" hook */
76 struct ng_one2many_link many[NG_ONE2MANY_MAX_LINKS];
77 u_int16_t nextMany; /* next round-robin */
78 u_int16_t numActiveMany; /* # active "many" */
79 u_int16_t activeMany[NG_ONE2MANY_MAX_LINKS];
80};
81typedef struct ng_one2many_private *priv_p;
82
83/* Netgraph node methods */
84static ng_constructor_t ng_one2many_constructor;
85static ng_rcvmsg_t ng_one2many_rcvmsg;
86static ng_shutdown_t ng_one2many_shutdown;
87static ng_newhook_t ng_one2many_newhook;
88static ng_rcvdata_t ng_one2many_rcvdata;
89static ng_disconnect_t ng_one2many_disconnect;
90
91/* Other functions */
92static void ng_one2many_update_many(priv_p priv);
93static void ng_one2many_notify(priv_p priv, uint32_t cmd);
94
95/******************************************************************
96 NETGRAPH PARSE TYPES
97******************************************************************/
98
99/* Parse type for struct ng_one2many_config */
100static const struct ng_parse_fixedarray_info
101 ng_one2many_enableLinks_array_type_info = {
102 &ng_parse_uint8_type,
103 NG_ONE2MANY_MAX_LINKS
104};
105static const struct ng_parse_type ng_one2many_enableLinks_array_type = {
106 &ng_parse_fixedarray_type,
107 &ng_one2many_enableLinks_array_type_info,
108};
109static const struct ng_parse_struct_field ng_one2many_config_type_fields[]
110 = NG_ONE2MANY_CONFIG_TYPE_INFO(&ng_one2many_enableLinks_array_type);
111static const struct ng_parse_type ng_one2many_config_type = {
112 &ng_parse_struct_type,
113 &ng_one2many_config_type_fields
114};
115
116/* Parse type for struct ng_one2many_link_stats */
117static const struct ng_parse_struct_field ng_one2many_link_stats_type_fields[]
118 = NG_ONE2MANY_LINK_STATS_TYPE_INFO;
119static const struct ng_parse_type ng_one2many_link_stats_type = {
120 &ng_parse_struct_type,
121 &ng_one2many_link_stats_type_fields
122};
123
124/* List of commands and how to convert arguments to/from ASCII */
125static const struct ng_cmdlist ng_one2many_cmdlist[] = {
126 {
127 NGM_ONE2MANY_COOKIE,
128 NGM_ONE2MANY_SET_CONFIG,
129 "setconfig",
130 &ng_one2many_config_type,
131 NULL
132 },
133 {
134 NGM_ONE2MANY_COOKIE,
135 NGM_ONE2MANY_GET_CONFIG,
136 "getconfig",
137 NULL,
138 &ng_one2many_config_type
139 },
140 {
141 NGM_ONE2MANY_COOKIE,
142 NGM_ONE2MANY_GET_STATS,
143 "getstats",
144 &ng_parse_int32_type,
145 &ng_one2many_link_stats_type
146 },
147 {
148 NGM_ONE2MANY_COOKIE,
149 NGM_ONE2MANY_CLR_STATS,
150 "clrstats",
151 &ng_parse_int32_type,
152 NULL,
153 },
154 {
155 NGM_ONE2MANY_COOKIE,
156 NGM_ONE2MANY_GETCLR_STATS,
157 "getclrstats",
158 &ng_parse_int32_type,
159 &ng_one2many_link_stats_type
160 },
161 { 0 }
162};
163
164/* Node type descriptor */
165static struct ng_type ng_one2many_typestruct = {
166 .version = NG_ABI_VERSION,
167 .name = NG_ONE2MANY_NODE_TYPE,
168 .constructor = ng_one2many_constructor,
169 .rcvmsg = ng_one2many_rcvmsg,
170 .shutdown = ng_one2many_shutdown,
171 .newhook = ng_one2many_newhook,
172 .rcvdata = ng_one2many_rcvdata,
173 .disconnect = ng_one2many_disconnect,
174 .cmdlist = ng_one2many_cmdlist,
175};
176NETGRAPH_INIT(one2many, &ng_one2many_typestruct);
177
178/******************************************************************
179 NETGRAPH NODE METHODS
180******************************************************************/
181
182/*
183 * Node constructor
184 */
185static int
186ng_one2many_constructor(node_p node)
187{
188 priv_p priv;
189
190 /* Allocate and initialize private info */
fc025606
SW
191 priv = kmalloc(sizeof(*priv), M_NETGRAPH,
192 M_WAITOK | M_NULLOK | M_ZERO);
b06ebda0
MD
193 if (priv == NULL)
194 return (ENOMEM);
195 priv->conf.xmitAlg = NG_ONE2MANY_XMIT_ROUNDROBIN;
196 priv->conf.failAlg = NG_ONE2MANY_FAIL_MANUAL;
197
198 /* cross reference */
199 NG_NODE_SET_PRIVATE(node, priv);
200 priv->node = node;
201
202 /* Done */
203 return (0);
204}
205
206/*
207 * Method for attaching a new hook
208 */
209static int
210ng_one2many_newhook(node_p node, hook_p hook, const char *name)
211{
212 const priv_p priv = NG_NODE_PRIVATE(node);
213 struct ng_one2many_link *link;
214 int linkNum;
215 u_long i;
216
217 /* Which hook? */
218 if (strncmp(name, NG_ONE2MANY_HOOK_MANY_PREFIX,
219 strlen(NG_ONE2MANY_HOOK_MANY_PREFIX)) == 0) {
220 const char *cp;
221 char *eptr;
222
223 cp = name + strlen(NG_ONE2MANY_HOOK_MANY_PREFIX);
224 if (!isdigit(*cp) || (cp[0] == '0' && cp[1] != '\0'))
225 return (EINVAL);
226 i = strtoul(cp, &eptr, 10);
227 if (*eptr != '\0' || i < 0 || i >= NG_ONE2MANY_MAX_LINKS)
228 return (EINVAL);
229 linkNum = (int)i;
230 link = &priv->many[linkNum];
231 } else if (strcmp(name, NG_ONE2MANY_HOOK_ONE) == 0) {
232 linkNum = NG_ONE2MANY_ONE_LINKNUM;
233 link = &priv->one;
234 } else
235 return (EINVAL);
236
237 /* Is hook already connected? (should never happen) */
238 if (link->hook != NULL)
239 return (EISCONN);
240
241 /* Setup private info for this link */
242 NG_HOOK_SET_PRIVATE(hook, (void *)(intptr_t)linkNum);
243 link->hook = hook;
244 bzero(&link->stats, sizeof(link->stats));
245 if (linkNum != NG_ONE2MANY_ONE_LINKNUM) {
246 priv->conf.enabledLinks[linkNum] = 1; /* auto-enable link */
247 ng_one2many_update_many(priv);
248 }
249
250 /* Done */
251 return (0);
252}
253
254/*
255 * Receive a control message
256 */
257static int
258ng_one2many_rcvmsg(node_p node, item_p item, hook_p lasthook)
259{
260 const priv_p priv = NG_NODE_PRIVATE(node);
261 struct ng_mesg *resp = NULL;
262 int error = 0;
263 struct ng_mesg *msg;
264
265 NGI_GET_MSG(item, msg);
266 switch (msg->header.typecookie) {
267 case NGM_ONE2MANY_COOKIE:
268 switch (msg->header.cmd) {
269 case NGM_ONE2MANY_SET_CONFIG:
270 {
271 struct ng_one2many_config *conf;
272 int i;
273
274 /* Check that new configuration is valid */
275 if (msg->header.arglen != sizeof(*conf)) {
276 error = EINVAL;
277 break;
278 }
279 conf = (struct ng_one2many_config *)msg->data;
280 switch (conf->xmitAlg) {
281 case NG_ONE2MANY_XMIT_ROUNDROBIN:
282 case NG_ONE2MANY_XMIT_ALL:
283 break;
284 default:
285 error = EINVAL;
286 break;
287 }
288 switch (conf->failAlg) {
289 case NG_ONE2MANY_FAIL_MANUAL:
290 case NG_ONE2MANY_FAIL_NOTIFY:
291 break;
292 default:
293 error = EINVAL;
294 break;
295 }
296 if (error != 0)
297 break;
298
299 /* Normalized many link enabled bits */
300 for (i = 0; i < NG_ONE2MANY_MAX_LINKS; i++)
301 conf->enabledLinks[i] = !!conf->enabledLinks[i];
302
303 /* Copy config and reset */
304 bcopy(conf, &priv->conf, sizeof(*conf));
305 ng_one2many_update_many(priv);
306 break;
307 }
308 case NGM_ONE2MANY_GET_CONFIG:
309 {
310 struct ng_one2many_config *conf;
311
5a975a3d 312 NG_MKRESPONSE(resp, msg, sizeof(*conf), M_WAITOK | M_NULLOK);
b06ebda0
MD
313 if (resp == NULL) {
314 error = ENOMEM;
315 break;
316 }
317 conf = (struct ng_one2many_config *)resp->data;
318 bcopy(&priv->conf, conf, sizeof(priv->conf));
319 break;
320 }
321 case NGM_ONE2MANY_GET_STATS:
322 case NGM_ONE2MANY_CLR_STATS:
323 case NGM_ONE2MANY_GETCLR_STATS:
324 {
325 struct ng_one2many_link *link;
326 int linkNum;
327
328 /* Get link */
329 if (msg->header.arglen != sizeof(int32_t)) {
330 error = EINVAL;
331 break;
332 }
333 linkNum = *((int32_t *)msg->data);
334 if (linkNum == NG_ONE2MANY_ONE_LINKNUM)
335 link = &priv->one;
336 else if (linkNum >= 0
337 && linkNum < NG_ONE2MANY_MAX_LINKS) {
338 link = &priv->many[linkNum];
339 } else {
340 error = EINVAL;
341 break;
342 }
343
344 /* Get/clear stats */
345 if (msg->header.cmd != NGM_ONE2MANY_CLR_STATS) {
346 NG_MKRESPONSE(resp, msg,
5a975a3d 347 sizeof(link->stats), M_WAITOK | M_NULLOK);
b06ebda0
MD
348 if (resp == NULL) {
349 error = ENOMEM;
350 break;
351 }
352 bcopy(&link->stats,
353 resp->data, sizeof(link->stats));
354 }
355 if (msg->header.cmd != NGM_ONE2MANY_GET_STATS)
356 bzero(&link->stats, sizeof(link->stats));
357 break;
358 }
359 default:
360 error = EINVAL;
361 break;
362 }
363 break;
364 /*
365 * One of our downstreams notifies us of link change. If we are
366 * configured to listen to these message, then we remove/add
367 * this hook from array of active hooks.
368 */
369 case NGM_FLOW_COOKIE:
370 {
371 int linkNum;
372
373 if (priv->conf.failAlg != NG_ONE2MANY_FAIL_NOTIFY)
374 break;
375
376 if (lasthook == NULL)
377 break;
378
379 linkNum = (intptr_t)NG_HOOK_PRIVATE(lasthook);
380 if (linkNum == NG_ONE2MANY_ONE_LINKNUM)
381 break;
382
383 KASSERT((linkNum >= 0 && linkNum < NG_ONE2MANY_MAX_LINKS),
384 ("%s: linkNum=%d", __func__, linkNum));
385
386 switch (msg->header.cmd) {
387 case NGM_LINK_IS_UP:
388 priv->conf.enabledLinks[linkNum] = 1;
389 ng_one2many_update_many(priv);
390 break;
391 case NGM_LINK_IS_DOWN:
392 priv->conf.enabledLinks[linkNum] = 0;
393 ng_one2many_update_many(priv);
394 break;
395 default:
396 break;
397 }
398 break;
399 }
400 default:
401 error = EINVAL;
402 break;
403 }
404
405 /* Done */
406 NG_RESPOND_MSG(error, node, item, resp);
407 NG_FREE_MSG(msg);
408 return (error);
409}
410
411/*
412 * Receive data on a hook
413 */
414static int
415ng_one2many_rcvdata(hook_p hook, item_p item)
416{
417 const node_p node = NG_HOOK_NODE(hook);
418 const priv_p priv = NG_NODE_PRIVATE(node);
419 struct ng_one2many_link *src;
420 struct ng_one2many_link *dst = NULL;
421 int error = 0;
422 int linkNum;
423 int i;
424 struct mbuf *m;
425
426 m = NGI_M(item); /* just peaking, mbuf still owned by item */
427 /* Get link number */
428 linkNum = (intptr_t)NG_HOOK_PRIVATE(hook);
429 KASSERT(linkNum == NG_ONE2MANY_ONE_LINKNUM
430 || (linkNum >= 0 && linkNum < NG_ONE2MANY_MAX_LINKS),
431 ("%s: linkNum=%d", __func__, linkNum));
432
433 /* Figure out source link */
434 src = (linkNum == NG_ONE2MANY_ONE_LINKNUM) ?
435 &priv->one : &priv->many[linkNum];
436 KASSERT(src->hook != NULL, ("%s: no src%d", __func__, linkNum));
437
438 /* Update receive stats */
439 src->stats.recvPackets++;
440 src->stats.recvOctets += m->m_pkthdr.len;
441
442 /* Figure out destination link */
443 if (linkNum == NG_ONE2MANY_ONE_LINKNUM) {
444 if (priv->numActiveMany == 0) {
445 NG_FREE_ITEM(item);
446 return (ENOTCONN);
447 }
448 switch(priv->conf.xmitAlg) {
449 case NG_ONE2MANY_XMIT_ROUNDROBIN:
450 dst = &priv->many[priv->activeMany[priv->nextMany]];
451 priv->nextMany = (priv->nextMany + 1) % priv->numActiveMany;
452 break;
453 case NG_ONE2MANY_XMIT_ALL:
454 /* no need to copy data for the 1st one */
455 dst = &priv->many[priv->activeMany[0]];
456
457 /* make copies of data and send for all links
458 * except the first one, which we'll do last
459 */
460 for (i = 1; i < priv->numActiveMany; i++) {
461 struct mbuf *m2;
462 struct ng_one2many_link *mdst;
463
464 mdst = &priv->many[priv->activeMany[i]];
5a975a3d 465 m2 = m_dup(m, MB_DONTWAIT); /* XXX m_copypacket() */
b06ebda0
MD
466 if (m2 == NULL) {
467 mdst->stats.memoryFailures++;
468 NG_FREE_ITEM(item);
469 NG_FREE_M(m);
470 return (ENOBUFS);
471 }
472 /* Update transmit stats */
473 mdst->stats.xmitPackets++;
474 mdst->stats.xmitOctets += m->m_pkthdr.len;
475 NG_SEND_DATA_ONLY(error, mdst->hook, m2);
476 }
477 break;
478#ifdef INVARIANTS
479 default:
480 panic("%s: invalid xmitAlg", __func__);
481#endif
482 }
483 } else {
484 dst = &priv->one;
485 }
486
487 /* Update transmit stats */
488 dst->stats.xmitPackets++;
489 dst->stats.xmitOctets += m->m_pkthdr.len;
490
491 /* Deliver packet */
492 NG_FWD_ITEM_HOOK(error, item, dst->hook);
493 return (error);
494}
495
496/*
497 * Shutdown node
498 */
499static int
500ng_one2many_shutdown(node_p node)
501{
502 const priv_p priv = NG_NODE_PRIVATE(node);
503
504 KASSERT(priv->numActiveMany == 0,
505 ("%s: numActiveMany=%d", __func__, priv->numActiveMany));
fc025606 506 kfree(priv, M_NETGRAPH);
b06ebda0
MD
507 NG_NODE_SET_PRIVATE(node, NULL);
508 NG_NODE_UNREF(node);
509 return (0);
510}
511
512/*
513 * Hook disconnection.
514 */
515static int
516ng_one2many_disconnect(hook_p hook)
517{
518 const priv_p priv = NG_NODE_PRIVATE(NG_HOOK_NODE(hook));
519 int linkNum;
520
521 /* Get link number */
522 linkNum = (intptr_t)NG_HOOK_PRIVATE(hook);
523 KASSERT(linkNum == NG_ONE2MANY_ONE_LINKNUM
524 || (linkNum >= 0 && linkNum < NG_ONE2MANY_MAX_LINKS),
525 ("%s: linkNum=%d", __func__, linkNum));
526
527 /* Nuke the link */
528 if (linkNum == NG_ONE2MANY_ONE_LINKNUM)
529 priv->one.hook = NULL;
530 else {
531 priv->many[linkNum].hook = NULL;
532 priv->conf.enabledLinks[linkNum] = 0;
533 ng_one2many_update_many(priv);
534 }
535
536 /* If no hooks left, go away */
537 if ((NG_NODE_NUMHOOKS(NG_HOOK_NODE(hook)) == 0)
538 && (NG_NODE_IS_VALID(NG_HOOK_NODE(hook))))
539 ng_rmnode_self(NG_HOOK_NODE(hook));
540 return (0);
541}
542
543/******************************************************************
544 OTHER FUNCTIONS
545******************************************************************/
546
547/*
548 * Update internal state after the addition or removal of a "many" link
549 */
550static void
551ng_one2many_update_many(priv_p priv)
552{
553 uint16_t saveActive = priv->numActiveMany;
554 int linkNum;
555
556 /* Update list of which "many" links are up */
557 priv->numActiveMany = 0;
558 for (linkNum = 0; linkNum < NG_ONE2MANY_MAX_LINKS; linkNum++) {
559 switch (priv->conf.failAlg) {
560 case NG_ONE2MANY_FAIL_MANUAL:
561 case NG_ONE2MANY_FAIL_NOTIFY:
562 if (priv->many[linkNum].hook != NULL
563 && priv->conf.enabledLinks[linkNum]) {
564 priv->activeMany[priv->numActiveMany] = linkNum;
565 priv->numActiveMany++;
566 }
567 break;
568#ifdef INVARIANTS
569 default:
570 panic("%s: invalid failAlg", __func__);
571#endif
572 }
573 }
574
575 if (priv->numActiveMany == 0 && saveActive > 0)
576 ng_one2many_notify(priv, NGM_LINK_IS_DOWN);
577
578 if (saveActive == 0 && priv->numActiveMany > 0)
579 ng_one2many_notify(priv, NGM_LINK_IS_UP);
580
581 /* Update transmit algorithm state */
582 switch (priv->conf.xmitAlg) {
583 case NG_ONE2MANY_XMIT_ROUNDROBIN:
584 if (priv->numActiveMany > 0)
585 priv->nextMany %= priv->numActiveMany;
586 break;
587 case NG_ONE2MANY_XMIT_ALL:
588 break;
589#ifdef INVARIANTS
590 default:
591 panic("%s: invalid xmitAlg", __func__);
592#endif
593 }
594}
595
596/*
597 * Notify upstream if we are out of links, or we have at least one link.
598 */
599static void
600ng_one2many_notify(priv_p priv, uint32_t cmd)
601{
602 struct ng_mesg *msg;
603 int dummy_error = 0;
604
605 if (priv->one.hook == NULL)
606 return;
607
5a975a3d 608 NG_MKMESSAGE(msg, NGM_FLOW_COOKIE, cmd, 0, M_WAITOK | M_NULLOK);
b06ebda0
MD
609 if (msg != NULL)
610 NG_SEND_MSG_HOOK(dummy_error, priv->node, msg, priv->one.hook, 0);
611}