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

setup skeleton for offset calculation

Browse Source
This commit is contained in:
varun-r-mallya
2025-10-15 04:42:38 +05:30
parent 11e8e72188
commit 69b73003ca
2 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
pythonbpf/vmlinux_parser/class_handler.py
View File

@ -71,6 +71,7 @@ def process_vmlinux_post_ast(
if len(field_elem) == 2: if len(field_elem) == 2:
field_name, field_type = field_elem field_name, field_type = field_elem
elif len(field_elem) == 3: elif len(field_elem) == 3:
raise NotImplementedError("Bitfields are not supported in the current version")
field_name, field_type, bitfield_size = field_elem field_name, field_type, bitfield_size = field_elem
field_table[field_name] = [field_type, bitfield_size] field_table[field_name] = [field_type, bitfield_size]
elif hasattr(class_obj, "__annotations__"): elif hasattr(class_obj, "__annotations__"):

13
pythonbpf/vmlinux_parser/dependency_node.py
View File

@ -13,6 +13,7 @@ class Field:
containing_type: Optional[Any] containing_type: Optional[Any]
type_size: Optional[int] type_size: Optional[int]
bitfield_size: Optional[int] bitfield_size: Optional[int]
offset: int
value: Any = None value: Any = None
ready: bool = False ready: bool = False
@ -60,6 +61,10 @@ class Field:
if mark_ready: if mark_ready:
self.ready = True self.ready = True
def set_offset(self, offset: int) -> None:
"""Set the offset of this field"""
self.offset = offset
@dataclass @dataclass
class DependencyNode: class DependencyNode:
@ -109,6 +114,7 @@ class DependencyNode:
depends_on: Optional[list[str]] = None depends_on: Optional[list[str]] = None
fields: Dict[str, Field] = field(default_factory=dict) fields: Dict[str, Field] = field(default_factory=dict)
_ready_cache: Optional[bool] = field(default=None, repr=False) _ready_cache: Optional[bool] = field(default=None, repr=False)
current_offset: int = 0
def add_field( def add_field(
self, self,
@ -120,6 +126,7 @@ class DependencyNode:
ctype_complex_type: Optional[int] = None, ctype_complex_type: Optional[int] = None,
bitfield_size: Optional[int] = None, bitfield_size: Optional[int] = None,
ready: bool = False, ready: bool = False,
offset: int = 0,
) -> None: ) -> None:
"""Add a field to the node with an optional initial value and readiness state.""" """Add a field to the node with an optional initial value and readiness state."""
if self.depends_on is None: if self.depends_on is None:
@ -133,6 +140,7 @@ class DependencyNode:
type_size=type_size, type_size=type_size,
ctype_complex_type=ctype_complex_type, ctype_complex_type=ctype_complex_type,
bitfield_size=bitfield_size, bitfield_size=bitfield_size,
offset=offset
) )
# Invalidate readiness cache # Invalidate readiness cache
self._ready_cache = None self._ready_cache = None
@ -209,9 +217,14 @@ class DependencyNode:
raise KeyError(f"Field '{name}' does not exist in node '{self.name}'") raise KeyError(f"Field '{name}' does not exist in node '{self.name}'")
self.fields[name].set_ready(is_ready) self.fields[name].set_ready(is_ready)
self.fields[name].set_offset(self.current_offset)
self.current_offset += self._calculate_size(name)
# Invalidate readiness cache # Invalidate readiness cache
self._ready_cache = None self._ready_cache = None
def _calculate_size(self, name: str) -> int:
pass
@property @property
def is_ready(self) -> bool: def is_ready(self) -> bool:
"""Check if the node is ready (all fields are ready).""" """Check if the node is ready (all fields are ready)."""