bpf: Limit caller's stack depth 256 for subprogs with tailcalls

[ Upstream commit 7f6e4312e15a5c370e84eaa685879b6bdcc717e4 ]

Protect against potential stack overflow that might happen when bpf2bpf
calls get combined with tailcalls. Limit the caller's stack depth for
such case down to 256 so that the worst case scenario would result in 8k
stack size (32 which is tailcall limit * 256 = 8k).

Suggested-by: Alexei Starovoitov <ast@kernel.org>
Signed-off-by: Maciej Fijalkowski <maciej.fijalkowski@intel.com>
Signed-off-by: Alexei Starovoitov <ast@kernel.org>
Signed-off-by: Sasha Levin <sashal@kernel.org>

diff --git a/include/linux/bpf_verifier.h b/include/linux/bpf_verifier.h
index ca08db4ffb5f74f6ae4655322499d4df4cdd2d72..ce3f5231aa698dd02a7b4671b5e13b2b6d37335e 100644 (file)
--- a/include/linux/bpf_verifier.h
+++ b/include/linux/bpf_verifier.h
@@ -358,6 +358,7 @@ struct bpf_subprog_info {
        u32 start; /* insn idx of function entry point */
        u32 linfo_idx; /* The idx to the main_prog->aux->linfo */
        u16 stack_depth; /* max. stack depth used by this function */
+       bool has_tail_call;
 };
 
 /* single container for all structs

diff --git a/kernel/bpf/verifier.c b/kernel/bpf/verifier.c
index c953dfbbaa6a9b27272d4a274824dd50927bbc2b..12eb9e47d101c8c051e34a9319f52c5d018e48eb 100644 (file)
--- a/kernel/bpf/verifier.c
+++ b/kernel/bpf/verifier.c
@@ -1470,6 +1470,10 @@ static int check_subprogs(struct bpf_verifier_env *env)
        for (i = 0; i < insn_cnt; i++) {
                u8 code = insn[i].code;
 
+               if (code == (BPF_JMP | BPF_CALL) &&
+                   insn[i].imm == BPF_FUNC_tail_call &&
+                   insn[i].src_reg != BPF_PSEUDO_CALL)
+                       subprog[cur_subprog].has_tail_call = true;
                if (BPF_CLASS(code) != BPF_JMP && BPF_CLASS(code) != BPF_JMP32)
                        goto next;
                if (BPF_OP(code) == BPF_EXIT || BPF_OP(code) == BPF_CALL)
@@ -2951,6 +2955,31 @@ static int check_max_stack_depth(struct bpf_verifier_env *env)
        int ret_prog[MAX_CALL_FRAMES];
 
 process_func:
+       /* protect against potential stack overflow that might happen when
+        * bpf2bpf calls get combined with tailcalls. Limit the caller's stack
+        * depth for such case down to 256 so that the worst case scenario
+        * would result in 8k stack size (32 which is tailcall limit * 256 =
+        * 8k).
+        *
+        * To get the idea what might happen, see an example:
+        * func1 -> sub rsp, 128
+        *  subfunc1 -> sub rsp, 256
+        *  tailcall1 -> add rsp, 256
+        *   func2 -> sub rsp, 192 (total stack size = 128 + 192 = 320)
+        *   subfunc2 -> sub rsp, 64
+        *   subfunc22 -> sub rsp, 128
+        *   tailcall2 -> add rsp, 128
+        *    func3 -> sub rsp, 32 (total stack size 128 + 192 + 64 + 32 = 416)
+        *
+        * tailcall will unwind the current stack frame but it will not get rid
+        * of caller's stack as shown on the example above.
+        */
+       if (idx && subprog[idx].has_tail_call && depth >= 256) {
+               verbose(env,
+                       "tail_calls are not allowed when call stack of previous frames is %d bytes. Too large\n",
+                       depth);
+               return -EACCES;
+       }
        /* round up to 32-bytes, since this is granularity
         * of interpreter stack size
         */