From 0d691865bc088b40a7cf5d97db19c3bddb18bf97 Mon Sep 17 00:00:00 2001 From: Pragyansh Chaturvedi Date: Tue, 30 Sep 2025 11:42:14 +0530 Subject: [PATCH] Add is_map in maps_pass --- pythonbpf/maps_pass.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/pythonbpf/maps_pass.py b/pythonbpf/maps_pass.py index 38deecb..0553996 100644 --- a/pythonbpf/maps_pass.py +++ b/pythonbpf/maps_pass.py @@ -8,19 +8,20 @@ map_sym_tab = {} def maps_proc(tree, module, chunks): for func_node in chunks: - # Check if this function is a map - is_map = False - for decorator in func_node.decorator_list: - if isinstance(decorator, ast.Name) and decorator.id == "map": - is_map = True - break - if is_map: + if is_map(func_node): print(f"Found BPF map: {func_node.name}") process_bpf_map(func_node, module) continue return map_sym_tab +def is_map(func_node): + return any( + isinstance(decorator, ast.Name) and decorator.id == "map" + for decorator in func_node.decorator_list + ) + + BPF_MAP_MAPPINGS = { "HASH": 1, # BPF_MAP_TYPE_HASH "PERF_EVENT_ARRAY": 4, # BPF_MAP_TYPE_PERF_EVENT_ARRAY