remove unary assign

Author: varun-r-mallya

examples/execve3.py

@@ -49,8 +49,6 @@ def hello_again(ctx: c_void_p) -> c_int64:
     keema = 8 * 9
     keesa = 10 - 11
     keeda = 10 / 5
-    # below does not spit IR
-    keeda += 1
     return c_int64(0)
 
 @bpf

pythonbpf/unary_and_binary_ops.py pythonbpf/binary_ops.pypythonbpf/unary_and_binary_ops.py renamed to pythonbpf/binary_ops.py

@@ -33,26 +33,3 @@ def handle_binary_op(rval, module, builder, func, local_sym_tab, map_sym_tab):
         SyntaxError("Unsupported binary operation")
 
     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

pythonbpf/functions_pass.py

@@ -3,7 +3,7 @@
 
 from .bpf_helper_handler import helper_func_list, handle_helper_call
 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):
@@ -22,17 +22,6 @@ def get_probe_string(func_node):
                     return arg.value
     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):
     """Handle assignment statements in the function body."""
     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):
         handle_assign(func, module, builder, stmt, map_sym_tab, local_sym_tab)
     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):
         handle_if(func, module, builder, stmt, map_sym_tab, local_sym_tab)
     elif isinstance(stmt, ast.Return):