libstdc++-v3/testsuite/20_util/allocator_members.cc

   1 // 2001-06-14  Benjamin Kosnik  <bkoz@redhat.com>
   2
   3 // Copyright (C) 2001, 2002 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 2, 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 COPYING.  If not, write to the Free
  18 // Software Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307,
  19 // USA.
  20
  21 // 20.4.1.1 allocator members
  22
  23 #include <memory>
  24 #include <stdexcept>
  25 #include <cstdlib>
  26 #include <testsuite_hooks.h>
  27
  28 struct gnu { };
  29
  30 bool check_new = false;
  31 bool check_delete = false;
  32
  33 void*
  34 operator new(std::size_t n) throw(std::bad_alloc)
  35 {
  36   check_new = true;
  37   return std::malloc(n);
  38 }
  39
  40 void operator delete(void *v) throw()
  41 {
  42   check_delete = true;
  43   return std::free(v);
  44 }
  45
  46 void test01()
  47 {
  48   bool test __attribute__((unused)) = true;
  49   std::allocator<gnu> obj;
  50
  51   // XXX These should work for various size allocation and
  52   // deallocations.  Currently, they only work as expected for sizes >
  53   // _MAX_BYTES as defined in stl_alloc.h, which happes to be 128.
  54   gnu* pobj = obj.allocate(256);
  55   VERIFY( check_new );
  56
  57   obj.deallocate(pobj, 256);
  58   VERIFY( check_delete );
  59 }
  60
  61 // libstdc++/8230
  62 void test02()
  63 {
  64   bool test __attribute__((unused)) = true;
  65   try
  66     {
  67       std::allocator<int> alloc;
  68       const std::allocator<int>::size_type n = alloc.max_size();
  69       int* p = alloc.allocate(n + 1);
  70       p[n] = 2002;
  71     }
  72   catch(const std::bad_alloc& e)
  73     {
  74       // Allowed.
  75       test = true;
  76     }
  77   catch(...)
  78     {
  79       test = false;
  80     }
  81   VERIFY( test );
  82 }
  83
  84 int main()
  85 {
  86   test01();
  87   test02();
  88   return 0;
  89 }