mirror of
https://github.com/varun-r-mallya/Python-BPF.git
synced 2025-12-31 21:06:25 +00:00
remove unary assign
This commit is contained in:
@ -49,8 +49,6 @@ def hello_again(ctx: c_void_p) -> c_int64:
|
|||||||
keema = 8 * 9
|
keema = 8 * 9
|
||||||
keesa = 10 - 11
|
keesa = 10 - 11
|
||||||
keeda = 10 / 5
|
keeda = 10 / 5
|
||||||
# below does not spit IR
|
|
||||||
keeda += 1
|
|
||||||
return c_int64(0)
|
return c_int64(0)
|
||||||
|
|
||||||
@bpf
|
@bpf
|
||||||
|
|||||||
@ -33,26 +33,3 @@ def handle_binary_op(rval, module, builder, func, local_sym_tab, map_sym_tab):
|
|||||||
SyntaxError("Unsupported binary operation")
|
SyntaxError("Unsupported binary operation")
|
||||||
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
def handle_unary_op(rval, module, builder, func, local_sym_tab, map_sym_tab):
|
|
||||||
print("UNARY ASSIGNMENT DOES NOT WORK")
|
|
||||||
return
|
|
||||||
# TODO: heavy fixxing needed
|
|
||||||
operand = rval.gay
|
|
||||||
op = rval.op
|
|
||||||
if isinstance(operand, ast.Name):
|
|
||||||
operand = local_sym_tab[operand.id]
|
|
||||||
elif isinstance(operand, ast.Constant):
|
|
||||||
operand = ir.Constant(ir.IntType(64), operand.value)
|
|
||||||
else:
|
|
||||||
SyntaxError("Unsupported operand type")
|
|
||||||
|
|
||||||
if isinstance(op, ast.UAdd):
|
|
||||||
result = builder.add(ir.Constant(ir.IntType(64), 0), operand)
|
|
||||||
elif isinstance(op, ast.USub):
|
|
||||||
result = builder.sub(ir.Constant(ir.IntType(64), 0), operand)
|
|
||||||
else:
|
|
||||||
result = "all my homies hate type errors"
|
|
||||||
SyntaxError("Unsupported unary operation")
|
|
||||||
|
|
||||||
return result
|
|
||||||
@ -3,7 +3,7 @@ import ast
|
|||||||
|
|
||||||
from .bpf_helper_handler import helper_func_list, handle_helper_call
|
from .bpf_helper_handler import helper_func_list, handle_helper_call
|
||||||
from .type_deducer import ctypes_to_ir
|
from .type_deducer import ctypes_to_ir
|
||||||
from .unary_and_binary_ops import handle_binary_op, handle_unary_op
|
from .binary_ops import handle_binary_op
|
||||||
|
|
||||||
|
|
||||||
def get_probe_string(func_node):
|
def get_probe_string(func_node):
|
||||||
@ -22,17 +22,6 @@ def get_probe_string(func_node):
|
|||||||
return arg.value
|
return arg.value
|
||||||
return "helper"
|
return "helper"
|
||||||
|
|
||||||
def handle_unary_assign(func, module, builder, stmt, map_sym_tab, local_sym_tab):
|
|
||||||
"""Handle unary assignment statements in the function body."""
|
|
||||||
SyntaxError("Unary assignment not supported")
|
|
||||||
target = stmt.target
|
|
||||||
if not isinstance(target, ast.Name):
|
|
||||||
SyntaxError("Unsupported assignment target")
|
|
||||||
return
|
|
||||||
else:
|
|
||||||
handle_unary_op(func, module, builder, stmt, map_sym_tab, local_sym_tab)
|
|
||||||
return
|
|
||||||
|
|
||||||
def handle_assign(func, module, builder, stmt, map_sym_tab, local_sym_tab):
|
def handle_assign(func, module, builder, stmt, map_sym_tab, local_sym_tab):
|
||||||
"""Handle assignment statements in the function body."""
|
"""Handle assignment statements in the function body."""
|
||||||
if len(stmt.targets) != 1:
|
if len(stmt.targets) != 1:
|
||||||
@ -256,7 +245,7 @@ def process_stmt(func, module, builder, stmt, local_sym_tab, map_sym_tab, did_re
|
|||||||
elif isinstance(stmt, ast.Assign):
|
elif isinstance(stmt, ast.Assign):
|
||||||
handle_assign(func, module, builder, stmt, map_sym_tab, local_sym_tab)
|
handle_assign(func, module, builder, stmt, map_sym_tab, local_sym_tab)
|
||||||
elif isinstance(stmt, ast.AugAssign):
|
elif isinstance(stmt, ast.AugAssign):
|
||||||
handle_unary_assign(func, module, builder, stmt, map_sym_tab, local_sym_tab)
|
raise SyntaxError("Augmented assignment not supported")
|
||||||
elif isinstance(stmt, ast.If):
|
elif isinstance(stmt, ast.If):
|
||||||
handle_if(func, module, builder, stmt, map_sym_tab, local_sym_tab)
|
handle_if(func, module, builder, stmt, map_sym_tab, local_sym_tab)
|
||||||
elif isinstance(stmt, ast.Return):
|
elif isinstance(stmt, ast.Return):
|
||||||
|
|||||||
Reference in New Issue
Block a user