Alexey Ivanov | cc01a9c | 2019-01-16 09:50:46 -0800 | [diff] [blame] | 1 | #!/usr/bin/python |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 2 | # |
| 3 | # bitehist.py Block I/O size histogram. |
Brendan Gregg | 310ab53 | 2016-07-24 13:34:40 -0700 | [diff] [blame] | 4 | # For Linux, uses BCC, eBPF. Embedded C. |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 5 | # |
Xiaozhou Liu | 10c8bd3 | 2019-02-03 15:50:21 +0800 | [diff] [blame] | 6 | # Written as a basic example of using histograms to show a distribution. |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 7 | # |
Xiaozhou Liu | 10c8bd3 | 2019-02-03 15:50:21 +0800 | [diff] [blame] | 8 | # A Ctrl-C will print the gathered histogram then exit. |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 9 | # |
| 10 | # Copyright (c) 2015 Brendan Gregg. |
| 11 | # Licensed under the Apache License, Version 2.0 (the "License") |
| 12 | # |
| 13 | # 15-Aug-2015 Brendan Gregg Created this. |
Xiaozhou Liu | 10c8bd3 | 2019-02-03 15:50:21 +0800 | [diff] [blame] | 14 | # 03-Feb-2019 Xiaozhou Liu added linear histogram. |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 15 | |
Javier Honduvilla Coto | fe966bb | 2018-05-10 20:45:45 +0200 | [diff] [blame] | 16 | from __future__ import print_function |
Brenden Blanco | c35989d | 2015-09-02 18:04:07 -0700 | [diff] [blame] | 17 | from bcc import BPF |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 18 | from time import sleep |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 19 | |
| 20 | # load BPF program |
Brendan Gregg | 310ab53 | 2016-07-24 13:34:40 -0700 | [diff] [blame] | 21 | b = BPF(text=""" |
| 22 | #include <uapi/linux/ptrace.h> |
xingfeng2510 | 109453e | 2022-03-04 16:57:51 +0800 | [diff] [blame] | 23 | #include <linux/blk-mq.h> |
Brendan Gregg | 310ab53 | 2016-07-24 13:34:40 -0700 | [diff] [blame] | 24 | |
| 25 | BPF_HISTOGRAM(dist); |
Xiaozhou Liu | 10c8bd3 | 2019-02-03 15:50:21 +0800 | [diff] [blame] | 26 | BPF_HISTOGRAM(dist_linear); |
Brendan Gregg | 310ab53 | 2016-07-24 13:34:40 -0700 | [diff] [blame] | 27 | |
xingfeng2510 | 109453e | 2022-03-04 16:57:51 +0800 | [diff] [blame] | 28 | int trace_req_done(struct pt_regs *ctx, struct request *req) |
Brendan Gregg | 310ab53 | 2016-07-24 13:34:40 -0700 | [diff] [blame] | 29 | { |
xingfeng2510 | 109453e | 2022-03-04 16:57:51 +0800 | [diff] [blame] | 30 | dist.increment(bpf_log2l(req->__data_len / 1024)); |
| 31 | dist_linear.increment(req->__data_len / 1024); |
| 32 | return 0; |
Brendan Gregg | 310ab53 | 2016-07-24 13:34:40 -0700 | [diff] [blame] | 33 | } |
| 34 | """) |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 35 | |
xingfeng2510 | 109453e | 2022-03-04 16:57:51 +0800 | [diff] [blame] | 36 | if BPF.get_kprobe_functions(b'__blk_account_io_done'): |
| 37 | b.attach_kprobe(event="__blk_account_io_done", fn_name="trace_req_done") |
| 38 | else: |
| 39 | b.attach_kprobe(event="blk_account_io_done", fn_name="trace_req_done") |
| 40 | |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 41 | # header |
| 42 | print("Tracing... Hit Ctrl-C to end.") |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 43 | |
Brendan Gregg | 0823f56 | 2015-09-25 11:07:35 -0700 | [diff] [blame] | 44 | # trace until Ctrl-C |
Brendan Gregg | f32a67c | 2015-09-07 14:42:12 -0700 | [diff] [blame] | 45 | try: |
xingfeng2510 | 109453e | 2022-03-04 16:57:51 +0800 | [diff] [blame] | 46 | sleep(99999999) |
Brendan Gregg | f32a67c | 2015-09-07 14:42:12 -0700 | [diff] [blame] | 47 | except KeyboardInterrupt: |
xingfeng2510 | 109453e | 2022-03-04 16:57:51 +0800 | [diff] [blame] | 48 | print() |
Brendan Gregg | f32a67c | 2015-09-07 14:42:12 -0700 | [diff] [blame] | 49 | |
Brendan Gregg | 0823f56 | 2015-09-25 11:07:35 -0700 | [diff] [blame] | 50 | # output |
Xiaozhou Liu | 10c8bd3 | 2019-02-03 15:50:21 +0800 | [diff] [blame] | 51 | print("log2 histogram") |
| 52 | print("~~~~~~~~~~~~~~") |
Brendan Gregg | 665c5b0 | 2015-09-21 11:55:52 -0700 | [diff] [blame] | 53 | b["dist"].print_log2_hist("kbytes") |
Xiaozhou Liu | 10c8bd3 | 2019-02-03 15:50:21 +0800 | [diff] [blame] | 54 | |
| 55 | print("\nlinear histogram") |
| 56 | print("~~~~~~~~~~~~~~~~") |
| 57 | b["dist_linear"].print_linear_hist("kbytes") |