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

rename get_key_ptr to get_or_create_ptr_from_arg

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-10-01 18:21:42 +05:30
parent 7f6c318069
commit 8dd2746411
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
pythonbpf/helper/helper_utils.py
View File

@ -37,14 +37,14 @@ def create_int_constant_ptr(value, builder, int_width=64):
return ptr
def get_key_ptr(key_arg, builder, local_sym_tab):
def get_or_create_ptr_from_arg(arg, builder, local_sym_tab):
"""Extract key pointer from the call arguments."""
if isinstance(key_arg, ast.Name):
key_ptr = get_var_ptr_from_name(key_arg.id, local_sym_tab)
elif isinstance(key_arg, ast.Constant) and isinstance(key_arg.value, int):
key_ptr = create_int_constant_ptr(key_arg.value, builder)
if isinstance(arg, ast.Name):
ptr = get_var_ptr_from_name(arg.id, local_sym_tab)
elif isinstance(arg, ast.Constant) and isinstance(arg.value, int):
ptr = create_int_constant_ptr(arg.value, builder)
else:
raise NotImplementedError(
"Only simple variable names are supported as keys in map lookup.")
return key_ptr
return ptr