bpf: allow for key-less BTF in array map

Given we'll be reusing BPF array maps for global data/bss/rodata
sections, we need a way to associate BTF DataSec type as its map
value type. In usual cases we have this ugly BPF_ANNOTATE_KV_PAIR()
macro hack e.g. via 38d5d3b3d5db ("bpf: Introduce BPF_ANNOTATE_KV_PAIR")
to get initial map to type association going. While more use cases
for it are discouraged, this also won't work for global data since
the use of array map is a BPF loader detail and therefore unknown
at compilation time. For array maps with just a single entry we make
an exception in terms of BTF in that key type is declared optional
if value type is of DataSec type. The latter LLVM is guaranteed to
emit and it also aligns with how we regard global data maps as just
a plain buffer area reusing existing map facilities for allowing
things like introspection with existing tools.

Signed-off-by: Daniel Borkmann <daniel@iogearbox.net>
Acked-by: Martin KaFai Lau <kafai@fb.com>
Signed-off-by: Alexei Starovoitov <ast@kernel.org>

1 files changed, 11 insertions, 4 deletions

diff --git a/kernel/bpf/syscall.c b/kernel/bpf/syscall.c
index 198c9680bf0d..438199e2eca4 100644
--- a/kernel/bpf/syscall.c
+++ b/kernel/bpf/syscall.c
@@ -504,9 +504,16 @@ static int map_check_btf(struct bpf_map *map, const struct btf *btf,
         u32 key_size, value_size;
         int ret = 0;
 
-        key_type = btf_type_id_size(btf, &btf_key_id, &key_size);
-        if (!key_type || key_size != map->key_size)
-                return -EINVAL;
+        /* Some maps allow key to be unspecified. */
+        if (btf_key_id) {
+                key_type = btf_type_id_size(btf, &btf_key_id, &key_size);
+                if (!key_type || key_size != map->key_size)
+                        return -EINVAL;
+        } else {
+                key_type = btf_type_by_id(btf, 0);
+                if (!map->ops->map_check_btf)
+                        return -EINVAL;
+        }
 
         value_type = btf_type_id_size(btf, &btf_value_id, &value_size);
         if (!value_type || value_size != map->value_size)
@@ -573,7 +580,7 @@ static int map_create(union bpf_attr *attr)
         if (attr->btf_key_type_id || attr->btf_value_type_id) {
                 struct btf *btf;
 
-                if (!attr->btf_key_type_id || !attr->btf_value_type_id) {
+                if (!attr->btf_value_type_id) {
                         err = -EINVAL;
                         goto free_map_nouncharge;
                 }