libstdc++-v3/testsuite/ext/allocators.cc

   1 // 2001-11-25  Phil Edwards  <pme@gcc.gnu.org>
   2 //
   3 // Copyright (C) 2001, 2003 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 <cstdlib>
  24 #include <memory>
  25 #include <ext/pool_allocator.h>
  26 #include <ext/debug_allocator.h>
  27 #include <ext/malloc_allocator.h>
  28 #include <testsuite_hooks.h>
  29
  30 using __gnu_cxx::__malloc_alloc;
  31 using __gnu_cxx::__debug_alloc;
  32 using __gnu_cxx::__pool_alloc;
  33
  34 template class __malloc_alloc<3>;
  35 template class __debug_alloc<__malloc_alloc<3> >;
  36 template class __pool_alloc<true, 3>;
  37 template class __pool_alloc<false, 3>;
  38
  39 struct big
  40 {
  41   long f[15];
  42 };
  43
  44
  45 bool         new_called;
  46 bool         delete_called;
  47 std::size_t  requested;
  48
  49 void*
  50 operator new(std::size_t n) throw(std::bad_alloc)
  51 {
  52   new_called = true;
  53   requested = n;
  54   return std::malloc(n);
  55 }
  56
  57 void
  58 operator delete(void *v) throw()
  59 {
  60   delete_called = true;
  61   return std::free(v);
  62 }
  63
  64 template<typename Alloc, bool uses_global_new_and_delete>
  65 void check_allocator()
  66 {
  67   bool test __attribute__((unused)) = true;
  68   new_called = false;
  69   delete_called = false;
  70   requested = 0;
  71
  72   std::__allocator<big, Alloc>   a;
  73   big *p = a.allocate(10);
  74   if (uses_global_new_and_delete)
  75     VERIFY( requested >= (10 * 15 * sizeof(long)) );
  76
  77   // Touch the far end of supposedly-allocated memory to check that we got
  78   // all of it.  Why "3"?  Because it's my favorite integer between e and pi.
  79   p[9].f[14] = 3;
  80   VERIFY( new_called == uses_global_new_and_delete );
  81   a.deallocate(p,10);
  82   VERIFY( delete_called == uses_global_new_and_delete );
  83 }
  84
  85 // These just help tracking down error messages.
  86 void test01() { check_allocator<__malloc_alloc<3>, false>(); }
  87 void test02() { check_allocator<__debug_alloc<__malloc_alloc<3> >, false>(); }
  88 void test03() { check_allocator<__pool_alloc<true, 3>, true>(); }
  89 void test04() { check_allocator<__pool_alloc<false, 3>, true>(); }
  90
  91 int main()
  92 {
  93   test01();
  94   test02();
  95   test03();
  96   test04();
  97   return 0;
  98 }
  99