from pythonbpf import bpf, map, section, bpfglobal, compile
from pythonbpf.helpers import bpf_ktime_get_ns
from pythonbpf.maps import HashMap

from ctypes import c_void_p, c_int64, c_int32, c_uint64

@bpf
@map
def last() -> HashMap:
    return HashMap(key_type=c_uint64, value_type=c_uint64, max_entries=1)


@bpf
@section("tracepoint/syscalls/sys_enter_execve")
def hello(ctx: c_void_p) -> c_int32:
    print("entered")
    print("multi constant support")
    return c_int32(0)


@bpf
@section("tracepoint/syscalls/sys_exit_execve")
def hello_again(ctx: c_void_p) -> c_int64:
    print("exited")
    key = 0
    tsp = last().lookup(key)
    if tsp:
        delta = (bpf_ktime_get_ns() - tsp.value)
        if delta < 1000000000:
            print("execve called within last second")
        last().delete(key)
    ts = bpf_ktime_get_ns()
    last().update(key, ts)
    return c_int64(0)


@bpf
@bpfglobal
def LICENSE() -> str:
    return "GPL"

compile()