[deliverable/linux.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[];
};

struct cgroup_netprio_state {
	struct cgroup_subsys_state css;
};

extern void sock_update_netprioidx(struct sock *sk, struct task_struct *task);

#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_subsys_state(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_subsys_state(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, task)

#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
8c215228 NH	28	struct cgroup_netprio_state {
8c215228 NH	29	struct cgroup_subsys_state css;
8c215228 NH	30	};
8c215228 NH	31
406a3c63	32	extern void sock_update_netprioidx(struct sock sk, struct task_struct task);
5bc1421e	33
2b73bc65 NH	34	#if IS_BUILTIN(CONFIG_NETPRIO_CGROUP)
	35
	36	static inline u32 task_netprioidx(struct task_struct *p)
5bc1421e	37	{
88d642fa	38	struct cgroup_subsys_state *css;
2b73bc65 NH	39	u32 idx;
	40
	41	rcu_read_lock();
88d642fa TH	42	css = task_subsys_state(p, net_prio_subsys_id);
88d642fa TH	43	idx = css->cgroup->id;
2b73bc65 NH	44	rcu_read_unlock();
	45	return idx;
	46	}
	47
	48	#elif IS_MODULE(CONFIG_NETPRIO_CGROUP)
	49
	50	static inline u32 task_netprioidx(struct task_struct *p)
	51	{
8a8e04df	52	struct cgroup_subsys_state *css;
2b73bc65 NH	53	u32 idx = 0;
	54
	55	rcu_read_lock();
8a8e04df DW	56	css = task_subsys_state(p, net_prio_subsys_id);
8a8e04df DW	57	if (css)
88d642fa	58	idx = css->cgroup->id;
2b73bc65 NH	59	rcu_read_unlock();
2b73bc65 NH	60	return idx;
5bc1421e	61	}
51e4e7fa	62	#endif
5bc1421e	63
51e4e7fa	64	#else /* !CONFIG_NETPRIO_CGROUP */
5bc1421e	65
2b73bc65 NH	66	static inline u32 task_netprioidx(struct task_struct *p)
	67	{
	68	return 0;
	69	}
	70
406a3c63	71	#define sock_update_netprioidx(sk, task)
51e4e7fa DW	72
51e4e7fa DW	73	#endif /* CONFIG_NETPRIO_CGROUP */
5bc1421e NH	74
5bc1421e NH	75	#endif /* _NET_CLS_CGROUP_H */