llvm/test/Transforms/KCFI/kcfi.ll

   1 ; RUN: opt -S -passes=kcfi %s | FileCheck --check-prefixes=CHECK,NOARM %s
   2 ; RUN: %if arm-registered-target %{ opt -S -passes=kcfi -mtriple=thumbv7m-unknown-linux-gnu %s | FileCheck --check-prefixes=CHECK,ARM %s %}
   3
   4 ; CHECK-LABEL: define void @f1(
   5 define void @f1(ptr noundef %x) {
   6   ; ARM:        %[[#FINT:]] = ptrtoint ptr %x to i32
   7   ; ARM-NEXT:   %[[#FAND:]] = and i32 %[[#FINT]], -2
   8   ; ARM-NEXT:   %[[#FPTR:]] = inttoptr i32 %[[#FAND]] to ptr
   9   ; ARM-NEXT:   %[[#GEPI:]] = getelementptr inbounds i32, ptr %[[#FPTR]], i32 -1
  10   ; NOARM:      %[[#GEPI:]] = getelementptr inbounds i32, ptr %x, i32 -1
  11   ; CHECK-NEXT: %[[#LOAD:]] = load i32, ptr %[[#GEPI]], align 4
  12   ; CHECK-NEXT: %[[#ICMP:]] = icmp ne i32 %[[#LOAD]], 12345678
  13   ; CHECK-NEXT: br i1 %[[#ICMP]], label %[[#TRAP:]], label %[[#CALL:]], !prof ![[#WEIGHTS:]]
  14   ; CHECK:      [[#TRAP]]:
  15   ; CHECK-NEXT: call void @llvm.debugtrap()
  16   ; CHECK-NEXT: br label %[[#CALL]]
  17   ; CHECK:      [[#CALL]]:
  18   ; CHECK-NEXT: call void %x()
  19   ; CHECK-NOT:  [ "kcfi"(i32 12345678) ]
  20   call void %x() [ "kcfi"(i32 12345678) ]
  21   ret void
  22 }
  23
  24 !llvm.module.flags = !{!0}
  25 !0 = !{i32 4, !"kcfi", i32 1}
  26 ; CHECK: ![[#WEIGHTS]] = !{!"branch_weights", i32 1, i32 1048575}