Blame view
net/sched/cls_cgroup.c
4.98 KB
f40092373
|
1 2 3 4 5 6 7 8 9 10 11 12 |
/* * net/sched/cls_cgroup.c Control Group Classifier * * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public License * as published by the Free Software Foundation; either version * 2 of the License, or (at your option) any later version. * * Authors: Thomas Graf <tgraf@suug.ch> */ #include <linux/module.h> |
5a0e3ad6a
|
13 |
#include <linux/slab.h> |
f40092373
|
14 |
#include <linux/skbuff.h> |
f84517253
|
15 |
#include <linux/rcupdate.h> |
f40092373
|
16 17 |
#include <net/rtnetlink.h> #include <net/pkt_cls.h> |
f84517253
|
18 19 |
#include <net/sock.h> #include <net/cls_cgroup.h> |
f40092373
|
20 |
|
cc7ec456f
|
21 |
struct cls_cgroup_head { |
f40092373
|
22 23 24 25 |
u32 handle; struct tcf_exts exts; struct tcf_ematch_tree ematches; }; |
dc7f9f6e8
|
26 |
static int cls_cgroup_classify(struct sk_buff *skb, const struct tcf_proto *tp, |
f40092373
|
27 28 29 |
struct tcf_result *res) { struct cls_cgroup_head *head = tp->root; |
e65fcfd63
|
30 |
u32 classid; |
f40092373
|
31 |
|
f84517253
|
32 33 34 |
rcu_read_lock(); classid = task_cls_state(current)->classid; rcu_read_unlock(); |
f40092373
|
35 36 37 38 39 40 41 42 43 44 |
/* * Due to the nature of the classifier it is required to ignore all * packets originating from softirq context as accessing `current' * would lead to false results. * * This test assumes that all callers of dev_queue_xmit() explicitely * disable bh. Knowing this, it is possible to detect softirq based * calls by looking at the number of nested bh disable calls because * softirqs always disables bh. */ |
75e1056f5
|
45 |
if (in_serving_softirq()) { |
f84517253
|
46 47 48 49 50 |
/* If there is an sk_classid we'll use that. */ if (!skb->sk) return -1; classid = skb->sk->sk_classid; } |
f40092373
|
51 |
|
e65fcfd63
|
52 53 54 55 56 57 58 59 60 |
if (!classid) return -1; if (!tcf_em_tree_match(skb, &head->ematches, NULL)) return -1; res->classid = classid; res->class = 0; return tcf_exts_exec(skb, &head->exts, res); |
f40092373
|
61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
} static unsigned long cls_cgroup_get(struct tcf_proto *tp, u32 handle) { return 0UL; } static void cls_cgroup_put(struct tcf_proto *tp, unsigned long f) { } static int cls_cgroup_init(struct tcf_proto *tp) { return 0; } |
f40092373
|
76 77 78 |
static const struct nla_policy cgroup_policy[TCA_CGROUP_MAX + 1] = { [TCA_CGROUP_EMATCHES] = { .type = NLA_NESTED }, }; |
c1b52739e
|
79 |
static int cls_cgroup_change(struct net *net, struct sk_buff *in_skb, |
af4c6641f
|
80 |
struct tcf_proto *tp, unsigned long base, |
f40092373
|
81 |
u32 handle, struct nlattr **tca, |
2f7ef2f87
|
82 |
unsigned long *arg, bool ovr) |
f40092373
|
83 |
{ |
cc7ec456f
|
84 |
struct nlattr *tb[TCA_CGROUP_MAX + 1]; |
f40092373
|
85 86 87 88 |
struct cls_cgroup_head *head = tp->root; struct tcf_ematch_tree t; struct tcf_exts e; int err; |
52ea3a56a
|
89 90 |
if (!tca[TCA_OPTIONS]) return -EINVAL; |
f40092373
|
91 92 93 94 95 96 97 |
if (head == NULL) { if (!handle) return -EINVAL; head = kzalloc(sizeof(*head), GFP_KERNEL); if (head == NULL) return -ENOBUFS; |
5da57f422
|
98 |
tcf_exts_init(&head->exts, TCA_CGROUP_ACT, TCA_CGROUP_POLICE); |
f40092373
|
99 100 101 102 103 104 105 106 107 108 109 110 111 112 |
head->handle = handle; tcf_tree_lock(tp); tp->root = head; tcf_tree_unlock(tp); } if (handle != head->handle) return -ENOENT; err = nla_parse_nested(tb, TCA_CGROUP_MAX, tca[TCA_OPTIONS], cgroup_policy); if (err < 0) return err; |
5da57f422
|
113 |
tcf_exts_init(&e, TCA_CGROUP_ACT, TCA_CGROUP_POLICE); |
2f7ef2f87
|
114 |
err = tcf_exts_validate(net, tp, tb, tca[TCA_RATE], &e, ovr); |
f40092373
|
115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 |
if (err < 0) return err; err = tcf_em_tree_validate(tp, tb[TCA_CGROUP_EMATCHES], &t); if (err < 0) return err; tcf_exts_change(tp, &head->exts, &e); tcf_em_tree_change(tp, &head->ematches, &t); return 0; } static void cls_cgroup_destroy(struct tcf_proto *tp) { |
47a1a1d4b
|
130 |
struct cls_cgroup_head *head = tp->root; |
f40092373
|
131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 |
if (head) { tcf_exts_destroy(tp, &head->exts); tcf_em_tree_destroy(tp, &head->ematches); kfree(head); } } static int cls_cgroup_delete(struct tcf_proto *tp, unsigned long arg) { return -EOPNOTSUPP; } static void cls_cgroup_walk(struct tcf_proto *tp, struct tcf_walker *arg) { struct cls_cgroup_head *head = tp->root; if (arg->count < arg->skip) goto skip; if (arg->fn(tp, (unsigned long) head, arg) < 0) { arg->stop = 1; return; } skip: arg->count++; } |
832d1d5bf
|
158 |
static int cls_cgroup_dump(struct net *net, struct tcf_proto *tp, unsigned long fh, |
f40092373
|
159 160 161 162 163 164 165 166 167 168 169 |
struct sk_buff *skb, struct tcmsg *t) { struct cls_cgroup_head *head = tp->root; unsigned char *b = skb_tail_pointer(skb); struct nlattr *nest; t->tcm_handle = head->handle; nest = nla_nest_start(skb, TCA_OPTIONS); if (nest == NULL) goto nla_put_failure; |
5da57f422
|
170 |
if (tcf_exts_dump(skb, &head->exts) < 0 || |
f40092373
|
171 172 173 174 |
tcf_em_tree_dump(skb, &head->ematches, TCA_CGROUP_EMATCHES) < 0) goto nla_put_failure; nla_nest_end(skb, nest); |
5da57f422
|
175 |
if (tcf_exts_dump_stats(skb, &head->exts) < 0) |
f40092373
|
176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 |
goto nla_put_failure; return skb->len; nla_put_failure: nlmsg_trim(skb, b); return -1; } static struct tcf_proto_ops cls_cgroup_ops __read_mostly = { .kind = "cgroup", .init = cls_cgroup_init, .change = cls_cgroup_change, .classify = cls_cgroup_classify, .destroy = cls_cgroup_destroy, .get = cls_cgroup_get, .put = cls_cgroup_put, .delete = cls_cgroup_delete, .walk = cls_cgroup_walk, .dump = cls_cgroup_dump, .owner = THIS_MODULE, }; static int __init init_cgroup_cls(void) { |
fe1217c4f
|
201 |
return register_tcf_proto_ops(&cls_cgroup_ops); |
f40092373
|
202 203 204 205 206 207 208 209 210 211 |
} static void __exit exit_cgroup_cls(void) { unregister_tcf_proto_ops(&cls_cgroup_ops); } module_init(init_cgroup_cls); module_exit(exit_cgroup_cls); MODULE_LICENSE("GPL"); |