blob: ec5cc8435238695d5fda1282ee8ae7e1890dd2a7 [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
Jiri Pirko0c6965d2014-11-05 20:51:51 +01002 * net/sched/act_gact.c Generic actions
Linus Torvalds1da177e2005-04-16 15:20:36 -07003 *
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 * copyright Jamal Hadi Salim (2002-4)
10 *
11 */
12
Linus Torvalds1da177e2005-04-16 15:20:36 -070013#include <linux/types.h>
14#include <linux/kernel.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070015#include <linux/string.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070016#include <linux/errno.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070017#include <linux/skbuff.h>
18#include <linux/rtnetlink.h>
19#include <linux/module.h>
20#include <linux/init.h>
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -070021#include <net/netlink.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070022#include <net/pkt_sched.h>
23#include <linux/tc_act/tc_gact.h>
24#include <net/tc_act/tc_gact.h>
25
David S. Millere9ce1cd2006-08-21 23:54:55 -070026#define GACT_TAB_MASK 15
Linus Torvalds1da177e2005-04-16 15:20:36 -070027
WANG Congddf97cc2016-02-22 15:57:53 -080028static int gact_net_id;
29
Linus Torvalds1da177e2005-04-16 15:20:36 -070030#ifdef CONFIG_GACT_PROB
David S. Millere9ce1cd2006-08-21 23:54:55 -070031static int gact_net_rand(struct tcf_gact *gact)
Linus Torvalds1da177e2005-04-16 15:20:36 -070032{
Eric Dumazetcef5ecf2015-07-06 05:18:05 -070033 smp_rmb(); /* coupled with smp_wmb() in tcf_gact_init() */
34 if (prandom_u32() % gact->tcfg_pval)
David S. Millere9ce1cd2006-08-21 23:54:55 -070035 return gact->tcf_action;
36 return gact->tcfg_paction;
Linus Torvalds1da177e2005-04-16 15:20:36 -070037}
38
David S. Millere9ce1cd2006-08-21 23:54:55 -070039static int gact_determ(struct tcf_gact *gact)
Linus Torvalds1da177e2005-04-16 15:20:36 -070040{
Eric Dumazetcc6510a2015-07-06 05:18:06 -070041 u32 pack = atomic_inc_return(&gact->packets);
42
Eric Dumazetcef5ecf2015-07-06 05:18:05 -070043 smp_rmb(); /* coupled with smp_wmb() in tcf_gact_init() */
Eric Dumazetcc6510a2015-07-06 05:18:06 -070044 if (pack % gact->tcfg_pval)
David S. Millere9ce1cd2006-08-21 23:54:55 -070045 return gact->tcf_action;
46 return gact->tcfg_paction;
Linus Torvalds1da177e2005-04-16 15:20:36 -070047}
48
David S. Millere9ce1cd2006-08-21 23:54:55 -070049typedef int (*g_rand)(struct tcf_gact *gact);
Eric Dumazetcc7ec452011-01-19 19:26:56 +000050static g_rand gact_rand[MAX_RAND] = { NULL, gact_net_rand, gact_determ };
David S. Millere9ce1cd2006-08-21 23:54:55 -070051#endif /* CONFIG_GACT_PROB */
Linus Torvalds1da177e2005-04-16 15:20:36 -070052
Patrick McHardy53b2bf32008-01-23 20:36:30 -080053static const struct nla_policy gact_policy[TCA_GACT_MAX + 1] = {
54 [TCA_GACT_PARMS] = { .len = sizeof(struct tc_gact) },
55 [TCA_GACT_PROB] = { .len = sizeof(struct tc_gact_p) },
56};
57
Benjamin LaHaisec1b52732013-01-14 05:15:39 +000058static int tcf_gact_init(struct net *net, struct nlattr *nla,
59 struct nlattr *est, struct tc_action *a,
60 int ovr, int bind)
Linus Torvalds1da177e2005-04-16 15:20:36 -070061{
WANG Congddf97cc2016-02-22 15:57:53 -080062 struct tc_action_net *tn = net_generic(net, gact_net_id);
Patrick McHardy7ba699c2008-01-22 22:11:50 -080063 struct nlattr *tb[TCA_GACT_MAX + 1];
Linus Torvalds1da177e2005-04-16 15:20:36 -070064 struct tc_gact *parm;
David S. Millere9ce1cd2006-08-21 23:54:55 -070065 struct tcf_gact *gact;
Linus Torvalds1da177e2005-04-16 15:20:36 -070066 int ret = 0;
Patrick McHardycee63722008-01-23 20:33:32 -080067 int err;
Hiroaki SHIMODA696ecdc2012-08-03 19:57:52 +090068#ifdef CONFIG_GACT_PROB
69 struct tc_gact_p *p_parm = NULL;
70#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -070071
Patrick McHardycee63722008-01-23 20:33:32 -080072 if (nla == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -070073 return -EINVAL;
74
Patrick McHardy53b2bf32008-01-23 20:36:30 -080075 err = nla_parse_nested(tb, TCA_GACT_MAX, nla, gact_policy);
Patrick McHardycee63722008-01-23 20:33:32 -080076 if (err < 0)
77 return err;
78
Patrick McHardy53b2bf32008-01-23 20:36:30 -080079 if (tb[TCA_GACT_PARMS] == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -070080 return -EINVAL;
Patrick McHardy7ba699c2008-01-22 22:11:50 -080081 parm = nla_data(tb[TCA_GACT_PARMS]);
Linus Torvalds1da177e2005-04-16 15:20:36 -070082
Patrick McHardy53b2bf32008-01-23 20:36:30 -080083#ifndef CONFIG_GACT_PROB
Patrick McHardy7ba699c2008-01-22 22:11:50 -080084 if (tb[TCA_GACT_PROB] != NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -070085 return -EOPNOTSUPP;
Hiroaki SHIMODA696ecdc2012-08-03 19:57:52 +090086#else
87 if (tb[TCA_GACT_PROB]) {
88 p_parm = nla_data(tb[TCA_GACT_PROB]);
89 if (p_parm->ptype >= MAX_RAND)
90 return -EINVAL;
91 }
Linus Torvalds1da177e2005-04-16 15:20:36 -070092#endif
93
WANG Congddf97cc2016-02-22 15:57:53 -080094 if (!tcf_hash_check(tn, parm->index, a, bind)) {
95 ret = tcf_hash_create(tn, parm->index, est, a,
96 sizeof(*gact), bind, true);
WANG Cong86062032014-02-11 17:07:31 -080097 if (ret)
98 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -070099 ret = ACT_P_CREATED;
100 } else {
Jamal Hadi Salim1a293212013-12-23 08:02:11 -0500101 if (bind)/* dont override defaults */
102 return 0;
WANG Cong86062032014-02-11 17:07:31 -0800103 tcf_hash_release(a, bind);
Jamal Hadi Salim1a293212013-12-23 08:02:11 -0500104 if (!ovr)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700105 return -EEXIST;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700106 }
107
WANG Cong86062032014-02-11 17:07:31 -0800108 gact = to_gact(a);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700109
Eric Dumazet56e5d1c2015-07-06 05:18:08 -0700110 ASSERT_RTNL();
David S. Millere9ce1cd2006-08-21 23:54:55 -0700111 gact->tcf_action = parm->action;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700112#ifdef CONFIG_GACT_PROB
Hiroaki SHIMODA696ecdc2012-08-03 19:57:52 +0900113 if (p_parm) {
David S. Millere9ce1cd2006-08-21 23:54:55 -0700114 gact->tcfg_paction = p_parm->paction;
Eric Dumazetcef5ecf2015-07-06 05:18:05 -0700115 gact->tcfg_pval = max_t(u16, 1, p_parm->pval);
116 /* Make sure tcfg_pval is written before tcfg_ptype
117 * coupled with smp_rmb() in gact_net_rand() & gact_determ()
118 */
119 smp_wmb();
David S. Millere9ce1cd2006-08-21 23:54:55 -0700120 gact->tcfg_ptype = p_parm->ptype;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700121 }
122#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -0700123 if (ret == ACT_P_CREATED)
WANG Congddf97cc2016-02-22 15:57:53 -0800124 tcf_hash_insert(tn, a);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700125 return ret;
126}
127
Eric Dumazetdc7f9f62011-07-05 23:25:42 +0000128static int tcf_gact(struct sk_buff *skb, const struct tc_action *a,
129 struct tcf_result *res)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700130{
David S. Millere9ce1cd2006-08-21 23:54:55 -0700131 struct tcf_gact *gact = a->priv;
Eric Dumazet56e5d1c2015-07-06 05:18:08 -0700132 int action = READ_ONCE(gact->tcf_action);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700133
Linus Torvalds1da177e2005-04-16 15:20:36 -0700134#ifdef CONFIG_GACT_PROB
Eric Dumazet8f2ae965b2015-07-06 05:18:07 -0700135 {
136 u32 ptype = READ_ONCE(gact->tcfg_ptype);
137
138 if (ptype)
139 action = gact_rand[ptype](gact);
140 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700141#endif
Eric Dumazet56e5d1c2015-07-06 05:18:08 -0700142 bstats_cpu_update(this_cpu_ptr(gact->common.cpu_bstats), skb);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700143 if (action == TC_ACT_SHOT)
Eric Dumazet56e5d1c2015-07-06 05:18:08 -0700144 qstats_drop_inc(this_cpu_ptr(gact->common.cpu_qstats));
145
146 tcf_lastuse_update(&gact->tcf_tm);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700147
148 return action;
149}
150
Amir Vadai9fea47d2016-05-13 12:55:36 +0000151static void tcf_gact_stats_update(struct tc_action *a, u64 bytes, u32 packets,
152 u64 lastuse)
153{
154 struct tcf_gact *gact = a->priv;
155 int action = READ_ONCE(gact->tcf_action);
156 struct tcf_t *tm = &gact->tcf_tm;
157
158 _bstats_cpu_update(this_cpu_ptr(gact->common.cpu_bstats), bytes, packets);
159 if (action == TC_ACT_SHOT)
160 this_cpu_ptr(gact->common.cpu_qstats)->drops += packets;
161
162 tm->lastuse = lastuse;
163}
164
David S. Millere9ce1cd2006-08-21 23:54:55 -0700165static int tcf_gact_dump(struct sk_buff *skb, struct tc_action *a, int bind, int ref)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700166{
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -0700167 unsigned char *b = skb_tail_pointer(skb);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700168 struct tcf_gact *gact = a->priv;
Eric Dumazet1c40be12010-08-16 20:04:22 +0000169 struct tc_gact opt = {
170 .index = gact->tcf_index,
171 .refcnt = gact->tcf_refcnt - ref,
172 .bindcnt = gact->tcf_bindcnt - bind,
173 .action = gact->tcf_action,
174 };
Linus Torvalds1da177e2005-04-16 15:20:36 -0700175 struct tcf_t t;
176
David S. Miller1b34ec42012-03-29 05:11:39 -0400177 if (nla_put(skb, TCA_GACT_PARMS, sizeof(opt), &opt))
178 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700179#ifdef CONFIG_GACT_PROB
David S. Millere9ce1cd2006-08-21 23:54:55 -0700180 if (gact->tcfg_ptype) {
Eric Dumazet1c40be12010-08-16 20:04:22 +0000181 struct tc_gact_p p_opt = {
182 .paction = gact->tcfg_paction,
183 .pval = gact->tcfg_pval,
184 .ptype = gact->tcfg_ptype,
185 };
186
David S. Miller1b34ec42012-03-29 05:11:39 -0400187 if (nla_put(skb, TCA_GACT_PROB, sizeof(p_opt), &p_opt))
188 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700189 }
190#endif
David S. Millere9ce1cd2006-08-21 23:54:55 -0700191 t.install = jiffies_to_clock_t(jiffies - gact->tcf_tm.install);
192 t.lastuse = jiffies_to_clock_t(jiffies - gact->tcf_tm.lastuse);
193 t.expires = jiffies_to_clock_t(gact->tcf_tm.expires);
Nicolas Dichtel98545182016-04-26 10:06:18 +0200194 if (nla_put_64bit(skb, TCA_GACT_TM, sizeof(t), &t, TCA_GACT_PAD))
David S. Miller1b34ec42012-03-29 05:11:39 -0400195 goto nla_put_failure;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700196 return skb->len;
197
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800198nla_put_failure:
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -0700199 nlmsg_trim(skb, b);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700200 return -1;
201}
202
WANG Congddf97cc2016-02-22 15:57:53 -0800203static int tcf_gact_walker(struct net *net, struct sk_buff *skb,
204 struct netlink_callback *cb, int type,
205 struct tc_action *a)
206{
207 struct tc_action_net *tn = net_generic(net, gact_net_id);
208
209 return tcf_generic_walker(tn, skb, cb, type, a);
210}
211
212static int tcf_gact_search(struct net *net, struct tc_action *a, u32 index)
213{
214 struct tc_action_net *tn = net_generic(net, gact_net_id);
215
216 return tcf_hash_search(tn, a, index);
217}
218
Linus Torvalds1da177e2005-04-16 15:20:36 -0700219static struct tc_action_ops act_gact_ops = {
220 .kind = "gact",
221 .type = TCA_ACT_GACT,
Linus Torvalds1da177e2005-04-16 15:20:36 -0700222 .owner = THIS_MODULE,
223 .act = tcf_gact,
Amir Vadai9fea47d2016-05-13 12:55:36 +0000224 .stats_update = tcf_gact_stats_update,
Linus Torvalds1da177e2005-04-16 15:20:36 -0700225 .dump = tcf_gact_dump,
Linus Torvalds1da177e2005-04-16 15:20:36 -0700226 .init = tcf_gact_init,
WANG Congddf97cc2016-02-22 15:57:53 -0800227 .walk = tcf_gact_walker,
228 .lookup = tcf_gact_search,
229};
230
231static __net_init int gact_init_net(struct net *net)
232{
233 struct tc_action_net *tn = net_generic(net, gact_net_id);
234
235 return tc_action_net_init(tn, &act_gact_ops, GACT_TAB_MASK);
236}
237
238static void __net_exit gact_exit_net(struct net *net)
239{
240 struct tc_action_net *tn = net_generic(net, gact_net_id);
241
242 tc_action_net_exit(tn);
243}
244
245static struct pernet_operations gact_net_ops = {
246 .init = gact_init_net,
247 .exit = gact_exit_net,
248 .id = &gact_net_id,
249 .size = sizeof(struct tc_action_net),
Linus Torvalds1da177e2005-04-16 15:20:36 -0700250};
251
252MODULE_AUTHOR("Jamal Hadi Salim(2002-4)");
253MODULE_DESCRIPTION("Generic Classifier actions");
254MODULE_LICENSE("GPL");
255
David S. Millere9ce1cd2006-08-21 23:54:55 -0700256static int __init gact_init_module(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700257{
258#ifdef CONFIG_GACT_PROB
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000259 pr_info("GACT probability on\n");
Linus Torvalds1da177e2005-04-16 15:20:36 -0700260#else
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000261 pr_info("GACT probability NOT on\n");
Linus Torvalds1da177e2005-04-16 15:20:36 -0700262#endif
WANG Congddf97cc2016-02-22 15:57:53 -0800263
264 return tcf_register_action(&act_gact_ops, &gact_net_ops);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700265}
266
David S. Millere9ce1cd2006-08-21 23:54:55 -0700267static void __exit gact_cleanup_module(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700268{
WANG Congddf97cc2016-02-22 15:57:53 -0800269 tcf_unregister_action(&act_gact_ops, &gact_net_ops);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700270}
271
272module_init(gact_init_module);
273module_exit(gact_cleanup_module);