2 * Copyright (c) 2010 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 *conf;
55 conf = (const struct ng_patch_config *)(buf -
56 offsetof(struct ng_patch_config, ops));
61 static const struct ng_parse_struct_field ng_patch_op_type_fields[]
62 = NG_PATCH_OP_TYPE_INFO;
63 static const struct ng_parse_type ng_patch_op_type = {
64 &ng_parse_struct_type,
65 &ng_patch_op_type_fields
68 static const struct ng_parse_array_info ng_patch_ops_array_info = {
70 &ng_patch_config_getlen
72 static const struct ng_parse_type ng_patch_ops_array_type = {
74 &ng_patch_ops_array_info
77 static const struct ng_parse_struct_field ng_patch_config_type_fields[]
78 = NG_PATCH_CONFIG_TYPE_INFO;
79 static const struct ng_parse_type ng_patch_config_type = {
80 &ng_parse_struct_type,
81 &ng_patch_config_type_fields
84 static const struct ng_parse_struct_field ng_patch_stats_fields[]
85 = NG_PATCH_STATS_TYPE_INFO;
86 static const struct ng_parse_type ng_patch_stats_type = {
87 &ng_parse_struct_type,
88 &ng_patch_stats_fields
91 static const struct ng_cmdlist ng_patch_cmdlist[] = {
103 &ng_patch_config_type,
122 NGM_PATCH_GETCLR_STATS,
130 static struct ng_type typestruct = {
131 .version = NG_ABI_VERSION,
132 .name = NG_PATCH_NODE_TYPE,
133 .constructor = ng_patch_constructor,
134 .rcvmsg = ng_patch_rcvmsg,
135 .shutdown = ng_patch_shutdown,
136 .newhook = ng_patch_newhook,
137 .rcvdata = ng_patch_rcvdata,
138 .disconnect = ng_patch_disconnect,
139 .cmdlist = ng_patch_cmdlist,
142 NETGRAPH_INIT(patch, &typestruct);
152 struct ng_patch_priv {
155 struct ng_patch_config *config;
156 union patch_val *val;
157 struct ng_patch_stats stats;
159 typedef struct ng_patch_priv *priv_p;
161 #define NG_PATCH_CONF_SIZE(count) (sizeof(struct ng_patch_config) + \
162 (count) * sizeof(struct ng_patch_op))
164 static void do_patch(priv_p conf, struct mbuf *m);
167 ng_patch_constructor(node_p node)
171 privdata = malloc(sizeof(*privdata), M_NETGRAPH, M_WAITOK | M_ZERO);
172 NG_NODE_SET_PRIVATE(node, privdata);
174 privdata->out = NULL;
175 privdata->config = NULL;
180 ng_patch_newhook(node_p node, hook_p hook, const char *name)
182 const priv_p privp = NG_NODE_PRIVATE(node);
184 if (strncmp(name, NG_PATCH_HOOK_IN, strlen(NG_PATCH_HOOK_IN)) == 0) {
186 } else if (strncmp(name, NG_PATCH_HOOK_OUT,
187 strlen(NG_PATCH_HOOK_OUT)) == 0) {
195 ng_patch_rcvmsg(node_p node, item_p item, hook_p lasthook)
197 const priv_p privp = NG_NODE_PRIVATE(node);
198 struct ng_patch_config *conf, *newconf;
199 union patch_val *newval;
201 struct ng_mesg *resp;
206 NGI_GET_MSG(item, msg);
207 switch (msg->header.typecookie) {
208 case NGM_PATCH_COOKIE:
209 switch (msg->header.cmd) {
210 case NGM_PATCH_GETCONFIG:
211 if (privp->config == NULL)
213 NG_MKRESPONSE(resp, msg,
214 NG_PATCH_CONF_SIZE(privp->config->count),
216 bcopy(privp->config, resp->data,
217 NG_PATCH_CONF_SIZE(privp->config->count));
219 case NGM_PATCH_SETCONFIG:
221 if (msg->header.arglen <
222 sizeof(struct ng_patch_config)) {
227 conf = (struct ng_patch_config *)msg->data;
228 if (msg->header.arglen <
229 NG_PATCH_CONF_SIZE(conf->count)) {
234 for(i = 0; i < conf->count; i++) {
235 switch(conf->ops[i].length) {
249 conf->csum_flags &= CSUM_IP | CSUM_TCP | CSUM_UDP |
254 NG_PATCH_CONF_SIZE(conf->count),
255 M_NETGRAPH, M_WAITOK);
256 newval = malloc(conf->count *
257 sizeof(union patch_val), M_NETGRAPH,
259 for(i = 0; i < conf->count; i++) {
260 switch (conf->ops[i].length) {
280 NG_PATCH_CONF_SIZE(conf->count));
281 if (privp->val != NULL)
282 free(privp->val, M_NETGRAPH);
284 if (privp->config != NULL)
285 free(privp->config, M_NETGRAPH);
286 privp->config = newconf;
290 case NGM_PATCH_GETCLR_STATS:
293 case NGM_PATCH_GET_STATS:
294 NG_MKRESPONSE(resp, msg, sizeof(struct ng_patch_stats),
296 bcopy(&(privp->stats), resp->data,
297 sizeof(struct ng_patch_stats));
300 /* else FALLTHROUGH */
301 case NGM_PATCH_CLR_STATS:
302 bzero(&(privp->stats), sizeof(struct ng_patch_stats));
314 NG_RESPOND_MSG(error, node, item, resp);
320 do_patch(priv_p privp, struct mbuf *m)
322 struct ng_patch_config *conf;
326 conf = privp->config;
328 for(i = 0; i < conf->count; i++) {
329 if (conf->ops[i].offset + conf->ops[i].length >
333 /* for "=" operation we don't need to copy data from mbuf */
334 if (conf->ops[i].mode != NG_PATCH_MODE_SET) {
335 m_copydata(m, conf->ops[i].offset,
336 conf->ops[i].length, (caddr_t)&buf);
339 switch (conf->ops[i].length) {
341 switch (conf->ops[i].mode) {
342 case NG_PATCH_MODE_SET:
343 *((uint8_t *)&buf) = privp->val[i].v1;
345 case NG_PATCH_MODE_ADD:
346 *((uint8_t *)&buf) += privp->val[i].v1;
348 case NG_PATCH_MODE_SUB:
349 *((uint8_t *)&buf) -= privp->val[i].v1;
351 case NG_PATCH_MODE_MUL:
352 *((uint8_t *)&buf) *= privp->val[i].v1;
354 case NG_PATCH_MODE_DIV:
355 *((uint8_t *)&buf) /= privp->val[i].v1;
357 case NG_PATCH_MODE_NEG:
358 *((int8_t *)&buf) = - *((int8_t *)&buf);
360 case NG_PATCH_MODE_AND:
361 *((uint8_t *)&buf) &= privp->val[i].v1;
363 case NG_PATCH_MODE_OR:
364 *((uint8_t *)&buf) |= privp->val[i].v1;
366 case NG_PATCH_MODE_XOR:
367 *((uint8_t *)&buf) ^= privp->val[i].v1;
369 case NG_PATCH_MODE_SHL:
370 *((uint8_t *)&buf) <<= privp->val[i].v1;
372 case NG_PATCH_MODE_SHR:
373 *((uint8_t *)&buf) >>= privp->val[i].v1;
378 *((int16_t *)&buf) = ntohs(*((int16_t *)&buf));
379 switch (conf->ops[i].mode) {
380 case NG_PATCH_MODE_SET:
381 *((uint16_t *)&buf) = privp->val[i].v2;
383 case NG_PATCH_MODE_ADD:
384 *((uint16_t *)&buf) += privp->val[i].v2;
386 case NG_PATCH_MODE_SUB:
387 *((uint16_t *)&buf) -= privp->val[i].v2;
389 case NG_PATCH_MODE_MUL:
390 *((uint16_t *)&buf) *= privp->val[i].v2;
392 case NG_PATCH_MODE_DIV:
393 *((uint16_t *)&buf) /= privp->val[i].v2;
395 case NG_PATCH_MODE_NEG:
396 *((int16_t *)&buf) = - *((int16_t *)&buf);
398 case NG_PATCH_MODE_AND:
399 *((uint16_t *)&buf) &= privp->val[i].v2;
401 case NG_PATCH_MODE_OR:
402 *((uint16_t *)&buf) |= privp->val[i].v2;
404 case NG_PATCH_MODE_XOR:
405 *((uint16_t *)&buf) ^= privp->val[i].v2;
407 case NG_PATCH_MODE_SHL:
408 *((uint16_t *)&buf) <<= privp->val[i].v2;
410 case NG_PATCH_MODE_SHR:
411 *((uint16_t *)&buf) >>= privp->val[i].v2;
414 *((int16_t *)&buf) = htons(*((int16_t *)&buf));
417 *((int32_t *)&buf) = ntohl(*((int32_t *)&buf));
418 switch (conf->ops[i].mode) {
419 case NG_PATCH_MODE_SET:
420 *((uint32_t *)&buf) = privp->val[i].v4;
422 case NG_PATCH_MODE_ADD:
423 *((uint32_t *)&buf) += privp->val[i].v4;
425 case NG_PATCH_MODE_SUB:
426 *((uint32_t *)&buf) -= privp->val[i].v4;
428 case NG_PATCH_MODE_MUL:
429 *((uint32_t *)&buf) *= privp->val[i].v4;
431 case NG_PATCH_MODE_DIV:
432 *((uint32_t *)&buf) /= privp->val[i].v4;
434 case NG_PATCH_MODE_NEG:
435 *((int32_t *)&buf) = - *((int32_t *)&buf);
437 case NG_PATCH_MODE_AND:
438 *((uint32_t *)&buf) &= privp->val[i].v4;
440 case NG_PATCH_MODE_OR:
441 *((uint32_t *)&buf) |= privp->val[i].v4;
443 case NG_PATCH_MODE_XOR:
444 *((uint32_t *)&buf) ^= privp->val[i].v4;
446 case NG_PATCH_MODE_SHL:
447 *((uint32_t *)&buf) <<= privp->val[i].v4;
449 case NG_PATCH_MODE_SHR:
450 *((uint32_t *)&buf) >>= privp->val[i].v4;
453 *((int32_t *)&buf) = htonl(*((int32_t *)&buf));
456 *((int64_t *)&buf) = be64toh(*((int64_t *)&buf));
457 switch (conf->ops[i].mode) {
458 case NG_PATCH_MODE_SET:
459 *((uint64_t *)&buf) = privp->val[i].v8;
461 case NG_PATCH_MODE_ADD:
462 *((uint64_t *)&buf) += privp->val[i].v8;
464 case NG_PATCH_MODE_SUB:
465 *((uint64_t *)&buf) -= privp->val[i].v8;
467 case NG_PATCH_MODE_MUL:
468 *((uint64_t *)&buf) *= privp->val[i].v8;
470 case NG_PATCH_MODE_DIV:
471 *((uint64_t *)&buf) /= privp->val[i].v8;
473 case NG_PATCH_MODE_NEG:
474 *((int64_t *)&buf) = - *((int64_t *)&buf);
476 case NG_PATCH_MODE_AND:
477 *((uint64_t *)&buf) &= privp->val[i].v8;
479 case NG_PATCH_MODE_OR:
480 *((uint64_t *)&buf) |= privp->val[i].v8;
482 case NG_PATCH_MODE_XOR:
483 *((uint64_t *)&buf) ^= privp->val[i].v8;
485 case NG_PATCH_MODE_SHL:
486 *((uint64_t *)&buf) <<= privp->val[i].v8;
488 case NG_PATCH_MODE_SHR:
489 *((uint64_t *)&buf) >>= privp->val[i].v8;
492 *((int64_t *)&buf) = htobe64(*((int64_t *)&buf));
496 m_copyback(m, conf->ops[i].offset, conf->ops[i].length,
501 privp->stats.patched++;
505 ng_patch_rcvdata(hook_p hook, item_p item)
507 const priv_p priv = NG_NODE_PRIVATE(NG_HOOK_NODE(hook));
512 priv->stats.received++;
514 if (priv->config != NULL && hook == priv->in &&
515 (m->m_flags & M_PKTHDR) != 0) {
516 m = m_unshare(m,M_NOWAIT);
518 priv->stats.dropped++;
523 m->m_pkthdr.csum_flags |= priv->config->csum_flags;
527 if (hook == priv->in) {
528 /* return frames on 'in' hook if 'out' not connected */
529 if (priv->out != NULL)
534 if (hook == priv->out && priv->in != NULL)
537 if (target == NULL) {
538 priv->stats.dropped++;
543 NG_FWD_NEW_DATA(error, item, target, m);
548 ng_patch_shutdown(node_p node)
550 const priv_p privdata = NG_NODE_PRIVATE(node);
552 if (privdata->val != NULL)
553 free(privdata->val, M_NETGRAPH);
554 if (privdata->config != NULL)
555 free(privdata->config, M_NETGRAPH);
556 NG_NODE_SET_PRIVATE(node, NULL);
558 free(privdata, M_NETGRAPH);
563 ng_patch_disconnect(hook_p hook)
567 priv = NG_NODE_PRIVATE(NG_HOOK_NODE(hook));
568 if (hook == priv->in) {
571 if (hook == priv->out) {
574 if (NG_NODE_NUMHOOKS(NG_HOOK_NODE(hook)) == 0 &&
575 NG_NODE_IS_VALID(NG_HOOK_NODE(hook))) /* already shutting down? */
576 ng_rmnode_self(NG_HOOK_NODE(hook));