From 2b3635fe20d748b1188d4cc006336e8551b3a498 Mon Sep 17 00:00:00 2001 From: varun-r-mallya Date: Sat, 22 Nov 2025 01:48:44 +0530 Subject: [PATCH] format chore --- pythonbpf/assign_pass.py | 8 ++++++-- pythonbpf/helper/__init__.py | 6 +++++- pythonbpf/vmlinux_parser/vmlinux_exports_handler.py | 11 +++++------ 3 files changed, 16 insertions(+), 9 deletions(-) diff --git a/pythonbpf/assign_pass.py b/pythonbpf/assign_pass.py index 0d69a04..5d73cf3 100644 --- a/pythonbpf/assign_pass.py +++ b/pythonbpf/assign_pass.py @@ -150,7 +150,9 @@ def handle_variable_assignment( return False val, val_type = val_result - logger.info(f"Evaluated value for {var_name}: {val} of type {val_type}, expected {var_type}") + logger.info( + f"Evaluated value for {var_name}: {val} of type {val_type}, expected {var_type}" + ) if val_type != var_type: # Handle vmlinux struct pointers - they're represented as Python classes but are i64 pointers @@ -161,7 +163,9 @@ def handle_variable_assignment( # Convert pointer to i64 using ptrtoint if isinstance(val.type, ir.PointerType): val = builder.ptrtoint(val, ir.IntType(64)) - logger.info(f"Converted vmlinux struct pointer to i64 using ptrtoint") + logger.info( + "Converted vmlinux struct pointer to i64 using ptrtoint" + ) builder.store(val, var_ptr) logger.info(f"Assigned vmlinux struct pointer to {var_name} (i64)") return True diff --git a/pythonbpf/helper/__init__.py b/pythonbpf/helper/__init__.py index e92828a..1730635 100644 --- a/pythonbpf/helper/__init__.py +++ b/pythonbpf/helper/__init__.py @@ -1,6 +1,10 @@ from .helper_registry import HelperHandlerRegistry from .helper_utils import reset_scratch_pool -from .bpf_helper_handler import handle_helper_call, emit_probe_read_kernel_str_call, emit_probe_read_kernel_call +from .bpf_helper_handler import ( + handle_helper_call, + emit_probe_read_kernel_str_call, + emit_probe_read_kernel_call, +) from .helpers import ( ktime, pid, diff --git a/pythonbpf/vmlinux_parser/vmlinux_exports_handler.py b/pythonbpf/vmlinux_parser/vmlinux_exports_handler.py index aa6f52d..c26cac9 100644 --- a/pythonbpf/vmlinux_parser/vmlinux_exports_handler.py +++ b/pythonbpf/vmlinux_parser/vmlinux_exports_handler.py @@ -173,7 +173,9 @@ class VmlinuxHandler: if field_size_bits in [8, 16, 32, 64]: int_width = field_size_bits - logger.info(f"Determined field size: {int_width} bits ({field_size_bytes} bytes)") + logger.info( + f"Determined field size: {int_width} bits ({field_size_bytes} bytes)" + ) # Special handling for struct_xdp_md i32 fields if struct_name == "struct_xdp_md" and int_width == 32: @@ -212,11 +214,8 @@ class VmlinuxHandler: # %result = call i64 inttoptr (i64 113 to ptr)(ptr %local_storage, i32 %size, ptr %passed) from pythonbpf.helper import emit_probe_read_kernel_call - result = emit_probe_read_kernel_call( - builder, - local_storage_i8_ptr, - field_size_bytes, - field_ptr + emit_probe_read_kernel_call( + builder, local_storage_i8_ptr, field_size_bytes, field_ptr ) # Load the value from local storage