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

Fix imports and type issues for bpf_probe_read

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-11-01 13:50:23 +05:30
parent 2257c175ed
commit ec2ea835e5
3 changed files with 30 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
pythonbpf/helper/bpf_helper_handler.py
View File

@ -476,23 +476,20 @@ def bpf_probe_read_emitter(
if len(call.args) != 3:
logger.warn("Expected 3 args for probe_read helper")
return
dst_ptr, _ = get_ptr_from_arg(
call.args[0], func, module, builder, local_sym_tab, map_sym_tab, struct_sym_tab
dst_ptr = get_or_create_ptr_from_arg(
func, module, call.args[0], builder, local_sym_tab, map_sym_tab, struct_sym_tab
)
size_val = (
get_int_value_from_arg(
call.args[1],
func,
module,
builder,
local_sym_tab,
map_sym_tab,
struct_sym_tab,
)
& 0xFFFFFFFF
size_val = get_int_value_from_arg(
call.args[1],
func,
module,
builder,
local_sym_tab,
map_sym_tab,
struct_sym_tab,
)
src_ptr, _ = get_ptr_from_arg(
call.args[2], func, module, builder, local_sym_tab, map_sym_tab, struct_sym_tab
src_ptr = get_or_create_ptr_from_arg(
func, module, call.args[2], builder, local_sym_tab, map_sym_tab, struct_sym_tab
)
fn_type = ir.FunctionType(
ir.IntType(64),
@ -507,7 +504,7 @@ def bpf_probe_read_emitter(
fn_ptr,
[
builder.bitcast(dst_ptr, ir.PointerType()),
ir.Constant(ir.IntType(32), size_val),
builder.trunc(size_val, ir.IntType(32)),
builder.bitcast(src_ptr, ir.PointerType()),
],
tail=False,