1 //===------------------------ memory_resource.cpp -------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 #include "experimental/memory_resource"
10
11 #ifndef _LIBCPP_HAS_NO_ATOMIC_HEADER
12 #include "atomic"
13 #elif !defined(_LIBCPP_HAS_NO_THREADS)
14 #include "mutex"
15 #if defined(__ELF__) && defined(_LIBCPP_LINK_PTHREAD_LIB)
16 #pragma comment(lib, "pthread")
17 #endif
18 #endif
19
20 _LIBCPP_BEGIN_NAMESPACE_LFTS_PMR
21
22 // memory_resource
23
24 //memory_resource::~memory_resource() {}
25
26 // new_delete_resource()
27
28 class _LIBCPP_TYPE_VIS __new_delete_memory_resource_imp
29 : public memory_resource
30 {
do_allocate(size_t size,size_t align)31 void *do_allocate(size_t size, size_t align) override {
32 #ifdef _LIBCPP_HAS_NO_ALIGNED_ALLOCATION
33 if (__is_overaligned_for_new(align))
34 __throw_bad_alloc();
35 #endif
36 return _VSTD::__libcpp_allocate(size, align);
37 }
38
do_deallocate(void * p,size_t n,size_t align)39 void do_deallocate(void *p, size_t n, size_t align) override {
40 _VSTD::__libcpp_deallocate(p, n, align);
41 }
42
do_is_equal(memory_resource const & other) const43 bool do_is_equal(memory_resource const & other) const noexcept override
44 { return &other == this; }
45
46 public:
47 ~__new_delete_memory_resource_imp() override = default;
48 };
49
50 // null_memory_resource()
51
52 class _LIBCPP_TYPE_VIS __null_memory_resource_imp
53 : public memory_resource
54 {
55 public:
56 ~__null_memory_resource_imp() = default;
57
58 protected:
do_allocate(size_t,size_t)59 virtual void* do_allocate(size_t, size_t) {
60 __throw_bad_alloc();
61 }
do_deallocate(void *,size_t,size_t)62 virtual void do_deallocate(void *, size_t, size_t) {}
do_is_equal(memory_resource const & __other) const63 virtual bool do_is_equal(memory_resource const & __other) const noexcept
64 { return &__other == this; }
65 };
66
67 namespace {
68
69 union ResourceInitHelper {
70 struct {
71 __new_delete_memory_resource_imp new_delete_res;
72 __null_memory_resource_imp null_res;
73 } resources;
74 char dummy;
ResourceInitHelper()75 _LIBCPP_CONSTEXPR_AFTER_CXX11 ResourceInitHelper() : resources() {}
~ResourceInitHelper()76 ~ResourceInitHelper() {}
77 };
78
79 # 79 "memory_resource.cpp" 1 3
80 _LIBCPP_SAFE_STATIC ResourceInitHelper res_init _LIBCPP_INIT_PRIORITY_MAX;
81 # 81 "memory_resource.cpp" 2
82
83 } // end namespace
84
85
new_delete_resource()86 memory_resource * new_delete_resource() noexcept {
87 return &res_init.resources.new_delete_res;
88 }
89
null_memory_resource()90 memory_resource * null_memory_resource() noexcept {
91 return &res_init.resources.null_res;
92 }
93
94 // default_memory_resource()
95
96 static memory_resource *
__default_memory_resource(bool set=false,memory_resource * new_res=nullptr)97 __default_memory_resource(bool set = false, memory_resource * new_res = nullptr) noexcept
98 {
99 #ifndef _LIBCPP_HAS_NO_ATOMIC_HEADER
100 _LIBCPP_SAFE_STATIC static atomic<memory_resource*> __res =
101 ATOMIC_VAR_INIT(&res_init.resources.new_delete_res);
102 if (set) {
103 new_res = new_res ? new_res : new_delete_resource();
104 // TODO: Can a weaker ordering be used?
105 return _VSTD::atomic_exchange_explicit(
106 &__res, new_res, memory_order_acq_rel);
107 }
108 else {
109 return _VSTD::atomic_load_explicit(
110 &__res, memory_order_acquire);
111 }
112 #elif !defined(_LIBCPP_HAS_NO_THREADS)
113 _LIBCPP_SAFE_STATIC static memory_resource * res = &res_init.resources.new_delete_res;
114 static mutex res_lock;
115 if (set) {
116 new_res = new_res ? new_res : new_delete_resource();
117 lock_guard<mutex> guard(res_lock);
118 memory_resource * old_res = res;
119 res = new_res;
120 return old_res;
121 } else {
122 lock_guard<mutex> guard(res_lock);
123 return res;
124 }
125 #else
126 _LIBCPP_SAFE_STATIC static memory_resource* res = &res_init.resources.new_delete_res;
127 if (set) {
128 new_res = new_res ? new_res : new_delete_resource();
129 memory_resource * old_res = res;
130 res = new_res;
131 return old_res;
132 } else {
133 return res;
134 }
135 #endif
136 }
137
get_default_resource()138 memory_resource * get_default_resource() noexcept
139 {
140 return __default_memory_resource();
141 }
142
set_default_resource(memory_resource * __new_res)143 memory_resource * set_default_resource(memory_resource * __new_res) noexcept
144 {
145 return __default_memory_resource(true, __new_res);
146 }
147
148 _LIBCPP_END_NAMESPACE_LFTS_PMR
149