format chore

Author: varun-r-mallya

pythonbpf/binary_ops.py

@@ -44,12 +44,12 @@ def get_operand_value(operand, builder, local_sym_tab):
 def handle_binary_op_impl(rval, builder, local_sym_tab):
     """
     Handle binary operations and emit corresponding LLVM IR instructions.
-    
+
     Args:
         rval: The AST BinOp node representing the binary operation
         builder: LLVM IR builder for emitting instructions
         local_sym_tab: Symbol table mapping variable names to their IR representations
-    
+
     Returns:
         The LLVM IR value representing the result of the binary operation
     """
@@ -83,13 +83,13 @@ def handle_binary_op_impl(rval, builder, local_sym_tab):
 def handle_binary_op(rval, builder, var_name, local_sym_tab):
     """
     Handle binary operations and optionally store the result to a variable.
-    
+
     Args:
         rval: The AST BinOp node representing the binary operation
         builder: LLVM IR builder for emitting instructions
         var_name: Optional variable name to store the result
         local_sym_tab: Symbol table mapping variable names to their IR representations
-    
+
     Returns:
         A tuple of (result_value, result_type)
     """

pythonbpf/codegen.py

@@ -47,7 +47,7 @@ def find_bpf_chunks(tree):
 def processor(source_code, filename, module):
     """
     Process Python source code and convert BPF-decorated functions to LLVM IR.
-    
+
     Args:
         source_code: The Python source code to process
         filename: The name of the source file
@@ -74,12 +74,12 @@ def processor(source_code, filename, module):
 def compile_to_ir(filename: str, output: str, loglevel=logging.INFO):
     """
     Compile a Python BPF program to LLVM IR.
-    
+
     Args:
         filename: Path to the Python source file containing BPF programs
         output: Path where the LLVM IR (.ll) file will be written
         loglevel: Logging level for compilation messages
-    
+
     Returns:
         Path to the generated LLVM IR file
     """
@@ -158,13 +158,13 @@ def compile_to_ir(filename: str, output: str, loglevel=logging.INFO):
 def compile(loglevel=logging.INFO) -> bool:
     """
     Compile the calling Python BPF program to an object file.
-    
+
     This function should be called from a Python file containing BPF programs.
     It will compile the calling file to LLVM IR and then to a BPF object file.
-    
+
     Args:
         loglevel: Logging level for compilation messages
-    
+
     Returns:
         True if compilation succeeded, False otherwise
     """
@@ -203,13 +203,13 @@ def compile(loglevel=logging.INFO) -> bool:
 def BPF(loglevel=logging.INFO) -> BpfProgram:
     """
     Compile the calling Python BPF program and return a BpfProgram object.
-    
+
     This function compiles the calling file's BPF programs to an object file
     and loads it into a BpfProgram object for immediate use.
-    
+
     Args:
         loglevel: Logging level for compilation messages
-    
+
     Returns:
         A BpfProgram object that can be used to load and attach BPF programs
     """

pythonbpf/debuginfo/debug_info_generator.py

@@ -10,15 +10,15 @@
 class DebugInfoGenerator:
     """
     Generator for DWARF/BTF debug information in LLVM IR modules.
-    
+
     This class provides methods to create debug metadata for BPF programs,
     including types, structs, globals, and compilation units.
     """
-    
+
     def __init__(self, module):
         """
         Initialize the debug info generator.
-        
+
         Args:
             module: LLVM IR module to attach debug info to
         """
@@ -28,7 +28,7 @@ def __init__(self, module):
     def generate_file_metadata(self, filename, dirname):
         """
         Generate file metadata for debug info.
-        
+
         Args:
             filename: Name of the source file
             dirname: Directory containing the source file
@@ -46,7 +46,7 @@ def generate_debug_cu(
     ):
         """
         Generate debug compile unit metadata.
-        
+
         Args:
             language: DWARF language code (e.g., DW_LANG_C11)
             producer: Compiler/producer string
@@ -114,11 +114,11 @@ def create_array_type(self, base_type: Any, count: int) -> Any:
     def _compute_array_size(base_type: Any, count: int) -> int:
         """
         Compute the size of an array in bits.
-        
+
         Args:
             base_type: The base type of the array
             count: Number of elements in the array
-        
+
         Returns:
             Total size in bits
         """

pythonbpf/debuginfo/dtypes.py

@@ -5,6 +5,7 @@
 
 class DwarfBehaviorEnum:
     """DWARF module flag behavior constants for LLVM."""
+
     ERROR_IF_MISMATCH = ir.Constant(ir.IntType(32), 1)
     WARNING_IF_MISMATCH = ir.Constant(ir.IntType(32), 2)
     OVERRIDE_USE_LARGEST = ir.Constant(ir.IntType(32), 7)

pythonbpf/decorators.py

@@ -33,13 +33,14 @@ def struct(cls):
 def section(name: str):
     """
     Decorator to specify the ELF section name for a BPF program.
-    
+
     Args:
         name: The section name (e.g., 'xdp', 'tracepoint/syscalls/sys_enter_execve')
-    
+
     Returns:
         A decorator function that marks the function with the section name
     """
+
     def wrapper(fn):
         """Decorator that sets the section name on the function."""
         fn._section = name

pythonbpf/expr/expr_pass.py

@@ -342,7 +342,7 @@ def eval_expr(
 ):
     """
     Evaluate an expression and return its LLVM IR value and type.
-    
+
     Args:
         func: The LLVM IR function being built
         module: The LLVM IR module
