1 files changed, 32 insertions, 0 deletions

diff --git a/kernel/bpf/arraymap.c b/kernel/bpf/arraymap.c
index c72e0d8e1e65..1a6e9861d554 100644
--- a/kernel/bpf/arraymap.c
+++ b/kernel/bpf/arraymap.c
@@ -160,6 +160,36 @@ static void *array_map_lookup_elem(struct bpf_map *map, void *key)
         return array->value + array->elem_size * (index & array->index_mask);
 }
 
+static int array_map_direct_value_addr(const struct bpf_map *map, u64 *imm,
+                                       u32 off)
+{
+        struct bpf_array *array = container_of(map, struct bpf_array, map);
+
+        if (map->max_entries != 1)
+                return -ENOTSUPP;
+        if (off >= map->value_size)
+                return -EINVAL;
+
+        *imm = (unsigned long)array->value;
+        return 0;
+}
+
+static int array_map_direct_value_meta(const struct bpf_map *map, u64 imm,
+                                       u32 *off)
+{
+        struct bpf_array *array = container_of(map, struct bpf_array, map);
+        u64 base = (unsigned long)array->value;
+        u64 range = array->elem_size;
+
+        if (map->max_entries != 1)
+                return -ENOTSUPP;
+        if (imm < base || imm >= base + range)
+                return -ENOENT;
+
+        *off = imm - base;
+        return 0;
+}
+
 /* emit BPF instructions equivalent to C code of array_map_lookup_elem() */
 static u32 array_map_gen_lookup(struct bpf_map *map, struct bpf_insn *insn_buf)
 {
@@ -419,6 +449,8 @@ const struct bpf_map_ops array_map_ops = {
         .map_update_elem = array_map_update_elem,
         .map_delete_elem = array_map_delete_elem,
         .map_gen_lookup = array_map_gen_lookup,
+        .map_direct_value_addr = array_map_direct_value_addr,
+        .map_direct_value_meta = array_map_direct_value_meta,
         .map_seq_show_elem = array_map_seq_show_elem,
         .map_check_btf = array_map_check_btf,
 };