From 0ce5add39b7307006861568c64e559cfd4884705 Mon Sep 17 00:00:00 2001 From: varun-r-mallya Date: Mon, 8 Dec 2025 16:22:30 +0530 Subject: [PATCH] compact the disksnoop.py example Signed-off-by: varun-r-mallya --- BCC-Examples/disksnoop.py | 20 +++++--------------- 1 file changed, 5 insertions(+), 15 deletions(-) diff --git a/BCC-Examples/disksnoop.py b/BCC-Examples/disksnoop.py index 73a737b..47a12e5 100644 --- a/BCC-Examples/disksnoop.py +++ b/BCC-Examples/disksnoop.py @@ -1,11 +1,10 @@ -from vmlinux import struct_request, struct_pt_regs -from pythonbpf import bpf, section, bpfglobal, compile, map +from ctypes import c_int32, c_int64, c_uint64 + +from vmlinux import struct_pt_regs, struct_request + +from pythonbpf import bpf, bpfglobal, compile, map, section from pythonbpf.helper import ktime from pythonbpf.maps import HashMap -from ctypes import c_int64, c_uint64, c_int32 - -# Constants -REQ_WRITE = 1 # from include/linux/blk_types.h @bpf @@ -17,24 +16,15 @@ def start() -> HashMap: @bpf @section("kprobe/blk_mq_end_request") def trace_completion(ctx: struct_pt_regs) -> c_int64: - # Get request pointer from first argument req_ptr = ctx.di req = struct_request(ctx.di) - # Print: data_len, cmd_flags, latency_us data_len = req.__data_len cmd_flags = req.cmd_flags - # Lookup start timestamp req_tsp = start.lookup(req_ptr) if req_tsp: - # Calculate delta in nanoseconds delta = ktime() - req_tsp - - # Convert to microseconds for printing delta_us = delta // 1000 - print(f"{data_len} {cmd_flags:x} {delta_us}\n") - - # Delete the entry start.delete(req_ptr) return c_int64(0)