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

Fix usage of map_params to use BPFMapType

Browse Source
This commit is contained in:
Pragyansh Chaturvedi
2025-09-30 13:28:18 +05:30
parent 1a1f2cf634
commit 9ae1b6ce15
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pythonbpf/maps_pass.py
View File

@ -37,8 +37,7 @@ BPF_MAP_MAPPINGS = {
def create_bpf_map(module, map_name, map_params):
"""Create a BPF map in the module with the given parameters and debug info"""
map_type_str = map_params.get("type", "HASH")
map_type = BPF_MAP_MAPPINGS.get(map_type_str)
map_type = map_params.get("type", BPFMapType.HASH).value
# Create the anonymous struct type for BPF map
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)
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", {
"tag": dc.DW_TAG_array_type,
"baseType": uint_type,