From f830fbe8ba4d25111fdd7b5d43b6212636a87f70 Mon Sep 17 00:00:00 2001
From: varun-r-mallya <varunrmallya@gmail.com>
Date: Wed, 10 Sep 2025 11:40:07 +0530
Subject: [PATCH] add bool assignment support

---
 examples/execve3.py         |  6 +++++-
 pythonbpf/functions_pass.py | 18 ++++++++++++++++--
 2 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/examples/execve3.py b/examples/execve3.py
index 40db272..4f3d753 100644
--- a/examples/execve3.py
+++ b/examples/execve3.py
@@ -30,8 +30,12 @@ def hello_again(ctx: c_void_p) -> c_int64:
 #        if delta < 1000000000:
 #            print("execve called within last second")
 #        last().delete(key)
-    if True:
+    x = False
+    x = True
+    if x:
         print("we prevailed")
+    if x:
+        print("we did not prevail")
     ts = ktime()
 #    last().update(key, ts)
     return c_int64(0)
diff --git a/pythonbpf/functions_pass.py b/pythonbpf/functions_pass.py
index 23f84a6..2713028 100644
--- a/pythonbpf/functions_pass.py
+++ b/pythonbpf/functions_pass.py
@@ -37,7 +37,15 @@ def handle_assign(func, module, builder, stmt, map_sym_tab, local_sym_tab):
     var_name = target.id
     rval = stmt.value
     if isinstance(rval, ast.Constant):
-        if isinstance(rval.value, int):
+        if isinstance(rval.value, bool):
+            if rval.value:
+                builder.store(ir.Constant(ir.IntType(1), 1),
+                              local_sym_tab[var_name])
+            else:
+                builder.store(ir.Constant(ir.IntType(1), 0),
+                              local_sym_tab[var_name])
+            print(f"Assigned constant {rval.value} to {var_name}")
+        elif isinstance(rval.value, int):
             # Assume c_int64 for now
             # var = builder.alloca(ir.IntType(64), name=var_name)
             # var.align = 8
@@ -232,7 +240,13 @@ def process_func_body(module, builder, func_node, func, ret_type, map_sym_tab):
                     print("Unsupported assignment call function type")
                     continue
             elif isinstance(rval, ast.Constant):
-                if isinstance(rval.value, int):
+                if isinstance(rval.value, bool):
+                    ir_type = ir.IntType(1)
+                    var = builder.alloca(ir_type, name=var_name)
+                    var.align = 1
+                    print(
+                        f"Pre-allocated variable {var_name} of type c_bool")
+                elif isinstance(rval.value, int):
                     # Assume c_int64 for now
                     ir_type = ir.IntType(64)
                     var = builder.alloca(ir_type, name=var_name)