Quuxplusone updated this revision to Diff 154041.
Quuxplusone added a comment.
Updated to no longer check "size != 0".
Also rolled in some drive-by cosmetic refactoring that I'd done later in my
branch: these functions aren't in a public header and don't need to be uglified.
Repository:
rCXX libc++
https://reviews.llvm.org/D47344
Files:
src/experimental/memory_resource.cpp
Index: src/experimental/memory_resource.cpp
===================================================================
--- src/experimental/memory_resource.cpp
+++ src/experimental/memory_resource.cpp
@@ -8,6 +8,7 @@
//===----------------------------------------------------------------------===//
#include "experimental/memory_resource"
+#include "memory"
#ifndef _LIBCPP_HAS_NO_ATOMIC_HEADER
#include "atomic"
@@ -23,38 +24,47 @@
// new_delete_resource()
+#ifdef _LIBCPP_HAS_NO_ALIGNED_ALLOCATION
+static bool is_aligned_to(void *ptr, size_t align)
+{
+ void *p2 = ptr;
+ size_t space = 1;
+ void *result = _VSTD::align(align, 1, p2, space);
+ return (result == ptr);
+}
+#endif
+
class _LIBCPP_TYPE_VIS __new_delete_memory_resource_imp
: public memory_resource
{
-public:
- ~__new_delete_memory_resource_imp() = default;
-
-protected:
- virtual void* do_allocate(size_t __size, size_t __align)
- { return _VSTD::__libcpp_allocate(__size, __align); /* FIXME */}
+ void *do_allocate(size_t bytes, size_t align) override
+ {
+ void *result = _VSTD::__libcpp_allocate(bytes, align);
+#ifdef _LIBCPP_HAS_NO_ALIGNED_ALLOCATION
+ if (!is_aligned_to(result, align)) {
+ _VSTD::__libcpp_deallocate(result, align);
+ __throw_bad_alloc();
+ }
+#endif
+ return result;
+ }
- virtual void do_deallocate(void * __p, size_t, size_t __align)
- { _VSTD::__libcpp_deallocate(__p, __align); /* FIXME */ }
+ void do_deallocate(void *p, size_t, size_t align) override
+ { _VSTD::__libcpp_deallocate(p, align); }
- virtual bool do_is_equal(memory_resource const & __other) const _NOEXCEPT
- { return &__other == this; }
+ bool do_is_equal(const memory_resource& other) const _NOEXCEPT override
+ { return &other == this; }
};
// null_memory_resource()
class _LIBCPP_TYPE_VIS __null_memory_resource_imp
: public memory_resource
{
-public:
- ~__null_memory_resource_imp() = default;
-
-protected:
- virtual void* do_allocate(size_t, size_t) {
- __throw_bad_alloc();
- }
- virtual void do_deallocate(void *, size_t, size_t) {}
- virtual bool do_is_equal(memory_resource const & __other) const _NOEXCEPT
- { return &__other == this; }
+ void *do_allocate(size_t, size_t) override { __throw_bad_alloc(); }
+ void do_deallocate(void *, size_t, size_t) override {}
+ bool do_is_equal(const memory_resource& other) const _NOEXCEPT override
+ { return &other == this; }
};
namespace {
Index: src/experimental/memory_resource.cpp
===================================================================
--- src/experimental/memory_resource.cpp
+++ src/experimental/memory_resource.cpp
@@ -8,6 +8,7 @@
//===----------------------------------------------------------------------===//
#include "experimental/memory_resource"
+#include "memory"
#ifndef _LIBCPP_HAS_NO_ATOMIC_HEADER
#include "atomic"
@@ -23,38 +24,47 @@
// new_delete_resource()
+#ifdef _LIBCPP_HAS_NO_ALIGNED_ALLOCATION
+static bool is_aligned_to(void *ptr, size_t align)
+{
+ void *p2 = ptr;
+ size_t space = 1;
+ void *result = _VSTD::align(align, 1, p2, space);
+ return (result == ptr);
+}
+#endif
+
class _LIBCPP_TYPE_VIS __new_delete_memory_resource_imp
: public memory_resource
{
-public:
- ~__new_delete_memory_resource_imp() = default;
-
-protected:
- virtual void* do_allocate(size_t __size, size_t __align)
- { return _VSTD::__libcpp_allocate(__size, __align); /* FIXME */}
+ void *do_allocate(size_t bytes, size_t align) override
+ {
+ void *result = _VSTD::__libcpp_allocate(bytes, align);
+#ifdef _LIBCPP_HAS_NO_ALIGNED_ALLOCATION
+ if (!is_aligned_to(result, align)) {
+ _VSTD::__libcpp_deallocate(result, align);
+ __throw_bad_alloc();
+ }
+#endif
+ return result;
+ }
- virtual void do_deallocate(void * __p, size_t, size_t __align)
- { _VSTD::__libcpp_deallocate(__p, __align); /* FIXME */ }
+ void do_deallocate(void *p, size_t, size_t align) override
+ { _VSTD::__libcpp_deallocate(p, align); }
- virtual bool do_is_equal(memory_resource const & __other) const _NOEXCEPT
- { return &__other == this; }
+ bool do_is_equal(const memory_resource& other) const _NOEXCEPT override
+ { return &other == this; }
};
// null_memory_resource()
class _LIBCPP_TYPE_VIS __null_memory_resource_imp
: public memory_resource
{
-public:
- ~__null_memory_resource_imp() = default;
-
-protected:
- virtual void* do_allocate(size_t, size_t) {
- __throw_bad_alloc();
- }
- virtual void do_deallocate(void *, size_t, size_t) {}
- virtual bool do_is_equal(memory_resource const & __other) const _NOEXCEPT
- { return &__other == this; }
+ void *do_allocate(size_t, size_t) override { __throw_bad_alloc(); }
+ void do_deallocate(void *, size_t, size_t) override {}
+ bool do_is_equal(const memory_resource& other) const _NOEXCEPT override
+ { return &other == this; }
};
namespace {
_______________________________________________
cfe-commits mailing list
cfe-commits@lists.llvm.org
http://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
