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 39 A::A() {} 40 B::B() {} 41 C::C() {} 42 D::D() {} 43 44 void A::f() { 45 } 46 47 void B::g() { 48 } 49 50 void D::f() { 51 } 52 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" 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" 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" 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" 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"))) 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"))) 117 void df3(D *d) { 118 // CHECK-NOT: call i1 @llvm.bitset.test 119 d->f(); 120 } 121 122 D d; 123 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" 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