Brendan Gregg | ebdb242 | 2015-08-18 16:53:41 -0700 | [diff] [blame] | 1 | #!/usr/bin/python |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 2 | # |
| 3 | # disksnoop.py Trace block device I/O: basic version of iosnoop. |
| 4 | # For Linux, uses BCC, eBPF. See .c file. |
| 5 | # |
| 6 | # Written as a basic example of tracing latency. |
| 7 | # |
| 8 | # Copyright (c) 2015 Brendan Gregg. |
| 9 | # Licensed under the Apache License, Version 2.0 (the "License") |
| 10 | # |
| 11 | # 11-Aug-2015 Brendan Gregg Created this. |
| 12 | |
Brendan Gregg | 762b1b4 | 2015-08-18 16:49:48 -0700 | [diff] [blame] | 13 | from __future__ import print_function |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 14 | from bpf import BPF |
| 15 | import sys |
| 16 | |
| 17 | REQ_WRITE = 1 # from include/linux/blk_types.h |
| 18 | |
| 19 | # load BPF program |
| 20 | b = BPF(src_file="disksnoop.c") |
Brenden Blanco | 5eef65e | 2015-08-19 15:39:19 -0700 | [diff] [blame] | 21 | b.attach_kprobe(event="blk_start_request", fn_name="do_request") |
| 22 | b.attach_kprobe(event="blk_update_request", fn_name="do_completion") |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 23 | |
| 24 | # header |
Brendan Gregg | 762b1b4 | 2015-08-18 16:49:48 -0700 | [diff] [blame] | 25 | print("%-18s %-2s %-7s %8s" % ("TIME(s)", "T", "BYTES", "LAT(ms)")) |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 26 | |
| 27 | # open trace pipe |
| 28 | try: |
| 29 | trace = open("/sys/kernel/debug/tracing/trace_pipe", "r") |
| 30 | except: |
Brendan Gregg | 762b1b4 | 2015-08-18 16:49:48 -0700 | [diff] [blame] | 31 | print("ERROR: opening trace_pipe", file=sys.stderr) |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 32 | exit(1) |
| 33 | |
| 34 | # format output |
| 35 | while 1: |
Brendan Gregg | 39e1373 | 2015-08-26 20:16:29 +1000 | [diff] [blame] | 36 | (task, pid, cpu, flags, ts, msg) = b.trace_readline_fields() |
| 37 | (bytes_s, bflags_s, us_s) = msg.split() |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 38 | |
Brendan Gregg | 39e1373 | 2015-08-26 20:16:29 +1000 | [diff] [blame] | 39 | if int(bflags_s, 16) & REQ_WRITE: |
Brendan Gregg | 48fbc3e | 2015-08-18 14:56:14 -0700 | [diff] [blame] | 40 | type_s = "W" |
| 41 | elif bytes_s == "0": # see blk_fill_rwbs() for logic |
| 42 | type_s = "M" |
| 43 | else: |
| 44 | type_s = "R" |
| 45 | ms = float(int(us_s, 10)) / 1000 |
| 46 | |
Brendan Gregg | 39e1373 | 2015-08-26 20:16:29 +1000 | [diff] [blame] | 47 | print("%-18.9f %-2s %-7s %8.2f" % (ts, type_s, bytes_s, ms)) |