2 * m_xt.c xtables based targets
3 * utilities mostly ripped from iptables <duh, its the linux way>
5 * This program is free software; you can distribute it and/or
6 * modify it under the terms of the GNU General Public License
7 * as published by the Free Software Foundation; either version
8 * 2 of the License, or (at your option) any later version.
10 * Authors: J Hadi Salim (hadi@cyberus.ca)
14 #include <sys/socket.h>
15 #include <netinet/in.h>
16 #include <arpa/inet.h>
19 #include <linux/netfilter.h>
20 #include <linux/netfilter_ipv4/ip_tables.h>
24 #include <linux/tc_act/tc_ipt.h>
39 # define XT_LIB_DIR "/lib/xtables"
43 #define ALIGN(x,a) __ALIGN_MASK(x,(typeof(x))(a)-1)
44 #define __ALIGN_MASK(x,mask) (((x)+(mask))&~(mask))
47 static const char *tname = "mangle";
51 static const char *ipthooks[] = {
59 static struct option original_opts[] = {
68 static struct xtables_globals tcipt_globals = {
70 .program_name = "tc-ipt",
71 .program_version = "0.2",
72 .orig_opts = original_opts,
73 .opts = original_opts,
78 * we may need to check for version mismatch
81 build_st(struct xtables_target *target, struct xt_entry_target *t)
85 XT_ALIGN(sizeof (struct xt_entry_target)) + target->size;
88 target->t = xtables_calloc(1, size);
89 target->t->u.target_size = size;
90 strcpy(target->t->u.user.name, target->name);
91 target->t->u.user.revision = target->revision;
93 if (target->init != NULL)
94 target->init(target->t);
102 inline void set_lib_dir(void)
105 lib_dir = getenv("XTABLES_LIBDIR");
107 lib_dir = getenv("IPTABLES_LIB_DIR");
109 fprintf(stderr, "using deprecated IPTABLES_LIB_DIR \n");
112 lib_dir = XT_LIB_DIR;
116 static int parse_ipt(struct action_util *a,int *argc_p,
117 char ***argv_p, int tca_id, struct nlmsghdr *n)
119 struct xtables_target *m = NULL;
124 char **argv = *argv_p;
125 int argc = 0, iargc = 0;
129 __u32 hook = 0, index = 0;
131 xtables_init_all(&tcipt_globals, NFPROTO_IPV4);
136 for (i = 0; i < rargc; i++) {
137 if (NULL == argv[i] || 0 == strcmp(argv[i], "action")) {
145 fprintf(stderr,"bad arguements to ipt %d vs %d \n", argc, rargc);
150 c = getopt_long(argc, argv, "j:", tcipt_globals.opts, NULL);
155 m = xtables_find_target(optarg, XTF_TRY_LOAD);
158 if (0 > build_st(m, NULL)) {
159 printf(" %s error \n", m->name);
163 xtables_merge_options(
164 #if (XTABLES_VERSION_CODE >= 6)
165 tcipt_globals.orig_opts,
171 fprintf(stderr," failed to find target %s\n\n", optarg);
178 memset(&fw, 0, sizeof (fw));
180 m->parse(c - m->option_offset, argv, 0,
181 &m->tflags, NULL, &m->t);
183 fprintf(stderr," failed to find target %s\n\n", optarg);
193 if (iargc > optind) {
194 if (matches(argv[optind], "index") == 0) {
195 if (get_u32(&index, argv[optind + 1], 10)) {
196 fprintf(stderr, "Illegal \"index\"\n");
197 xtables_free_opts(1);
207 fprintf(stderr," ipt Parser BAD!! (%s)\n", *argv);
211 /* check that we passed the correct parameters to the target */
212 if (m && m->final_check)
213 m->final_check(m->tflags);
216 struct tcmsg *t = NLMSG_DATA(n);
217 if (t->tcm_parent != TC_H_ROOT
218 && t->tcm_parent == TC_H_MAJ(TC_H_INGRESS)) {
219 hook = NF_IP_PRE_ROUTING;
221 hook = NF_IP_POST_ROUTING;
225 tail = NLMSG_TAIL(n);
226 addattr_l(n, MAX_MSG, tca_id, NULL, 0);
227 fprintf(stdout, "tablename: %s hook: %s\n ", tname, ipthooks[hook]);
228 fprintf(stdout, "\ttarget: ");
231 m->print(NULL, m->t, 0);
232 fprintf(stdout, " index %d\n", index);
234 if (strlen(tname) > 16) {
238 size = 1 + strlen(tname);
240 strncpy(k, tname, size);
242 addattr_l(n, MAX_MSG, TCA_IPT_TABLE, k, size);
243 addattr_l(n, MAX_MSG, TCA_IPT_HOOK, &hook, 4);
244 addattr_l(n, MAX_MSG, TCA_IPT_INDEX, &index, 4);
246 addattr_l(n, MAX_MSG, TCA_IPT_TARG, m->t, m->t->u.target_size);
247 tail->rta_len = (void *) NLMSG_TAIL(n) - (void *) tail;
251 *argc_p = rargc - iargc;
255 xtables_free_opts(1);
256 /* Clear flags if target will be used again */
259 /* Free allocated memory */
269 print_ipt(struct action_util *au,FILE * f, struct rtattr *arg)
271 struct rtattr *tb[TCA_IPT_MAX + 1];
272 struct xt_entry_target *t = NULL;
277 xtables_init_all(&tcipt_globals, NFPROTO_IPV4);
280 parse_rtattr_nested(tb, TCA_IPT_MAX, arg);
282 if (tb[TCA_IPT_TABLE] == NULL) {
283 fprintf(f, "[NULL ipt table name ] assuming mangle ");
285 fprintf(f, "tablename: %s ",
286 (char *) RTA_DATA(tb[TCA_IPT_TABLE]));
289 if (tb[TCA_IPT_HOOK] == NULL) {
290 fprintf(f, "[NULL ipt hook name ]\n ");
294 hook = *(__u32 *) RTA_DATA(tb[TCA_IPT_HOOK]);
295 fprintf(f, " hook: %s \n", ipthooks[hook]);
298 if (tb[TCA_IPT_TARG] == NULL) {
299 fprintf(f, "\t[NULL ipt target parameters ] \n");
302 struct xtables_target *m = NULL;
303 t = RTA_DATA(tb[TCA_IPT_TARG]);
304 m = xtables_find_target(t->u.user.name, XTF_TRY_LOAD);
306 if (0 > build_st(m, t)) {
307 fprintf(stderr, " %s error \n", m->name);
312 xtables_merge_options(
313 #if (XTABLES_VERSION_CODE >= 6)
314 tcipt_globals.orig_opts,
320 fprintf(stderr, " failed to find target %s\n\n",
324 fprintf(f, "\ttarget ");
325 m->print(NULL, m->t, 0);
326 if (tb[TCA_IPT_INDEX] == NULL) {
327 fprintf(f, " [NULL ipt target index ]\n");
330 index = *(__u32 *) RTA_DATA(tb[TCA_IPT_INDEX]);
331 fprintf(f, " \n\tindex %d", index);
334 if (tb[TCA_IPT_CNT]) {
335 struct tc_cnt *c = RTA_DATA(tb[TCA_IPT_CNT]);;
336 fprintf(f, " ref %d bind %d", c->refcnt, c->bindcnt);
339 if (tb[TCA_IPT_TM]) {
340 struct tcf_t *tm = RTA_DATA(tb[TCA_IPT_TM]);
347 xtables_free_opts(1);
352 struct action_util ipt_action_util = {
354 .parse_aopt = parse_ipt,
355 .print_aopt = print_ipt,