1 //===----------------------------------------------------------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is dual licensed under the MIT and the University of Illinois Open 6 // Source Licenses. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 // test sized operator delete by replacing unsized operator delete. 11 12 // UNSUPPORTED: asan, msan 13 14 #include <new> 15 #include <cstddef> 16 #include <cstdlib> 17 #include <cassert> 18 #include <limits> 19 20 int delete_called = 0; 21 int delete_nothrow_called = 0; 22 23 void operator delete(void* p) throw() 24 { 25 ++delete_called; 26 delete_nothrow_called; 27 std::free(p); 28 } 29 30 void operator delete(void* p, const std::nothrow_t&) throw() 31 { 32 ++delete_nothrow_called; 33 std::free(p); 34 } 35 36 int new_handler_called = 0; 37 38 void new_handler() 39 { 40 ++new_handler_called; 41 std::set_new_handler(0); 42 } 43 44 bool A_constructed = false; 45 46 struct A 47 { 48 A() {A_constructed = true;} 49 ~A() {A_constructed = false;} 50 }; 51 52 int main() 53 { 54 std::set_new_handler(new_handler); 55 try 56 { 57 void* vp = operator new (std::numeric_limits<std::size_t>::max()); 58 assert(false); 59 } 60 catch (std::bad_alloc&) 61 { 62 assert(new_handler_called == 1); 63 } 64 catch (...) 65 { 66 assert(false); 67 } 68 A* ap = new A; 69 assert(ap); 70 assert(A_constructed); 71 assert(!delete_called); 72 assert(!delete_nothrow_called); 73 delete ap; 74 assert(!A_constructed); 75 assert(delete_called == 1); 76 assert(!delete_nothrow_called); 77 } 78