libstdc++-v3/testsuite/23_containers/vector/zero_sized_allocations.cc

   1 // Copyright (C) 2007-2019 Free Software Foundation, Inc.
   2 //
   3 // This file is part of the GNU ISO C++ Library.  This library is free
   4 // software; you can redistribute it and/or modify it under the
   5 // terms of the GNU General Public License as published by the
   6 // Free Software Foundation; either version 3, or (at your option)
   7 // any later version.
   8 //
   9 // This library is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU General Public License along
  15 // with this library; see the file COPYING3.  If not see
  16 // <http://www.gnu.org/licenses/>.
  17 //
  18
  19 #include <vector>
  20 #include <cstdlib>
  21 #include <testsuite_hooks.h>
  22
  23 unsigned int zero_sized_news = 0;
  24
  25 void *operator new(std::size_t size) THROW (std::bad_alloc)
  26 {
  27   /* malloc(0) is unpredictable; avoid it.  */
  28   if (size == 0)
  29     {
  30       size = 1;
  31       ++zero_sized_news;
  32     }
  33
  34   void *p = std::malloc(size);
  35
  36   if (p == 0)
  37     throw std::bad_alloc();
  38
  39   return p;
  40 }
  41
  42 void operator delete(void *ptr) throw()
  43 {
  44   if (ptr != 0)
  45     std::free(ptr);
  46 }
  47
  48 #if __cpp_sized_deallocation
  49 void operator delete(void *ptr, std::size_t) throw()
  50 {
  51   if (ptr != 0)
  52     std::free(ptr);
  53 }
  54 #endif
  55
  56 // http://gcc.gnu.org/ml/libstdc++/2007-09/msg00006.html
  57 void test01()
  58 {
  59   std::vector<std::vector<int> > *v;
  60   VERIFY( zero_sized_news == 0 );
  61
  62   v = new std::vector<std::vector<int> >;
  63   VERIFY( zero_sized_news == 0 );
  64
  65   v->resize(10);
  66   delete v;
  67   VERIFY( zero_sized_news == 0 );
  68 }
  69
  70 int main()
  71 {
  72   test01();
  73   return 0;
  74 }