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 Luostatic 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 Luoint 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