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 28 void sock_update_netprioidx(struct sock *sk); 29 30 #if IS_BUILTIN(CONFIG_CGROUP_NET_PRIO) 31 static inline u32 task_netprioidx(struct task_struct *p) 32 { 33 struct cgroup_subsys_state *css; 34 u32 idx; 35 36 rcu_read_lock(); 37 css = task_css(p, net_prio_subsys_id); 38 idx = css->cgroup->id; 39 rcu_read_unlock(); 40 return idx; 41 } 42 #elif IS_MODULE(CONFIG_CGROUP_NET_PRIO) 43 static inline u32 task_netprioidx(struct task_struct *p) 44 { 45 struct cgroup_subsys_state *css; 46 u32 idx = 0; 47 48 rcu_read_lock(); 49 css = task_css(p, net_prio_subsys_id); 50 if (css) 51 idx = css->cgroup->id; 52 rcu_read_unlock(); 53 return idx; 54 } 55 #endif 56 #else /* !CONFIG_CGROUP_NET_PRIO */ 57 static inline u32 task_netprioidx(struct task_struct *p) 58 { 59 return 0; 60 } 61 62 #define sock_update_netprioidx(sk) 63 64 #endif /* CONFIG_CGROUP_NET_PRIO */ 65 #endif /* _NET_CLS_CGROUP_H */ 66