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 ctype subclass identifier

Browse Source 
Signed-off-by: varun-r-mallya <varunrmallya@gmail.com>
This commit is contained in:
varun-r-mallya
2025-10-11 20:02:20 +05:30
parent 06773c895f
commit 7c559840f0
2 changed files with 25 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
pythonbpf/vmlinux_parser/vmlinux_class_handler.py
View File

@ -30,6 +30,7 @@ def process_vmlinux_class(node, llvm_module, handler: DependencyHandler):
if current_symbol_name not in symbols_in_module: if current_symbol_name not in symbols_in_module:
raise ImportError(f"{current_symbol_name} not present in module vmlinux") raise ImportError(f"{current_symbol_name} not present in module vmlinux")
logger.info(f"Resolving vmlinux class {current_symbol_name}") logger.info(f"Resolving vmlinux class {current_symbol_name}")
logger.debug(f"Current handler state: {handler.is_ready} readiness and {handler.get_all_nodes()} all nodes")
field_table = {} # should contain the field and it's type. field_table = {} # should contain the field and it's type.
# Get the class object from the module # Get the class object from the module
@ -65,13 +66,33 @@ def process_vmlinux_class(node, llvm_module, handler: DependencyHandler):
new_dep_node.add_field(elem_name, elem_type, ready=True) new_dep_node.add_field(elem_name, elem_type, ready=True)
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 print("elem_name:", elem_name, "elem_type:", elem_type)
temp_node = type('TempNode', (), # currently fails when a non-normal type appears which is basically everytime
{'name': elem_type.__name__ if hasattr(elem_type, '__name__') else str(elem_type)})() identify_ctypes_type(elem_type)
if process_vmlinux_class(temp_node, llvm_module, handler): symbol_name = elem_type.__name__ if hasattr(elem_type, '__name__') else str(elem_type)
vmlinux_symbol = getattr(imported_module, symbol_name)
if process_vmlinux_class(vmlinux_symbol, llvm_module, handler):
new_dep_node.set_field_ready(elem_name, True) new_dep_node.set_field_ready(elem_name, True)
else: else:
raise ValueError(f"{elem_name} with type {elem_type} not supported in recursive resolver") 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)
logger.info(f"added node: {current_symbol_name}")
return True return True
def identify_ctypes_type(t):
if isinstance(t, type): # t is a type/class
if issubclass(t, ctypes.Array):
print("Array type")
print("Element type:", t._type_)
print("Length:", t._length_)
elif issubclass(t, ctypes._Pointer):
print("Pointer type")
print("Points to:", t._type_)
elif issubclass(t, ctypes._SimpleCData):
print("Scalar type")
print("Base type:", t)
else:
print("Other ctypes type")
else:
print("Instance, not type")

1
tests/failing_tests/xdp_pass.py
View File

@ -2,7 +2,6 @@ from pythonbpf import bpf, map, section, bpfglobal, compile, compile_to_ir
from pythonbpf.maps import HashMap from pythonbpf.maps import HashMap
from pythonbpf.helper import XDP_PASS from pythonbpf.helper import XDP_PASS
from vmlinux import struct_xdp_md from vmlinux import struct_xdp_md
# from vmlinux import struct_ring_buffer_per_cpu
from ctypes import c_int64 from ctypes import c_int64
# Instructions to how to run this program # Instructions to how to run this program