forked from OSchip/llvm-project
				
			
		
			
				
	
	
		
			55 lines
		
	
	
		
			1.1 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			55 lines
		
	
	
		
			1.1 KiB
		
	
	
	
		
			C++
		
	
	
	
| //===----------------------------------------------------------------------===//
 | |
| //
 | |
| //                     The LLVM Compiler Infrastructure
 | |
| //
 | |
| // This file is dual licensed under the MIT and the University of Illinois Open
 | |
| // Source Licenses. See LICENSE.TXT for details.
 | |
| //
 | |
| //===----------------------------------------------------------------------===//
 | |
| 
 | |
|  // test operator new
 | |
| 
 | |
| #include <new>
 | |
| #include <cstddef>
 | |
| #include <cassert>
 | |
| #include <limits>
 | |
| 
 | |
| int new_handler_called = 0;
 | |
| 
 | |
| void new_handler()
 | |
| {
 | |
|     ++new_handler_called;
 | |
|     std::set_new_handler(0);
 | |
| }
 | |
| 
 | |
| bool A_constructed = false;
 | |
| 
 | |
| struct A
 | |
| {
 | |
|     A() {A_constructed = true;}
 | |
|     ~A() {A_constructed = false;}
 | |
| };
 | |
| 
 | |
| int main()
 | |
| {
 | |
|     std::set_new_handler(new_handler);
 | |
|     try
 | |
|     {
 | |
|         void* vp = operator new (std::numeric_limits<std::size_t>::max());
 | |
|         assert(false);
 | |
|     }
 | |
|     catch (std::bad_alloc&)
 | |
|     {
 | |
|         assert(new_handler_called == 1);
 | |
|     }
 | |
|     catch (...)
 | |
|     {
 | |
|         assert(false);
 | |
|     }
 | |
|     A* ap = new A;
 | |
|     assert(ap);
 | |
|     assert(A_constructed);
 | |
|     delete ap;
 | |
|     assert(!A_constructed);
 | |
| }
 |