[PATCH] libstdc++: Remove constraints on std::generator::promise_type::operator new
Jonathan Wakely
jwakely@redhat.com
Wed Dec 11 10:58:02 GMT 2024
This was approved in Wrocław as LWG 3900, so that passing an incorrect
argument intended as an allocator will be ill-formed, instead of
silently ignored.
This also renames the template parameters and function parameters for
the allocators, to align with the names in the standard. I found it too
confusing to have a parameter _Alloc which doesn't correspond to Alloc
in the standard. Rename _Alloc to _Allocator (which the standard calls
Allocator) and rename _Na to _Alloc (which the standard calls Alloc).
libstdc++-v3/ChangeLog:
* include/std/generator (_Promise_alloc): Rename template
parameter. Use __alloc_rebind to rebind allocator.
(_Promise_alloc::operator new): Replace constraints with a
static_assert in the body. Rename allocator parameter.
(_Promise_alloc<void>::_M_allocate): Rename allocator parameter.
Use __alloc_rebind to rebind allocator.
(_Promise_alloc<void>::operator new): Rename allocator
parameter.
* testsuite/24_iterators/range_generators/alloc.cc: New test.
* testsuite/24_iterators/range_generators/lwg3900.cc: New test.
---
Tested x86_64-linux.
libstdc++-v3/include/std/generator | 55 +++++++++----------
.../24_iterators/range_generators/alloc.cc | 45 +++++++++++++++
.../24_iterators/range_generators/lwg3900.cc | 16 ++++++
3 files changed, 87 insertions(+), 29 deletions(-)
create mode 100644 libstdc++-v3/testsuite/24_iterators/range_generators/alloc.cc
create mode 100644 libstdc++-v3/testsuite/24_iterators/range_generators/lwg3900.cc
diff --git a/libstdc++-v3/include/std/generator b/libstdc++-v3/include/std/generator
index 0a14e70064e..16b953e90af 100644
--- a/libstdc++-v3/include/std/generator
+++ b/libstdc++-v3/include/std/generator
@@ -416,13 +416,11 @@ _GLIBCXX_BEGIN_NAMESPACE_VERSION
concept _Stateless_alloc = (allocator_traits<_All>::is_always_equal::value
&& default_initializable<_All>);
- template<typename _Alloc>
+ template<typename _Allocator>
class _Promise_alloc
{
- using _ATr = allocator_traits<_Alloc>;
- using _Rebound = typename _ATr::template rebind_alloc<_Alloc_block>;
- using _Rebound_ATr = typename _ATr
- ::template rebind_traits<_Alloc_block>;
+ using _Rebound = __alloc_rebind<_Allocator, _Alloc_block>;
+ using _Rebound_ATr = allocator_traits<_Rebound>;
static_assert(is_pointer_v<typename _Rebound_ATr::pointer>,
"Must use allocators for true pointers with generators");
@@ -465,30 +463,31 @@ _GLIBCXX_BEGIN_NAMESPACE_VERSION
public:
void*
operator new(std::size_t __sz)
- requires default_initializable<_Rebound> // _Alloc is non-void
+ requires default_initializable<_Rebound> // _Allocator is non-void
{ return _M_allocate({}, __sz); }
- template<typename _Na, typename... _Args>
+ // _GLIBCXX_RESOLVE_LIB_DEFECTS
+ // 3900. The allocator_arg_t overloads of promise_type::operator new
+ // should not be constrained
+ template<typename _Alloc, typename... _Args>
void*
operator new(std::size_t __sz,
- allocator_arg_t, const _Na& __na,
+ allocator_arg_t, const _Alloc& __a,
const _Args&...)
- requires convertible_to<const _Na&, _Alloc>
{
- return _M_allocate(static_cast<_Rebound>(static_cast<_Alloc>(__na)),
- __sz);
+ static_assert(convertible_to<const _Alloc&, _Allocator>);
+ return _M_allocate(_Rebound(_Allocator(__a)), __sz);
}
- template<typename _This, typename _Na, typename... _Args>
+ template<typename _This, typename _Alloc, typename... _Args>
void*
operator new(std::size_t __sz,
const _This&,
- allocator_arg_t, const _Na& __na,
+ allocator_arg_t, const _Alloc& __a,
const _Args&...)
- requires convertible_to<const _Na&, _Alloc>
{
- return _M_allocate(static_cast<_Rebound>(static_cast<_Alloc>(__na)),
- __sz);
+ static_assert(convertible_to<const _Alloc&, _Allocator>);
+ return _M_allocate(_Rebound(_Allocator(__a)), __sz);
}
void
@@ -578,20 +577,18 @@ _GLIBCXX_BEGIN_NAMESPACE_VERSION
}
}
- template<typename _Na>
+ template<typename _Alloc>
static void*
- _M_allocate(const _Na& __na, std::size_t __csz)
+ _M_allocate(const _Alloc& __a, std::size_t __csz)
{
- using _Rebound = typename std::allocator_traits<_Na>
- ::template rebind_alloc<_Alloc_block>;
- using _Rebound_ATr = typename std::allocator_traits<_Na>
- ::template rebind_traits<_Alloc_block>;
+ using _Rebound = __alloc_rebind<_Alloc, _Alloc_block>;
+ using _Rebound_ATr = allocator_traits<_Rebound>;
static_assert(is_pointer_v<typename _Rebound_ATr::pointer>,
"Must use allocators for true pointers with generators");
_Dealloc_fn __d = &_M_deallocator<_Rebound>;
- auto __b = static_cast<_Rebound>(__na);
+ auto __b = static_cast<_Rebound>(__a);
auto __asz = _M_alloc_size<_Rebound>(__csz);
auto __nblk = _Alloc_block::_M_cnt(__asz);
void* __p = __b.allocate(__nblk);
@@ -619,20 +616,20 @@ _GLIBCXX_BEGIN_NAMESPACE_VERSION
return __p;
}
- template<typename _Na, typename... _Args>
+ template<typename _Alloc, typename... _Args>
void*
operator new(std::size_t __sz,
- allocator_arg_t, const _Na& __na,
+ allocator_arg_t, const _Alloc& __a,
const _Args&...)
- { return _M_allocate(__na, __sz); }
+ { return _M_allocate(__a, __sz); }
- template<typename _This, typename _Na, typename... _Args>
+ template<typename _This, typename _Alloc, typename... _Args>
void*
operator new(std::size_t __sz,
const _This&,
- allocator_arg_t, const _Na& __na,
+ allocator_arg_t, const _Alloc& __a,
const _Args&...)
- { return _M_allocate(__na, __sz); }
+ { return _M_allocate(__a, __sz); }
void
operator delete(void* __ptr, std::size_t __sz) noexcept
diff --git a/libstdc++-v3/testsuite/24_iterators/range_generators/alloc.cc b/libstdc++-v3/testsuite/24_iterators/range_generators/alloc.cc
new file mode 100644
index 00000000000..8671da748dd
--- /dev/null
+++ b/libstdc++-v3/testsuite/24_iterators/range_generators/alloc.cc
@@ -0,0 +1,45 @@
+// { dg-do run { target c++23 } }
+
+#include <generator>
+#include <testsuite_hooks.h>
+#include <testsuite_allocator.h>
+
+template<typename... Args>
+std::pmr::generator<int>
+gen(Args...)
+{
+ co_yield 1;
+ co_yield 2;
+}
+
+int
+main()
+{
+ __gnu_test::memory_resource mr;
+ for (auto _ : gen())
+ VERIFY(mr.number_of_active_allocations() == 0);
+
+ for (auto _ : gen(std::allocator_arg))
+ VERIFY(mr.number_of_active_allocations() == 0);
+
+ for (auto _ : gen(std::allocator_arg, std::pmr::new_delete_resource()))
+ VERIFY(mr.number_of_active_allocations() == 0);
+
+#if __cpp_exceptions
+ try {
+ for (auto _ : gen(std::allocator_arg, std::pmr::null_memory_resource()))
+ VERIFY(false);
+ } catch (const std::bad_alloc&) {
+ }
+#endif
+
+ VERIFY(mr.number_of_active_allocations() == 0);
+
+ for (auto _ : gen(std::allocator_arg , &mr))
+ VERIFY(mr.number_of_active_allocations() == 1);
+
+ for (auto _ : gen("const This& argument", std::allocator_arg , &mr))
+ VERIFY(mr.number_of_active_allocations() == 1);
+
+ VERIFY(mr.number_of_active_allocations() == 0);
+}
diff --git a/libstdc++-v3/testsuite/24_iterators/range_generators/lwg3900.cc b/libstdc++-v3/testsuite/24_iterators/range_generators/lwg3900.cc
new file mode 100644
index 00000000000..957879e8862
--- /dev/null
+++ b/libstdc++-v3/testsuite/24_iterators/range_generators/lwg3900.cc
@@ -0,0 +1,16 @@
+// { dg-do compile { target c++23 } }
+
+// LWG 3900. allocator_arg_t overloads of generator::promise_type::operator new
+// should not be constrained
+
+#include <generator>
+#include <memory_resource>
+
+std::pmr::generator<int>
+bar(std::allocator_arg_t, std::pmr::memory_resource& mr) // { dg-error "here" }
+{
+ co_yield 3;
+}
+
+// { dg-error "static assertion failed" "" { target *-*-* } 0 }
+// { dg-error "no matching function .*memory_resource&" "" { target *-*-* } 0 }
--
2.47.1
More information about the Libstdc++
mailing list