[PATCH v5 4/5] cgroup: helper for checking rstat participation of css

[JP Kobryn <inwardvessel@gmail.com>]

There are a few places where a conditional check is performed to validate a
given css on its rstat participation. This new helper tries to make the
code more readable where this check is performed.

Signed-off-by: JP Kobryn <inwardvessel@gmail.com>
---
 kernel/cgroup/rstat.c | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/kernel/cgroup/rstat.c b/kernel/cgroup/rstat.c
index e1e9dd7de705..15bc7ab458dc 100644
--- a/kernel/cgroup/rstat.c
+++ b/kernel/cgroup/rstat.c
@@ -14,6 +14,17 @@ static DEFINE_PER_CPU(raw_spinlock_t, rstat_base_cpu_lock);
 
 static void cgroup_base_stat_flush(struct cgroup *cgrp, int cpu);
 
+/*
+ * Determines whether a given css can participate in rstat.
+ * css's that are cgroup::self use rstat for base stats.
+ * Other css's associated with a subsystem use rstat when they
+ * define the ss->css_rstat_flush callback.
+ */
+static inline bool is_rstat_css(struct cgroup_subsys_state *css)
+{
+	return css_is_cgroup(css) || css->ss->css_rstat_flush != NULL;
+}
+
 static struct css_rstat_cpu *css_rstat_cpu(
 		struct cgroup_subsys_state *css, int cpu)
 {
@@ -119,7 +130,7 @@ __bpf_kfunc void css_rstat_updated(struct cgroup_subsys_state *css, int cpu)
 	 * Since bpf programs can call this function, prevent access to
 	 * uninitialized rstat pointers.
 	 */
-	if (!css_is_cgroup(css) && css->ss->css_rstat_flush == NULL)
+	if (!is_rstat_css(css))
 		return;
 
 	/*
@@ -390,7 +401,7 @@ __bpf_kfunc void css_rstat_flush(struct cgroup_subsys_state *css)
 	 * Since bpf programs can call this function, prevent access to
 	 * uninitialized rstat pointers.
 	 */
-	if (!is_cgroup && css->ss->css_rstat_flush == NULL)
+	if (!is_rstat_css(css))
 		return;
 
 	might_sleep();
@@ -464,7 +475,7 @@ void css_rstat_exit(struct cgroup_subsys_state *css)
 {
 	int cpu;
 
-	if (!css_is_cgroup(css) && css->ss->css_rstat_flush == NULL)
+	if (!is_rstat_css(css))
 		return;
 
 	css_rstat_flush(css);
-- 
2.47.1