From a756f5e4b71e07aadb0f16168c811d8c12edda20 Mon Sep 17 00:00:00 2001 From: Pragyansh Chaturvedi Date: Fri, 10 Oct 2025 23:55:12 +0530 Subject: [PATCH] Add passing helper test for assignment --- tests/passing_tests/assign/helper.py | 34 ++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 tests/passing_tests/assign/helper.py diff --git a/tests/passing_tests/assign/helper.py b/tests/passing_tests/assign/helper.py new file mode 100644 index 0000000..9809a9c --- /dev/null +++ b/tests/passing_tests/assign/helper.py @@ -0,0 +1,34 @@ +from pythonbpf import bpf, map, section, bpfglobal, compile +from ctypes import c_void_p, c_int64, c_uint64 +from pythonbpf.maps import HashMap + +# NOTE: An example of i64** assignment with binops on the RHS + + +@bpf +@map +def last() -> HashMap: + return HashMap(key=c_uint64, value=c_uint64, max_entries=3) + + +@bpf +@section("tracepoint/syscalls/sys_enter_execve") +def hello_world(ctx: c_void_p) -> c_int64: + last.update(0, 1) + x = last.lookup(0) + print(f"{x}") + x = x + 1 + if x == 2: + print("Hello, World!") + else: + print("Goodbye, World!") + return + + +@bpf +@bpfglobal +def LICENSE() -> str: + return "GPL" + + +compile()