Merge branch 'master' into bcc_examples

pythonbpf/allocation_pass.py

@@ -5,6 +5,7 @@ from llvmlite import ir
 from dataclasses import dataclass
 from typing import Any
 from pythonbpf.helper import HelperHandlerRegistry
+from .expr import VmlinuxHandlerRegistry
 from pythonbpf.type_deducer import ctypes_to_ir
 
 logger = logging.getLogger(__name__)
@@ -64,6 +65,15 @@ def handle_assign_allocation(builder, stmt, local_sym_tab, structs_sym_tab):
         if var_name in local_sym_tab:
             logger.debug(f"Variable {var_name} already allocated, skipping")
             continue
+            
+        # When allocating a variable, check if it's a vmlinux struct type
+        if isinstance(stmt.value, ast.Name) and VmlinuxHandlerRegistry.is_vmlinux_struct(
+            stmt.value.id
+        ):
+            # Handle vmlinux struct allocation
+            # This requires more implementation
+            print(stmt.value)
+            pass
 
         # Determine type and allocate based on rval
         if isinstance(rval, ast.Call):
@@ -85,7 +95,6 @@ def handle_assign_allocation(builder, stmt, local_sym_tab, structs_sym_tab):
                 f"Unsupported assignment value type for {var_name}: {type(rval).__name__}"
             )
 
-
 def _allocate_for_call(builder, var_name, rval, local_sym_tab, structs_sym_tab):
     """Allocate memory for variable assigned from a call."""