1 /*
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
16 
17 #include <linux/cgroup.h>
18 #include <linux/hardirq.h>
19 #include <linux/rcupdate.h>
20 
21 #if IS_ENABLED(CONFIG_CGROUP_NET_PRIO)
22 struct netprio_map {
23 	struct rcu_head rcu;
24 	u32 priomap_len;
25 	u32 priomap[];
26 };
27 
task_netprioidx(struct task_struct * p)28 static inline u32 task_netprioidx(struct task_struct *p)
29 {
30 	struct cgroup_subsys_state *css;
31 	u32 idx;
32 
33 	rcu_read_lock();
34 	css = task_css(p, net_prio_cgrp_id);
35 	idx = css->cgroup->id;
36 	rcu_read_unlock();
37 	return idx;
38 }
39 
sock_update_netprioidx(struct sock_cgroup_data * skcd)40 static inline void sock_update_netprioidx(struct sock_cgroup_data *skcd)
41 {
42 	if (in_interrupt())
43 		return;
44 
45 	sock_cgroup_set_prioidx(skcd, task_netprioidx(current));
46 }
47 
48 #else /* !CONFIG_CGROUP_NET_PRIO */
49 
task_netprioidx(struct task_struct * p)50 static inline u32 task_netprioidx(struct task_struct *p)
51 {
52 	return 0;
53 }
54 
sock_update_netprioidx(struct sock_cgroup_data * skcd)55 static inline void sock_update_netprioidx(struct sock_cgroup_data *skcd)
56 {
57 }
58 
59 #endif /* CONFIG_CGROUP_NET_PRIO */
60 #endif  /* _NET_CLS_CGROUP_H */
61