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

try to separate out ast node from vmlinux type

Browse Source
This commit is contained in:
varun-r-mallya
2025-10-12 01:59:14 +05:30
parent 8743ea17f3
commit 8774277000
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pythonbpf/vmlinux_parser/class_handler.py
View File

@ -13,9 +13,16 @@ def get_module_symbols(module_name: str):
imported_module = importlib.import_module(module_name)
return [name for name in dir(imported_module)], imported_module
def process_vmlinux_class(node, llvm_module, handler: DependencyHandler):
symbols_in_module, imported_module = get_module_symbols("vmlinux")
if node.name in symbols_in_module:
vmlinux_type = getattr(imported_module, node.name)
process_vmlinux_post_ast(vmlinux_type, llvm_module, handler)
else:
raise ImportError(f"{node.name} not in vmlinux")
# Recursive function that gets all the dependent classes and adds them to handler
def process_vmlinux_class(node, llvm_module, handler: DependencyHandler, processing_stack=None):
def process_vmlinux_post_ast(node, llvm_module, handler: DependencyHandler, processing_stack=None):
"""
Recursively process vmlinux classes and their dependencies.
@ -142,7 +149,7 @@ def process_vmlinux_class(node, llvm_module, handler: DependencyHandler, process
# Recursively process the dependency
if vmlinux_symbol is not None:
if process_vmlinux_class(vmlinux_symbol, llvm_module, handler, processing_stack):
if process_vmlinux_post_ast(vmlinux_symbol, llvm_module, handler, processing_stack):
new_dep_node.set_field_ready(elem_name, True)
else:
raise ValueError(