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

Call eval_expr properly within get_operand_value

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-10-11 03:21:09 +05:30
parent 91a3fe140d
commit c9bbe1ffd8
2 changed files with 42 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
pythonbpf/binary_ops.py
View File

@ -8,7 +8,9 @@ from pythonbpf.expr import get_base_type_and_depth, deref_to_depth, eval_expr
logger: Logger = logging.getLogger(__name__) logger: Logger = logging.getLogger(__name__)
def get_operand_value(func, operand, builder, local_sym_tab): def get_operand_value(
func, module, operand, builder, local_sym_tab, map_sym_tab, structs_sym_tab=None
):
"""Extract the value from an operand, handling variables and constants.""" """Extract the value from an operand, handling variables and constants."""
logger.info(f"Getting operand value for: {ast.dump(operand)}") logger.info(f"Getting operand value for: {ast.dump(operand)}")
if isinstance(operand, ast.Name): if isinstance(operand, ast.Name):
@ -26,10 +28,14 @@ def get_operand_value(func, operand, builder, local_sym_tab):
return cst return cst
raise TypeError(f"Unsupported constant type: {type(operand.value)}") raise TypeError(f"Unsupported constant type: {type(operand.value)}")
elif isinstance(operand, ast.BinOp): elif isinstance(operand, ast.BinOp):
res = handle_binary_op_impl(func, operand, builder, local_sym_tab) res = handle_binary_op_impl(
func, module, operand, builder, local_sym_tab, map_sym_tab, structs_sym_tab
)
return res return res
elif isinstance(operand, ast.Call): else:
res = eval_expr(func, None, builder, operand, local_sym_tab, {}, {}) res = eval_expr(
func, module, builder, operand, local_sym_tab, map_sym_tab, structs_sym_tab
)
if res is None: if res is None:
raise ValueError(f"Failed to evaluate call expression: {operand}") raise ValueError(f"Failed to evaluate call expression: {operand}")
val, _ = res val, _ = res
@ -37,10 +43,16 @@ def get_operand_value(func, operand, builder, local_sym_tab):
raise TypeError(f"Unsupported operand type: {type(operand)}") raise TypeError(f"Unsupported operand type: {type(operand)}")
def handle_binary_op_impl(func, rval, builder, local_sym_tab): def handle_binary_op_impl(
func, module, rval, builder, local_sym_tab, map_sym_tab, structs_sym_tab=None
):
op = rval.op op = rval.op
left = get_operand_value(func, rval.left, builder, local_sym_tab) left = get_operand_value(
right = get_operand_value(func, rval.right, builder, local_sym_tab) func, module, rval.left, builder, local_sym_tab, map_sym_tab, structs_sym_tab
)
right = get_operand_value(
func, module, rval.right, builder, local_sym_tab, map_sym_tab, structs_sym_tab
)
logger.info(f"left is {left}, right is {right}, op is {op}") logger.info(f"left is {left}, right is {right}, op is {op}")
# NOTE: Before doing the operation, if the operands are integers # NOTE: Before doing the operation, if the operands are integers
@ -73,8 +85,19 @@ def handle_binary_op_impl(func, rval, builder, local_sym_tab):
raise SyntaxError("Unsupported binary operation") raise SyntaxError("Unsupported binary operation")
def handle_binary_op(func, rval, builder, var_name, local_sym_tab): def handle_binary_op(
result = handle_binary_op_impl(func, rval, builder, local_sym_tab) func,
module,
rval,
builder,
var_name,
local_sym_tab,
map_sym_tab,
structs_sym_tab=None,
):
result = handle_binary_op_impl(
func, module, rval, builder, local_sym_tab, map_sym_tab, structs_sym_tab
)
if var_name and var_name in local_sym_tab: if var_name and var_name in local_sym_tab:
logger.info( logger.info(
f"Storing result {result} into variable {local_sym_tab[var_name].var}" f"Storing result {result} into variable {local_sym_tab[var_name].var}"

11
pythonbpf/expr/expr_pass.py
View File

@ -402,7 +402,16 @@ def eval_expr(
elif isinstance(expr, ast.BinOp): elif isinstance(expr, ast.BinOp):
from pythonbpf.binary_ops import handle_binary_op from pythonbpf.binary_ops import handle_binary_op
return handle_binary_op(func, expr, builder, None, local_sym_tab) return handle_binary_op(
func,
module,
expr,
builder,
None,
local_sym_tab,
map_sym_tab,
structs_sym_tab,
)
elif isinstance(expr, ast.Compare): elif isinstance(expr, ast.Compare):
return _handle_compare( return _handle_compare(
func, module, builder, expr, local_sym_tab, map_sym_tab, structs_sym_tab func, module, builder, expr, local_sym_tab, map_sym_tab, structs_sym_tab