add debug info storage on assignment_info.py dataclass

Author: varun-r-mallya

pythonbpf/expr/vmlinux_registry.py

@@ -39,6 +39,12 @@ def handle_attribute(cls, expr, local_sym_tab, module, builder):
             )
         return None
 
+    @classmethod
+    def get_struct_debug_info(cls, name):
+        if cls._handler is None:
+            return False
+        return cls._handler.get_struct_debug_info(name)
+
     @classmethod
     def is_vmlinux_struct(cls, name):
         """Check if a name refers to a vmlinux struct"""

pythonbpf/functions/function_debug_info.py

@@ -2,8 +2,20 @@
 
 import llvmlite.ir as ir
 
+from pythonbpf.debuginfo import DebugInfoGenerator
+from pythonbpf.expr import VmlinuxHandlerRegistry
+
 
 def generate_function_debug_info(
     func_node: ast.FunctionDef, module: ir.Module, func: ir.Function
 ):
+    generator = DebugInfoGenerator(module)
+    leading_argument = func_node.args.args[0]
+    leading_argument_name = leading_argument.arg
+    # TODO: add ctypes handling as well here
+    print(leading_argument.arg, leading_argument.annotation.id)
+    context_debug_info = VmlinuxHandlerRegistry.get_struct_debug_info(
+        name=leading_argument.annotation.id
+    )
+    print(context_debug_info)
     pass

pythonbpf/vmlinux_parser/assignment_info.py

@@ -33,3 +33,4 @@ class AssignmentInfo:
     #   Value is a tuple that contains the global variable representing that field
     #   along with all the information about that field as a Field type.
     members: Optional[Dict[str, tuple[ir.GlobalVariable, Field]]]  # For structs.
+    debug_info: Any

pythonbpf/vmlinux_parser/import_detector.py

@@ -148,6 +148,7 @@ def process_vmlinux_assign(node, module, assignments: dict[str, AssignmentInfo])
                 pointer_level=None,
                 signature=None,
                 members=None,
+                debug_info=None,
             )
             logger.info(
                 f"Added assignment: {target_name} = {node.value.value!r} of type {type(node.value.value)}"

pythonbpf/vmlinux_parser/ir_gen/ir_generation.py

@@ -73,9 +73,8 @@ def struct_processor(self, struct, processing_stack=None):
                             )
 
             # Generate IR first to populate field names
-            self.generated_debug_info.append(
-                (struct, self.gen_ir(struct, self.generated_debug_info))
-            )
+            struct_debug_info = self.gen_ir(struct, self.generated_debug_info)
+            self.generated_debug_info.append((struct, struct_debug_info))
 
             # Fill the assignments dictionary with struct information
             if struct.name not in self.assignments:
@@ -105,6 +104,7 @@ def struct_processor(self, struct, processing_stack=None):
                     pointer_level=None,
                     signature=None,
                     members=members_dict,
+                    debug_info=struct_debug_info,
                 )
                 logger.info(f"Added struct assignment info for {struct.name}")
 

pythonbpf/vmlinux_parser/vmlinux_exports_handler.py

@@ -1,4 +1,6 @@
 import logging
+from typing import Any
+
 from llvmlite import ir
 
 from pythonbpf.local_symbol import LocalSymbol
@@ -40,6 +42,15 @@ def is_vmlinux_enum(self, name):
             and self.vmlinux_symtab[name].value_type == AssignmentType.CONSTANT
         )
 
+    def get_struct_debug_info(self, name: str) -> Any:
+        if (
+            name in self.vmlinux_symtab
+            and self.vmlinux_symtab[name].value_type == AssignmentType.STRUCT
+        ):
+            return self.vmlinux_symtab[name].debug_info
+        else:
+            raise ValueError(f"{name} is not a vmlinux struct type")
+
     def get_vmlinux_struct_type(self, name):
         """Check if name is a vmlinux struct type"""
         if (