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 #include <linux/cgroup.h> 17 #include <linux/hardirq.h> 18 #include <linux/rcupdate.h> 19 20 21 struct netprio_map { 22 struct rcu_head rcu; 23 u32 priomap_len; 24 u32 priomap[]; 25 }; 26 27 #ifdef CONFIG_CGROUPS 28 29 struct cgroup_netprio_state { 30 struct cgroup_subsys_state css; 31 u32 prioidx; 32 }; 33 34 #ifndef CONFIG_NETPRIO_CGROUP 35 extern int net_prio_subsys_id; 36 #endif 37 38 extern void sock_update_netprioidx(struct sock *sk); 39 40 #if IS_BUILTIN(CONFIG_NETPRIO_CGROUP) 41 task_netprioidx(struct task_struct * p)42static inline u32 task_netprioidx(struct task_struct *p) 43 { 44 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 task_netprioidx(struct task_struct * p)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; 73 } 74 75 #else 76 task_netprioidx(struct task_struct * p)77static inline u32 task_netprioidx(struct task_struct *p) 78 { 79 return 0; 80 } 81 82 #endif /* CONFIG_NETPRIO_CGROUP */ 83 84 #else 85 #define sock_update_netprioidx(sk) 86 #endif 87 88 #endif /* _NET_CLS_CGROUP_H */ 89