[linux-2.6-block.git] / include / net / netprio_cgroup.h

/*
 * netprio_cgroup.h			Control Group Priority set
 *
 *
 * Authors:	Neil Horman <nhorman@tuxdriver.com>
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the Free
 * Software Foundation; either version 2 of the License, or (at your option)
 * any later version.
 *
 */

#ifndef _NETPRIO_CGROUP_H
#define _NETPRIO_CGROUP_H
#include <linux/cgroup.h>
#include <linux/hardirq.h>
#include <linux/rcupdate.h>


#if IS_ENABLED(CONFIG_NETPRIO_CGROUP)
struct netprio_map {
	struct rcu_head rcu;
	u32 priomap_len;
	u32 priomap[];
};

void sock_update_netprioidx(struct sock *sk);

#if IS_BUILTIN(CONFIG_NETPRIO_CGROUP)

static inline u32 task_netprioidx(struct task_struct *p)
{
	struct cgroup_subsys_state *css;
	u32 idx;

	rcu_read_lock();
	css = task_css(p, net_prio_subsys_id);
	idx = css->cgroup->id;
	rcu_read_unlock();
	return idx;
}

#elif IS_MODULE(CONFIG_NETPRIO_CGROUP)

static inline u32 task_netprioidx(struct task_struct *p)
{
	struct cgroup_subsys_state *css;
	u32 idx = 0;

	rcu_read_lock();
	css = task_css(p, net_prio_subsys_id);
	if (css)
		idx = css->cgroup->id;
	rcu_read_unlock();
	return idx;
}
#endif

#else /* !CONFIG_NETPRIO_CGROUP */

static inline u32 task_netprioidx(struct task_struct *p)
{
	return 0;
}

#define sock_update_netprioidx(sk)

#endif /* CONFIG_NETPRIO_CGROUP */

#endif  /* _NET_CLS_CGROUP_H */
Commit	Line	Data
5bc1421e NH	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
5bc1421e NH	16	#include <linux/cgroup.h>
	17	#include <linux/hardirq.h>
	18	#include <linux/rcupdate.h>
	19
5bc1421e	20
51e4e7fa	21	#if IS_ENABLED(CONFIG_NETPRIO_CGROUP)
5bc1421e NH	22	struct netprio_map {
	23	struct rcu_head rcu;
	24	u32 priomap_len;
	25	u32 priomap[];
	26	};
	27
37830721	28	void sock_update_netprioidx(struct sock *sk);
5bc1421e	29
2b73bc65 NH	30	#if IS_BUILTIN(CONFIG_NETPRIO_CGROUP)
	31
	32	static inline u32 task_netprioidx(struct task_struct *p)
5bc1421e	33	{
88d642fa	34	struct cgroup_subsys_state *css;
2b73bc65 NH	35	u32 idx;
	36
	37	rcu_read_lock();
8af01f56	38	css = task_css(p, net_prio_subsys_id);
88d642fa	39	idx = css->cgroup->id;
2b73bc65 NH	40	rcu_read_unlock();
	41	return idx;
	42	}
	43
	44	#elif IS_MODULE(CONFIG_NETPRIO_CGROUP)
	45
	46	static inline u32 task_netprioidx(struct task_struct *p)
	47	{
8a8e04df	48	struct cgroup_subsys_state *css;
2b73bc65 NH	49	u32 idx = 0;
	50
	51	rcu_read_lock();
8af01f56	52	css = task_css(p, net_prio_subsys_id);
8a8e04df	53	if (css)
88d642fa	54	idx = css->cgroup->id;
2b73bc65 NH	55	rcu_read_unlock();
2b73bc65 NH	56	return idx;
5bc1421e	57	}
51e4e7fa	58	#endif
5bc1421e	59
51e4e7fa	60	#else /* !CONFIG_NETPRIO_CGROUP */
5bc1421e	61
2b73bc65 NH	62	static inline u32 task_netprioidx(struct task_struct *p)
	63	{
	64	return 0;
	65	}
	66
6ffd4641	67	#define sock_update_netprioidx(sk)
51e4e7fa DW	68
51e4e7fa DW	69	#endif /* CONFIG_NETPRIO_CGROUP */
5bc1421e NH	70
5bc1421e NH	71	#endif /* _NET_CLS_CGROUP_H */