Clang Project

clang_source_code/test/CodeGenCXX/type-metadata.cpp
1// Tests for the cfi-vcall feature:
2// RUN: %clang_cc1 -flto -flto-unit -triple x86_64-unknown-linux -fvisibility hidden -fsanitize=cfi-vcall -fsanitize-trap=cfi-vcall -emit-llvm -o - %s | FileCheck --check-prefix=CFI --check-prefix=CFI-NVT --check-prefix=ITANIUM --check-prefix=TT-ITANIUM --check-prefix=NDIAG %s
3// RUN: %clang_cc1 -flto -flto-unit -triple x86_64-unknown-linux -fvisibility hidden -fsanitize=cfi-vcall -emit-llvm -o - %s | FileCheck --check-prefix=CFI --check-prefix=CFI-NVT --check-prefix=ITANIUM --check-prefix=TT-ITANIUM --check-prefix=ITANIUM-DIAG --check-prefix=DIAG --check-prefix=DIAG-ABORT %s
4// RUN: %clang_cc1 -flto -flto-unit -triple x86_64-unknown-linux -fvisibility hidden -fsanitize=cfi-vcall -fsanitize-recover=cfi-vcall -emit-llvm -o - %s | FileCheck --check-prefix=CFI --check-prefix=CFI-NVT --check-prefix=ITANIUM --check-prefix=TT-ITANIUM --check-prefix=ITANIUM-DIAG --check-prefix=DIAG --check-prefix=DIAG-RECOVER %s
5// RUN: %clang_cc1 -flto -flto-unit -triple x86_64-pc-windows-msvc -fsanitize=cfi-vcall -fsanitize-trap=cfi-vcall -emit-llvm -o - %s | FileCheck --check-prefix=CFI --check-prefix=CFI-NVT --check-prefix=MS --check-prefix=TT-MS --check-prefix=NDIAG %s
6
7// Tests for the whole-program-vtables feature:
8// RUN: %clang_cc1 -flto -flto-unit -triple x86_64-unknown-linux -fvisibility hidden -fwhole-program-vtables -emit-llvm -o - %s | FileCheck --check-prefix=VTABLE-OPT --check-prefix=ITANIUM --check-prefix=TT-ITANIUM %s
9// RUN: %clang_cc1 -flto -flto-unit -triple x86_64-pc-windows-msvc -fwhole-program-vtables -emit-llvm -o - %s | FileCheck --check-prefix=VTABLE-OPT --check-prefix=MS --check-prefix=TT-MS %s
10
11// Tests for cfi + whole-program-vtables:
12// RUN: %clang_cc1 -flto -flto-unit -triple x86_64-unknown-linux -fvisibility hidden -fsanitize=cfi-vcall -fsanitize-trap=cfi-vcall -fwhole-program-vtables -emit-llvm -o - %s | FileCheck --check-prefix=CFI --check-prefix=CFI-VT --check-prefix=ITANIUM --check-prefix=TC-ITANIUM %s
13// RUN: %clang_cc1 -flto -flto-unit -triple x86_64-pc-windows-msvc -fsanitize=cfi-vcall -fsanitize-trap=cfi-vcall -fwhole-program-vtables -emit-llvm -o - %s | FileCheck --check-prefix=CFI --check-prefix=CFI-VT --check-prefix=MS --check-prefix=TC-MS %s
14
15// ITANIUM: @_ZTV1A = {{[^!]*}}, !type [[A16:![0-9]+]]
16// ITANIUM-DIAG-SAME: !type [[ALL16:![0-9]+]]
17// ITANIUM-SAME: !type [[AF16:![0-9]+]]
18
19// ITANIUM: @_ZTV1B = {{[^!]*}}, !type [[A32:![0-9]+]]
20// ITANIUM-DIAG-SAME: !type [[ALL32:![0-9]+]]
21// ITANIUM-SAME: !type [[AF32:![0-9]+]]
22// ITANIUM-SAME: !type [[AF40:![0-9]+]]
23// ITANIUM-SAME: !type [[AF48:![0-9]+]]
24// ITANIUM-SAME: !type [[B32:![0-9]+]]
25// ITANIUM-DIAG-SAME: !type [[ALL32]]
26// ITANIUM-SAME: !type [[BF32:![0-9]+]]
27// ITANIUM-SAME: !type [[BF40:![0-9]+]]
28// ITANIUM-SAME: !type [[BF48:![0-9]+]]
29
30// ITANIUM: @_ZTV1C = {{[^!]*}}, !type [[A32]]
31// ITANIUM-DIAG-SAME: !type [[ALL32]]
32// ITANIUM-SAME: !type [[AF32]]
33// ITANIUM-SAME: !type [[C32:![0-9]+]]
34// ITANIUM-DIAG-SAME: !type [[ALL32]]
35// ITANIUM-SAME: !type [[CF32:![0-9]+]]
36
37// DIAG: @[[SRC:.*]] = private unnamed_addr constant [{{.*}} x i8] c"{{.*}}type-metadata.cpp\00", align 1
38// DIAG: @[[TYPE:.*]] = private unnamed_addr constant { i16, i16, [4 x i8] } { i16 -1, i16 0, [4 x i8] c"'A'\00" }
39// DIAG: @[[BADTYPESTATIC:.*]] = private unnamed_addr global { i8, { [{{.*}} x i8]*, i32, i32 }, { i16, i16, [4 x i8] }* } { i8 0, { [{{.*}} x i8]*, i32, i32 } { [{{.*}} x i8]* @[[SRC]], i32 123, i32 3 }, { i16, i16, [4 x i8] }* @[[TYPE]] }
40
41// ITANIUM: @_ZTVN12_GLOBAL__N_11DE = {{[^!]*}}, !type [[A32]]
42// ITANIUM-DIAG-SAME: !type [[ALL32]]
43// ITANIUM-SAME: !type [[AF32]]
44// ITANIUM-SAME: !type [[AF40]]
45// ITANIUM-SAME: !type [[AF48]]
46// ITANIUM-SAME: !type [[B32]]
47// ITANIUM-DIAG-SAME: !type [[ALL32]]
48// ITANIUM-SAME: !type [[BF32]]
49// ITANIUM-SAME: !type [[BF40]]
50// ITANIUM-SAME: !type [[BF48]]
51// ITANIUM-SAME: !type [[C88:![0-9]+]]
52// ITANIUM-DIAG-SAME: !type [[ALL88:![0-9]+]]
53// ITANIUM-SAME: !type [[CF32]]
54// ITANIUM-SAME: !type [[CF40:![0-9]+]]
55// ITANIUM-SAME: !type [[CF48:![0-9]+]]
56// ITANIUM-SAME: !type [[D32:![0-9]+]]
57// ITANIUM-DIAG-SAME: !type [[ALL32]]
58// ITANIUM-SAME: !type [[DF32:![0-9]+]]
59// ITANIUM-SAME: !type [[DF40:![0-9]+]]
60// ITANIUM-SAME: !type [[DF48:![0-9]+]]
61
62// ITANIUM: @_ZTCN12_GLOBAL__N_11DE0_1B = {{[^!]*}}, !type [[A32]]
63// ITANIUM-DIAG-SAME: !type [[ALL32]]
64// ITANIUM-SAME: !type [[B32]]
65// ITANIUM-DIAG-SAME: !type [[ALL32]]
66
67// ITANIUM: @_ZTCN12_GLOBAL__N_11DE8_1C = {{[^!]*}}, !type [[A64:![0-9]+]]
68// ITANIUM-DIAG-SAME: !type [[ALL64:![0-9]+]]
69// ITANIUM-SAME: !type [[AF64:![0-9]+]]
70// ITANIUM-SAME: !type [[C32]]
71// ITANIUM-DIAG-SAME: !type [[ALL32]]
72// ITANIUM-SAME: !type [[CF64:![0-9]+]]
73
74// ITANIUM: @_ZTVZ3foovE2FA = {{[^!]*}}, !type [[A16]]
75// ITANIUM-DIAG-SAME: !type [[ALL16]]
76// ITANIUM-SAME: !type [[AF16]]
77// ITANIUM-SAME: !type [[FA16:![0-9]+]]
78// ITANIUM-DIAG-SAME: !type [[ALL16]]
79// ITANIUM-SAME: !type [[FAF16:![0-9]+]]
80
81// MS: comdat($"??_7A@@6B@"), !type [[A8:![0-9]+]]
82// MS: comdat($"??_7B@@6B0@@"), !type [[B8:![0-9]+]]
83// MS: comdat($"??_7B@@6BA@@@"), !type [[A8]]
84// MS: comdat($"??_7C@@6B@"), !type [[A8]]
85// MS: comdat($"??_7D@?A0x{{[^@]*}}@@6BB@@@"), !type [[B8]], !type [[D8:![0-9]+]]
86// MS: comdat($"??_7D@?A0x{{[^@]*}}@@6BA@@@"), !type [[A8]]
87// MS: comdat($"??_7FA@?1??foo@@YAXXZ@6B@"), !type [[A8]], !type [[FA8:![0-9]+]]
88
89struct A {
90  A();
91  virtual void f();
92};
93
94struct B : virtual A {
95  B();
96  virtual void g();
97  virtual void h();
98};
99
100struct C : virtual A {
101  C();
102};
103
104namespace {
105
106struct D : B, C {
107  D();
108  virtual void f();
109  virtual void h();
110};
111
112}
113
114A::A() {}
115B::B() {}
116C::C() {}
117D::D() {}
118
119void A::f() {
120}
121
122void B::g() {
123}
124
125void D::f() {
126}
127
128void D::h() {
129}
130
131// ITANIUM: define hidden void @_Z2afP1A
132// MS: define dso_local void @"?af@@YAXPEAUA@@@Z"
133void af(A *a) {
134  // TT-ITANIUM: [[P:%[^ ]*]] = call i1 @llvm.type.test(i8* [[VT:%[^ ]*]], metadata !"_ZTS1A")
135  // TT-MS: [[P:%[^ ]*]] = call i1 @llvm.type.test(i8* [[VT:%[^ ]*]], metadata !"?AUA@@")
136  // TC-ITANIUM: [[PAIR:%[^ ]*]] = call { i8*, i1 } @llvm.type.checked.load(i8* {{%[^ ]*}}, i32 0, metadata !"_ZTS1A")
137  // TC-MS: [[PAIR:%[^ ]*]] = call { i8*, i1 } @llvm.type.checked.load(i8* {{%[^ ]*}}, i32 0, metadata !"?AUA@@")
138  // CFI-VT: [[P:%[^ ]*]] = extractvalue { i8*, i1 } [[PAIR]], 1
139  // DIAG-NEXT: [[VTVALID0:%[^ ]*]] = call i1 @llvm.type.test(i8* [[VT]], metadata !"all-vtables")
140  // VTABLE-OPT: call void @llvm.assume(i1 [[P]])
141  // CFI-NEXT: br i1 [[P]], label %[[CONTBB:[^ ,]*]], label %[[TRAPBB:[^ ,]*]]
142  // CFI-NEXT: {{^$}}
143
144  // CFI: [[TRAPBB]]
145  // NDIAG-NEXT: call void @llvm.trap()
146  // NDIAG-NEXT: unreachable
147  // DIAG-NEXT: [[VTINT:%[^ ]*]] = ptrtoint i8* [[VT]] to i64
148  // DIAG-NEXT: [[VTVALID:%[^ ]*]] = zext i1 [[VTVALID0]] to i64
149  // DIAG-ABORT-NEXT: call void @__ubsan_handle_cfi_check_fail_abort(i8* getelementptr inbounds ({{.*}} @[[BADTYPESTATIC]], i32 0, i32 0), i64 [[VTINT]], i64 [[VTVALID]])
150  // DIAG-ABORT-NEXT: unreachable
151  // DIAG-RECOVER-NEXT: call void @__ubsan_handle_cfi_check_fail(i8* getelementptr inbounds ({{.*}} @[[BADTYPESTATIC]], i32 0, i32 0), i64 [[VTINT]], i64 [[VTVALID]])
152  // DIAG-RECOVER-NEXT: br label %[[CONTBB]]
153
154  // CFI: [[CONTBB]]
155  // CFI-NVT: [[PTR:%[^ ]*]] = load
156  // CFI-VT: [[PTRI8:%[^ ]*]] = extractvalue { i8*, i1 } [[PAIR]], 0
157  // CFI-VT: [[PTR:%[^ ]*]] = bitcast i8* [[PTRI8]] to
158  // CFI: call void [[PTR]]
159#line 123
160  a->f();
161}
162
163// ITANIUM: define internal void @_Z3df1PN12_GLOBAL__N_11DE
164// MS: define internal void @"?df1@@YAXPEAUD@?A0x{{[^@]*}}@@@Z"
165void df1(D *d) {
166  // TT-ITANIUM: {{%[^ ]*}} = call i1 @llvm.type.test(i8* {{%[^ ]*}}, metadata ![[DTYPE:[0-9]+]])
167  // TT-MS: {{%[^ ]*}} = call i1 @llvm.type.test(i8* {{%[^ ]*}}, metadata !"?AUA@@")
168  // TC-ITANIUM: {{%[^ ]*}} = call { i8*, i1 } @llvm.type.checked.load(i8* {{%[^ ]*}}, i32 0, metadata ![[DTYPE:[0-9]+]])
169  // TC-MS: {{%[^ ]*}} = call { i8*, i1 } @llvm.type.checked.load(i8* {{%[^ ]*}}, i32 0, metadata !"?AUA@@")
170  d->f();
171}
172
173// ITANIUM: define internal void @_Z3dg1PN12_GLOBAL__N_11DE
174// MS: define internal void @"?dg1@@YAXPEAUD@?A0x{{[^@]*}}@@@Z"
175void dg1(D *d) {
176  // TT-ITANIUM: {{%[^ ]*}} = call i1 @llvm.type.test(i8* {{%[^ ]*}}, metadata !"_ZTS1B")
177  // TT-MS: {{%[^ ]*}} = call i1 @llvm.type.test(i8* {{%[^ ]*}}, metadata !"?AUB@@")
178  // TC-ITANIUM: {{%[^ ]*}} = call { i8*, i1 } @llvm.type.checked.load(i8* {{%[^ ]*}}, i32 8, metadata !"_ZTS1B")
179  // TC-MS: {{%[^ ]*}} = call { i8*, i1 } @llvm.type.checked.load(i8* {{%[^ ]*}}, i32 0, metadata !"?AUB@@")
180  d->g();
181}
182
183// ITANIUM: define internal void @_Z3dh1PN12_GLOBAL__N_11DE
184// MS: define internal void @"?dh1@@YAXPEAUD@?A0x{{[^@]*}}@@@Z"
185void dh1(D *d) {
186  // TT-ITANIUM: {{%[^ ]*}} = call i1 @llvm.type.test(i8* {{%[^ ]*}}, metadata ![[DTYPE]])
187  // TT-MS: {{%[^ ]*}} = call i1 @llvm.type.test(i8* {{%[^ ]*}}, metadata ![[DTYPE:[0-9]+]])
188  // TC-ITANIUM: {{%[^ ]*}} = call { i8*, i1 } @llvm.type.checked.load(i8* {{%[^ ]*}}, i32 16, metadata ![[DTYPE]])
189  // TC-MS: {{%[^ ]*}} = call { i8*, i1 } @llvm.type.checked.load(i8* {{%[^ ]*}}, i32 8, metadata ![[DTYPE:[0-9]+]])
190  d->h();
191}
192
193// ITANIUM: define internal void @_Z3df2PN12_GLOBAL__N_11DE
194// MS: define internal void @"?df2@@YAXPEAUD@?A0x{{[^@]*}}@@@Z"
195__attribute__((no_sanitize("cfi")))
196void df2(D *d) {
197  // CFI-NVT-NOT: call i1 @llvm.type.test
198  // CFI-VT: [[P:%[^ ]*]] = call i1 @llvm.type.test
199  // CFI-VT: call void @llvm.assume(i1 [[P]])
200  d->f();
201}
202
203// ITANIUM: define internal void @_Z3df3PN12_GLOBAL__N_11DE
204// MS: define internal void @"?df3@@YAXPEAUD@?A0x{{[^@]*}}@@@Z"
205__attribute__((no_sanitize("address"))) __attribute__((no_sanitize("cfi-vcall")))
206void df3(D *d) {
207  // CFI-NVT-NOT: call i1 @llvm.type.test
208  // CFI-VT: [[P:%[^ ]*]] = call i1 @llvm.type.test
209  // CFI-VT: call void @llvm.assume(i1 [[P]])
210  d->f();
211}
212
213D d;
214
215void foo() {
216  df1(&d);
217  dg1(&d);
218  dh1(&d);
219  df2(&d);
220  df3(&d);
221
222  struct FA : A {
223    void f() {}
224  } fa;
225  af(&fa);
226}
227
228namespace test2 {
229
230struct A {
231  virtual void m_fn1();
232};
233struct B {
234  virtual void m_fn2();
235};
236struct C : B, A {};
237struct D : C {
238  void m_fn1();
239};
240
241// ITANIUM: define hidden void @_ZN5test21fEPNS_1DE
242// MS: define dso_local void @"?f@test2@@YAXPEAUD@1@@Z"
243void f(D *d) {
244  // TT-ITANIUM: {{%[^ ]*}} = call i1 @llvm.type.test(i8* {{%[^ ]*}}, metadata !"_ZTSN5test21DE")
245  // TT-MS: {{%[^ ]*}} = call i1 @llvm.type.test(i8* {{%[^ ]*}}, metadata !"?AUA@test2@@")
246  // TC-ITANIUM: {{%[^ ]*}} = call { i8*, i1 } @llvm.type.checked.load(i8* {{%[^ ]*}}, i32 8, metadata !"_ZTSN5test21DE")
247  // TC-MS: {{%[^ ]*}} = call { i8*, i1 } @llvm.type.checked.load(i8* {{%[^ ]*}}, i32 0, metadata !"?AUA@test2@@")
248  d->m_fn1();
249}
250
251}
252
253// ITANIUM: [[A16]] = !{i64 16, !"_ZTS1A"}
254// ITANIUM-DIAG: [[ALL16]] = !{i64 16, !"all-vtables"}
255// ITANIUM: [[AF16]] = !{i64 16, !"_ZTSM1AFvvE.virtual"}
256// ITANIUM: [[A32]] = !{i64 32, !"_ZTS1A"}
257// ITANIUM-DIAG: [[ALL32]] = !{i64 32, !"all-vtables"}
258// ITANIUM: [[AF32]] = !{i64 32, !"_ZTSM1AFvvE.virtual"}
259// ITANIUM: [[AF40]] = !{i64 40, !"_ZTSM1AFvvE.virtual"}
260// ITANIUM: [[AF48]] = !{i64 48, !"_ZTSM1AFvvE.virtual"}
261// ITANIUM: [[B32]] = !{i64 32, !"_ZTS1B"}
262// ITANIUM: [[BF32]] = !{i64 32, !"_ZTSM1BFvvE.virtual"}
263// ITANIUM: [[BF40]] = !{i64 40, !"_ZTSM1BFvvE.virtual"}
264// ITANIUM: [[BF48]] = !{i64 48, !"_ZTSM1BFvvE.virtual"}
265// ITANIUM: [[C32]] = !{i64 32, !"_ZTS1C"}
266// ITANIUM: [[CF32]] = !{i64 32, !"_ZTSM1CFvvE.virtual"}
267// ITANIUM: [[C88]] = !{i64 88, !"_ZTS1C"}
268// ITANIUM-DIAG: [[ALL88]] = !{i64 88, !"all-vtables"}
269// ITANIUM: [[CF40]] = !{i64 40, !"_ZTSM1CFvvE.virtual"}
270// ITANIUM: [[CF48]] = !{i64 48, !"_ZTSM1CFvvE.virtual"}
271// ITANIUM: [[D32]] = !{i64 32, [[D_ID:![0-9]+]]}
272// ITANIUM: [[D_ID]] = distinct !{}
273// ITANIUM: [[DF32]] = !{i64 32, [[DF_ID:![0-9]+]]}
274// ITANIUM: [[DF_ID]] = distinct !{}
275// ITANIUM: [[DF40]] = !{i64 40, [[DF_ID]]}
276// ITANIUM: [[DF48]] = !{i64 48, [[DF_ID]]}
277// ITANIUM: [[A64]] = !{i64 64, !"_ZTS1A"}
278// ITANIUM-DIAG: [[ALL64]] = !{i64 64, !"all-vtables"}
279// ITANIUM: [[AF64]] = !{i64 64, !"_ZTSM1AFvvE.virtual"}
280// ITANIUM: [[CF64]] = !{i64 64, !"_ZTSM1CFvvE.virtual"}
281// ITANIUM: [[FA16]] = !{i64 16, [[FA_ID:![0-9]+]]}
282// ITANIUM: [[FA_ID]] = distinct !{}
283// ITANIUM: [[FAF16]] = !{i64 16, [[FAF_ID:![0-9]+]]}
284// ITANIUM: [[FAF_ID]] = distinct !{}
285
286// MS: [[A8]] = !{i64 8, !"?AUA@@"}
287// MS: [[B8]] = !{i64 8, !"?AUB@@"}
288// MS: [[D8]] = !{i64 8, [[D_ID:![0-9]+]]}
289// MS: [[D_ID]] = distinct !{}
290// MS: [[FA8]] = !{i64 8, [[FA_ID:![0-9]+]]}
291// MS: [[FA_ID]] = distinct !{}
292