2 * Copyright (C) 2010 by Maxim Ignatenko <gelraen.ua@gmail.com>
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
28 #include <sys/cdefs.h>
29 __FBSDID("$FreeBSD$");
31 #include <sys/param.h>
32 #include <sys/systm.h>
33 #include <sys/kernel.h>
34 #include <sys/endian.h>
35 #include <sys/malloc.h>
37 #include <netgraph/ng_message.h>
38 #include <netgraph/ng_parse.h>
39 #include <netgraph/ng_patch.h>
40 #include <netgraph/netgraph.h>
42 static ng_constructor_t ng_patch_constructor;
43 static ng_rcvmsg_t ng_patch_rcvmsg;
44 static ng_shutdown_t ng_patch_shutdown;
45 static ng_newhook_t ng_patch_newhook;
46 static ng_rcvdata_t ng_patch_rcvdata;
47 static ng_disconnect_t ng_patch_disconnect;
50 ng_patch_config_getlen(const struct ng_parse_type *type,
51 const u_char *start, const u_char *buf)
53 const struct ng_patch_config *p;
55 p = (const struct ng_patch_config *)(buf -
56 offsetof(struct ng_patch_config, ops));
60 static const struct ng_parse_struct_field ng_patch_op_type_fields[]
61 = NG_PATCH_OP_TYPE_INFO;
62 static const struct ng_parse_type ng_patch_op_type = {
63 &ng_parse_struct_type,
64 &ng_patch_op_type_fields
67 static const struct ng_parse_array_info ng_patch_confarr_info = {
69 &ng_patch_config_getlen
71 static const struct ng_parse_type ng_patch_confarr_type = {
73 &ng_patch_confarr_info
76 static const struct ng_parse_struct_field ng_patch_config_type_fields[]
77 = NG_PATCH_CONFIG_TYPE_INFO;
78 static const struct ng_parse_type ng_patch_config_type = {
79 &ng_parse_struct_type,
80 &ng_patch_config_type_fields
83 static const struct ng_parse_struct_field ng_patch_stats_fields[]
84 = NG_PATCH_STATS_TYPE_INFO;
85 static const struct ng_parse_type ng_patch_stats_type = {
86 &ng_parse_struct_type,
87 &ng_patch_stats_fields
90 static const struct ng_cmdlist ng_patch_cmdlist[] = {
102 &ng_patch_config_type,
121 NGM_PATCH_GETCLR_STATS,
129 static struct ng_type typestruct = {
130 .version = NG_ABI_VERSION,
131 .name = NG_PATCH_NODE_TYPE,
132 .constructor = ng_patch_constructor,
133 .rcvmsg = ng_patch_rcvmsg,
134 .shutdown = ng_patch_shutdown,
135 .newhook = ng_patch_newhook,
136 .rcvdata = ng_patch_rcvdata,
137 .disconnect = ng_patch_disconnect,
138 .cmdlist = ng_patch_cmdlist,
140 NETGRAPH_INIT(patch, &typestruct);
149 struct ng_patch_priv {
152 struct ng_patch_config *config;
153 union patch_val *val;
154 struct ng_patch_stats stats;
156 typedef struct ng_patch_priv *priv_p;
158 #define NG_PATCH_CONF_SIZE(count) (sizeof(struct ng_patch_config) + \
159 (count) * sizeof(struct ng_patch_op))
161 static void do_patch(priv_p conf, struct mbuf *m);
164 ng_patch_constructor(node_p node)
168 privdata = malloc(sizeof(*privdata), M_NETGRAPH, M_WAITOK | M_ZERO);
169 NG_NODE_SET_PRIVATE(node, privdata);
171 privdata->out = NULL;
172 privdata->config = NULL;
177 ng_patch_newhook(node_p node, hook_p hook, const char *name)
179 const priv_p privp = NG_NODE_PRIVATE(node);
181 if (strncmp(name, NG_PATCH_HOOK_IN, strlen(NG_PATCH_HOOK_IN)) == 0) {
183 } else if (strncmp(name, NG_PATCH_HOOK_OUT,
184 strlen(NG_PATCH_HOOK_OUT)) == 0) {
192 ng_patch_rcvmsg(node_p node, item_p item, hook_p lasthook)
194 const priv_p privp = NG_NODE_PRIVATE(node);
195 struct ng_patch_config *conf, *newconf;
196 union patch_val *newval;
198 struct ng_mesg *resp;
203 NGI_GET_MSG(item, msg);
204 switch (msg->header.typecookie) {
205 case NGM_PATCH_COOKIE:
206 switch (msg->header.cmd) {
207 case NGM_PATCH_GETCONFIG:
208 if (privp->config == NULL)
210 NG_MKRESPONSE(resp, msg,
211 NG_PATCH_CONF_SIZE(privp->config->count),
213 bcopy(privp->config, resp->data,
214 NG_PATCH_CONF_SIZE(privp->config->count));
216 case NGM_PATCH_SETCONFIG:
218 if (msg->header.arglen <
219 sizeof(struct ng_patch_config)) {
224 conf = (struct ng_patch_config *)msg->data;
225 if (msg->header.arglen <
226 NG_PATCH_CONF_SIZE(conf->count)) {
231 for(i = 0; i < conf->count; i++) {
232 switch(conf->ops[i].length) {
246 conf->csum_flags &= CSUM_IP | CSUM_TCP | CSUM_UDP |
251 NG_PATCH_CONF_SIZE(conf->count),
252 M_NETGRAPH, M_WAITOK);
253 newval = malloc(conf->count *
254 sizeof(union patch_val), M_NETGRAPH,
256 for(i = 0; i < conf->count; i++) {
257 switch (conf->ops[i].length) {
277 NG_PATCH_CONF_SIZE(conf->count));
278 if (privp->val != NULL)
279 free(privp->val, M_NETGRAPH);
281 if (privp->config != NULL)
282 free(privp->config, M_NETGRAPH);
283 privp->config = newconf;
287 case NGM_PATCH_GETCLR_STATS:
290 case NGM_PATCH_GET_STATS:
291 NG_MKRESPONSE(resp, msg, sizeof(struct ng_patch_stats),
293 bcopy(&(privp->stats), resp->data,
294 sizeof(struct ng_patch_stats));
297 /* else FALLTHROUGH */
298 case NGM_PATCH_CLR_STATS:
299 bzero(&(privp->stats), sizeof(struct ng_patch_stats));
311 NG_RESPOND_MSG(error, node, item, resp);
317 do_patch(priv_p privp, struct mbuf *m)
319 struct ng_patch_config *conf;
323 conf = privp->config;
325 for(i = 0; i < conf->count; i++) {
326 if (conf->ops[i].offset + conf->ops[i].length >
330 /* for "=" operation we don't need to copy data from mbuf */
331 if (conf->ops[i].mode != NG_PATCH_MODE_SET) {
332 m_copydata(m, conf->ops[i].offset,
333 conf->ops[i].length, (caddr_t)&buf);
336 switch (conf->ops[i].length) {
338 switch (conf->ops[i].mode) {
339 case NG_PATCH_MODE_SET:
340 *((uint8_t *)&buf) = privp->val[i].v1;
342 case NG_PATCH_MODE_ADD:
343 *((uint8_t *)&buf) += privp->val[i].v1;
345 case NG_PATCH_MODE_SUB:
346 *((uint8_t *)&buf) -= privp->val[i].v1;
348 case NG_PATCH_MODE_MUL:
349 *((uint8_t *)&buf) *= privp->val[i].v1;
351 case NG_PATCH_MODE_DIV:
352 *((uint8_t *)&buf) /= privp->val[i].v1;
354 case NG_PATCH_MODE_NEG:
355 *((int8_t *)&buf) = - *((int8_t *)&buf);
357 case NG_PATCH_MODE_AND:
358 *((uint8_t *)&buf) &= privp->val[i].v1;
360 case NG_PATCH_MODE_OR:
361 *((uint8_t *)&buf) |= privp->val[i].v1;
363 case NG_PATCH_MODE_XOR:
364 *((uint8_t *)&buf) ^= privp->val[i].v1;
366 case NG_PATCH_MODE_SHL:
367 *((uint8_t *)&buf) <<= privp->val[i].v1;
369 case NG_PATCH_MODE_SHR:
370 *((uint8_t *)&buf) >>= privp->val[i].v1;
375 *((int16_t *)&buf) = ntohs(*((int16_t *)&buf));
376 switch (conf->ops[i].mode) {
377 case NG_PATCH_MODE_SET:
378 *((uint16_t *)&buf) = privp->val[i].v2;
380 case NG_PATCH_MODE_ADD:
381 *((uint16_t *)&buf) += privp->val[i].v2;
383 case NG_PATCH_MODE_SUB:
384 *((uint16_t *)&buf) -= privp->val[i].v2;
386 case NG_PATCH_MODE_MUL:
387 *((uint16_t *)&buf) *= privp->val[i].v2;
389 case NG_PATCH_MODE_DIV:
390 *((uint16_t *)&buf) /= privp->val[i].v2;
392 case NG_PATCH_MODE_NEG:
393 *((int16_t *)&buf) = - *((int16_t *)&buf);
395 case NG_PATCH_MODE_AND:
396 *((uint16_t *)&buf) &= privp->val[i].v2;
398 case NG_PATCH_MODE_OR:
399 *((uint16_t *)&buf) |= privp->val[i].v2;
401 case NG_PATCH_MODE_XOR:
402 *((uint16_t *)&buf) ^= privp->val[i].v2;
404 case NG_PATCH_MODE_SHL:
405 *((uint16_t *)&buf) <<= privp->val[i].v2;
407 case NG_PATCH_MODE_SHR:
408 *((uint16_t *)&buf) >>= privp->val[i].v2;
411 *((int16_t *)&buf) = htons(*((int16_t *)&buf));
414 *((int32_t *)&buf) = ntohl(*((int32_t *)&buf));
415 switch (conf->ops[i].mode) {
416 case NG_PATCH_MODE_SET:
417 *((uint32_t *)&buf) = privp->val[i].v4;
419 case NG_PATCH_MODE_ADD:
420 *((uint32_t *)&buf) += privp->val[i].v4;
422 case NG_PATCH_MODE_SUB:
423 *((uint32_t *)&buf) -= privp->val[i].v4;
425 case NG_PATCH_MODE_MUL:
426 *((uint32_t *)&buf) *= privp->val[i].v4;
428 case NG_PATCH_MODE_DIV:
429 *((uint32_t *)&buf) /= privp->val[i].v4;
431 case NG_PATCH_MODE_NEG:
432 *((int32_t *)&buf) = - *((int32_t *)&buf);
434 case NG_PATCH_MODE_AND:
435 *((uint32_t *)&buf) &= privp->val[i].v4;
437 case NG_PATCH_MODE_OR:
438 *((uint32_t *)&buf) |= privp->val[i].v4;
440 case NG_PATCH_MODE_XOR:
441 *((uint32_t *)&buf) ^= privp->val[i].v4;
443 case NG_PATCH_MODE_SHL:
444 *((uint32_t *)&buf) <<= privp->val[i].v4;
446 case NG_PATCH_MODE_SHR:
447 *((uint32_t *)&buf) >>= privp->val[i].v4;
450 *((int32_t *)&buf) = htonl(*((int32_t *)&buf));
453 *((int64_t *)&buf) = be64toh(*((int64_t *)&buf));
454 switch (conf->ops[i].mode) {
455 case NG_PATCH_MODE_SET:
456 *((uint64_t *)&buf) = privp->val[i].v8;
458 case NG_PATCH_MODE_ADD:
459 *((uint64_t *)&buf) += privp->val[i].v8;
461 case NG_PATCH_MODE_SUB:
462 *((uint64_t *)&buf) -= privp->val[i].v8;
464 case NG_PATCH_MODE_MUL:
465 *((uint64_t *)&buf) *= privp->val[i].v8;
467 case NG_PATCH_MODE_DIV:
468 *((uint64_t *)&buf) /= privp->val[i].v8;
470 case NG_PATCH_MODE_NEG:
471 *((int64_t *)&buf) = - *((int64_t *)&buf);
473 case NG_PATCH_MODE_AND:
474 *((uint64_t *)&buf) &= privp->val[i].v8;
476 case NG_PATCH_MODE_OR:
477 *((uint64_t *)&buf) |= privp->val[i].v8;
479 case NG_PATCH_MODE_XOR:
480 *((uint64_t *)&buf) ^= privp->val[i].v8;
482 case NG_PATCH_MODE_SHL:
483 *((uint64_t *)&buf) <<= privp->val[i].v8;
485 case NG_PATCH_MODE_SHR:
486 *((uint64_t *)&buf) >>= privp->val[i].v8;
489 *((int64_t *)&buf) = htobe64(*((int64_t *)&buf));
493 m_copyback(m, conf->ops[i].offset, conf->ops[i].length,
498 privp->stats.patched++;
502 ng_patch_rcvdata(hook_p hook, item_p item)
504 const priv_p priv = NG_NODE_PRIVATE(NG_HOOK_NODE(hook));
509 priv->stats.received++;
511 if (priv->config != NULL && hook == priv->in &&
512 (m->m_flags & M_PKTHDR) != 0) {
513 m = m_unshare(m,M_NOWAIT);
515 priv->stats.dropped++;
520 m->m_pkthdr.csum_flags |= priv->config->csum_flags;
524 if (hook == priv->in) {
525 /* return frames on 'in' hook if 'out' not connected */
526 if (priv->out != NULL)
531 if (hook == priv->out && priv->in != NULL)
534 if (target == NULL) {
535 priv->stats.dropped++;
540 NG_FWD_NEW_DATA(error, item, target, m);
545 ng_patch_shutdown(node_p node)
547 const priv_p privdata = NG_NODE_PRIVATE(node);
549 if (privdata->val != NULL)
550 free(privdata->val, M_NETGRAPH);
551 if (privdata->config != NULL)
552 free(privdata->config, M_NETGRAPH);
553 NG_NODE_SET_PRIVATE(node, NULL);
555 free(privdata, M_NETGRAPH);
560 ng_patch_disconnect(hook_p hook)
564 priv = NG_NODE_PRIVATE(NG_HOOK_NODE(hook));
565 if (hook == priv->in) {
568 if (hook == priv->out) {
571 if (NG_NODE_NUMHOOKS(NG_HOOK_NODE(hook)) == 0 &&
572 NG_NODE_IS_VALID(NG_HOOK_NODE(hook))) /* already shutting down? */
573 ng_rmnode_self(NG_HOOK_NODE(hook));