Add bpf_ktime_get_ns helper function without IR gen

This commit is contained in:
2025-09-07 20:58:54 +05:30
parent 3b5425ca1b
commit d87decd25e
4 changed files with 12 additions and 4 deletions

View File

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

View File

@ -1,7 +1,7 @@
from llvmlite import ir
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
def get_probe_string(func_node):
@ -29,6 +29,8 @@ def process_func_body(module, builder, func_node, func, ret_type):
call = stmt.value
if isinstance(call.func, ast.Name) and call.func.id == "print":
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):
if stmt.value is None:
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()