Hi, On 10/22/2024 11:59 AM, Alexei Starovoitov wrote: > On Mon, Oct 21, 2024 at 7:02 AM Hou Tao <houtao@xxxxxxxxxxxxxxx> wrote: >> Hi, >> >> On 10/12/2024 12:29 AM, Alexei Starovoitov wrote: >>> On Tue, Oct 8, 2024 at 2:02 AM Hou Tao <houtao@xxxxxxxxxxxxxxx> wrote: >>>> +#define MAX_DYNPTR_CNT_IN_MAP_KEY 4 >>>> + >>>> static int map_check_btf(struct bpf_map *map, struct bpf_token *token, >>>> const struct btf *btf, u32 btf_key_id, u32 btf_value_id) >>>> { >>>> @@ -1103,6 +1113,40 @@ static int map_check_btf(struct bpf_map *map, struct bpf_token *token, >>>> if (!value_type || value_size != map->value_size) >>>> return -EINVAL; >>>> >>>> + if (btf_type_is_dynptr(btf, key_type)) >>>> + map->key_record = btf_new_bpf_dynptr_record(); >>>> + else >>>> + map->key_record = btf_parse_fields(btf, key_type, BPF_DYNPTR, map->key_size); >>>> + if (!IS_ERR_OR_NULL(map->key_record)) { >>>> + if (map->key_record->cnt > MAX_DYNPTR_CNT_IN_MAP_KEY) { >>>> + ret = -E2BIG; >>>> + goto free_map_tab; >>> Took me a while to grasp that map->key_record is only for dynptr fields >>> and map->record is for the rest except dynptr fields. >>> >>> Maybe rename key_record to dynptr_fields ? >>> Or at least add a comment to struct bpf_map to explain >>> what each btf_record is for. >> I was trying to rename map->record to map->value_record, however, I was >> afraid that it may introduce too much churn, so I didn't do that. But I >> think it is a good idea to add comments for both btf_record. And >> considering that only bpf_dynptr is enabled for map key, renaming it to >> dynptr_fields seems reasonable as well. >>> It's kinda arbitrary decision to support multiple dynptr-s per key >>> while other fields are not. >>> Maybe worth looking at generalizing it a bit so single btf_record >>> can have multiple of certain field kinds? >>> In addition to btf_record->cnt you'd need btf_record->dynptr_cnt >>> but that would be easier to extend in the future ? >> Map value has already supported multiple kptrs or bpf_list_node. > fwiw I believe we reached the dead end there. > The whole support for bpf_list and bpf_rb_tree may get deprecated > and removed. The expected users didn't materialize. OK. > >> And in >> the discussion [1], I thought multiple dynptr support in map key is >> necessary, so I enabled it. >> >> [1]: >> https://lore.kernel.org/bpf/CAADnVQJWaBRB=P-ZNkppwm=0tZaT3qP8PKLLJ2S5SSA2-S8mxg@xxxxxxxxxxxxxx/ > Sure. That's a different reasoning and use case. > I'm proposing to use a single btf_record with different cnt-s. > The current btf_record->cnt will stay as-is indicating total number of fields > while btf_record->dynptr_cnt will be just for these dynptrs you're introducing. > Then you won't need two top level btf_record-s. I misunderstood your suggestion yesterday. Now I see what you are suggesting. However, it seems using a separated counter for different kinds of btf_field will only benefit dynptr field, because other types doesn't need to iterate all field instead they just need to find one through binary search. And I don't understand why only one btf_record will be enough, because these two btf_records are derived from map key and map value respectively.