varun-r-mallya/python-bpf
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/Python-BPF.git synced 2026-02-10 07:00:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow local symbols to be used within return

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-10-03 19:50:56 +05:30
parent 3f061750cf
commit be05b5d102
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
pythonbpf/functions_pass.py
View File

@ -417,6 +417,19 @@ def process_stmt(
) )
builder.ret(val[0]) builder.ret(val[0])
did_return = True did_return = True
elif isinstance(stmt.value.args[0], ast.Name):
if stmt.value.args[0].id in local_sym_tab:
var = local_sym_tab[stmt.value.args[0].id].var
val = builder.load(var)
if val.type != ret_type:
raise ValueError(
"Return type mismatch: expected"
f"{ret_type}, got {val.type}"
)
builder.ret(val)
did_return = True
else:
raise ValueError("Failed to evaluate return expression")
elif isinstance(stmt.value, ast.Name): elif isinstance(stmt.value, ast.Name):
if stmt.value.id == "XDP_PASS": if stmt.value.id == "XDP_PASS":
builder.ret(ir.Constant(ret_type, 2)) builder.ret(ir.Constant(ret_type, 2))