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 var assigning for helpers

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-09-10 01:49:14 +05:30
parent dafd6d18ab
commit 55c9b2ebe1
3 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/execve2.py
View File

@ -25,7 +25,7 @@ def hello_again(ctx: c_void_p) -> c_int64:
key = 0 key = 0
tsp = last().lookup(key) tsp = last().lookup(key)
print(tsp) print(tsp)
ktime() ts = ktime()
return c_int64(0) return c_int64(0)

4
pythonbpf/bpf_helper_handler.py
View File

@ -107,7 +107,7 @@ def handle_helper_call(call, module, builder, func, local_sym_tab=None, map_sym_
func_name = call.func.id func_name = call.func.id
if func_name in helper_func_list: if func_name in helper_func_list:
# it is not a map method call # it is not a map method call
helper_func_list[func_name](call, module, builder, func) return helper_func_list[func_name](call, module, builder, func)
else: else:
raise NotImplementedError( raise NotImplementedError(
f"Function {func_name} is not implemented as a helper function.") f"Function {func_name} is not implemented as a helper function.")
@ -119,7 +119,7 @@ def handle_helper_call(call, module, builder, func, local_sym_tab=None, map_sym_
if map_sym_tab and map_name in map_sym_tab: if map_sym_tab and map_name in map_sym_tab:
map_ptr = map_sym_tab[map_name] map_ptr = map_sym_tab[map_name]
if method_name in helper_func_list: if method_name in helper_func_list:
helper_func_list[method_name]( return helper_func_list[method_name](
call, map_ptr, module, builder, local_sym_tab) call, map_ptr, module, builder, local_sym_tab)
else: else:
raise NotImplementedError( raise NotImplementedError(

11
pythonbpf/functions_pass.py
View File

@ -39,7 +39,6 @@ def handle_assign(module, builder, stmt, map_sym_tab, local_sym_tab):
if isinstance(rval, ast.Constant): if isinstance(rval, ast.Constant):
if isinstance(rval.value, int): if isinstance(rval.value, int):
# Assume c_int64 for now # Assume c_int64 for now
# TODO: make symtab for this
var = builder.alloca(ir.IntType(64), name=var_name) var = builder.alloca(ir.IntType(64), name=var_name)
var.align = 8 var.align = 8
builder.store(ir.Constant(ir.IntType(64), rval.value), var) builder.store(ir.Constant(ir.IntType(64), rval.value), var)
@ -57,6 +56,14 @@ def handle_assign(module, builder, stmt, map_sym_tab, local_sym_tab):
print(f"Assigned {call_type} constant " print(f"Assigned {call_type} constant "
f"{rval.args[0].value} to {var_name}") f"{rval.args[0].value} to {var_name}")
local_sym_tab[var_name] = var local_sym_tab[var_name] = var
elif call_type in helper_func_list:
var = builder.alloca(ir.IntType(64), name=var_name)
var.align = 8
val = handle_helper_call(
rval, module, builder, None, local_sym_tab, map_sym_tab)
builder.store(val, var)
local_sym_tab[var_name] = var
print(f"Assigned constant {rval.func.id} to {var_name}")
else: else:
print(f"Unsupported assignment call type: {call_type}") print(f"Unsupported assignment call type: {call_type}")
elif isinstance(rval.func, ast.Attribute): elif isinstance(rval.func, ast.Attribute):
@ -70,6 +77,8 @@ def handle_assign(module, builder, stmt, map_sym_tab, local_sym_tab):
rval, module, builder, None, local_sym_tab, map_sym_tab) rval, module, builder, None, local_sym_tab, map_sym_tab)
else: else:
print("Unsupported assignment call structure") print("Unsupported assignment call structure")
else:
print("Unsupported assignment call function type")
def process_func_body(module, builder, func_node, func, ret_type, map_sym_tab): def process_func_body(module, builder, func_node, func, ret_type, map_sym_tab):