test/std/language.support/support.dynamic/new.delete/new.delete.single/new_replace.pass.cpp

   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 replacement
  11
  12 // UNSUPPORTED: sanitizer-new-delete
  13
  14 #include <new>
  15 #include <cstddef>
  16 #include <cstdlib>
  17 #include <cassert>
  18 #include <limits>
  19
  20 int new_called = 0;
  21
  22 void* operator new(std::size_t s) throw(std::bad_alloc)
  23 {
  24     ++new_called;
  25     void* ret = std::malloc(s);
  26     if (!ret) std::abort(); // placate MSVC's unchecked malloc warning
  27     return ret;
  28 }
  29
  30 void  operator delete(void* p) throw()
  31 {
  32     --new_called;
  33     std::free(p);
  34 }
  35
  36 bool A_constructed = false;
  37
  38 struct A
  39 {
  40     A() {A_constructed = true;}
  41     ~A() {A_constructed = false;}
  42 };
  43
  44 int main()
  45 {
  46     A* ap = new A;
  47     assert(ap);
  48     assert(A_constructed);
  49     assert(new_called);
  50     delete ap;
  51     assert(!A_constructed);
  52     assert(!new_called);
  53 }