Pirama Arumuga Nainar | 799172d | 2016-03-03 15:50:30 -0800 | [diff] [blame^] | 1 | // RUN: %clangxx_msan %s -O0 -fsanitize=memory -fsanitize-memory-use-after-dtor -o %t && MSAN_OPTIONS=poison_in_dtor=1 %run %t >%t.out 2>&1 |
| 2 | |
| 3 | // RUN: %clangxx_msan %s -O1 -fsanitize=memory -fsanitize-memory-use-after-dtor -o %t && MSAN_OPTIONS=poison_in_dtor=1 %run %t >%t.out 2>&1 |
| 4 | |
| 5 | // RUN: %clangxx_msan %s -O2 -fsanitize=memory -fsanitize-memory-use-after-dtor -o %t && MSAN_OPTIONS=poison_in_dtor=1 %run %t >%t.out 2>&1 |
| 6 | |
| 7 | #include <sanitizer/msan_interface.h> |
| 8 | #include <assert.h> |
| 9 | |
| 10 | template <class T> class Vector { |
| 11 | public: |
| 12 | int size; |
| 13 | ~Vector() { |
| 14 | assert(__msan_test_shadow(&this->size, sizeof(this->size)) == -1); |
| 15 | } |
| 16 | }; |
| 17 | |
| 18 | struct VirtualBase { |
| 19 | public: |
| 20 | Vector<int> virtual_v; |
| 21 | int virtual_a; |
| 22 | // Pointer to subclass member |
| 23 | int *intermediate_a_ptr; |
| 24 | |
| 25 | VirtualBase() { |
| 26 | virtual_v.size = 1; |
| 27 | virtual_a = 9; |
| 28 | } |
| 29 | void set_ptr(int *intermediate_a) { |
| 30 | this->intermediate_a_ptr = intermediate_a; |
| 31 | } |
| 32 | virtual ~VirtualBase() { |
| 33 | assert(__msan_test_shadow(&virtual_v, sizeof(virtual_v)) == -1); |
| 34 | assert(__msan_test_shadow(&virtual_a, sizeof(virtual_a)) == -1); |
| 35 | // Derived class member is poisoned |
| 36 | assert(__msan_test_shadow(intermediate_a_ptr, |
| 37 | sizeof(*intermediate_a_ptr)) != -1); |
| 38 | } |
| 39 | }; |
| 40 | |
| 41 | struct Intermediate : virtual public VirtualBase { |
| 42 | public: |
| 43 | int intermediate_a; |
| 44 | |
| 45 | Intermediate() { intermediate_a = 5; } |
| 46 | virtual ~Intermediate() { |
| 47 | assert(__msan_test_shadow(&this->intermediate_a, |
| 48 | sizeof(this->intermediate_a)) == -1); |
| 49 | // Members inherited from VirtualBase unpoisoned |
| 50 | assert(__msan_test_shadow(&virtual_v, sizeof(virtual_v)) == -1); |
| 51 | assert(__msan_test_shadow(&virtual_a, sizeof(virtual_a)) == -1); |
| 52 | assert(__msan_test_shadow(intermediate_a_ptr, |
| 53 | sizeof(*intermediate_a_ptr)) == -1); |
| 54 | } |
| 55 | }; |
| 56 | |
| 57 | struct Base { |
| 58 | int base_a; |
| 59 | Vector<int> base_v; |
| 60 | double base_b; |
| 61 | // Pointers to subclass members |
| 62 | int *derived_a_ptr; |
| 63 | Vector<int> *derived_v1_ptr; |
| 64 | Vector<int> *derived_v2_ptr; |
| 65 | double *derived_b_ptr; |
| 66 | double *derived_c_ptr; |
| 67 | |
| 68 | Base(int *derived_a, Vector<int> *derived_v1, Vector<int> *derived_v2, |
| 69 | double *derived_b, double *derived_c) { |
| 70 | base_a = 2; |
| 71 | base_v.size = 1; |
| 72 | base_b = 13.2324; |
| 73 | derived_a_ptr = derived_a; |
| 74 | derived_v1_ptr = derived_v1; |
| 75 | derived_v2_ptr = derived_v2; |
| 76 | derived_b_ptr = derived_b; |
| 77 | derived_c_ptr = derived_c; |
| 78 | } |
| 79 | virtual ~Base() { |
| 80 | assert(__msan_test_shadow(&base_a, sizeof(base_a)) == -1); |
| 81 | assert(__msan_test_shadow(&base_v, sizeof(base_v)) == -1); |
| 82 | assert(__msan_test_shadow(&base_b, sizeof(base_b)) == -1); |
| 83 | // Derived class members are poisoned |
| 84 | assert(__msan_test_shadow(derived_a_ptr, sizeof(*derived_a_ptr)) != -1); |
| 85 | assert(__msan_test_shadow(derived_v1_ptr, sizeof(*derived_v1_ptr)) != -1); |
| 86 | assert(__msan_test_shadow(derived_v2_ptr, sizeof(*derived_v2_ptr)) != -1); |
| 87 | assert(__msan_test_shadow(derived_b_ptr, sizeof(*derived_b_ptr)) != -1); |
| 88 | assert(__msan_test_shadow(derived_c_ptr, sizeof(*derived_c_ptr)) != -1); |
| 89 | } |
| 90 | }; |
| 91 | |
| 92 | struct Derived : public Base, public Intermediate { |
| 93 | int derived_a; |
| 94 | Vector<int> derived_v1; |
| 95 | Vector<int> derived_v2; |
| 96 | double derived_b; |
| 97 | double derived_c; |
| 98 | |
| 99 | Derived() |
| 100 | : Base(&derived_a, &derived_v1, &derived_v2, &derived_b, &derived_c) { |
| 101 | derived_a = 5; |
| 102 | derived_v1.size = 1; |
| 103 | derived_v2.size = 1; |
| 104 | derived_b = 7; |
| 105 | derived_c = 10; |
| 106 | } |
| 107 | ~Derived() { |
| 108 | assert(__msan_test_shadow(&derived_a, sizeof(derived_a)) == -1); |
| 109 | assert(__msan_test_shadow(&derived_v1, sizeof(derived_v1)) == -1); |
| 110 | assert(__msan_test_shadow(&derived_v2, sizeof(derived_v2)) == -1); |
| 111 | assert(__msan_test_shadow(&derived_b, sizeof(derived_b)) == -1); |
| 112 | assert(__msan_test_shadow(&derived_c, sizeof(derived_c)) == -1); |
| 113 | } |
| 114 | }; |
| 115 | |
| 116 | int main() { |
| 117 | Derived *d = new Derived(); |
| 118 | d->set_ptr(&d->intermediate_a); |
| 119 | |
| 120 | // Keep track of members of VirtualBase, since the virtual base table |
| 121 | // is inaccessible after destruction |
| 122 | Vector<int> *temp_virtual_v = &d->virtual_v; |
| 123 | int *temp_virtual_a = &d->virtual_a; |
| 124 | int **temp_intermediate_a_ptr = &d->intermediate_a_ptr; |
| 125 | |
| 126 | d->~Derived(); |
| 127 | assert(__msan_test_shadow(&d->derived_a, sizeof(d->derived_a)) != -1); |
| 128 | assert(__msan_test_shadow(&d->derived_v1, sizeof(d->derived_v1)) != -1); |
| 129 | assert(__msan_test_shadow(&d->derived_v2, sizeof(d->derived_v2)) != -1); |
| 130 | assert(__msan_test_shadow(&d->derived_b, sizeof(d->derived_b)) != -1); |
| 131 | assert(__msan_test_shadow(&d->derived_c, sizeof(d->derived_c)) != -1); |
| 132 | |
| 133 | // Inherited from base |
| 134 | assert(__msan_test_shadow(&d->base_a, sizeof(d->base_a)) != -1); |
| 135 | assert(__msan_test_shadow(&d->base_v, sizeof(d->base_v)) != -1); |
| 136 | assert(__msan_test_shadow(&d->base_b, sizeof(d->base_b)) != -1); |
| 137 | assert(__msan_test_shadow(&d->derived_a_ptr, sizeof(d->derived_a_ptr)) != -1); |
| 138 | assert(__msan_test_shadow(&d->derived_v1_ptr, sizeof(d->derived_v1_ptr)) != |
| 139 | -1); |
| 140 | assert(__msan_test_shadow(&d->derived_v2_ptr, sizeof(d->derived_v2_ptr)) != |
| 141 | -1); |
| 142 | assert(__msan_test_shadow(&d->derived_b_ptr, sizeof(d->derived_b_ptr)) != -1); |
| 143 | assert(__msan_test_shadow(&d->derived_c_ptr, sizeof(d->derived_c_ptr)) != -1); |
| 144 | |
| 145 | // Inherited from intermediate |
| 146 | assert(__msan_test_shadow(temp_virtual_v, sizeof(*temp_virtual_v)) != -1); |
| 147 | assert(__msan_test_shadow(temp_virtual_a, sizeof(*temp_virtual_a)) != -1); |
| 148 | assert(__msan_test_shadow(temp_intermediate_a_ptr, |
| 149 | sizeof(*temp_intermediate_a_ptr)) != -1); |
| 150 | |
| 151 | return 0; |
| 152 | } |