blob: fb881144fa015078c03e21ace0f9391ca0aea104 [file] [log] [blame]
Thomas Gleixner2874c5f2019-05-27 08:55:01 +02001// SPDX-License-Identifier: GPL-2.0-or-later
Thomas Graff4009232008-11-07 22:56:00 -08002/*
3 * net/sched/cls_cgroup.c Control Group Classifier
4 *
Thomas Graff4009232008-11-07 22:56:00 -08005 * Authors: Thomas Graf <tgraf@suug.ch>
6 */
7
8#include <linux/module.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +09009#include <linux/slab.h>
Thomas Graff4009232008-11-07 22:56:00 -080010#include <linux/skbuff.h>
Herbert Xuf8451722010-05-24 00:12:34 -070011#include <linux/rcupdate.h>
Thomas Graff4009232008-11-07 22:56:00 -080012#include <net/rtnetlink.h>
13#include <net/pkt_cls.h>
Herbert Xuf8451722010-05-24 00:12:34 -070014#include <net/sock.h>
15#include <net/cls_cgroup.h>
Thomas Graff4009232008-11-07 22:56:00 -080016
Eric Dumazetcc7ec452011-01-19 19:26:56 +000017struct cls_cgroup_head {
Thomas Graff4009232008-11-07 22:56:00 -080018 u32 handle;
19 struct tcf_exts exts;
20 struct tcf_ematch_tree ematches;
John Fastabend952313b2014-09-12 20:06:26 -070021 struct tcf_proto *tp;
Cong Wangaaa908f2018-05-23 15:26:53 -070022 struct rcu_work rwork;
Thomas Graff4009232008-11-07 22:56:00 -080023};
24
Eric Dumazetdc7f9f62011-07-05 23:25:42 +000025static int cls_cgroup_classify(struct sk_buff *skb, const struct tcf_proto *tp,
Thomas Graff4009232008-11-07 22:56:00 -080026 struct tcf_result *res)
27{
John Fastabend952313b2014-09-12 20:06:26 -070028 struct cls_cgroup_head *head = rcu_dereference_bh(tp->root);
Daniel Borkmannb87a1732015-07-15 14:21:41 +020029 u32 classid = task_get_classid(skb);
Thomas Graff4009232008-11-07 22:56:00 -080030
Matteo Croce594725d2019-05-02 17:13:18 +020031 if (unlikely(!head))
32 return -1;
Paul Menagee65fcfd2009-05-26 20:47:02 -070033 if (!classid)
34 return -1;
Paul Menagee65fcfd2009-05-26 20:47:02 -070035 if (!tcf_em_tree_match(skb, &head->ematches, NULL))
36 return -1;
37
38 res->classid = classid;
39 res->class = 0;
Daniel Borkmannb87a1732015-07-15 14:21:41 +020040
Paul Menagee65fcfd2009-05-26 20:47:02 -070041 return tcf_exts_exec(skb, &head->exts, res);
Thomas Graff4009232008-11-07 22:56:00 -080042}
43
WANG Cong8113c092017-08-04 21:31:43 -070044static void *cls_cgroup_get(struct tcf_proto *tp, u32 handle)
Thomas Graff4009232008-11-07 22:56:00 -080045{
WANG Cong8113c092017-08-04 21:31:43 -070046 return NULL;
Thomas Graff4009232008-11-07 22:56:00 -080047}
48
Thomas Graff4009232008-11-07 22:56:00 -080049static int cls_cgroup_init(struct tcf_proto *tp)
50{
51 return 0;
52}
53
Thomas Graff4009232008-11-07 22:56:00 -080054static const struct nla_policy cgroup_policy[TCA_CGROUP_MAX + 1] = {
55 [TCA_CGROUP_EMATCHES] = { .type = NLA_NESTED },
56};
57
Cong Wanged148162017-11-06 13:47:22 -080058static void __cls_cgroup_destroy(struct cls_cgroup_head *head)
59{
60 tcf_exts_destroy(&head->exts);
61 tcf_em_tree_destroy(&head->ematches);
62 tcf_exts_put_net(&head->exts);
63 kfree(head);
64}
65
Cong Wangb1b5b042017-10-26 18:24:31 -070066static void cls_cgroup_destroy_work(struct work_struct *work)
67{
Cong Wangaaa908f2018-05-23 15:26:53 -070068 struct cls_cgroup_head *head = container_of(to_rcu_work(work),
Cong Wangb1b5b042017-10-26 18:24:31 -070069 struct cls_cgroup_head,
Cong Wangaaa908f2018-05-23 15:26:53 -070070 rwork);
Cong Wangb1b5b042017-10-26 18:24:31 -070071 rtnl_lock();
Cong Wanged148162017-11-06 13:47:22 -080072 __cls_cgroup_destroy(head);
Cong Wangb1b5b042017-10-26 18:24:31 -070073 rtnl_unlock();
74}
75
Benjamin LaHaisec1b52732013-01-14 05:15:39 +000076static int cls_cgroup_change(struct net *net, struct sk_buff *in_skb,
Eric W. Biedermanaf4c6642012-05-25 13:42:45 -060077 struct tcf_proto *tp, unsigned long base,
Thomas Graff4009232008-11-07 22:56:00 -080078 u32 handle, struct nlattr **tca,
Vlad Buslov12db03b2019-02-11 10:55:45 +020079 void **arg, bool ovr, bool rtnl_held,
Alexander Aring7306db32018-01-18 11:20:51 -050080 struct netlink_ext_ack *extack)
Thomas Graff4009232008-11-07 22:56:00 -080081{
Eric Dumazetcc7ec452011-01-19 19:26:56 +000082 struct nlattr *tb[TCA_CGROUP_MAX + 1];
John Fastabend952313b2014-09-12 20:06:26 -070083 struct cls_cgroup_head *head = rtnl_dereference(tp->root);
84 struct cls_cgroup_head *new;
Thomas Graff4009232008-11-07 22:56:00 -080085 int err;
86
Minoru Usui52ea3a52009-06-09 04:03:09 -070087 if (!tca[TCA_OPTIONS])
88 return -EINVAL;
89
John Fastabend952313b2014-09-12 20:06:26 -070090 if (!head && !handle)
91 return -EINVAL;
Thomas Graff4009232008-11-07 22:56:00 -080092
John Fastabend952313b2014-09-12 20:06:26 -070093 if (head && handle != head->handle)
Thomas Graff4009232008-11-07 22:56:00 -080094 return -ENOENT;
95
John Fastabend952313b2014-09-12 20:06:26 -070096 new = kzalloc(sizeof(*head), GFP_KERNEL);
97 if (!new)
98 return -ENOBUFS;
99
Cong Wang14215102019-02-20 21:37:42 -0800100 err = tcf_exts_init(&new->exts, net, TCA_CGROUP_ACT, TCA_CGROUP_POLICE);
WANG Congb9a24bb2016-08-19 12:36:54 -0700101 if (err < 0)
102 goto errout;
Jiri Pirko18b54272014-12-02 18:00:36 +0100103 new->handle = handle;
John Fastabend952313b2014-09-12 20:06:26 -0700104 new->tp = tp;
Johannes Berg8cb08172019-04-26 14:07:28 +0200105 err = nla_parse_nested_deprecated(tb, TCA_CGROUP_MAX,
106 tca[TCA_OPTIONS], cgroup_policy,
107 NULL);
Thomas Graff4009232008-11-07 22:56:00 -0800108 if (err < 0)
John Fastabendd14cbfc82014-09-15 23:31:17 -0700109 goto errout;
Thomas Graff4009232008-11-07 22:56:00 -0800110
Alexander Aring50a56192018-01-18 11:20:52 -0500111 err = tcf_exts_validate(net, tp, tb, tca[TCA_RATE], &new->exts, ovr,
Vlad Buslovec6743a2019-02-11 10:55:43 +0200112 true, extack);
Thomas Graff4009232008-11-07 22:56:00 -0800113 if (err < 0)
John Fastabendd14cbfc82014-09-15 23:31:17 -0700114 goto errout;
Thomas Graff4009232008-11-07 22:56:00 -0800115
Jiri Pirko4ebc1e32017-08-04 14:28:57 +0200116 err = tcf_em_tree_validate(tp, tb[TCA_CGROUP_EMATCHES], &new->ematches);
Jiri Pirko8cc62512017-08-04 14:29:11 +0200117 if (err < 0)
John Fastabendd14cbfc82014-09-15 23:31:17 -0700118 goto errout;
Thomas Graff4009232008-11-07 22:56:00 -0800119
John Fastabend952313b2014-09-12 20:06:26 -0700120 rcu_assign_pointer(tp->root, new);
Cong Wanged148162017-11-06 13:47:22 -0800121 if (head) {
122 tcf_exts_get_net(&head->exts);
Cong Wangaaa908f2018-05-23 15:26:53 -0700123 tcf_queue_work(&head->rwork, cls_cgroup_destroy_work);
Cong Wanged148162017-11-06 13:47:22 -0800124 }
Thomas Graff4009232008-11-07 22:56:00 -0800125 return 0;
John Fastabendd14cbfc82014-09-15 23:31:17 -0700126errout:
WANG Congb9a24bb2016-08-19 12:36:54 -0700127 tcf_exts_destroy(&new->exts);
John Fastabendd14cbfc82014-09-15 23:31:17 -0700128 kfree(new);
129 return err;
Thomas Graff4009232008-11-07 22:56:00 -0800130}
131
Vlad Buslov12db03b2019-02-11 10:55:45 +0200132static void cls_cgroup_destroy(struct tcf_proto *tp, bool rtnl_held,
Jakub Kicinski715df5e2018-01-24 12:54:13 -0800133 struct netlink_ext_ack *extack)
Thomas Graff4009232008-11-07 22:56:00 -0800134{
John Fastabend952313b2014-09-12 20:06:26 -0700135 struct cls_cgroup_head *head = rtnl_dereference(tp->root);
Thomas Graff4009232008-11-07 22:56:00 -0800136
Daniel Borkmannd9363772016-11-27 01:18:01 +0100137 /* Head can still be NULL due to cls_cgroup_init(). */
Cong Wanged148162017-11-06 13:47:22 -0800138 if (head) {
139 if (tcf_exts_get_net(&head->exts))
Cong Wangaaa908f2018-05-23 15:26:53 -0700140 tcf_queue_work(&head->rwork, cls_cgroup_destroy_work);
Cong Wanged148162017-11-06 13:47:22 -0800141 else
142 __cls_cgroup_destroy(head);
143 }
Thomas Graff4009232008-11-07 22:56:00 -0800144}
145
Alexander Aring571acf22018-01-18 11:20:53 -0500146static int cls_cgroup_delete(struct tcf_proto *tp, void *arg, bool *last,
Vlad Buslov12db03b2019-02-11 10:55:45 +0200147 bool rtnl_held, struct netlink_ext_ack *extack)
Thomas Graff4009232008-11-07 22:56:00 -0800148{
149 return -EOPNOTSUPP;
150}
151
Vlad Buslov12db03b2019-02-11 10:55:45 +0200152static void cls_cgroup_walk(struct tcf_proto *tp, struct tcf_walker *arg,
153 bool rtnl_held)
Thomas Graff4009232008-11-07 22:56:00 -0800154{
John Fastabend952313b2014-09-12 20:06:26 -0700155 struct cls_cgroup_head *head = rtnl_dereference(tp->root);
Thomas Graff4009232008-11-07 22:56:00 -0800156
157 if (arg->count < arg->skip)
158 goto skip;
159
Vlad Buslov8b58d122019-02-15 17:18:44 +0200160 if (!head)
161 return;
WANG Cong8113c092017-08-04 21:31:43 -0700162 if (arg->fn(tp, head, arg) < 0) {
Thomas Graff4009232008-11-07 22:56:00 -0800163 arg->stop = 1;
164 return;
165 }
166skip:
167 arg->count++;
168}
169
WANG Cong8113c092017-08-04 21:31:43 -0700170static int cls_cgroup_dump(struct net *net, struct tcf_proto *tp, void *fh,
Vlad Buslov12db03b2019-02-11 10:55:45 +0200171 struct sk_buff *skb, struct tcmsg *t, bool rtnl_held)
Thomas Graff4009232008-11-07 22:56:00 -0800172{
John Fastabend952313b2014-09-12 20:06:26 -0700173 struct cls_cgroup_head *head = rtnl_dereference(tp->root);
Thomas Graff4009232008-11-07 22:56:00 -0800174 struct nlattr *nest;
175
176 t->tcm_handle = head->handle;
177
Michal Kubecekae0be8d2019-04-26 11:13:06 +0200178 nest = nla_nest_start_noflag(skb, TCA_OPTIONS);
Thomas Graff4009232008-11-07 22:56:00 -0800179 if (nest == NULL)
180 goto nla_put_failure;
181
WANG Cong5da57f42013-12-15 20:15:07 -0800182 if (tcf_exts_dump(skb, &head->exts) < 0 ||
Thomas Graff4009232008-11-07 22:56:00 -0800183 tcf_em_tree_dump(skb, &head->ematches, TCA_CGROUP_EMATCHES) < 0)
184 goto nla_put_failure;
185
186 nla_nest_end(skb, nest);
187
WANG Cong5da57f42013-12-15 20:15:07 -0800188 if (tcf_exts_dump_stats(skb, &head->exts) < 0)
Thomas Graff4009232008-11-07 22:56:00 -0800189 goto nla_put_failure;
190
191 return skb->len;
192
193nla_put_failure:
Jiri Pirko6ea3b442014-12-09 22:23:29 +0100194 nla_nest_cancel(skb, nest);
Thomas Graff4009232008-11-07 22:56:00 -0800195 return -1;
196}
197
198static struct tcf_proto_ops cls_cgroup_ops __read_mostly = {
199 .kind = "cgroup",
200 .init = cls_cgroup_init,
201 .change = cls_cgroup_change,
202 .classify = cls_cgroup_classify,
203 .destroy = cls_cgroup_destroy,
204 .get = cls_cgroup_get,
Thomas Graff4009232008-11-07 22:56:00 -0800205 .delete = cls_cgroup_delete,
206 .walk = cls_cgroup_walk,
207 .dump = cls_cgroup_dump,
208 .owner = THIS_MODULE,
209};
210
211static int __init init_cgroup_cls(void)
212{
Daniel Borkmannfe1217c2013-12-29 18:27:10 +0100213 return register_tcf_proto_ops(&cls_cgroup_ops);
Thomas Graff4009232008-11-07 22:56:00 -0800214}
215
216static void __exit exit_cgroup_cls(void)
217{
218 unregister_tcf_proto_ops(&cls_cgroup_ops);
Thomas Graff4009232008-11-07 22:56:00 -0800219}
220
221module_init(init_cgroup_cls);
222module_exit(exit_cgroup_cls);
223MODULE_LICENSE("GPL");