blob: 189e3c5b3d098a0d224a8fb2a04b1bd76e47b3ac [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
2 * net/sched/police.c Input police filter.
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 * Authors: Alexey Kuznetsov, <kuznet@ms2.inr.ac.ru>
10 * J Hadi Salim (action changes)
11 */
12
Linus Torvalds1da177e2005-04-16 15:20:36 -070013#include <linux/module.h>
14#include <linux/types.h>
15#include <linux/kernel.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070016#include <linux/string.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070017#include <linux/errno.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070018#include <linux/skbuff.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070019#include <linux/rtnetlink.h>
20#include <linux/init.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +090021#include <linux/slab.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070022#include <net/act_api.h>
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -070023#include <net/netlink.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070024
Jiri Pirko0e243212013-02-12 00:12:06 +000025struct tcf_police {
26 struct tcf_common common;
27 int tcfp_result;
28 u32 tcfp_ewma_rate;
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +000029 s64 tcfp_burst;
Jiri Pirko0e243212013-02-12 00:12:06 +000030 u32 tcfp_mtu;
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +000031 s64 tcfp_toks;
32 s64 tcfp_ptoks;
33 s64 tcfp_mtu_ptoks;
34 s64 tcfp_t_c;
35 struct psched_ratecfg rate;
36 bool rate_present;
37 struct psched_ratecfg peak;
38 bool peak_present;
Jiri Pirko0e243212013-02-12 00:12:06 +000039};
40#define to_police(pc) \
41 container_of(pc, struct tcf_police, common)
42
David S. Millere9ce1cd2006-08-21 23:54:55 -070043#define POL_TAB_MASK 15
44static struct tcf_common *tcf_police_ht[POL_TAB_MASK + 1];
45static u32 police_idx_gen;
Linus Torvalds1da177e2005-04-16 15:20:36 -070046static DEFINE_RWLOCK(police_lock);
47
David S. Millere9ce1cd2006-08-21 23:54:55 -070048static struct tcf_hashinfo police_hash_info = {
49 .htab = tcf_police_ht,
50 .hmask = POL_TAB_MASK,
51 .lock = &police_lock,
52};
53
Patrick McHardy1e9b3d52006-11-30 19:54:05 -080054/* old policer structure from before tc actions */
Eric Dumazetcc7ec452011-01-19 19:26:56 +000055struct tc_police_compat {
Patrick McHardy1e9b3d52006-11-30 19:54:05 -080056 u32 index;
57 int action;
58 u32 limit;
59 u32 burst;
60 u32 mtu;
61 struct tc_ratespec rate;
62 struct tc_ratespec peakrate;
63};
64
Linus Torvalds1da177e2005-04-16 15:20:36 -070065/* Each policer is serialized by its individual spinlock */
66
Jamal Hadi Salim83b950c2006-04-06 22:24:22 -070067static int tcf_act_police_walker(struct sk_buff *skb, struct netlink_callback *cb,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +090068 int type, struct tc_action *a)
Linus Torvalds1da177e2005-04-16 15:20:36 -070069{
David S. Millere9ce1cd2006-08-21 23:54:55 -070070 struct tcf_common *p;
Linus Torvalds1da177e2005-04-16 15:20:36 -070071 int err = 0, index = -1, i = 0, s_i = 0, n_i = 0;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -080072 struct nlattr *nest;
Linus Torvalds1da177e2005-04-16 15:20:36 -070073
Jamal Hadi Salime1e992e2007-09-12 16:32:59 +020074 read_lock_bh(&police_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -070075
76 s_i = cb->args[0];
77
David S. Millere9ce1cd2006-08-21 23:54:55 -070078 for (i = 0; i < (POL_TAB_MASK + 1); i++) {
79 p = tcf_police_ht[tcf_hash(i, POL_TAB_MASK)];
Linus Torvalds1da177e2005-04-16 15:20:36 -070080
David S. Millere9ce1cd2006-08-21 23:54:55 -070081 for (; p; p = p->tcfc_next) {
Linus Torvalds1da177e2005-04-16 15:20:36 -070082 index++;
83 if (index < s_i)
84 continue;
85 a->priv = p;
86 a->order = index;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -080087 nest = nla_nest_start(skb, a->order);
88 if (nest == NULL)
89 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -070090 if (type == RTM_DELACTION)
91 err = tcf_action_dump_1(skb, a, 0, 1);
92 else
93 err = tcf_action_dump_1(skb, a, 0, 0);
94 if (err < 0) {
95 index--;
Patrick McHardy4b3550ef2008-01-23 20:34:11 -080096 nla_nest_cancel(skb, nest);
Linus Torvalds1da177e2005-04-16 15:20:36 -070097 goto done;
98 }
Patrick McHardy4b3550ef2008-01-23 20:34:11 -080099 nla_nest_end(skb, nest);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700100 n_i++;
101 }
102 }
103done:
Jamal Hadi Salime1e992e2007-09-12 16:32:59 +0200104 read_unlock_bh(&police_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700105 if (n_i)
106 cb->args[0] += n_i;
107 return n_i;
108
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800109nla_put_failure:
Patrick McHardy4b3550ef2008-01-23 20:34:11 -0800110 nla_nest_cancel(skb, nest);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700111 goto done;
112}
Linus Torvalds1da177e2005-04-16 15:20:36 -0700113
Patrick McHardyc3bc7cf2007-07-15 00:03:05 -0700114static void tcf_police_destroy(struct tcf_police *p)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700115{
David S. Millere9ce1cd2006-08-21 23:54:55 -0700116 unsigned int h = tcf_hash(p->tcf_index, POL_TAB_MASK);
117 struct tcf_common **p1p;
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900118
David S. Millere9ce1cd2006-08-21 23:54:55 -0700119 for (p1p = &tcf_police_ht[h]; *p1p; p1p = &(*p1p)->tcfc_next) {
120 if (*p1p == &p->common) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700121 write_lock_bh(&police_lock);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700122 *p1p = p->tcf_next;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700123 write_unlock_bh(&police_lock);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700124 gen_kill_estimator(&p->tcf_bstats,
125 &p->tcf_rate_est);
Eric Dumazetc7de2cf2010-06-09 02:09:23 +0000126 /*
127 * gen_estimator est_timer() might access p->tcf_lock
128 * or bstats, wait a RCU grace period before freeing p
129 */
Lai Jiangshan5957b1a2011-03-15 17:58:00 +0800130 kfree_rcu(p, tcf_rcu);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700131 return;
132 }
133 }
Ilpo Järvinen547b7922008-07-25 21:43:18 -0700134 WARN_ON(1);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700135}
136
Patrick McHardy53b2bf32008-01-23 20:36:30 -0800137static const struct nla_policy police_policy[TCA_POLICE_MAX + 1] = {
138 [TCA_POLICE_RATE] = { .len = TC_RTAB_SIZE },
139 [TCA_POLICE_PEAKRATE] = { .len = TC_RTAB_SIZE },
140 [TCA_POLICE_AVRATE] = { .type = NLA_U32 },
141 [TCA_POLICE_RESULT] = { .type = NLA_U32 },
142};
143
Benjamin LaHaisec1b52732013-01-14 05:15:39 +0000144static int tcf_act_police_locate(struct net *net, struct nlattr *nla,
145 struct nlattr *est, struct tc_action *a,
146 int ovr, int bind)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700147{
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000148 unsigned int h;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700149 int ret = 0, err;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800150 struct nlattr *tb[TCA_POLICE_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -0700151 struct tc_police *parm;
David S. Millere9ce1cd2006-08-21 23:54:55 -0700152 struct tcf_police *police;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700153 struct qdisc_rate_table *R_tab = NULL, *P_tab = NULL;
Patrick McHardy1e9b3d52006-11-30 19:54:05 -0800154 int size;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700155
Patrick McHardycee63722008-01-23 20:33:32 -0800156 if (nla == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700157 return -EINVAL;
158
Patrick McHardy53b2bf32008-01-23 20:36:30 -0800159 err = nla_parse_nested(tb, TCA_POLICE_MAX, nla, police_policy);
Patrick McHardycee63722008-01-23 20:33:32 -0800160 if (err < 0)
161 return err;
162
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800163 if (tb[TCA_POLICE_TBF] == NULL)
Patrick McHardy1e9b3d52006-11-30 19:54:05 -0800164 return -EINVAL;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800165 size = nla_len(tb[TCA_POLICE_TBF]);
Patrick McHardy1e9b3d52006-11-30 19:54:05 -0800166 if (size != sizeof(*parm) && size != sizeof(struct tc_police_compat))
Linus Torvalds1da177e2005-04-16 15:20:36 -0700167 return -EINVAL;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800168 parm = nla_data(tb[TCA_POLICE_TBF]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700169
David S. Millere9ce1cd2006-08-21 23:54:55 -0700170 if (parm->index) {
171 struct tcf_common *pc;
172
173 pc = tcf_hash_lookup(parm->index, &police_hash_info);
174 if (pc != NULL) {
175 a->priv = pc;
176 police = to_police(pc);
177 if (bind) {
178 police->tcf_bindcnt += 1;
179 police->tcf_refcnt += 1;
180 }
181 if (ovr)
182 goto override;
183 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700184 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700185 }
186
David S. Millere9ce1cd2006-08-21 23:54:55 -0700187 police = kzalloc(sizeof(*police), GFP_KERNEL);
188 if (police == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700189 return -ENOMEM;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700190 ret = ACT_P_CREATED;
David S. Millere9ce1cd2006-08-21 23:54:55 -0700191 police->tcf_refcnt = 1;
192 spin_lock_init(&police->tcf_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700193 if (bind)
David S. Millere9ce1cd2006-08-21 23:54:55 -0700194 police->tcf_bindcnt = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700195override:
196 if (parm->rate.rate) {
197 err = -ENOMEM;
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800198 R_tab = qdisc_get_rtab(&parm->rate, tb[TCA_POLICE_RATE]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700199 if (R_tab == NULL)
200 goto failure;
Stephen Hemmingerc1b56872008-11-25 21:14:06 -0800201
Linus Torvalds1da177e2005-04-16 15:20:36 -0700202 if (parm->peakrate.rate) {
203 P_tab = qdisc_get_rtab(&parm->peakrate,
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800204 tb[TCA_POLICE_PEAKRATE]);
Stephen Hemminger71bcb092008-11-25 21:13:31 -0800205 if (P_tab == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700206 goto failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700207 }
208 }
Stephen Hemminger71bcb092008-11-25 21:13:31 -0800209
David S. Millere9ce1cd2006-08-21 23:54:55 -0700210 spin_lock_bh(&police->tcf_lock);
Stephen Hemminger71bcb092008-11-25 21:13:31 -0800211 if (est) {
212 err = gen_replace_estimator(&police->tcf_bstats,
213 &police->tcf_rate_est,
214 &police->tcf_lock, est);
215 if (err)
216 goto failure_unlock;
Jarek Poplawskia883bf52009-03-04 17:38:10 -0800217 } else if (tb[TCA_POLICE_AVRATE] &&
218 (ret == ACT_P_CREATED ||
219 !gen_estimator_active(&police->tcf_bstats,
220 &police->tcf_rate_est))) {
221 err = -EINVAL;
222 goto failure_unlock;
Stephen Hemminger71bcb092008-11-25 21:13:31 -0800223 }
224
225 /* No failure allowed after this point */
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000226 police->tcfp_mtu = parm->mtu;
227 if (police->tcfp_mtu == 0) {
228 police->tcfp_mtu = ~0;
229 if (R_tab)
230 police->tcfp_mtu = 255 << R_tab->rate.cell_log;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700231 }
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000232 if (R_tab) {
233 police->rate_present = true;
Eric Dumazet01cb71d2013-06-02 13:55:05 +0000234 psched_ratecfg_precompute(&police->rate, &R_tab->rate);
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000235 qdisc_put_rtab(R_tab);
236 } else {
237 police->rate_present = false;
238 }
239 if (P_tab) {
240 police->peak_present = true;
Eric Dumazet01cb71d2013-06-02 13:55:05 +0000241 psched_ratecfg_precompute(&police->peak, &P_tab->rate);
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000242 qdisc_put_rtab(P_tab);
243 } else {
244 police->peak_present = false;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700245 }
246
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800247 if (tb[TCA_POLICE_RESULT])
Patrick McHardy1587bac2008-01-23 20:35:03 -0800248 police->tcfp_result = nla_get_u32(tb[TCA_POLICE_RESULT]);
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000249 police->tcfp_burst = PSCHED_TICKS2NS(parm->burst);
250 police->tcfp_toks = police->tcfp_burst;
251 if (police->peak_present) {
252 police->tcfp_mtu_ptoks = (s64) psched_l2t_ns(&police->peak,
253 police->tcfp_mtu);
254 police->tcfp_ptoks = police->tcfp_mtu_ptoks;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700255 }
David S. Millere9ce1cd2006-08-21 23:54:55 -0700256 police->tcf_action = parm->action;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700257
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800258 if (tb[TCA_POLICE_AVRATE])
Patrick McHardy1587bac2008-01-23 20:35:03 -0800259 police->tcfp_ewma_rate = nla_get_u32(tb[TCA_POLICE_AVRATE]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700260
David S. Millere9ce1cd2006-08-21 23:54:55 -0700261 spin_unlock_bh(&police->tcf_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700262 if (ret != ACT_P_CREATED)
263 return ret;
264
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000265 police->tcfp_t_c = ktime_to_ns(ktime_get());
David S. Millere9ce1cd2006-08-21 23:54:55 -0700266 police->tcf_index = parm->index ? parm->index :
267 tcf_hash_new_index(&police_idx_gen, &police_hash_info);
268 h = tcf_hash(police->tcf_index, POL_TAB_MASK);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700269 write_lock_bh(&police_lock);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700270 police->tcf_next = tcf_police_ht[h];
271 tcf_police_ht[h] = &police->common;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700272 write_unlock_bh(&police_lock);
273
David S. Millere9ce1cd2006-08-21 23:54:55 -0700274 a->priv = police;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700275 return ret;
276
Stephen Hemminger71bcb092008-11-25 21:13:31 -0800277failure_unlock:
278 spin_unlock_bh(&police->tcf_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700279failure:
Stephen Hemminger71bcb092008-11-25 21:13:31 -0800280 if (P_tab)
281 qdisc_put_rtab(P_tab);
282 if (R_tab)
283 qdisc_put_rtab(R_tab);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700284 if (ret == ACT_P_CREATED)
David S. Millere9ce1cd2006-08-21 23:54:55 -0700285 kfree(police);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700286 return err;
287}
288
289static int tcf_act_police_cleanup(struct tc_action *a, int bind)
290{
David S. Millere9ce1cd2006-08-21 23:54:55 -0700291 struct tcf_police *p = a->priv;
Patrick McHardyc3bc7cf2007-07-15 00:03:05 -0700292 int ret = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700293
Patrick McHardyc3bc7cf2007-07-15 00:03:05 -0700294 if (p != NULL) {
295 if (bind)
296 p->tcf_bindcnt--;
297
298 p->tcf_refcnt--;
299 if (p->tcf_refcnt <= 0 && !p->tcf_bindcnt) {
300 tcf_police_destroy(p);
301 ret = 1;
302 }
303 }
304 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700305}
306
Eric Dumazetdc7f9f62011-07-05 23:25:42 +0000307static int tcf_act_police(struct sk_buff *skb, const struct tc_action *a,
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +0900308 struct tcf_result *res)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700309{
David S. Millere9ce1cd2006-08-21 23:54:55 -0700310 struct tcf_police *police = a->priv;
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000311 s64 now;
312 s64 toks;
313 s64 ptoks = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700314
David S. Millere9ce1cd2006-08-21 23:54:55 -0700315 spin_lock(&police->tcf_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700316
Eric Dumazetbfe0d022011-01-09 08:30:54 +0000317 bstats_update(&police->tcf_bstats, skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700318
David S. Millere9ce1cd2006-08-21 23:54:55 -0700319 if (police->tcfp_ewma_rate &&
320 police->tcf_rate_est.bps >= police->tcfp_ewma_rate) {
321 police->tcf_qstats.overlimits++;
Jarek Poplawskib9647582009-06-16 08:33:55 +0000322 if (police->tcf_action == TC_ACT_SHOT)
323 police->tcf_qstats.drops++;
David S. Millere9ce1cd2006-08-21 23:54:55 -0700324 spin_unlock(&police->tcf_lock);
325 return police->tcf_action;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700326 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700327
Jussi Kivilinna0abf77e2008-07-20 00:08:27 -0700328 if (qdisc_pkt_len(skb) <= police->tcfp_mtu) {
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000329 if (!police->rate_present) {
David S. Millere9ce1cd2006-08-21 23:54:55 -0700330 spin_unlock(&police->tcf_lock);
331 return police->tcfp_result;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700332 }
333
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000334 now = ktime_to_ns(ktime_get());
335 toks = min_t(s64, now - police->tcfp_t_c,
336 police->tcfp_burst);
337 if (police->peak_present) {
David S. Millere9ce1cd2006-08-21 23:54:55 -0700338 ptoks = toks + police->tcfp_ptoks;
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000339 if (ptoks > police->tcfp_mtu_ptoks)
340 ptoks = police->tcfp_mtu_ptoks;
341 ptoks -= (s64) psched_l2t_ns(&police->peak,
342 qdisc_pkt_len(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700343 }
David S. Millere9ce1cd2006-08-21 23:54:55 -0700344 toks += police->tcfp_toks;
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000345 if (toks > police->tcfp_burst)
David S. Millere9ce1cd2006-08-21 23:54:55 -0700346 toks = police->tcfp_burst;
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000347 toks -= (s64) psched_l2t_ns(&police->rate, qdisc_pkt_len(skb));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700348 if ((toks|ptoks) >= 0) {
David S. Millere9ce1cd2006-08-21 23:54:55 -0700349 police->tcfp_t_c = now;
350 police->tcfp_toks = toks;
351 police->tcfp_ptoks = ptoks;
352 spin_unlock(&police->tcf_lock);
353 return police->tcfp_result;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700354 }
355 }
356
David S. Millere9ce1cd2006-08-21 23:54:55 -0700357 police->tcf_qstats.overlimits++;
Jarek Poplawskib9647582009-06-16 08:33:55 +0000358 if (police->tcf_action == TC_ACT_SHOT)
359 police->tcf_qstats.drops++;
David S. Millere9ce1cd2006-08-21 23:54:55 -0700360 spin_unlock(&police->tcf_lock);
361 return police->tcf_action;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700362}
363
364static int
365tcf_act_police_dump(struct sk_buff *skb, struct tc_action *a, int bind, int ref)
366{
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -0700367 unsigned char *b = skb_tail_pointer(skb);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700368 struct tcf_police *police = a->priv;
Jeff Mahoney0f04cfd2010-08-31 13:21:42 +0000369 struct tc_police opt = {
370 .index = police->tcf_index,
371 .action = police->tcf_action,
372 .mtu = police->tcfp_mtu,
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000373 .burst = PSCHED_NS2TICKS(police->tcfp_burst),
Jeff Mahoney0f04cfd2010-08-31 13:21:42 +0000374 .refcnt = police->tcf_refcnt - ref,
375 .bindcnt = police->tcf_bindcnt - bind,
376 };
Linus Torvalds1da177e2005-04-16 15:20:36 -0700377
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000378 if (police->rate_present)
Eric Dumazet01cb71d2013-06-02 13:55:05 +0000379 psched_ratecfg_getrate(&opt.rate, &police->rate);
Jiri Pirkoc6d14ff2013-02-12 00:12:07 +0000380 if (police->peak_present)
Eric Dumazet01cb71d2013-06-02 13:55:05 +0000381 psched_ratecfg_getrate(&opt.peakrate, &police->peak);
David S. Miller1b34ec42012-03-29 05:11:39 -0400382 if (nla_put(skb, TCA_POLICE_TBF, sizeof(opt), &opt))
383 goto nla_put_failure;
384 if (police->tcfp_result &&
385 nla_put_u32(skb, TCA_POLICE_RESULT, police->tcfp_result))
386 goto nla_put_failure;
387 if (police->tcfp_ewma_rate &&
388 nla_put_u32(skb, TCA_POLICE_AVRATE, police->tcfp_ewma_rate))
389 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700390 return skb->len;
391
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800392nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -0700393 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700394 return -1;
395}
396
397MODULE_AUTHOR("Alexey Kuznetsov");
398MODULE_DESCRIPTION("Policing actions");
399MODULE_LICENSE("GPL");
400
401static struct tc_action_ops act_police_ops = {
402 .kind = "police",
David S. Millere9ce1cd2006-08-21 23:54:55 -0700403 .hinfo = &police_hash_info,
Linus Torvalds1da177e2005-04-16 15:20:36 -0700404 .type = TCA_ID_POLICE,
405 .capab = TCA_CAP_NONE,
406 .owner = THIS_MODULE,
407 .act = tcf_act_police,
408 .dump = tcf_act_police_dump,
409 .cleanup = tcf_act_police_cleanup,
David S. Millere9ce1cd2006-08-21 23:54:55 -0700410 .lookup = tcf_hash_search,
Linus Torvalds1da177e2005-04-16 15:20:36 -0700411 .init = tcf_act_police_locate,
Jamal Hadi Salim83b950c2006-04-06 22:24:22 -0700412 .walk = tcf_act_police_walker
Linus Torvalds1da177e2005-04-16 15:20:36 -0700413};
414
415static int __init
416police_init_module(void)
417{
418 return tcf_register_action(&act_police_ops);
419}
420
421static void __exit
422police_cleanup_module(void)
423{
424 tcf_unregister_action(&act_police_ops);
425}
426
427module_init(police_init_module);
428module_exit(police_cleanup_module);