Commit | Line | Data |
---|---|---|
1da177e4 LT |
1 | /* |
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 | ||
1da177e4 LT |
14 | #include <linux/types.h> |
15 | #include <linux/kernel.h> | |
1da177e4 | 16 | #include <linux/string.h> |
1da177e4 | 17 | #include <linux/errno.h> |
5a0e3ad6 | 18 | #include <linux/slab.h> |
1da177e4 | 19 | #include <linux/skbuff.h> |
1da177e4 LT |
20 | #include <linux/init.h> |
21 | #include <linux/kmod.h> | |
ab27cfb8 | 22 | #include <linux/err.h> |
3a9a231d | 23 | #include <linux/module.h> |
b854272b DL |
24 | #include <net/net_namespace.h> |
25 | #include <net/sock.h> | |
1da177e4 LT |
26 | #include <net/sch_generic.h> |
27 | #include <net/act_api.h> | |
dc5fc579 | 28 | #include <net/netlink.h> |
1da177e4 | 29 | |
e9ce1cd3 DM |
30 | void tcf_hash_destroy(struct tcf_common *p, struct tcf_hashinfo *hinfo) |
31 | { | |
32 | unsigned int h = tcf_hash(p->tcfc_index, hinfo->hmask); | |
33 | struct tcf_common **p1p; | |
34 | ||
35 | for (p1p = &hinfo->htab[h]; *p1p; p1p = &(*p1p)->tcfc_next) { | |
36 | if (*p1p == p) { | |
37 | write_lock_bh(hinfo->lock); | |
38 | *p1p = p->tcfc_next; | |
39 | write_unlock_bh(hinfo->lock); | |
e9ce1cd3 DM |
40 | gen_kill_estimator(&p->tcfc_bstats, |
41 | &p->tcfc_rate_est); | |
c7de2cf0 ED |
42 | /* |
43 | * gen_estimator est_timer() might access p->tcfc_lock | |
44 | * or bstats, wait a RCU grace period before freeing p | |
45 | */ | |
f5c8593c | 46 | kfree_rcu(p, tcfc_rcu); |
e9ce1cd3 DM |
47 | return; |
48 | } | |
49 | } | |
547b792c | 50 | WARN_ON(1); |
e9ce1cd3 DM |
51 | } |
52 | EXPORT_SYMBOL(tcf_hash_destroy); | |
53 | ||
54 | int tcf_hash_release(struct tcf_common *p, int bind, | |
55 | struct tcf_hashinfo *hinfo) | |
56 | { | |
57 | int ret = 0; | |
58 | ||
59 | if (p) { | |
60 | if (bind) | |
61 | p->tcfc_bindcnt--; | |
62 | ||
63 | p->tcfc_refcnt--; | |
10297b99 | 64 | if (p->tcfc_bindcnt <= 0 && p->tcfc_refcnt <= 0) { |
e9ce1cd3 DM |
65 | tcf_hash_destroy(p, hinfo); |
66 | ret = 1; | |
67 | } | |
68 | } | |
69 | return ret; | |
70 | } | |
71 | EXPORT_SYMBOL(tcf_hash_release); | |
72 | ||
73 | static int tcf_dump_walker(struct sk_buff *skb, struct netlink_callback *cb, | |
74 | struct tc_action *a, struct tcf_hashinfo *hinfo) | |
75 | { | |
76 | struct tcf_common *p; | |
cc7ec456 | 77 | int err = 0, index = -1, i = 0, s_i = 0, n_i = 0; |
4b3550ef | 78 | struct nlattr *nest; |
e9ce1cd3 | 79 | |
e1e992e5 | 80 | read_lock_bh(hinfo->lock); |
e9ce1cd3 DM |
81 | |
82 | s_i = cb->args[0]; | |
83 | ||
84 | for (i = 0; i < (hinfo->hmask + 1); i++) { | |
85 | p = hinfo->htab[tcf_hash(i, hinfo->hmask)]; | |
86 | ||
87 | for (; p; p = p->tcfc_next) { | |
88 | index++; | |
89 | if (index < s_i) | |
90 | continue; | |
91 | a->priv = p; | |
92 | a->order = n_i; | |
4b3550ef PM |
93 | |
94 | nest = nla_nest_start(skb, a->order); | |
95 | if (nest == NULL) | |
96 | goto nla_put_failure; | |
e9ce1cd3 DM |
97 | err = tcf_action_dump_1(skb, a, 0, 0); |
98 | if (err < 0) { | |
99 | index--; | |
4b3550ef | 100 | nlmsg_trim(skb, nest); |
e9ce1cd3 DM |
101 | goto done; |
102 | } | |
4b3550ef | 103 | nla_nest_end(skb, nest); |
e9ce1cd3 DM |
104 | n_i++; |
105 | if (n_i >= TCA_ACT_MAX_PRIO) | |
106 | goto done; | |
107 | } | |
108 | } | |
109 | done: | |
e1e992e5 | 110 | read_unlock_bh(hinfo->lock); |
e9ce1cd3 DM |
111 | if (n_i) |
112 | cb->args[0] += n_i; | |
113 | return n_i; | |
114 | ||
7ba699c6 | 115 | nla_put_failure: |
4b3550ef | 116 | nla_nest_cancel(skb, nest); |
e9ce1cd3 DM |
117 | goto done; |
118 | } | |
119 | ||
120 | static int tcf_del_walker(struct sk_buff *skb, struct tc_action *a, | |
121 | struct tcf_hashinfo *hinfo) | |
122 | { | |
123 | struct tcf_common *p, *s_p; | |
4b3550ef | 124 | struct nlattr *nest; |
cc7ec456 | 125 | int i = 0, n_i = 0; |
e9ce1cd3 | 126 | |
4b3550ef PM |
127 | nest = nla_nest_start(skb, a->order); |
128 | if (nest == NULL) | |
129 | goto nla_put_failure; | |
1b34ec43 DM |
130 | if (nla_put_string(skb, TCA_KIND, a->ops->kind)) |
131 | goto nla_put_failure; | |
e9ce1cd3 DM |
132 | for (i = 0; i < (hinfo->hmask + 1); i++) { |
133 | p = hinfo->htab[tcf_hash(i, hinfo->hmask)]; | |
134 | ||
135 | while (p != NULL) { | |
136 | s_p = p->tcfc_next; | |
137 | if (ACT_P_DELETED == tcf_hash_release(p, 0, hinfo)) | |
cc7ec456 | 138 | module_put(a->ops->owner); |
e9ce1cd3 DM |
139 | n_i++; |
140 | p = s_p; | |
141 | } | |
142 | } | |
1b34ec43 DM |
143 | if (nla_put_u32(skb, TCA_FCNT, n_i)) |
144 | goto nla_put_failure; | |
4b3550ef | 145 | nla_nest_end(skb, nest); |
e9ce1cd3 DM |
146 | |
147 | return n_i; | |
7ba699c6 | 148 | nla_put_failure: |
4b3550ef | 149 | nla_nest_cancel(skb, nest); |
e9ce1cd3 DM |
150 | return -EINVAL; |
151 | } | |
152 | ||
153 | int tcf_generic_walker(struct sk_buff *skb, struct netlink_callback *cb, | |
154 | int type, struct tc_action *a) | |
155 | { | |
156 | struct tcf_hashinfo *hinfo = a->ops->hinfo; | |
157 | ||
158 | if (type == RTM_DELACTION) { | |
159 | return tcf_del_walker(skb, a, hinfo); | |
160 | } else if (type == RTM_GETACTION) { | |
161 | return tcf_dump_walker(skb, cb, a, hinfo); | |
162 | } else { | |
6ff9c364 | 163 | WARN(1, "tcf_generic_walker: unknown action %d\n", type); |
e9ce1cd3 DM |
164 | return -EINVAL; |
165 | } | |
166 | } | |
167 | EXPORT_SYMBOL(tcf_generic_walker); | |
168 | ||
169 | struct tcf_common *tcf_hash_lookup(u32 index, struct tcf_hashinfo *hinfo) | |
170 | { | |
171 | struct tcf_common *p; | |
172 | ||
e1e992e5 | 173 | read_lock_bh(hinfo->lock); |
e9ce1cd3 DM |
174 | for (p = hinfo->htab[tcf_hash(index, hinfo->hmask)]; p; |
175 | p = p->tcfc_next) { | |
176 | if (p->tcfc_index == index) | |
177 | break; | |
178 | } | |
e1e992e5 | 179 | read_unlock_bh(hinfo->lock); |
e9ce1cd3 DM |
180 | |
181 | return p; | |
182 | } | |
183 | EXPORT_SYMBOL(tcf_hash_lookup); | |
184 | ||
185 | u32 tcf_hash_new_index(u32 *idx_gen, struct tcf_hashinfo *hinfo) | |
186 | { | |
187 | u32 val = *idx_gen; | |
188 | ||
189 | do { | |
190 | if (++val == 0) | |
191 | val = 1; | |
192 | } while (tcf_hash_lookup(val, hinfo)); | |
193 | ||
194 | return (*idx_gen = val); | |
195 | } | |
196 | EXPORT_SYMBOL(tcf_hash_new_index); | |
197 | ||
198 | int tcf_hash_search(struct tc_action *a, u32 index) | |
199 | { | |
200 | struct tcf_hashinfo *hinfo = a->ops->hinfo; | |
201 | struct tcf_common *p = tcf_hash_lookup(index, hinfo); | |
202 | ||
203 | if (p) { | |
204 | a->priv = p; | |
205 | return 1; | |
206 | } | |
207 | return 0; | |
208 | } | |
209 | EXPORT_SYMBOL(tcf_hash_search); | |
210 | ||
211 | struct tcf_common *tcf_hash_check(u32 index, struct tc_action *a, int bind, | |
212 | struct tcf_hashinfo *hinfo) | |
213 | { | |
214 | struct tcf_common *p = NULL; | |
215 | if (index && (p = tcf_hash_lookup(index, hinfo)) != NULL) { | |
76aab2c1 | 216 | if (bind) |
e9ce1cd3 | 217 | p->tcfc_bindcnt++; |
76aab2c1 | 218 | p->tcfc_refcnt++; |
e9ce1cd3 DM |
219 | a->priv = p; |
220 | } | |
221 | return p; | |
222 | } | |
223 | EXPORT_SYMBOL(tcf_hash_check); | |
224 | ||
0e991ec6 SH |
225 | struct tcf_common *tcf_hash_create(u32 index, struct nlattr *est, |
226 | struct tc_action *a, int size, int bind, | |
227 | u32 *idx_gen, struct tcf_hashinfo *hinfo) | |
e9ce1cd3 DM |
228 | { |
229 | struct tcf_common *p = kzalloc(size, GFP_KERNEL); | |
230 | ||
231 | if (unlikely(!p)) | |
0e991ec6 | 232 | return ERR_PTR(-ENOMEM); |
e9ce1cd3 DM |
233 | p->tcfc_refcnt = 1; |
234 | if (bind) | |
235 | p->tcfc_bindcnt = 1; | |
236 | ||
237 | spin_lock_init(&p->tcfc_lock); | |
e9ce1cd3 DM |
238 | p->tcfc_index = index ? index : tcf_hash_new_index(idx_gen, hinfo); |
239 | p->tcfc_tm.install = jiffies; | |
240 | p->tcfc_tm.lastuse = jiffies; | |
0e991ec6 SH |
241 | if (est) { |
242 | int err = gen_new_estimator(&p->tcfc_bstats, &p->tcfc_rate_est, | |
243 | &p->tcfc_lock, est); | |
244 | if (err) { | |
245 | kfree(p); | |
246 | return ERR_PTR(err); | |
247 | } | |
248 | } | |
249 | ||
e9ce1cd3 DM |
250 | a->priv = (void *) p; |
251 | return p; | |
252 | } | |
253 | EXPORT_SYMBOL(tcf_hash_create); | |
254 | ||
255 | void tcf_hash_insert(struct tcf_common *p, struct tcf_hashinfo *hinfo) | |
256 | { | |
257 | unsigned int h = tcf_hash(p->tcfc_index, hinfo->hmask); | |
258 | ||
259 | write_lock_bh(hinfo->lock); | |
260 | p->tcfc_next = hinfo->htab[h]; | |
261 | hinfo->htab[h] = p; | |
262 | write_unlock_bh(hinfo->lock); | |
263 | } | |
264 | EXPORT_SYMBOL(tcf_hash_insert); | |
1da177e4 LT |
265 | |
266 | static struct tc_action_ops *act_base = NULL; | |
267 | static DEFINE_RWLOCK(act_mod_lock); | |
268 | ||
269 | int tcf_register_action(struct tc_action_ops *act) | |
270 | { | |
271 | struct tc_action_ops *a, **ap; | |
272 | ||
76c82d7a JHS |
273 | /* Must supply act, dump, cleanup and init */ |
274 | if (!act->act || !act->dump || !act->cleanup || !act->init) | |
275 | return -EINVAL; | |
276 | ||
63ef6174 JHS |
277 | if (!act->lookup) |
278 | act->lookup = tcf_hash_search; | |
279 | ||
1da177e4 LT |
280 | write_lock(&act_mod_lock); |
281 | for (ap = &act_base; (a = *ap) != NULL; ap = &a->next) { | |
282 | if (act->type == a->type || (strcmp(act->kind, a->kind) == 0)) { | |
283 | write_unlock(&act_mod_lock); | |
284 | return -EEXIST; | |
285 | } | |
286 | } | |
287 | act->next = NULL; | |
288 | *ap = act; | |
289 | write_unlock(&act_mod_lock); | |
290 | return 0; | |
291 | } | |
62e3ba1b | 292 | EXPORT_SYMBOL(tcf_register_action); |
1da177e4 LT |
293 | |
294 | int tcf_unregister_action(struct tc_action_ops *act) | |
295 | { | |
296 | struct tc_action_ops *a, **ap; | |
297 | int err = -ENOENT; | |
298 | ||
299 | write_lock(&act_mod_lock); | |
300 | for (ap = &act_base; (a = *ap) != NULL; ap = &a->next) | |
301 | if (a == act) | |
302 | break; | |
303 | if (a) { | |
304 | *ap = a->next; | |
305 | a->next = NULL; | |
306 | err = 0; | |
307 | } | |
308 | write_unlock(&act_mod_lock); | |
309 | return err; | |
310 | } | |
62e3ba1b | 311 | EXPORT_SYMBOL(tcf_unregister_action); |
1da177e4 LT |
312 | |
313 | /* lookup by name */ | |
314 | static struct tc_action_ops *tc_lookup_action_n(char *kind) | |
315 | { | |
316 | struct tc_action_ops *a = NULL; | |
317 | ||
318 | if (kind) { | |
319 | read_lock(&act_mod_lock); | |
320 | for (a = act_base; a; a = a->next) { | |
321 | if (strcmp(kind, a->kind) == 0) { | |
322 | if (!try_module_get(a->owner)) { | |
323 | read_unlock(&act_mod_lock); | |
324 | return NULL; | |
325 | } | |
326 | break; | |
327 | } | |
328 | } | |
329 | read_unlock(&act_mod_lock); | |
330 | } | |
331 | return a; | |
332 | } | |
333 | ||
7ba699c6 PM |
334 | /* lookup by nlattr */ |
335 | static struct tc_action_ops *tc_lookup_action(struct nlattr *kind) | |
1da177e4 LT |
336 | { |
337 | struct tc_action_ops *a = NULL; | |
338 | ||
339 | if (kind) { | |
340 | read_lock(&act_mod_lock); | |
341 | for (a = act_base; a; a = a->next) { | |
7ba699c6 | 342 | if (nla_strcmp(kind, a->kind) == 0) { |
1da177e4 LT |
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 | ||
355 | #if 0 | |
356 | /* lookup by id */ | |
357 | static struct tc_action_ops *tc_lookup_action_id(u32 type) | |
358 | { | |
359 | struct tc_action_ops *a = NULL; | |
360 | ||
361 | if (type) { | |
362 | read_lock(&act_mod_lock); | |
363 | for (a = act_base; a; a = a->next) { | |
364 | if (a->type == type) { | |
365 | if (!try_module_get(a->owner)) { | |
366 | read_unlock(&act_mod_lock); | |
367 | return NULL; | |
368 | } | |
369 | break; | |
370 | } | |
371 | } | |
372 | read_unlock(&act_mod_lock); | |
373 | } | |
374 | return a; | |
375 | } | |
376 | #endif | |
377 | ||
dc7f9f6e | 378 | int tcf_action_exec(struct sk_buff *skb, const struct tc_action *act, |
10297b99 | 379 | struct tcf_result *res) |
1da177e4 | 380 | { |
dc7f9f6e | 381 | const struct tc_action *a; |
1da177e4 LT |
382 | int ret = -1; |
383 | ||
384 | if (skb->tc_verd & TC_NCLS) { | |
385 | skb->tc_verd = CLR_TC_NCLS(skb->tc_verd); | |
1da177e4 LT |
386 | ret = TC_ACT_OK; |
387 | goto exec_done; | |
388 | } | |
389 | while ((a = act) != NULL) { | |
390 | repeat: | |
76c82d7a | 391 | if (a->ops) { |
f43c5a0d | 392 | ret = a->ops->act(skb, a, res); |
1da177e4 LT |
393 | if (TC_MUNGED & skb->tc_verd) { |
394 | /* copied already, allow trampling */ | |
395 | skb->tc_verd = SET_TC_OK2MUNGE(skb->tc_verd); | |
396 | skb->tc_verd = CLR_TC_MUNGED(skb->tc_verd); | |
397 | } | |
1da177e4 LT |
398 | if (ret == TC_ACT_REPEAT) |
399 | goto repeat; /* we need a ttl - JHS */ | |
14d50e78 HS |
400 | if (ret != TC_ACT_PIPE) |
401 | goto exec_done; | |
1da177e4 LT |
402 | } |
403 | act = a->next; | |
404 | } | |
405 | exec_done: | |
1da177e4 LT |
406 | return ret; |
407 | } | |
62e3ba1b | 408 | EXPORT_SYMBOL(tcf_action_exec); |
1da177e4 LT |
409 | |
410 | void tcf_action_destroy(struct tc_action *act, int bind) | |
411 | { | |
412 | struct tc_action *a; | |
413 | ||
414 | for (a = act; a; a = act) { | |
76c82d7a | 415 | if (a->ops) { |
1da177e4 LT |
416 | if (a->ops->cleanup(a, bind) == ACT_P_DELETED) |
417 | module_put(a->ops->owner); | |
418 | act = act->next; | |
419 | kfree(a); | |
6ff9c364 | 420 | } else { |
421 | /*FIXME: Remove later - catch insertion bugs*/ | |
422 | WARN(1, "tcf_action_destroy: BUG? destroying NULL ops\n"); | |
1da177e4 LT |
423 | act = act->next; |
424 | kfree(a); | |
425 | } | |
426 | } | |
427 | } | |
428 | ||
429 | int | |
430 | tcf_action_dump_old(struct sk_buff *skb, struct tc_action *a, int bind, int ref) | |
431 | { | |
432 | int err = -EINVAL; | |
433 | ||
76c82d7a | 434 | if (a->ops == NULL) |
1da177e4 LT |
435 | return err; |
436 | return a->ops->dump(skb, a, bind, ref); | |
437 | } | |
438 | ||
439 | int | |
440 | tcf_action_dump_1(struct sk_buff *skb, struct tc_action *a, int bind, int ref) | |
441 | { | |
442 | int err = -EINVAL; | |
27a884dc | 443 | unsigned char *b = skb_tail_pointer(skb); |
4b3550ef | 444 | struct nlattr *nest; |
1da177e4 | 445 | |
76c82d7a | 446 | if (a->ops == NULL) |
1da177e4 LT |
447 | return err; |
448 | ||
1b34ec43 DM |
449 | if (nla_put_string(skb, TCA_KIND, a->ops->kind)) |
450 | goto nla_put_failure; | |
1da177e4 | 451 | if (tcf_action_copy_stats(skb, a, 0)) |
7ba699c6 | 452 | goto nla_put_failure; |
4b3550ef PM |
453 | nest = nla_nest_start(skb, TCA_OPTIONS); |
454 | if (nest == NULL) | |
455 | goto nla_put_failure; | |
cc7ec456 ED |
456 | err = tcf_action_dump_old(skb, a, bind, ref); |
457 | if (err > 0) { | |
4b3550ef | 458 | nla_nest_end(skb, nest); |
1da177e4 LT |
459 | return err; |
460 | } | |
461 | ||
7ba699c6 | 462 | nla_put_failure: |
dc5fc579 | 463 | nlmsg_trim(skb, b); |
1da177e4 LT |
464 | return -1; |
465 | } | |
62e3ba1b | 466 | EXPORT_SYMBOL(tcf_action_dump_1); |
1da177e4 LT |
467 | |
468 | int | |
469 | tcf_action_dump(struct sk_buff *skb, struct tc_action *act, int bind, int ref) | |
470 | { | |
471 | struct tc_action *a; | |
472 | int err = -EINVAL; | |
4b3550ef | 473 | struct nlattr *nest; |
1da177e4 LT |
474 | |
475 | while ((a = act) != NULL) { | |
1da177e4 | 476 | act = a->next; |
4b3550ef PM |
477 | nest = nla_nest_start(skb, a->order); |
478 | if (nest == NULL) | |
479 | goto nla_put_failure; | |
1da177e4 LT |
480 | err = tcf_action_dump_1(skb, a, bind, ref); |
481 | if (err < 0) | |
4fe683f5 | 482 | goto errout; |
4b3550ef | 483 | nla_nest_end(skb, nest); |
1da177e4 LT |
484 | } |
485 | ||
486 | return 0; | |
487 | ||
7ba699c6 | 488 | nla_put_failure: |
4fe683f5 TG |
489 | err = -EINVAL; |
490 | errout: | |
4b3550ef | 491 | nla_nest_cancel(skb, nest); |
4fe683f5 | 492 | return err; |
1da177e4 LT |
493 | } |
494 | ||
c1b52739 BL |
495 | struct tc_action *tcf_action_init_1(struct net *net, struct nlattr *nla, |
496 | struct nlattr *est, char *name, int ovr, | |
497 | int bind) | |
1da177e4 LT |
498 | { |
499 | struct tc_action *a; | |
500 | struct tc_action_ops *a_o; | |
501 | char act_name[IFNAMSIZ]; | |
cc7ec456 | 502 | struct nlattr *tb[TCA_ACT_MAX + 1]; |
7ba699c6 | 503 | struct nlattr *kind; |
ab27cfb8 | 504 | int err; |
1da177e4 | 505 | |
1da177e4 | 506 | if (name == NULL) { |
cee63723 PM |
507 | err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL); |
508 | if (err < 0) | |
1da177e4 | 509 | goto err_out; |
cee63723 | 510 | err = -EINVAL; |
7ba699c6 | 511 | kind = tb[TCA_ACT_KIND]; |
1da177e4 LT |
512 | if (kind == NULL) |
513 | goto err_out; | |
7ba699c6 | 514 | if (nla_strlcpy(act_name, kind, IFNAMSIZ) >= IFNAMSIZ) |
1da177e4 LT |
515 | goto err_out; |
516 | } else { | |
cee63723 | 517 | err = -EINVAL; |
1da177e4 LT |
518 | if (strlcpy(act_name, name, IFNAMSIZ) >= IFNAMSIZ) |
519 | goto err_out; | |
520 | } | |
521 | ||
522 | a_o = tc_lookup_action_n(act_name); | |
523 | if (a_o == NULL) { | |
95a5afca | 524 | #ifdef CONFIG_MODULES |
1da177e4 | 525 | rtnl_unlock(); |
4bba3925 | 526 | request_module("act_%s", act_name); |
1da177e4 LT |
527 | rtnl_lock(); |
528 | ||
529 | a_o = tc_lookup_action_n(act_name); | |
530 | ||
531 | /* We dropped the RTNL semaphore in order to | |
532 | * perform the module load. So, even if we | |
533 | * succeeded in loading the module we have to | |
534 | * tell the caller to replay the request. We | |
535 | * indicate this using -EAGAIN. | |
536 | */ | |
537 | if (a_o != NULL) { | |
ab27cfb8 | 538 | err = -EAGAIN; |
1da177e4 LT |
539 | goto err_mod; |
540 | } | |
541 | #endif | |
ab27cfb8 | 542 | err = -ENOENT; |
1da177e4 LT |
543 | goto err_out; |
544 | } | |
545 | ||
ab27cfb8 | 546 | err = -ENOMEM; |
0da974f4 | 547 | a = kzalloc(sizeof(*a), GFP_KERNEL); |
1da177e4 LT |
548 | if (a == NULL) |
549 | goto err_mod; | |
1da177e4 LT |
550 | |
551 | /* backward compatibility for policer */ | |
552 | if (name == NULL) | |
c1b52739 | 553 | err = a_o->init(net, tb[TCA_ACT_OPTIONS], est, a, ovr, bind); |
1da177e4 | 554 | else |
c1b52739 | 555 | err = a_o->init(net, nla, est, a, ovr, bind); |
ab27cfb8 | 556 | if (err < 0) |
1da177e4 LT |
557 | goto err_free; |
558 | ||
559 | /* module count goes up only when brand new policy is created | |
cc7ec456 ED |
560 | * if it exists and is only bound to in a_o->init() then |
561 | * ACT_P_CREATED is not returned (a zero is). | |
562 | */ | |
ab27cfb8 | 563 | if (err != ACT_P_CREATED) |
1da177e4 LT |
564 | module_put(a_o->owner); |
565 | a->ops = a_o; | |
1da177e4 | 566 | |
1da177e4 LT |
567 | return a; |
568 | ||
569 | err_free: | |
570 | kfree(a); | |
571 | err_mod: | |
572 | module_put(a_o->owner); | |
573 | err_out: | |
ab27cfb8 | 574 | return ERR_PTR(err); |
1da177e4 LT |
575 | } |
576 | ||
c1b52739 BL |
577 | struct tc_action *tcf_action_init(struct net *net, struct nlattr *nla, |
578 | struct nlattr *est, char *name, int ovr, | |
579 | int bind) | |
1da177e4 | 580 | { |
cc7ec456 | 581 | struct nlattr *tb[TCA_ACT_MAX_PRIO + 1]; |
1da177e4 | 582 | struct tc_action *head = NULL, *act, *act_prev = NULL; |
cee63723 | 583 | int err; |
1da177e4 LT |
584 | int i; |
585 | ||
cee63723 PM |
586 | err = nla_parse_nested(tb, TCA_ACT_MAX_PRIO, nla, NULL); |
587 | if (err < 0) | |
588 | return ERR_PTR(err); | |
1da177e4 | 589 | |
7ba699c6 | 590 | for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) { |
c1b52739 | 591 | act = tcf_action_init_1(net, tb[i], est, name, ovr, bind); |
ab27cfb8 | 592 | if (IS_ERR(act)) |
1da177e4 | 593 | goto err; |
7ba699c6 | 594 | act->order = i; |
1da177e4 LT |
595 | |
596 | if (head == NULL) | |
597 | head = act; | |
598 | else | |
599 | act_prev->next = act; | |
600 | act_prev = act; | |
601 | } | |
602 | return head; | |
603 | ||
604 | err: | |
605 | if (head != NULL) | |
606 | tcf_action_destroy(head, bind); | |
ab27cfb8 | 607 | return act; |
1da177e4 LT |
608 | } |
609 | ||
610 | int tcf_action_copy_stats(struct sk_buff *skb, struct tc_action *a, | |
611 | int compat_mode) | |
612 | { | |
613 | int err = 0; | |
614 | struct gnet_dump d; | |
615 | struct tcf_act_hdr *h = a->priv; | |
10297b99 | 616 | |
1da177e4 LT |
617 | if (h == NULL) |
618 | goto errout; | |
619 | ||
620 | /* compat_mode being true specifies a call that is supposed | |
06fe9fb4 | 621 | * to add additional backward compatibility statistic TLVs. |
1da177e4 LT |
622 | */ |
623 | if (compat_mode) { | |
624 | if (a->type == TCA_OLD_COMPAT) | |
625 | err = gnet_stats_start_copy_compat(skb, 0, | |
4bdf3991 | 626 | TCA_STATS, TCA_XSTATS, &h->tcf_lock, &d); |
1da177e4 LT |
627 | else |
628 | return 0; | |
629 | } else | |
630 | err = gnet_stats_start_copy(skb, TCA_ACT_STATS, | |
4bdf3991 | 631 | &h->tcf_lock, &d); |
1da177e4 LT |
632 | |
633 | if (err < 0) | |
634 | goto errout; | |
635 | ||
636 | if (a->ops != NULL && a->ops->get_stats != NULL) | |
637 | if (a->ops->get_stats(skb, a) < 0) | |
638 | goto errout; | |
639 | ||
e9ce1cd3 | 640 | if (gnet_stats_copy_basic(&d, &h->tcf_bstats) < 0 || |
d250a5f9 ED |
641 | gnet_stats_copy_rate_est(&d, &h->tcf_bstats, |
642 | &h->tcf_rate_est) < 0 || | |
e9ce1cd3 | 643 | gnet_stats_copy_queue(&d, &h->tcf_qstats) < 0) |
1da177e4 LT |
644 | goto errout; |
645 | ||
646 | if (gnet_stats_finish_copy(&d) < 0) | |
647 | goto errout; | |
648 | ||
649 | return 0; | |
650 | ||
651 | errout: | |
652 | return -1; | |
653 | } | |
654 | ||
655 | static int | |
15e47304 | 656 | tca_get_fill(struct sk_buff *skb, struct tc_action *a, u32 portid, u32 seq, |
10297b99 | 657 | u16 flags, int event, int bind, int ref) |
1da177e4 LT |
658 | { |
659 | struct tcamsg *t; | |
660 | struct nlmsghdr *nlh; | |
27a884dc | 661 | unsigned char *b = skb_tail_pointer(skb); |
4b3550ef | 662 | struct nlattr *nest; |
1da177e4 | 663 | |
15e47304 | 664 | nlh = nlmsg_put(skb, portid, seq, event, sizeof(*t), flags); |
8b00a53c DM |
665 | if (!nlh) |
666 | goto out_nlmsg_trim; | |
667 | t = nlmsg_data(nlh); | |
1da177e4 | 668 | t->tca_family = AF_UNSPEC; |
9ef1d4c7 PM |
669 | t->tca__pad1 = 0; |
670 | t->tca__pad2 = 0; | |
10297b99 | 671 | |
4b3550ef PM |
672 | nest = nla_nest_start(skb, TCA_ACT_TAB); |
673 | if (nest == NULL) | |
8b00a53c | 674 | goto out_nlmsg_trim; |
1da177e4 LT |
675 | |
676 | if (tcf_action_dump(skb, a, bind, ref) < 0) | |
8b00a53c | 677 | goto out_nlmsg_trim; |
1da177e4 | 678 | |
4b3550ef | 679 | nla_nest_end(skb, nest); |
10297b99 | 680 | |
27a884dc | 681 | nlh->nlmsg_len = skb_tail_pointer(skb) - b; |
1da177e4 LT |
682 | return skb->len; |
683 | ||
8b00a53c | 684 | out_nlmsg_trim: |
dc5fc579 | 685 | nlmsg_trim(skb, b); |
1da177e4 LT |
686 | return -1; |
687 | } | |
688 | ||
689 | static int | |
15e47304 | 690 | act_get_notify(struct net *net, u32 portid, struct nlmsghdr *n, |
7316ae88 | 691 | struct tc_action *a, int event) |
1da177e4 LT |
692 | { |
693 | struct sk_buff *skb; | |
1da177e4 LT |
694 | |
695 | skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL); | |
696 | if (!skb) | |
697 | return -ENOBUFS; | |
15e47304 | 698 | if (tca_get_fill(skb, a, portid, n->nlmsg_seq, 0, event, 0, 0) <= 0) { |
1da177e4 LT |
699 | kfree_skb(skb); |
700 | return -EINVAL; | |
701 | } | |
2942e900 | 702 | |
15e47304 | 703 | return rtnl_unicast(skb, net, portid); |
1da177e4 LT |
704 | } |
705 | ||
706 | static struct tc_action * | |
15e47304 | 707 | tcf_action_get_1(struct nlattr *nla, struct nlmsghdr *n, u32 portid) |
1da177e4 | 708 | { |
cc7ec456 | 709 | struct nlattr *tb[TCA_ACT_MAX + 1]; |
1da177e4 LT |
710 | struct tc_action *a; |
711 | int index; | |
ab27cfb8 | 712 | int err; |
1da177e4 | 713 | |
cee63723 PM |
714 | err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL); |
715 | if (err < 0) | |
ab27cfb8 | 716 | goto err_out; |
1da177e4 | 717 | |
cee63723 | 718 | err = -EINVAL; |
7ba699c6 PM |
719 | if (tb[TCA_ACT_INDEX] == NULL || |
720 | nla_len(tb[TCA_ACT_INDEX]) < sizeof(index)) | |
ab27cfb8 | 721 | goto err_out; |
1587bac4 | 722 | index = nla_get_u32(tb[TCA_ACT_INDEX]); |
1da177e4 | 723 | |
ab27cfb8 | 724 | err = -ENOMEM; |
0da974f4 | 725 | a = kzalloc(sizeof(struct tc_action), GFP_KERNEL); |
1da177e4 | 726 | if (a == NULL) |
ab27cfb8 | 727 | goto err_out; |
1da177e4 | 728 | |
ab27cfb8 | 729 | err = -EINVAL; |
7ba699c6 | 730 | a->ops = tc_lookup_action(tb[TCA_ACT_KIND]); |
1da177e4 LT |
731 | if (a->ops == NULL) |
732 | goto err_free; | |
ab27cfb8 | 733 | err = -ENOENT; |
1da177e4 LT |
734 | if (a->ops->lookup(a, index) == 0) |
735 | goto err_mod; | |
736 | ||
737 | module_put(a->ops->owner); | |
1da177e4 | 738 | return a; |
ab27cfb8 | 739 | |
1da177e4 LT |
740 | err_mod: |
741 | module_put(a->ops->owner); | |
742 | err_free: | |
743 | kfree(a); | |
ab27cfb8 PM |
744 | err_out: |
745 | return ERR_PTR(err); | |
1da177e4 LT |
746 | } |
747 | ||
748 | static void cleanup_a(struct tc_action *act) | |
749 | { | |
750 | struct tc_action *a; | |
751 | ||
752 | for (a = act; a; a = act) { | |
753 | act = a->next; | |
754 | kfree(a); | |
755 | } | |
756 | } | |
757 | ||
758 | static struct tc_action *create_a(int i) | |
759 | { | |
760 | struct tc_action *act; | |
761 | ||
0da974f4 | 762 | act = kzalloc(sizeof(*act), GFP_KERNEL); |
1da177e4 | 763 | if (act == NULL) { |
6ff9c364 | 764 | pr_debug("create_a: failed to alloc!\n"); |
1da177e4 LT |
765 | return NULL; |
766 | } | |
1da177e4 LT |
767 | act->order = i; |
768 | return act; | |
769 | } | |
770 | ||
7316ae88 | 771 | static int tca_action_flush(struct net *net, struct nlattr *nla, |
15e47304 | 772 | struct nlmsghdr *n, u32 portid) |
1da177e4 LT |
773 | { |
774 | struct sk_buff *skb; | |
775 | unsigned char *b; | |
776 | struct nlmsghdr *nlh; | |
777 | struct tcamsg *t; | |
778 | struct netlink_callback dcb; | |
4b3550ef | 779 | struct nlattr *nest; |
cc7ec456 | 780 | struct nlattr *tb[TCA_ACT_MAX + 1]; |
7ba699c6 | 781 | struct nlattr *kind; |
1da177e4 | 782 | struct tc_action *a = create_a(0); |
36723873 | 783 | int err = -ENOMEM; |
1da177e4 LT |
784 | |
785 | if (a == NULL) { | |
6ff9c364 | 786 | pr_debug("tca_action_flush: couldnt create tc_action\n"); |
1da177e4 LT |
787 | return err; |
788 | } | |
789 | ||
790 | skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL); | |
791 | if (!skb) { | |
6ff9c364 | 792 | pr_debug("tca_action_flush: failed skb alloc\n"); |
1da177e4 | 793 | kfree(a); |
36723873 | 794 | return err; |
1da177e4 LT |
795 | } |
796 | ||
27a884dc | 797 | b = skb_tail_pointer(skb); |
1da177e4 | 798 | |
cee63723 PM |
799 | err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL); |
800 | if (err < 0) | |
1da177e4 LT |
801 | goto err_out; |
802 | ||
cee63723 | 803 | err = -EINVAL; |
7ba699c6 | 804 | kind = tb[TCA_ACT_KIND]; |
1da177e4 LT |
805 | a->ops = tc_lookup_action(kind); |
806 | if (a->ops == NULL) | |
807 | goto err_out; | |
808 | ||
15e47304 | 809 | nlh = nlmsg_put(skb, portid, n->nlmsg_seq, RTM_DELACTION, sizeof(*t), 0); |
8b00a53c DM |
810 | if (!nlh) |
811 | goto out_module_put; | |
812 | t = nlmsg_data(nlh); | |
1da177e4 | 813 | t->tca_family = AF_UNSPEC; |
9ef1d4c7 PM |
814 | t->tca__pad1 = 0; |
815 | t->tca__pad2 = 0; | |
1da177e4 | 816 | |
4b3550ef PM |
817 | nest = nla_nest_start(skb, TCA_ACT_TAB); |
818 | if (nest == NULL) | |
8b00a53c | 819 | goto out_module_put; |
1da177e4 LT |
820 | |
821 | err = a->ops->walk(skb, &dcb, RTM_DELACTION, a); | |
822 | if (err < 0) | |
8b00a53c | 823 | goto out_module_put; |
f97017cd JHS |
824 | if (err == 0) |
825 | goto noflush_out; | |
1da177e4 | 826 | |
4b3550ef | 827 | nla_nest_end(skb, nest); |
1da177e4 | 828 | |
27a884dc | 829 | nlh->nlmsg_len = skb_tail_pointer(skb) - b; |
1da177e4 LT |
830 | nlh->nlmsg_flags |= NLM_F_ROOT; |
831 | module_put(a->ops->owner); | |
832 | kfree(a); | |
15e47304 | 833 | err = rtnetlink_send(skb, net, portid, RTNLGRP_TC, |
cc7ec456 | 834 | n->nlmsg_flags & NLM_F_ECHO); |
1da177e4 LT |
835 | if (err > 0) |
836 | return 0; | |
837 | ||
838 | return err; | |
839 | ||
8b00a53c | 840 | out_module_put: |
ebbaeab1 | 841 | module_put(a->ops->owner); |
1da177e4 | 842 | err_out: |
f97017cd | 843 | noflush_out: |
1da177e4 LT |
844 | kfree_skb(skb); |
845 | kfree(a); | |
846 | return err; | |
847 | } | |
848 | ||
849 | static int | |
7316ae88 | 850 | tca_action_gd(struct net *net, struct nlattr *nla, struct nlmsghdr *n, |
15e47304 | 851 | u32 portid, int event) |
1da177e4 | 852 | { |
cee63723 | 853 | int i, ret; |
cc7ec456 | 854 | struct nlattr *tb[TCA_ACT_MAX_PRIO + 1]; |
1da177e4 LT |
855 | struct tc_action *head = NULL, *act, *act_prev = NULL; |
856 | ||
cee63723 PM |
857 | ret = nla_parse_nested(tb, TCA_ACT_MAX_PRIO, nla, NULL); |
858 | if (ret < 0) | |
859 | return ret; | |
1da177e4 | 860 | |
cc7ec456 | 861 | if (event == RTM_DELACTION && n->nlmsg_flags & NLM_F_ROOT) { |
f97017cd | 862 | if (tb[1] != NULL) |
15e47304 | 863 | return tca_action_flush(net, tb[1], n, portid); |
f97017cd JHS |
864 | else |
865 | return -EINVAL; | |
1da177e4 LT |
866 | } |
867 | ||
7ba699c6 | 868 | for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) { |
15e47304 | 869 | act = tcf_action_get_1(tb[i], n, portid); |
ab27cfb8 PM |
870 | if (IS_ERR(act)) { |
871 | ret = PTR_ERR(act); | |
1da177e4 | 872 | goto err; |
ab27cfb8 | 873 | } |
7ba699c6 | 874 | act->order = i; |
1da177e4 LT |
875 | |
876 | if (head == NULL) | |
877 | head = act; | |
878 | else | |
879 | act_prev->next = act; | |
880 | act_prev = act; | |
881 | } | |
882 | ||
883 | if (event == RTM_GETACTION) | |
15e47304 | 884 | ret = act_get_notify(net, portid, n, head, event); |
1da177e4 LT |
885 | else { /* delete */ |
886 | struct sk_buff *skb; | |
887 | ||
888 | skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL); | |
889 | if (!skb) { | |
890 | ret = -ENOBUFS; | |
891 | goto err; | |
892 | } | |
893 | ||
15e47304 | 894 | if (tca_get_fill(skb, head, portid, n->nlmsg_seq, 0, event, |
10297b99 | 895 | 0, 1) <= 0) { |
1da177e4 LT |
896 | kfree_skb(skb); |
897 | ret = -EINVAL; | |
898 | goto err; | |
899 | } | |
900 | ||
901 | /* now do the delete */ | |
902 | tcf_action_destroy(head, 0); | |
15e47304 | 903 | ret = rtnetlink_send(skb, net, portid, RTNLGRP_TC, |
cc7ec456 | 904 | n->nlmsg_flags & NLM_F_ECHO); |
1da177e4 LT |
905 | if (ret > 0) |
906 | return 0; | |
907 | return ret; | |
908 | } | |
909 | err: | |
910 | cleanup_a(head); | |
911 | return ret; | |
912 | } | |
913 | ||
7316ae88 | 914 | static int tcf_add_notify(struct net *net, struct tc_action *a, |
15e47304 | 915 | u32 portid, u32 seq, int event, u16 flags) |
1da177e4 LT |
916 | { |
917 | struct tcamsg *t; | |
918 | struct nlmsghdr *nlh; | |
919 | struct sk_buff *skb; | |
4b3550ef | 920 | struct nlattr *nest; |
1da177e4 LT |
921 | unsigned char *b; |
922 | int err = 0; | |
923 | ||
924 | skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL); | |
925 | if (!skb) | |
926 | return -ENOBUFS; | |
927 | ||
27a884dc | 928 | b = skb_tail_pointer(skb); |
1da177e4 | 929 | |
15e47304 | 930 | nlh = nlmsg_put(skb, portid, seq, event, sizeof(*t), flags); |
8b00a53c DM |
931 | if (!nlh) |
932 | goto out_kfree_skb; | |
933 | t = nlmsg_data(nlh); | |
1da177e4 | 934 | t->tca_family = AF_UNSPEC; |
9ef1d4c7 PM |
935 | t->tca__pad1 = 0; |
936 | t->tca__pad2 = 0; | |
937 | ||
4b3550ef PM |
938 | nest = nla_nest_start(skb, TCA_ACT_TAB); |
939 | if (nest == NULL) | |
8b00a53c | 940 | goto out_kfree_skb; |
1da177e4 LT |
941 | |
942 | if (tcf_action_dump(skb, a, 0, 0) < 0) | |
8b00a53c | 943 | goto out_kfree_skb; |
1da177e4 | 944 | |
4b3550ef | 945 | nla_nest_end(skb, nest); |
10297b99 | 946 | |
27a884dc | 947 | nlh->nlmsg_len = skb_tail_pointer(skb) - b; |
ac6d439d | 948 | NETLINK_CB(skb).dst_group = RTNLGRP_TC; |
10297b99 | 949 | |
15e47304 | 950 | err = rtnetlink_send(skb, net, portid, RTNLGRP_TC, flags & NLM_F_ECHO); |
1da177e4 LT |
951 | if (err > 0) |
952 | err = 0; | |
953 | return err; | |
954 | ||
8b00a53c | 955 | out_kfree_skb: |
f6e57464 | 956 | kfree_skb(skb); |
1da177e4 LT |
957 | return -1; |
958 | } | |
959 | ||
10297b99 | 960 | |
1da177e4 | 961 | static int |
7316ae88 | 962 | tcf_action_add(struct net *net, struct nlattr *nla, struct nlmsghdr *n, |
15e47304 | 963 | u32 portid, int ovr) |
1da177e4 LT |
964 | { |
965 | int ret = 0; | |
966 | struct tc_action *act; | |
967 | struct tc_action *a; | |
968 | u32 seq = n->nlmsg_seq; | |
969 | ||
c1b52739 | 970 | act = tcf_action_init(net, nla, NULL, NULL, ovr, 0); |
1da177e4 LT |
971 | if (act == NULL) |
972 | goto done; | |
ab27cfb8 PM |
973 | if (IS_ERR(act)) { |
974 | ret = PTR_ERR(act); | |
975 | goto done; | |
976 | } | |
1da177e4 LT |
977 | |
978 | /* dump then free all the actions after update; inserted policy | |
979 | * stays intact | |
cc7ec456 | 980 | */ |
15e47304 | 981 | ret = tcf_add_notify(net, act, portid, seq, RTM_NEWACTION, n->nlmsg_flags); |
1da177e4 LT |
982 | for (a = act; a; a = act) { |
983 | act = a->next; | |
984 | kfree(a); | |
985 | } | |
986 | done: | |
987 | return ret; | |
988 | } | |
989 | ||
661d2967 | 990 | static int tc_ctl_action(struct sk_buff *skb, struct nlmsghdr *n) |
1da177e4 | 991 | { |
3b1e0a65 | 992 | struct net *net = sock_net(skb->sk); |
7ba699c6 | 993 | struct nlattr *tca[TCA_ACT_MAX + 1]; |
15e47304 | 994 | u32 portid = skb ? NETLINK_CB(skb).portid : 0; |
1da177e4 LT |
995 | int ret = 0, ovr = 0; |
996 | ||
dfc47ef8 EB |
997 | if ((n->nlmsg_type != RTM_GETACTION) && !capable(CAP_NET_ADMIN)) |
998 | return -EPERM; | |
999 | ||
7ba699c6 PM |
1000 | ret = nlmsg_parse(n, sizeof(struct tcamsg), tca, TCA_ACT_MAX, NULL); |
1001 | if (ret < 0) | |
1002 | return ret; | |
1003 | ||
1004 | if (tca[TCA_ACT_TAB] == NULL) { | |
6ff9c364 | 1005 | pr_notice("tc_ctl_action: received NO action attribs\n"); |
1da177e4 LT |
1006 | return -EINVAL; |
1007 | } | |
1008 | ||
cc7ec456 | 1009 | /* n->nlmsg_flags & NLM_F_CREATE */ |
1da177e4 LT |
1010 | switch (n->nlmsg_type) { |
1011 | case RTM_NEWACTION: | |
1012 | /* we are going to assume all other flags | |
25985edc | 1013 | * imply create only if it doesn't exist |
1da177e4 LT |
1014 | * Note that CREATE | EXCL implies that |
1015 | * but since we want avoid ambiguity (eg when flags | |
1016 | * is zero) then just set this | |
1017 | */ | |
cc7ec456 | 1018 | if (n->nlmsg_flags & NLM_F_REPLACE) |
1da177e4 LT |
1019 | ovr = 1; |
1020 | replay: | |
15e47304 | 1021 | ret = tcf_action_add(net, tca[TCA_ACT_TAB], n, portid, ovr); |
1da177e4 LT |
1022 | if (ret == -EAGAIN) |
1023 | goto replay; | |
1024 | break; | |
1025 | case RTM_DELACTION: | |
7316ae88 | 1026 | ret = tca_action_gd(net, tca[TCA_ACT_TAB], n, |
15e47304 | 1027 | portid, RTM_DELACTION); |
1da177e4 LT |
1028 | break; |
1029 | case RTM_GETACTION: | |
7316ae88 | 1030 | ret = tca_action_gd(net, tca[TCA_ACT_TAB], n, |
15e47304 | 1031 | portid, RTM_GETACTION); |
1da177e4 LT |
1032 | break; |
1033 | default: | |
1034 | BUG(); | |
1035 | } | |
1036 | ||
1037 | return ret; | |
1038 | } | |
1039 | ||
7ba699c6 | 1040 | static struct nlattr * |
3a6c2b41 | 1041 | find_dump_kind(const struct nlmsghdr *n) |
1da177e4 | 1042 | { |
cc7ec456 | 1043 | struct nlattr *tb1, *tb2[TCA_ACT_MAX + 1]; |
7ba699c6 PM |
1044 | struct nlattr *tb[TCA_ACT_MAX_PRIO + 1]; |
1045 | struct nlattr *nla[TCAA_MAX + 1]; | |
1046 | struct nlattr *kind; | |
1da177e4 | 1047 | |
c96c9471 | 1048 | if (nlmsg_parse(n, sizeof(struct tcamsg), nla, TCAA_MAX, NULL) < 0) |
1da177e4 | 1049 | return NULL; |
7ba699c6 | 1050 | tb1 = nla[TCA_ACT_TAB]; |
1da177e4 LT |
1051 | if (tb1 == NULL) |
1052 | return NULL; | |
1053 | ||
7ba699c6 PM |
1054 | if (nla_parse(tb, TCA_ACT_MAX_PRIO, nla_data(tb1), |
1055 | NLMSG_ALIGN(nla_len(tb1)), NULL) < 0) | |
1da177e4 | 1056 | return NULL; |
1da177e4 | 1057 | |
6d834e04 PM |
1058 | if (tb[1] == NULL) |
1059 | return NULL; | |
1060 | if (nla_parse(tb2, TCA_ACT_MAX, nla_data(tb[1]), | |
1061 | nla_len(tb[1]), NULL) < 0) | |
1da177e4 | 1062 | return NULL; |
7ba699c6 | 1063 | kind = tb2[TCA_ACT_KIND]; |
1da177e4 | 1064 | |
26dab893 | 1065 | return kind; |
1da177e4 LT |
1066 | } |
1067 | ||
1068 | static int | |
1069 | tc_dump_action(struct sk_buff *skb, struct netlink_callback *cb) | |
1070 | { | |
1071 | struct nlmsghdr *nlh; | |
27a884dc | 1072 | unsigned char *b = skb_tail_pointer(skb); |
4b3550ef | 1073 | struct nlattr *nest; |
1da177e4 LT |
1074 | struct tc_action_ops *a_o; |
1075 | struct tc_action a; | |
1076 | int ret = 0; | |
8b00a53c | 1077 | struct tcamsg *t = (struct tcamsg *) nlmsg_data(cb->nlh); |
7ba699c6 | 1078 | struct nlattr *kind = find_dump_kind(cb->nlh); |
1da177e4 LT |
1079 | |
1080 | if (kind == NULL) { | |
6ff9c364 | 1081 | pr_info("tc_dump_action: action bad kind\n"); |
1da177e4 LT |
1082 | return 0; |
1083 | } | |
1084 | ||
26dab893 | 1085 | a_o = tc_lookup_action(kind); |
cc7ec456 | 1086 | if (a_o == NULL) |
1da177e4 | 1087 | return 0; |
1da177e4 LT |
1088 | |
1089 | memset(&a, 0, sizeof(struct tc_action)); | |
1090 | a.ops = a_o; | |
1091 | ||
1092 | if (a_o->walk == NULL) { | |
6ff9c364 | 1093 | WARN(1, "tc_dump_action: %s !capable of dumping table\n", |
1094 | a_o->kind); | |
8b00a53c | 1095 | goto out_module_put; |
1da177e4 LT |
1096 | } |
1097 | ||
15e47304 | 1098 | nlh = nlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq, |
8b00a53c DM |
1099 | cb->nlh->nlmsg_type, sizeof(*t), 0); |
1100 | if (!nlh) | |
1101 | goto out_module_put; | |
1102 | t = nlmsg_data(nlh); | |
1da177e4 | 1103 | t->tca_family = AF_UNSPEC; |
9ef1d4c7 PM |
1104 | t->tca__pad1 = 0; |
1105 | t->tca__pad2 = 0; | |
1da177e4 | 1106 | |
4b3550ef PM |
1107 | nest = nla_nest_start(skb, TCA_ACT_TAB); |
1108 | if (nest == NULL) | |
8b00a53c | 1109 | goto out_module_put; |
1da177e4 LT |
1110 | |
1111 | ret = a_o->walk(skb, cb, RTM_GETACTION, &a); | |
1112 | if (ret < 0) | |
8b00a53c | 1113 | goto out_module_put; |
1da177e4 LT |
1114 | |
1115 | if (ret > 0) { | |
4b3550ef | 1116 | nla_nest_end(skb, nest); |
1da177e4 LT |
1117 | ret = skb->len; |
1118 | } else | |
4b3550ef | 1119 | nla_nest_cancel(skb, nest); |
1da177e4 | 1120 | |
27a884dc | 1121 | nlh->nlmsg_len = skb_tail_pointer(skb) - b; |
15e47304 | 1122 | if (NETLINK_CB(cb->skb).portid && ret) |
1da177e4 LT |
1123 | nlh->nlmsg_flags |= NLM_F_MULTI; |
1124 | module_put(a_o->owner); | |
1125 | return skb->len; | |
1126 | ||
8b00a53c | 1127 | out_module_put: |
1da177e4 | 1128 | module_put(a_o->owner); |
dc5fc579 | 1129 | nlmsg_trim(skb, b); |
1da177e4 LT |
1130 | return skb->len; |
1131 | } | |
1132 | ||
1133 | static int __init tc_action_init(void) | |
1134 | { | |
c7ac8679 GR |
1135 | rtnl_register(PF_UNSPEC, RTM_NEWACTION, tc_ctl_action, NULL, NULL); |
1136 | rtnl_register(PF_UNSPEC, RTM_DELACTION, tc_ctl_action, NULL, NULL); | |
1137 | rtnl_register(PF_UNSPEC, RTM_GETACTION, tc_ctl_action, tc_dump_action, | |
1138 | NULL); | |
1da177e4 | 1139 | |
1da177e4 LT |
1140 | return 0; |
1141 | } | |
1142 | ||
1143 | subsys_initcall(tc_action_init); |