From 5cbd9a531e0190530a89204c6e35d44d648a9c0f Mon Sep 17 00:00:00 2001
From: Pragyansh Chaturvedi <pragyanshchaturvedi18@gmail.com>
Date: Mon, 27 Oct 2025 01:08:56 +0530
Subject: [PATCH] Add bpf_get_prandom_u32 helper

---
 pythonbpf/helper/__init__.py           |  3 ++-
 pythonbpf/helper/bpf_helper_handler.py | 23 +++++++++++++++++++++++
 pythonbpf/helper/helpers.py            |  5 +++++
 3 files changed, 30 insertions(+), 1 deletion(-)

diff --git a/pythonbpf/helper/__init__.py b/pythonbpf/helper/__init__.py
index 2f9c347..5c38137 100644
--- a/pythonbpf/helper/__init__.py
+++ b/pythonbpf/helper/__init__.py
@@ -1,7 +1,7 @@
 from .helper_registry import HelperHandlerRegistry
 from .helper_utils import reset_scratch_pool
 from .bpf_helper_handler import handle_helper_call, emit_probe_read_kernel_str_call
-from .helpers import ktime, pid, deref, comm, probe_read_str, XDP_DROP, XDP_PASS
+from .helpers import ktime, pid, deref, comm, probe_read_str, random, XDP_DROP, XDP_PASS
 
 
 # Register the helper handler with expr module
@@ -65,6 +65,7 @@ __all__ = [
     "deref",
     "comm",
     "probe_read_str",
+    "random",
     "XDP_DROP",
     "XDP_PASS",
 ]
diff --git a/pythonbpf/helper/bpf_helper_handler.py b/pythonbpf/helper/bpf_helper_handler.py
index 7868677..fa0e1e1 100644
--- a/pythonbpf/helper/bpf_helper_handler.py
+++ b/pythonbpf/helper/bpf_helper_handler.py
@@ -25,6 +25,7 @@ class BPFHelperID(Enum):
     BPF_MAP_DELETE_ELEM = 3
     BPF_KTIME_GET_NS = 5
     BPF_PRINTK = 6
+    BPF_GET_PRANDOM_U32 = 7
     BPF_GET_CURRENT_PID_TGID = 14
     BPF_GET_CURRENT_COMM = 16
     BPF_PERF_EVENT_OUTPUT = 25
@@ -433,6 +434,28 @@ def bpf_probe_read_kernel_str_emitter(
     return result, ir.IntType(64)
 
 
+@HelperHandlerRegistry.register("random")
+def bpf_get_prandom_u32_emitter(
+    call,
+    map_ptr,
+    module,
+    builder,
+    func,
+    local_sym_tab=None,
+    struct_sym_tab=None,
+    map_sym_tab=None,
+):
+    """
+    Emit LLVM IR for bpf_get_prandom_u32 helper function call.
+    """
+    helper_id = ir.Constant(ir.IntType(64), BPFHelperID.BPF_GET_PRANDOM_U32.value)
+    fn_type = ir.FunctionType(ir.IntType(32), [], var_arg=False)
+    fn_ptr_type = ir.PointerType(fn_type)
+    fn_ptr = builder.inttoptr(helper_id, fn_ptr_type)
+    result = builder.call(fn_ptr, [], tail=False)
+    return result, ir.IntType(32)
+
+
 def handle_helper_call(
     call,
     module,
diff --git a/pythonbpf/helper/helpers.py b/pythonbpf/helper/helpers.py
index cb1a8e1..f8f2bd6 100644
--- a/pythonbpf/helper/helpers.py
+++ b/pythonbpf/helper/helpers.py
@@ -27,6 +27,11 @@ def probe_read_str(dst, src):
     return ctypes.c_int64(0)
 
 
+def random():
+    """get a pseudorandom u32 number"""
+    return ctypes.c_int32(0)
+
+
 XDP_ABORTED = ctypes.c_int64(0)
 XDP_DROP = ctypes.c_int64(1)
 XDP_PASS = ctypes.c_int64(2)