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()