llvm-project/llvm/test/CodeGen/BPF/adjust-opt-speculative1.ll
Yonghong Song ddf1864ace BPF: add AdjustOpt IR pass to generate verifier friendly codes
Add an IR phase right before main module optimization.
This is to modify IR to restrict certain downward optimizations
in order to generate verifier friendly code.
  > prevent certain instcombine optimizations, handling both
    in-block/cross-block instcombines.
  > avoid speculative code motion if the variable used in
    condition is also used in the later blocks.

Internally, a bpf IR builtin
  result = __builtin_bpf_passthrough(seq_num, result)
is used to enforce ordering. This builtin is only used
during target independent IR optimizations and it will
be removed at the beginning of target dependent IR
optimizations.

For example, removing the following workaround,
  --- a/tools/testing/selftests/bpf/progs/test_sysctl_loop1.c
  +++ b/tools/testing/selftests/bpf/progs/test_sysctl_loop1.c
  @@ -47,7 +47,7 @@ int sysctl_tcp_mem(struct bpf_sysctl *ctx)
          /* a workaround to prevent compiler from generating
           * codes verifier cannot handle yet.
           */
  -       volatile int ret;
  +       int ret;
this patch is able to generate code which passed the verifier.

To disable optimization, users need to use "opt" command like below:
  clang -target bpf -O2 -S -emit-llvm -Xclang -disable-llvm-passes test.c
  // disable icmp serialization
  opt -O2 -bpf-disable-serialize-icmp test.ll | llvm-dis > t.ll
  // disable avoid-speculation
  opt -O2 -bpf-disable-avoid-speculation test.ll | llvm-dis > t.ll
  llc t.ll

Differential Revision: https://reviews.llvm.org/D85570
2020-10-07 08:49:10 -07:00

85 lines
3.3 KiB
LLVM

; RUN: opt -O2 -mtriple=bpf-pc-linux %s | llvm-dis > %t1
; RUN: llc %t1 -o - | FileCheck -check-prefixes=CHECK-COMMON,CHECK %s
; RUN: opt -O2 -mtriple=bpf-pc-linux -bpf-disable-avoid-speculation %s | llvm-dis > %t1
; RUN: llc %t1 -o - | FileCheck -check-prefixes=CHECK-COMMON,CHECK-DISABLE %s
;
; Source:
; unsigned long foo();
; void *test(void *p) {
; unsigned long ret = foo();
; if (ret <= 7)
; p += ret;
; return p;
; }
; Compilation flag:
; clang -target bpf -O2 -S -emit-llvm -Xclang -disable-llvm-passes test.c
; Function Attrs: nounwind
define dso_local i8* @test(i8* %p) #0 {
entry:
%p.addr = alloca i8*, align 8
%ret = alloca i64, align 8
store i8* %p, i8** %p.addr, align 8, !tbaa !2
%0 = bitcast i64* %ret to i8*
call void @llvm.lifetime.start.p0i8(i64 8, i8* %0) #3
%call = call i64 bitcast (i64 (...)* @foo to i64 ()*)()
store i64 %call, i64* %ret, align 8, !tbaa !6
%1 = load i64, i64* %ret, align 8, !tbaa !6
%cmp = icmp ule i64 %1, 7
br i1 %cmp, label %if.then, label %if.end
if.then: ; preds = %entry
%2 = load i64, i64* %ret, align 8, !tbaa !6
%3 = load i8*, i8** %p.addr, align 8, !tbaa !2
%add.ptr = getelementptr i8, i8* %3, i64 %2
store i8* %add.ptr, i8** %p.addr, align 8, !tbaa !2
br label %if.end
if.end: ; preds = %if.then, %entry
%4 = load i8*, i8** %p.addr, align 8, !tbaa !2
%5 = bitcast i64* %ret to i8*
call void @llvm.lifetime.end.p0i8(i64 8, i8* %5) #3
ret i8* %4
}
; CHECK-COMMON: [[REG6:r[0-9]+]] = r1
; CHECK-COMMON: call foo
; CHECK: if r0 > 7 goto [[LABEL:.*]]
; CHECK: [[REG6]] += r0
; CHECK: [[LABEL]]:
; CHECK: r0 = [[REG6]]
; CHECK-DISABLE: r0 = [[REG6]]
; CHECK-DISABLE: r0 += [[REG1:r[0-9]+]]
; CHECK-DISABLE: [[REG2:r[0-9]+]] = 8
; CHECK-DISABLE: if [[REG2]] > [[REG1]] goto [[LABEL:.*]]
; CHECK-DISABLE: r0 = [[REG6]]
; CHECK-DISABLE: [[LABEL]]:
; CHECK-COMMON: exit
; Function Attrs: argmemonly nounwind willreturn
declare void @llvm.lifetime.start.p0i8(i64 immarg, i8* nocapture) #1
declare dso_local i64 @foo(...) #2
; Function Attrs: argmemonly nounwind willreturn
declare void @llvm.lifetime.end.p0i8(i64 immarg, i8* nocapture) #1
attributes #0 = { nounwind "disable-tail-calls"="false" "frame-pointer"="all" "less-precise-fpmad"="false" "min-legal-vector-width"="0" "no-infs-fp-math"="false" "no-jump-tables"="false" "no-nans-fp-math"="false" "no-signed-zeros-fp-math"="false" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "unsafe-fp-math"="false" "use-soft-float"="false" }
attributes #1 = { argmemonly nounwind willreturn }
attributes #2 = { "disable-tail-calls"="false" "frame-pointer"="all" "less-precise-fpmad"="false" "no-infs-fp-math"="false" "no-nans-fp-math"="false" "no-signed-zeros-fp-math"="false" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "unsafe-fp-math"="false" "use-soft-float"="false" }
attributes #3 = { nounwind }
!llvm.module.flags = !{!0}
!llvm.ident = !{!1}
!0 = !{i32 1, !"wchar_size", i32 4}
!1 = !{!"clang version 12.0.0 (https://github.com/llvm/llvm-project.git ca9c5433a6c31e372092fcd8bfd0e4fddd7e8784)"}
!2 = !{!3, !3, i64 0}
!3 = !{!"any pointer", !4, i64 0}
!4 = !{!"omnipotent char", !5, i64 0}
!5 = !{!"Simple C/C++ TBAA"}
!6 = !{!7, !7, i64 0}
!7 = !{!"long", !4, i64 0}