Commit f062226d authored by Yafang Shao's avatar Yafang Shao Committed by Alexei Starovoitov
Browse files

bpf: bpf_struct_ops memory usage 



A new helper is introduced to calculate bpf_struct_ops memory usage.

The result as follows,

- before
1: struct_ops  name count_map  flags 0x0
        key 4B  value 256B  max_entries 1  memlock 4096B
        btf_id 73

- after
1: struct_ops  name count_map  flags 0x0
        key 4B  value 256B  max_entries 1  memlock 5016B
        btf_id 73

Signed-off-by: default avatarYafang Shao <laoar.shao@gmail.com>
Link: https://lore.kernel.org/r/20230305124615.12358-13-laoar.shao@gmail.com


Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
parent c6e66b42

kernel/bpf/bpf_struct_ops.c

+16 −0
Original line number Diff line number Diff line
@@ -641,6 +641,21 @@ static struct bpf_map *bpf_struct_ops_map_alloc(union bpf_attr *attr) 
	return map;

}



static u64 bpf_struct_ops_map_mem_usage(const struct bpf_map *map)

{

	struct bpf_struct_ops_map *st_map = (struct bpf_struct_ops_map *)map;

	const struct bpf_struct_ops *st_ops = st_map->st_ops;

	const struct btf_type *vt = st_ops->value_type;

	u64 usage;



	usage = sizeof(*st_map) +

			vt->size - sizeof(struct bpf_struct_ops_value);

	usage += vt->size;

	usage += btf_type_vlen(vt) * sizeof(struct bpf_links *);

	usage += PAGE_SIZE;

	return usage;

}



BTF_ID_LIST_SINGLE(bpf_struct_ops_map_btf_ids, struct, bpf_struct_ops_map)

const struct bpf_map_ops bpf_struct_ops_map_ops = {

	.map_alloc_check = bpf_struct_ops_map_alloc_check,
@@ -651,6 +666,7 @@ const struct bpf_map_ops bpf_struct_ops_map_ops = { 
	.map_delete_elem = bpf_struct_ops_map_delete_elem,

	.map_update_elem = bpf_struct_ops_map_update_elem,

	.map_seq_show_elem = bpf_struct_ops_map_seq_show_elem,

	.map_mem_usage = bpf_struct_ops_map_mem_usage,

	.map_btf_id = &bpf_struct_ops_map_btf_ids[0],

};