libstdc++-v3/testsuite/20_util/shared_ptr/creation/alloc.cc

   1 // { dg-do run { target c++11 } }
   2
   3 // Copyright (C) 2007-2024 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.6.6.2 Template class shared_ptr [util.smartptr.shared]
  21
  22 #include <memory>
  23 #include <testsuite_hooks.h>
  24 #include <testsuite_allocator.h>
  25
  26 using __gnu_test::tracker_allocator_counter;
  27 using __gnu_test::tracker_allocator;
  28
  29 struct A
  30 {
  31   A(int i, double d, char c = '\0') : i(i), d(d), c(c) { ++ctor_count; }
  32   explicit A(int i) : i(i), d(), c() { ++ctor_count; }
  33   A() : i(), d(), c() { ++ctor_count; }
  34   ~A() { ++dtor_count; }
  35   int i;
  36   double d;
  37   char c;
  38   static int ctor_count;
  39   static int dtor_count;
  40 };
  41 int A::ctor_count = 0;
  42 int A::dtor_count = 0;
  43
  44 struct reset_count_struct
  45 {
  46   ~reset_count_struct()
  47   {
  48     A::ctor_count = 0;
  49     A::dtor_count = 0;
  50     tracker_allocator_counter::reset();
  51   }
  52 };
  53
  54 // 20.6.6.2.6 shared_ptr creation [util.smartptr.shared.create]
  55
  56 void
  57 test01()
  58 {
  59   reset_count_struct __attribute__((unused)) reset;
  60
  61   {
  62     std::shared_ptr<A> p1 = std::allocate_shared<A>(tracker_allocator<A>());
  63     VERIFY( p1.get() != 0 );
  64     VERIFY( p1.use_count() == 1 );
  65     VERIFY( A::ctor_count == 1 );
  66     VERIFY( tracker_allocator_counter::get_allocation_count() > sizeof(A) );
  67   }
  68   VERIFY( A::ctor_count == A::dtor_count );
  69   VERIFY( tracker_allocator_counter::get_allocation_count()
  70           == tracker_allocator_counter::get_deallocation_count() );
  71 }
  72
  73 void
  74 test02()
  75 {
  76   reset_count_struct __attribute__((unused)) reset;
  77
  78   std::shared_ptr<A> p1;
  79
  80   p1 = std::allocate_shared<A>(tracker_allocator<A>(), 1);
  81   VERIFY( A::ctor_count == 1 );
  82   VERIFY( tracker_allocator_counter::get_allocation_count() > sizeof(A) );
  83
  84   p1 = std::allocate_shared<A>(tracker_allocator<A>(), 1, 2.0);
  85   VERIFY( A::ctor_count == 2 );
  86   VERIFY( A::dtor_count == 1 );
  87   VERIFY( tracker_allocator_counter::get_deallocation_count() > sizeof(A) );
  88
  89   p1 = std::allocate_shared<A>(tracker_allocator<A>(), 1, 2.0, '3');
  90   VERIFY( A::ctor_count == 3 );
  91   VERIFY( A::dtor_count == 2 );
  92   VERIFY( p1->i == 1 );
  93   VERIFY( p1->d == 2.0 );
  94   VERIFY( p1->c == '3' );
  95
  96   p1 = std::shared_ptr<A>();
  97   VERIFY( A::ctor_count == A::dtor_count );
  98   VERIFY( tracker_allocator_counter::get_allocation_count()
  99           == tracker_allocator_counter::get_deallocation_count() );
 100 }
 101
 102 void
 103 test03()
 104 {
 105   __gnu_test::CustomPointerAlloc<int> alloc;
 106   auto p = std::allocate_shared<int>(alloc, 1);
 107   VERIFY( *p == 1 );
 108 }
 109
 110 int
 111 main()
 112 {
 113   test01();
 114   test02();
 115   test03();
 116   return 0;
 117 }