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> |
| 23 | #include <linux/blkdev.h> |
| 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 | |
| 28 | int kprobe__blk_account_io_completion(struct pt_regs *ctx, struct request *req) |
| 29 | { |
| 30 | dist.increment(bpf_log2l(req->__data_len / 1024)); |
Xiaozhou Liu | 10c8bd3 | 2019-02-03 15:50:21 +0800 | [diff] [blame^] | 31 | dist_linear.increment(req->__data_len / 1024); |
Brendan Gregg | 310ab53 | 2016-07-24 13:34:40 -0700 | [diff] [blame] | 32 | return 0; |
| 33 | } |
| 34 | """) |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 35 | |
| 36 | # header |
| 37 | print("Tracing... Hit Ctrl-C to end.") |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 38 | |
Brendan Gregg | 0823f56 | 2015-09-25 11:07:35 -0700 | [diff] [blame] | 39 | # trace until Ctrl-C |
Brendan Gregg | f32a67c | 2015-09-07 14:42:12 -0700 | [diff] [blame] | 40 | try: |
| 41 | sleep(99999999) |
| 42 | except KeyboardInterrupt: |
Javier Honduvilla Coto | fe966bb | 2018-05-10 20:45:45 +0200 | [diff] [blame] | 43 | print() |
Brendan Gregg | f32a67c | 2015-09-07 14:42:12 -0700 | [diff] [blame] | 44 | |
Brendan Gregg | 0823f56 | 2015-09-25 11:07:35 -0700 | [diff] [blame] | 45 | # output |
Xiaozhou Liu | 10c8bd3 | 2019-02-03 15:50:21 +0800 | [diff] [blame^] | 46 | print("log2 histogram") |
| 47 | print("~~~~~~~~~~~~~~") |
Brendan Gregg | 665c5b0 | 2015-09-21 11:55:52 -0700 | [diff] [blame] | 48 | b["dist"].print_log2_hist("kbytes") |
Xiaozhou Liu | 10c8bd3 | 2019-02-03 15:50:21 +0800 | [diff] [blame^] | 49 | |
| 50 | print("\nlinear histogram") |
| 51 | print("~~~~~~~~~~~~~~~~") |
| 52 | b["dist_linear"].print_linear_hist("kbytes") |