cgroup: improve css_from_dir() into css_tryget_from_dir()

css_from_dir() returns the matching css (cgroup_subsys_state) given a
dentry and subsystem.  The function doesn't pin the css before
returning and requires the caller to be holding RCU read lock or
cgroup_mutex and handling pinning on the caller side.

Given that users of the function are likely to want to pin the
returned css (both existing users do) and that getting and putting
css's are very cheap, there's no reason for the interface to be tricky
like this.

Rename css_from_dir() to css_tryget_from_dir() and make it try to pin
the found css and return it only if pinning succeeded.  The callers
are updated so that they no longer do RCU locking and pinning around
the function and just use the returned css.

This will also ease converting cgroup to kernfs.

Signed-off-by: Tejun Heo <tj@kernel.org>
Acked-by: Michal Hocko <mhocko@suse.cz>
Acked-by: Li Zefan <lizefan@huawei.com>
Cc: Steven Rostedt <rostedt@goodmis.org>
Cc: Frederic Weisbecker <fweisbec@gmail.com>
Cc: Ingo Molnar <mingo@redhat.com>
Cc: Johannes Weiner <hannes@cmpxchg.org>
Cc: Balbir Singh <bsingharora@gmail.com>
Cc: KAMEZAWA Hiroyuki <kamezawa.hiroyu@jp.fujitsu.com>

1 files changed, 7 insertions, 9 deletions

diff --git a/mm/memcontrol.c b/mm/memcontrol.c
index 04a97bce2270..102ab48ffa13 100644
--- a/mm/memcontrol.c
+++ b/mm/memcontrol.c
@@ -6183,17 +6183,15 @@ static int memcg_write_event_control(struct cgroup_subsys_state *css,
          * automatically removed on cgroup destruction but the removal is
          * asynchronous, so take an extra ref on @css.
          */
-        rcu_read_lock();
-
+        cfile_css = css_tryget_from_dir(cfile.file->f_dentry->d_parent,
+                                        &memory_cgrp_subsys);
         ret = -EINVAL;
-        cfile_css = css_from_dir(cfile.file->f_dentry->d_parent,
-                                 &memory_cgrp_subsys);
-        if (cfile_css == css && css_tryget(css))
-                ret = 0;
-
-        rcu_read_unlock();
-        if (ret)
+        if (IS_ERR(cfile_css))
                 goto out_put_cfile;
+        if (cfile_css != css) {
+                css_put(cfile_css);
+                goto out_put_cfile;
+        }
 
         ret = event->register_event(memcg, event->eventfd, buffer);
         if (ret)