1 // Copyright (C) 2018-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 // { dg-options "-std=gnu++17" }
19 // { dg-do run { target c++17 } }
20 
21 #include <memory_resource>
22 #include <testsuite_hooks.h>
23 #include <testsuite_allocator.h>
24 
eq(const std::pmr::pool_options & lhs,const std::pmr::pool_options & rhs)25 bool eq(const std::pmr::pool_options& lhs, const std::pmr::pool_options& rhs)
26 {
27   return lhs.max_blocks_per_chunk == rhs.max_blocks_per_chunk
28     && lhs.largest_required_pool_block == rhs.largest_required_pool_block;
29 }
30 
31 void
test01()32 test01()
33 {
34   std::pmr::unsynchronized_pool_resource r0;
35   const std::pmr::pool_options opts = r0.options();
36   VERIFY( opts.max_blocks_per_chunk != 0 );
37   VERIFY( opts.largest_required_pool_block != 0 );
38 
39   std::pmr::unsynchronized_pool_resource r1(opts);
40   const auto opts1 = r1.options();
41   VERIFY( eq(opts, opts1) );
42 
43   std::pmr::unsynchronized_pool_resource r2(std::pmr::pool_options{0, 0});
44   const auto opts2 = r2.options();
45   VERIFY( eq(opts, opts2) );
46 }
47 
48 void
test02()49 test02()
50 {
51   std::pmr::pool_options opts{0, 0};
52   std::size_t num_allocs = 0;
53 
54   __gnu_test::memory_resource test_mr;
55 
56   std::pmr::unsynchronized_pool_resource r1(opts, &test_mr);
57   opts = r1.options();
58   // opts.largest_required_pool_block should be set to the block size of
59   // the largest pool (this is a GNU extension). Confirm this by checking
60   // that allocations larger than opts.largest_required_pool_block come
61   // directly from the upstream allocator, test_mr, not from r1's pools.
62 
63   // The following should result in a "large" allocation direct from upstream:
64   (void) r1.allocate(opts.largest_required_pool_block + 1);
65   num_allocs = test_mr.number_of_active_allocations();
66   // This should result in another "large" allocation direct from upstream:
67   (void) r1.allocate(opts.largest_required_pool_block + 1);
68   // Which means the number of upstream allocations should have increased:
69   VERIFY( test_mr.number_of_active_allocations() > num_allocs );
70   r1.release();
71 
72   // Repeat with a user-specified block size:
73   opts.largest_required_pool_block = 64;
74   std::pmr::unsynchronized_pool_resource r2(opts, &test_mr);
75   opts = r2.options();
76   (void) r2.allocate(opts.largest_required_pool_block + 1);
77   num_allocs = test_mr.number_of_active_allocations();
78   (void) r2.allocate(opts.largest_required_pool_block + 1);
79   VERIFY( test_mr.number_of_active_allocations() > num_allocs );
80   r2.release();
81 
82   // Repeat with an odd user-specified block size:
83   opts.largest_required_pool_block = 71;
84   std::pmr::unsynchronized_pool_resource r3(opts, &test_mr);
85   opts = r3.options();
86   (void) r3.allocate(opts.largest_required_pool_block + 1);
87   num_allocs = test_mr.number_of_active_allocations();
88   (void) r3.allocate(opts.largest_required_pool_block + 1);
89   VERIFY( test_mr.number_of_active_allocations() > num_allocs );
90   r3.release();
91 }
92 
93 int
main()94 main()
95 {
96   test01();
97   test02();
98 }
99