Minki/linux
1
0
Fork 1
mirror of https://github.com/torvalds/linux.git synced 2024-11-23 20:51:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

cgroup: make css_set_rwsem a spinlock and rename it to css_set_lock

Browse Source 
css_set_rwsem is the inner lock protecting css_sets and is accessed
from hot paths such as fork and exit.  Internally, it has no reason to
be a rwsem or even mutex.  There are no internal blocking operations
while holding it.  This was rwsem because css task iteration used to
expose it to external iterator users.  As the previous patch updated
css task iteration such that the locking is not leaked to its users,
there's no reason to keep it a rwsem.

This patch converts css_set_rwsem to a spinlock and rename it to
css_set_lock.  It uses bh-safe operations as a planned usage needs to
access it from RCU callback context.

Signed-off-by: Tejun Heo <tj@kernel.org>
This commit is contained in:
Tejun Heo 2015-10-15 16:41:53 -04:00
parent ed27b9f7a1
commit f0d9a5f175
2 changed files with 74 additions and 75 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
include/linux/cgroup.h
View File

@ -13,7 +13,6 @@
#include <linux/nodemask.h> #include <linux/nodemask.h>
#include <linux/rculist.h> #include <linux/rculist.h>
#include <linux/cgroupstats.h> #include <linux/cgroupstats.h>
#include <linux/rwsem.h>
#include <linux/fs.h> #include <linux/fs.h>
#include <linux/seq_file.h> #include <linux/seq_file.h>
#include <linux/kernfs.h> #include <linux/kernfs.h>
@ -367,11 +366,11 @@ static inline void css_put_many(struct cgroup_subsys_state *css, unsigned int n)
*/ */
#ifdef CONFIG_PROVE_RCU #ifdef CONFIG_PROVE_RCU
extern struct mutex cgroup_mutex; extern struct mutex cgroup_mutex;
extern struct rw_semaphore css_set_rwsem; extern spinlock_t css_set_lock;
#define task_css_set_check(task, __c) \ #define task_css_set_check(task, __c) \
rcu_dereference_check((task)->cgroups, \ rcu_dereference_check((task)->cgroups, \
lockdep_is_held(&cgroup_mutex) || \ lockdep_is_held(&cgroup_mutex) || \
lockdep_is_held(&css_set_rwsem) || \ lockdep_is_held(&css_set_lock) || \
((task)->flags & PF_EXITING) || (__c)) ((task)->flags & PF_EXITING) || (__c))
#else #else
#define task_css_set_check(task, __c) \ #define task_css_set_check(task, __c) \

144
kernel/cgroup.c
View File

