blob: 9cf2d5ef38d967ad7d67202bbd82f3bd933a31dd [file] [log] [blame]
Herbert Xuf8451722010-05-24 00:12:34 -07001/*
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>
Daniel Borkmannfe1217c2013-12-29 18:27:10 +010019#include <net/sock.h>
Herbert Xuf8451722010-05-24 00:12:34 -070020
Daniel Borkmannfe1217c2013-12-29 18:27:10 +010021#ifdef CONFIG_CGROUP_NET_CLASSID
22struct cgroup_cls_state {
Herbert Xuf8451722010-05-24 00:12:34 -070023 struct cgroup_subsys_state css;
24 u32 classid;
25};
26
Daniel Borkmannfe1217c2013-12-29 18:27:10 +010027struct cgroup_cls_state *task_cls_state(struct task_struct *p);
Daniel Wagnerf3419802012-09-12 16:12:01 +020028
Herbert Xuf8451722010-05-24 00:12:34 -070029static inline u32 task_cls_classid(struct task_struct *p)
30{
Daniel Wagner920750c2012-10-25 04:16:56 +000031 u32 classid;
Li Zefan3fb5a992010-09-02 15:42:43 +000032
Herbert Xuf8451722010-05-24 00:12:34 -070033 if (in_interrupt())
34 return 0;
35
Li Zefan3fb5a992010-09-02 15:42:43 +000036 rcu_read_lock();
Tejun Heo8af01f52013-08-08 20:11:22 -040037 classid = container_of(task_css(p, net_cls_subsys_id),
Li Zefan3fb5a992010-09-02 15:42:43 +000038 struct cgroup_cls_state, css)->classid;
39 rcu_read_unlock();
40
41 return classid;
Herbert Xuf8451722010-05-24 00:12:34 -070042}
Daniel Borkmannfe1217c2013-12-29 18:27:10 +010043
44static inline void sock_update_classid(struct sock *sk)
Herbert Xuf8451722010-05-24 00:12:34 -070045{
Daniel Borkmannfe1217c2013-12-29 18:27:10 +010046 u32 classid;
Herbert Xuf8451722010-05-24 00:12:34 -070047
Daniel Borkmannfe1217c2013-12-29 18:27:10 +010048 classid = task_cls_classid(current);
49 if (classid != sk->sk_classid)
50 sk->sk_classid = classid;
Herbert Xuf8451722010-05-24 00:12:34 -070051}
Daniel Borkmannfe1217c2013-12-29 18:27:10 +010052#else /* !CONFIG_CGROUP_NET_CLASSID */
Zefan Li211d2f972013-04-08 20:03:35 +000053static inline void sock_update_classid(struct sock *sk)
Daniel Wagnerf3419802012-09-12 16:12:01 +020054{
55}
Daniel Borkmannfe1217c2013-12-29 18:27:10 +010056#endif /* CONFIG_CGROUP_NET_CLASSID */
Herbert Xuf8451722010-05-24 00:12:34 -070057#endif /* _NET_CLS_CGROUP_H */