libstdc++-v3/testsuite/20_util/allocator/1.cc

   1 // 2001-06-14  Benjamin Kosnik  <bkoz@redhat.com>
   2
   3 // Copyright (C) 2001-2021 Free Software Foundation, Inc.
   4 //
   5 // This file is part of the GNU ISO C++ Library.  This library is free
   6 // software; you can redistribute it and/or modify it under the
   7 // terms of the GNU General Public License as published by the
   8 // Free Software Foundation; either version 3, or (at your option)
   9 // any later version.
  10
  11 // This library is distributed in the hope that it will be useful,
  12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 // GNU General Public License for more details.
  15
  16 // You should have received a copy of the GNU General Public License along
  17 // with this library; see the file COPYING3.  If not see
  18 // <http://www.gnu.org/licenses/>.
  19
  20 // 20.4.1.1 allocator members
  21
  22 #include <memory>
  23 #include <stdexcept>
  24 #include <cstdlib>
  25 #include <testsuite_hooks.h>
  26
  27 #if __cplusplus >= 201103L
  28 # define NOTHROW noexcept
  29 #else
  30 # define NOTHROW throw()
  31 #endif
  32
  33 struct gnu { };
  34
  35 bool check_new = false;
  36 bool check_delete = false;
  37
  38 void*
  39 operator new(std::size_t n) THROW(std::bad_alloc)
  40 {
  41   check_new = true;
  42   return std::malloc(n);
  43 }
  44
  45 void operator delete(void *v) NOTHROW
  46 {
  47   check_delete = true;
  48   return std::free(v);
  49 }
  50
  51 #if __cpp_sized_deallocation
  52 void operator delete(void *v, std::size_t) NOTHROW
  53 {
  54   ::operator delete(v);
  55 }
  56 #endif
  57
  58 void test01()
  59 {
  60   std::allocator<gnu> obj;
  61
  62   // NB: These should work for various size allocation and
  63   // deallocations.  Currently, they only work as expected for sizes >
  64   // _MAX_BYTES as defined in stl_alloc.h, which happes to be 128.
  65   gnu* pobj = obj.allocate(256);
  66   VERIFY( check_new );
  67
  68   obj.deallocate(pobj, 256);
  69   VERIFY( check_delete );
  70 }
  71
  72 int main()
  73 {
  74   test01();
  75   return 0;
  76 }
  77