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 the number of args for skb_store_bytes by making the first arg implicit

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-11-03 21:11:16 +05:30
parent 67c9d9b932
commit 5e371787eb
2 changed files with 12 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
pythonbpf/helper/bpf_helper_handler.py
View File

@ -583,16 +583,14 @@ def bpf_skb_store_bytes_emitter(
Expected call signature: skb_store_bytes(skb, offset, from, len, flags) Expected call signature: skb_store_bytes(skb, offset, from, len, flags)
""" """
if len(call.args) not in (4, 5): if len(call.args) not in (3, 4):
raise ValueError( raise ValueError(
f"skb_store_bytes expects 4 or 5 args (skb, offset, from, len, flags), got {len(call.args)}" f"skb_store_bytes expects 3 or 4 args (offset, from, len, flags), got {len(call.args)}"
) )
skb_ptr = get_or_create_ptr_from_arg( skb_ptr = func.args[0] # First argument to the function is skb
func, module, call.args[0], builder, local_sym_tab, map_sym_tab, struct_sym_tab
)
offset_val = get_int_value_from_arg( offset_val = get_int_value_from_arg(
call.args[1], call.args[0],
func, func,
module, module,
builder, builder,
@ -601,10 +599,10 @@ def bpf_skb_store_bytes_emitter(
struct_sym_tab, struct_sym_tab,
) )
from_ptr = get_or_create_ptr_from_arg( from_ptr = get_or_create_ptr_from_arg(
func, module, call.args[2], builder, local_sym_tab, map_sym_tab, struct_sym_tab func, module, call.args[1], builder, local_sym_tab, map_sym_tab, struct_sym_tab
) )
len_val = get_int_value_from_arg( len_val = get_int_value_from_arg(
call.args[3], call.args[2],
func, func,
module, module,
builder, builder,
@ -612,10 +610,11 @@ def bpf_skb_store_bytes_emitter(
map_sym_tab, map_sym_tab,
struct_sym_tab, struct_sym_tab,
) )
if len(call.args) == 5: if len(call.args) == 4:
flags_val = get_flags_val(call.args[4], builder, local_sym_tab) flags_val = get_flags_val(call.args[3], builder, local_sym_tab)
else: else:
flags_val = ir.Constant(ir.IntType(64), 0) flags_val = 0
flags = ir.Constant(ir.IntType(64), flags_val)
fn_type = ir.FunctionType( fn_type = ir.FunctionType(
ir.IntType(64), ir.IntType(64),
[ [
@ -638,7 +637,7 @@ def bpf_skb_store_bytes_emitter(
builder.trunc(offset_val, ir.IntType(32)), builder.trunc(offset_val, ir.IntType(32)),
builder.bitcast(from_ptr, ir.PointerType()), builder.bitcast(from_ptr, ir.PointerType()),
builder.trunc(len_val, ir.IntType(32)), builder.trunc(len_val, ir.IntType(32)),
flags_val, flags,
], ],
tail=False, tail=False,
) )

2
pythonbpf/helper/helpers.py
View File

@ -47,7 +47,7 @@ def uid():
return ctypes.c_int32(0) return ctypes.c_int32(0)
def skb_store_bytes(skb, offset, from_buf, size, flags=0): def skb_store_bytes(offset, from_buf, size, flags=0):
"""store bytes into a socket buffer""" """store bytes into a socket buffer"""
return ctypes.c_int64(0) return ctypes.c_int64(0)