Fix usage of map_params to use BPFMapType

This commit is contained in:
Pragyansh Chaturvedi
2025-09-30 13:28:18 +05:30
parent 1a1f2cf634
commit 9ae1b6ce15

View File

@ -37,8 +37,7 @@ BPF_MAP_MAPPINGS = {
def create_bpf_map(module, map_name, map_params): def create_bpf_map(module, map_name, map_params):
"""Create a BPF map in the module with the given parameters and debug info""" """Create a BPF map in the module with the given parameters and debug info"""
map_type_str = map_params.get("type", "HASH") map_type = map_params.get("type", BPFMapType.HASH).value
map_type = BPF_MAP_MAPPINGS.get(map_type_str)
# Create the anonymous struct type for BPF map # Create the anonymous struct type for BPF map
map_struct_type = ir.LiteralStructType( map_struct_type = ir.LiteralStructType(
@ -82,7 +81,7 @@ def create_map_debug_info(module, map_global, map_name, map_params):
# Create array type for map type field (array of 1 unsigned int) # Create array type for map type field (array of 1 unsigned int)
array_subrange = module.add_debug_info( array_subrange = module.add_debug_info(
"DISubrange", {"count": BPF_MAP_MAPPINGS[map_params.get("type", "HASH")]}) "DISubrange", {"count": map_params.get("type", BPFMapType.HASH).value})
array_type = module.add_debug_info("DICompositeType", { array_type = module.add_debug_info("DICompositeType", {
"tag": dc.DW_TAG_array_type, "tag": dc.DW_TAG_array_type,
"baseType": uint_type, "baseType": uint_type,