1 // RUN: %clangxx_cfi -o %t1 %s 2 // RUN: %expect_crash %t1 2>&1 | FileCheck --check-prefix=CFI %s 3 // RUN: %expect_crash %t1 x 2>&1 | FileCheck --check-prefix=CFI %s 4 5 // RUN: %clangxx_cfi -DB32 -o %t2 %s 6 // RUN: %expect_crash %t2 2>&1 | FileCheck --check-prefix=CFI %s 7 // RUN: %expect_crash %t2 x 2>&1 | FileCheck --check-prefix=CFI %s 8 9 // RUN: %clangxx_cfi -DB64 -o %t3 %s 10 // RUN: %expect_crash %t3 2>&1 | FileCheck --check-prefix=CFI %s 11 // RUN: %expect_crash %t3 x 2>&1 | FileCheck --check-prefix=CFI %s 12 13 // RUN: %clangxx_cfi -DBM -o %t4 %s 14 // RUN: %expect_crash %t4 2>&1 | FileCheck --check-prefix=CFI %s 15 // RUN: %expect_crash %t4 x 2>&1 | FileCheck --check-prefix=CFI %s 16 17 // RUN: %clangxx -o %t5 %s 18 // RUN: %t5 2>&1 | FileCheck --check-prefix=NCFI %s 19 // RUN: %t5 x 2>&1 | FileCheck --check-prefix=NCFI %s 20 21 // RUN: %clangxx_cfi_diag -o %t6 %s 22 // RUN: %t6 2>&1 | FileCheck --check-prefix=CFI-DIAG2 %s 23 // RUN: %t6 x 2>&1 | FileCheck --check-prefix=CFI-DIAG1 %s 24 25 // Tests that the CFI mechanism is sensitive to multiple inheritance and only 26 // permits calls via virtual tables for the correct base class. 27 28 // REQUIRES: cxxabi 29 30 #include <stdio.h> 31 #include "utils.h" 32 33 struct A { 34 virtual void f() = 0; 35 }; 36 37 struct B { 38 virtual void g() = 0; 39 }; 40 41 struct C : A, B { 42 virtual void f(), g(); 43 }; 44 45 void C::f() {} 46 void C::g() {} 47 48 int main(int argc, char **argv) { 49 create_derivers<A>(); 50 create_derivers<B>(); 51 52 C *c = new C; 53 break_optimization(c); 54 55 // CFI: 1 56 // NCFI: 1 57 fprintf(stderr, "1\n"); 58 59 if (argc > 1) { 60 A *a = c; 61 // CFI-DIAG1: runtime error: control flow integrity check for type 'B' failed during cast to unrelated type 62 // CFI-DIAG1-NEXT: note: vtable is of type '{{(struct )?}}C' 63 ((B *)a)->g(); // UB here 64 } else { 65 // CFI-DIAG2: runtime error: control flow integrity check for type 'A' failed during cast to unrelated type 66 // CFI-DIAG2-NEXT: note: vtable is of type '{{(struct )?}}C' 67 B *b = c; 68 ((A *)b)->f(); // UB here 69 } 70 71 // CFI-NOT: {{^2$}} 72 // NCFI: {{^2$}} 73 fprintf(stderr, "2\n"); 74 } 75
