diff options
Diffstat (limited to 'examples/tracing/disksnoop.py')
-rwxr-xr-x | examples/tracing/disksnoop.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/examples/tracing/disksnoop.py b/examples/tracing/disksnoop.py index a35e1abd..7b6891b7 100755 --- a/examples/tracing/disksnoop.py +++ b/examples/tracing/disksnoop.py @@ -19,7 +19,7 @@ REQ_WRITE = 1 # from include/linux/blk_types.h # load BPF program b = BPF(text=""" #include <uapi/linux/ptrace.h> -#include <linux/blkdev.h> +#include <linux/blk-mq.h> BPF_HASH(start, struct request *); @@ -46,7 +46,10 @@ void trace_completion(struct pt_regs *ctx, struct request *req) { if BPF.get_kprobe_functions(b'blk_start_request'): b.attach_kprobe(event="blk_start_request", fn_name="trace_start") b.attach_kprobe(event="blk_mq_start_request", fn_name="trace_start") -b.attach_kprobe(event="blk_account_io_done", fn_name="trace_completion") +if BPF.get_kprobe_functions(b'__blk_account_io_done'): + b.attach_kprobe(event="__blk_account_io_done", fn_name="trace_completion") +else: + b.attach_kprobe(event="blk_account_io_done", fn_name="trace_completion") # header print("%-18s %-2s %-7s %8s" % ("TIME(s)", "T", "BYTES", "LAT(ms)")) |