2 * net/sched/cls_fw.c Classifier mapping ipchains' fwmark to traffic class.
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public License
6 * as published by the Free Software Foundation; either version
7 * 2 of the License, or (at your option) any later version.
9 * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
12 * Karlis Peisenieks <karlis@mt.lv> : 990415 : fw_walk off by one
13 * Karlis Peisenieks <karlis@mt.lv> : 990415 : fw_delete killed all the filter (and kernel).
14 * Alex <alex@pilotsoft.com> : 2004xxyy: Added Action extension
16 * JHS: We should remove the CONFIG_NET_CLS_IND from here
17 * eventually when the meta match extension is made available
21 #include <linux/module.h>
22 #include <linux/slab.h>
23 #include <linux/types.h>
24 #include <linux/kernel.h>
25 #include <linux/string.h>
26 #include <linux/errno.h>
27 #include <linux/skbuff.h>
28 #include <net/netlink.h>
29 #include <net/act_api.h>
30 #include <net/pkt_cls.h>
31 #include <net/sch_generic.h>
37 struct fw_filter __rcu *ht[HTSIZE];
42 struct fw_filter __rcu *next;
44 struct tcf_result res;
45 #ifdef CONFIG_NET_CLS_IND
47 #endif /* CONFIG_NET_CLS_IND */
51 struct work_struct work;
56 static u32 fw_hash(u32 handle)
58 handle ^= (handle >> 16);
59 handle ^= (handle >> 8);
60 return handle % HTSIZE;
63 static int fw_classify(struct sk_buff *skb, const struct tcf_proto *tp,
64 struct tcf_result *res)
66 struct fw_head *head = rcu_dereference_bh(tp->root);
74 for (f = rcu_dereference_bh(head->ht[fw_hash(id)]); f;
75 f = rcu_dereference_bh(f->next)) {
78 #ifdef CONFIG_NET_CLS_IND
79 if (!tcf_match_indev(skb, f->ifindex))
81 #endif /* CONFIG_NET_CLS_IND */
82 r = tcf_exts_exec(skb, &f->exts, res);
90 struct Qdisc *q = tcf_block_q(tp->chain->block);
92 /* Old method: classify the packet using its skb mark. */
93 if (id && (TC_H_MAJ(id) == 0 ||
94 !(TC_H_MAJ(id ^ q->handle)))) {
104 static void *fw_get(struct tcf_proto *tp, u32 handle)
106 struct fw_head *head = rtnl_dereference(tp->root);
112 f = rtnl_dereference(head->ht[fw_hash(handle)]);
113 for (; f; f = rtnl_dereference(f->next)) {
120 static int fw_init(struct tcf_proto *tp)
122 /* We don't allocate fw_head here, because in the old method
123 * we don't need it at all.
128 static void __fw_delete_filter(struct fw_filter *f)
130 tcf_exts_destroy(&f->exts);
131 tcf_exts_put_net(&f->exts);
135 static void fw_delete_filter_work(struct work_struct *work)
137 struct fw_filter *f = container_of(work, struct fw_filter, work);
140 __fw_delete_filter(f);
144 static void fw_delete_filter(struct rcu_head *head)
146 struct fw_filter *f = container_of(head, struct fw_filter, rcu);
148 INIT_WORK(&f->work, fw_delete_filter_work);
149 tcf_queue_work(&f->work);
152 static void fw_destroy(struct tcf_proto *tp)
154 struct fw_head *head = rtnl_dereference(tp->root);
161 for (h = 0; h < HTSIZE; h++) {
162 while ((f = rtnl_dereference(head->ht[h])) != NULL) {
163 RCU_INIT_POINTER(head->ht[h],
164 rtnl_dereference(f->next));
165 tcf_unbind_filter(tp, &f->res);
166 if (tcf_exts_get_net(&f->exts))
167 call_rcu(&f->rcu, fw_delete_filter);
169 __fw_delete_filter(f);
172 kfree_rcu(head, rcu);
175 static int fw_delete(struct tcf_proto *tp, void *arg, bool *last)
177 struct fw_head *head = rtnl_dereference(tp->root);
178 struct fw_filter *f = arg;
179 struct fw_filter __rcu **fp;
180 struct fw_filter *pfp;
184 if (head == NULL || f == NULL)
187 fp = &head->ht[fw_hash(f->id)];
189 for (pfp = rtnl_dereference(*fp); pfp;
190 fp = &pfp->next, pfp = rtnl_dereference(*fp)) {
192 RCU_INIT_POINTER(*fp, rtnl_dereference(f->next));
193 tcf_unbind_filter(tp, &f->res);
194 tcf_exts_get_net(&f->exts);
195 call_rcu(&f->rcu, fw_delete_filter);
202 for (h = 0; h < HTSIZE; h++) {
203 if (rcu_access_pointer(head->ht[h])) {
213 static const struct nla_policy fw_policy[TCA_FW_MAX + 1] = {
214 [TCA_FW_CLASSID] = { .type = NLA_U32 },
215 [TCA_FW_INDEV] = { .type = NLA_STRING, .len = IFNAMSIZ },
216 [TCA_FW_MASK] = { .type = NLA_U32 },
219 static int fw_set_parms(struct net *net, struct tcf_proto *tp,
220 struct fw_filter *f, struct nlattr **tb,
221 struct nlattr **tca, unsigned long base, bool ovr)
223 struct fw_head *head = rtnl_dereference(tp->root);
227 err = tcf_exts_validate(net, tp, tb, tca[TCA_RATE], &f->exts, ovr);
231 if (tb[TCA_FW_CLASSID]) {
232 f->res.classid = nla_get_u32(tb[TCA_FW_CLASSID]);
233 tcf_bind_filter(tp, &f->res, base);
236 #ifdef CONFIG_NET_CLS_IND
237 if (tb[TCA_FW_INDEV]) {
239 ret = tcf_change_indev(net, tb[TCA_FW_INDEV]);
244 #endif /* CONFIG_NET_CLS_IND */
247 if (tb[TCA_FW_MASK]) {
248 mask = nla_get_u32(tb[TCA_FW_MASK]);
249 if (mask != head->mask)
251 } else if (head->mask != 0xFFFFFFFF)
257 static int fw_change(struct net *net, struct sk_buff *in_skb,
258 struct tcf_proto *tp, unsigned long base,
259 u32 handle, struct nlattr **tca, void **arg,
262 struct fw_head *head = rtnl_dereference(tp->root);
263 struct fw_filter *f = *arg;
264 struct nlattr *opt = tca[TCA_OPTIONS];
265 struct nlattr *tb[TCA_FW_MAX + 1];
269 return handle ? -EINVAL : 0; /* Succeed if it is old method. */
271 err = nla_parse_nested(tb, TCA_FW_MAX, opt, fw_policy, NULL);
276 struct fw_filter *pfp, *fnew;
277 struct fw_filter __rcu **fp;
279 if (f->id != handle && handle)
282 fnew = kzalloc(sizeof(struct fw_filter), GFP_KERNEL);
288 #ifdef CONFIG_NET_CLS_IND
289 fnew->ifindex = f->ifindex;
290 #endif /* CONFIG_NET_CLS_IND */
293 err = tcf_exts_init(&fnew->exts, TCA_FW_ACT, TCA_FW_POLICE);
299 err = fw_set_parms(net, tp, fnew, tb, tca, base, ovr);
301 tcf_exts_destroy(&fnew->exts);
306 fp = &head->ht[fw_hash(fnew->id)];
307 for (pfp = rtnl_dereference(*fp); pfp;
308 fp = &pfp->next, pfp = rtnl_dereference(*fp))
312 RCU_INIT_POINTER(fnew->next, rtnl_dereference(pfp->next));
313 rcu_assign_pointer(*fp, fnew);
314 tcf_unbind_filter(tp, &f->res);
315 tcf_exts_get_net(&f->exts);
316 call_rcu(&f->rcu, fw_delete_filter);
326 u32 mask = 0xFFFFFFFF;
328 mask = nla_get_u32(tb[TCA_FW_MASK]);
330 head = kzalloc(sizeof(*head), GFP_KERNEL);
335 rcu_assign_pointer(tp->root, head);
338 f = kzalloc(sizeof(struct fw_filter), GFP_KERNEL);
342 err = tcf_exts_init(&f->exts, TCA_FW_ACT, TCA_FW_POLICE);
348 err = fw_set_parms(net, tp, f, tb, tca, base, ovr);
352 RCU_INIT_POINTER(f->next, head->ht[fw_hash(handle)]);
353 rcu_assign_pointer(head->ht[fw_hash(handle)], f);
359 tcf_exts_destroy(&f->exts);
364 static void fw_walk(struct tcf_proto *tp, struct tcf_walker *arg)
366 struct fw_head *head = rtnl_dereference(tp->root);
375 for (h = 0; h < HTSIZE; h++) {
378 for (f = rtnl_dereference(head->ht[h]); f;
379 f = rtnl_dereference(f->next)) {
380 if (arg->count < arg->skip) {
384 if (arg->fn(tp, f, arg) < 0) {
393 static int fw_dump(struct net *net, struct tcf_proto *tp, void *fh,
394 struct sk_buff *skb, struct tcmsg *t)
396 struct fw_head *head = rtnl_dereference(tp->root);
397 struct fw_filter *f = fh;
403 t->tcm_handle = f->id;
405 if (!f->res.classid && !tcf_exts_has_actions(&f->exts))
408 nest = nla_nest_start(skb, TCA_OPTIONS);
410 goto nla_put_failure;
412 if (f->res.classid &&
413 nla_put_u32(skb, TCA_FW_CLASSID, f->res.classid))
414 goto nla_put_failure;
415 #ifdef CONFIG_NET_CLS_IND
417 struct net_device *dev;
418 dev = __dev_get_by_index(net, f->ifindex);
419 if (dev && nla_put_string(skb, TCA_FW_INDEV, dev->name))
420 goto nla_put_failure;
422 #endif /* CONFIG_NET_CLS_IND */
423 if (head->mask != 0xFFFFFFFF &&
424 nla_put_u32(skb, TCA_FW_MASK, head->mask))
425 goto nla_put_failure;
427 if (tcf_exts_dump(skb, &f->exts) < 0)
428 goto nla_put_failure;
430 nla_nest_end(skb, nest);
432 if (tcf_exts_dump_stats(skb, &f->exts) < 0)
433 goto nla_put_failure;
438 nla_nest_cancel(skb, nest);
442 static void fw_bind_class(void *fh, u32 classid, unsigned long cl)
444 struct fw_filter *f = fh;
446 if (f && f->res.classid == classid)
450 static struct tcf_proto_ops cls_fw_ops __read_mostly = {
452 .classify = fw_classify,
454 .destroy = fw_destroy,
460 .bind_class = fw_bind_class,
461 .owner = THIS_MODULE,
464 static int __init init_fw(void)
466 return register_tcf_proto_ops(&cls_fw_ops);
469 static void __exit exit_fw(void)
471 unregister_tcf_proto_ops(&cls_fw_ops);
476 MODULE_LICENSE("GPL");