parse context from first function argument to local symbol table

pythonbpf/vmlinux_parser/import_detector.py

@@ -2,7 +2,6 @@ import ast
 import logging
 import importlib
 import inspect
-import llvmlite.ir as ir
 
 from .assignment_info import AssignmentInfo, AssignmentType
 from .dependency_handler import DependencyHandler

pythonbpf/vmlinux_parser/vmlinux_exports_handler.py

@@ -39,6 +39,16 @@ class VmlinuxHandler:
             and self.vmlinux_symtab[name]["value_type"] == AssignmentType.CONSTANT
         )
 
+    def get_vmlinux_struct_type(self, name):
+        """Check if name is a vmlinux struct type"""
+        if (
+            name in self.vmlinux_symtab
+            and self.vmlinux_symtab[name]["value_type"] == AssignmentType.STRUCT
+        ):
+            return self.vmlinux_symtab[name]["python_type"]
+        else:
+            raise ValueError(f"{name} is not a vmlinux struct type")
+
     def is_vmlinux_struct(self, name):
         """Check if name is a vmlinux struct"""
         return (