1*fe0dd9d4SHao Luo // SPDX-License-Identifier: GPL-2.0
2*fe0dd9d4SHao Luo /* Copyright (c) 2022 Google */
3*fe0dd9d4SHao Luo 
4*fe0dd9d4SHao Luo #include "bpf_iter.h"
5*fe0dd9d4SHao Luo #include <bpf/bpf_helpers.h>
6*fe0dd9d4SHao Luo #include <bpf/bpf_tracing.h>
7*fe0dd9d4SHao Luo 
8*fe0dd9d4SHao Luo char _license[] SEC("license") = "GPL";
9*fe0dd9d4SHao Luo int terminate_early = 0;
10*fe0dd9d4SHao Luo u64 terminal_cgroup = 0;
11*fe0dd9d4SHao Luo 
cgroup_id(struct cgroup * cgrp)12*fe0dd9d4SHao Luo static inline u64 cgroup_id(struct cgroup *cgrp)
13*fe0dd9d4SHao Luo {
14*fe0dd9d4SHao Luo 	return cgrp->kn->id;
15*fe0dd9d4SHao Luo }
16*fe0dd9d4SHao Luo 
17*fe0dd9d4SHao Luo SEC("iter/cgroup")
cgroup_id_printer(struct bpf_iter__cgroup * ctx)18*fe0dd9d4SHao Luo int cgroup_id_printer(struct bpf_iter__cgroup *ctx)
19*fe0dd9d4SHao Luo {
20*fe0dd9d4SHao Luo 	struct seq_file *seq = ctx->meta->seq;
21*fe0dd9d4SHao Luo 	struct cgroup *cgrp = ctx->cgroup;
22*fe0dd9d4SHao Luo 
23*fe0dd9d4SHao Luo 	/* epilogue */
24*fe0dd9d4SHao Luo 	if (cgrp == NULL) {
25*fe0dd9d4SHao Luo 		BPF_SEQ_PRINTF(seq, "epilogue\n");
26*fe0dd9d4SHao Luo 		return 0;
27*fe0dd9d4SHao Luo 	}
28*fe0dd9d4SHao Luo 
29*fe0dd9d4SHao Luo 	/* prologue */
30*fe0dd9d4SHao Luo 	if (ctx->meta->seq_num == 0)
31*fe0dd9d4SHao Luo 		BPF_SEQ_PRINTF(seq, "prologue\n");
32*fe0dd9d4SHao Luo 
33*fe0dd9d4SHao Luo 	BPF_SEQ_PRINTF(seq, "%8llu\n", cgroup_id(cgrp));
34*fe0dd9d4SHao Luo 
35*fe0dd9d4SHao Luo 	if (terminal_cgroup == cgroup_id(cgrp))
36*fe0dd9d4SHao Luo 		return 1;
37*fe0dd9d4SHao Luo 
38*fe0dd9d4SHao Luo 	return terminate_early ? 1 : 0;
39*fe0dd9d4SHao Luo }
40