varun-r-mallya/python-bpf
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/Python-BPF.git synced 2025-12-31 21:06:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

add error message

Browse Source
This commit is contained in:
varun-r-mallya
2025-10-11 19:05:21 +05:30
parent 1e3d775865
commit 06773c895f
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pythonbpf/vmlinux_parser/vmlinux_class_handler.py
View File

@ -8,11 +8,13 @@ import ctypes
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@lru_cache(maxsize=1) @lru_cache(maxsize=1)
def get_module_symbols(module_name: str): def get_module_symbols(module_name: str):
imported_module = importlib.import_module(module_name) imported_module = importlib.import_module(module_name)
return [name for name in dir(imported_module)], imported_module return [name for name in dir(imported_module)], imported_module
# Recursive function that gets all the dependent classes and adds them to handler # Recursive function that gets all the dependent classes and adds them to handler
def process_vmlinux_class(node, llvm_module, handler: DependencyHandler): def process_vmlinux_class(node, llvm_module, handler: DependencyHandler):
symbols_in_module, imported_module = get_module_symbols("vmlinux") symbols_in_module, imported_module = get_module_symbols("vmlinux")
@ -64,11 +66,12 @@ def process_vmlinux_class(node, llvm_module, handler: DependencyHandler):
elif module_name == "vmlinux": elif module_name == "vmlinux":
new_dep_node.add_field(elem_name, elem_type, ready=False) new_dep_node.add_field(elem_name, elem_type, ready=False)
# Create a temporary node-like object for recursion # Create a temporary node-like object for recursion
temp_node = type('TempNode', (), {'name': elem_type.__name__ if hasattr(elem_type, '__name__') else str(elem_type)})() temp_node = type('TempNode', (),
{'name': elem_type.__name__ if hasattr(elem_type, '__name__') else str(elem_type)})()
if process_vmlinux_class(temp_node, llvm_module, handler): if process_vmlinux_class(temp_node, llvm_module, handler):
new_dep_node.set_field_ready(elem_name, True) new_dep_node.set_field_ready(elem_name, True)
else: else:
print(f"[other] {elem_name} -> {elem_type}") raise ValueError(f"{elem_name} with type {elem_type} not supported in recursive resolver")
handler.add_node(new_dep_node) handler.add_node(new_dep_node)
return True return True