1 files changed, 26 insertions, 0 deletions

diff --git a/kernel/bpf/arraymap.c b/kernel/bpf/arraymap.c
index 96e9c5c1dfc9..98c0f00c3f5e 100644
--- a/kernel/bpf/arraymap.c
+++ b/kernel/bpf/arraymap.c
@@ -606,6 +606,31 @@ static void *array_of_map_lookup_elem(struct bpf_map *map, void *key)
         return READ_ONCE(*inner_map);
 }
 
+static u32 array_of_map_gen_lookup(struct bpf_map *map,
+                                   struct bpf_insn *insn_buf)
+{
+        u32 elem_size = round_up(map->value_size, 8);
+        struct bpf_insn *insn = insn_buf;
+        const int ret = BPF_REG_0;
+        const int map_ptr = BPF_REG_1;
+        const int index = BPF_REG_2;
+
+        *insn++ = BPF_ALU64_IMM(BPF_ADD, map_ptr, offsetof(struct bpf_array, value));
+        *insn++ = BPF_LDX_MEM(BPF_W, ret, index, 0);
+        *insn++ = BPF_JMP_IMM(BPF_JGE, ret, map->max_entries, 5);
+        if (is_power_of_2(elem_size))
+                *insn++ = BPF_ALU64_IMM(BPF_LSH, ret, ilog2(elem_size));
+        else
+                *insn++ = BPF_ALU64_IMM(BPF_MUL, ret, elem_size);
+        *insn++ = BPF_ALU64_REG(BPF_ADD, ret, map_ptr);
+        *insn++ = BPF_LDX_MEM(BPF_DW, ret, ret, 0);
+        *insn++ = BPF_JMP_IMM(BPF_JEQ, ret, 0, 1);
+        *insn++ = BPF_JMP_IMM(BPF_JA, 0, 0, 1);
+        *insn++ = BPF_MOV64_IMM(ret, 0);
+
+        return insn - insn_buf;
+}
+
 const struct bpf_map_ops array_of_maps_map_ops = {
         .map_alloc = array_of_map_alloc,
         .map_free = array_of_map_free,
@@ -615,4 +640,5 @@ const struct bpf_map_ops array_of_maps_map_ops = {
         .map_fd_get_ptr = bpf_map_fd_get_ptr,
         .map_fd_put_ptr = bpf_map_fd_put_ptr,
         .map_fd_sys_lookup_elem = bpf_map_fd_sys_lookup_elem,
+        .map_gen_lookup = array_of_map_gen_lookup,
 };