| 1 | //===----------------------------------------------------------------------===// |
|---|
| 2 | // |
|---|
| 3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
|---|
| 4 | // See https://llvm.org/LICENSE.txt for license information. |
|---|
| 5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
|---|
| 6 | // |
|---|
| 7 | //===----------------------------------------------------------------------===// |
|---|
| 8 | |
|---|
| 9 | // Reports leaks after https://github.com/llvm/llvm-project/pull/66285 |
|---|
| 10 | // UNSUPPORTED: lsan |
|---|
| 11 | |
|---|
| 12 | // UNSUPPORTED: no-exceptions |
|---|
| 13 | |
|---|
| 14 | #include "cxxabi.h" |
|---|
| 15 | |
|---|
| 16 | #include <cassert> |
|---|
| 17 | #include <cstdlib> |
|---|
| 18 | #include <exception> |
|---|
| 19 | |
|---|
| 20 | void my_terminate () { exit ( status: 0 ); } |
|---|
| 21 | |
|---|
| 22 | // Wrapper routines |
|---|
| 23 | void *my_alloc2 ( size_t sz ) { |
|---|
| 24 | void *p = std::malloc ( size: sz ); |
|---|
| 25 | // std::printf ( "Allocated %ld bytes at %lx\n", sz, (unsigned long) p ); |
|---|
| 26 | return p; |
|---|
| 27 | } |
|---|
| 28 | |
|---|
| 29 | void my_dealloc2 ( void *p ) { |
|---|
| 30 | // std::printf ( "Freeing %lx\n", (unsigned long) p ); |
|---|
| 31 | std::free ( ptr: p ); |
|---|
| 32 | } |
|---|
| 33 | |
|---|
| 34 | void my_dealloc3 ( void *p, size_t ) { |
|---|
| 35 | // std::printf ( "Freeing %lx (size %ld)\n", (unsigned long) p, sz ); |
|---|
| 36 | std::free ( ptr: p ); |
|---|
| 37 | } |
|---|
| 38 | |
|---|
| 39 | void my_construct ( void *) { |
|---|
| 40 | // std::printf ( "Constructing %lx\n", (unsigned long) p ); |
|---|
| 41 | } |
|---|
| 42 | |
|---|
| 43 | void my_destruct ( void *) { |
|---|
| 44 | // std::printf ( "Destructing %lx\n", (unsigned long) p ); |
|---|
| 45 | } |
|---|
| 46 | |
|---|
| 47 | int gCounter; |
|---|
| 48 | void count_construct ( void * ) { ++gCounter; } |
|---|
| 49 | void count_destruct ( void * ) { --gCounter; } |
|---|
| 50 | |
|---|
| 51 | |
|---|
| 52 | int gConstructorCounter; |
|---|
| 53 | int gConstructorThrowTarget; |
|---|
| 54 | int gDestructorCounter; |
|---|
| 55 | int gDestructorThrowTarget; |
|---|
| 56 | void throw_construct ( void * ) { if ( gConstructorCounter == gConstructorThrowTarget ) throw 1; ++gConstructorCounter; } |
|---|
| 57 | void throw_destruct ( void * ) { if ( ++gDestructorCounter == gDestructorThrowTarget ) throw 2; } |
|---|
| 58 | |
|---|
| 59 | struct vec_on_stack { |
|---|
| 60 | void *storage; |
|---|
| 61 | vec_on_stack () : storage ( __cxxabiv1::__cxa_vec_new ( element_count: 10, element_size: 40, padding_size: 8, constructor: throw_construct, destructor: throw_destruct )) {} |
|---|
| 62 | ~vec_on_stack () { __cxxabiv1::__cxa_vec_delete ( array_address: storage, element_size: 40, padding_size: 8, destructor: throw_destruct ); } |
|---|
| 63 | }; |
|---|
| 64 | |
|---|
| 65 | |
|---|
| 66 | // Make sure the constructors and destructors are matched |
|---|
| 67 | void test_exception_in_destructor ( ) { |
|---|
| 68 | |
|---|
| 69 | // Try throwing from a destructor while unwinding the stack -- should abort |
|---|
| 70 | gConstructorCounter = gDestructorCounter = 0; |
|---|
| 71 | gConstructorThrowTarget = -1; |
|---|
| 72 | gDestructorThrowTarget = 5; |
|---|
| 73 | try { |
|---|
| 74 | vec_on_stack v; |
|---|
| 75 | throw 3; |
|---|
| 76 | } catch ( int i ) { |
|---|
| 77 | |
|---|
| 78 | } |
|---|
| 79 | |
|---|
| 80 | assert(false && "should never get here" ); |
|---|
| 81 | } |
|---|
| 82 | |
|---|
| 83 | |
|---|
| 84 | |
|---|
| 85 | int main () { |
|---|
| 86 | std::set_terminate ( my_terminate ); |
|---|
| 87 | test_exception_in_destructor (); |
|---|
| 88 | return 1; // we failed if we get here |
|---|
| 89 | } |
|---|
| 90 | |
|---|
