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

Remove unnecessary return artifacts from get_operand_value

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-10-11 03:06:24 +05:30
parent c2c17741e5
commit 91a3fe140d
1 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pythonbpf/binary_ops.py
View File

@ -18,33 +18,31 @@ def get_operand_value(func, operand, builder, local_sym_tab):
base_type, depth = get_base_type_and_depth(var_type) base_type, depth = get_base_type_and_depth(var_type)
logger.info(f"var is {var}, base_type is {base_type}, depth is {depth}") logger.info(f"var is {var}, base_type is {base_type}, depth is {depth}")
val = deref_to_depth(func, builder, var, depth) val = deref_to_depth(func, builder, var, depth)
return val, [val], var return val
raise ValueError(f"Undefined variable: {operand.id}") raise ValueError(f"Undefined variable: {operand.id}")
elif isinstance(operand, ast.Constant): elif isinstance(operand, ast.Constant):
if isinstance(operand.value, int): if isinstance(operand.value, int):
cst = ir.Constant(ir.IntType(64), int(operand.value)) cst = ir.Constant(ir.IntType(64), int(operand.value))
return cst, [cst], None 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, operand, builder, local_sym_tab)
return res, [res], None return res
elif isinstance(operand, ast.Call): elif isinstance(operand, ast.Call):
res = eval_expr(func, None, builder, operand, local_sym_tab, {}, {}) res = eval_expr(func, None, builder, operand, local_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, val_type = res val, _ = res
return val, [val], None return val
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, rval, builder, local_sym_tab):
op = rval.op op = rval.op
left, lchain, _ = get_operand_value(func, rval.left, builder, local_sym_tab) left = get_operand_value(func, rval.left, builder, local_sym_tab)
right, rchain, _ = get_operand_value(func, rval.right, builder, local_sym_tab) right = get_operand_value(func, rval.right, builder, local_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}")
logger.info(f"left chain: {lchain}, right chain: {rchain}")
# NOTE: Before doing the operation, if the operands are integers # NOTE: Before doing the operation, if the operands are integers
# we always extend them to i64. The assignment to LHS will take # we always extend them to i64. The assignment to LHS will take
# care of truncation if needed. # care of truncation if needed.