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 operator new
11 
12 // asan and msan will not call the new handler.
13 // UNSUPPORTED: asan, msan
14 
15 #include <new>
16 #include <cstddef>
17 #include <cassert>
18 #include <limits>
19 
20 int new_handler_called = 0;
21 
new_handler()22 void new_handler()
23 {
24     ++new_handler_called;
25     std::set_new_handler(0);
26 }
27 
28 bool A_constructed = false;
29 
30 struct A
31 {
AA32     A() {A_constructed = true;}
~AA33     ~A() {A_constructed = false;}
34 };
35 
main()36 int main()
37 {
38     std::set_new_handler(new_handler);
39     try
40     {
41         void* vp = operator new (std::numeric_limits<std::size_t>::max());
42         assert(false);
43     }
44     catch (std::bad_alloc&)
45     {
46         assert(new_handler_called == 1);
47     }
48     catch (...)
49     {
50         assert(false);
51     }
52     A* ap = new A;
53     assert(ap);
54     assert(A_constructed);
55     delete ap;
56     assert(!A_constructed);
57 }
58