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 map type support to process_bpf_map

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-09-20 02:19:17 +05:30
parent 69d0cf2e0e
commit 67fc3f9562
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
pythonbpf/maps_pass.py
View File

@ -231,6 +231,10 @@ def process_bpf_map(func_node, module):
map_name = func_node.name map_name = func_node.name
print(f"Processing BPF map: {map_name}") print(f"Processing BPF map: {map_name}")
BPF_MAP_TYPES = {"HashMap": process_hash_map, # BPF_MAP_TYPE_HASH
"PerfEventArray": process_perf_event_map, # BPF_MAP_TYPE_PERF_EVENT_ARRAY
}
# For now, assume single return statement # For now, assume single return statement
return_stmt = None return_stmt = None
for stmt in func_node.body: for stmt in func_node.body:
@ -243,7 +247,12 @@ def process_bpf_map(func_node, module):
rval = return_stmt.value rval = return_stmt.value
# Handle only HashMap maps # Handle only HashMap maps
if isinstance(rval, ast.Call) and isinstance(rval.func, ast.Name) and rval.func.id == "HashMap": if isinstance(rval, ast.Call) and isinstance(rval.func, ast.Name):
process_hash_map(map_name, rval, module) if rval.func.id in BPF_MAP_TYPES:
handler = BPF_MAP_TYPES[rval.func.id]
handler(map_name, rval, module)
else:
print(f"Unknown map type {rval.func.id}, defaulting to HashMap")
process_hash_map(map_name, rval, module)
else: else:
raise ValueError("Function under @map must return a map") raise ValueError("Function under @map must return a map")