| /* |
| * m_connmark.c Connection tracking marking import |
| * |
| * Copyright (c) 2011 Felix Fietkau <nbd@openwrt.org> |
| * |
| * This program is free software; you can redistribute it and/or modify it |
| * under the terms and conditions of the GNU General Public License, |
| * version 2, as published by the Free Software Foundation. |
| * |
| * This program is distributed in the hope it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for |
| * more details. |
| * |
| * You should have received a copy of the GNU General Public License along with |
| * this program; if not, see <http://www.gnu.org/licenses>. |
| */ |
| |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <unistd.h> |
| #include <string.h> |
| #include "utils.h" |
| #include "tc_util.h" |
| #include <linux/tc_act/tc_connmark.h> |
| |
| static void |
| explain(void) |
| { |
| fprintf(stderr, |
| "Usage: ... connmark [zone ZONE] [CONTROL] [index <INDEX>]\n" |
| "where :\n" |
| "\tZONE is the conntrack zone\n" |
| "\tCONTROL := reclassify | pipe | drop | continue | ok |\n" |
| "\t goto chain <CHAIN_INDEX>\n"); |
| } |
| |
| static void |
| usage(void) |
| { |
| explain(); |
| exit(-1); |
| } |
| |
| static int |
| parse_connmark(struct action_util *a, int *argc_p, char ***argv_p, int tca_id, |
| struct nlmsghdr *n) |
| { |
| struct tc_connmark sel = {}; |
| char **argv = *argv_p; |
| int argc = *argc_p; |
| int ok = 0; |
| struct rtattr *tail; |
| |
| while (argc > 0) { |
| if (matches(*argv, "connmark") == 0) { |
| ok = 1; |
| argc--; |
| argv++; |
| } else if (matches(*argv, "help") == 0) { |
| usage(); |
| } else { |
| break; |
| } |
| |
| } |
| |
| if (!ok) { |
| explain(); |
| return -1; |
| } |
| |
| if (argc) { |
| if (matches(*argv, "zone") == 0) { |
| NEXT_ARG(); |
| if (get_u16(&sel.zone, *argv, 10)) { |
| fprintf(stderr, "connmark: Illegal \"zone\"\n"); |
| return -1; |
| } |
| argc--; |
| argv++; |
| } |
| } |
| |
| parse_action_control_dflt(&argc, &argv, &sel.action, false, TC_ACT_PIPE); |
| |
| if (argc) { |
| if (matches(*argv, "index") == 0) { |
| NEXT_ARG(); |
| if (get_u32(&sel.index, *argv, 10)) { |
| fprintf(stderr, "connmark: Illegal \"index\"\n"); |
| return -1; |
| } |
| argc--; |
| argv++; |
| } |
| } |
| |
| tail = addattr_nest(n, MAX_MSG, tca_id); |
| addattr_l(n, MAX_MSG, TCA_CONNMARK_PARMS, &sel, sizeof(sel)); |
| addattr_nest_end(n, tail); |
| |
| *argc_p = argc; |
| *argv_p = argv; |
| return 0; |
| } |
| |
| static int print_connmark(struct action_util *au, FILE *f, struct rtattr *arg) |
| { |
| struct rtattr *tb[TCA_CONNMARK_MAX + 1]; |
| struct tc_connmark *ci; |
| |
| if (arg == NULL) |
| return -1; |
| |
| parse_rtattr_nested(tb, TCA_CONNMARK_MAX, arg); |
| if (tb[TCA_CONNMARK_PARMS] == NULL) { |
| fprintf(stderr, "Missing connmark parameters\n"); |
| return -1; |
| } |
| |
| ci = RTA_DATA(tb[TCA_CONNMARK_PARMS]); |
| |
| print_string(PRINT_ANY, "kind", "%s ", "connmark"); |
| print_uint(PRINT_ANY, "zone", "zone %u", ci->zone); |
| print_action_control(f, " ", ci->action, ""); |
| |
| print_nl(); |
| print_uint(PRINT_ANY, "index", "\t index %u", ci->index); |
| print_int(PRINT_ANY, "ref", " ref %d", ci->refcnt); |
| print_int(PRINT_ANY, "bind", " bind %d", ci->bindcnt); |
| |
| if (show_stats) { |
| if (tb[TCA_CONNMARK_TM]) { |
| struct tcf_t *tm = RTA_DATA(tb[TCA_CONNMARK_TM]); |
| |
| print_tm(f, tm); |
| } |
| } |
| print_nl(); |
| |
| return 0; |
| } |
| |
| struct action_util connmark_action_util = { |
| .id = "connmark", |
| .parse_aopt = parse_connmark, |
| .print_aopt = print_connmark, |
| }; |