summaryrefslogtreecommitdiff
path: root/libstdc++-v3/testsuite/20_util/monotonic_buffer_resource/deallocate.cc
blob: db835935079bf8a7cc1627aff7ca2ed7450d4e8e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
// Copyright (C) 2018 Free Software Foundation, Inc.
//
// This file is part of the GNU ISO C++ Library.  This library is free
// software; you can redistribute it and/or modify it under the
// terms of the GNU General Public License as published by the
// Free Software Foundation; either version 3, or (at your option)
// any later version.

// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License along
// with this library; see the file COPYING3.  If not see
// <http://www.gnu.org/licenses/>.

// { dg-options "-std=gnu++17" }
// { dg-do run { target c++17 } }
// { dg-require-cstdint "" }

#include <memory_resource>
#include <testsuite_allocator.h>

struct resource : __gnu_test::memory_resource
{
  int allocate_calls = 0;
  int deallocate_calls = 0;

  void*
  do_allocate(std::size_t bytes, std::size_t align) override
  {
    ++allocate_calls;
    return __gnu_test::memory_resource::do_allocate(bytes, align);
  }

  void
  do_deallocate(void* p, std::size_t bytes, std::size_t align) override
  {
    ++deallocate_calls;
    __gnu_test::memory_resource::do_deallocate(p, bytes, align);
  }
};

void
test01()
{
  resource r;

  // test that it's possible to deallocate after each of the constructors
  {
    std::pmr::monotonic_buffer_resource mr(&r);
    auto p = mr.allocate(1024);
    VERIFY( p != nullptr );
    const std::uintptr_t pi = reinterpret_cast<std::uintptr_t>(p);
    mr.deallocate(p, 1024);
    VERIFY( r.deallocate_calls == 0 );
    auto q = mr.allocate(1024);
    VERIFY( q != nullptr );
    VERIFY( pi != reinterpret_cast<std::uintptr_t>(q) );
    mr.deallocate(q, 1024);
    VERIFY( r.deallocate_calls == 0 );
  }
  VERIFY( r.deallocate_calls == r.allocate_calls );
  VERIFY( r.number_of_active_allocations() == 0 );
  {
    r.deallocate_calls = r.allocate_calls = 0;
    std::pmr::monotonic_buffer_resource mr(128, &r);
    auto p = mr.allocate(64);
    VERIFY( p != nullptr );
    const std::uintptr_t pi = reinterpret_cast<std::uintptr_t>(p);
    mr.deallocate(p, 64);
    auto q = mr.allocate(1024);
    VERIFY( q != nullptr );
    VERIFY( p != q );
    VERIFY( pi != reinterpret_cast<std::uintptr_t>(q) );
    mr.deallocate(q, 1024);
    VERIFY( r.deallocate_calls == 0 );
  }
  VERIFY( r.number_of_active_allocations() == 0 );
  {
    r.deallocate_calls = r.allocate_calls = 0;
    unsigned char buf[64];
    std::pmr::monotonic_buffer_resource mr((void*)buf, sizeof(buf), &r);
    auto p = mr.allocate(64);
    VERIFY( p != nullptr );
    const std::uintptr_t pi = reinterpret_cast<std::uintptr_t>(p);
    mr.deallocate(p, 64);
    auto q = mr.allocate(1024);
    VERIFY( q != nullptr );
    VERIFY( p != q );
    VERIFY( pi != reinterpret_cast<std::uintptr_t>(q) );
    mr.deallocate(q, 1024);
    VERIFY( r.deallocate_calls == 0 );
  }
  VERIFY( r.deallocate_calls == r.allocate_calls );
  VERIFY( r.number_of_active_allocations() == 0 );
}

int
main()
{
  test01();
}