| xj | b04a402 | 2021-11-25 15:01:52 +0800 | [diff] [blame] | 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 |  | 
|  | 14 | #include <linux/types.h> | 
|  | 15 | #include <linux/kernel.h> | 
|  | 16 | #include <linux/string.h> | 
|  | 17 | #include <linux/errno.h> | 
|  | 18 | #include <linux/slab.h> | 
|  | 19 | #include <linux/skbuff.h> | 
|  | 20 | #include <linux/init.h> | 
|  | 21 | #include <linux/kmod.h> | 
|  | 22 | #include <linux/err.h> | 
|  | 23 | #include <linux/module.h> | 
|  | 24 | #include <linux/rhashtable.h> | 
|  | 25 | #include <linux/list.h> | 
|  | 26 | #include <net/net_namespace.h> | 
|  | 27 | #include <net/sock.h> | 
|  | 28 | #include <net/sch_generic.h> | 
|  | 29 | #include <net/pkt_cls.h> | 
|  | 30 | #include <net/act_api.h> | 
|  | 31 | #include <net/netlink.h> | 
|  | 32 |  | 
|  | 33 | static int tcf_action_goto_chain_init(struct tc_action *a, struct tcf_proto *tp) | 
|  | 34 | { | 
|  | 35 | u32 chain_index = a->tcfa_action & TC_ACT_EXT_VAL_MASK; | 
|  | 36 |  | 
|  | 37 | if (!tp) | 
|  | 38 | return -EINVAL; | 
|  | 39 | a->goto_chain = tcf_chain_get_by_act(tp->chain->block, chain_index); | 
|  | 40 | if (!a->goto_chain) | 
|  | 41 | return -ENOMEM; | 
|  | 42 | return 0; | 
|  | 43 | } | 
|  | 44 |  | 
|  | 45 | static void tcf_action_goto_chain_fini(struct tc_action *a) | 
|  | 46 | { | 
|  | 47 | tcf_chain_put_by_act(a->goto_chain); | 
|  | 48 | } | 
|  | 49 |  | 
|  | 50 | static void tcf_action_goto_chain_exec(const struct tc_action *a, | 
|  | 51 | struct tcf_result *res) | 
|  | 52 | { | 
|  | 53 | const struct tcf_chain *chain = a->goto_chain; | 
|  | 54 |  | 
|  | 55 | res->goto_tp = rcu_dereference_bh(chain->filter_chain); | 
|  | 56 | } | 
|  | 57 |  | 
|  | 58 | static void tcf_free_cookie_rcu(struct rcu_head *p) | 
|  | 59 | { | 
|  | 60 | struct tc_cookie *cookie = container_of(p, struct tc_cookie, rcu); | 
|  | 61 |  | 
|  | 62 | kfree(cookie->data); | 
|  | 63 | kfree(cookie); | 
|  | 64 | } | 
|  | 65 |  | 
|  | 66 | static void tcf_set_action_cookie(struct tc_cookie __rcu **old_cookie, | 
|  | 67 | struct tc_cookie *new_cookie) | 
|  | 68 | { | 
|  | 69 | struct tc_cookie *old; | 
|  | 70 |  | 
|  | 71 | old = xchg((__force struct tc_cookie **)old_cookie, new_cookie); | 
|  | 72 | if (old) | 
|  | 73 | call_rcu(&old->rcu, tcf_free_cookie_rcu); | 
|  | 74 | } | 
|  | 75 |  | 
|  | 76 | /* XXX: For standalone actions, we don't need a RCU grace period either, because | 
|  | 77 | * actions are always connected to filters and filters are already destroyed in | 
|  | 78 | * RCU callbacks, so after a RCU grace period actions are already disconnected | 
|  | 79 | * from filters. Readers later can not find us. | 
|  | 80 | */ | 
|  | 81 | static void free_tcf(struct tc_action *p) | 
|  | 82 | { | 
|  | 83 | free_percpu(p->cpu_bstats); | 
|  | 84 | free_percpu(p->cpu_qstats); | 
|  | 85 |  | 
|  | 86 | tcf_set_action_cookie(&p->act_cookie, NULL); | 
|  | 87 | if (p->goto_chain) | 
|  | 88 | tcf_action_goto_chain_fini(p); | 
|  | 89 |  | 
|  | 90 | kfree(p); | 
|  | 91 | } | 
|  | 92 |  | 
|  | 93 | static void tcf_action_cleanup(struct tc_action *p) | 
|  | 94 | { | 
|  | 95 | if (p->ops->cleanup) | 
|  | 96 | p->ops->cleanup(p); | 
|  | 97 |  | 
|  | 98 | gen_kill_estimator(&p->tcfa_rate_est); | 
|  | 99 | free_tcf(p); | 
|  | 100 | } | 
|  | 101 |  | 
|  | 102 | static int __tcf_action_put(struct tc_action *p, bool bind) | 
|  | 103 | { | 
|  | 104 | struct tcf_idrinfo *idrinfo = p->idrinfo; | 
|  | 105 |  | 
|  | 106 | if (refcount_dec_and_lock(&p->tcfa_refcnt, &idrinfo->lock)) { | 
|  | 107 | if (bind) | 
|  | 108 | atomic_dec(&p->tcfa_bindcnt); | 
|  | 109 | idr_remove(&idrinfo->action_idr, p->tcfa_index); | 
|  | 110 | spin_unlock(&idrinfo->lock); | 
|  | 111 |  | 
|  | 112 | tcf_action_cleanup(p); | 
|  | 113 | return 1; | 
|  | 114 | } | 
|  | 115 |  | 
|  | 116 | if (bind) | 
|  | 117 | atomic_dec(&p->tcfa_bindcnt); | 
|  | 118 |  | 
|  | 119 | return 0; | 
|  | 120 | } | 
|  | 121 |  | 
|  | 122 | int __tcf_idr_release(struct tc_action *p, bool bind, bool strict) | 
|  | 123 | { | 
|  | 124 | int ret = 0; | 
|  | 125 |  | 
|  | 126 | /* Release with strict==1 and bind==0 is only called through act API | 
|  | 127 | * interface (classifiers always bind). Only case when action with | 
|  | 128 | * positive reference count and zero bind count can exist is when it was | 
|  | 129 | * also created with act API (unbinding last classifier will destroy the | 
|  | 130 | * action if it was created by classifier). So only case when bind count | 
|  | 131 | * can be changed after initial check is when unbound action is | 
|  | 132 | * destroyed by act API while classifier binds to action with same id | 
|  | 133 | * concurrently. This result either creation of new action(same behavior | 
|  | 134 | * as before), or reusing existing action if concurrent process | 
|  | 135 | * increments reference count before action is deleted. Both scenarios | 
|  | 136 | * are acceptable. | 
|  | 137 | */ | 
|  | 138 | if (p) { | 
|  | 139 | if (!bind && strict && atomic_read(&p->tcfa_bindcnt) > 0) | 
|  | 140 | return -EPERM; | 
|  | 141 |  | 
|  | 142 | if (__tcf_action_put(p, bind)) | 
|  | 143 | ret = ACT_P_DELETED; | 
|  | 144 | } | 
|  | 145 |  | 
|  | 146 | return ret; | 
|  | 147 | } | 
|  | 148 | EXPORT_SYMBOL(__tcf_idr_release); | 
|  | 149 |  | 
|  | 150 | static size_t tcf_action_shared_attrs_size(const struct tc_action *act) | 
|  | 151 | { | 
|  | 152 | struct tc_cookie *act_cookie; | 
|  | 153 | u32 cookie_len = 0; | 
|  | 154 |  | 
|  | 155 | rcu_read_lock(); | 
|  | 156 | act_cookie = rcu_dereference(act->act_cookie); | 
|  | 157 |  | 
|  | 158 | if (act_cookie) | 
|  | 159 | cookie_len = nla_total_size(act_cookie->len); | 
|  | 160 | rcu_read_unlock(); | 
|  | 161 |  | 
|  | 162 | return  nla_total_size(0) /* action number nested */ | 
|  | 163 | + nla_total_size(IFNAMSIZ) /* TCA_ACT_KIND */ | 
|  | 164 | + cookie_len /* TCA_ACT_COOKIE */ | 
|  | 165 | + nla_total_size(0) /* TCA_ACT_STATS nested */ | 
|  | 166 | /* TCA_STATS_BASIC */ | 
|  | 167 | + nla_total_size_64bit(sizeof(struct gnet_stats_basic)) | 
|  | 168 | /* TCA_STATS_QUEUE */ | 
|  | 169 | + nla_total_size_64bit(sizeof(struct gnet_stats_queue)) | 
|  | 170 | + nla_total_size(0) /* TCA_OPTIONS nested */ | 
|  | 171 | + nla_total_size(sizeof(struct tcf_t)); /* TCA_GACT_TM */ | 
|  | 172 | } | 
|  | 173 |  | 
|  | 174 | static size_t tcf_action_full_attrs_size(size_t sz) | 
|  | 175 | { | 
|  | 176 | return NLMSG_HDRLEN                     /* struct nlmsghdr */ | 
|  | 177 | + sizeof(struct tcamsg) | 
|  | 178 | + nla_total_size(0)             /* TCA_ACT_TAB nested */ | 
|  | 179 | + sz; | 
|  | 180 | } | 
|  | 181 |  | 
|  | 182 | static size_t tcf_action_fill_size(const struct tc_action *act) | 
|  | 183 | { | 
|  | 184 | size_t sz = tcf_action_shared_attrs_size(act); | 
|  | 185 |  | 
|  | 186 | if (act->ops->get_fill_size) | 
|  | 187 | return act->ops->get_fill_size(act) + sz; | 
|  | 188 | return sz; | 
|  | 189 | } | 
|  | 190 |  | 
|  | 191 | static int tcf_dump_walker(struct tcf_idrinfo *idrinfo, struct sk_buff *skb, | 
|  | 192 | struct netlink_callback *cb) | 
|  | 193 | { | 
|  | 194 | int err = 0, index = -1, s_i = 0, n_i = 0; | 
|  | 195 | u32 act_flags = cb->args[2]; | 
|  | 196 | unsigned long jiffy_since = cb->args[3]; | 
|  | 197 | struct nlattr *nest; | 
|  | 198 | struct idr *idr = &idrinfo->action_idr; | 
|  | 199 | struct tc_action *p; | 
|  | 200 | unsigned long id = 1; | 
|  | 201 |  | 
|  | 202 | spin_lock(&idrinfo->lock); | 
|  | 203 |  | 
|  | 204 | s_i = cb->args[0]; | 
|  | 205 |  | 
|  | 206 | idr_for_each_entry_ul(idr, p, id) { | 
|  | 207 | index++; | 
|  | 208 | if (index < s_i) | 
|  | 209 | continue; | 
|  | 210 |  | 
|  | 211 | if (jiffy_since && | 
|  | 212 | time_after(jiffy_since, | 
|  | 213 | (unsigned long)p->tcfa_tm.lastuse)) | 
|  | 214 | continue; | 
|  | 215 |  | 
|  | 216 | nest = nla_nest_start(skb, n_i); | 
|  | 217 | if (!nest) { | 
|  | 218 | index--; | 
|  | 219 | goto nla_put_failure; | 
|  | 220 | } | 
|  | 221 | err = tcf_action_dump_1(skb, p, 0, 0); | 
|  | 222 | if (err < 0) { | 
|  | 223 | index--; | 
|  | 224 | nlmsg_trim(skb, nest); | 
|  | 225 | goto done; | 
|  | 226 | } | 
|  | 227 | nla_nest_end(skb, nest); | 
|  | 228 | n_i++; | 
|  | 229 | if (!(act_flags & TCA_FLAG_LARGE_DUMP_ON) && | 
|  | 230 | n_i >= TCA_ACT_MAX_PRIO) | 
|  | 231 | goto done; | 
|  | 232 | } | 
|  | 233 | done: | 
|  | 234 | if (index >= 0) | 
|  | 235 | cb->args[0] = index + 1; | 
|  | 236 |  | 
|  | 237 | spin_unlock(&idrinfo->lock); | 
|  | 238 | if (n_i) { | 
|  | 239 | if (act_flags & TCA_FLAG_LARGE_DUMP_ON) | 
|  | 240 | cb->args[1] = n_i; | 
|  | 241 | } | 
|  | 242 | return n_i; | 
|  | 243 |  | 
|  | 244 | nla_put_failure: | 
|  | 245 | nla_nest_cancel(skb, nest); | 
|  | 246 | goto done; | 
|  | 247 | } | 
|  | 248 |  | 
|  | 249 | static int tcf_del_walker(struct tcf_idrinfo *idrinfo, struct sk_buff *skb, | 
|  | 250 | const struct tc_action_ops *ops) | 
|  | 251 | { | 
|  | 252 | struct nlattr *nest; | 
|  | 253 | int n_i = 0; | 
|  | 254 | int ret = -EINVAL; | 
|  | 255 | struct idr *idr = &idrinfo->action_idr; | 
|  | 256 | struct tc_action *p; | 
|  | 257 | unsigned long id = 1; | 
|  | 258 |  | 
|  | 259 | nest = nla_nest_start(skb, 0); | 
|  | 260 | if (nest == NULL) | 
|  | 261 | goto nla_put_failure; | 
|  | 262 | if (nla_put_string(skb, TCA_KIND, ops->kind)) | 
|  | 263 | goto nla_put_failure; | 
|  | 264 |  | 
|  | 265 | idr_for_each_entry_ul(idr, p, id) { | 
|  | 266 | ret = __tcf_idr_release(p, false, true); | 
|  | 267 | if (ret == ACT_P_DELETED) { | 
|  | 268 | module_put(ops->owner); | 
|  | 269 | n_i++; | 
|  | 270 | } else if (ret < 0) { | 
|  | 271 | goto nla_put_failure; | 
|  | 272 | } | 
|  | 273 | } | 
|  | 274 | if (nla_put_u32(skb, TCA_FCNT, n_i)) | 
|  | 275 | goto nla_put_failure; | 
|  | 276 | nla_nest_end(skb, nest); | 
|  | 277 |  | 
|  | 278 | return n_i; | 
|  | 279 | nla_put_failure: | 
|  | 280 | nla_nest_cancel(skb, nest); | 
|  | 281 | return ret; | 
|  | 282 | } | 
|  | 283 |  | 
|  | 284 | int tcf_generic_walker(struct tc_action_net *tn, struct sk_buff *skb, | 
|  | 285 | struct netlink_callback *cb, int type, | 
|  | 286 | const struct tc_action_ops *ops, | 
|  | 287 | struct netlink_ext_ack *extack) | 
|  | 288 | { | 
|  | 289 | struct tcf_idrinfo *idrinfo = tn->idrinfo; | 
|  | 290 |  | 
|  | 291 | if (type == RTM_DELACTION) { | 
|  | 292 | return tcf_del_walker(idrinfo, skb, ops); | 
|  | 293 | } else if (type == RTM_GETACTION) { | 
|  | 294 | return tcf_dump_walker(idrinfo, skb, cb); | 
|  | 295 | } else { | 
|  | 296 | WARN(1, "tcf_generic_walker: unknown command %d\n", type); | 
|  | 297 | NL_SET_ERR_MSG(extack, "tcf_generic_walker: unknown command"); | 
|  | 298 | return -EINVAL; | 
|  | 299 | } | 
|  | 300 | } | 
|  | 301 | EXPORT_SYMBOL(tcf_generic_walker); | 
|  | 302 |  | 
|  | 303 | int tcf_idr_search(struct tc_action_net *tn, struct tc_action **a, u32 index) | 
|  | 304 | { | 
|  | 305 | struct tcf_idrinfo *idrinfo = tn->idrinfo; | 
|  | 306 | struct tc_action *p; | 
|  | 307 |  | 
|  | 308 | spin_lock(&idrinfo->lock); | 
|  | 309 | p = idr_find(&idrinfo->action_idr, index); | 
|  | 310 | if (IS_ERR(p)) | 
|  | 311 | p = NULL; | 
|  | 312 | else if (p) | 
|  | 313 | refcount_inc(&p->tcfa_refcnt); | 
|  | 314 | spin_unlock(&idrinfo->lock); | 
|  | 315 |  | 
|  | 316 | if (p) { | 
|  | 317 | *a = p; | 
|  | 318 | return true; | 
|  | 319 | } | 
|  | 320 | return false; | 
|  | 321 | } | 
|  | 322 | EXPORT_SYMBOL(tcf_idr_search); | 
|  | 323 |  | 
|  | 324 | static int tcf_idr_delete_index(struct tcf_idrinfo *idrinfo, u32 index) | 
|  | 325 | { | 
|  | 326 | struct tc_action *p; | 
|  | 327 | int ret = 0; | 
|  | 328 |  | 
|  | 329 | spin_lock(&idrinfo->lock); | 
|  | 330 | p = idr_find(&idrinfo->action_idr, index); | 
|  | 331 | if (!p) { | 
|  | 332 | spin_unlock(&idrinfo->lock); | 
|  | 333 | return -ENOENT; | 
|  | 334 | } | 
|  | 335 |  | 
|  | 336 | if (!atomic_read(&p->tcfa_bindcnt)) { | 
|  | 337 | if (refcount_dec_and_test(&p->tcfa_refcnt)) { | 
|  | 338 | struct module *owner = p->ops->owner; | 
|  | 339 |  | 
|  | 340 | WARN_ON(p != idr_remove(&idrinfo->action_idr, | 
|  | 341 | p->tcfa_index)); | 
|  | 342 | spin_unlock(&idrinfo->lock); | 
|  | 343 |  | 
|  | 344 | tcf_action_cleanup(p); | 
|  | 345 | module_put(owner); | 
|  | 346 | return 0; | 
|  | 347 | } | 
|  | 348 | ret = 0; | 
|  | 349 | } else { | 
|  | 350 | ret = -EPERM; | 
|  | 351 | } | 
|  | 352 |  | 
|  | 353 | spin_unlock(&idrinfo->lock); | 
|  | 354 | return ret; | 
|  | 355 | } | 
|  | 356 |  | 
|  | 357 | int tcf_idr_create(struct tc_action_net *tn, u32 index, struct nlattr *est, | 
|  | 358 | struct tc_action **a, const struct tc_action_ops *ops, | 
|  | 359 | int bind, bool cpustats) | 
|  | 360 | { | 
|  | 361 | struct tc_action *p = kzalloc(ops->size, GFP_KERNEL); | 
|  | 362 | struct tcf_idrinfo *idrinfo = tn->idrinfo; | 
|  | 363 | int err = -ENOMEM; | 
|  | 364 |  | 
|  | 365 | if (unlikely(!p)) | 
|  | 366 | return -ENOMEM; | 
|  | 367 | refcount_set(&p->tcfa_refcnt, 1); | 
|  | 368 | if (bind) | 
|  | 369 | atomic_set(&p->tcfa_bindcnt, 1); | 
|  | 370 |  | 
|  | 371 | if (cpustats) { | 
|  | 372 | p->cpu_bstats = netdev_alloc_pcpu_stats(struct gnet_stats_basic_cpu); | 
|  | 373 | if (!p->cpu_bstats) | 
|  | 374 | goto err1; | 
|  | 375 | p->cpu_qstats = alloc_percpu(struct gnet_stats_queue); | 
|  | 376 | if (!p->cpu_qstats) | 
|  | 377 | goto err2; | 
|  | 378 | } | 
|  | 379 | spin_lock_init(&p->tcfa_lock); | 
|  | 380 | p->tcfa_index = index; | 
|  | 381 | p->tcfa_tm.install = jiffies; | 
|  | 382 | p->tcfa_tm.lastuse = jiffies; | 
|  | 383 | p->tcfa_tm.firstuse = 0; | 
|  | 384 | if (est) { | 
|  | 385 | err = gen_new_estimator(&p->tcfa_bstats, p->cpu_bstats, | 
|  | 386 | &p->tcfa_rate_est, | 
|  | 387 | &p->tcfa_lock, NULL, est); | 
|  | 388 | if (err) | 
|  | 389 | goto err3; | 
|  | 390 | } | 
|  | 391 |  | 
|  | 392 | p->idrinfo = idrinfo; | 
|  | 393 | p->ops = ops; | 
|  | 394 | *a = p; | 
|  | 395 | return 0; | 
|  | 396 | err3: | 
|  | 397 | free_percpu(p->cpu_qstats); | 
|  | 398 | err2: | 
|  | 399 | free_percpu(p->cpu_bstats); | 
|  | 400 | err1: | 
|  | 401 | kfree(p); | 
|  | 402 | return err; | 
|  | 403 | } | 
|  | 404 | EXPORT_SYMBOL(tcf_idr_create); | 
|  | 405 |  | 
|  | 406 | void tcf_idr_insert(struct tc_action_net *tn, struct tc_action *a) | 
|  | 407 | { | 
|  | 408 | struct tcf_idrinfo *idrinfo = tn->idrinfo; | 
|  | 409 |  | 
|  | 410 | spin_lock(&idrinfo->lock); | 
|  | 411 | /* Replace ERR_PTR(-EBUSY) allocated by tcf_idr_check_alloc */ | 
|  | 412 | WARN_ON(!IS_ERR(idr_replace(&idrinfo->action_idr, a, a->tcfa_index))); | 
|  | 413 | spin_unlock(&idrinfo->lock); | 
|  | 414 | } | 
|  | 415 | EXPORT_SYMBOL(tcf_idr_insert); | 
|  | 416 |  | 
|  | 417 | /* Cleanup idr index that was allocated but not initialized. */ | 
|  | 418 |  | 
|  | 419 | void tcf_idr_cleanup(struct tc_action_net *tn, u32 index) | 
|  | 420 | { | 
|  | 421 | struct tcf_idrinfo *idrinfo = tn->idrinfo; | 
|  | 422 |  | 
|  | 423 | spin_lock(&idrinfo->lock); | 
|  | 424 | /* Remove ERR_PTR(-EBUSY) allocated by tcf_idr_check_alloc */ | 
|  | 425 | WARN_ON(!IS_ERR(idr_remove(&idrinfo->action_idr, index))); | 
|  | 426 | spin_unlock(&idrinfo->lock); | 
|  | 427 | } | 
|  | 428 | EXPORT_SYMBOL(tcf_idr_cleanup); | 
|  | 429 |  | 
|  | 430 | /* Check if action with specified index exists. If actions is found, increments | 
|  | 431 | * its reference and bind counters, and return 1. Otherwise insert temporary | 
|  | 432 | * error pointer (to prevent concurrent users from inserting actions with same | 
|  | 433 | * index) and return 0. | 
|  | 434 | */ | 
|  | 435 |  | 
|  | 436 | int tcf_idr_check_alloc(struct tc_action_net *tn, u32 *index, | 
|  | 437 | struct tc_action **a, int bind) | 
|  | 438 | { | 
|  | 439 | struct tcf_idrinfo *idrinfo = tn->idrinfo; | 
|  | 440 | struct tc_action *p; | 
|  | 441 | int ret; | 
|  | 442 |  | 
|  | 443 | again: | 
|  | 444 | spin_lock(&idrinfo->lock); | 
|  | 445 | if (*index) { | 
|  | 446 | p = idr_find(&idrinfo->action_idr, *index); | 
|  | 447 | if (IS_ERR(p)) { | 
|  | 448 | /* This means that another process allocated | 
|  | 449 | * index but did not assign the pointer yet. | 
|  | 450 | */ | 
|  | 451 | spin_unlock(&idrinfo->lock); | 
|  | 452 | goto again; | 
|  | 453 | } | 
|  | 454 |  | 
|  | 455 | if (p) { | 
|  | 456 | refcount_inc(&p->tcfa_refcnt); | 
|  | 457 | if (bind) | 
|  | 458 | atomic_inc(&p->tcfa_bindcnt); | 
|  | 459 | *a = p; | 
|  | 460 | ret = 1; | 
|  | 461 | } else { | 
|  | 462 | *a = NULL; | 
|  | 463 | ret = idr_alloc_u32(&idrinfo->action_idr, NULL, index, | 
|  | 464 | *index, GFP_ATOMIC); | 
|  | 465 | if (!ret) | 
|  | 466 | idr_replace(&idrinfo->action_idr, | 
|  | 467 | ERR_PTR(-EBUSY), *index); | 
|  | 468 | } | 
|  | 469 | } else { | 
|  | 470 | *index = 1; | 
|  | 471 | *a = NULL; | 
|  | 472 | ret = idr_alloc_u32(&idrinfo->action_idr, NULL, index, | 
|  | 473 | UINT_MAX, GFP_ATOMIC); | 
|  | 474 | if (!ret) | 
|  | 475 | idr_replace(&idrinfo->action_idr, ERR_PTR(-EBUSY), | 
|  | 476 | *index); | 
|  | 477 | } | 
|  | 478 | spin_unlock(&idrinfo->lock); | 
|  | 479 | return ret; | 
|  | 480 | } | 
|  | 481 | EXPORT_SYMBOL(tcf_idr_check_alloc); | 
|  | 482 |  | 
|  | 483 | void tcf_idrinfo_destroy(const struct tc_action_ops *ops, | 
|  | 484 | struct tcf_idrinfo *idrinfo) | 
|  | 485 | { | 
|  | 486 | struct idr *idr = &idrinfo->action_idr; | 
|  | 487 | struct tc_action *p; | 
|  | 488 | int ret; | 
|  | 489 | unsigned long id = 1; | 
|  | 490 |  | 
|  | 491 | idr_for_each_entry_ul(idr, p, id) { | 
|  | 492 | ret = __tcf_idr_release(p, false, true); | 
|  | 493 | if (ret == ACT_P_DELETED) | 
|  | 494 | module_put(ops->owner); | 
|  | 495 | else if (ret < 0) | 
|  | 496 | return; | 
|  | 497 | } | 
|  | 498 | idr_destroy(&idrinfo->action_idr); | 
|  | 499 | } | 
|  | 500 | EXPORT_SYMBOL(tcf_idrinfo_destroy); | 
|  | 501 |  | 
|  | 502 | static LIST_HEAD(act_base); | 
|  | 503 | static DEFINE_RWLOCK(act_mod_lock); | 
|  | 504 |  | 
|  | 505 | int tcf_register_action(struct tc_action_ops *act, | 
|  | 506 | struct pernet_operations *ops) | 
|  | 507 | { | 
|  | 508 | struct tc_action_ops *a; | 
|  | 509 | int ret; | 
|  | 510 |  | 
|  | 511 | if (!act->act || !act->dump || !act->init || !act->walk || !act->lookup) | 
|  | 512 | return -EINVAL; | 
|  | 513 |  | 
|  | 514 | /* We have to register pernet ops before making the action ops visible, | 
|  | 515 | * otherwise tcf_action_init_1() could get a partially initialized | 
|  | 516 | * netns. | 
|  | 517 | */ | 
|  | 518 | ret = register_pernet_subsys(ops); | 
|  | 519 | if (ret) | 
|  | 520 | return ret; | 
|  | 521 |  | 
|  | 522 | write_lock(&act_mod_lock); | 
|  | 523 | list_for_each_entry(a, &act_base, head) { | 
|  | 524 | if (act->type == a->type || (strcmp(act->kind, a->kind) == 0)) { | 
|  | 525 | write_unlock(&act_mod_lock); | 
|  | 526 | unregister_pernet_subsys(ops); | 
|  | 527 | return -EEXIST; | 
|  | 528 | } | 
|  | 529 | } | 
|  | 530 | list_add_tail(&act->head, &act_base); | 
|  | 531 | write_unlock(&act_mod_lock); | 
|  | 532 |  | 
|  | 533 | return 0; | 
|  | 534 | } | 
|  | 535 | EXPORT_SYMBOL(tcf_register_action); | 
|  | 536 |  | 
|  | 537 | int tcf_unregister_action(struct tc_action_ops *act, | 
|  | 538 | struct pernet_operations *ops) | 
|  | 539 | { | 
|  | 540 | struct tc_action_ops *a; | 
|  | 541 | int err = -ENOENT; | 
|  | 542 |  | 
|  | 543 | write_lock(&act_mod_lock); | 
|  | 544 | list_for_each_entry(a, &act_base, head) { | 
|  | 545 | if (a == act) { | 
|  | 546 | list_del(&act->head); | 
|  | 547 | err = 0; | 
|  | 548 | break; | 
|  | 549 | } | 
|  | 550 | } | 
|  | 551 | write_unlock(&act_mod_lock); | 
|  | 552 | if (!err) | 
|  | 553 | unregister_pernet_subsys(ops); | 
|  | 554 | return err; | 
|  | 555 | } | 
|  | 556 | EXPORT_SYMBOL(tcf_unregister_action); | 
|  | 557 |  | 
|  | 558 | /* lookup by name */ | 
|  | 559 | static struct tc_action_ops *tc_lookup_action_n(char *kind) | 
|  | 560 | { | 
|  | 561 | struct tc_action_ops *a, *res = NULL; | 
|  | 562 |  | 
|  | 563 | if (kind) { | 
|  | 564 | read_lock(&act_mod_lock); | 
|  | 565 | list_for_each_entry(a, &act_base, head) { | 
|  | 566 | if (strcmp(kind, a->kind) == 0) { | 
|  | 567 | if (try_module_get(a->owner)) | 
|  | 568 | res = a; | 
|  | 569 | break; | 
|  | 570 | } | 
|  | 571 | } | 
|  | 572 | read_unlock(&act_mod_lock); | 
|  | 573 | } | 
|  | 574 | return res; | 
|  | 575 | } | 
|  | 576 |  | 
|  | 577 | /* lookup by nlattr */ | 
|  | 578 | static struct tc_action_ops *tc_lookup_action(struct nlattr *kind) | 
|  | 579 | { | 
|  | 580 | struct tc_action_ops *a, *res = NULL; | 
|  | 581 |  | 
|  | 582 | if (kind) { | 
|  | 583 | read_lock(&act_mod_lock); | 
|  | 584 | list_for_each_entry(a, &act_base, head) { | 
|  | 585 | if (nla_strcmp(kind, a->kind) == 0) { | 
|  | 586 | if (try_module_get(a->owner)) | 
|  | 587 | res = a; | 
|  | 588 | break; | 
|  | 589 | } | 
|  | 590 | } | 
|  | 591 | read_unlock(&act_mod_lock); | 
|  | 592 | } | 
|  | 593 | return res; | 
|  | 594 | } | 
|  | 595 |  | 
|  | 596 | /*TCA_ACT_MAX_PRIO is 32, there count upto 32 */ | 
|  | 597 | #define TCA_ACT_MAX_PRIO_MASK 0x1FF | 
|  | 598 | int tcf_action_exec(struct sk_buff *skb, struct tc_action **actions, | 
|  | 599 | int nr_actions, struct tcf_result *res) | 
|  | 600 | { | 
|  | 601 | u32 jmp_prgcnt = 0; | 
|  | 602 | u32 jmp_ttl = TCA_ACT_MAX_PRIO; /*matches actions per filter */ | 
|  | 603 | int i; | 
|  | 604 | int ret = TC_ACT_OK; | 
|  | 605 |  | 
|  | 606 | if (skb_skip_tc_classify(skb)) | 
|  | 607 | return TC_ACT_OK; | 
|  | 608 |  | 
|  | 609 | restart_act_graph: | 
|  | 610 | for (i = 0; i < nr_actions; i++) { | 
|  | 611 | const struct tc_action *a = actions[i]; | 
|  | 612 |  | 
|  | 613 | if (jmp_prgcnt > 0) { | 
|  | 614 | jmp_prgcnt -= 1; | 
|  | 615 | continue; | 
|  | 616 | } | 
|  | 617 | repeat: | 
|  | 618 | ret = a->ops->act(skb, a, res); | 
|  | 619 | if (ret == TC_ACT_REPEAT) | 
|  | 620 | goto repeat;	/* we need a ttl - JHS */ | 
|  | 621 |  | 
|  | 622 | if (TC_ACT_EXT_CMP(ret, TC_ACT_JUMP)) { | 
|  | 623 | jmp_prgcnt = ret & TCA_ACT_MAX_PRIO_MASK; | 
|  | 624 | if (!jmp_prgcnt || (jmp_prgcnt > nr_actions)) { | 
|  | 625 | /* faulty opcode, stop pipeline */ | 
|  | 626 | return TC_ACT_OK; | 
|  | 627 | } else { | 
|  | 628 | jmp_ttl -= 1; | 
|  | 629 | if (jmp_ttl > 0) | 
|  | 630 | goto restart_act_graph; | 
|  | 631 | else /* faulty graph, stop pipeline */ | 
|  | 632 | return TC_ACT_OK; | 
|  | 633 | } | 
|  | 634 | } else if (TC_ACT_EXT_CMP(ret, TC_ACT_GOTO_CHAIN)) { | 
|  | 635 | tcf_action_goto_chain_exec(a, res); | 
|  | 636 | } | 
|  | 637 |  | 
|  | 638 | if (ret != TC_ACT_PIPE) | 
|  | 639 | break; | 
|  | 640 | } | 
|  | 641 |  | 
|  | 642 | return ret; | 
|  | 643 | } | 
|  | 644 | EXPORT_SYMBOL(tcf_action_exec); | 
|  | 645 |  | 
|  | 646 | int tcf_action_destroy(struct tc_action *actions[], int bind) | 
|  | 647 | { | 
|  | 648 | const struct tc_action_ops *ops; | 
|  | 649 | struct tc_action *a; | 
|  | 650 | int ret = 0, i; | 
|  | 651 |  | 
|  | 652 | for (i = 0; i < TCA_ACT_MAX_PRIO && actions[i]; i++) { | 
|  | 653 | a = actions[i]; | 
|  | 654 | actions[i] = NULL; | 
|  | 655 | ops = a->ops; | 
|  | 656 | ret = __tcf_idr_release(a, bind, true); | 
|  | 657 | if (ret == ACT_P_DELETED) | 
|  | 658 | module_put(ops->owner); | 
|  | 659 | else if (ret < 0) | 
|  | 660 | return ret; | 
|  | 661 | } | 
|  | 662 | return ret; | 
|  | 663 | } | 
|  | 664 |  | 
|  | 665 | static int tcf_action_destroy_1(struct tc_action *a, int bind) | 
|  | 666 | { | 
|  | 667 | struct tc_action *actions[] = { a, NULL }; | 
|  | 668 |  | 
|  | 669 | return tcf_action_destroy(actions, bind); | 
|  | 670 | } | 
|  | 671 |  | 
|  | 672 | static int tcf_action_put(struct tc_action *p) | 
|  | 673 | { | 
|  | 674 | return __tcf_action_put(p, false); | 
|  | 675 | } | 
|  | 676 |  | 
|  | 677 | /* Put all actions in this array, skip those NULL's. */ | 
|  | 678 | static void tcf_action_put_many(struct tc_action *actions[]) | 
|  | 679 | { | 
|  | 680 | int i; | 
|  | 681 |  | 
|  | 682 | for (i = 0; i < TCA_ACT_MAX_PRIO; i++) { | 
|  | 683 | struct tc_action *a = actions[i]; | 
|  | 684 | const struct tc_action_ops *ops; | 
|  | 685 |  | 
|  | 686 | if (!a) | 
|  | 687 | continue; | 
|  | 688 | ops = a->ops; | 
|  | 689 | if (tcf_action_put(a)) | 
|  | 690 | module_put(ops->owner); | 
|  | 691 | } | 
|  | 692 | } | 
|  | 693 |  | 
|  | 694 | int | 
|  | 695 | tcf_action_dump_old(struct sk_buff *skb, struct tc_action *a, int bind, int ref) | 
|  | 696 | { | 
|  | 697 | return a->ops->dump(skb, a, bind, ref); | 
|  | 698 | } | 
|  | 699 |  | 
|  | 700 | int | 
|  | 701 | tcf_action_dump_1(struct sk_buff *skb, struct tc_action *a, int bind, int ref) | 
|  | 702 | { | 
|  | 703 | int err = -EINVAL; | 
|  | 704 | unsigned char *b = skb_tail_pointer(skb); | 
|  | 705 | struct nlattr *nest; | 
|  | 706 | struct tc_cookie *cookie; | 
|  | 707 |  | 
|  | 708 | if (nla_put_string(skb, TCA_KIND, a->ops->kind)) | 
|  | 709 | goto nla_put_failure; | 
|  | 710 | if (tcf_action_copy_stats(skb, a, 0)) | 
|  | 711 | goto nla_put_failure; | 
|  | 712 |  | 
|  | 713 | rcu_read_lock(); | 
|  | 714 | cookie = rcu_dereference(a->act_cookie); | 
|  | 715 | if (cookie) { | 
|  | 716 | if (nla_put(skb, TCA_ACT_COOKIE, cookie->len, cookie->data)) { | 
|  | 717 | rcu_read_unlock(); | 
|  | 718 | goto nla_put_failure; | 
|  | 719 | } | 
|  | 720 | } | 
|  | 721 | rcu_read_unlock(); | 
|  | 722 |  | 
|  | 723 | nest = nla_nest_start(skb, TCA_OPTIONS); | 
|  | 724 | if (nest == NULL) | 
|  | 725 | goto nla_put_failure; | 
|  | 726 | err = tcf_action_dump_old(skb, a, bind, ref); | 
|  | 727 | if (err > 0) { | 
|  | 728 | nla_nest_end(skb, nest); | 
|  | 729 | return err; | 
|  | 730 | } | 
|  | 731 |  | 
|  | 732 | nla_put_failure: | 
|  | 733 | nlmsg_trim(skb, b); | 
|  | 734 | return -1; | 
|  | 735 | } | 
|  | 736 | EXPORT_SYMBOL(tcf_action_dump_1); | 
|  | 737 |  | 
|  | 738 | int tcf_action_dump(struct sk_buff *skb, struct tc_action *actions[], | 
|  | 739 | int bind, int ref) | 
|  | 740 | { | 
|  | 741 | struct tc_action *a; | 
|  | 742 | int err = -EINVAL, i; | 
|  | 743 | struct nlattr *nest; | 
|  | 744 |  | 
|  | 745 | for (i = 0; i < TCA_ACT_MAX_PRIO && actions[i]; i++) { | 
|  | 746 | a = actions[i]; | 
|  | 747 | nest = nla_nest_start(skb, i + 1); | 
|  | 748 | if (nest == NULL) | 
|  | 749 | goto nla_put_failure; | 
|  | 750 | err = tcf_action_dump_1(skb, a, bind, ref); | 
|  | 751 | if (err < 0) | 
|  | 752 | goto errout; | 
|  | 753 | nla_nest_end(skb, nest); | 
|  | 754 | } | 
|  | 755 |  | 
|  | 756 | return 0; | 
|  | 757 |  | 
|  | 758 | nla_put_failure: | 
|  | 759 | err = -EINVAL; | 
|  | 760 | errout: | 
|  | 761 | nla_nest_cancel(skb, nest); | 
|  | 762 | return err; | 
|  | 763 | } | 
|  | 764 |  | 
|  | 765 | static struct tc_cookie *nla_memdup_cookie(struct nlattr **tb) | 
|  | 766 | { | 
|  | 767 | struct tc_cookie *c = kzalloc(sizeof(*c), GFP_KERNEL); | 
|  | 768 | if (!c) | 
|  | 769 | return NULL; | 
|  | 770 |  | 
|  | 771 | c->data = nla_memdup(tb[TCA_ACT_COOKIE], GFP_KERNEL); | 
|  | 772 | if (!c->data) { | 
|  | 773 | kfree(c); | 
|  | 774 | return NULL; | 
|  | 775 | } | 
|  | 776 | c->len = nla_len(tb[TCA_ACT_COOKIE]); | 
|  | 777 |  | 
|  | 778 | return c; | 
|  | 779 | } | 
|  | 780 |  | 
|  | 781 | static bool tcf_action_valid(int action) | 
|  | 782 | { | 
|  | 783 | int opcode = TC_ACT_EXT_OPCODE(action); | 
|  | 784 |  | 
|  | 785 | if (!opcode) | 
|  | 786 | return action <= TC_ACT_VALUE_MAX; | 
|  | 787 | return opcode <= TC_ACT_EXT_OPCODE_MAX || action == TC_ACT_UNSPEC; | 
|  | 788 | } | 
|  | 789 |  | 
|  | 790 | struct tc_action *tcf_action_init_1(struct net *net, struct tcf_proto *tp, | 
|  | 791 | struct nlattr *nla, struct nlattr *est, | 
|  | 792 | char *name, int ovr, int bind, | 
|  | 793 | bool rtnl_held, | 
|  | 794 | struct netlink_ext_ack *extack) | 
|  | 795 | { | 
|  | 796 | struct tc_action *a; | 
|  | 797 | struct tc_action_ops *a_o; | 
|  | 798 | struct tc_cookie *cookie = NULL; | 
|  | 799 | char act_name[IFNAMSIZ]; | 
|  | 800 | struct nlattr *tb[TCA_ACT_MAX + 1]; | 
|  | 801 | struct nlattr *kind; | 
|  | 802 | int err; | 
|  | 803 |  | 
|  | 804 | if (name == NULL) { | 
|  | 805 | err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL, extack); | 
|  | 806 | if (err < 0) | 
|  | 807 | goto err_out; | 
|  | 808 | err = -EINVAL; | 
|  | 809 | kind = tb[TCA_ACT_KIND]; | 
|  | 810 | if (!kind) { | 
|  | 811 | NL_SET_ERR_MSG(extack, "TC action kind must be specified"); | 
|  | 812 | goto err_out; | 
|  | 813 | } | 
|  | 814 | if (nla_strlcpy(act_name, kind, IFNAMSIZ) >= IFNAMSIZ) { | 
|  | 815 | NL_SET_ERR_MSG(extack, "TC action name too long"); | 
|  | 816 | goto err_out; | 
|  | 817 | } | 
|  | 818 | if (tb[TCA_ACT_COOKIE]) { | 
|  | 819 | int cklen = nla_len(tb[TCA_ACT_COOKIE]); | 
|  | 820 |  | 
|  | 821 | if (cklen > TC_COOKIE_MAX_SIZE) { | 
|  | 822 | NL_SET_ERR_MSG(extack, "TC cookie size above the maximum"); | 
|  | 823 | goto err_out; | 
|  | 824 | } | 
|  | 825 |  | 
|  | 826 | cookie = nla_memdup_cookie(tb); | 
|  | 827 | if (!cookie) { | 
|  | 828 | NL_SET_ERR_MSG(extack, "No memory to generate TC cookie"); | 
|  | 829 | err = -ENOMEM; | 
|  | 830 | goto err_out; | 
|  | 831 | } | 
|  | 832 | } | 
|  | 833 | } else { | 
|  | 834 | if (strlcpy(act_name, name, IFNAMSIZ) >= IFNAMSIZ) { | 
|  | 835 | NL_SET_ERR_MSG(extack, "TC action name too long"); | 
|  | 836 | err = -EINVAL; | 
|  | 837 | goto err_out; | 
|  | 838 | } | 
|  | 839 | } | 
|  | 840 |  | 
|  | 841 | a_o = tc_lookup_action_n(act_name); | 
|  | 842 | if (a_o == NULL) { | 
|  | 843 | #ifdef CONFIG_MODULES | 
|  | 844 | if (rtnl_held) | 
|  | 845 | rtnl_unlock(); | 
|  | 846 | request_module("act_%s", act_name); | 
|  | 847 | if (rtnl_held) | 
|  | 848 | rtnl_lock(); | 
|  | 849 |  | 
|  | 850 | a_o = tc_lookup_action_n(act_name); | 
|  | 851 |  | 
|  | 852 | /* We dropped the RTNL semaphore in order to | 
|  | 853 | * perform the module load.  So, even if we | 
|  | 854 | * succeeded in loading the module we have to | 
|  | 855 | * tell the caller to replay the request.  We | 
|  | 856 | * indicate this using -EAGAIN. | 
|  | 857 | */ | 
|  | 858 | if (a_o != NULL) { | 
|  | 859 | err = -EAGAIN; | 
|  | 860 | goto err_mod; | 
|  | 861 | } | 
|  | 862 | #endif | 
|  | 863 | NL_SET_ERR_MSG(extack, "Failed to load TC action module"); | 
|  | 864 | err = -ENOENT; | 
|  | 865 | goto err_out; | 
|  | 866 | } | 
|  | 867 |  | 
|  | 868 | /* backward compatibility for policer */ | 
|  | 869 | if (name == NULL) | 
|  | 870 | err = a_o->init(net, tb[TCA_ACT_OPTIONS], est, &a, ovr, bind, | 
|  | 871 | rtnl_held, extack); | 
|  | 872 | else | 
|  | 873 | err = a_o->init(net, nla, est, &a, ovr, bind, rtnl_held, | 
|  | 874 | extack); | 
|  | 875 | if (err < 0) | 
|  | 876 | goto err_mod; | 
|  | 877 |  | 
|  | 878 | if (!name && tb[TCA_ACT_COOKIE]) | 
|  | 879 | tcf_set_action_cookie(&a->act_cookie, cookie); | 
|  | 880 |  | 
|  | 881 | /* module count goes up only when brand new policy is created | 
|  | 882 | * if it exists and is only bound to in a_o->init() then | 
|  | 883 | * ACT_P_CREATED is not returned (a zero is). | 
|  | 884 | */ | 
|  | 885 | if (err != ACT_P_CREATED) | 
|  | 886 | module_put(a_o->owner); | 
|  | 887 |  | 
|  | 888 | if (TC_ACT_EXT_CMP(a->tcfa_action, TC_ACT_GOTO_CHAIN)) { | 
|  | 889 | err = tcf_action_goto_chain_init(a, tp); | 
|  | 890 | if (err) { | 
|  | 891 | tcf_action_destroy_1(a, bind); | 
|  | 892 | NL_SET_ERR_MSG(extack, "Failed to init TC action chain"); | 
|  | 893 | return ERR_PTR(err); | 
|  | 894 | } | 
|  | 895 | } | 
|  | 896 |  | 
|  | 897 | if (!tcf_action_valid(a->tcfa_action)) { | 
|  | 898 | tcf_action_destroy_1(a, bind); | 
|  | 899 | NL_SET_ERR_MSG(extack, "Invalid control action value"); | 
|  | 900 | return ERR_PTR(-EINVAL); | 
|  | 901 | } | 
|  | 902 |  | 
|  | 903 | return a; | 
|  | 904 |  | 
|  | 905 | err_mod: | 
|  | 906 | module_put(a_o->owner); | 
|  | 907 | err_out: | 
|  | 908 | if (cookie) { | 
|  | 909 | kfree(cookie->data); | 
|  | 910 | kfree(cookie); | 
|  | 911 | } | 
|  | 912 | return ERR_PTR(err); | 
|  | 913 | } | 
|  | 914 |  | 
|  | 915 | /* Returns numbers of initialized actions or negative error. */ | 
|  | 916 |  | 
|  | 917 | int tcf_action_init(struct net *net, struct tcf_proto *tp, struct nlattr *nla, | 
|  | 918 | struct nlattr *est, char *name, int ovr, int bind, | 
|  | 919 | struct tc_action *actions[], size_t *attr_size, | 
|  | 920 | bool rtnl_held, struct netlink_ext_ack *extack) | 
|  | 921 | { | 
|  | 922 | struct nlattr *tb[TCA_ACT_MAX_PRIO + 1]; | 
|  | 923 | struct tc_action *act; | 
|  | 924 | size_t sz = 0; | 
|  | 925 | int err; | 
|  | 926 | int i; | 
|  | 927 |  | 
|  | 928 | err = nla_parse_nested(tb, TCA_ACT_MAX_PRIO, nla, NULL, extack); | 
|  | 929 | if (err < 0) | 
|  | 930 | return err; | 
|  | 931 |  | 
|  | 932 | for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) { | 
|  | 933 | act = tcf_action_init_1(net, tp, tb[i], est, name, ovr, bind, | 
|  | 934 | rtnl_held, extack); | 
|  | 935 | if (IS_ERR(act)) { | 
|  | 936 | err = PTR_ERR(act); | 
|  | 937 | goto err; | 
|  | 938 | } | 
|  | 939 | act->order = i; | 
|  | 940 | sz += tcf_action_fill_size(act); | 
|  | 941 | /* Start from index 0 */ | 
|  | 942 | actions[i - 1] = act; | 
|  | 943 | } | 
|  | 944 |  | 
|  | 945 | *attr_size = tcf_action_full_attrs_size(sz); | 
|  | 946 | return i - 1; | 
|  | 947 |  | 
|  | 948 | err: | 
|  | 949 | tcf_action_destroy(actions, bind); | 
|  | 950 | return err; | 
|  | 951 | } | 
|  | 952 |  | 
|  | 953 | int tcf_action_copy_stats(struct sk_buff *skb, struct tc_action *p, | 
|  | 954 | int compat_mode) | 
|  | 955 | { | 
|  | 956 | int err = 0; | 
|  | 957 | struct gnet_dump d; | 
|  | 958 |  | 
|  | 959 | if (p == NULL) | 
|  | 960 | goto errout; | 
|  | 961 |  | 
|  | 962 | /* compat_mode being true specifies a call that is supposed | 
|  | 963 | * to add additional backward compatibility statistic TLVs. | 
|  | 964 | */ | 
|  | 965 | if (compat_mode) { | 
|  | 966 | if (p->type == TCA_OLD_COMPAT) | 
|  | 967 | err = gnet_stats_start_copy_compat(skb, 0, | 
|  | 968 | TCA_STATS, | 
|  | 969 | TCA_XSTATS, | 
|  | 970 | &p->tcfa_lock, &d, | 
|  | 971 | TCA_PAD); | 
|  | 972 | else | 
|  | 973 | return 0; | 
|  | 974 | } else | 
|  | 975 | err = gnet_stats_start_copy(skb, TCA_ACT_STATS, | 
|  | 976 | &p->tcfa_lock, &d, TCA_ACT_PAD); | 
|  | 977 |  | 
|  | 978 | if (err < 0) | 
|  | 979 | goto errout; | 
|  | 980 |  | 
|  | 981 | if (gnet_stats_copy_basic(NULL, &d, p->cpu_bstats, &p->tcfa_bstats) < 0 || | 
|  | 982 | gnet_stats_copy_rate_est(&d, &p->tcfa_rate_est) < 0 || | 
|  | 983 | gnet_stats_copy_queue(&d, p->cpu_qstats, | 
|  | 984 | &p->tcfa_qstats, | 
|  | 985 | p->tcfa_qstats.qlen) < 0) | 
|  | 986 | goto errout; | 
|  | 987 |  | 
|  | 988 | if (gnet_stats_finish_copy(&d) < 0) | 
|  | 989 | goto errout; | 
|  | 990 |  | 
|  | 991 | return 0; | 
|  | 992 |  | 
|  | 993 | errout: | 
|  | 994 | return -1; | 
|  | 995 | } | 
|  | 996 |  | 
|  | 997 | static int tca_get_fill(struct sk_buff *skb, struct tc_action *actions[], | 
|  | 998 | u32 portid, u32 seq, u16 flags, int event, int bind, | 
|  | 999 | int ref) | 
|  | 1000 | { | 
|  | 1001 | struct tcamsg *t; | 
|  | 1002 | struct nlmsghdr *nlh; | 
|  | 1003 | unsigned char *b = skb_tail_pointer(skb); | 
|  | 1004 | struct nlattr *nest; | 
|  | 1005 |  | 
|  | 1006 | nlh = nlmsg_put(skb, portid, seq, event, sizeof(*t), flags); | 
|  | 1007 | if (!nlh) | 
|  | 1008 | goto out_nlmsg_trim; | 
|  | 1009 | t = nlmsg_data(nlh); | 
|  | 1010 | t->tca_family = AF_UNSPEC; | 
|  | 1011 | t->tca__pad1 = 0; | 
|  | 1012 | t->tca__pad2 = 0; | 
|  | 1013 |  | 
|  | 1014 | nest = nla_nest_start(skb, TCA_ACT_TAB); | 
|  | 1015 | if (!nest) | 
|  | 1016 | goto out_nlmsg_trim; | 
|  | 1017 |  | 
|  | 1018 | if (tcf_action_dump(skb, actions, bind, ref) < 0) | 
|  | 1019 | goto out_nlmsg_trim; | 
|  | 1020 |  | 
|  | 1021 | nla_nest_end(skb, nest); | 
|  | 1022 |  | 
|  | 1023 | nlh->nlmsg_len = skb_tail_pointer(skb) - b; | 
|  | 1024 | return skb->len; | 
|  | 1025 |  | 
|  | 1026 | out_nlmsg_trim: | 
|  | 1027 | nlmsg_trim(skb, b); | 
|  | 1028 | return -1; | 
|  | 1029 | } | 
|  | 1030 |  | 
|  | 1031 | static int | 
|  | 1032 | tcf_get_notify(struct net *net, u32 portid, struct nlmsghdr *n, | 
|  | 1033 | struct tc_action *actions[], int event, | 
|  | 1034 | struct netlink_ext_ack *extack) | 
|  | 1035 | { | 
|  | 1036 | struct sk_buff *skb; | 
|  | 1037 |  | 
|  | 1038 | skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL); | 
|  | 1039 | if (!skb) | 
|  | 1040 | return -ENOBUFS; | 
|  | 1041 | if (tca_get_fill(skb, actions, portid, n->nlmsg_seq, 0, event, | 
|  | 1042 | 0, 1) <= 0) { | 
|  | 1043 | NL_SET_ERR_MSG(extack, "Failed to fill netlink attributes while adding TC action"); | 
|  | 1044 | kfree_skb(skb); | 
|  | 1045 | return -EINVAL; | 
|  | 1046 | } | 
|  | 1047 |  | 
|  | 1048 | return rtnl_unicast(skb, net, portid); | 
|  | 1049 | } | 
|  | 1050 |  | 
|  | 1051 | static struct tc_action *tcf_action_get_1(struct net *net, struct nlattr *nla, | 
|  | 1052 | struct nlmsghdr *n, u32 portid, | 
|  | 1053 | struct netlink_ext_ack *extack) | 
|  | 1054 | { | 
|  | 1055 | struct nlattr *tb[TCA_ACT_MAX + 1]; | 
|  | 1056 | const struct tc_action_ops *ops; | 
|  | 1057 | struct tc_action *a; | 
|  | 1058 | int index; | 
|  | 1059 | int err; | 
|  | 1060 |  | 
|  | 1061 | err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL, extack); | 
|  | 1062 | if (err < 0) | 
|  | 1063 | goto err_out; | 
|  | 1064 |  | 
|  | 1065 | err = -EINVAL; | 
|  | 1066 | if (tb[TCA_ACT_INDEX] == NULL || | 
|  | 1067 | nla_len(tb[TCA_ACT_INDEX]) < sizeof(index)) { | 
|  | 1068 | NL_SET_ERR_MSG(extack, "Invalid TC action index value"); | 
|  | 1069 | goto err_out; | 
|  | 1070 | } | 
|  | 1071 | index = nla_get_u32(tb[TCA_ACT_INDEX]); | 
|  | 1072 |  | 
|  | 1073 | err = -EINVAL; | 
|  | 1074 | ops = tc_lookup_action(tb[TCA_ACT_KIND]); | 
|  | 1075 | if (!ops) { /* could happen in batch of actions */ | 
|  | 1076 | NL_SET_ERR_MSG(extack, "Specified TC action not found"); | 
|  | 1077 | goto err_out; | 
|  | 1078 | } | 
|  | 1079 | err = -ENOENT; | 
|  | 1080 | if (ops->lookup(net, &a, index, extack) == 0) | 
|  | 1081 | goto err_mod; | 
|  | 1082 |  | 
|  | 1083 | module_put(ops->owner); | 
|  | 1084 | return a; | 
|  | 1085 |  | 
|  | 1086 | err_mod: | 
|  | 1087 | module_put(ops->owner); | 
|  | 1088 | err_out: | 
|  | 1089 | return ERR_PTR(err); | 
|  | 1090 | } | 
|  | 1091 |  | 
|  | 1092 | static int tca_action_flush(struct net *net, struct nlattr *nla, | 
|  | 1093 | struct nlmsghdr *n, u32 portid, | 
|  | 1094 | struct netlink_ext_ack *extack) | 
|  | 1095 | { | 
|  | 1096 | struct sk_buff *skb; | 
|  | 1097 | unsigned char *b; | 
|  | 1098 | struct nlmsghdr *nlh; | 
|  | 1099 | struct tcamsg *t; | 
|  | 1100 | struct netlink_callback dcb; | 
|  | 1101 | struct nlattr *nest; | 
|  | 1102 | struct nlattr *tb[TCA_ACT_MAX + 1]; | 
|  | 1103 | const struct tc_action_ops *ops; | 
|  | 1104 | struct nlattr *kind; | 
|  | 1105 | int err = -ENOMEM; | 
|  | 1106 |  | 
|  | 1107 | skb = alloc_skb(NLMSG_GOODSIZE, GFP_KERNEL); | 
|  | 1108 | if (!skb) | 
|  | 1109 | return err; | 
|  | 1110 |  | 
|  | 1111 | b = skb_tail_pointer(skb); | 
|  | 1112 |  | 
|  | 1113 | err = nla_parse_nested(tb, TCA_ACT_MAX, nla, NULL, extack); | 
|  | 1114 | if (err < 0) | 
|  | 1115 | goto err_out; | 
|  | 1116 |  | 
|  | 1117 | err = -EINVAL; | 
|  | 1118 | kind = tb[TCA_ACT_KIND]; | 
|  | 1119 | ops = tc_lookup_action(kind); | 
|  | 1120 | if (!ops) { /*some idjot trying to flush unknown action */ | 
|  | 1121 | NL_SET_ERR_MSG(extack, "Cannot flush unknown TC action"); | 
|  | 1122 | goto err_out; | 
|  | 1123 | } | 
|  | 1124 |  | 
|  | 1125 | nlh = nlmsg_put(skb, portid, n->nlmsg_seq, RTM_DELACTION, | 
|  | 1126 | sizeof(*t), 0); | 
|  | 1127 | if (!nlh) { | 
|  | 1128 | NL_SET_ERR_MSG(extack, "Failed to create TC action flush notification"); | 
|  | 1129 | goto out_module_put; | 
|  | 1130 | } | 
|  | 1131 | t = nlmsg_data(nlh); | 
|  | 1132 | t->tca_family = AF_UNSPEC; | 
|  | 1133 | t->tca__pad1 = 0; | 
|  | 1134 | t->tca__pad2 = 0; | 
|  | 1135 |  | 
|  | 1136 | nest = nla_nest_start(skb, TCA_ACT_TAB); | 
|  | 1137 | if (!nest) { | 
|  | 1138 | NL_SET_ERR_MSG(extack, "Failed to add new netlink message"); | 
|  | 1139 | goto out_module_put; | 
|  | 1140 | } | 
|  | 1141 |  | 
|  | 1142 | err = ops->walk(net, skb, &dcb, RTM_DELACTION, ops, extack); | 
|  | 1143 | if (err <= 0) { | 
|  | 1144 | nla_nest_cancel(skb, nest); | 
|  | 1145 | goto out_module_put; | 
|  | 1146 | } | 
|  | 1147 |  | 
|  | 1148 | nla_nest_end(skb, nest); | 
|  | 1149 |  | 
|  | 1150 | nlh->nlmsg_len = skb_tail_pointer(skb) - b; | 
|  | 1151 | nlh->nlmsg_flags |= NLM_F_ROOT; | 
|  | 1152 | module_put(ops->owner); | 
|  | 1153 | err = rtnetlink_send(skb, net, portid, RTNLGRP_TC, | 
|  | 1154 | n->nlmsg_flags & NLM_F_ECHO); | 
|  | 1155 | if (err > 0) | 
|  | 1156 | return 0; | 
|  | 1157 | if (err < 0) | 
|  | 1158 | NL_SET_ERR_MSG(extack, "Failed to send TC action flush notification"); | 
|  | 1159 |  | 
|  | 1160 | return err; | 
|  | 1161 |  | 
|  | 1162 | out_module_put: | 
|  | 1163 | module_put(ops->owner); | 
|  | 1164 | err_out: | 
|  | 1165 | kfree_skb(skb); | 
|  | 1166 | return err; | 
|  | 1167 | } | 
|  | 1168 |  | 
|  | 1169 | static int tcf_action_delete(struct net *net, struct tc_action *actions[]) | 
|  | 1170 | { | 
|  | 1171 | int i; | 
|  | 1172 |  | 
|  | 1173 | for (i = 0; i < TCA_ACT_MAX_PRIO && actions[i]; i++) { | 
|  | 1174 | struct tc_action *a = actions[i]; | 
|  | 1175 | const struct tc_action_ops *ops = a->ops; | 
|  | 1176 | /* Actions can be deleted concurrently so we must save their | 
|  | 1177 | * type and id to search again after reference is released. | 
|  | 1178 | */ | 
|  | 1179 | struct tcf_idrinfo *idrinfo = a->idrinfo; | 
|  | 1180 | u32 act_index = a->tcfa_index; | 
|  | 1181 |  | 
|  | 1182 | actions[i] = NULL; | 
|  | 1183 | if (tcf_action_put(a)) { | 
|  | 1184 | /* last reference, action was deleted concurrently */ | 
|  | 1185 | module_put(ops->owner); | 
|  | 1186 | } else  { | 
|  | 1187 | int ret; | 
|  | 1188 |  | 
|  | 1189 | /* now do the delete */ | 
|  | 1190 | ret = tcf_idr_delete_index(idrinfo, act_index); | 
|  | 1191 | if (ret < 0) | 
|  | 1192 | return ret; | 
|  | 1193 | } | 
|  | 1194 | } | 
|  | 1195 | return 0; | 
|  | 1196 | } | 
|  | 1197 |  | 
|  | 1198 | static int | 
|  | 1199 | tcf_del_notify(struct net *net, struct nlmsghdr *n, struct tc_action *actions[], | 
|  | 1200 | u32 portid, size_t attr_size, struct netlink_ext_ack *extack) | 
|  | 1201 | { | 
|  | 1202 | int ret; | 
|  | 1203 | struct sk_buff *skb; | 
|  | 1204 |  | 
|  | 1205 | skb = alloc_skb(attr_size <= NLMSG_GOODSIZE ? NLMSG_GOODSIZE : attr_size, | 
|  | 1206 | GFP_KERNEL); | 
|  | 1207 | if (!skb) | 
|  | 1208 | return -ENOBUFS; | 
|  | 1209 |  | 
|  | 1210 | if (tca_get_fill(skb, actions, portid, n->nlmsg_seq, 0, RTM_DELACTION, | 
|  | 1211 | 0, 2) <= 0) { | 
|  | 1212 | NL_SET_ERR_MSG(extack, "Failed to fill netlink TC action attributes"); | 
|  | 1213 | kfree_skb(skb); | 
|  | 1214 | return -EINVAL; | 
|  | 1215 | } | 
|  | 1216 |  | 
|  | 1217 | /* now do the delete */ | 
|  | 1218 | ret = tcf_action_delete(net, actions); | 
|  | 1219 | if (ret < 0) { | 
|  | 1220 | NL_SET_ERR_MSG(extack, "Failed to delete TC action"); | 
|  | 1221 | kfree_skb(skb); | 
|  | 1222 | return ret; | 
|  | 1223 | } | 
|  | 1224 |  | 
|  | 1225 | ret = rtnetlink_send(skb, net, portid, RTNLGRP_TC, | 
|  | 1226 | n->nlmsg_flags & NLM_F_ECHO); | 
|  | 1227 | if (ret > 0) | 
|  | 1228 | return 0; | 
|  | 1229 | return ret; | 
|  | 1230 | } | 
|  | 1231 |  | 
|  | 1232 | static int | 
|  | 1233 | tca_action_gd(struct net *net, struct nlattr *nla, struct nlmsghdr *n, | 
|  | 1234 | u32 portid, int event, struct netlink_ext_ack *extack) | 
|  | 1235 | { | 
|  | 1236 | int i, ret; | 
|  | 1237 | struct nlattr *tb[TCA_ACT_MAX_PRIO + 1]; | 
|  | 1238 | struct tc_action *act; | 
|  | 1239 | size_t attr_size = 0; | 
|  | 1240 | struct tc_action *actions[TCA_ACT_MAX_PRIO] = {}; | 
|  | 1241 |  | 
|  | 1242 | ret = nla_parse_nested(tb, TCA_ACT_MAX_PRIO, nla, NULL, extack); | 
|  | 1243 | if (ret < 0) | 
|  | 1244 | return ret; | 
|  | 1245 |  | 
|  | 1246 | if (event == RTM_DELACTION && n->nlmsg_flags & NLM_F_ROOT) { | 
|  | 1247 | if (tb[1]) | 
|  | 1248 | return tca_action_flush(net, tb[1], n, portid, extack); | 
|  | 1249 |  | 
|  | 1250 | NL_SET_ERR_MSG(extack, "Invalid netlink attributes while flushing TC action"); | 
|  | 1251 | return -EINVAL; | 
|  | 1252 | } | 
|  | 1253 |  | 
|  | 1254 | for (i = 1; i <= TCA_ACT_MAX_PRIO && tb[i]; i++) { | 
|  | 1255 | act = tcf_action_get_1(net, tb[i], n, portid, extack); | 
|  | 1256 | if (IS_ERR(act)) { | 
|  | 1257 | ret = PTR_ERR(act); | 
|  | 1258 | goto err; | 
|  | 1259 | } | 
|  | 1260 | attr_size += tcf_action_fill_size(act); | 
|  | 1261 | actions[i - 1] = act; | 
|  | 1262 | } | 
|  | 1263 |  | 
|  | 1264 | attr_size = tcf_action_full_attrs_size(attr_size); | 
|  | 1265 |  | 
|  | 1266 | if (event == RTM_GETACTION) | 
|  | 1267 | ret = tcf_get_notify(net, portid, n, actions, event, extack); | 
|  | 1268 | else { /* delete */ | 
|  | 1269 | ret = tcf_del_notify(net, n, actions, portid, attr_size, extack); | 
|  | 1270 | if (ret) | 
|  | 1271 | goto err; | 
|  | 1272 | return 0; | 
|  | 1273 | } | 
|  | 1274 | err: | 
|  | 1275 | tcf_action_put_many(actions); | 
|  | 1276 | return ret; | 
|  | 1277 | } | 
|  | 1278 |  | 
|  | 1279 | static int | 
|  | 1280 | tcf_add_notify(struct net *net, struct nlmsghdr *n, struct tc_action *actions[], | 
|  | 1281 | u32 portid, size_t attr_size, struct netlink_ext_ack *extack) | 
|  | 1282 | { | 
|  | 1283 | struct sk_buff *skb; | 
|  | 1284 | int err = 0; | 
|  | 1285 |  | 
|  | 1286 | skb = alloc_skb(attr_size <= NLMSG_GOODSIZE ? NLMSG_GOODSIZE : attr_size, | 
|  | 1287 | GFP_KERNEL); | 
|  | 1288 | if (!skb) | 
|  | 1289 | return -ENOBUFS; | 
|  | 1290 |  | 
|  | 1291 | if (tca_get_fill(skb, actions, portid, n->nlmsg_seq, n->nlmsg_flags, | 
|  | 1292 | RTM_NEWACTION, 0, 0) <= 0) { | 
|  | 1293 | NL_SET_ERR_MSG(extack, "Failed to fill netlink attributes while adding TC action"); | 
|  | 1294 | kfree_skb(skb); | 
|  | 1295 | return -EINVAL; | 
|  | 1296 | } | 
|  | 1297 |  | 
|  | 1298 | err = rtnetlink_send(skb, net, portid, RTNLGRP_TC, | 
|  | 1299 | n->nlmsg_flags & NLM_F_ECHO); | 
|  | 1300 | if (err > 0) | 
|  | 1301 | err = 0; | 
|  | 1302 | return err; | 
|  | 1303 | } | 
|  | 1304 |  | 
|  | 1305 | static int tcf_action_add(struct net *net, struct nlattr *nla, | 
|  | 1306 | struct nlmsghdr *n, u32 portid, int ovr, | 
|  | 1307 | struct netlink_ext_ack *extack) | 
|  | 1308 | { | 
|  | 1309 | size_t attr_size = 0; | 
|  | 1310 | int loop, ret; | 
|  | 1311 | struct tc_action *actions[TCA_ACT_MAX_PRIO] = {}; | 
|  | 1312 |  | 
|  | 1313 | for (loop = 0; loop < 10; loop++) { | 
|  | 1314 | ret = tcf_action_init(net, NULL, nla, NULL, NULL, ovr, 0, | 
|  | 1315 | actions, &attr_size, true, extack); | 
|  | 1316 | if (ret != -EAGAIN) | 
|  | 1317 | break; | 
|  | 1318 | } | 
|  | 1319 |  | 
|  | 1320 | if (ret < 0) | 
|  | 1321 | return ret; | 
|  | 1322 | ret = tcf_add_notify(net, n, actions, portid, attr_size, extack); | 
|  | 1323 | if (ovr) | 
|  | 1324 | tcf_action_put_many(actions); | 
|  | 1325 |  | 
|  | 1326 | return ret; | 
|  | 1327 | } | 
|  | 1328 |  | 
|  | 1329 | static u32 tcaa_root_flags_allowed = TCA_FLAG_LARGE_DUMP_ON; | 
|  | 1330 | static const struct nla_policy tcaa_policy[TCA_ROOT_MAX + 1] = { | 
|  | 1331 | [TCA_ROOT_FLAGS] = { .type = NLA_BITFIELD32, | 
|  | 1332 | .validation_data = &tcaa_root_flags_allowed }, | 
|  | 1333 | [TCA_ROOT_TIME_DELTA]      = { .type = NLA_U32 }, | 
|  | 1334 | }; | 
|  | 1335 |  | 
|  | 1336 | static int tc_ctl_action(struct sk_buff *skb, struct nlmsghdr *n, | 
|  | 1337 | struct netlink_ext_ack *extack) | 
|  | 1338 | { | 
|  | 1339 | struct net *net = sock_net(skb->sk); | 
|  | 1340 | struct nlattr *tca[TCA_ROOT_MAX + 1]; | 
|  | 1341 | u32 portid = skb ? NETLINK_CB(skb).portid : 0; | 
|  | 1342 | int ret = 0, ovr = 0; | 
|  | 1343 |  | 
|  | 1344 | if ((n->nlmsg_type != RTM_GETACTION) && | 
|  | 1345 | !netlink_capable(skb, CAP_NET_ADMIN)) | 
|  | 1346 | return -EPERM; | 
|  | 1347 |  | 
|  | 1348 | ret = nlmsg_parse(n, sizeof(struct tcamsg), tca, TCA_ROOT_MAX, NULL, | 
|  | 1349 | extack); | 
|  | 1350 | if (ret < 0) | 
|  | 1351 | return ret; | 
|  | 1352 |  | 
|  | 1353 | if (tca[TCA_ACT_TAB] == NULL) { | 
|  | 1354 | NL_SET_ERR_MSG(extack, "Netlink action attributes missing"); | 
|  | 1355 | return -EINVAL; | 
|  | 1356 | } | 
|  | 1357 |  | 
|  | 1358 | /* n->nlmsg_flags & NLM_F_CREATE */ | 
|  | 1359 | switch (n->nlmsg_type) { | 
|  | 1360 | case RTM_NEWACTION: | 
|  | 1361 | /* we are going to assume all other flags | 
|  | 1362 | * imply create only if it doesn't exist | 
|  | 1363 | * Note that CREATE | EXCL implies that | 
|  | 1364 | * but since we want avoid ambiguity (eg when flags | 
|  | 1365 | * is zero) then just set this | 
|  | 1366 | */ | 
|  | 1367 | if (n->nlmsg_flags & NLM_F_REPLACE) | 
|  | 1368 | ovr = 1; | 
|  | 1369 | ret = tcf_action_add(net, tca[TCA_ACT_TAB], n, portid, ovr, | 
|  | 1370 | extack); | 
|  | 1371 | break; | 
|  | 1372 | case RTM_DELACTION: | 
|  | 1373 | ret = tca_action_gd(net, tca[TCA_ACT_TAB], n, | 
|  | 1374 | portid, RTM_DELACTION, extack); | 
|  | 1375 | break; | 
|  | 1376 | case RTM_GETACTION: | 
|  | 1377 | ret = tca_action_gd(net, tca[TCA_ACT_TAB], n, | 
|  | 1378 | portid, RTM_GETACTION, extack); | 
|  | 1379 | break; | 
|  | 1380 | default: | 
|  | 1381 | BUG(); | 
|  | 1382 | } | 
|  | 1383 |  | 
|  | 1384 | return ret; | 
|  | 1385 | } | 
|  | 1386 |  | 
|  | 1387 | static struct nlattr *find_dump_kind(struct nlattr **nla) | 
|  | 1388 | { | 
|  | 1389 | struct nlattr *tb1, *tb2[TCA_ACT_MAX + 1]; | 
|  | 1390 | struct nlattr *tb[TCA_ACT_MAX_PRIO + 1]; | 
|  | 1391 | struct nlattr *kind; | 
|  | 1392 |  | 
|  | 1393 | tb1 = nla[TCA_ACT_TAB]; | 
|  | 1394 | if (tb1 == NULL) | 
|  | 1395 | return NULL; | 
|  | 1396 |  | 
|  | 1397 | if (nla_parse(tb, TCA_ACT_MAX_PRIO, nla_data(tb1), | 
|  | 1398 | NLMSG_ALIGN(nla_len(tb1)), NULL, NULL) < 0) | 
|  | 1399 | return NULL; | 
|  | 1400 |  | 
|  | 1401 | if (tb[1] == NULL) | 
|  | 1402 | return NULL; | 
|  | 1403 | if (nla_parse_nested(tb2, TCA_ACT_MAX, tb[1], NULL, NULL) < 0) | 
|  | 1404 | return NULL; | 
|  | 1405 | kind = tb2[TCA_ACT_KIND]; | 
|  | 1406 |  | 
|  | 1407 | return kind; | 
|  | 1408 | } | 
|  | 1409 |  | 
|  | 1410 | static int tc_dump_action(struct sk_buff *skb, struct netlink_callback *cb) | 
|  | 1411 | { | 
|  | 1412 | struct net *net = sock_net(skb->sk); | 
|  | 1413 | struct nlmsghdr *nlh; | 
|  | 1414 | unsigned char *b = skb_tail_pointer(skb); | 
|  | 1415 | struct nlattr *nest; | 
|  | 1416 | struct tc_action_ops *a_o; | 
|  | 1417 | int ret = 0; | 
|  | 1418 | struct tcamsg *t = (struct tcamsg *) nlmsg_data(cb->nlh); | 
|  | 1419 | struct nlattr *tb[TCA_ROOT_MAX + 1]; | 
|  | 1420 | struct nlattr *count_attr = NULL; | 
|  | 1421 | unsigned long jiffy_since = 0; | 
|  | 1422 | struct nlattr *kind = NULL; | 
|  | 1423 | struct nla_bitfield32 bf; | 
|  | 1424 | u32 msecs_since = 0; | 
|  | 1425 | u32 act_count = 0; | 
|  | 1426 |  | 
|  | 1427 | ret = nlmsg_parse(cb->nlh, sizeof(struct tcamsg), tb, TCA_ROOT_MAX, | 
|  | 1428 | tcaa_policy, NULL); | 
|  | 1429 | if (ret < 0) | 
|  | 1430 | return ret; | 
|  | 1431 |  | 
|  | 1432 | kind = find_dump_kind(tb); | 
|  | 1433 | if (kind == NULL) { | 
|  | 1434 | pr_info("tc_dump_action: action bad kind\n"); | 
|  | 1435 | return 0; | 
|  | 1436 | } | 
|  | 1437 |  | 
|  | 1438 | a_o = tc_lookup_action(kind); | 
|  | 1439 | if (a_o == NULL) | 
|  | 1440 | return 0; | 
|  | 1441 |  | 
|  | 1442 | cb->args[2] = 0; | 
|  | 1443 | if (tb[TCA_ROOT_FLAGS]) { | 
|  | 1444 | bf = nla_get_bitfield32(tb[TCA_ROOT_FLAGS]); | 
|  | 1445 | cb->args[2] = bf.value; | 
|  | 1446 | } | 
|  | 1447 |  | 
|  | 1448 | if (tb[TCA_ROOT_TIME_DELTA]) { | 
|  | 1449 | msecs_since = nla_get_u32(tb[TCA_ROOT_TIME_DELTA]); | 
|  | 1450 | } | 
|  | 1451 |  | 
|  | 1452 | nlh = nlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq, | 
|  | 1453 | cb->nlh->nlmsg_type, sizeof(*t), 0); | 
|  | 1454 | if (!nlh) | 
|  | 1455 | goto out_module_put; | 
|  | 1456 |  | 
|  | 1457 | if (msecs_since) | 
|  | 1458 | jiffy_since = jiffies - msecs_to_jiffies(msecs_since); | 
|  | 1459 |  | 
|  | 1460 | t = nlmsg_data(nlh); | 
|  | 1461 | t->tca_family = AF_UNSPEC; | 
|  | 1462 | t->tca__pad1 = 0; | 
|  | 1463 | t->tca__pad2 = 0; | 
|  | 1464 | cb->args[3] = jiffy_since; | 
|  | 1465 | count_attr = nla_reserve(skb, TCA_ROOT_COUNT, sizeof(u32)); | 
|  | 1466 | if (!count_attr) | 
|  | 1467 | goto out_module_put; | 
|  | 1468 |  | 
|  | 1469 | nest = nla_nest_start(skb, TCA_ACT_TAB); | 
|  | 1470 | if (nest == NULL) | 
|  | 1471 | goto out_module_put; | 
|  | 1472 |  | 
|  | 1473 | ret = a_o->walk(net, skb, cb, RTM_GETACTION, a_o, NULL); | 
|  | 1474 | if (ret < 0) | 
|  | 1475 | goto out_module_put; | 
|  | 1476 |  | 
|  | 1477 | if (ret > 0) { | 
|  | 1478 | nla_nest_end(skb, nest); | 
|  | 1479 | ret = skb->len; | 
|  | 1480 | act_count = cb->args[1]; | 
|  | 1481 | memcpy(nla_data(count_attr), &act_count, sizeof(u32)); | 
|  | 1482 | cb->args[1] = 0; | 
|  | 1483 | } else | 
|  | 1484 | nlmsg_trim(skb, b); | 
|  | 1485 |  | 
|  | 1486 | nlh->nlmsg_len = skb_tail_pointer(skb) - b; | 
|  | 1487 | if (NETLINK_CB(cb->skb).portid && ret) | 
|  | 1488 | nlh->nlmsg_flags |= NLM_F_MULTI; | 
|  | 1489 | module_put(a_o->owner); | 
|  | 1490 | return skb->len; | 
|  | 1491 |  | 
|  | 1492 | out_module_put: | 
|  | 1493 | module_put(a_o->owner); | 
|  | 1494 | nlmsg_trim(skb, b); | 
|  | 1495 | return skb->len; | 
|  | 1496 | } | 
|  | 1497 |  | 
|  | 1498 | struct tcf_action_net { | 
|  | 1499 | struct rhashtable egdev_ht; | 
|  | 1500 | }; | 
|  | 1501 |  | 
|  | 1502 | static unsigned int tcf_action_net_id; | 
|  | 1503 |  | 
|  | 1504 | struct tcf_action_egdev_cb { | 
|  | 1505 | struct list_head list; | 
|  | 1506 | tc_setup_cb_t *cb; | 
|  | 1507 | void *cb_priv; | 
|  | 1508 | }; | 
|  | 1509 |  | 
|  | 1510 | struct tcf_action_egdev { | 
|  | 1511 | struct rhash_head ht_node; | 
|  | 1512 | const struct net_device *dev; | 
|  | 1513 | unsigned int refcnt; | 
|  | 1514 | struct list_head cb_list; | 
|  | 1515 | }; | 
|  | 1516 |  | 
|  | 1517 | static const struct rhashtable_params tcf_action_egdev_ht_params = { | 
|  | 1518 | .key_offset = offsetof(struct tcf_action_egdev, dev), | 
|  | 1519 | .head_offset = offsetof(struct tcf_action_egdev, ht_node), | 
|  | 1520 | .key_len = sizeof(const struct net_device *), | 
|  | 1521 | }; | 
|  | 1522 |  | 
|  | 1523 | static struct tcf_action_egdev * | 
|  | 1524 | tcf_action_egdev_lookup(const struct net_device *dev) | 
|  | 1525 | { | 
|  | 1526 | struct net *net = dev_net(dev); | 
|  | 1527 | struct tcf_action_net *tan = net_generic(net, tcf_action_net_id); | 
|  | 1528 |  | 
|  | 1529 | return rhashtable_lookup_fast(&tan->egdev_ht, &dev, | 
|  | 1530 | tcf_action_egdev_ht_params); | 
|  | 1531 | } | 
|  | 1532 |  | 
|  | 1533 | static struct tcf_action_egdev * | 
|  | 1534 | tcf_action_egdev_get(const struct net_device *dev) | 
|  | 1535 | { | 
|  | 1536 | struct tcf_action_egdev *egdev; | 
|  | 1537 | struct tcf_action_net *tan; | 
|  | 1538 |  | 
|  | 1539 | egdev = tcf_action_egdev_lookup(dev); | 
|  | 1540 | if (egdev) | 
|  | 1541 | goto inc_ref; | 
|  | 1542 |  | 
|  | 1543 | egdev = kzalloc(sizeof(*egdev), GFP_KERNEL); | 
|  | 1544 | if (!egdev) | 
|  | 1545 | return NULL; | 
|  | 1546 | INIT_LIST_HEAD(&egdev->cb_list); | 
|  | 1547 | egdev->dev = dev; | 
|  | 1548 | tan = net_generic(dev_net(dev), tcf_action_net_id); | 
|  | 1549 | rhashtable_insert_fast(&tan->egdev_ht, &egdev->ht_node, | 
|  | 1550 | tcf_action_egdev_ht_params); | 
|  | 1551 |  | 
|  | 1552 | inc_ref: | 
|  | 1553 | egdev->refcnt++; | 
|  | 1554 | return egdev; | 
|  | 1555 | } | 
|  | 1556 |  | 
|  | 1557 | static void tcf_action_egdev_put(struct tcf_action_egdev *egdev) | 
|  | 1558 | { | 
|  | 1559 | struct tcf_action_net *tan; | 
|  | 1560 |  | 
|  | 1561 | if (--egdev->refcnt) | 
|  | 1562 | return; | 
|  | 1563 | tan = net_generic(dev_net(egdev->dev), tcf_action_net_id); | 
|  | 1564 | rhashtable_remove_fast(&tan->egdev_ht, &egdev->ht_node, | 
|  | 1565 | tcf_action_egdev_ht_params); | 
|  | 1566 | kfree(egdev); | 
|  | 1567 | } | 
|  | 1568 |  | 
|  | 1569 | static struct tcf_action_egdev_cb * | 
|  | 1570 | tcf_action_egdev_cb_lookup(struct tcf_action_egdev *egdev, | 
|  | 1571 | tc_setup_cb_t *cb, void *cb_priv) | 
|  | 1572 | { | 
|  | 1573 | struct tcf_action_egdev_cb *egdev_cb; | 
|  | 1574 |  | 
|  | 1575 | list_for_each_entry(egdev_cb, &egdev->cb_list, list) | 
|  | 1576 | if (egdev_cb->cb == cb && egdev_cb->cb_priv == cb_priv) | 
|  | 1577 | return egdev_cb; | 
|  | 1578 | return NULL; | 
|  | 1579 | } | 
|  | 1580 |  | 
|  | 1581 | static int tcf_action_egdev_cb_call(struct tcf_action_egdev *egdev, | 
|  | 1582 | enum tc_setup_type type, | 
|  | 1583 | void *type_data, bool err_stop) | 
|  | 1584 | { | 
|  | 1585 | struct tcf_action_egdev_cb *egdev_cb; | 
|  | 1586 | int ok_count = 0; | 
|  | 1587 | int err; | 
|  | 1588 |  | 
|  | 1589 | list_for_each_entry(egdev_cb, &egdev->cb_list, list) { | 
|  | 1590 | err = egdev_cb->cb(type, type_data, egdev_cb->cb_priv); | 
|  | 1591 | if (err) { | 
|  | 1592 | if (err_stop) | 
|  | 1593 | return err; | 
|  | 1594 | } else { | 
|  | 1595 | ok_count++; | 
|  | 1596 | } | 
|  | 1597 | } | 
|  | 1598 | return ok_count; | 
|  | 1599 | } | 
|  | 1600 |  | 
|  | 1601 | static int tcf_action_egdev_cb_add(struct tcf_action_egdev *egdev, | 
|  | 1602 | tc_setup_cb_t *cb, void *cb_priv) | 
|  | 1603 | { | 
|  | 1604 | struct tcf_action_egdev_cb *egdev_cb; | 
|  | 1605 |  | 
|  | 1606 | egdev_cb = tcf_action_egdev_cb_lookup(egdev, cb, cb_priv); | 
|  | 1607 | if (WARN_ON(egdev_cb)) | 
|  | 1608 | return -EEXIST; | 
|  | 1609 | egdev_cb = kzalloc(sizeof(*egdev_cb), GFP_KERNEL); | 
|  | 1610 | if (!egdev_cb) | 
|  | 1611 | return -ENOMEM; | 
|  | 1612 | egdev_cb->cb = cb; | 
|  | 1613 | egdev_cb->cb_priv = cb_priv; | 
|  | 1614 | list_add(&egdev_cb->list, &egdev->cb_list); | 
|  | 1615 | return 0; | 
|  | 1616 | } | 
|  | 1617 |  | 
|  | 1618 | static void tcf_action_egdev_cb_del(struct tcf_action_egdev *egdev, | 
|  | 1619 | tc_setup_cb_t *cb, void *cb_priv) | 
|  | 1620 | { | 
|  | 1621 | struct tcf_action_egdev_cb *egdev_cb; | 
|  | 1622 |  | 
|  | 1623 | egdev_cb = tcf_action_egdev_cb_lookup(egdev, cb, cb_priv); | 
|  | 1624 | if (WARN_ON(!egdev_cb)) | 
|  | 1625 | return; | 
|  | 1626 | list_del(&egdev_cb->list); | 
|  | 1627 | kfree(egdev_cb); | 
|  | 1628 | } | 
|  | 1629 |  | 
|  | 1630 | static int __tc_setup_cb_egdev_register(const struct net_device *dev, | 
|  | 1631 | tc_setup_cb_t *cb, void *cb_priv) | 
|  | 1632 | { | 
|  | 1633 | struct tcf_action_egdev *egdev = tcf_action_egdev_get(dev); | 
|  | 1634 | int err; | 
|  | 1635 |  | 
|  | 1636 | if (!egdev) | 
|  | 1637 | return -ENOMEM; | 
|  | 1638 | err = tcf_action_egdev_cb_add(egdev, cb, cb_priv); | 
|  | 1639 | if (err) | 
|  | 1640 | goto err_cb_add; | 
|  | 1641 | return 0; | 
|  | 1642 |  | 
|  | 1643 | err_cb_add: | 
|  | 1644 | tcf_action_egdev_put(egdev); | 
|  | 1645 | return err; | 
|  | 1646 | } | 
|  | 1647 | int tc_setup_cb_egdev_register(const struct net_device *dev, | 
|  | 1648 | tc_setup_cb_t *cb, void *cb_priv) | 
|  | 1649 | { | 
|  | 1650 | int err; | 
|  | 1651 |  | 
|  | 1652 | rtnl_lock(); | 
|  | 1653 | err = __tc_setup_cb_egdev_register(dev, cb, cb_priv); | 
|  | 1654 | rtnl_unlock(); | 
|  | 1655 | return err; | 
|  | 1656 | } | 
|  | 1657 | EXPORT_SYMBOL_GPL(tc_setup_cb_egdev_register); | 
|  | 1658 |  | 
|  | 1659 | static void __tc_setup_cb_egdev_unregister(const struct net_device *dev, | 
|  | 1660 | tc_setup_cb_t *cb, void *cb_priv) | 
|  | 1661 | { | 
|  | 1662 | struct tcf_action_egdev *egdev = tcf_action_egdev_lookup(dev); | 
|  | 1663 |  | 
|  | 1664 | if (WARN_ON(!egdev)) | 
|  | 1665 | return; | 
|  | 1666 | tcf_action_egdev_cb_del(egdev, cb, cb_priv); | 
|  | 1667 | tcf_action_egdev_put(egdev); | 
|  | 1668 | } | 
|  | 1669 | void tc_setup_cb_egdev_unregister(const struct net_device *dev, | 
|  | 1670 | tc_setup_cb_t *cb, void *cb_priv) | 
|  | 1671 | { | 
|  | 1672 | rtnl_lock(); | 
|  | 1673 | __tc_setup_cb_egdev_unregister(dev, cb, cb_priv); | 
|  | 1674 | rtnl_unlock(); | 
|  | 1675 | } | 
|  | 1676 | EXPORT_SYMBOL_GPL(tc_setup_cb_egdev_unregister); | 
|  | 1677 |  | 
|  | 1678 | int tc_setup_cb_egdev_call(const struct net_device *dev, | 
|  | 1679 | enum tc_setup_type type, void *type_data, | 
|  | 1680 | bool err_stop) | 
|  | 1681 | { | 
|  | 1682 | struct tcf_action_egdev *egdev = tcf_action_egdev_lookup(dev); | 
|  | 1683 |  | 
|  | 1684 | if (!egdev) | 
|  | 1685 | return 0; | 
|  | 1686 | return tcf_action_egdev_cb_call(egdev, type, type_data, err_stop); | 
|  | 1687 | } | 
|  | 1688 | EXPORT_SYMBOL_GPL(tc_setup_cb_egdev_call); | 
|  | 1689 |  | 
|  | 1690 | static __net_init int tcf_action_net_init(struct net *net) | 
|  | 1691 | { | 
|  | 1692 | struct tcf_action_net *tan = net_generic(net, tcf_action_net_id); | 
|  | 1693 |  | 
|  | 1694 | return rhashtable_init(&tan->egdev_ht, &tcf_action_egdev_ht_params); | 
|  | 1695 | } | 
|  | 1696 |  | 
|  | 1697 | static void __net_exit tcf_action_net_exit(struct net *net) | 
|  | 1698 | { | 
|  | 1699 | struct tcf_action_net *tan = net_generic(net, tcf_action_net_id); | 
|  | 1700 |  | 
|  | 1701 | rhashtable_destroy(&tan->egdev_ht); | 
|  | 1702 | } | 
|  | 1703 |  | 
|  | 1704 | static struct pernet_operations tcf_action_net_ops = { | 
|  | 1705 | .init = tcf_action_net_init, | 
|  | 1706 | .exit = tcf_action_net_exit, | 
|  | 1707 | .id = &tcf_action_net_id, | 
|  | 1708 | .size = sizeof(struct tcf_action_net), | 
|  | 1709 | }; | 
|  | 1710 |  | 
|  | 1711 | static int __init tc_action_init(void) | 
|  | 1712 | { | 
|  | 1713 | int err; | 
|  | 1714 |  | 
|  | 1715 | err = register_pernet_subsys(&tcf_action_net_ops); | 
|  | 1716 | if (err) | 
|  | 1717 | return err; | 
|  | 1718 |  | 
|  | 1719 | rtnl_register(PF_UNSPEC, RTM_NEWACTION, tc_ctl_action, NULL, 0); | 
|  | 1720 | rtnl_register(PF_UNSPEC, RTM_DELACTION, tc_ctl_action, NULL, 0); | 
|  | 1721 | rtnl_register(PF_UNSPEC, RTM_GETACTION, tc_ctl_action, tc_dump_action, | 
|  | 1722 | 0); | 
|  | 1723 |  | 
|  | 1724 | return 0; | 
|  | 1725 | } | 
|  | 1726 |  | 
|  | 1727 | subsys_initcall(tc_action_init); |