support binary ops with vmlinux enums

Author: varun-r-mallya

pythonbpf/functions/functions_pass.py

@@ -356,9 +356,7 @@ def process_func_body(
         builder.ret(ir.Constant(ir.IntType(64), 0))
 
 
-def process_bpf_chunk(
-    func_node, module, return_type, map_sym_tab, structs_sym_tab
-):
+def process_bpf_chunk(func_node, module, return_type, map_sym_tab, structs_sym_tab):
     """Process a single BPF chunk (function) and emit corresponding LLVM IR."""
 
     func_name = func_node.name
@@ -420,7 +418,7 @@ def func_proc(tree, module, chunks, map_sym_tab, structs_sym_tab):
             module,
             ctypes_to_ir(infer_return_type(func_node)),
             map_sym_tab,
-            structs_sym_tab
+            structs_sym_tab,
         )
 
 

pythonbpf/vmlinux_parser/vmlinux_exports_handler.py

@@ -71,7 +71,7 @@ def handle_vmlinux_struct_field(
         """Handle access to vmlinux struct fields"""
         # Check if it's a variable of vmlinux struct type
         if struct_var_name in local_sym_tab:
-            var_info = local_sym_tab[struct_var_name]
+            var_info = local_sym_tab[struct_var_name]  # noqa: F841
             # Need to check if this variable is a vmlinux struct
             # This will depend on how you track vmlinux struct types in your symbol table
             logger.info(