1*a71a9546SAutomerger Merge Worker /*
2*a71a9546SAutomerger Merge Worker * (C) 2011 by Pablo Neira Ayuso <[email protected]>
3*a71a9546SAutomerger Merge Worker * (C) 2011 by Intra2Net AG <http://www.intra2net.com>
4*a71a9546SAutomerger Merge Worker *
5*a71a9546SAutomerger Merge Worker * This program is free software; you can redistribute it and/or modify
6*a71a9546SAutomerger Merge Worker * it under the terms of the GNU General Public License version 2 (or
7*a71a9546SAutomerger Merge Worker * any later at your option) as published by the Free Software Foundation.
8*a71a9546SAutomerger Merge Worker */
9*a71a9546SAutomerger Merge Worker #include <stdbool.h>
10*a71a9546SAutomerger Merge Worker #include <stdlib.h>
11*a71a9546SAutomerger Merge Worker #include <stdio.h>
12*a71a9546SAutomerger Merge Worker #include <string.h>
13*a71a9546SAutomerger Merge Worker #include <getopt.h>
14*a71a9546SAutomerger Merge Worker #include <xtables.h>
15*a71a9546SAutomerger Merge Worker
16*a71a9546SAutomerger Merge Worker #include <linux/netfilter/x_tables.h>
17*a71a9546SAutomerger Merge Worker #include <linux/netfilter/xt_nfacct.h>
18*a71a9546SAutomerger Merge Worker
19*a71a9546SAutomerger Merge Worker enum {
20*a71a9546SAutomerger Merge Worker O_NAME = 0,
21*a71a9546SAutomerger Merge Worker };
22*a71a9546SAutomerger Merge Worker
23*a71a9546SAutomerger Merge Worker #define s struct xt_nfacct_match_info
24*a71a9546SAutomerger Merge Worker static const struct xt_option_entry nfacct_opts[] = {
25*a71a9546SAutomerger Merge Worker {.name = "nfacct-name", .id = O_NAME, .type = XTTYPE_STRING,
26*a71a9546SAutomerger Merge Worker .min = 1, .flags = XTOPT_MAND|XTOPT_PUT, XTOPT_POINTER(s, name)},
27*a71a9546SAutomerger Merge Worker XTOPT_TABLEEND,
28*a71a9546SAutomerger Merge Worker };
29*a71a9546SAutomerger Merge Worker #undef s
30*a71a9546SAutomerger Merge Worker
nfacct_help(void)31*a71a9546SAutomerger Merge Worker static void nfacct_help(void)
32*a71a9546SAutomerger Merge Worker {
33*a71a9546SAutomerger Merge Worker printf("nfacct match options:\n"
34*a71a9546SAutomerger Merge Worker " --nfacct-name STRING Name of accouting area\n");
35*a71a9546SAutomerger Merge Worker }
36*a71a9546SAutomerger Merge Worker
nfacct_parse(struct xt_option_call * cb)37*a71a9546SAutomerger Merge Worker static void nfacct_parse(struct xt_option_call *cb)
38*a71a9546SAutomerger Merge Worker {
39*a71a9546SAutomerger Merge Worker xtables_option_parse(cb);
40*a71a9546SAutomerger Merge Worker switch (cb->entry->id) {
41*a71a9546SAutomerger Merge Worker case O_NAME:
42*a71a9546SAutomerger Merge Worker if (strchr(cb->arg, '\n') != NULL)
43*a71a9546SAutomerger Merge Worker xtables_error(PARAMETER_PROBLEM,
44*a71a9546SAutomerger Merge Worker "Newlines not allowed in --nfacct-name");
45*a71a9546SAutomerger Merge Worker break;
46*a71a9546SAutomerger Merge Worker }
47*a71a9546SAutomerger Merge Worker }
48*a71a9546SAutomerger Merge Worker
49*a71a9546SAutomerger Merge Worker static void
nfacct_print_name(const struct xt_nfacct_match_info * info,char * name)50*a71a9546SAutomerger Merge Worker nfacct_print_name(const struct xt_nfacct_match_info *info, char *name)
51*a71a9546SAutomerger Merge Worker {
52*a71a9546SAutomerger Merge Worker printf(" %snfacct-name ", name);
53*a71a9546SAutomerger Merge Worker xtables_save_string(info->name);
54*a71a9546SAutomerger Merge Worker }
55*a71a9546SAutomerger Merge Worker
nfacct_print(const void * ip,const struct xt_entry_match * match,int numeric)56*a71a9546SAutomerger Merge Worker static void nfacct_print(const void *ip, const struct xt_entry_match *match,
57*a71a9546SAutomerger Merge Worker int numeric)
58*a71a9546SAutomerger Merge Worker {
59*a71a9546SAutomerger Merge Worker const struct xt_nfacct_match_info *info =
60*a71a9546SAutomerger Merge Worker (struct xt_nfacct_match_info *)match->data;
61*a71a9546SAutomerger Merge Worker
62*a71a9546SAutomerger Merge Worker nfacct_print_name(info, "");
63*a71a9546SAutomerger Merge Worker }
64*a71a9546SAutomerger Merge Worker
nfacct_save(const void * ip,const struct xt_entry_match * match)65*a71a9546SAutomerger Merge Worker static void nfacct_save(const void *ip, const struct xt_entry_match *match)
66*a71a9546SAutomerger Merge Worker {
67*a71a9546SAutomerger Merge Worker const struct xt_nfacct_match_info *info =
68*a71a9546SAutomerger Merge Worker (struct xt_nfacct_match_info *)match->data;
69*a71a9546SAutomerger Merge Worker
70*a71a9546SAutomerger Merge Worker nfacct_print_name(info, "--");
71*a71a9546SAutomerger Merge Worker }
72*a71a9546SAutomerger Merge Worker
73*a71a9546SAutomerger Merge Worker static struct xtables_match nfacct_matches[] = {
74*a71a9546SAutomerger Merge Worker {
75*a71a9546SAutomerger Merge Worker .family = NFPROTO_UNSPEC,
76*a71a9546SAutomerger Merge Worker .revision = 0,
77*a71a9546SAutomerger Merge Worker .name = "nfacct",
78*a71a9546SAutomerger Merge Worker .version = XTABLES_VERSION,
79*a71a9546SAutomerger Merge Worker .size = XT_ALIGN(sizeof(struct xt_nfacct_match_info)),
80*a71a9546SAutomerger Merge Worker .userspacesize = offsetof(struct xt_nfacct_match_info, nfacct),
81*a71a9546SAutomerger Merge Worker .help = nfacct_help,
82*a71a9546SAutomerger Merge Worker .x6_parse = nfacct_parse,
83*a71a9546SAutomerger Merge Worker .print = nfacct_print,
84*a71a9546SAutomerger Merge Worker .save = nfacct_save,
85*a71a9546SAutomerger Merge Worker .x6_options = nfacct_opts,
86*a71a9546SAutomerger Merge Worker },
87*a71a9546SAutomerger Merge Worker {
88*a71a9546SAutomerger Merge Worker .family = NFPROTO_UNSPEC,
89*a71a9546SAutomerger Merge Worker .revision = 1,
90*a71a9546SAutomerger Merge Worker .name = "nfacct",
91*a71a9546SAutomerger Merge Worker .version = XTABLES_VERSION,
92*a71a9546SAutomerger Merge Worker .size = XT_ALIGN(sizeof(struct xt_nfacct_match_info_v1)),
93*a71a9546SAutomerger Merge Worker .userspacesize = offsetof(struct xt_nfacct_match_info_v1, nfacct),
94*a71a9546SAutomerger Merge Worker .help = nfacct_help,
95*a71a9546SAutomerger Merge Worker .x6_parse = nfacct_parse,
96*a71a9546SAutomerger Merge Worker .print = nfacct_print,
97*a71a9546SAutomerger Merge Worker .save = nfacct_save,
98*a71a9546SAutomerger Merge Worker .x6_options = nfacct_opts,
99*a71a9546SAutomerger Merge Worker },
100*a71a9546SAutomerger Merge Worker };
101*a71a9546SAutomerger Merge Worker
_init(void)102*a71a9546SAutomerger Merge Worker void _init(void)
103*a71a9546SAutomerger Merge Worker {
104*a71a9546SAutomerger Merge Worker xtables_register_matches(nfacct_matches, ARRAY_SIZE(nfacct_matches));
105*a71a9546SAutomerger Merge Worker }
106