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

Use logger instead of prints in map_pass

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-09-30 19:44:29 +05:30
parent 9fa362ec6a
commit 744aa3fbdf
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
pythonbpf/maps/maps_pass.py
View File

@ -3,6 +3,9 @@ from llvmlite import ir
from pythonbpf import dwarf_constants as dc from pythonbpf import dwarf_constants as dc
from enum import Enum from enum import Enum
from .maps_utils import MapProcessorRegistry from .maps_utils import MapProcessorRegistry
import logging
logger = logging.getLogger(__name__)
def maps_proc(tree, module, chunks): def maps_proc(tree, module, chunks):
@ -46,7 +49,7 @@ def create_bpf_map(module, map_name, map_params):
# Generate debug info for BTF # Generate debug info for BTF
create_map_debug_info(module, map_global, map_name, map_params) create_map_debug_info(module, map_global, map_name, map_params)
print(f"Created BPF map: {map_name}") logger.info(f"Created BPF map: {map_name} with params {map_params}")
return map_global return map_global
@ -182,7 +185,7 @@ def create_map_debug_info(module, map_global, map_name, map_params):
@MapProcessorRegistry.register("HashMap") @MapProcessorRegistry.register("HashMap")
def process_hash_map(map_name, rval, module): def process_hash_map(map_name, rval, module):
"""Process a BPF_HASH map declaration""" """Process a BPF_HASH map declaration"""
print(f"Creating HashMap map: {map_name}") logger.info(f"Processing HashMap: {map_name}")
map_params = {"type": BPFMapType.HASH} map_params = {"type": BPFMapType.HASH}
# Assuming order: key_type, value_type, max_entries # Assuming order: key_type, value_type, max_entries
@ -206,14 +209,14 @@ def process_hash_map(map_name, rval, module):
if isinstance(const_val, (int, str)): if isinstance(const_val, (int, str)):
map_params["max_entries"] = const_val map_params["max_entries"] = const_val
print(f"Map parameters: {map_params}") logger.info(f"Map parameters: {map_params}")
return create_bpf_map(module, map_name, map_params) return create_bpf_map(module, map_name, map_params)
@MapProcessorRegistry.register("PerfEventArray") @MapProcessorRegistry.register("PerfEventArray")
def process_perf_event_map(map_name, rval, module): def process_perf_event_map(map_name, rval, module):
"""Process a BPF_PERF_EVENT_ARRAY map declaration""" """Process a BPF_PERF_EVENT_ARRAY map declaration"""
print(f"Creating PerfEventArray map: {map_name}") logger.info(f"Processing PerfEventArray: {map_name}")
map_params = {"type": BPFMapType.PERF_EVENT_ARRAY} map_params = {"type": BPFMapType.PERF_EVENT_ARRAY}
if len(rval.args) >= 1 and isinstance(rval.args[0], ast.Name): if len(rval.args) >= 1 and isinstance(rval.args[0], ast.Name):
@ -228,14 +231,14 @@ def process_perf_event_map(map_name, rval, module):
isinstance(keyword.value, ast.Name): isinstance(keyword.value, ast.Name):
map_params["value_size"] = keyword.value.id map_params["value_size"] = keyword.value.id
print(f"Map parameters: {map_params}") logger.info(f"Map parameters: {map_params}")
return create_bpf_map(module, map_name, map_params) return create_bpf_map(module, map_name, map_params)
def process_bpf_map(func_node, module): def process_bpf_map(func_node, module):
"""Process a BPF map (a function decorated with @map)""" """Process a BPF map (a function decorated with @map)"""
map_name = func_node.name map_name = func_node.name
print(f"Processing BPF map: {map_name}") logger.info(f"Processing BPF map: {map_name}")
# For now, assume single return statement # For now, assume single return statement
return_stmt = None return_stmt = None
@ -253,7 +256,8 @@ def process_bpf_map(func_node, module):
if handler: if handler:
return handler(map_name, rval, module) return handler(map_name, rval, module)
else: else:
print(f"Unknown map type {rval.func.id}, defaulting to HashMap") logger.warning(f"Unknown map type {
rval.func.id}, defaulting to HashMap")
process_hash_map(map_name, rval, module) 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")