]>
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> | |
19 | ||
8fb974c9 | 20 | #if IS_ENABLED(CONFIG_NET_CLS_CGROUP) |
f8451725 HX |
21 | struct cgroup_cls_state |
22 | { | |
23 | struct cgroup_subsys_state css; | |
24 | u32 classid; | |
25 | }; | |
26 | ||
211d2f97 | 27 | extern void sock_update_classid(struct sock *sk); |
f3419807 | 28 | |
8fb974c9 | 29 | #if IS_BUILTIN(CONFIG_NET_CLS_CGROUP) |
f8451725 HX |
30 | static inline u32 task_cls_classid(struct task_struct *p) |
31 | { | |
920750ce | 32 | u32 classid; |
3fb5a991 | 33 | |
f8451725 HX |
34 | if (in_interrupt()) |
35 | return 0; | |
36 | ||
3fb5a991 | 37 | rcu_read_lock(); |
8af01f56 | 38 | classid = container_of(task_css(p, net_cls_subsys_id), |
3fb5a991 LZ |
39 | struct cgroup_cls_state, css)->classid; |
40 | rcu_read_unlock(); | |
41 | ||
42 | return classid; | |
f8451725 | 43 | } |
8fb974c9 | 44 | #elif IS_MODULE(CONFIG_NET_CLS_CGROUP) |
f8451725 HX |
45 | static inline u32 task_cls_classid(struct task_struct *p) |
46 | { | |
8a8e04df | 47 | struct cgroup_subsys_state *css; |
ea16f912 | 48 | u32 classid = 0; |
f8451725 HX |
49 | |
50 | if (in_interrupt()) | |
51 | return 0; | |
52 | ||
53 | rcu_read_lock(); | |
8af01f56 | 54 | css = task_css(p, net_cls_subsys_id); |
8a8e04df DW |
55 | if (css) |
56 | classid = container_of(css, | |
f8451725 HX |
57 | struct cgroup_cls_state, css)->classid; |
58 | rcu_read_unlock(); | |
59 | ||
60 | return classid; | |
61 | } | |
62 | #endif | |
8fb974c9 | 63 | #else /* !CGROUP_NET_CLS_CGROUP */ |
211d2f97 | 64 | static inline void sock_update_classid(struct sock *sk) |
f3419807 DW |
65 | { |
66 | } | |
67 | ||
f8451725 HX |
68 | static inline u32 task_cls_classid(struct task_struct *p) |
69 | { | |
70 | return 0; | |
71 | } | |
8fb974c9 | 72 | #endif /* CGROUP_NET_CLS_CGROUP */ |
f8451725 | 73 | #endif /* _NET_CLS_CGROUP_H */ |