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

Pass map_sym_tab to handle_struct_field_assign

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-10-12 10:37:20 +05:30
parent 64e44d0d58
commit 08c0ccf0ac
2 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pythonbpf/assign_pass.py
View File

@ -7,7 +7,7 @@ logger = logging.getLogger(__name__)
def handle_struct_field_assignment( def handle_struct_field_assignment(
func, module, builder, target, rval, local_sym_tab, structs_sym_tab func, module, builder, target, rval, local_sym_tab, map_sym_tab, structs_sym_tab
): ):
"""Handle struct field assignment (obj.field = value).""" """Handle struct field assignment (obj.field = value)."""
@ -27,7 +27,9 @@ def handle_struct_field_assignment(
# Get field pointer and evaluate value # Get field pointer and evaluate value
field_ptr = struct_info.gep(builder, local_sym_tab[var_name].var, field_name) field_ptr = struct_info.gep(builder, local_sym_tab[var_name].var, field_name)
val = eval_expr(func, module, builder, rval, local_sym_tab, None, structs_sym_tab) val = eval_expr(
func, module, builder, rval, local_sym_tab, map_sym_tab, structs_sym_tab
)
if val is None: if val is None:
logger.error(f"Failed to evaluate value for {var_name}.{field_name}") logger.error(f"Failed to evaluate value for {var_name}.{field_name}")

9
pythonbpf/functions/functions_pass.py
View File

@ -84,7 +84,14 @@ def handle_assign(
if isinstance(target, ast.Attribute): if isinstance(target, ast.Attribute):
# NOTE: Struct field assignment case: pkt.field = value # NOTE: Struct field assignment case: pkt.field = value
handle_struct_field_assignment( handle_struct_field_assignment(
func, module, builder, target, rval, local_sym_tab, structs_sym_tab func,
module,
builder,
target,
rval,
local_sym_tab,
map_sym_tab,
structs_sym_tab,
) )
return return