@@ -351,7 +351,7 @@ def eval_expr(
         local_sym_tab: Local symbol table
         map_sym_tab: Map symbol table
         structs_sym_tab: Struct symbol table
-    
+
     Returns:
         A tuple of (value, type) or None if evaluation fails
     """

pythonbpf/expr/type_normalization.py

@@ -26,10 +26,10 @@
 def _get_base_type_and_depth(ir_type):
     """
     Get the base type and pointer depth for an LLVM IR type.
-    
+
     Args:
         ir_type: The LLVM IR type to analyze
-    
+
     Returns:
         A tuple of (base_type, depth) where depth is the number of pointer levels
     """
@@ -44,13 +44,13 @@ def _get_base_type_and_depth(ir_type):
 def _deref_to_depth(func, builder, val, target_depth):
     """
     Dereference a pointer to a certain depth with null checks.
-    
+
     Args:
         func: The LLVM IR function being built
         builder: LLVM IR builder
         val: The pointer value to dereference
         target_depth: Number of levels to dereference
-    
+
     Returns:
         The dereferenced value, or None if dereferencing fails
     """
@@ -101,13 +101,13 @@ def _deref_to_depth(func, builder, val, target_depth):
 def _normalize_types(func, builder, lhs, rhs):
     """
     Normalize types for comparison by casting or dereferencing as needed.
-    
+
     Args:
         func: The LLVM IR function being built
         builder: LLVM IR builder
         lhs: Left-hand side value
         rhs: Right-hand side value
-    
+
     Returns:
         A tuple of (normalized_lhs, normalized_rhs) or (None, None) on error
     """
@@ -138,11 +138,11 @@ def _normalize_types(func, builder, lhs, rhs):
 def convert_to_bool(builder, val):
     """
     Convert an LLVM IR value to a boolean (i1) type.
-    
+
     Args:
         builder: LLVM IR builder
         val: The value to convert
-    
+
     Returns:
         An i1 boolean value
     """
@@ -158,14 +158,14 @@ def convert_to_bool(builder, val):
 def handle_comparator(func, builder, op, lhs, rhs):
     """
     Handle comparison operations between two values.
-    
+
     Args:
         func: The LLVM IR function being built
         builder: LLVM IR builder
         op: The AST comparison operator node
         lhs: Left-hand side value
         rhs: Right-hand side value
-    
+
     Returns:
         A tuple of (result, ir.IntType(1)) or None on error
     """

pythonbpf/functions/functions_pass.py

@@ -27,12 +27,13 @@
 class LocalSymbol:
     """
     Represents a local variable in a BPF function.
-    
+
     Attributes:
         var: LLVM IR alloca instruction for the variable
         ir_type: LLVM IR type of the variable
         metadata: Optional metadata (e.g., struct type name)
     """
+
     var: ir.AllocaInstr
     ir_type: ir.Type
     metadata: Any = None
@@ -262,7 +263,7 @@ def handle_cond(
 ):
     """
     Evaluate a condition expression and convert it to a boolean value.
-    
+
     Args:
         func: The LLVM IR function being built
         module: The LLVM IR module
@@ -271,7 +272,7 @@ def handle_cond(
         local_sym_tab: Local symbol table
         map_sym_tab: Map symbol table
         structs_sym_tab: Struct symbol table
-    
+
     Returns:
         LLVM IR boolean value representing the condition result
     """
@@ -332,13 +333,13 @@ def handle_if(
 def handle_return(builder, stmt, local_sym_tab, ret_type):
     """
     Handle return statements in BPF functions.
-    
+
     Args:
         builder: LLVM IR builder
         stmt: The AST Return node
         local_sym_tab: Local symbol table
         ret_type: Expected return type
-    
+
     Returns:
         True if a return was emitted, False otherwise
     """
@@ -375,7 +376,7 @@ def process_stmt(
 ):
     """
     Process a single statement in a BPF function.
-    
+
     Args:
         func: The LLVM IR function being built
         module: The LLVM IR module
@@ -386,7 +387,7 @@ def process_stmt(
         structs_sym_tab: Struct symbol table
         did_return: Whether a return has been emitted
         ret_type: Expected return type
-    
+
     Returns:
         True if a return was emitted, False otherwise
     """
@@ -426,10 +427,10 @@ def allocate_mem(
 ):
     """
     Pre-allocate stack memory for local variables in a BPF function.
-    
+
     This function scans the function body and creates alloca instructions
     for all local variables before processing the function statements.
-    
+
     Args:
         module: The LLVM IR module
         builder: LLVM IR builder
@@ -439,7 +440,7 @@ def allocate_mem(
         map_sym_tab: Map symbol table
         local_sym_tab: Local symbol table to populate
         structs_sym_tab: Struct symbol table
-    
+
     Returns:
         Updated local symbol table
     """
@@ -638,7 +639,7 @@ def process_bpf_chunk(func_node, module, return_type, map_sym_tab, structs_sym_t
 def func_proc(tree, module, chunks, map_sym_tab, structs_sym_tab):
     """
     Process all BPF function chunks and generate LLVM IR.
-    
+
     Args:
         tree: The Python AST (not used in current implementation)
         module: The LLVM IR module to add functions to
@@ -673,13 +674,13 @@ def func_proc(tree, module, chunks, map_sym_tab, structs_sym_tab):
 def infer_return_type(func_node: ast.FunctionDef):
     """
     Infer the return type of a BPF function from annotations or return statements.
-    
+
     Args:
         func_node: The AST function node
-    
+
     Returns:
         String representation of the return type (e.g., 'c_int64')
-    
+
     Raises:
         TypeError: If func_node is not a FunctionDef
     """