varun-r-mallya/python-bpf
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/Python-BPF.git synced 2025-12-31 21:06:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add bpf_ktime_get_ns helper function without IR gen

Browse Source
This commit is contained in:
varun-r-mallya
2025-09-07 20:58:54 +05:30
parent 3b5425ca1b
commit d87decd25e
4 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/execve2.py
View File

@ -1,6 +1,6 @@
from pythonbpf.decorators import bpf, section from pythonbpf.decorators import bpf, section
from ctypes import c_void_p, c_int64, c_int32 from ctypes import c_void_p, c_int64, c_int32
from pythonbpf.helpers import bpf_ktime_get_ns
@bpf @bpf
@section("tracepoint/syscalls/sys_enter_execve") @section("tracepoint/syscalls/sys_enter_execve")
@ -11,8 +11,9 @@ def hello(ctx: c_void_p) -> c_int32:
@bpf @bpf
@section("tracepoint/syscalls/sys_exit_execve") @section("tracepoint/syscalls/sys_exit_execve")
def hello_again(ctx: c_void_p) -> c_int32: def hello_again(ctx: c_void_p) -> c_int64:
print("exited") print("exited")
return c_int32(0) ts = bpf_ktime_get_ns()
return c_int64(0)
LICENSE = "GPL" LICENSE = "GPL"

3
pythonbpf/bpf_helper_handler.py
View File

@ -1,6 +1,9 @@
import ast import ast
from llvmlite import ir from llvmlite import ir
def bpf_ktime_get_ns_emitter(call, module, builder, func):
pass
def bpf_printk_emitter(call, module, builder, func): def bpf_printk_emitter(call, module, builder, func):
if not hasattr(func, "_fmt_counter"): if not hasattr(func, "_fmt_counter"):
func._fmt_counter = 0 func._fmt_counter = 0

4
pythonbpf/functions_pass.py
View File

@ -1,7 +1,7 @@
from llvmlite import ir from llvmlite import ir
import ast import ast
from .bpf_helper_handler import bpf_printk_emitter from .bpf_helper_handler import bpf_printk_emitter, bpf_ktime_get_ns_emitter
from .type_deducer import ctypes_to_ir from .type_deducer import ctypes_to_ir
def get_probe_string(func_node): def get_probe_string(func_node):
@ -29,6 +29,8 @@ def process_func_body(module, builder, func_node, func, ret_type):
call = stmt.value call = stmt.value
if isinstance(call.func, ast.Name) and call.func.id == "print": if isinstance(call.func, ast.Name) and call.func.id == "print":
bpf_printk_emitter(call, module, builder, func) bpf_printk_emitter(call, module, builder, func)
if isinstance(call.func, ast.Name) and call.func.id == "bpf_ktime_get_ns":
bpf_ktime_get_ns_emitter(call, module, builder, func)
elif isinstance(stmt, ast.Return): elif isinstance(stmt, ast.Return):
if stmt.value is None: if stmt.value is None:
builder.ret(ir.Constant(ir.IntType(32), 0)) builder.ret(ir.Constant(ir.IntType(32), 0))

2
pythonbpf/helpers.py Normal file
View File

@ -0,0 +1,2 @@
def bpf_ktime_get_ns():
return bpf_ktime_get_ns()