Commit d8234d47 authored by Chuyi Zhou's avatar Chuyi Zhou Committed by Alexei Starovoitov
Browse files

selftests/bpf: Add test for using css_task iter in sleepable progs 



This Patch add a test to prove css_task iter can be used in normal
sleepable progs.

Signed-off-by: default avatarChuyi Zhou <zhouchuyi@bytedance.com>
Acked-by: default avatarYonghong Song <yonghong.song@linux.dev>
Link: https://lore.kernel.org/r/20231031050438.93297-4-zhouchuyi@bytedance.com


Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
parent f49843af

tools/testing/selftests/bpf/prog_tests/iters.c

+1 −0
Original line number Diff line number Diff line
@@ -294,6 +294,7 @@ void test_iters(void) 
	RUN_TESTS(iters_state_safety);

	RUN_TESTS(iters_looping);

	RUN_TESTS(iters);

	RUN_TESTS(iters_css_task);



	if (env.has_testmod)

		RUN_TESTS(iters_testmod_seq);

tools/testing/selftests/bpf/progs/iters_css_task.c

+19 −0
Original line number Diff line number Diff line
@@ -89,3 +89,22 @@ int cgroup_id_printer(struct bpf_iter__cgroup *ctx) 
	bpf_cgroup_release(acquired);

	return 0;

}



SEC("?fentry.s/" SYS_PREFIX "sys_getpgid")

int BPF_PROG(iter_css_task_for_each_sleep)

{

	u64 cgrp_id = bpf_get_current_cgroup_id();

	struct cgroup *cgrp = bpf_cgroup_from_id(cgrp_id);

	struct cgroup_subsys_state *css;

	struct task_struct *task;



	if (cgrp == NULL)

		return 0;

	css = &cgrp->self;



	bpf_for_each(css_task, task, css, CSS_TASK_ITER_PROCS) {



	}

	bpf_cgroup_release(cgrp);

	return 0;

}