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 if statement barebones

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-09-10 03:12:25 +05:30
parent 617aac973c
commit 357ad7cb99
1 changed files with 24 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
pythonbpf/functions_pass.py
View File

@ -81,6 +81,10 @@ def handle_assign(module, builder, stmt, map_sym_tab, local_sym_tab):
print("Unsupported assignment call function type") print("Unsupported assignment call function type")
def handle_if_statement(module, builder, stmt, map_sym_tab, local_sym_tab):
pass
def handle_expr(module, builder, expr, local_sym_tab, map_sym_tab): def handle_expr(module, builder, expr, local_sym_tab, map_sym_tab):
"""Handle expression statements in the function body.""" """Handle expression statements in the function body."""
call = expr.value call = expr.value
@ -94,6 +98,24 @@ def handle_expr(module, builder, expr, local_sym_tab, map_sym_tab):
print("Unsupported expression statement") print("Unsupported expression statement")
def handle_if(module, builder, stmt, map_sym_tab, local_sym_tab):
"""Handle if statements in the function body."""
func = builder.block.parent
then_block = func.append_basic_block(name="if.then")
merge_block = func.append_basic_block(name="if.end")
cond = stmt.test
builder.cbranch(cond, then_block, merge_block)
builder.position_at_end(then_block)
for s in stmt.body:
pass
if not builder.block.is_terminated:
builder.branch(merge_block)
builder.position_at_end(merge_block)
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):
"""Process the body of a bpf function""" """Process the body of a bpf function"""
# TODO: A lot. We just have print -> bpf_trace_printk for now # TODO: A lot. We just have print -> bpf_trace_printk for now
@ -106,6 +128,8 @@ def process_func_body(module, builder, func_node, func, ret_type, map_sym_tab):
handle_expr(module, builder, stmt, local_sym_tab, map_sym_tab) handle_expr(module, builder, stmt, local_sym_tab, map_sym_tab)
elif isinstance(stmt, ast.Assign): elif isinstance(stmt, ast.Assign):
handle_assign(module, builder, stmt, map_sym_tab, local_sym_tab) handle_assign(module, builder, stmt, map_sym_tab, local_sym_tab)
elif isinstance(stmt, ast.If):
handle_if(module, builder, stmt, map_sym_tab, local_sym_tab)
elif isinstance(stmt, ast.Return): elif isinstance(stmt, ast.Return):
if stmt.value is None: if stmt.value is None:
builder.ret(ir.Constant(ir.IntType(32), 0)) builder.ret(ir.Constant(ir.IntType(32), 0))