Blame view
net/sched/act_ipt.c
7.42 KB
1da177e4c Linux-2.6.12-rc2 |
1 2 3 4 5 6 7 8 9 10 11 12 |
/* * net/sched/ipt.c iptables target interface * *TODO: Add other tables. For now we only support the ipv4 table targets * * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public License * as published by the Free Software Foundation; either version * 2 of the License, or (at your option) any later version. * * Copyright: Jamal Hadi Salim (2002-4) */ |
1da177e4c Linux-2.6.12-rc2 |
13 14 |
#include <linux/types.h> #include <linux/kernel.h> |
1da177e4c Linux-2.6.12-rc2 |
15 |
#include <linux/string.h> |
1da177e4c Linux-2.6.12-rc2 |
16 |
#include <linux/errno.h> |
1da177e4c Linux-2.6.12-rc2 |
17 18 19 20 |
#include <linux/skbuff.h> #include <linux/rtnetlink.h> #include <linux/module.h> #include <linux/init.h> |
5a0e3ad6a include cleanup: ... |
21 |
#include <linux/slab.h> |
dc5fc579b [NETLINK]: Use nl... |
22 |
#include <net/netlink.h> |
1da177e4c Linux-2.6.12-rc2 |
23 24 25 26 27 |
#include <net/pkt_sched.h> #include <linux/tc_act/tc_ipt.h> #include <net/tc_act/tc_ipt.h> #include <linux/netfilter_ipv4/ip_tables.h> |
1da177e4c Linux-2.6.12-rc2 |
28 |
|
e9ce1cd3c [PKT_SCHED]: Kill... |
29 30 31 |
#define IPT_TAB_MASK 15 static struct tcf_common *tcf_ipt_ht[IPT_TAB_MASK + 1]; static u32 ipt_idx_gen; |
1da177e4c Linux-2.6.12-rc2 |
32 |
static DEFINE_RWLOCK(ipt_lock); |
e9ce1cd3c [PKT_SCHED]: Kill... |
33 34 35 36 37 |
static struct tcf_hashinfo ipt_hash_info = { .htab = tcf_ipt_ht, .hmask = IPT_TAB_MASK, .lock = &ipt_lock, }; |
1da177e4c Linux-2.6.12-rc2 |
38 |
|
87a2e70db netfilter: xtable... |
39 |
static int ipt_init_target(struct xt_entry_target *t, char *table, unsigned int hook) |
1da177e4c Linux-2.6.12-rc2 |
40 |
{ |
af5d6dc20 netfilter: xtable... |
41 |
struct xt_tgchk_param par; |
e60a13e03 [NETFILTER]: {ip,... |
42 |
struct xt_target *target; |
1da177e4c Linux-2.6.12-rc2 |
43 |
int ret = 0; |
239a87c87 [NET_SCHED]: act_... |
44 45 |
target = xt_request_find_target(AF_INET, t->u.user.name, t->u.user.revision); |
d2a7b6bad netfilter: xtable... |
46 47 |
if (IS_ERR(target)) return PTR_ERR(target); |
1da177e4c Linux-2.6.12-rc2 |
48 |
|
1da177e4c Linux-2.6.12-rc2 |
49 |
t->u.kernel.target = target; |
af5d6dc20 netfilter: xtable... |
50 51 52 53 54 |
par.table = table; par.entryinfo = NULL; par.target = target; par.targinfo = t->data; par.hook_mask = hook; |
916a917df netfilter: xtable... |
55 |
par.family = NFPROTO_IPV4; |
1da177e4c Linux-2.6.12-rc2 |
56 |
|
916a917df netfilter: xtable... |
57 |
ret = xt_check_target(&par, t->u.target_size - sizeof(*t), 0, false); |
367c67900 netfilter: xtable... |
58 |
if (ret < 0) { |
239a87c87 [NET_SCHED]: act_... |
59 |
module_put(t->u.kernel.target->me); |
18118cdbf [NETFILTER]: ipt ... |
60 |
return ret; |
239a87c87 [NET_SCHED]: act_... |
61 |
} |
367c67900 netfilter: xtable... |
62 |
return 0; |
1da177e4c Linux-2.6.12-rc2 |
63 |
} |
87a2e70db netfilter: xtable... |
64 |
static void ipt_destroy_target(struct xt_entry_target *t) |
1da177e4c Linux-2.6.12-rc2 |
65 |
{ |
a2df1648b netfilter: xtable... |
66 67 68 69 70 71 72 |
struct xt_tgdtor_param par = { .target = t->u.kernel.target, .targinfo = t->data, }; if (par.target->destroy != NULL) par.target->destroy(&par); module_put(par.target->me); |
1da177e4c Linux-2.6.12-rc2 |
73 |
} |
e9ce1cd3c [PKT_SCHED]: Kill... |
74 |
static int tcf_ipt_release(struct tcf_ipt *ipt, int bind) |
1da177e4c Linux-2.6.12-rc2 |
75 76 |
{ int ret = 0; |
e9ce1cd3c [PKT_SCHED]: Kill... |
77 |
if (ipt) { |
1da177e4c Linux-2.6.12-rc2 |
78 |
if (bind) |
e9ce1cd3c [PKT_SCHED]: Kill... |
79 80 81 82 83 84 85 |
ipt->tcf_bindcnt--; ipt->tcf_refcnt--; if (ipt->tcf_bindcnt <= 0 && ipt->tcf_refcnt <= 0) { ipt_destroy_target(ipt->tcfi_t); kfree(ipt->tcfi_tname); kfree(ipt->tcfi_t); tcf_hash_destroy(&ipt->common, &ipt_hash_info); |
1da177e4c Linux-2.6.12-rc2 |
86 87 88 89 90 |
ret = ACT_P_DELETED; } } return ret; } |
53b2bf3f8 [NET_SCHED]: Use ... |
91 92 93 94 |
static const struct nla_policy ipt_policy[TCA_IPT_MAX + 1] = { [TCA_IPT_TABLE] = { .type = NLA_STRING, .len = IFNAMSIZ }, [TCA_IPT_HOOK] = { .type = NLA_U32 }, [TCA_IPT_INDEX] = { .type = NLA_U32 }, |
87a2e70db netfilter: xtable... |
95 |
[TCA_IPT_TARG] = { .len = sizeof(struct xt_entry_target) }, |
53b2bf3f8 [NET_SCHED]: Use ... |
96 |
}; |
7ba699c60 [NET_SCHED]: Conv... |
97 |
static int tcf_ipt_init(struct nlattr *nla, struct nlattr *est, |
e9ce1cd3c [PKT_SCHED]: Kill... |
98 |
struct tc_action *a, int ovr, int bind) |
1da177e4c Linux-2.6.12-rc2 |
99 |
{ |
7ba699c60 [NET_SCHED]: Conv... |
100 |
struct nlattr *tb[TCA_IPT_MAX + 1]; |
e9ce1cd3c [PKT_SCHED]: Kill... |
101 102 |
struct tcf_ipt *ipt; struct tcf_common *pc; |
87a2e70db netfilter: xtable... |
103 |
struct xt_entry_target *td, *t; |
1da177e4c Linux-2.6.12-rc2 |
104 105 106 107 |
char *tname; int ret = 0, err; u32 hook = 0; u32 index = 0; |
cee63723b [NET_SCHED]: Prop... |
108 |
if (nla == NULL) |
1da177e4c Linux-2.6.12-rc2 |
109 |
return -EINVAL; |
53b2bf3f8 [NET_SCHED]: Use ... |
110 |
err = nla_parse_nested(tb, TCA_IPT_MAX, nla, ipt_policy); |
cee63723b [NET_SCHED]: Prop... |
111 112 |
if (err < 0) return err; |
53b2bf3f8 [NET_SCHED]: Use ... |
113 |
if (tb[TCA_IPT_HOOK] == NULL) |
1da177e4c Linux-2.6.12-rc2 |
114 |
return -EINVAL; |
53b2bf3f8 [NET_SCHED]: Use ... |
115 |
if (tb[TCA_IPT_TARG] == NULL) |
1da177e4c Linux-2.6.12-rc2 |
116 |
return -EINVAL; |
53b2bf3f8 [NET_SCHED]: Use ... |
117 |
|
87a2e70db netfilter: xtable... |
118 |
td = (struct xt_entry_target *)nla_data(tb[TCA_IPT_TARG]); |
7ba699c60 [NET_SCHED]: Conv... |
119 |
if (nla_len(tb[TCA_IPT_TARG]) < td->u.target_size) |
1da177e4c Linux-2.6.12-rc2 |
120 |
return -EINVAL; |
53b2bf3f8 [NET_SCHED]: Use ... |
121 |
if (tb[TCA_IPT_INDEX] != NULL) |
1587bac49 [NET_SCHED]: Use ... |
122 |
index = nla_get_u32(tb[TCA_IPT_INDEX]); |
1da177e4c Linux-2.6.12-rc2 |
123 |
|
e9ce1cd3c [PKT_SCHED]: Kill... |
124 125 126 127 |
pc = tcf_hash_check(index, a, bind, &ipt_hash_info); if (!pc) { pc = tcf_hash_create(index, est, a, sizeof(*ipt), bind, &ipt_idx_gen, &ipt_hash_info); |
0e991ec6a tc: propogate err... |
128 |
if (IS_ERR(pc)) |
cc7ec456f net_sched: cleanups |
129 |
return PTR_ERR(pc); |
1da177e4c Linux-2.6.12-rc2 |
130 131 132 |
ret = ACT_P_CREATED; } else { if (!ovr) { |
e9ce1cd3c [PKT_SCHED]: Kill... |
133 |
tcf_ipt_release(to_ipt(pc), bind); |
1da177e4c Linux-2.6.12-rc2 |
134 135 136 |
return -EEXIST; } } |
e9ce1cd3c [PKT_SCHED]: Kill... |
137 |
ipt = to_ipt(pc); |
1da177e4c Linux-2.6.12-rc2 |
138 |
|
1587bac49 [NET_SCHED]: Use ... |
139 |
hook = nla_get_u32(tb[TCA_IPT_HOOK]); |
1da177e4c Linux-2.6.12-rc2 |
140 141 142 |
err = -ENOMEM; tname = kmalloc(IFNAMSIZ, GFP_KERNEL); |
e9ce1cd3c [PKT_SCHED]: Kill... |
143 |
if (unlikely(!tname)) |
1da177e4c Linux-2.6.12-rc2 |
144 |
goto err1; |
7ba699c60 [NET_SCHED]: Conv... |
145 146 |
if (tb[TCA_IPT_TABLE] == NULL || nla_strlcpy(tname, tb[TCA_IPT_TABLE], IFNAMSIZ) >= IFNAMSIZ) |
1da177e4c Linux-2.6.12-rc2 |
147 |
strcpy(tname, "mangle"); |
c7b1b2497 [SCHED]: Use kmem... |
148 |
t = kmemdup(td, td->u.target_size, GFP_KERNEL); |
e9ce1cd3c [PKT_SCHED]: Kill... |
149 |
if (unlikely(!t)) |
1da177e4c Linux-2.6.12-rc2 |
150 |
goto err2; |
1da177e4c Linux-2.6.12-rc2 |
151 |
|
cc7ec456f net_sched: cleanups |
152 153 |
err = ipt_init_target(t, tname, hook); if (err < 0) |
1da177e4c Linux-2.6.12-rc2 |
154 |
goto err3; |
e9ce1cd3c [PKT_SCHED]: Kill... |
155 |
spin_lock_bh(&ipt->tcf_lock); |
1da177e4c Linux-2.6.12-rc2 |
156 |
if (ret != ACT_P_CREATED) { |
e9ce1cd3c [PKT_SCHED]: Kill... |
157 158 159 |
ipt_destroy_target(ipt->tcfi_t); kfree(ipt->tcfi_tname); kfree(ipt->tcfi_t); |
1da177e4c Linux-2.6.12-rc2 |
160 |
} |
e9ce1cd3c [PKT_SCHED]: Kill... |
161 162 163 164 |
ipt->tcfi_tname = tname; ipt->tcfi_t = t; ipt->tcfi_hook = hook; spin_unlock_bh(&ipt->tcf_lock); |
1da177e4c Linux-2.6.12-rc2 |
165 |
if (ret == ACT_P_CREATED) |
e9ce1cd3c [PKT_SCHED]: Kill... |
166 |
tcf_hash_insert(pc, &ipt_hash_info); |
1da177e4c Linux-2.6.12-rc2 |
167 168 169 170 171 172 173 |
return ret; err3: kfree(t); err2: kfree(tname); err1: |
e9ce1cd3c [PKT_SCHED]: Kill... |
174 |
kfree(pc); |
1da177e4c Linux-2.6.12-rc2 |
175 176 |
return err; } |
e9ce1cd3c [PKT_SCHED]: Kill... |
177 |
static int tcf_ipt_cleanup(struct tc_action *a, int bind) |
1da177e4c Linux-2.6.12-rc2 |
178 |
{ |
e9ce1cd3c [PKT_SCHED]: Kill... |
179 180 |
struct tcf_ipt *ipt = a->priv; return tcf_ipt_release(ipt, bind); |
1da177e4c Linux-2.6.12-rc2 |
181 |
} |
dc7f9f6e8 net: sched: const... |
182 |
static int tcf_ipt(struct sk_buff *skb, const struct tc_action *a, |
e9ce1cd3c [PKT_SCHED]: Kill... |
183 |
struct tcf_result *res) |
1da177e4c Linux-2.6.12-rc2 |
184 185 |
{ int ret = 0, result = 0; |
e9ce1cd3c [PKT_SCHED]: Kill... |
186 |
struct tcf_ipt *ipt = a->priv; |
4b560b447 netfilter: xtable... |
187 |
struct xt_action_param par; |
1da177e4c Linux-2.6.12-rc2 |
188 189 190 191 192 |
if (skb_cloned(skb)) { if (pskb_expand_head(skb, 0, 0, GFP_ATOMIC)) return TC_ACT_UNSPEC; } |
e9ce1cd3c [PKT_SCHED]: Kill... |
193 |
spin_lock(&ipt->tcf_lock); |
1da177e4c Linux-2.6.12-rc2 |
194 |
|
e9ce1cd3c [PKT_SCHED]: Kill... |
195 |
ipt->tcf_tm.lastuse = jiffies; |
bfe0d0298 net_sched: factor... |
196 |
bstats_update(&ipt->tcf_bstats, skb); |
1da177e4c Linux-2.6.12-rc2 |
197 198 |
/* yes, we have to worry about both in and out dev |
cc7ec456f net_sched: cleanups |
199 200 201 |
* worry later - danger - this API seems to have changed * from earlier kernels */ |
7eb355865 netfilter: xtable... |
202 203 204 205 206 207 |
par.in = skb->dev; par.out = NULL; par.hooknum = ipt->tcfi_hook; par.target = ipt->tcfi_t->u.kernel.target; par.targinfo = ipt->tcfi_t->data; ret = par.target->target(skb, &par); |
1da177e4c Linux-2.6.12-rc2 |
208 209 210 211 212 213 |
switch (ret) { case NF_ACCEPT: result = TC_ACT_OK; break; case NF_DROP: result = TC_ACT_SHOT; |
e9ce1cd3c [PKT_SCHED]: Kill... |
214 |
ipt->tcf_qstats.drops++; |
1da177e4c Linux-2.6.12-rc2 |
215 |
break; |
243bf6e29 netfilter: xtable... |
216 |
case XT_CONTINUE: |
1da177e4c Linux-2.6.12-rc2 |
217 218 219 220 |
result = TC_ACT_PIPE; break; default: if (net_ratelimit()) |
6ff9c3644 net sched: printk... |
221 222 223 |
pr_notice("tc filter: Bogus netfilter code" " %d assume ACCEPT ", ret); |
1da177e4c Linux-2.6.12-rc2 |
224 225 226 |
result = TC_POLICE_OK; break; } |
e9ce1cd3c [PKT_SCHED]: Kill... |
227 |
spin_unlock(&ipt->tcf_lock); |
1da177e4c Linux-2.6.12-rc2 |
228 229 230 |
return result; } |
e9ce1cd3c [PKT_SCHED]: Kill... |
231 |
static int tcf_ipt_dump(struct sk_buff *skb, struct tc_action *a, int bind, int ref) |
1da177e4c Linux-2.6.12-rc2 |
232 |
{ |
27a884dc3 [SK_BUFF]: Conver... |
233 |
unsigned char *b = skb_tail_pointer(skb); |
e9ce1cd3c [PKT_SCHED]: Kill... |
234 |
struct tcf_ipt *ipt = a->priv; |
87a2e70db netfilter: xtable... |
235 |
struct xt_entry_target *t; |
1da177e4c Linux-2.6.12-rc2 |
236 237 |
struct tcf_t tm; struct tc_cnt c; |
1da177e4c Linux-2.6.12-rc2 |
238 239 |
/* for simple targets kernel size == user size |
cc7ec456f net_sched: cleanups |
240 241 242 |
* user name = target name * for foolproof you need to not assume this */ |
1da177e4c Linux-2.6.12-rc2 |
243 |
|
c7b1b2497 [SCHED]: Use kmem... |
244 |
t = kmemdup(ipt->tcfi_t, ipt->tcfi_t->u.user.target_size, GFP_ATOMIC); |
e9ce1cd3c [PKT_SCHED]: Kill... |
245 |
if (unlikely(!t)) |
7ba699c60 [NET_SCHED]: Conv... |
246 |
goto nla_put_failure; |
1da177e4c Linux-2.6.12-rc2 |
247 |
|
e9ce1cd3c [PKT_SCHED]: Kill... |
248 249 |
c.bindcnt = ipt->tcf_bindcnt - bind; c.refcnt = ipt->tcf_refcnt - ref; |
e9ce1cd3c [PKT_SCHED]: Kill... |
250 |
strcpy(t->u.user.name, ipt->tcfi_t->u.kernel.target->name); |
7ba699c60 [NET_SCHED]: Conv... |
251 |
NLA_PUT(skb, TCA_IPT_TARG, ipt->tcfi_t->u.user.target_size, t); |
24beeab53 [NET_SCHED]: Use ... |
252 253 |
NLA_PUT_U32(skb, TCA_IPT_INDEX, ipt->tcf_index); NLA_PUT_U32(skb, TCA_IPT_HOOK, ipt->tcfi_hook); |
7ba699c60 [NET_SCHED]: Conv... |
254 |
NLA_PUT(skb, TCA_IPT_CNT, sizeof(struct tc_cnt), &c); |
57e1c487a [NET_SCHED]: Use ... |
255 |
NLA_PUT_STRING(skb, TCA_IPT_TABLE, ipt->tcfi_tname); |
e9ce1cd3c [PKT_SCHED]: Kill... |
256 257 258 |
tm.install = jiffies_to_clock_t(jiffies - ipt->tcf_tm.install); tm.lastuse = jiffies_to_clock_t(jiffies - ipt->tcf_tm.lastuse); tm.expires = jiffies_to_clock_t(ipt->tcf_tm.expires); |
7ba699c60 [NET_SCHED]: Conv... |
259 |
NLA_PUT(skb, TCA_IPT_TM, sizeof (tm), &tm); |
1da177e4c Linux-2.6.12-rc2 |
260 261 |
kfree(t); return skb->len; |
7ba699c60 [NET_SCHED]: Conv... |
262 |
nla_put_failure: |
dc5fc579b [NETLINK]: Use nl... |
263 |
nlmsg_trim(skb, b); |
1da177e4c Linux-2.6.12-rc2 |
264 265 266 267 268 269 |
kfree(t); return -1; } static struct tc_action_ops act_ipt_ops = { .kind = "ipt", |
e9ce1cd3c [PKT_SCHED]: Kill... |
270 |
.hinfo = &ipt_hash_info, |
1da177e4c Linux-2.6.12-rc2 |
271 272 273 274 275 276 277 278 279 280 281 282 283 284 |
.type = TCA_ACT_IPT, .capab = TCA_CAP_NONE, .owner = THIS_MODULE, .act = tcf_ipt, .dump = tcf_ipt_dump, .cleanup = tcf_ipt_cleanup, .lookup = tcf_hash_search, .init = tcf_ipt_init, .walk = tcf_generic_walker }; MODULE_AUTHOR("Jamal Hadi Salim(2002-4)"); MODULE_DESCRIPTION("Iptables target actions"); MODULE_LICENSE("GPL"); |
e9ce1cd3c [PKT_SCHED]: Kill... |
285 |
static int __init ipt_init_module(void) |
1da177e4c Linux-2.6.12-rc2 |
286 287 288 |
{ return tcf_register_action(&act_ipt_ops); } |
e9ce1cd3c [PKT_SCHED]: Kill... |
289 |
static void __exit ipt_cleanup_module(void) |
1da177e4c Linux-2.6.12-rc2 |
290 291 292 293 294 295 |
{ tcf_unregister_action(&act_ipt_ops); } module_init(ipt_init_module); module_exit(ipt_cleanup_module); |