Merge branch 'support-alu32_arsh'

Jiong Wang says:

====================
BPF_ALU | BPF_ARSH | BPF_* were rejected by commit: 7891a87efc71
("bpf: arsh is not supported in 32 bit alu thus reject it"). As explained
in the commit message, this is due to there is no complete support for them
on interpreter and various JIT compilation back-ends.

This patch set is a follow-up which completes the missing bits. This also
pave the way for running bpf program compiled with ALU32 instruction
enabled by specifing -mattr=+alu32 to LLVM for which case there is likely
to have more BPF_ALU | BPF_ARSH insns that will trigger the rejection code.

test_verifier.c is updated accordingly.

I have tested this patch set on x86-64 and NFP, I need help of review and
test on the arch changes (mips/ppc/s390).

Note, there might be merge confict on mips change which is better to be
applied on top of:

  commit: 20b880a05f06 ("mips: bpf: fix encoding bug for mm_srlv32_op"),

which is on mips-fixes branch at the moment.

Thanks.

v1->v2:
 - Fix ppc implementation bug. Should zero high bits explicitly.
====================

Signed-off-by: Alexei Starovoitov <ast@kernel.org>

1 files changed, 6 insertions, 0 deletions

diff --git a/arch/powerpc/net/bpf_jit_comp64.c b/arch/powerpc/net/bpf_jit_comp64.c
index 17482f5de3e2..7dc81877057d 100644
--- a/arch/powerpc/net/bpf_jit_comp64.c
+++ b/arch/powerpc/net/bpf_jit_comp64.c
@@ -529,9 +529,15 @@ static int bpf_jit_build_body(struct bpf_prog *fp, u32 *image,
                         if (imm != 0)
                                 PPC_SRDI(dst_reg, dst_reg, imm);
                         break;
+                case BPF_ALU | BPF_ARSH | BPF_X: /* (s32) dst >>= src */
+                        PPC_SRAW(dst_reg, dst_reg, src_reg);
+                        goto bpf_alu32_trunc;
                 case BPF_ALU64 | BPF_ARSH | BPF_X: /* (s64) dst >>= src */
                         PPC_SRAD(dst_reg, dst_reg, src_reg);
                         break;
+                case BPF_ALU | BPF_ARSH | BPF_K: /* (s32) dst >>= imm */
+                        PPC_SRAWI(dst_reg, dst_reg, imm);
+                        goto bpf_alu32_trunc;
                 case BPF_ALU64 | BPF_ARSH | BPF_K: /* (s64) dst >>= imm */
                         if (imm != 0)
                                 PPC_SRADI(dst_reg, dst_reg, imm);