1; RUN: llc -mtriple=i386 %s -o - | FileCheck --check-prefixes=CHECK,NOFSECT,32 %s
2; RUN: llc -mtriple=x86_64 %s -o - | FileCheck --check-prefixes=CHECK,NOFSECT,64 %s
3; RUN: llc -mtriple=x86_64 -function-sections %s -o - | FileCheck --check-prefixes=CHECK,FSECT,64 %s
4
5define void @f0() "patchable-function-entry"="0" {
6; CHECK-LABEL: f0:
7; CHECK-NEXT: .Lfunc_begin0:
8; CHECK-NOT:   nop
9; CHECK:       ret
10; CHECK-NOT:   .section __patchable_function_entries
11  ret void
12}
13
14define void @f1() "patchable-function-entry"="1" {
15; CHECK-LABEL: f1:
16; CHECK-NEXT: .Lfunc_begin1:
17; CHECK:       nop
18; CHECK-NEXT:  ret
19; CHECK:       .section __patchable_function_entries,"awo",@progbits,f1,unique,0
20; 32:          .p2align 2
21; 32-NEXT:     .long .Lfunc_begin1
22; 64:          .p2align 3
23; 64-NEXT:     .quad .Lfunc_begin1
24  ret void
25}
26
27;; Without -function-sections, f2 is in the same text section as f1.
28;; They share the __patchable_function_entries section.
29;; With -function-sections, f1 and f2 are in different text sections.
30;; Use separate __patchable_function_entries.
31define void @f2() "patchable-function-entry"="2" {
32; CHECK-LABEL: f2:
33; CHECK-NEXT: .Lfunc_begin2:
34; 32-COUNT-2:  nop
35; 64:          xchgw %ax, %ax
36; CHECK-NEXT:  ret
37; NOFSECT:     .section __patchable_function_entries,"awo",@progbits,f1,unique,0
38; FSECT:       .section __patchable_function_entries,"awo",@progbits,f2,unique,1
39; 32:          .p2align 2
40; 32-NEXT:     .long .Lfunc_begin2
41; 64:          .p2align 3
42; 64-NEXT:     .quad .Lfunc_begin2
43  ret void
44}
45
46$f3 = comdat any
47define void @f3() "patchable-function-entry"="3" comdat {
48; CHECK-LABEL: f3:
49; CHECK-NEXT: .Lfunc_begin3:
50; 32-COUNT-3:  nop
51; 64:          nopl (%rax)
52; CHECK:       ret
53; NOFSECT:     .section __patchable_function_entries,"aGwo",@progbits,f3,comdat,f3,unique,1
54; FSECT:       .section __patchable_function_entries,"aGwo",@progbits,f3,comdat,f3,unique,2
55; 32:          .p2align 2
56; 32-NEXT:     .long .Lfunc_begin3
57; 64:          .p2align 3
58; 64-NEXT:     .quad .Lfunc_begin3
59  ret void
60}
61
62$f5 = comdat any
63define void @f5() "patchable-function-entry"="5" comdat {
64; CHECK-LABEL: f5:
65; CHECK-NEXT: .Lfunc_begin4:
66; 32-COUNT-5:  nop
67; 64:          nopl 8(%rax,%rax)
68; CHECK-NEXT:  ret
69; NOFSECT      .section __patchable_function_entries,"aGwo",@progbits,f5,comdat,f5,unique,2
70; FSECT:       .section __patchable_function_entries,"aGwo",@progbits,f5,comdat,f5,unique,3
71; 32:          .p2align 2
72; 32-NEXT:     .long .Lfunc_begin4
73; 64:          .p2align 3
74; 64-NEXT:     .quad .Lfunc_begin4
75  ret void
76}
77
78;; -fpatchable-function-entry=3,2
79;; "patchable-function-prefix" emits data before the function entry label.
80;; We emit 1-byte NOPs before the function entry, so that with a partial patch,
81;; the remaining instructions do not need to be modified.
82define void @f3_2() "patchable-function-entry"="1" "patchable-function-prefix"="2" {
83; CHECK-LABEL: .type f3_2,@function
84; CHECK-NEXT: .Ltmp0: # @f3_2
85; CHECK-NEXT:  nop
86; CHECK-NEXT:  nop
87; CHECK-NEXT: f3_2:
88; CHECK:      # %bb.0:
89; CHECK-NEXT:  nop
90; CHECK-NEXT:  ret
91;; .size does not include the prefix.
92; CHECK:      .Lfunc_end5:
93; CHECK-NEXT: .size f3_2, .Lfunc_end5-f3_2
94; NOFSECT     .section __patchable_function_entries,"awo",@progbits,f0,unique,0
95; FSECT:      .section __patchable_function_entries,"awo",@progbits,f3_2,unique,4
96; 32:         .p2align 2
97; 32-NEXT:    .long .Ltmp0
98; 64:         .p2align 3
99; 64-NEXT:    .quad .Ltmp0
100  %frame = alloca i8, i32 16
101  ret void
102}
103