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

format chore

Browse Source
This commit is contained in:
varun-r-mallya
2025-10-16 04:10:24 +05:30
parent de02731ea1
commit 0f5c1fa752
4 changed files with 70 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
pythonbpf/vmlinux_parser/ir_gen/ir_generation.py
View File

@ -33,7 +33,9 @@ class IRGenerator:
# Detect circular dependency
if struct.name in processing_stack:
logger.info(f"Circular dependency detected for {struct.name}, skipping recursive processing")
logger.info(
f"Circular dependency detected for {struct.name}, skipping recursive processing"
)
# For circular dependencies, we can either:
# 1. Use forward declarations (opaque pointers)
# 2. Mark as incomplete and process later
@ -54,9 +56,13 @@ class IRGenerator:
if dependency in self.handler.nodes:
dep_node_from_dependency = self.handler[dependency]
# Pass the processing_stack down to track circular refs
self.struct_processor(dep_node_from_dependency, processing_stack)
self.struct_processor(
dep_node_from_dependency, processing_stack
)
else:
raise RuntimeError(f"Warning: Dependency {dependency} not found in handler")
raise RuntimeError(
f"Warning: Dependency {dependency} not found in handler"
)
# Actual processor logic here after dependencies are resolved
self.gen_ir(struct)
@ -74,13 +80,17 @@ class IRGenerator:
field_index = 0
for field_name, field in struct.fields.items():
# does not take arrays and similar types into consideration yet.
if field.ctype_complex_type is not None and issubclass(field.ctype_complex_type, ctypes.Array):
if field.ctype_complex_type is not None and issubclass(
field.ctype_complex_type, ctypes.Array
):
array_size = field.type_size
containing_type = field.containing_type
if containing_type.__module__ == ctypes.__name__:
containing_type_size = ctypes.sizeof(containing_type)
for i in range(0,array_size):
field_co_re_name = self._struct_name_generator(struct, field, field_index, True, i, containing_type_size)
for i in range(0, array_size):
field_co_re_name = self._struct_name_generator(
struct, field, field_index, True, i, containing_type_size
)
globvar = ir.GlobalVariable(
self.llvm_module, ir.IntType(64), name=field_co_re_name
)
@ -91,9 +101,13 @@ class IRGenerator:
array_size = field.type_size
containing_type = field.containing_type
if containing_type.__module__ == "vmlinux":
containing_type_size = self.handler[containing_type.__name__].current_offset
for i in range(0,array_size):
field_co_re_name = self._struct_name_generator(struct, field, field_index, True, i, containing_type_size)
containing_type_size = self.handler[
containing_type.__name__
].current_offset
for i in range(0, array_size):
field_co_re_name = self._struct_name_generator(
struct, field, field_index, True, i, containing_type_size
)
globvar = ir.GlobalVariable(
self.llvm_module, ir.IntType(64), name=field_co_re_name
)
@ -101,7 +115,9 @@ class IRGenerator:
globvar.set_metadata("llvm.preserve.access.index", debug_info)
field_index += 1
else:
field_co_re_name = self._struct_name_generator(struct, field, field_index)
field_co_re_name = self._struct_name_generator(
struct, field, field_index
)
field_index += 1
globvar = ir.GlobalVariable(
self.llvm_module, ir.IntType(64), name=field_co_re_name
@ -110,15 +126,21 @@ class IRGenerator:
globvar.set_metadata("llvm.preserve.access.index", debug_info)
def _struct_name_generator(
self, struct: DependencyNode, field, field_index: int, is_indexed: bool=False, index: Optional[int]=None, containing_type_size: Optional[int]=None
self,
struct: DependencyNode,
field,
field_index: int,
is_indexed: bool = False,
index: int = 0,
containing_type_size: Optional[int] = None,
) -> str:
if is_indexed:
name = (
"llvm."
+ struct.name.removeprefix("struct_")
+ f":0:{field.offset + index*containing_type_size}"
+ "$"
+ f"0:{field_index}:{index}"
"llvm."
+ struct.name.removeprefix("struct_")
+ f":0:{field.offset + index * containing_type_size}"
+ "$"
+ f"0:{field_index}:{index}"
)
return name
elif struct.name.startswith("struct_"):