xref: /llvm-project/llvm/test/CodeGen/X86/xray-attribute-instrumentation.ll (revision e6bf48d11047e970cb24554a01b65b566d6b5d22)
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
5define 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.
40define 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)
45Test:
46  %cond = icmp eq i32 %i, 0
47  br i1 %cond, label %IsEqual, label %NotEqual
48IsEqual:
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)
54NotEqual:
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
89