blob: 0b8eb235bc136fff1eeb349f72c549a5ce4df3df [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
2 * net/sched/act_api.c Packet action API.
3 *
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.
8 *
9 * Author: Jamal Hadi Salim
10 *
11 *
12 */
13
Linus Torvalds1da177e2005-04-16 15:20:36 -070014#include <linux/types.h>
15#include <linux/kernel.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070016#include <linux/string.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070017#include <linux/errno.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070018#include <linux/skbuff.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070019#include <linux/init.h>
20#include <linux/kmod.h>
Patrick McHardyab27cfb2008-01-23 20:33:13 -080021#include <linux/err.h>
Denis V. Lunevb8542722007-12-01 00:21:31 +110022#include <net/net_namespace.h>
23#include <net/sock.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070024#include <net/sch_generic.h>
25#include <net/act_api.h>
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -070026#include <net/netlink.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070027
David S. Millere9ce1cd2006-08-21 23:54:55 -070028void tcf_hash_destroy(struct tcf_common *p, struct tcf_hashinfo *hinfo)
29{
30 unsigned int h = tcf_hash(p->tcfc_index, hinfo->hmask);
31 struct tcf_common **p1p;
32
33 for (p1p = &hinfo->htab[h]; *p1p; p1p = &(*p1p)->tcfc_next) {
34 if (*p1p == p) {
35 write_lock_bh(hinfo->lock);
36 *p1p = p->tcfc_next;
37 write_unlock_bh(hinfo->lock);
David S. Millere9ce1cd2006-08-21 23:54:55 -070038 gen_kill_estimator(&p->tcfc_bstats,
39 &p->tcfc_rate_est);
David S. Millere9ce1cd2006-08-21 23:54:55 -070040 kfree(p);
41 return;
42 }
43 }
44 BUG_TRAP(0);
45}
46EXPORT_SYMBOL(tcf_hash_destroy);
47
48int tcf_hash_release(struct tcf_common *p, int bind,
49 struct tcf_hashinfo *hinfo)
50{
51 int ret = 0;
52
53 if (p) {
54 if (bind)
55 p->tcfc_bindcnt--;
56
57 p->tcfc_refcnt--;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +090058 if (p->tcfc_bindcnt <= 0 && p->tcfc_refcnt <= 0) {
David S. Millere9ce1cd2006-08-21 23:54:55 -070059 tcf_hash_destroy(p, hinfo);
60 ret = 1;
61 }
62 }
63 return ret;
64}
65EXPORT_SYMBOL(tcf_hash_release);
66
67static int tcf_dump_walker(struct sk_buff *skb, struct netlink_callback *cb,
68 struct tc_action *a, struct tcf_hashinfo *hinfo)
69{
70 struct tcf_common *p;
71 int err = 0, index = -1,i = 0, s_i = 0, n_i = 0;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -080072 struct nlattr *nest;
David S. Millere9ce1cd2006-08-21 23:54:55 -070073
Jamal Hadi Salime1e992e2007-09-12 16:32:59 +020074 read_lock_bh(hinfo->lock);
David S. Millere9ce1cd2006-08-21 23:54:55 -070075
76 s_i = cb->args[0];
77
78 for (i = 0; i < (hinfo->hmask + 1); i++) {
79 p = hinfo->htab[tcf_hash(i, hinfo->hmask)];
80
81 for (; p; p = p->tcfc_next) {
82 index++;
83 if (index < s_i)
84 continue;
85 a->priv = p;
86 a->order = n_i;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -080087
88 nest = nla_nest_start(skb, a->order);
89 if (nest == NULL)
90 goto nla_put_failure;
David S. Millere9ce1cd2006-08-21 23:54:55 -070091 err = tcf_action_dump_1(skb, a, 0, 0);
92 if (err < 0) {
93 index--;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -080094 nlmsg_trim(skb, nest);
David S. Millere9ce1cd2006-08-21 23:54:55 -070095 goto done;
96 }
Patrick McHardy4b3550ef2008-01-23 20:34:11 -080097 nla_nest_end(skb, nest);
David S. Millere9ce1cd2006-08-21 23:54:55 -070098 n_i++;
99 if (n_i >= TCA_ACT_MAX_PRIO)
100 goto done;
101 }
102 }
103done:
Jamal Hadi Salime1e992e2007-09-12 16:32:59 +0200104 read_unlock_bh(hinfo->lock);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700105 if (n_i)
106 cb->args[0] += n_i;
107 return n_i;
108
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800109nla_put_failure:
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800110 nla_nest_cancel(skb, nest);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700111 goto done;
112}
113
114static int tcf_del_walker(struct sk_buff *skb, struct tc_action *a,
115 struct tcf_hashinfo *hinfo)
116{
117 struct tcf_common *p, *s_p;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800118 struct nlattr *nest;
David S. Millere9ce1cd2006-08-21 23:54:55 -0700119 int i= 0, n_i = 0;
120
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800121 nest = nla_nest_start(skb, a->order);
122 if (nest == NULL)
123 goto nla_put_failure;
Patrick McHardy57e1c482008-01-23 20:34:28 -0800124 NLA_PUT_STRING(skb, TCA_KIND, a->ops->kind);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700125 for (i = 0; i < (hinfo->hmask + 1); i++) {
126 p = hinfo->htab[tcf_hash(i, hinfo->hmask)];
127
128 while (p != NULL) {
129 s_p = p->tcfc_next;
130 if (ACT_P_DELETED == tcf_hash_release(p, 0, hinfo))
131 module_put(a->ops->owner);
132 n_i++;
133 p = s_p;
134 }
135 }
Patrick McHardy24beeab2008-01-23 20:34:48 -0800136 NLA_PUT_U32(skb, TCA_FCNT, n_i);
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800137 nla_nest_end(skb, nest);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700138
139 return n_i;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800140nla_put_failure:
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800141 nla_nest_cancel(skb, nest);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700142 return -EINVAL;
143}
144
145int tcf_generic_walker(struct sk_buff *skb, struct netlink_callback *cb,
146 int type, struct tc_action *a)
147{
148 struct tcf_hashinfo *hinfo = a->ops->hinfo;
149
150 if (type == RTM_DELACTION) {
151 return tcf_del_walker(skb, a, hinfo);
152 } else if (type == RTM_GETACTION) {
153 return tcf_dump_walker(skb, cb, a, hinfo);
154 } else {
155 printk("tcf_generic_walker: unknown action %d\n", type);
156 return -EINVAL;
157 }
158}
159EXPORT_SYMBOL(tcf_generic_walker);
160
161struct tcf_common *tcf_hash_lookup(u32 index, struct tcf_hashinfo *hinfo)
162{
163 struct tcf_common *p;
164
Jamal Hadi Salime1e992e2007-09-12 16:32:59 +0200165 read_lock_bh(hinfo->lock);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700166 for (p = hinfo->htab[tcf_hash(index, hinfo->hmask)]; p;
167 p = p->tcfc_next) {
168 if (p->tcfc_index == index)
169 break;
170 }
Jamal Hadi Salime1e992e2007-09-12 16:32:59 +0200171 read_unlock_bh(hinfo->lock);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700172
173 return p;
174}
175EXPORT_SYMBOL(tcf_hash_lookup);
176
177u32 tcf_hash_new_index(u32 *idx_gen, struct tcf_hashinfo *hinfo)
178{
179 u32 val = *idx_gen;
180
181 do {
182 if (++val == 0)
183 val = 1;
184 } while (tcf_hash_lookup(val, hinfo));
185
186 return (*idx_gen = val);
187}
188EXPORT_SYMBOL(tcf_hash_new_index);
189
190int tcf_hash_search(struct tc_action *a, u32 index)
191{
192 struct tcf_hashinfo *hinfo = a->ops->hinfo;
193 struct tcf_common *p = tcf_hash_lookup(index, hinfo);
194
195 if (p) {
196 a->priv = p;
197 return 1;
198 }
199 return 0;
200}
201EXPORT_SYMBOL(tcf_hash_search);
202
203struct tcf_common *tcf_hash_check(u32 index, struct tc_action *a, int bind,
204 struct tcf_hashinfo *hinfo)
205{
206 struct tcf_common *p = NULL;
207 if (index && (p = tcf_hash_lookup(index, hinfo)) != NULL) {
208 if (bind) {
209 p->tcfc_bindcnt++;
210 p->tcfc_refcnt++;
211 }
212 a->priv = p;
213 }
214 return p;
215}
216EXPORT_SYMBOL(tcf_hash_check);
217
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800218struct tcf_common *tcf_hash_create(u32 index, struct nlattr *est, struct tc_action *a, int size, int bind, u32 *idx_gen, struct tcf_hashinfo *hinfo)
David S. Millere9ce1cd2006-08-21 23:54:55 -0700219{
220 struct tcf_common *p = kzalloc(size, GFP_KERNEL);
221
222 if (unlikely(!p))
223 return p;
224 p->tcfc_refcnt = 1;
225 if (bind)
226 p->tcfc_bindcnt = 1;
227
228 spin_lock_init(&p->tcfc_lock);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700229 p->tcfc_index = index ? index : tcf_hash_new_index(idx_gen, hinfo);
230 p->tcfc_tm.install = jiffies;
231 p->tcfc_tm.lastuse = jiffies;
David S. Millere9ce1cd2006-08-21 23:54:55 -0700232 if (est)
233 gen_new_estimator(&p->tcfc_bstats, &p->tcfc_rate_est,
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800234 &p->tcfc_lock, est);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700235 a->priv = (void *) p;
236 return p;
237}
238EXPORT_SYMBOL(tcf_hash_create);
239
240void tcf_hash_insert(struct tcf_common *p, struct tcf_hashinfo *hinfo)
241{
242 unsigned int h = tcf_hash(p->tcfc_index, hinfo->hmask);
243
244 write_lock_bh(hinfo->lock);
245 p->tcfc_next = hinfo->htab[h];
246 hinfo->htab[h] = p;
247 write_unlock_bh(hinfo->lock);
248}
249EXPORT_SYMBOL(tcf_hash_insert);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700250
251static struct tc_action_ops *act_base = NULL;
252static DEFINE_RWLOCK(act_mod_lock);
253
254int tcf_register_action(struct tc_action_ops *act)
255{
256 struct tc_action_ops *a, **ap;
257
258 write_lock(&act_mod_lock);
259 for (ap = &act_base; (a = *ap) != NULL; ap = &a->next) {
260 if (act->type == a->type || (strcmp(act->kind, a->kind) == 0)) {
261 write_unlock(&act_mod_lock);
262 return -EEXIST;
263 }
264 }
265 act->next = NULL;
266 *ap = act;
267 write_unlock(&act_mod_lock);
268 return 0;
269}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800270EXPORT_SYMBOL(tcf_register_action);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700271
272int tcf_unregister_action(struct tc_action_ops *act)
273{
274 struct tc_action_ops *a, **ap;
275 int err = -ENOENT;
276
277 write_lock(&act_mod_lock);
278 for (ap = &act_base; (a = *ap) != NULL; ap = &a->next)
279 if (a == act)
280 break;
281 if (a) {
282 *ap = a->next;
283 a->next = NULL;
284 err = 0;
285 }
286 write_unlock(&act_mod_lock);
287 return err;
288}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800289EXPORT_SYMBOL(tcf_unregister_action);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700290
291/* lookup by name */
292static struct tc_action_ops *tc_lookup_action_n(char *kind)
293{
294 struct tc_action_ops *a = NULL;
295
296 if (kind) {
297 read_lock(&act_mod_lock);
298 for (a = act_base; a; a = a->next) {
299 if (strcmp(kind, a->kind) == 0) {
300 if (!try_module_get(a->owner)) {
301 read_unlock(&act_mod_lock);
302 return NULL;
303 }
304 break;
305 }
306 }
307 read_unlock(&act_mod_lock);
308 }
309 return a;
310}
311
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800312/* lookup by nlattr */
313static struct tc_action_ops *tc_lookup_action(struct nlattr *kind)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700314{
315 struct tc_action_ops *a = NULL;
316
317 if (kind) {
318 read_lock(&act_mod_lock);
319 for (a = act_base; a; a = a->next) {
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800320 if (nla_strcmp(kind, a->kind) == 0) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700321 if (!try_module_get(a->owner)) {
322 read_unlock(&act_mod_lock);
323 return NULL;
324 }
325 break;
326 }
327 }
328 read_unlock(&act_mod_lock);
329 }
330 return a;
331}
332
333#if 0
334/* lookup by id */
335static struct tc_action_ops *tc_lookup_action_id(u32 type)
336{
337 struct tc_action_ops *a = NULL;
338
339 if (type) {
340 read_lock(&act_mod_lock);
341 for (a = act_base; a; a = a->next) {
342 if (a->type == type) {
343 if (!try_module_get(a->owner)) {
344 read_unlock(&act_mod_lock);
345 return NULL;
346 }
347 break;
348 }
349 }
350 read_unlock(&act_mod_lock);
351 }
352 return a;
353}
354#endif
355
356int tcf_action_exec(struct sk_buff *skb, struct tc_action *act,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900357 struct tcf_result *res)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700358{
359 struct tc_action *a;
360 int ret = -1;
361
362 if (skb->tc_verd & TC_NCLS) {
363 skb->tc_verd = CLR_TC_NCLS(skb->tc_verd);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700364 ret = TC_ACT_OK;
365 goto exec_done;
366 }
367 while ((a = act) != NULL) {
368repeat:
369 if (a->ops && a->ops->act) {
Patrick McHardyf43c5a02006-01-08 22:15:34 -0800370 ret = a->ops->act(skb, a, res);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700371 if (TC_MUNGED & skb->tc_verd) {
372 /* copied already, allow trampling */
373 skb->tc_verd = SET_TC_OK2MUNGE(skb->tc_verd);
374 skb->tc_verd = CLR_TC_MUNGED(skb->tc_verd);
375 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700376 if (ret == TC_ACT_REPEAT)
377 goto repeat; /* we need a ttl - JHS */
J Hadi Salim14d50e72005-05-03 16:29:13 -0700378 if (ret != TC_ACT_PIPE)
379 goto exec_done;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700380 }
381 act = a->next;
382 }
383exec_done:
Linus Torvalds1da177e2005-04-16 15:20:36 -0700384 return ret;
385}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800386EXPORT_SYMBOL(tcf_action_exec);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700387
388void tcf_action_destroy(struct tc_action *act, int bind)
389{
390 struct tc_action *a;
391
392 for (a = act; a; a = act) {
393 if (a->ops && a->ops->cleanup) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700394 if (a->ops->cleanup(a, bind) == ACT_P_DELETED)
395 module_put(a->ops->owner);
396 act = act->next;
397 kfree(a);
398 } else { /*FIXME: Remove later - catch insertion bugs*/
399 printk("tcf_action_destroy: BUG? destroying NULL ops\n");
400 act = act->next;
401 kfree(a);
402 }
403 }
404}
405
406int
407tcf_action_dump_old(struct sk_buff *skb, struct tc_action *a, int bind, int ref)
408{
409 int err = -EINVAL;
410
411 if (a->ops == NULL || a->ops->dump == NULL)
412 return err;
413 return a->ops->dump(skb, a, bind, ref);
414}
415
416int
417tcf_action_dump_1(struct sk_buff *skb, struct tc_action *a, int bind, int ref)
418{
419 int err = -EINVAL;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -0700420 unsigned char *b = skb_tail_pointer(skb);
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800421 struct nlattr *nest;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700422
423 if (a->ops == NULL || a->ops->dump == NULL)
424 return err;
425
Patrick McHardy57e1c482008-01-23 20:34:28 -0800426 NLA_PUT_STRING(skb, TCA_KIND, a->ops->kind);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700427 if (tcf_action_copy_stats(skb, a, 0))
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800428 goto nla_put_failure;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800429 nest = nla_nest_start(skb, TCA_OPTIONS);
430 if (nest == NULL)
431 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700432 if ((err = tcf_action_dump_old(skb, a, bind, ref)) > 0) {
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800433 nla_nest_end(skb, nest);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700434 return err;
435 }
436
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800437nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -0700438 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700439 return -1;
440}
Patrick McHardy62e3ba12008-01-22 22:10:23 -0800441EXPORT_SYMBOL(tcf_action_dump_1);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700442
443int
444tcf_action_dump(struct sk_buff *skb, struct tc_action *act, int bind, int ref)
445{
446 struct tc_action *a;
447 int err = -EINVAL;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800448 struct nlattr *nest;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700449
450 while ((a = act) != NULL) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700451 act = a->next;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800452 nest = nla_nest_start(skb, a->order);
453 if (nest == NULL)
454 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700455 err = tcf_action_dump_1(skb, a, bind, ref);
456 if (err < 0)
Thomas Graf4fe683f2006-07-05 20:47:28 -0700457 goto errout;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800458 nla_nest_end(skb, nest);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700459 }
460
461 return 0;
462
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800463nla_put_failure:
Thomas Graf4fe683f2006-07-05 20:47:28 -0700464 err = -EINVAL;
465errout:
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800466 nla_nest_cancel(skb, nest);
Thomas Graf4fe683f2006-07-05 20:47:28 -0700467 return err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700468}
469
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800470struct tc_action *tcf_action_init_1(struct nlattr *nla, struct nlattr *est,
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800471 char *name, int ovr, int bind)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700472{
473 struct tc_action *a;
474 struct tc_action_ops *a_o;
475 char act_name[IFNAMSIZ];
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800476 struct nlattr *tb[TCA_ACT_MAX+1];
477 struct nlattr *kind;
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800478 int err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700479
Linus Torvalds1da177e2005-04-16 15:20:36 -0700480 if (name == NULL) {
Patrick McHardycee63722008-01-23 20:33:32 -0800481 err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL);
482 if (err < 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700483 goto err_out;
Patrick McHardycee63722008-01-23 20:33:32 -0800484 err = -EINVAL;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800485 kind = tb[TCA_ACT_KIND];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700486 if (kind == NULL)
487 goto err_out;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800488 if (nla_strlcpy(act_name, kind, IFNAMSIZ) >= IFNAMSIZ)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700489 goto err_out;
490 } else {
Patrick McHardycee63722008-01-23 20:33:32 -0800491 err = -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700492 if (strlcpy(act_name, name, IFNAMSIZ) >= IFNAMSIZ)
493 goto err_out;
494 }
495
496 a_o = tc_lookup_action_n(act_name);
497 if (a_o == NULL) {
498#ifdef CONFIG_KMOD
499 rtnl_unlock();
Patrick McHardy4bba3922006-01-08 22:22:14 -0800500 request_module("act_%s", act_name);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700501 rtnl_lock();
502
503 a_o = tc_lookup_action_n(act_name);
504
505 /* We dropped the RTNL semaphore in order to
506 * perform the module load. So, even if we
507 * succeeded in loading the module we have to
508 * tell the caller to replay the request. We
509 * indicate this using -EAGAIN.
510 */
511 if (a_o != NULL) {
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800512 err = -EAGAIN;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700513 goto err_mod;
514 }
515#endif
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800516 err = -ENOENT;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700517 goto err_out;
518 }
519
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800520 err = -ENOMEM;
Panagiotis Issaris0da974f2006-07-21 14:51:30 -0700521 a = kzalloc(sizeof(*a), GFP_KERNEL);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700522 if (a == NULL)
523 goto err_mod;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700524
525 /* backward compatibility for policer */
526 if (name == NULL)
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800527 err = a_o->init(tb[TCA_ACT_OPTIONS], est, a, ovr, bind);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700528 else
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800529 err = a_o->init(nla, est, a, ovr, bind);
530 if (err < 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700531 goto err_free;
532
533 /* module count goes up only when brand new policy is created
534 if it exists and is only bound to in a_o->init() then
535 ACT_P_CREATED is not returned (a zero is).
536 */
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800537 if (err != ACT_P_CREATED)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700538 module_put(a_o->owner);
539 a->ops = a_o;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700540
Linus Torvalds1da177e2005-04-16 15:20:36 -0700541 return a;
542
543err_free:
544 kfree(a);
545err_mod:
546 module_put(a_o->owner);
547err_out:
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800548 return ERR_PTR(err);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700549}
550
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800551struct tc_action *tcf_action_init(struct nlattr *nla, struct nlattr *est,
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800552 char *name, int ovr, int bind)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700553{
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800554 struct nlattr *tb[TCA_ACT_MAX_PRIO+1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700555 struct tc_action *head = NULL, *act, *act_prev = NULL;
Patrick McHardycee63722008-01-23 20:33:32 -0800556 int err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700557 int i;
558
Patrick McHardycee63722008-01-23 20:33:32 -0800559 err = nla_parse_nested(tb, TCA_ACT_MAX_PRIO, nla, NULL);
560 if (err < 0)
561 return ERR_PTR(err);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700562
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800563 for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) {
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800564 act = tcf_action_init_1(tb[i], est, name, ovr, bind);
565 if (IS_ERR(act))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700566 goto err;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800567 act->order = i;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700568
569 if (head == NULL)
570 head = act;
571 else
572 act_prev->next = act;
573 act_prev = act;
574 }
575 return head;
576
577err:
578 if (head != NULL)
579 tcf_action_destroy(head, bind);
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800580 return act;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700581}
582
583int tcf_action_copy_stats(struct sk_buff *skb, struct tc_action *a,
584 int compat_mode)
585{
586 int err = 0;
587 struct gnet_dump d;
588 struct tcf_act_hdr *h = a->priv;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900589
Linus Torvalds1da177e2005-04-16 15:20:36 -0700590 if (h == NULL)
591 goto errout;
592
593 /* compat_mode being true specifies a call that is supposed
594 * to add additional backward compatiblity statistic TLVs.
595 */
596 if (compat_mode) {
597 if (a->type == TCA_OLD_COMPAT)
598 err = gnet_stats_start_copy_compat(skb, 0,
Patrick McHardy4bdf3992007-07-02 22:47:37 -0700599 TCA_STATS, TCA_XSTATS, &h->tcf_lock, &d);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700600 else
601 return 0;
602 } else
603 err = gnet_stats_start_copy(skb, TCA_ACT_STATS,
Patrick McHardy4bdf3992007-07-02 22:47:37 -0700604 &h->tcf_lock, &d);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700605
606 if (err < 0)
607 goto errout;
608
609 if (a->ops != NULL && a->ops->get_stats != NULL)
610 if (a->ops->get_stats(skb, a) < 0)
611 goto errout;
612
David S. Millere9ce1cd2006-08-21 23:54:55 -0700613 if (gnet_stats_copy_basic(&d, &h->tcf_bstats) < 0 ||
David S. Millere9ce1cd2006-08-21 23:54:55 -0700614 gnet_stats_copy_rate_est(&d, &h->tcf_rate_est) < 0 ||
David S. Millere9ce1cd2006-08-21 23:54:55 -0700615 gnet_stats_copy_queue(&d, &h->tcf_qstats) < 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700616 goto errout;
617
618 if (gnet_stats_finish_copy(&d) < 0)
619 goto errout;
620
621 return 0;
622
623errout:
624 return -1;
625}
626
627static int
628tca_get_fill(struct sk_buff *skb, struct tc_action *a, u32 pid, u32 seq,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900629 u16 flags, int event, int bind, int ref)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700630{
631 struct tcamsg *t;
632 struct nlmsghdr *nlh;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -0700633 unsigned char *b = skb_tail_pointer(skb);
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800634 struct nlattr *nest;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700635
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -0700636 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*t), flags);
637
Linus Torvalds1da177e2005-04-16 15:20:36 -0700638 t = NLMSG_DATA(nlh);
639 t->tca_family = AF_UNSPEC;
Patrick McHardy9ef1d4c2005-06-28 12:55:30 -0700640 t->tca__pad1 = 0;
641 t->tca__pad2 = 0;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900642
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800643 nest = nla_nest_start(skb, TCA_ACT_TAB);
644 if (nest == NULL)
645 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700646
647 if (tcf_action_dump(skb, a, bind, ref) < 0)
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800648 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700649
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800650 nla_nest_end(skb, nest);
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900651
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -0700652 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700653 return skb->len;
654
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800655nla_put_failure:
Linus Torvalds1da177e2005-04-16 15:20:36 -0700656nlmsg_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -0700657 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700658 return -1;
659}
660
661static int
662act_get_notify(u32 pid, struct nlmsghdr *n, struct tc_action *a, int event)
663{
664 struct sk_buff *skb;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700665
666 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
667 if (!skb)
668 return -ENOBUFS;
669 if (tca_get_fill(skb, a, pid, n->nlmsg_seq, 0, event, 0, 0) <= 0) {
670 kfree_skb(skb);
671 return -EINVAL;
672 }
Thomas Graf2942e902006-08-15 00:30:25 -0700673
Denis V. Lunev97c53ca2007-11-19 22:26:51 -0800674 return rtnl_unicast(skb, &init_net, pid);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700675}
676
677static struct tc_action *
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800678tcf_action_get_1(struct nlattr *nla, struct nlmsghdr *n, u32 pid)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700679{
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800680 struct nlattr *tb[TCA_ACT_MAX+1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700681 struct tc_action *a;
682 int index;
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800683 int err;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700684
Patrick McHardycee63722008-01-23 20:33:32 -0800685 err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL);
686 if (err < 0)
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800687 goto err_out;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700688
Patrick McHardycee63722008-01-23 20:33:32 -0800689 err = -EINVAL;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800690 if (tb[TCA_ACT_INDEX] == NULL ||
691 nla_len(tb[TCA_ACT_INDEX]) < sizeof(index))
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800692 goto err_out;
Patrick McHardy1587bac2008-01-23 20:35:03 -0800693 index = nla_get_u32(tb[TCA_ACT_INDEX]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700694
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800695 err = -ENOMEM;
Panagiotis Issaris0da974f2006-07-21 14:51:30 -0700696 a = kzalloc(sizeof(struct tc_action), GFP_KERNEL);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700697 if (a == NULL)
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800698 goto err_out;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700699
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800700 err = -EINVAL;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800701 a->ops = tc_lookup_action(tb[TCA_ACT_KIND]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700702 if (a->ops == NULL)
703 goto err_free;
704 if (a->ops->lookup == NULL)
705 goto err_mod;
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800706 err = -ENOENT;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700707 if (a->ops->lookup(a, index) == 0)
708 goto err_mod;
709
710 module_put(a->ops->owner);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700711 return a;
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800712
Linus Torvalds1da177e2005-04-16 15:20:36 -0700713err_mod:
714 module_put(a->ops->owner);
715err_free:
716 kfree(a);
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800717err_out:
718 return ERR_PTR(err);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700719}
720
721static void cleanup_a(struct tc_action *act)
722{
723 struct tc_action *a;
724
725 for (a = act; a; a = act) {
726 act = a->next;
727 kfree(a);
728 }
729}
730
731static struct tc_action *create_a(int i)
732{
733 struct tc_action *act;
734
Panagiotis Issaris0da974f2006-07-21 14:51:30 -0700735 act = kzalloc(sizeof(*act), GFP_KERNEL);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700736 if (act == NULL) {
737 printk("create_a: failed to alloc!\n");
738 return NULL;
739 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700740 act->order = i;
741 return act;
742}
743
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800744static int tca_action_flush(struct nlattr *nla, struct nlmsghdr *n, u32 pid)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700745{
746 struct sk_buff *skb;
747 unsigned char *b;
748 struct nlmsghdr *nlh;
749 struct tcamsg *t;
750 struct netlink_callback dcb;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800751 struct nlattr *nest;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800752 struct nlattr *tb[TCA_ACT_MAX+1];
753 struct nlattr *kind;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700754 struct tc_action *a = create_a(0);
755 int err = -EINVAL;
756
757 if (a == NULL) {
758 printk("tca_action_flush: couldnt create tc_action\n");
759 return err;
760 }
761
762 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
763 if (!skb) {
764 printk("tca_action_flush: failed skb alloc\n");
765 kfree(a);
766 return -ENOBUFS;
767 }
768
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -0700769 b = skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700770
Patrick McHardycee63722008-01-23 20:33:32 -0800771 err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL);
772 if (err < 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700773 goto err_out;
774
Patrick McHardycee63722008-01-23 20:33:32 -0800775 err = -EINVAL;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800776 kind = tb[TCA_ACT_KIND];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700777 a->ops = tc_lookup_action(kind);
778 if (a->ops == NULL)
779 goto err_out;
780
781 nlh = NLMSG_PUT(skb, pid, n->nlmsg_seq, RTM_DELACTION, sizeof(*t));
782 t = NLMSG_DATA(nlh);
783 t->tca_family = AF_UNSPEC;
Patrick McHardy9ef1d4c2005-06-28 12:55:30 -0700784 t->tca__pad1 = 0;
785 t->tca__pad2 = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700786
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800787 nest = nla_nest_start(skb, TCA_ACT_TAB);
788 if (nest == NULL)
789 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700790
791 err = a->ops->walk(skb, &dcb, RTM_DELACTION, a);
792 if (err < 0)
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800793 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700794
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800795 nla_nest_end(skb, nest);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700796
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -0700797 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700798 nlh->nlmsg_flags |= NLM_F_ROOT;
799 module_put(a->ops->owner);
800 kfree(a);
Denis V. Lunev97c53ca2007-11-19 22:26:51 -0800801 err = rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, n->nlmsg_flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700802 if (err > 0)
803 return 0;
804
805 return err;
806
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800807nla_put_failure:
Linus Torvalds1da177e2005-04-16 15:20:36 -0700808nlmsg_failure:
Thomas Grafebbaeab2006-07-09 11:36:23 -0700809 module_put(a->ops->owner);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700810err_out:
811 kfree_skb(skb);
812 kfree(a);
813 return err;
814}
815
816static int
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800817tca_action_gd(struct nlattr *nla, struct nlmsghdr *n, u32 pid, int event)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700818{
Patrick McHardycee63722008-01-23 20:33:32 -0800819 int i, ret;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800820 struct nlattr *tb[TCA_ACT_MAX_PRIO+1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700821 struct tc_action *head = NULL, *act, *act_prev = NULL;
822
Patrick McHardycee63722008-01-23 20:33:32 -0800823 ret = nla_parse_nested(tb, TCA_ACT_MAX_PRIO, nla, NULL);
824 if (ret < 0)
825 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700826
827 if (event == RTM_DELACTION && n->nlmsg_flags&NLM_F_ROOT) {
828 if (tb[0] != NULL && tb[1] == NULL)
829 return tca_action_flush(tb[0], n, pid);
830 }
831
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800832 for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) {
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800833 act = tcf_action_get_1(tb[i], n, pid);
834 if (IS_ERR(act)) {
835 ret = PTR_ERR(act);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700836 goto err;
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800837 }
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800838 act->order = i;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700839
840 if (head == NULL)
841 head = act;
842 else
843 act_prev->next = act;
844 act_prev = act;
845 }
846
847 if (event == RTM_GETACTION)
848 ret = act_get_notify(pid, n, head, event);
849 else { /* delete */
850 struct sk_buff *skb;
851
852 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
853 if (!skb) {
854 ret = -ENOBUFS;
855 goto err;
856 }
857
858 if (tca_get_fill(skb, head, pid, n->nlmsg_seq, 0, event,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900859 0, 1) <= 0) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700860 kfree_skb(skb);
861 ret = -EINVAL;
862 goto err;
863 }
864
865 /* now do the delete */
866 tcf_action_destroy(head, 0);
Denis V. Lunev97c53ca2007-11-19 22:26:51 -0800867 ret = rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900868 n->nlmsg_flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700869 if (ret > 0)
870 return 0;
871 return ret;
872 }
873err:
874 cleanup_a(head);
875 return ret;
876}
877
878static int tcf_add_notify(struct tc_action *a, u32 pid, u32 seq, int event,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900879 u16 flags)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700880{
881 struct tcamsg *t;
882 struct nlmsghdr *nlh;
883 struct sk_buff *skb;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800884 struct nlattr *nest;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700885 unsigned char *b;
886 int err = 0;
887
888 skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL);
889 if (!skb)
890 return -ENOBUFS;
891
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -0700892 b = skb_tail_pointer(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700893
Jamal Hadi Salime431b8c2005-06-18 22:55:31 -0700894 nlh = NLMSG_NEW(skb, pid, seq, event, sizeof(*t), flags);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700895 t = NLMSG_DATA(nlh);
896 t->tca_family = AF_UNSPEC;
Patrick McHardy9ef1d4c2005-06-28 12:55:30 -0700897 t->tca__pad1 = 0;
898 t->tca__pad2 = 0;
899
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800900 nest = nla_nest_start(skb, TCA_ACT_TAB);
901 if (nest == NULL)
902 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700903
904 if (tcf_action_dump(skb, a, 0, 0) < 0)
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800905 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700906
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800907 nla_nest_end(skb, nest);
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900908
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -0700909 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Patrick McHardyac6d4392005-08-14 19:29:52 -0700910 NETLINK_CB(skb).dst_group = RTNLGRP_TC;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900911
Denis V. Lunev97c53ca2007-11-19 22:26:51 -0800912 err = rtnetlink_send(skb, &init_net, pid, RTNLGRP_TC, flags&NLM_F_ECHO);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700913 if (err > 0)
914 err = 0;
915 return err;
916
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800917nla_put_failure:
Linus Torvalds1da177e2005-04-16 15:20:36 -0700918nlmsg_failure:
Patrick McHardyf6e57462006-03-12 20:33:22 -0800919 kfree_skb(skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700920 return -1;
921}
922
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900923
Linus Torvalds1da177e2005-04-16 15:20:36 -0700924static int
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800925tcf_action_add(struct nlattr *nla, struct nlmsghdr *n, u32 pid, int ovr)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700926{
927 int ret = 0;
928 struct tc_action *act;
929 struct tc_action *a;
930 u32 seq = n->nlmsg_seq;
931
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800932 act = tcf_action_init(nla, NULL, NULL, ovr, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700933 if (act == NULL)
934 goto done;
Patrick McHardyab27cfb2008-01-23 20:33:13 -0800935 if (IS_ERR(act)) {
936 ret = PTR_ERR(act);
937 goto done;
938 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700939
940 /* dump then free all the actions after update; inserted policy
941 * stays intact
942 * */
943 ret = tcf_add_notify(act, pid, seq, RTM_NEWACTION, n->nlmsg_flags);
944 for (a = act; a; a = act) {
945 act = a->next;
946 kfree(a);
947 }
948done:
949 return ret;
950}
951
952static int tc_ctl_action(struct sk_buff *skb, struct nlmsghdr *n, void *arg)
953{
Denis V. Lunevb8542722007-12-01 00:21:31 +1100954 struct net *net = skb->sk->sk_net;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800955 struct nlattr *tca[TCA_ACT_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700956 u32 pid = skb ? NETLINK_CB(skb).pid : 0;
957 int ret = 0, ovr = 0;
958
Denis V. Lunevb8542722007-12-01 00:21:31 +1100959 if (net != &init_net)
960 return -EINVAL;
961
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800962 ret = nlmsg_parse(n, sizeof(struct tcamsg), tca, TCA_ACT_MAX, NULL);
963 if (ret < 0)
964 return ret;
965
966 if (tca[TCA_ACT_TAB] == NULL) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700967 printk("tc_ctl_action: received NO action attribs\n");
968 return -EINVAL;
969 }
970
971 /* n->nlmsg_flags&NLM_F_CREATE
972 * */
973 switch (n->nlmsg_type) {
974 case RTM_NEWACTION:
975 /* we are going to assume all other flags
976 * imply create only if it doesnt exist
977 * Note that CREATE | EXCL implies that
978 * but since we want avoid ambiguity (eg when flags
979 * is zero) then just set this
980 */
981 if (n->nlmsg_flags&NLM_F_REPLACE)
982 ovr = 1;
983replay:
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800984 ret = tcf_action_add(tca[TCA_ACT_TAB], n, pid, ovr);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700985 if (ret == -EAGAIN)
986 goto replay;
987 break;
988 case RTM_DELACTION:
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800989 ret = tca_action_gd(tca[TCA_ACT_TAB], n, pid, RTM_DELACTION);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700990 break;
991 case RTM_GETACTION:
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800992 ret = tca_action_gd(tca[TCA_ACT_TAB], n, pid, RTM_GETACTION);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700993 break;
994 default:
995 BUG();
996 }
997
998 return ret;
999}
1000
Patrick McHardy7ba699c2008-01-22 22:11:50 -08001001static struct nlattr *
Linus Torvalds1da177e2005-04-16 15:20:36 -07001002find_dump_kind(struct nlmsghdr *n)
1003{
Patrick McHardy7ba699c2008-01-22 22:11:50 -08001004 struct nlattr *tb1, *tb2[TCA_ACT_MAX+1];
1005 struct nlattr *tb[TCA_ACT_MAX_PRIO + 1];
1006 struct nlattr *nla[TCAA_MAX + 1];
1007 struct nlattr *kind;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001008
Patrick McHardyc96c9472008-01-23 20:32:58 -08001009 if (nlmsg_parse(n, sizeof(struct tcamsg), nla, TCAA_MAX, NULL) < 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001010 return NULL;
Patrick McHardy7ba699c2008-01-22 22:11:50 -08001011 tb1 = nla[TCA_ACT_TAB];
Linus Torvalds1da177e2005-04-16 15:20:36 -07001012 if (tb1 == NULL)
1013 return NULL;
1014
Patrick McHardy7ba699c2008-01-22 22:11:50 -08001015 if (nla_parse(tb, TCA_ACT_MAX_PRIO, nla_data(tb1),
1016 NLMSG_ALIGN(nla_len(tb1)), NULL) < 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001017 return NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001018
Patrick McHardy6d834e02008-01-23 20:32:42 -08001019 if (tb[1] == NULL)
1020 return NULL;
1021 if (nla_parse(tb2, TCA_ACT_MAX, nla_data(tb[1]),
1022 nla_len(tb[1]), NULL) < 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001023 return NULL;
Patrick McHardy7ba699c2008-01-22 22:11:50 -08001024 kind = tb2[TCA_ACT_KIND];
Linus Torvalds1da177e2005-04-16 15:20:36 -07001025
Thomas Graf26dab892006-07-05 20:45:06 -07001026 return kind;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001027}
1028
1029static int
1030tc_dump_action(struct sk_buff *skb, struct netlink_callback *cb)
1031{
Denis V. Lunevb8542722007-12-01 00:21:31 +11001032 struct net *net = skb->sk->sk_net;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001033 struct nlmsghdr *nlh;
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001034 unsigned char *b = skb_tail_pointer(skb);
Patrick McHardy4b3550ef2008-01-23 20:34:11 -08001035 struct nlattr *nest;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001036 struct tc_action_ops *a_o;
1037 struct tc_action a;
1038 int ret = 0;
1039 struct tcamsg *t = (struct tcamsg *) NLMSG_DATA(cb->nlh);
Patrick McHardy7ba699c2008-01-22 22:11:50 -08001040 struct nlattr *kind = find_dump_kind(cb->nlh);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001041
Denis V. Lunevb8542722007-12-01 00:21:31 +11001042 if (net != &init_net)
1043 return 0;
1044
Linus Torvalds1da177e2005-04-16 15:20:36 -07001045 if (kind == NULL) {
1046 printk("tc_dump_action: action bad kind\n");
1047 return 0;
1048 }
1049
Thomas Graf26dab892006-07-05 20:45:06 -07001050 a_o = tc_lookup_action(kind);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001051 if (a_o == NULL) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001052 return 0;
1053 }
1054
1055 memset(&a, 0, sizeof(struct tc_action));
1056 a.ops = a_o;
1057
1058 if (a_o->walk == NULL) {
Thomas Graf26dab892006-07-05 20:45:06 -07001059 printk("tc_dump_action: %s !capable of dumping table\n", a_o->kind);
Patrick McHardy7ba699c2008-01-22 22:11:50 -08001060 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001061 }
1062
1063 nlh = NLMSG_PUT(skb, NETLINK_CB(cb->skb).pid, cb->nlh->nlmsg_seq,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +09001064 cb->nlh->nlmsg_type, sizeof(*t));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001065 t = NLMSG_DATA(nlh);
1066 t->tca_family = AF_UNSPEC;
Patrick McHardy9ef1d4c2005-06-28 12:55:30 -07001067 t->tca__pad1 = 0;
1068 t->tca__pad2 = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001069
Patrick McHardy4b3550ef2008-01-23 20:34:11 -08001070 nest = nla_nest_start(skb, TCA_ACT_TAB);
1071 if (nest == NULL)
1072 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001073
1074 ret = a_o->walk(skb, cb, RTM_GETACTION, &a);
1075 if (ret < 0)
Patrick McHardy7ba699c2008-01-22 22:11:50 -08001076 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001077
1078 if (ret > 0) {
Patrick McHardy4b3550ef2008-01-23 20:34:11 -08001079 nla_nest_end(skb, nest);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001080 ret = skb->len;
1081 } else
Patrick McHardy4b3550ef2008-01-23 20:34:11 -08001082 nla_nest_cancel(skb, nest);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001083
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -07001084 nlh->nlmsg_len = skb_tail_pointer(skb) - b;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001085 if (NETLINK_CB(cb->skb).pid && ret)
1086 nlh->nlmsg_flags |= NLM_F_MULTI;
1087 module_put(a_o->owner);
1088 return skb->len;
1089
Patrick McHardy7ba699c2008-01-22 22:11:50 -08001090nla_put_failure:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001091nlmsg_failure:
1092 module_put(a_o->owner);
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -07001093 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001094 return skb->len;
1095}
1096
1097static int __init tc_action_init(void)
1098{
Thomas Graf708914c2007-03-22 11:56:59 -07001099 rtnl_register(PF_UNSPEC, RTM_NEWACTION, tc_ctl_action, NULL);
1100 rtnl_register(PF_UNSPEC, RTM_DELACTION, tc_ctl_action, NULL);
1101 rtnl_register(PF_UNSPEC, RTM_GETACTION, tc_ctl_action, tc_dump_action);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001102
Linus Torvalds1da177e2005-04-16 15:20:36 -07001103 return 0;
1104}
1105
1106subsys_initcall(tc_action_init);