llvm/test/CodeGen/X86/xray-attribute-instrumentation.ll

   1 ; RUN: llc -mtriple=x86_64-unknown-linux-gnu                       < %s | FileCheck %s --check-prefixes=CHECK,CHECK-LINUX
   2 ; RUN: llc -mtriple=x86_64-unknown-linux-gnu -relocation-model=pic < %s | FileCheck %s --check-prefixes=CHECK,CHECK-LINUX
   3 ; RUN: llc -mtriple=x86_64-darwin-unknown                          < %s | FileCheck %s --check-prefixes=CHECK,CHECK-MACOS
   4
   5 define i32 @foo() nounwind noinline uwtable "function-instrument"="xray-always" {
   6 ; CHECK:       .p2align 1
   7 ; CHECK-LABEL: Lxray_sled_0:
   8 ; CHECK:       .ascii "\353\t"
   9 ; CHECK-NEXT:  nopw 512(%rax,%rax)
  10   ret i32 0
  11 ; CHECK:       .p2align 1
  12 ; CHECK-LABEL: Lxray_sled_1:
  13 ; CHECK:       retq
  14 ; CHECK-NEXT:  nopw %cs:512(%rax,%rax)
  15 }
  16
  17 ; CHECK-LINUX-LABEL: .section xray_instr_map,"ao",@progbits,foo{{$}}
  18 ; CHECK-LINUX-LABEL: .Lxray_sleds_start0:
  19 ; CHECK-LINUX:         .quad .Lxray_sled_0
  20 ; CHECK-LINUX:         .quad .Lxray_sled_1
  21 ; CHECK-LINUX-LABEL: .Lxray_sleds_end0:
  22 ; CHECK-LINUX-LABEL: .section xray_fn_idx,"ao",@progbits,foo{{$}}
  23 ; CHECK-LINUX:       [[IDX:\.Lxray_fn_idx[0-9]+]]:
  24 ; CHECK-LINUX-NEXT:    .quad .Lxray_sleds_start0-[[IDX]]
  25 ; CHECK-LINUX-NEXT:    .quad 2
  26
  27 ; CHECK-MACOS-LABEL: .section __DATA,xray_instr_map,regular,live_support{{$}}
  28 ; CHECK-MACOS-LABEL: lxray_sleds_start0:
  29 ; CHECK-MACOS:         .quad Lxray_sled_0
  30 ; CHECK-MACOS:         .quad Lxray_sled_1
  31 ; CHECK-MACOS-LABEL: Lxray_sleds_end0:
  32 ; CHECK-MACOS-LABEL: .section __DATA,xray_fn_idx,regular,live_support{{$}}
  33 ; CHECK-MACOS:       [[IDX:lxray_fn_idx[0-9]+]]:
  34 ; CHECK-MACOS-NEXT:    .quad lxray_sleds_start0-[[IDX]]
  35 ; CHECK-MACOS-NEXT:    .quad 2
  36
  37
  38 ; We test multiple returns in a single function to make sure we're getting all
  39 ; of them with XRay instrumentation.
  40 define i32 @bar(i32 %i) nounwind noinline uwtable "function-instrument"="xray-always" {
  41 ; CHECK:       .p2align 1
  42 ; CHECK-LABEL: Lxray_sled_2:
  43 ; CHECK:       .ascii "\353\t"
  44 ; CHECK-NEXT:  nopw 512(%rax,%rax)
  45 Test:
  46   %cond = icmp eq i32 %i, 0
  47   br i1 %cond, label %IsEqual, label %NotEqual
  48 IsEqual:
  49   ret i32 0
  50 ; CHECK:       .p2align 1
  51 ; CHECK-LABEL: Lxray_sled_3:
  52 ; CHECK:       retq
  53 ; CHECK-NEXT:  nopw %cs:512(%rax,%rax)
  54 NotEqual:
  55   ret i32 1
  56 ; CHECK:       .p2align 1
  57 ; CHECK-LABEL: Lxray_sled_4:
  58 ; CHECK:       retq
  59 ; CHECK-NEXT:  nopw %cs:512(%rax,%rax)
  60 }
  61
  62 ; CHECK-LINUX-LABEL: .section xray_instr_map,"ao",@progbits,bar{{$}}
  63 ; CHECK-LINUX-LABEL: .Lxray_sleds_start1:
  64 ; CHECK-LINUX:       [[TMP:.Ltmp[0-9]+]]:
  65 ; CHECK-LINUX-NEXT:    .quad .Lxray_sled_2-[[TMP]]
  66 ; CHECK-LINUX:       [[TMP:.Ltmp[0-9]+]]:
  67 ; CHECK-LINUX-NEXT:    .quad .Lxray_sled_3-[[TMP]]
  68 ; CHECK-LINUX:       [[TMP:.Ltmp[0-9]+]]:
  69 ; CHECK-LINUX-NEXT:    .quad .Lxray_sled_4-[[TMP]]
  70 ; CHECK-LINUX-LABEL: .Lxray_sleds_end1:
  71 ; CHECK-LINUX-LABEL: .section xray_fn_idx,"ao",@progbits,bar{{$}}
  72 ; CHECK-LINUX:       [[IDX:\.Lxray_fn_idx[0-9]+]]:
  73 ; CHECK-LINUX-NEXT:    .quad .Lxray_sleds_start1-[[IDX]]
  74 ; CHECK-LINUX-NEXT:    .quad 3
  75
  76 ; CHECK-MACOS-LABEL: .section __DATA,xray_instr_map,regular,live_support{{$}}
  77 ; CHECK-MACOS-LABEL: lxray_sleds_start1:
  78 ; CHECK-MACOS:       [[TMP:Ltmp[0-9]+]]:
  79 ; CHECK-MACOS-NEXT:    .quad Lxray_sled_2-[[TMP]]
  80 ; CHECK-MACOS:       [[TMP:Ltmp[0-9]+]]:
  81 ; CHECK-MACOS-NEXT:    .quad Lxray_sled_3-[[TMP]]
  82 ; CHECK-MACOS:       [[TMP:Ltmp[0-9]+]]:
  83 ; CHECK-MACOS-NEXT:    .quad Lxray_sled_4-[[TMP]]
  84 ; CHECK-MACOS-LABEL: Lxray_sleds_end1:
  85 ; CHECK-MACOS-LABEL: .section __DATA,xray_fn_idx,regular,live_support{{$}}
  86 ; CHECK-MACOS:       [[IDX:lxray_fn_idx[0-9]+]]:
  87 ; CHECK-MACOS-NEXT:    .quad lxray_sleds_start1-[[IDX]]
  88 ; CHECK-MACOS-NEXT:    .quad 3