[PATCH bpf-next v6 03/12] bpf: Disallow interpreter fallback for BPF_ADDR_PERCPU insn

[Leon Hwang]

Since interpreter is unable to handle the 'insn_is_mov_percpu_addr()' insn,
require JIT in __bpf_prog_select_runtime() when the prog has the insn.

BTW, rename the helper bpf_map_supports_cpu_flags() to
bpf_map_is_percpu_map().

Fixes: 7bdbf7446305 ("bpf: add special internal-only MOV instruction to resolve 
per-CPU addrs")
Signed-off-by: Leon Hwang <leon.hw...@linux.dev>
---
 include/linux/bpf.h | 5 +++--
 kernel/bpf/core.c   | 1 +
 kernel/bpf/fixups.c | 5 +++++
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/include/linux/bpf.h b/include/linux/bpf.h
index 5f48a6ab8a1a..24ab17a25046 100644
--- a/include/linux/bpf.h
+++ b/include/linux/bpf.h
@@ -1781,6 +1781,7 @@ struct bpf_prog_aux {
        bool might_sleep;
        bool kprobe_write_ctx;
        bool has_addr_space_cast_insn;
+       bool has_addr_percpu_insn;
        struct {
                s32 keyring_serial;
                u8 keyring_type;
@@ -4167,7 +4168,7 @@ bpf_prog_update_insn_ptrs(struct bpf_prog *prog, u32 
*offsets, void *image)
 }
 #endif
 
-static inline bool bpf_map_supports_cpu_flags(enum bpf_map_type map_type)
+static inline bool bpf_map_is_percpu_map(enum bpf_map_type map_type)
 {
        switch (map_type) {
        case BPF_MAP_TYPE_PERCPU_ARRAY:
@@ -4194,7 +4195,7 @@ static inline int bpf_map_check_op_flags(struct bpf_map 
*map, u64 flags, u64 all
                return -EINVAL;
 
        if (flags & (BPF_F_CPU | BPF_F_ALL_CPUS)) {
-               if (!bpf_map_supports_cpu_flags(map->map_type))
+               if (!bpf_map_is_percpu_map(map->map_type))
                        return -EINVAL;
                if ((flags & BPF_F_CPU) && (flags & BPF_F_ALL_CPUS))
                        return -EINVAL;
diff --git a/kernel/bpf/core.c b/kernel/bpf/core.c
index 49398b5bd172..69203d58e0ad 100644
--- a/kernel/bpf/core.c
+++ b/kernel/bpf/core.c
@@ -2621,6 +2621,7 @@ struct bpf_prog *__bpf_prog_select_runtime(struct 
bpf_verifier_env *env, struct
 
        if (IS_ENABLED(CONFIG_BPF_JIT_ALWAYS_ON) ||
            fp->aux->has_addr_space_cast_insn ||
+           fp->aux->has_addr_percpu_insn ||
            bpf_prog_has_kfunc_call(fp))
                jit_needed = true;
 
diff --git a/kernel/bpf/fixups.c b/kernel/bpf/fixups.c
index 2d5958774b61..9552ddcf6936 100644
--- a/kernel/bpf/fixups.c
+++ b/kernel/bpf/fixups.c
@@ -2009,6 +2009,9 @@ int bpf_do_misc_fixups(struct bpf_verifier_env *env)
                                        return -EFAULT;
                                }
 
+                               if (bpf_map_is_percpu_map(map_ptr->map_type))
+                                       prog->aux->has_addr_percpu_insn = true;
+
                                new_prog = bpf_patch_insn_data(env, i + delta,
                                                               insn_buf, cnt);
                                if (!new_prog)
@@ -2113,6 +2116,7 @@ int bpf_do_misc_fixups(struct bpf_verifier_env *env)
                         * way, it's fine to back out this inlining logic
                         */
 #ifdef CONFIG_SMP
+                       prog->aux->has_addr_percpu_insn = true;
                        insn_buf[0] = BPF_MOV64_IMM(BPF_REG_0, (u32)(unsigned 
long)&cpu_number);
                        insn_buf[1] = BPF_MOV64_PERCPU_REG(BPF_REG_0, 
BPF_REG_0);
                        insn_buf[2] = BPF_LDX_MEM(BPF_W, BPF_REG_0, BPF_REG_0, 
0);
@@ -2134,6 +2138,7 @@ int bpf_do_misc_fixups(struct bpf_verifier_env *env)
                /* Implement bpf_get_current_task() and 
bpf_get_current_task_btf() inline. */
                if ((insn->imm == BPF_FUNC_get_current_task || insn->imm == 
BPF_FUNC_get_current_task_btf) &&
                    bpf_verifier_inlines_helper_call(env, insn->imm)) {
+                       prog->aux->has_addr_percpu_insn = true;
                        insn_buf[0] = BPF_MOV64_IMM(BPF_REG_0, (u32)(unsigned 
long)&current_task);
                        insn_buf[1] = BPF_MOV64_PERCPU_REG(BPF_REG_0, 
BPF_REG_0);
                        insn_buf[2] = BPF_LDX_MEM(BPF_DW, BPF_REG_0, BPF_REG_0, 
0);
-- 
2.54.0