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
,