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 support for tuple style multiiassignment in allocation_pass

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-10-14 23:06:43 +05:30
parent 4a79f9b9b2
commit dd3fc74d09
1 changed files with 46 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

79
pythonbpf/allocation_pass.py
View File

@ -25,41 +25,54 @@ class LocalSymbol:
def handle_assign_allocation(builder, stmt, local_sym_tab, structs_sym_tab): def handle_assign_allocation(builder, stmt, local_sym_tab, structs_sym_tab):
"""Handle memory allocation for assignment statements.""" """Handle memory allocation for assignment statements."""
# Validate assignment logger.info(f"Handling assignment for allocation: {ast.dump(stmt)}")
if len(stmt.targets) != 1:
logger.warning("Multi-target assignment not supported, skipping allocation")
return
target = stmt.targets[0] # NOTE: Support multi-target assignments (e.g.: a, b = 1, 2)
if isinstance(stmt.targets[0], ast.Tuple):
# Skip non-name targets (e.g., struct field assignments) if not isinstance(stmt.value, ast.Tuple):
if isinstance(target, ast.Attribute): logger.warning("Mismatched multi-target assignment, skipping allocation")
logger.debug(f"Struct field assignment to {target.attr}, no allocation needed") return
return targets = stmt.targets[0].elts
rvals = stmt.value.elts
if not isinstance(target, ast.Name): if len(targets) != len(rvals):
logger.warning(f"Unsupported assignment target type: {type(target).__name__}") logger.warning("Mismatched multi-target assignment, skipping allocation")
return return
var_name = target.id
rval = stmt.value
# Skip if already allocated
if var_name in local_sym_tab:
logger.debug(f"Variable {var_name} already allocated, skipping")
return
# Determine type and allocate based on rval
if isinstance(rval, ast.Call):
_allocate_for_call(builder, var_name, rval, local_sym_tab, structs_sym_tab)
elif isinstance(rval, ast.Constant):
_allocate_for_constant(builder, var_name, rval, local_sym_tab)
elif isinstance(rval, ast.BinOp):
_allocate_for_binop(builder, var_name, local_sym_tab)
else: else:
logger.warning( targets = stmt.targets
f"Unsupported assignment value type for {var_name}: {type(rval).__name__}" rvals = [stmt.value]
)
for target, rval in zip(targets, rvals):
# Skip non-name targets (e.g., struct field assignments)
if isinstance(target, ast.Attribute):
logger.debug(
f"Struct field assignment to {target.attr}, no allocation needed"
)
continue
if not isinstance(target, ast.Name):
logger.warning(
f"Unsupported assignment target type: {type(target).__name__}"
)
continue
var_name = target.id
# Skip if already allocated
if var_name in local_sym_tab:
logger.debug(f"Variable {var_name} already allocated, skipping")
continue
# Determine type and allocate based on rval
if isinstance(rval, ast.Call):
_allocate_for_call(builder, var_name, rval, local_sym_tab, structs_sym_tab)
elif isinstance(rval, ast.Constant):
_allocate_for_constant(builder, var_name, rval, local_sym_tab)
elif isinstance(rval, ast.BinOp):
_allocate_for_binop(builder, var_name, local_sym_tab)
else:
logger.warning(
f"Unsupported assignment value type for {var_name}: {type(rval).__name__}"
)
def _allocate_for_call(builder, var_name, rval, local_sym_tab, structs_sym_tab): def _allocate_for_call(builder, var_name, rval, local_sym_tab, structs_sym_tab):