[mirror_ubuntu-artful-kernel.git] / include / net / cls_cgroup.h

/*
 * cls_cgroup.h			Control Group Classifier
 *
 * Authors:	Thomas Graf <tgraf@suug.ch>
 *
 * 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 _NET_CLS_CGROUP_H
#define _NET_CLS_CGROUP_H

#include <linux/cgroup.h>
#include <linux/hardirq.h>
#include <linux/rcupdate.h>

#if IS_ENABLED(CONFIG_NET_CLS_CGROUP)
struct cgroup_cls_state
{
	struct cgroup_subsys_state css;
	u32 classid;
};

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

#if IS_BUILTIN(CONFIG_NET_CLS_CGROUP)
static inline u32 task_cls_classid(struct task_struct *p)
{
	u32 classid;

	if (in_interrupt())
		return 0;

	rcu_read_lock();
	classid = container_of(task_subsys_state(p, net_cls_subsys_id),
			       struct cgroup_cls_state, css)->classid;
	rcu_read_unlock();

	return classid;
}
#elif IS_MODULE(CONFIG_NET_CLS_CGROUP)
static inline u32 task_cls_classid(struct task_struct *p)
{
	struct cgroup_subsys_state *css;
	u32 classid = 0;

	if (in_interrupt())
		return 0;

	rcu_read_lock();
	css = task_subsys_state(p, net_cls_subsys_id);
	if (css)
		classid = container_of(css,
				       struct cgroup_cls_state, css)->classid;
	rcu_read_unlock();

	return classid;
}
#endif
#else /* !CGROUP_NET_CLS_CGROUP */
static inline void sock_update_classid(struct sock *sk, struct task_struct *task)
{
}

static inline u32 task_cls_classid(struct task_struct *p)
{
	return 0;
}
#endif /* CGROUP_NET_CLS_CGROUP */
#endif  /* _NET_CLS_CGROUP_H */
Commit	Line	Data
f8451725 HX	1	/*
	2	* cls_cgroup.h Control Group Classifier
	3	*
	4	* Authors: Thomas Graf <tgraf@suug.ch>
	5	*
	6	* This program is free software; you can redistribute it and/or modify it
	7	* under the terms of the GNU General Public License as published by the Free
	8	* Software Foundation; either version 2 of the License, or (at your option)
	9	* any later version.
	10	*
	11	*/
	12
	13	#ifndef _NET_CLS_CGROUP_H
	14	#define _NET_CLS_CGROUP_H
	15
	16	#include <linux/cgroup.h>
	17	#include <linux/hardirq.h>
	18	#include <linux/rcupdate.h>
	19
8fb974c9	20	#if IS_ENABLED(CONFIG_NET_CLS_CGROUP)
f8451725 HX	21	struct cgroup_cls_state
	22	{
	23	struct cgroup_subsys_state css;
	24	u32 classid;
	25	};
	26
fd9a08a7	27	extern void sock_update_classid(struct sock sk, struct task_struct task);
f3419807	28
8fb974c9	29	#if IS_BUILTIN(CONFIG_NET_CLS_CGROUP)
f8451725 HX	30	static inline u32 task_cls_classid(struct task_struct *p)
f8451725 HX	31	{
920750ce	32	u32 classid;
3fb5a991	33
f8451725 HX	34	if (in_interrupt())
	35	return 0;
	36
3fb5a991 LZ	37	rcu_read_lock();
	38	classid = container_of(task_subsys_state(p, net_cls_subsys_id),
	39	struct cgroup_cls_state, css)->classid;
	40	rcu_read_unlock();
	41
	42	return classid;
f8451725	43	}
8fb974c9	44	#elif IS_MODULE(CONFIG_NET_CLS_CGROUP)
f8451725 HX	45	static inline u32 task_cls_classid(struct task_struct *p)
f8451725 HX	46	{
8a8e04df	47	struct cgroup_subsys_state *css;
ea16f912	48	u32 classid = 0;
f8451725 HX	49
	50	if (in_interrupt())
	51	return 0;
	52
	53	rcu_read_lock();
8a8e04df DW	54	css = task_subsys_state(p, net_cls_subsys_id);
	55	if (css)
	56	classid = container_of(css,
f8451725 HX	57	struct cgroup_cls_state, css)->classid;
	58	rcu_read_unlock();
	59
	60	return classid;
	61	}
	62	#endif
8fb974c9	63	#else /* !CGROUP_NET_CLS_CGROUP */
dc95a2c0	64	static inline void sock_update_classid(struct sock sk, struct task_struct task)
f3419807 DW	65	{
	66	}
	67
f8451725 HX	68	static inline u32 task_cls_classid(struct task_struct *p)
	69	{
	70	return 0;
	71	}
8fb974c9	72	#endif /* CGROUP_NET_CLS_CGROUP */
f8451725	73	#endif /* _NET_CLS_CGROUP_H */