a554497024
From-SVN: r267494
185 lines
3.9 KiB
C++
185 lines
3.9 KiB
C++
// Copyright (C) 2018-2019 Free Software Foundation, Inc.
|
|
//
|
|
// This file is part of the GNU ISO C++ Library. This library is free
|
|
// software; you can redistribute it and/or modify it under the
|
|
// terms of the GNU General Public License as published by the
|
|
// Free Software Foundation; either version 3, or (at your option)
|
|
// any later version.
|
|
|
|
// This library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License along
|
|
// with this library; see the file COPYING3. If not see
|
|
// <http://www.gnu.org/licenses/>.
|
|
|
|
// { dg-do run }
|
|
// { dg-xfail-run-if "AIX operator new" { powerpc-ibm-aix* } }
|
|
|
|
#include <new>
|
|
#include <stdlib.h>
|
|
#include <testsuite_hooks.h>
|
|
|
|
// PR libstdc++/68210
|
|
|
|
struct MyBadAlloc: std::bad_alloc { };
|
|
|
|
static bool new_fail;
|
|
static bool bad_alloc_thrown;
|
|
static unsigned new_called;
|
|
static unsigned delete_called;
|
|
static unsigned new_vec_called;
|
|
static unsigned delete_vec_called;
|
|
static unsigned new_handler_called;
|
|
|
|
static void new_handler ()
|
|
{
|
|
if (new_handler_called++)
|
|
throw MyBadAlloc ();
|
|
}
|
|
|
|
void* operator new (size_t n)
|
|
{
|
|
static size_t cntr;
|
|
|
|
++new_called;
|
|
|
|
for ( ; ; ) {
|
|
if (void *p = new_fail ? 0 : malloc (n + sizeof n)) {
|
|
*static_cast<size_t*>(p) = ++cntr;
|
|
return static_cast<size_t*>(p) + 1;
|
|
}
|
|
|
|
if (std::new_handler h = std::set_new_handler (0)) {
|
|
std::set_new_handler (h);
|
|
h ();
|
|
}
|
|
else {
|
|
bad_alloc_thrown = true;
|
|
throw MyBadAlloc ();
|
|
}
|
|
}
|
|
}
|
|
|
|
void operator delete (void *p)
|
|
{
|
|
++delete_called;
|
|
if (p)
|
|
free (static_cast<size_t*>(p) - 1);
|
|
}
|
|
|
|
void* operator new[] (size_t n)
|
|
{
|
|
++new_vec_called;
|
|
return operator new(n);
|
|
}
|
|
|
|
void operator delete[] (void *p)
|
|
{
|
|
++delete_vec_called;
|
|
operator delete(p);
|
|
}
|
|
|
|
#if __cplusplus >= 201402L
|
|
void operator delete (void *p, std::size_t)
|
|
{
|
|
::operator delete(p);
|
|
}
|
|
void operator delete[] (void *p, std::size_t)
|
|
{
|
|
::operator delete[](p);
|
|
}
|
|
#endif
|
|
|
|
void init()
|
|
{
|
|
new_fail = false;
|
|
new_called = 0;
|
|
delete_called = 0;
|
|
new_vec_called = 0;
|
|
delete_vec_called = 0;
|
|
new_handler_called = 0;
|
|
std::set_new_handler (0);
|
|
}
|
|
|
|
void
|
|
test01()
|
|
{
|
|
init ();
|
|
|
|
void *p = operator new (1, std::nothrow);
|
|
|
|
VERIFY (p != 0);
|
|
VERIFY (1 == new_called);
|
|
VERIFY (0 == new_handler_called);
|
|
VERIFY (!bad_alloc_thrown);
|
|
|
|
operator delete (p, std::nothrow);
|
|
VERIFY( 1 == delete_called );
|
|
|
|
new_fail = true;
|
|
p = operator new (1, std::nothrow);
|
|
|
|
VERIFY (0 == p);
|
|
VERIFY (2 == new_called);
|
|
VERIFY (0 == new_handler_called);
|
|
VERIFY (bad_alloc_thrown);
|
|
|
|
new_fail = true;
|
|
bad_alloc_thrown = false;
|
|
std::set_new_handler (new_handler);
|
|
p = operator new (1, std::nothrow);
|
|
|
|
VERIFY (0 == p);
|
|
VERIFY (3 == new_called);
|
|
VERIFY (2 == new_handler_called);
|
|
VERIFY (!bad_alloc_thrown);
|
|
}
|
|
|
|
void
|
|
test02()
|
|
{
|
|
init ();
|
|
|
|
void *p = operator new[] (1, std::nothrow);
|
|
|
|
VERIFY (p != 0);
|
|
VERIFY (1 == new_called);
|
|
VERIFY (1 == new_vec_called);
|
|
VERIFY (0 == new_handler_called);
|
|
VERIFY (!bad_alloc_thrown);
|
|
|
|
operator delete[] (p, std::nothrow);
|
|
VERIFY( 1 == delete_called );
|
|
VERIFY( 1 == delete_vec_called );
|
|
|
|
new_fail = true;
|
|
p = operator new[] (1, std::nothrow);
|
|
|
|
VERIFY (0 == p);
|
|
VERIFY (2 == new_called);
|
|
VERIFY (2 == new_vec_called);
|
|
VERIFY (0 == new_handler_called);
|
|
VERIFY (bad_alloc_thrown);
|
|
|
|
new_fail = true;
|
|
bad_alloc_thrown = false;
|
|
std::set_new_handler (new_handler);
|
|
p = operator new[] (1, std::nothrow);
|
|
|
|
VERIFY (0 == p);
|
|
VERIFY (3 == new_called);
|
|
VERIFY (3 == new_vec_called);
|
|
VERIFY (2 == new_handler_called);
|
|
VERIFY (!bad_alloc_thrown);
|
|
}
|
|
|
|
|
|
int main()
|
|
{
|
|
test01();
|
|
test02();
|
|
}
|