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

Make bpf structs discoverable during chunk exploration

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-09-21 03:17:23 +05:30
parent 48c0a1f506
commit 780f53cd3f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pythonbpf/codegen.py
View File

@ -14,7 +14,7 @@ def find_bpf_chunks(tree):
"""Find all functions decorated with @bpf in the AST.""" """Find all functions decorated with @bpf in the AST."""
bpf_functions = [] bpf_functions = []
for node in ast.walk(tree): for node in ast.walk(tree):
if isinstance(node, ast.FunctionDef): if isinstance(node, ast.FunctionDef) or isinstance(node, ast.ClassDef):
for decorator in node.decorator_list: for decorator in node.decorator_list:
if isinstance(decorator, ast.Name) and decorator.id == "bpf": if isinstance(decorator, ast.Name) and decorator.id == "bpf":
bpf_functions.append(node) bpf_functions.append(node)
@ -28,7 +28,7 @@ def processor(source_code, filename, module):
bpf_chunks = find_bpf_chunks(tree) bpf_chunks = find_bpf_chunks(tree)
for func_node in bpf_chunks: for func_node in bpf_chunks:
print(f"Found BPF function: {func_node.name}") print(f"Found BPF function/struct: {func_node.name}")
map_sym_tab = maps_proc(tree, module, bpf_chunks) map_sym_tab = maps_proc(tree, module, bpf_chunks)
func_proc(tree, module, bpf_chunks, map_sym_tab) func_proc(tree, module, bpf_chunks, map_sym_tab)