test/std/language.support/support.dynamic/new.delete/new.delete.array/new_array_nothrow_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 [] nothrow by replacing only operator new
  11
  12 // UNSUPPORTED: sanitizer-new-delete
  13
  14 // TODO Investigate why UBSAN prevents new from calling our replacement.
  15 // XFAIL: ubsan
  16
  17
  18 #include <new>
  19 #include <cstddef>
  20 #include <cstdlib>
  21 #include <cassert>
  22 #include <limits>
  23
  24 int new_called = 0;
  25
  26 void* operator new(std::size_t s) throw(std::bad_alloc)
  27 {
  28     ++new_called;
  29     void* ret = std::malloc(s);
  30     if (!ret) std::abort(); // placate MSVC's unchecked malloc warning
  31     return  ret;
  32 }
  33
  34 void  operator delete(void* p) throw()
  35 {
  36     --new_called;
  37     std::free(p);
  38 }
  39
  40 volatile int A_constructed = 0;
  41
  42 struct A
  43 {
  44     A() {++A_constructed;}
  45     ~A() {--A_constructed;}
  46 };
  47
  48 int main()
  49 {
  50     A* ap = new (std::nothrow) A[3];
  51     assert(ap);
  52     assert(A_constructed == 3);
  53     assert(new_called);
  54     delete [] ap;
  55     assert(A_constructed == 0);
  56     assert(!new_called);
  57 }