1bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant//===----------------------------------------------------------------------===//
2bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant//
3f5256e16dfc425c1d466f6308d4026d529ce9e0bHoward Hinnant//                     The LLVM Compiler Infrastructure
4bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant//
5b64f8b07c104c6cc986570ac8ee0ed16a9f23976Howard Hinnant// This file is dual licensed under the MIT and the University of Illinois Open
6b64f8b07c104c6cc986570ac8ee0ed16a9f23976Howard Hinnant// Source Licenses. See LICENSE.TXT for details.
7bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant//
8bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant//===----------------------------------------------------------------------===//
9cf6dcc35e14a3d2dafe67cf76a200b2e1a671e7dHoward Hinnant
10bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant // test operator new replacement
11bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant
12bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <new>
13bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <cstddef>
14bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <cstdlib>
15bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <cassert>
16bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <limits>
17bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant
18bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantint new_called = 0;
19bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant
20bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantvoid* operator new(std::size_t s) throw(std::bad_alloc)
21bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{
22bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    ++new_called;
23bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    return std::malloc(s);
24bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant}
25bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant
26bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantvoid  operator delete(void* p) throw()
27bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{
28bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    --new_called;
29bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    std::free(p);
30bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant}
31bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant
32bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantbool A_constructed = false;
33bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant
34bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantstruct A
35bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{
36bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    A() {A_constructed = true;}
37bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    ~A() {A_constructed = false;}
38bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant};
39bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant
40bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantint main()
41bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{
42bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    A* ap = new A;
43bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    assert(ap);
44bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    assert(A_constructed);
45bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    assert(new_called);
46bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    delete ap;
47bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    assert(!A_constructed);
48bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant    assert(!new_called);
49bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant}
50