1 // RUN: %clang_cc1 -triple x86_64-unknown-linux -fsanitize=cfi-vcall -fsanitize-trap=cfi-vcall -emit-llvm -o - %s | FileCheck --check-prefix=CHECK --check-prefix=ITANIUM --check-prefix=NDIAG %s
2 // RUN: %clang_cc1 -triple x86_64-unknown-linux -fsanitize=cfi-vcall -emit-llvm -o - %s | FileCheck --check-prefix=CHECK --check-prefix=ITANIUM --check-prefix=DIAG --check-prefix=DIAG-ABORT %s
3 // RUN: %clang_cc1 -triple x86_64-unknown-linux -fsanitize=cfi-vcall -fsanitize-recover=cfi-vcall -emit-llvm -o - %s | FileCheck --check-prefix=CHECK --check-prefix=ITANIUM --check-prefix=DIAG --check-prefix=DIAG-RECOVER %s
4 // RUN: %clang_cc1 -triple x86_64-pc-windows-msvc -fsanitize=cfi-vcall -fsanitize-trap=cfi-vcall -emit-llvm -o - %s | FileCheck --check-prefix=CHECK --check-prefix=MS --check-prefix=NDIAG %s
5
6 // MS: @[[VTA:[0-9]*]] {{.*}} comdat($"\01??_7A@@6B@")
7 // MS: @[[VTB:[0-9]*]] {{.*}} comdat($"\01??_7B@@6B0@@")
8 // MS: @[[VTAinB:[0-9]*]] {{.*}} comdat($"\01??_7B@@6BA@@@")
9 // MS: @[[VTAinC:[0-9]*]] {{.*}} comdat($"\01??_7C@@6B@")
10 // MS: @[[VTBinD:[0-9]*]] {{.*}} comdat($"\01??_7D@?A@@6BB@@@")
11 // MS: @[[VTAinBinD:[0-9]*]] {{.*}} comdat($"\01??_7D@?A@@6BA@@@")
12 // MS: @[[VTFA:[0-9]*]] {{.*}} comdat($"\01??_7FA@?1??foo@@YAXXZ@6B@")
13
14 struct A {
15 A();
16 virtual void f();
17 };
18
19 struct B : virtual A {
20 B();
21 virtual void g();
22 virtual void h();
23 };
24
25 struct C : virtual A {
26 C();
27 };
28
29 namespace {
30
31 struct D : B, C {
32 D();
33 virtual void f();
34 virtual void h();
35 };
36
37 }
38
A()39 A::A() {}
B()40 B::B() {}
C()41 C::C() {}
D()42 D::D() {}
43
f()44 void A::f() {
45 }
46
g()47 void B::g() {
48 }
49
f()50 void D::f() {
51 }
52
h()53 void D::h() {
54 }
55
56 // DIAG: @[[SRC:.*]] = private unnamed_addr constant [{{.*}} x i8] c"{{.*}}cfi-vcall.cpp\00", align 1
57 // DIAG: @[[TYPE:.*]] = private unnamed_addr constant { i16, i16, [4 x i8] } { i16 -1, i16 0, [4 x i8] c"'A'\00" }
58 // DIAG: @[[BADTYPESTATIC:.*]] = private unnamed_addr global { { [{{.*}} x i8]*, i32, i32 }, { i16, i16, [4 x i8] }*, i8 } { { [{{.*}} x i8]*, i32, i32 } { [{{.*}} x i8]* @[[SRC]], i32 [[@LINE+21]], i32 3 }, { i16, i16, [4 x i8] }* @[[TYPE]], i8 0 }
59
60 // ITANIUM: define void @_Z2afP1A
61 // MS: define void @"\01?af@@YAXPEAUA@@@Z"
af(A * a)62 void af(A *a) {
63 // ITANIUM: [[P:%[^ ]*]] = call i1 @llvm.bitset.test(i8* [[VT:%[^ ]*]], metadata !"_ZTS1A")
64 // MS: [[P:%[^ ]*]] = call i1 @llvm.bitset.test(i8* [[VT:%[^ ]*]], metadata !"?AUA@@")
65 // CHECK-NEXT: br i1 [[P]], label %[[CONTBB:[^ ,]*]], label %[[TRAPBB:[^ ,]*]]
66 // CHECK-NEXT: {{^$}}
67
68 // CHECK: [[TRAPBB]]
69 // NDIAG-NEXT: call void @llvm.trap()
70 // NDIAG-NEXT: unreachable
71 // DIAG-NEXT: [[VTINT:%[^ ]*]] = ptrtoint i8* [[VT]] to i64
72 // DIAG-ABORT-NEXT: call void @__ubsan_handle_cfi_bad_type_abort(i8* bitcast ({{.*}} @[[BADTYPESTATIC]] to i8*), i64 [[VTINT]])
73 // DIAG-ABORT-NEXT: unreachable
74 // DIAG-RECOVER-NEXT: call void @__ubsan_handle_cfi_bad_type(i8* bitcast ({{.*}} @[[BADTYPESTATIC]] to i8*), i64 [[VTINT]])
75 // DIAG-RECOVER-NEXT: br label %[[CONTBB]]
76
77 // CHECK: [[CONTBB]]
78 // CHECK: call void %
79 a->f();
80 }
81
82 // ITANIUM: define internal void @_Z3df1PN12_GLOBAL__N_11DE
83 // MS: define internal void @"\01?df1@@YAXPEAUD@?A@@@Z"
df1(D * d)84 void df1(D *d) {
85 // ITANIUM: {{%[^ ]*}} = call i1 @llvm.bitset.test(i8* {{%[^ ]*}}, metadata ![[DTYPE:[0-9]+]])
86 // MS: {{%[^ ]*}} = call i1 @llvm.bitset.test(i8* {{%[^ ]*}}, metadata !"?AUA@@")
87 d->f();
88 }
89
90 // ITANIUM: define internal void @_Z3dg1PN12_GLOBAL__N_11DE
91 // MS: define internal void @"\01?dg1@@YAXPEAUD@?A@@@Z"
dg1(D * d)92 void dg1(D *d) {
93 // ITANIUM: {{%[^ ]*}} = call i1 @llvm.bitset.test(i8* {{%[^ ]*}}, metadata !"_ZTS1B")
94 // MS: {{%[^ ]*}} = call i1 @llvm.bitset.test(i8* {{%[^ ]*}}, metadata !"?AUB@@")
95 d->g();
96 }
97
98 // ITANIUM: define internal void @_Z3dh1PN12_GLOBAL__N_11DE
99 // MS: define internal void @"\01?dh1@@YAXPEAUD@?A@@@Z"
dh1(D * d)100 void dh1(D *d) {
101 // ITANIUM: {{%[^ ]*}} = call i1 @llvm.bitset.test(i8* {{%[^ ]*}}, metadata ![[DTYPE]])
102 // MS: {{%[^ ]*}} = call i1 @llvm.bitset.test(i8* {{%[^ ]*}}, metadata ![[DTYPE:[0-9]+]])
103 d->h();
104 }
105
106 // ITANIUM: define internal void @_Z3df2PN12_GLOBAL__N_11DE
107 // MS: define internal void @"\01?df2@@YAXPEAUD@?A@@@Z"
108 __attribute__((no_sanitize("cfi")))
df2(D * d)109 void df2(D *d) {
110 // CHECK-NOT: call i1 @llvm.bitset.test
111 d->f();
112 }
113
114 // ITANIUM: define internal void @_Z3df3PN12_GLOBAL__N_11DE
115 // MS: define internal void @"\01?df3@@YAXPEAUD@?A@@@Z"
116 __attribute__((no_sanitize("address"))) __attribute__((no_sanitize("cfi-vcall")))
df3(D * d)117 void df3(D *d) {
118 // CHECK-NOT: call i1 @llvm.bitset.test
119 d->f();
120 }
121
122 D d;
123
foo()124 void foo() {
125 df1(&d);
126 dg1(&d);
127 dh1(&d);
128 df2(&d);
129 df3(&d);
130
131 struct FA : A {
132 void f() {}
133 } fa;
134 af(&fa);
135 }
136
137 namespace test2 {
138
139 struct A {
140 virtual void m_fn1();
141 };
142 struct B {
143 virtual void m_fn2();
144 };
145 struct C : B, A {};
146 struct D : C {
147 void m_fn1();
148 };
149
150 // ITANIUM: define void @_ZN5test21fEPNS_1DE
151 // MS: define void @"\01?f@test2@@YAXPEAUD@1@@Z"
f(D * d)152 void f(D *d) {
153 // ITANIUM: {{%[^ ]*}} = call i1 @llvm.bitset.test(i8* {{%[^ ]*}}, metadata !"_ZTSN5test21DE")
154 // MS: {{%[^ ]*}} = call i1 @llvm.bitset.test(i8* {{%[^ ]*}}, metadata !"?AUA@test2@@")
155 d->m_fn1();
156 }
157
158 }
159
160 // Check for the expected number of elements (9 or 15 respectively).
161 // MS: !llvm.bitsets = !{[[X:[^,]*(,[^,]*){8}]]}
162 // ITANIUM: !llvm.bitsets = !{[[X:[^,]*(,[^,]*){14}]]}
163
164 // ITANIUM-DAG: !{!"_ZTS1A", [3 x i8*]* @_ZTV1A, i64 16}
165 // ITANIUM-DAG: !{!"_ZTS1A", [7 x i8*]* @_ZTCN12_GLOBAL__N_11DE0_1B, i64 32}
166 // ITANIUM-DAG: !{!"_ZTS1B", [7 x i8*]* @_ZTCN12_GLOBAL__N_11DE0_1B, i64 32}
167 // ITANIUM-DAG: !{!"_ZTS1A", [9 x i8*]* @_ZTCN12_GLOBAL__N_11DE8_1C, i64 64}
168 // ITANIUM-DAG: !{!"_ZTS1C", [9 x i8*]* @_ZTCN12_GLOBAL__N_11DE8_1C, i64 32}
169 // ITANIUM-DAG: !{!"_ZTS1A", [12 x i8*]* @_ZTVN12_GLOBAL__N_11DE, i64 32}
170 // ITANIUM-DAG: !{!"_ZTS1B", [12 x i8*]* @_ZTVN12_GLOBAL__N_11DE, i64 32}
171 // ITANIUM-DAG: !{!"_ZTS1C", [12 x i8*]* @_ZTVN12_GLOBAL__N_11DE, i64 88}
172 // ITANIUM-DAG: !{![[DTYPE]], [12 x i8*]* @_ZTVN12_GLOBAL__N_11DE, i64 32}
173 // ITANIUM-DAG: !{!"_ZTS1A", [7 x i8*]* @_ZTV1B, i64 32}
174 // ITANIUM-DAG: !{!"_ZTS1B", [7 x i8*]* @_ZTV1B, i64 32}
175 // ITANIUM-DAG: !{!"_ZTS1A", [5 x i8*]* @_ZTV1C, i64 32}
176 // ITANIUM-DAG: !{!"_ZTS1C", [5 x i8*]* @_ZTV1C, i64 32}
177 // ITANIUM-DAG: !{!"_ZTS1A", [3 x i8*]* @_ZTVZ3foovE2FA, i64 16}
178 // ITANIUM-DAG: !{!{{[0-9]+}}, [3 x i8*]* @_ZTVZ3foovE2FA, i64 16}
179
180 // MS-DAG: !{!"?AUA@@", [2 x i8*]* @[[VTA]], i64 8}
181 // MS-DAG: !{!"?AUB@@", [3 x i8*]* @[[VTB]], i64 8}
182 // MS-DAG: !{!"?AUA@@", [2 x i8*]* @[[VTAinB]], i64 8}
183 // MS-DAG: !{!"?AUA@@", [2 x i8*]* @[[VTAinC]], i64 8}
184 // MS-DAG: !{!"?AUB@@", [3 x i8*]* @[[VTBinD]], i64 8}
185 // MS-DAG: !{![[DTYPE]], [3 x i8*]* @[[VTBinD]], i64 8}
186 // MS-DAG: !{!"?AUA@@", [2 x i8*]* @[[VTAinBinD]], i64 8}
187 // MS-DAG: !{!"?AUA@@", [2 x i8*]* @[[VTFA]], i64 8}
188 // MS-DAG: !{!{{[0-9]+}}, [2 x i8*]* @[[VTFA]], i64 8}
189