continue adding --ebpf option to the python tools/ scripts
Several python tools allow their eBPF code to be printed to
stdout for debugging. There are other projects that would
like to share these program definitions however, instead of
duplicating code. We previously agreed on an --ebpf option
and we now continue adding it to more tools.
Signed-off-by: Nathan Scott <nathans@redhat.com>
diff --git a/tools/cpudist.py b/tools/cpudist.py
index 47658c0..4d7c9eb 100755
--- a/tools/cpudist.py
+++ b/tools/cpudist.py
@@ -44,6 +44,8 @@
help="output interval, in seconds")
parser.add_argument("count", nargs="?", default=99999999,
help="number of outputs")
+parser.add_argument("--ebpf", action="store_true",
+ help=argparse.SUPPRESS)
args = parser.parse_args()
countdown = int(args.count)
debug = 0
@@ -149,8 +151,10 @@
bpf_text = bpf_text.replace('STORAGE', 'BPF_HISTOGRAM(dist);')
bpf_text = bpf_text.replace('STORE',
'dist.increment(bpf_log2l(delta));')
-if debug:
+if debug or args.ebpf:
print(bpf_text)
+ if args.ebpf:
+ exit()
b = BPF(text=bpf_text)
b.attach_kprobe(event="finish_task_switch", fn_name="sched_switch")