Commit | Line | Data |
---|---|---|
f8451725 HX |
1 | /* |
2 | * cls_cgroup.h Control Group Classifier | |
3 | * | |
4 | * Authors: Thomas Graf <tgraf@suug.ch> | |
5 | * | |
6 | * This program is free software; you can redistribute it and/or modify it | |
7 | * under the terms of the GNU General Public License as published by the Free | |
8 | * Software Foundation; either version 2 of the License, or (at your option) | |
9 | * any later version. | |
10 | * | |
11 | */ | |
12 | ||
13 | #ifndef _NET_CLS_CGROUP_H | |
14 | #define _NET_CLS_CGROUP_H | |
15 | ||
16 | #include <linux/cgroup.h> | |
17 | #include <linux/hardirq.h> | |
18 | #include <linux/rcupdate.h> | |
fe1217c4 | 19 | #include <net/sock.h> |
f8451725 | 20 | |
fe1217c4 DB |
21 | #ifdef CONFIG_CGROUP_NET_CLASSID |
22 | struct cgroup_cls_state { | |
f8451725 HX |
23 | struct cgroup_subsys_state css; |
24 | u32 classid; | |
25 | }; | |
26 | ||
fe1217c4 | 27 | struct cgroup_cls_state *task_cls_state(struct task_struct *p); |
f3419807 | 28 | |
f8451725 HX |
29 | static inline u32 task_cls_classid(struct task_struct *p) |
30 | { | |
920750ce | 31 | u32 classid; |
3fb5a991 | 32 | |
f8451725 HX |
33 | if (in_interrupt()) |
34 | return 0; | |
35 | ||
3fb5a991 | 36 | rcu_read_lock(); |
073219e9 | 37 | classid = container_of(task_css(p, net_cls_cgrp_id), |
3fb5a991 LZ |
38 | struct cgroup_cls_state, css)->classid; |
39 | rcu_read_unlock(); | |
40 | ||
41 | return classid; | |
f8451725 | 42 | } |
f8451725 | 43 | |
211d2f97 | 44 | static inline void sock_update_classid(struct sock *sk) |
f3419807 | 45 | { |
fe1217c4 | 46 | u32 classid; |
f3419807 | 47 | |
fe1217c4 DB |
48 | classid = task_cls_classid(current); |
49 | if (classid != sk->sk_classid) | |
50 | sk->sk_classid = classid; | |
51 | } | |
b87a173e DB |
52 | |
53 | static inline u32 task_get_classid(const struct sk_buff *skb) | |
54 | { | |
55 | u32 classid = task_cls_state(current)->classid; | |
56 | ||
57 | /* Due to the nature of the classifier it is required to ignore all | |
58 | * packets originating from softirq context as accessing `current' | |
59 | * would lead to false results. | |
60 | * | |
61 | * This test assumes that all callers of dev_queue_xmit() explicitly | |
62 | * disable bh. Knowing this, it is possible to detect softirq based | |
63 | * calls by looking at the number of nested bh disable calls because | |
64 | * softirqs always disables bh. | |
65 | */ | |
66 | if (in_serving_softirq()) { | |
67 | /* If there is an sk_classid we'll use that. */ | |
68 | if (!skb->sk) | |
69 | return 0; | |
70 | ||
71 | classid = skb->sk->sk_classid; | |
72 | } | |
73 | ||
74 | return classid; | |
75 | } | |
fe1217c4 DB |
76 | #else /* !CONFIG_CGROUP_NET_CLASSID */ |
77 | static inline void sock_update_classid(struct sock *sk) | |
f8451725 | 78 | { |
f8451725 | 79 | } |
b87a173e DB |
80 | |
81 | static inline u32 task_get_classid(const struct sk_buff *skb) | |
82 | { | |
83 | return 0; | |
84 | } | |
fe1217c4 | 85 | #endif /* CONFIG_CGROUP_NET_CLASSID */ |
f8451725 | 86 | #endif /* _NET_CLS_CGROUP_H */ |