Alexey Ivanov | cc01a9c | 2019-01-16 09:50:46 -0800 | [diff] [blame] | 1 | #!/usr/bin/python |
mcaleavya | 6959dcf | 2016-02-06 12:00:34 +0000 | [diff] [blame] | 2 | # |
| 3 | # bitehist.py Block I/O size histogram. |
| 4 | # For Linux, uses BCC, eBPF. See .c file. |
| 5 | # |
| 6 | # USAGE: bitesize |
mcaleavya | 6959dcf | 2016-02-06 12:00:34 +0000 | [diff] [blame] | 7 | # |
Brendan Gregg | 6497a84 | 2016-02-06 17:08:38 -0800 | [diff] [blame] | 8 | # Ctrl-C will print the partially gathered histogram then exit. |
mcaleavya | 6959dcf | 2016-02-06 12:00:34 +0000 | [diff] [blame] | 9 | # |
| 10 | # Copyright (c) 2016 Allan McAleavy |
| 11 | # Licensed under the Apache License, Version 2.0 (the "License") |
| 12 | # |
| 13 | # 05-Feb-2016 Allan McAleavy ran pep8 against file |
Brendan Gregg | c7736ee | 2019-03-20 09:01:57 -0700 | [diff] [blame] | 14 | # 19-Mar-2019 Brendan Gregg Switched to use tracepoints. |
mcaleavya | 6959dcf | 2016-02-06 12:00:34 +0000 | [diff] [blame] | 15 | |
| 16 | from bcc import BPF |
| 17 | from time import sleep |
| 18 | |
| 19 | bpf_text = """ |
| 20 | #include <uapi/linux/ptrace.h> |
| 21 | #include <linux/blkdev.h> |
| 22 | |
Brendan Gregg | f6e92c9 | 2016-02-06 17:11:59 -0800 | [diff] [blame] | 23 | struct proc_key_t { |
mcaleavya | 6959dcf | 2016-02-06 12:00:34 +0000 | [diff] [blame] | 24 | char name[TASK_COMM_LEN]; |
| 25 | u64 slot; |
| 26 | }; |
| 27 | |
mcaleavya | 6959dcf | 2016-02-06 12:00:34 +0000 | [diff] [blame] | 28 | BPF_HISTOGRAM(dist, struct proc_key_t); |
mcaleavya | 6959dcf | 2016-02-06 12:00:34 +0000 | [diff] [blame] | 29 | |
Brendan Gregg | 6b05e61 | 2019-03-20 21:13:59 -0700 | [diff] [blame] | 30 | TRACEPOINT_PROBE(block, block_rq_issue) |
mcaleavya | 6959dcf | 2016-02-06 12:00:34 +0000 | [diff] [blame] | 31 | { |
Brendan Gregg | c7736ee | 2019-03-20 09:01:57 -0700 | [diff] [blame] | 32 | struct proc_key_t key = {.slot = bpf_log2l(args->bytes / 1024)}; |
Sumanth Korikkar | 7f6066d | 2020-05-20 10:49:56 -0500 | [diff] [blame] | 33 | bpf_probe_read_kernel(&key.name, sizeof(key.name), args->comm); |
zcy | 80242fb | 2021-07-02 00:12:32 +0800 | [diff] [blame] | 34 | dist.atomic_increment(key); |
Brendan Gregg | 6497a84 | 2016-02-06 17:08:38 -0800 | [diff] [blame] | 35 | return 0; |
mcaleavya | 6959dcf | 2016-02-06 12:00:34 +0000 | [diff] [blame] | 36 | } |
| 37 | """ |
| 38 | |
| 39 | # load BPF program |
| 40 | b = BPF(text=bpf_text) |
mcaleavya | 6959dcf | 2016-02-06 12:00:34 +0000 | [diff] [blame] | 41 | |
Brendan Gregg | c7736ee | 2019-03-20 09:01:57 -0700 | [diff] [blame] | 42 | print("Tracing block I/O... Hit Ctrl-C to end.") |
mcaleavya | 6959dcf | 2016-02-06 12:00:34 +0000 | [diff] [blame] | 43 | |
| 44 | # trace until Ctrl-C |
| 45 | dist = b.get_table("dist") |
| 46 | |
| 47 | try: |
| 48 | sleep(99999999) |
| 49 | except KeyboardInterrupt: |
Rafael F | 78948e4 | 2017-03-26 14:54:25 +0200 | [diff] [blame] | 50 | dist.print_log2_hist("Kbytes", "Process Name", |
| 51 | section_print_fn=bytes.decode) |