@ -45,7 +45,6 @@
#include <linux/sched.h> #include <linux/sched.h>
#include <linux/slab.h> #include <linux/slab.h>
#include <linux/spinlock.h> #include <linux/spinlock.h>
#include <linux/rwsem.h>
#include <linux/percpu-rwsem.h> #include <linux/percpu-rwsem.h>
#include <linux/string.h> #include <linux/string.h>
#include <linux/sort.h> #include <linux/sort.h>
@ -76,7 +75,7 @@
* cgroup_mutex is the master lock. Any modification to cgroup or its * cgroup_mutex is the master lock. Any modification to cgroup or its
* hierarchy must be performed while holding it. * hierarchy must be performed while holding it.
* *
* css_set_rwsem protects task->cgroups pointer, the list of css_set * css_set_lock protects task->cgroups pointer, the list of css_set
* objects, and the chain of tasks off each css_set. * objects, and the chain of tasks off each css_set.
* *
* These locks are exported if CONFIG_PROVE_RCU so that accessors in * These locks are exported if CONFIG_PROVE_RCU so that accessors in
@ -84,12 +83,12 @@
*/ */
#ifdef CONFIG_PROVE_RCU #ifdef CONFIG_PROVE_RCU
DEFINE_MUTEX(cgroup_mutex); DEFINE_MUTEX(cgroup_mutex);
DECLARE_RWSEM(css_set_rwsem); DEFINE_SPINLOCK(css_set_lock);
EXPORT_SYMBOL_GPL(cgroup_mutex); EXPORT_SYMBOL_GPL(cgroup_mutex);
EXPORT_SYMBOL_GPL(css_set_rwsem); EXPORT_SYMBOL_GPL(css_set_lock);
#else #else
static DEFINE_MUTEX(cgroup_mutex); static DEFINE_MUTEX(cgroup_mutex);
static DECLARE_RWSEM(css_set_rwsem); static DEFINE_SPINLOCK(css_set_lock);
#endif #endif
/* /*
@ -605,7 +604,7 @@ static int css_set_count = 1; /* 1 for init_css_set */
*/ */
static bool css_set_populated(struct css_set *cset) static bool css_set_populated(struct css_set *cset)
{ {
lockdep_assert_held(&css_set_rwsem); lockdep_assert_held(&css_set_lock);
return !list_empty(&cset->tasks) || !list_empty(&cset->mg_tasks); return !list_empty(&cset->tasks) || !list_empty(&cset->mg_tasks);
} }
@ -628,7 +627,7 @@ static bool css_set_populated(struct css_set *cset)
*/ */
static void cgroup_update_populated(struct cgroup *cgrp, bool populated) static void cgroup_update_populated(struct cgroup *cgrp, bool populated)
{ {
lockdep_assert_held(&css_set_rwsem); lockdep_assert_held(&css_set_lock);
do { do {
bool trigger; bool trigger;
@ -660,7 +659,7 @@ static void css_set_update_populated(struct css_set *cset, bool populated)
{ {
struct cgrp_cset_link *link; struct cgrp_cset_link *link;
lockdep_assert_held(&css_set_rwsem); lockdep_assert_held(&css_set_lock);
list_for_each_entry(link, &cset->cgrp_links, cgrp_link) list_for_each_entry(link, &cset->cgrp_links, cgrp_link)
cgroup_update_populated(link->cgrp, populated); cgroup_update_populated(link->cgrp, populated);
@ -685,7 +684,7 @@ static void css_set_move_task(struct task_struct *task,
struct css_set *from_cset, struct css_set *to_cset, struct css_set *from_cset, struct css_set *to_cset,
bool use_mg_tasks) bool use_mg_tasks)
{ {
lockdep_assert_held(&css_set_rwsem); lockdep_assert_held(&css_set_lock);
if (from_cset) { if (from_cset) {
struct css_task_iter *it, *pos; struct css_task_iter *it, *pos;
@ -755,7 +754,7 @@ static void put_css_set_locked(struct css_set *cset)
struct cgroup_subsys *ss; struct cgroup_subsys *ss;
int ssid; int ssid;
lockdep_assert_held(&css_set_rwsem); lockdep_assert_held(&css_set_lock);
if (!atomic_dec_and_test(&cset->refcount)) if (!atomic_dec_and_test(&cset->refcount))
return; return;
@ -787,9 +786,9 @@ static void put_css_set(struct css_set *cset)
if (atomic_add_unless(&cset->refcount, -1, 1)) if (atomic_add_unless(&cset->refcount, -1, 1))
return; return;
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
put_css_set_locked(cset); put_css_set_locked(cset);
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
} }
/* /*
@ -1012,11 +1011,11 @@ static struct css_set *find_css_set(struct css_set *old_cset,
/* First see if we already have a cgroup group that matches /* First see if we already have a cgroup group that matches
* the desired set */ * the desired set */
down_read(&css_set_rwsem); spin_lock_bh(&css_set_lock);
cset = find_existing_css_set(old_cset, cgrp, template); cset = find_existing_css_set(old_cset, cgrp, template);
if (cset) if (cset)
get_css_set(cset); get_css_set(cset);
up_read(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
if (cset) if (cset)
return cset; return cset;
@ -1044,7 +1043,7 @@ static struct css_set *find_css_set(struct css_set *old_cset,
* find_existing_css_set() */ * find_existing_css_set() */
memcpy(cset->subsys, template, sizeof(cset->subsys)); memcpy(cset->subsys, template, sizeof(cset->subsys));
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
/* Add reference counts and links from the new css_set. */ /* Add reference counts and links from the new css_set. */
list_for_each_entry(link, &old_cset->cgrp_links, cgrp_link) { list_for_each_entry(link, &old_cset->cgrp_links, cgrp_link) {
struct cgroup *c = link->cgrp; struct cgroup *c = link->cgrp;
@ -1066,7 +1065,7 @@ static struct css_set *find_css_set(struct css_set *old_cset,
list_add_tail(&cset->e_cset_node[ssid], list_add_tail(&cset->e_cset_node[ssid],
&cset->subsys[ssid]->cgroup->e_csets[ssid]); &cset->subsys[ssid]->cgroup->e_csets[ssid]);
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
return cset; return cset;
} }
@ -1130,14 +1129,15 @@ static void cgroup_destroy_root(struct cgroup_root *root)
* Release all the links from cset_links to this hierarchy's * Release all the links from cset_links to this hierarchy's
* root cgroup * root cgroup
*/ */
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
list_for_each_entry_safe(link, tmp_link, &cgrp->cset_links, cset_link) { list_for_each_entry_safe(link, tmp_link, &cgrp->cset_links, cset_link) {
list_del(&link->cset_link); list_del(&link->cset_link);
list_del(&link->cgrp_link); list_del(&link->cgrp_link);
kfree(link); kfree(link);
} }
up_write(&css_set_rwsem);
spin_unlock_bh(&css_set_lock);
if (!list_empty(&root->root_list)) { if (!list_empty(&root->root_list)) {
list_del(&root->root_list); list_del(&root->root_list);
@ -1159,7 +1159,7 @@ static struct cgroup *cset_cgroup_from_root(struct css_set *cset,
struct cgroup *res = NULL; struct cgroup *res = NULL;
lockdep_assert_held(&cgroup_mutex); lockdep_assert_held(&cgroup_mutex);
lockdep_assert_held(&css_set_rwsem); lockdep_assert_held(&css_set_lock);
if (cset == &init_css_set) { if (cset == &init_css_set) {
res = &root->cgrp; res = &root->cgrp;
@ -1182,7 +1182,7 @@ static struct cgroup *cset_cgroup_from_root(struct css_set *cset,
/* /*
* Return the cgroup for "task" from the given hierarchy. Must be * Return the cgroup for "task" from the given hierarchy. Must be
* called with cgroup_mutex and css_set_rwsem held. * called with cgroup_mutex and css_set_lock held.
*/ */
static struct cgroup *task_cgroup_from_root(struct task_struct *task, static struct cgroup *task_cgroup_from_root(struct task_struct *task,
struct cgroup_root *root) struct cgroup_root *root)
@ -1531,11 +1531,11 @@ static int rebind_subsystems(struct cgroup_root *dst_root,
ss->root = dst_root; ss->root = dst_root;
css->cgroup = dcgrp; css->cgroup = dcgrp;
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
hash_for_each(css_set_table, i, cset, hlist) hash_for_each(css_set_table, i, cset, hlist)
list_move_tail(&cset->e_cset_node[ss->id], list_move_tail(&cset->e_cset_node[ss->id],
&dcgrp->e_csets[ss->id]); &dcgrp->e_csets[ss->id]);
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
src_root->subsys_mask &= ~(1 << ssid); src_root->subsys_mask &= ~(1 << ssid);
scgrp->subtree_control &= ~(1 << ssid); scgrp->subtree_control &= ~(1 << ssid);
@ -1812,7 +1812,7 @@ static void cgroup_enable_task_cg_lists(void)
{ {
struct task_struct *p, *g; struct task_struct *p, *g;
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
if (use_task_css_set_links) if (use_task_css_set_links)
goto out_unlock; goto out_unlock;
@ -1851,7 +1851,7 @@ static void cgroup_enable_task_cg_lists(void)
} while_each_thread(g, p); } while_each_thread(g, p);
read_unlock(&tasklist_lock); read_unlock(&tasklist_lock);
out_unlock: out_unlock:
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
} }
static void init_cgroup_housekeeping(struct cgroup *cgrp) static void init_cgroup_housekeeping(struct cgroup *cgrp)
@ -1915,7 +1915,7 @@ static int cgroup_setup_root(struct cgroup_root *root, unsigned long ss_mask)
goto out; goto out;
/* /*
* We're accessing css_set_count without locking css_set_rwsem here, * We're accessing css_set_count without locking css_set_lock here,
* but that's OK - it can only be increased by someone holding * but that's OK - it can only be increased by someone holding
* cgroup_lock, and that's us. The worst that can happen is that we * cgroup_lock, and that's us. The worst that can happen is that we
* have some link structures left over * have some link structures left over
@ -1957,13 +1957,13 @@ static int cgroup_setup_root(struct cgroup_root *root, unsigned long ss_mask)
* Link the root cgroup in this hierarchy into all the css_set * Link the root cgroup in this hierarchy into all the css_set
* objects. * objects.
*/ */
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
hash_for_each(css_set_table, i, cset, hlist) { hash_for_each(css_set_table, i, cset, hlist) {
link_css_set(&tmp_links, cset, root_cgrp); link_css_set(&tmp_links, cset, root_cgrp);
if (css_set_populated(cset)) if (css_set_populated(cset))
cgroup_update_populated(root_cgrp, true); cgroup_update_populated(root_cgrp, true);
} }
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
BUG_ON(!list_empty(&root_cgrp->self.children)); BUG_ON(!list_empty(&root_cgrp->self.children));
BUG_ON(atomic_read(&root->nr_cgrps) != 1); BUG_ON(atomic_read(&root->nr_cgrps) != 1);
@ -2196,7 +2196,7 @@ char *task_cgroup_path(struct task_struct *task, char *buf, size_t buflen)
char *path = NULL; char *path = NULL;
mutex_lock(&cgroup_mutex); mutex_lock(&cgroup_mutex);
down_read(&css_set_rwsem); spin_lock_bh(&css_set_lock);
root = idr_get_next(&cgroup_hierarchy_idr, &hierarchy_id); root = idr_get_next(&cgroup_hierarchy_idr, &hierarchy_id);
@ -2209,7 +2209,7 @@ char *task_cgroup_path(struct task_struct *task, char *buf, size_t buflen)
path = buf; path = buf;
} }
up_read(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
mutex_unlock(&cgroup_mutex); mutex_unlock(&cgroup_mutex);
return path; return path;
} }
@ -2258,7 +2258,7 @@ static void cgroup_taskset_add(struct task_struct *task,
{ {
struct css_set *cset; struct css_set *cset;
lockdep_assert_held(&css_set_rwsem); lockdep_assert_held(&css_set_lock);
/* @task either already exited or can't exit until the end */ /* @task either already exited or can't exit until the end */
if (task->flags & PF_EXITING) if (task->flags & PF_EXITING)
@ -2364,7 +2364,7 @@ static int cgroup_taskset_migrate(struct cgroup_taskset *tset,
* the new cgroup. There are no failure cases after here, so this * the new cgroup. There are no failure cases after here, so this
* is the commit point. * is the commit point.
*/ */
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
list_for_each_entry(cset, &tset->src_csets, mg_node) { list_for_each_entry(cset, &tset->src_csets, mg_node) {
list_for_each_entry_safe(task, tmp_task, &cset->mg_tasks, cg_list) { list_for_each_entry_safe(task, tmp_task, &cset->mg_tasks, cg_list) {
struct css_set *from_cset = task_css_set(task); struct css_set *from_cset = task_css_set(task);
@ -2375,7 +2375,7 @@ static int cgroup_taskset_migrate(struct cgroup_taskset *tset,
put_css_set_locked(from_cset); put_css_set_locked(from_cset);
} }
} }
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
/* /*
* Migration is committed, all target tasks are now on dst_csets. * Migration is committed, all target tasks are now on dst_csets.
@ -2399,13 +2399,13 @@ out_cancel_attach:
css->ss->cancel_attach(css, tset); css->ss->cancel_attach(css, tset);
} }
out_release_tset: out_release_tset:
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
list_splice_init(&tset->dst_csets, &tset->src_csets); list_splice_init(&tset->dst_csets, &tset->src_csets);
list_for_each_entry_safe(cset, tmp_cset, &tset->src_csets, mg_node) { list_for_each_entry_safe(cset, tmp_cset, &tset->src_csets, mg_node) {
list_splice_tail_init(&cset->mg_tasks, &cset->tasks); list_splice_tail_init(&cset->mg_tasks, &cset->tasks);
list_del_init(&cset->mg_node); list_del_init(&cset->mg_node);
} }
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
return ret; return ret;
} }
@ -2422,14 +2422,14 @@ static void cgroup_migrate_finish(struct list_head *preloaded_csets)
lockdep_assert_held(&cgroup_mutex); lockdep_assert_held(&cgroup_mutex);
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
list_for_each_entry_safe(cset, tmp_cset, preloaded_csets, mg_preload_node) { list_for_each_entry_safe(cset, tmp_cset, preloaded_csets, mg_preload_node) {
cset->mg_src_cgrp = NULL; cset->mg_src_cgrp = NULL;
cset->mg_dst_cset = NULL; cset->mg_dst_cset = NULL;
list_del_init(&cset->mg_preload_node); list_del_init(&cset->mg_preload_node);
put_css_set_locked(cset); put_css_set_locked(cset);
} }
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
} }
/** /**
@ -2455,7 +2455,7 @@ static void cgroup_migrate_add_src(struct css_set *src_cset,
struct cgroup *src_cgrp; struct cgroup *src_cgrp;
lockdep_assert_held(&cgroup_mutex); lockdep_assert_held(&cgroup_mutex);
lockdep_assert_held(&css_set_rwsem); lockdep_assert_held(&css_set_lock);
src_cgrp = cset_cgroup_from_root(src_cset, dst_cgrp->root); src_cgrp = cset_cgroup_from_root(src_cset, dst_cgrp->root);
@ -2571,7 +2571,7 @@ static int cgroup_migrate(struct task_struct *leader, bool threadgroup,
* already PF_EXITING could be freed from underneath us unless we * already PF_EXITING could be freed from underneath us unless we
* take an rcu_read_lock. * take an rcu_read_lock.
*/ */
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
rcu_read_lock(); rcu_read_lock();
task = leader; task = leader;
do { do {
@ -2580,7 +2580,7 @@ static int cgroup_migrate(struct task_struct *leader, bool threadgroup,
break; break;
} while_each_thread(leader, task); } while_each_thread(leader, task);
rcu_read_unlock(); rcu_read_unlock();
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
return cgroup_taskset_migrate(&tset, cgrp); return cgroup_taskset_migrate(&tset, cgrp);
} }
@ -2601,7 +2601,7 @@ static int cgroup_attach_task(struct cgroup *dst_cgrp,
int ret; int ret;
/* look up all src csets */ /* look up all src csets */
down_read(&css_set_rwsem); spin_lock_bh(&css_set_lock);
rcu_read_lock(); rcu_read_lock();
task = leader; task = leader;
do { do {
@ -2611,7 +2611,7 @@ static int cgroup_attach_task(struct cgroup *dst_cgrp,
break; break;
} while_each_thread(leader, task); } while_each_thread(leader, task);
rcu_read_unlock(); rcu_read_unlock();
up_read(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
/* prepare dst csets and commit */ /* prepare dst csets and commit */
ret = cgroup_migrate_prepare_dst(dst_cgrp, &preloaded_csets); ret = cgroup_migrate_prepare_dst(dst_cgrp, &preloaded_csets);
@ -2644,9 +2644,9 @@ static int cgroup_procs_write_permission(struct task_struct *task,
struct cgroup *cgrp; struct cgroup *cgrp;
struct inode *inode; struct inode *inode;
down_read(&css_set_rwsem); spin_lock_bh(&css_set_lock);
cgrp = task_cgroup_from_root(task, &cgrp_dfl_root); cgrp = task_cgroup_from_root(task, &cgrp_dfl_root);
up_read(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
while (!cgroup_is_descendant(dst_cgrp, cgrp)) while (!cgroup_is_descendant(dst_cgrp, cgrp))
cgrp = cgroup_parent(cgrp); cgrp = cgroup_parent(cgrp);
@ -2743,9 +2743,9 @@ int cgroup_attach_task_all(struct task_struct *from, struct task_struct *tsk)
if (root == &cgrp_dfl_root) if (root == &cgrp_dfl_root)
continue; continue;
down_read(&css_set_rwsem); spin_lock_bh(&css_set_lock);
from_cgrp = task_cgroup_from_root(from, root); from_cgrp = task_cgroup_from_root(from, root);
up_read(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
retval = cgroup_attach_task(from_cgrp, tsk, false); retval = cgroup_attach_task(from_cgrp, tsk, false);
if (retval) if (retval)
@ -2870,7 +2870,7 @@ static int cgroup_update_dfl_csses(struct cgroup *cgrp)
percpu_down_write(&cgroup_threadgroup_rwsem); percpu_down_write(&cgroup_threadgroup_rwsem);
/* look up all csses currently attached to @cgrp's subtree */ /* look up all csses currently attached to @cgrp's subtree */
down_read(&css_set_rwsem); spin_lock_bh(&css_set_lock);
css_for_each_descendant_pre(css, cgroup_css(cgrp, NULL)) { css_for_each_descendant_pre(css, cgroup_css(cgrp, NULL)) {
struct cgrp_cset_link *link; struct cgrp_cset_link *link;
@ -2882,14 +2882,14 @@ static int cgroup_update_dfl_csses(struct cgroup *cgrp)
cgroup_migrate_add_src(link->cset, cgrp, cgroup_migrate_add_src(link->cset, cgrp,
&preloaded_csets); &preloaded_csets);
} }
up_read(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
/* NULL dst indicates self on default hierarchy */ /* NULL dst indicates self on default hierarchy */
ret = cgroup_migrate_prepare_dst(NULL, &preloaded_csets); ret = cgroup_migrate_prepare_dst(NULL, &preloaded_csets);
if (ret) if (ret)
goto out_finish; goto out_finish;
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
list_for_each_entry(src_cset, &preloaded_csets, mg_preload_node) { list_for_each_entry(src_cset, &preloaded_csets, mg_preload_node) {
struct task_struct *task, *ntask; struct task_struct *task, *ntask;
@ -2901,7 +2901,7 @@ static int cgroup_update_dfl_csses(struct cgroup *cgrp)
list_for_each_entry_safe(task, ntask, &src_cset->tasks, cg_list) list_for_each_entry_safe(task, ntask, &src_cset->tasks, cg_list)
cgroup_taskset_add(task, &tset); cgroup_taskset_add(task, &tset);
} }
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
ret = cgroup_taskset_migrate(&tset, cgrp); ret = cgroup_taskset_migrate(&tset, cgrp);
out_finish: out_finish:
@ -3577,10 +3577,10 @@ static int cgroup_task_count(const struct cgroup *cgrp)
int count = 0; int count = 0;
struct cgrp_cset_link *link; struct cgrp_cset_link *link;
down_read(&css_set_rwsem); spin_lock_bh(&css_set_lock);
list_for_each_entry(link, &cgrp->cset_links, cset_link) list_for_each_entry(link, &cgrp->cset_links, cset_link)
count += atomic_read(&link->cset->refcount); count += atomic_read(&link->cset->refcount);
up_read(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
return count; return count;
} }
@ -3823,7 +3823,7 @@ static void css_task_iter_advance_css_set(struct css_task_iter *it)
struct cgrp_cset_link *link; struct cgrp_cset_link *link;
struct css_set *cset; struct css_set *cset;
lockdep_assert_held(&css_set_rwsem); lockdep_assert_held(&css_set_lock);
/* Advance to the next non-empty css_set */ /* Advance to the next non-empty css_set */
do { do {
@ -3881,7 +3881,7 @@ static void css_task_iter_advance(struct css_task_iter *it)
{ {
struct list_head *l = it->task_pos; struct list_head *l = it->task_pos;
lockdep_assert_held(&css_set_rwsem); lockdep_assert_held(&css_set_lock);
WARN_ON_ONCE(!l); WARN_ON_ONCE(!l);
/* /*
@ -3918,7 +3918,7 @@ void css_task_iter_start(struct cgroup_subsys_state *css,
memset(it, 0, sizeof(*it)); memset(it, 0, sizeof(*it));
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
it->ss = css->ss; it->ss = css->ss;
@ -3931,7 +3931,7 @@ void css_task_iter_start(struct cgroup_subsys_state *css,
css_task_iter_advance_css_set(it); css_task_iter_advance_css_set(it);
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
} }
/** /**
@ -3950,14 +3950,14 @@ struct task_struct *css_task_iter_next(struct css_task_iter *it)
if (it->cur_task) if (it->cur_task)
put_task_struct(it->cur_task); put_task_struct(it->cur_task);
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
it->cur_task = list_entry(it->task_pos, struct task_struct, cg_list); it->cur_task = list_entry(it->task_pos, struct task_struct, cg_list);
get_task_struct(it->cur_task); get_task_struct(it->cur_task);
css_task_iter_advance(it); css_task_iter_advance(it);
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
return it->cur_task; return it->cur_task;
} }
@ -3971,10 +3971,10 @@ struct task_struct *css_task_iter_next(struct css_task_iter *it)
void css_task_iter_end(struct css_task_iter *it) void css_task_iter_end(struct css_task_iter *it)
{ {
if (it->cur_cset) { if (it->cur_cset) {
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
list_del(&it->iters_node); list_del(&it->iters_node);
put_css_set_locked(it->cur_cset); put_css_set_locked(it->cur_cset);
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
} }
if (it->cur_task) if (it->cur_task)
@ -4003,10 +4003,10 @@ int cgroup_transfer_tasks(struct cgroup *to, struct cgroup *from)
mutex_lock(&cgroup_mutex); mutex_lock(&cgroup_mutex);
/* all tasks in @from are being moved, all csets are source */ /* all tasks in @from are being moved, all csets are source */
down_read(&css_set_rwsem); spin_lock_bh(&css_set_lock);
list_for_each_entry(link, &from->cset_links, cset_link) list_for_each_entry(link, &from->cset_links, cset_link)
cgroup_migrate_add_src(link->cset, to, &preloaded_csets); cgroup_migrate_add_src(link->cset, to, &preloaded_csets);
up_read(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
ret = cgroup_migrate_prepare_dst(to, &preloaded_csets); ret = cgroup_migrate_prepare_dst(to, &preloaded_csets);
if (ret) if (ret)
@ -5359,7 +5359,7 @@ int proc_cgroup_show(struct seq_file *m, struct pid_namespace *ns,
goto out; goto out;
mutex_lock(&cgroup_mutex); mutex_lock(&cgroup_mutex);
down_read(&css_set_rwsem); spin_lock_bh(&css_set_lock);
for_each_root(root) { for_each_root(root) {
struct cgroup_subsys *ss; struct cgroup_subsys *ss;
@ -5391,7 +5391,7 @@ int proc_cgroup_show(struct seq_file *m, struct pid_namespace *ns,
retval = 0; retval = 0;
out_unlock: out_unlock:
up_read(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
mutex_unlock(&cgroup_mutex); mutex_unlock(&cgroup_mutex);
kfree(buf); kfree(buf);
out: out:
@ -5537,7 +5537,7 @@ void cgroup_post_fork(struct task_struct *child,
* @child during its iteration. * @child during its iteration.
* *
* If we won the race, @child is associated with %current's * If we won the race, @child is associated with %current's
* css_set. Grabbing css_set_rwsem guarantees both that the * css_set. Grabbing css_set_lock guarantees both that the
* association is stable, and, on completion of the parent's * association is stable, and, on completion of the parent's
* migration, @child is visible in the source of migration or * migration, @child is visible in the source of migration or
* already in the destination cgroup. This guarantee is necessary * already in the destination cgroup. This guarantee is necessary
@ -5552,13 +5552,13 @@ void cgroup_post_fork(struct task_struct *child,
if (use_task_css_set_links) { if (use_task_css_set_links) {
struct css_set *cset; struct css_set *cset;
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
cset = task_css_set(current); cset = task_css_set(current);
if (list_empty(&child->cg_list)) { if (list_empty(&child->cg_list)) {
get_css_set(cset); get_css_set(cset);
css_set_move_task(child, NULL, cset, false); css_set_move_task(child, NULL, cset, false);
} }
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
} }
/* /*
@ -5603,9 +5603,9 @@ void cgroup_exit(struct task_struct *tsk)
cset = task_css_set(tsk); cset = task_css_set(tsk);
if (!list_empty(&tsk->cg_list)) { if (!list_empty(&tsk->cg_list)) {
down_write(&css_set_rwsem); spin_lock_bh(&css_set_lock);
css_set_move_task(tsk, cset, NULL, false); css_set_move_task(tsk, cset, NULL, false);
up_write(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
put_cset = true; put_cset = true;
} }
@ -5823,7 +5823,7 @@ static int current_css_set_cg_links_read(struct seq_file *seq, void *v)
if (!name_buf) if (!name_buf)
return -ENOMEM; return -ENOMEM;
down_read(&css_set_rwsem); spin_lock_bh(&css_set_lock);
rcu_read_lock(); rcu_read_lock();
cset = rcu_dereference(current->cgroups); cset = rcu_dereference(current->cgroups);
list_for_each_entry(link, &cset->cgrp_links, cgrp_link) { list_for_each_entry(link, &cset->cgrp_links, cgrp_link) {
@ -5834,7 +5834,7 @@ static int current_css_set_cg_links_read(struct seq_file *seq, void *v)
c->root->hierarchy_id, name_buf); c->root->hierarchy_id, name_buf);
} }
rcu_read_unlock(); rcu_read_unlock();
up_read(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
kfree(name_buf); kfree(name_buf);
return 0; return 0;
} }
@ -5845,7 +5845,7 @@ static int cgroup_css_links_read(struct seq_file *seq, void *v)
struct cgroup_subsys_state *css = seq_css(seq); struct cgroup_subsys_state *css = seq_css(seq);
struct cgrp_cset_link *link; struct cgrp_cset_link *link;
down_read(&css_set_rwsem); spin_lock_bh(&css_set_lock);
list_for_each_entry(link, &css->cgroup->cset_links, cset_link) { list_for_each_entry(link, &css->cgroup->cset_links, cset_link) {
struct css_set *cset = link->cset; struct css_set *cset = link->cset;
struct task_struct *task; struct task_struct *task;
@ -5868,7 +5868,7 @@ static int cgroup_css_links_read(struct seq_file *seq, void *v)
overflow: overflow:
seq_puts(seq, " ...\n"); seq_puts(seq, " ...\n");
} }
up_read(&css_set_rwsem); spin_unlock_bh(&css_set_lock);
return 0; return 0;
} }