blob: d58fdec47597bccc095a9da28066bd6b67acfc2f [file] [log] [blame]
Neil Horman5bc14212011-11-22 05:10:51 +00001/*
2 * netprio_cgroup.h Control Group Priority set
3 *
4 *
5 * Authors: Neil Horman <nhorman@tuxdriver.com>
6 *
7 * This program is free software; you can redistribute it and/or modify it
8 * under the terms of the GNU General Public License as published by the Free
9 * Software Foundation; either version 2 of the License, or (at your option)
10 * any later version.
11 *
12 */
13
14#ifndef _NETPRIO_CGROUP_H
15#define _NETPRIO_CGROUP_H
Neil Horman5bc14212011-11-22 05:10:51 +000016#include <linux/cgroup.h>
17#include <linux/hardirq.h>
18#include <linux/rcupdate.h>
19
Neil Horman5bc14212011-11-22 05:10:51 +000020
21struct netprio_map {
22 struct rcu_head rcu;
23 u32 priomap_len;
24 u32 priomap[];
25};
26
27#ifdef CONFIG_CGROUPS
28
Neil Horman8c215222011-11-23 15:52:58 -050029struct cgroup_netprio_state {
30 struct cgroup_subsys_state css;
31 u32 prioidx;
32};
33
Neil Horman5bc14212011-11-22 05:10:51 +000034#ifndef CONFIG_NETPRIO_CGROUP
35extern int net_prio_subsys_id;
36#endif
37
38extern void sock_update_netprioidx(struct sock *sk);
39
Neil Horman2b73bc62012-02-10 05:43:38 +000040#if IS_BUILTIN(CONFIG_NETPRIO_CGROUP)
41
42static inline u32 task_netprioidx(struct task_struct *p)
Neil Horman5bc14212011-11-22 05:10:51 +000043{
Neil Horman2b73bc62012-02-10 05:43:38 +000044 struct cgroup_netprio_state *state;
45 u32 idx;
46
47 rcu_read_lock();
48 state = container_of(task_subsys_state(p, net_prio_subsys_id),
49 struct cgroup_netprio_state, css);
50 idx = state->prioidx;
51 rcu_read_unlock();
52 return idx;
53}
54
55#elif IS_MODULE(CONFIG_NETPRIO_CGROUP)
56
57static inline u32 task_netprioidx(struct task_struct *p)
58{
59 struct cgroup_netprio_state *state;
60 int subsys_id;
61 u32 idx = 0;
62
63 rcu_read_lock();
64 subsys_id = rcu_dereference_index_check(net_prio_subsys_id,
65 rcu_read_lock_held());
66 if (subsys_id >= 0) {
67 state = container_of(task_subsys_state(p, subsys_id),
68 struct cgroup_netprio_state, css);
69 idx = state->prioidx;
70 }
71 rcu_read_unlock();
72 return idx;
Neil Horman5bc14212011-11-22 05:10:51 +000073}
74
75#else
76
Neil Horman2b73bc62012-02-10 05:43:38 +000077static inline u32 task_netprioidx(struct task_struct *p)
78{
79 return 0;
80}
81
82#endif /* CONFIG_NETPRIO_CGROUP */
83
84#else
Neil Horman5bc14212011-11-22 05:10:51 +000085#define sock_update_netprioidx(sk)
Neil Horman5bc14212011-11-22 05:10:51 +000086#endif
87
88#endif /* _NET_CLS_CGROUP_H */