blob: 6120e56117ca14d2abeb855e20beb33ae7cd9896 [file] [log] [blame]
Thomas Gleixner2874c5f2019-05-27 08:55:01 +02001// SPDX-License-Identifier: GPL-2.0-or-later
Jamal Hadi Salimdb753072005-04-24 20:10:16 -07002/*
Jiri Pirko0c6965d2014-11-05 20:51:51 +01003 * net/sched/act_simple.c Simple example of an action
Jamal Hadi Salimdb753072005-04-24 20:10:16 -07004 *
Jamal Hadi Salimfa1b1cf2008-05-05 00:22:35 -07005 * Authors: Jamal Hadi Salim (2005-8)
Jamal Hadi Salimdb753072005-04-24 20:10:16 -07006 */
7
David S. Millercbdbf002005-04-25 12:15:01 -07008#include <linux/module.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +09009#include <linux/slab.h>
David S. Millercbdbf002005-04-25 12:15:01 -070010#include <linux/init.h>
Jamal Hadi Salimdb753072005-04-24 20:10:16 -070011#include <linux/kernel.h>
Jamal Hadi Salimdb753072005-04-24 20:10:16 -070012#include <linux/skbuff.h>
13#include <linux/rtnetlink.h>
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -070014#include <net/netlink.h>
Jamal Hadi Salimdb753072005-04-24 20:10:16 -070015#include <net/pkt_sched.h>
Davide Caratti4b006b02019-03-20 15:00:10 +010016#include <net/pkt_cls.h>
Jamal Hadi Salimdb753072005-04-24 20:10:16 -070017
Jamal Hadi Salimdb753072005-04-24 20:10:16 -070018#include <linux/tc_act/tc_defact.h>
19#include <net/tc_act/tc_defact.h>
20
Alexey Dobriyanc7d03a02016-11-17 04:58:21 +030021static unsigned int simp_net_id;
WANG Conga85a9702016-07-25 16:09:41 -070022static struct tc_action_ops act_simp_ops;
WANG Congddf97cc2016-02-22 15:57:53 -080023
Jamal Hadi Salimfa1b1cf2008-05-05 00:22:35 -070024#define SIMP_MAX_DATA 32
Jamal Hadi Salim798de372018-08-12 09:34:57 -040025static int tcf_simp_act(struct sk_buff *skb, const struct tc_action *a,
26 struct tcf_result *res)
Jamal Hadi Salimdb753072005-04-24 20:10:16 -070027{
WANG Conga85a9702016-07-25 16:09:41 -070028 struct tcf_defact *d = to_defact(a);
Jamal Hadi Salimdb753072005-04-24 20:10:16 -070029
David S. Millere9ce1cd2006-08-21 23:54:55 -070030 spin_lock(&d->tcf_lock);
Jamal Hadi Salim9c4a4e42016-06-06 06:32:53 -040031 tcf_lastuse_update(&d->tcf_tm);
Eric Dumazetbfe0d022011-01-09 08:30:54 +000032 bstats_update(&d->tcf_bstats, skb);
Jamal Hadi Salimdb753072005-04-24 20:10:16 -070033
YOSHIFUJI Hideaki10297b92007-02-09 23:25:16 +090034 /* print policy string followed by _ then packet count
35 * Example if this was the 3rd packet and the string was "hello"
36 * then it would look like "hello_3" (without quotes)
Eric Dumazetcc7ec452011-01-19 19:26:56 +000037 */
stephen hemminger6ff9c362010-05-12 06:37:05 +000038 pr_info("simple: %s_%d\n",
David S. Millere9ce1cd2006-08-21 23:54:55 -070039 (char *)d->tcfd_defdata, d->tcf_bstats.packets);
40 spin_unlock(&d->tcf_lock);
41 return d->tcf_action;
42}
43
Cong Wang9a63b252017-12-05 12:53:07 -080044static void tcf_simp_release(struct tc_action *a)
David S. Millere9ce1cd2006-08-21 23:54:55 -070045{
WANG Cong86062032014-02-11 17:07:31 -080046 struct tcf_defact *d = to_defact(a);
WANG Conga5b5c952014-02-11 17:07:32 -080047 kfree(d->tcfd_defdata);
David S. Millere9ce1cd2006-08-21 23:54:55 -070048}
49
Davide Caratti8d499532018-06-08 05:02:31 +020050static int alloc_defdata(struct tcf_defact *d, const struct nlattr *defdata)
David S. Millere9ce1cd2006-08-21 23:54:55 -070051{
Dan Carpenter0eff6832010-07-14 17:56:37 -070052 d->tcfd_defdata = kzalloc(SIMP_MAX_DATA, GFP_KERNEL);
David S. Millere9ce1cd2006-08-21 23:54:55 -070053 if (unlikely(!d->tcfd_defdata))
54 return -ENOMEM;
Davide Caratti8d499532018-06-08 05:02:31 +020055 nla_strlcpy(d->tcfd_defdata, defdata, SIMP_MAX_DATA);
David S. Millere9ce1cd2006-08-21 23:54:55 -070056 return 0;
57}
58
Davide Caratti4b006b02019-03-20 15:00:10 +010059static int reset_policy(struct tc_action *a, const struct nlattr *defdata,
60 struct tc_defact *p, struct tcf_proto *tp,
61 struct netlink_ext_ack *extack)
David S. Millere9ce1cd2006-08-21 23:54:55 -070062{
Davide Caratti4b006b02019-03-20 15:00:10 +010063 struct tcf_chain *goto_ch = NULL;
64 struct tcf_defact *d;
65 int err;
66
67 err = tcf_action_check_ctrlact(p->action, tp, &goto_ch, extack);
68 if (err < 0)
69 return err;
70 d = to_defact(a);
Jamal Hadi Salim9d1045a2008-05-06 00:10:24 -070071 spin_lock_bh(&d->tcf_lock);
Davide Caratti4b006b02019-03-20 15:00:10 +010072 goto_ch = tcf_action_set_ctrlact(a, p->action, goto_ch);
Jamal Hadi Salim9d1045a2008-05-06 00:10:24 -070073 memset(d->tcfd_defdata, 0, SIMP_MAX_DATA);
Davide Caratti8d499532018-06-08 05:02:31 +020074 nla_strlcpy(d->tcfd_defdata, defdata, SIMP_MAX_DATA);
Jamal Hadi Salim9d1045a2008-05-06 00:10:24 -070075 spin_unlock_bh(&d->tcf_lock);
Davide Caratti4b006b02019-03-20 15:00:10 +010076 if (goto_ch)
77 tcf_chain_put_by_act(goto_ch);
78 return 0;
David S. Millere9ce1cd2006-08-21 23:54:55 -070079}
80
Patrick McHardy53b2bf32008-01-23 20:36:30 -080081static const struct nla_policy simple_policy[TCA_DEF_MAX + 1] = {
82 [TCA_DEF_PARMS] = { .len = sizeof(struct tc_defact) },
Jamal Hadi Salimfa1b1cf2008-05-05 00:22:35 -070083 [TCA_DEF_DATA] = { .type = NLA_STRING, .len = SIMP_MAX_DATA },
Patrick McHardy53b2bf32008-01-23 20:36:30 -080084};
85
Benjamin LaHaisec1b52732013-01-14 05:15:39 +000086static int tcf_simp_init(struct net *net, struct nlattr *nla,
WANG Conga85a9702016-07-25 16:09:41 -070087 struct nlattr *est, struct tc_action **a,
Vlad Buslov789871b2018-07-05 17:24:25 +030088 int ovr, int bind, bool rtnl_held,
Davide Caratti85d09662019-03-20 14:59:59 +010089 struct tcf_proto *tp, struct netlink_ext_ack *extack)
David S. Millere9ce1cd2006-08-21 23:54:55 -070090{
WANG Congddf97cc2016-02-22 15:57:53 -080091 struct tc_action_net *tn = net_generic(net, simp_net_id);
Patrick McHardy7ba699c2008-01-22 22:11:50 -080092 struct nlattr *tb[TCA_DEF_MAX + 1];
Davide Caratti4b006b02019-03-20 15:00:10 +010093 struct tcf_chain *goto_ch = NULL;
David S. Millere9ce1cd2006-08-21 23:54:55 -070094 struct tc_defact *parm;
95 struct tcf_defact *d;
WANG Congb2313072016-06-13 13:46:28 -070096 bool exists = false;
97 int ret = 0, err;
Dmytro Linkin7be8ef22019-08-01 13:02:51 +000098 u32 index;
David S. Millere9ce1cd2006-08-21 23:54:55 -070099
Patrick McHardycee63722008-01-23 20:33:32 -0800100 if (nla == NULL)
David S. Millere9ce1cd2006-08-21 23:54:55 -0700101 return -EINVAL;
102
Johannes Berg8cb08172019-04-26 14:07:28 +0200103 err = nla_parse_nested_deprecated(tb, TCA_DEF_MAX, nla, simple_policy,
104 NULL);
Patrick McHardycee63722008-01-23 20:33:32 -0800105 if (err < 0)
106 return err;
107
Patrick McHardy53b2bf32008-01-23 20:36:30 -0800108 if (tb[TCA_DEF_PARMS] == NULL)
David S. Millere9ce1cd2006-08-21 23:54:55 -0700109 return -EINVAL;
110
Jamal Hadi Salimfa1b1cf2008-05-05 00:22:35 -0700111 parm = nla_data(tb[TCA_DEF_PARMS]);
Dmytro Linkin7be8ef22019-08-01 13:02:51 +0000112 index = parm->index;
113 err = tcf_idr_check_alloc(tn, &index, a, bind);
Vlad Buslov0190c1d2018-07-05 17:24:32 +0300114 if (err < 0)
115 return err;
116 exists = err;
Jamal Hadi Salim0e5538a2016-05-10 16:49:29 -0400117 if (exists && bind)
118 return 0;
119
120 if (tb[TCA_DEF_DATA] == NULL) {
121 if (exists)
Chris Mi65a206c2017-08-30 02:31:59 -0400122 tcf_idr_release(*a, bind);
Vlad Buslov0190c1d2018-07-05 17:24:32 +0300123 else
Dmytro Linkin7be8ef22019-08-01 13:02:51 +0000124 tcf_idr_cleanup(tn, index);
Jamal Hadi Salim0e5538a2016-05-10 16:49:29 -0400125 return -EINVAL;
126 }
127
Jamal Hadi Salim0e5538a2016-05-10 16:49:29 -0400128 if (!exists) {
Dmytro Linkin7be8ef22019-08-01 13:02:51 +0000129 ret = tcf_idr_create(tn, index, est, a,
Chris Mi65a206c2017-08-30 02:31:59 -0400130 &act_simp_ops, bind, false);
Vlad Buslov0190c1d2018-07-05 17:24:32 +0300131 if (ret) {
Dmytro Linkin7be8ef22019-08-01 13:02:51 +0000132 tcf_idr_cleanup(tn, index);
WANG Cong86062032014-02-11 17:07:31 -0800133 return ret;
Vlad Buslov0190c1d2018-07-05 17:24:32 +0300134 }
David S. Millere9ce1cd2006-08-21 23:54:55 -0700135
WANG Conga85a9702016-07-25 16:09:41 -0700136 d = to_defact(*a);
Davide Caratti4b006b02019-03-20 15:00:10 +0100137 err = tcf_action_check_ctrlact(parm->action, tp, &goto_ch,
138 extack);
139 if (err < 0)
140 goto release_idr;
141
142 err = alloc_defdata(d, tb[TCA_DEF_DATA]);
143 if (err < 0)
144 goto put_chain;
145
146 tcf_action_set_ctrlact(*a, parm->action, goto_ch);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700147 ret = ACT_P_CREATED;
148 } else {
Vlad Buslov4e8ddd72018-07-05 17:24:30 +0300149 if (!ovr) {
Davide Caratti4b006b02019-03-20 15:00:10 +0100150 err = -EEXIST;
151 goto release_idr;
Vlad Buslov4e8ddd72018-07-05 17:24:30 +0300152 }
Jamal Hadi Salim1a293212013-12-23 08:02:11 -0500153
Davide Caratti4b006b02019-03-20 15:00:10 +0100154 err = reset_policy(*a, tb[TCA_DEF_DATA], parm, tp, extack);
155 if (err)
156 goto release_idr;
David S. Millere9ce1cd2006-08-21 23:54:55 -0700157 }
158
David S. Millere9ce1cd2006-08-21 23:54:55 -0700159 if (ret == ACT_P_CREATED)
Chris Mi65a206c2017-08-30 02:31:59 -0400160 tcf_idr_insert(tn, *a);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700161 return ret;
Davide Caratti4b006b02019-03-20 15:00:10 +0100162put_chain:
163 if (goto_ch)
164 tcf_chain_put_by_act(goto_ch);
165release_idr:
166 tcf_idr_release(*a, bind);
167 return err;
David S. Millere9ce1cd2006-08-21 23:54:55 -0700168}
169
Eric Dumazetcc7ec452011-01-19 19:26:56 +0000170static int tcf_simp_dump(struct sk_buff *skb, struct tc_action *a,
171 int bind, int ref)
David S. Millere9ce1cd2006-08-21 23:54:55 -0700172{
Arnaldo Carvalho de Melo27a884d2007-04-19 20:29:13 -0700173 unsigned char *b = skb_tail_pointer(skb);
WANG Conga85a9702016-07-25 16:09:41 -0700174 struct tcf_defact *d = to_defact(a);
Eric Dumazet1c40be12010-08-16 20:04:22 +0000175 struct tc_defact opt = {
176 .index = d->tcf_index,
Vlad Buslov036bb442018-07-05 17:24:24 +0300177 .refcnt = refcount_read(&d->tcf_refcnt) - ref,
178 .bindcnt = atomic_read(&d->tcf_bindcnt) - bind,
Eric Dumazet1c40be12010-08-16 20:04:22 +0000179 };
David S. Millere9ce1cd2006-08-21 23:54:55 -0700180 struct tcf_t t;
181
Vlad Buslov5e481802018-08-10 20:51:48 +0300182 spin_lock_bh(&d->tcf_lock);
183 opt.action = d->tcf_action;
David S. Miller1b34ec42012-03-29 05:11:39 -0400184 if (nla_put(skb, TCA_DEF_PARMS, sizeof(opt), &opt) ||
185 nla_put_string(skb, TCA_DEF_DATA, d->tcfd_defdata))
186 goto nla_put_failure;
Jamal Hadi Salim48d8ee12016-06-06 06:32:55 -0400187
188 tcf_tm_dump(&t, &d->tcf_tm);
Nicolas Dichtel9854518e2016-04-26 10:06:18 +0200189 if (nla_put_64bit(skb, TCA_DEF_TM, sizeof(t), &t, TCA_DEF_PAD))
David S. Miller1b34ec42012-03-29 05:11:39 -0400190 goto nla_put_failure;
Vlad Buslov5e481802018-08-10 20:51:48 +0300191 spin_unlock_bh(&d->tcf_lock);
192
David S. Millere9ce1cd2006-08-21 23:54:55 -0700193 return skb->len;
194
Patrick McHardy7ba699c2008-01-22 22:11:50 -0800195nla_put_failure:
Vlad Buslov5e481802018-08-10 20:51:48 +0300196 spin_unlock_bh(&d->tcf_lock);
Arnaldo Carvalho de Melodc5fc572007-03-25 23:06:12 -0700197 nlmsg_trim(skb, b);
David S. Millere9ce1cd2006-08-21 23:54:55 -0700198 return -1;
Jamal Hadi Salimdb753072005-04-24 20:10:16 -0700199}
200
WANG Congddf97cc2016-02-22 15:57:53 -0800201static int tcf_simp_walker(struct net *net, struct sk_buff *skb,
202 struct netlink_callback *cb, int type,
Alexander Aring41780102018-02-15 10:54:58 -0500203 const struct tc_action_ops *ops,
204 struct netlink_ext_ack *extack)
WANG Congddf97cc2016-02-22 15:57:53 -0800205{
206 struct tc_action_net *tn = net_generic(net, simp_net_id);
207
Alexander Aringb3620142018-02-15 10:54:59 -0500208 return tcf_generic_walker(tn, skb, cb, type, ops, extack);
WANG Congddf97cc2016-02-22 15:57:53 -0800209}
210
Cong Wangf061b482018-08-29 10:15:35 -0700211static int tcf_simp_search(struct net *net, struct tc_action **a, u32 index)
WANG Congddf97cc2016-02-22 15:57:53 -0800212{
213 struct tc_action_net *tn = net_generic(net, simp_net_id);
214
Chris Mi65a206c2017-08-30 02:31:59 -0400215 return tcf_idr_search(tn, a, index);
WANG Congddf97cc2016-02-22 15:57:53 -0800216}
217
Jamal Hadi Salimdb753072005-04-24 20:10:16 -0700218static struct tc_action_ops act_simp_ops = {
David S. Millere9ce1cd2006-08-21 23:54:55 -0700219 .kind = "simple",
Eli Coheneddd2cf2019-02-10 14:25:00 +0200220 .id = TCA_ID_SIMP,
David S. Millere9ce1cd2006-08-21 23:54:55 -0700221 .owner = THIS_MODULE,
Jamal Hadi Salim798de372018-08-12 09:34:57 -0400222 .act = tcf_simp_act,
David S. Millere9ce1cd2006-08-21 23:54:55 -0700223 .dump = tcf_simp_dump,
WANG Cong86062032014-02-11 17:07:31 -0800224 .cleanup = tcf_simp_release,
David S. Millere9ce1cd2006-08-21 23:54:55 -0700225 .init = tcf_simp_init,
WANG Congddf97cc2016-02-22 15:57:53 -0800226 .walk = tcf_simp_walker,
227 .lookup = tcf_simp_search,
WANG Conga85a9702016-07-25 16:09:41 -0700228 .size = sizeof(struct tcf_defact),
WANG Congddf97cc2016-02-22 15:57:53 -0800229};
230
231static __net_init int simp_init_net(struct net *net)
232{
233 struct tc_action_net *tn = net_generic(net, simp_net_id);
234
Cong Wang981471b2019-08-25 10:01:32 -0700235 return tc_action_net_init(net, tn, &act_simp_ops);
WANG Congddf97cc2016-02-22 15:57:53 -0800236}
237
Cong Wang039af9c2017-12-11 15:35:03 -0800238static void __net_exit simp_exit_net(struct list_head *net_list)
WANG Congddf97cc2016-02-22 15:57:53 -0800239{
Cong Wang039af9c2017-12-11 15:35:03 -0800240 tc_action_net_exit(net_list, simp_net_id);
WANG Congddf97cc2016-02-22 15:57:53 -0800241}
242
243static struct pernet_operations simp_net_ops = {
244 .init = simp_init_net,
Cong Wang039af9c2017-12-11 15:35:03 -0800245 .exit_batch = simp_exit_net,
WANG Congddf97cc2016-02-22 15:57:53 -0800246 .id = &simp_net_id,
247 .size = sizeof(struct tc_action_net),
Jamal Hadi Salimdb753072005-04-24 20:10:16 -0700248};
249
250MODULE_AUTHOR("Jamal Hadi Salim(2005)");
251MODULE_DESCRIPTION("Simple example action");
252MODULE_LICENSE("GPL");
253
254static int __init simp_init_module(void)
255{
WANG Congddf97cc2016-02-22 15:57:53 -0800256 int ret = tcf_register_action(&act_simp_ops, &simp_net_ops);
Jamal Hadi Salimdb753072005-04-24 20:10:16 -0700257 if (!ret)
stephen hemminger6ff9c362010-05-12 06:37:05 +0000258 pr_info("Simple TC action Loaded\n");
Jamal Hadi Salimdb753072005-04-24 20:10:16 -0700259 return ret;
260}
261
262static void __exit simp_cleanup_module(void)
263{
WANG Congddf97cc2016-02-22 15:57:53 -0800264 tcf_unregister_action(&act_simp_ops, &simp_net_ops);
Jamal Hadi Salimdb753072005-04-24 20:10:16 -0700265}
266
267module_init(simp_init_module);
268module_exit(simp_cleanup_module);