18df9d23ea
This allows catching OOB accesses inside `unique_ptr<T[]>` when the size of the allocation is known. The size of the allocation can be known when the unique_ptr has been created with make_unique & friends or when the type necessitates an array cookie before the allocation. This is a re-aplpication of45a09d181which had been reverted inf11abac6due to unrelated CI failures.
133 lines
2.8 KiB
C++
133 lines
2.8 KiB
C++
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// <memory>
|
|
|
|
// unique_ptr
|
|
|
|
// pointer unique_ptr<T>::get() const noexcept;
|
|
// pointer unique_ptr<T[]>::get() const noexcept;
|
|
|
|
#include <memory>
|
|
#include <cassert>
|
|
#include <cstddef>
|
|
|
|
#include "test_macros.h"
|
|
|
|
template <class T>
|
|
TEST_CONSTEXPR_CXX23 void test_basic() {
|
|
// non-const element type
|
|
{
|
|
// non-const access
|
|
{
|
|
T* x = new T;
|
|
std::unique_ptr<T> ptr(x);
|
|
ASSERT_SAME_TYPE(decltype(ptr.get()), T*);
|
|
ASSERT_NOEXCEPT(ptr.get());
|
|
assert(ptr.get() == x);
|
|
}
|
|
|
|
// const access
|
|
{
|
|
T* x = new T;
|
|
std::unique_ptr<T> const ptr(x);
|
|
ASSERT_SAME_TYPE(decltype(ptr.get()), T*);
|
|
ASSERT_NOEXCEPT(ptr.get());
|
|
assert(ptr.get() == x);
|
|
}
|
|
}
|
|
|
|
// const element type
|
|
{
|
|
// non-const access
|
|
{
|
|
T* x = new T;
|
|
std::unique_ptr<T const> ptr(x);
|
|
ASSERT_SAME_TYPE(decltype(ptr.get()), T const*);
|
|
assert(ptr.get() == x);
|
|
}
|
|
|
|
// const access
|
|
{
|
|
T* x = new T;
|
|
std::unique_ptr<T const> const ptr(x);
|
|
ASSERT_SAME_TYPE(decltype(ptr.get()), T const*);
|
|
assert(ptr.get() == x);
|
|
}
|
|
}
|
|
|
|
// Same thing but for unique_ptr<T[]>
|
|
// non-const element type
|
|
{
|
|
// non-const access
|
|
{
|
|
T* x = new T[3];
|
|
std::unique_ptr<T[]> ptr(x);
|
|
ASSERT_SAME_TYPE(decltype(ptr.get()), T*);
|
|
ASSERT_NOEXCEPT(ptr.get());
|
|
assert(ptr.get() == x);
|
|
}
|
|
|
|
// const access
|
|
{
|
|
T* x = new T[3];
|
|
std::unique_ptr<T[]> const ptr(x);
|
|
ASSERT_SAME_TYPE(decltype(ptr.get()), T*);
|
|
ASSERT_NOEXCEPT(ptr.get());
|
|
assert(ptr.get() == x);
|
|
}
|
|
}
|
|
|
|
// const element type
|
|
{
|
|
// non-const access
|
|
{
|
|
T* x = new T[3];
|
|
std::unique_ptr<T const[]> ptr(x);
|
|
ASSERT_SAME_TYPE(decltype(ptr.get()), T const*);
|
|
assert(ptr.get() == x);
|
|
}
|
|
|
|
// const access
|
|
{
|
|
T* x = new T[3];
|
|
std::unique_ptr<T const[]> const ptr(x);
|
|
ASSERT_SAME_TYPE(decltype(ptr.get()), T const*);
|
|
assert(ptr.get() == x);
|
|
}
|
|
}
|
|
}
|
|
|
|
template <std::size_t Size>
|
|
struct WithSize {
|
|
char padding[Size];
|
|
};
|
|
|
|
TEST_CONSTEXPR_CXX23 bool test() {
|
|
test_basic<char>();
|
|
test_basic<int>();
|
|
test_basic<WithSize<1> >();
|
|
test_basic<WithSize<2> >();
|
|
test_basic<WithSize<3> >();
|
|
test_basic<WithSize<4> >();
|
|
test_basic<WithSize<8> >();
|
|
test_basic<WithSize<16> >();
|
|
test_basic<WithSize<256> >();
|
|
|
|
return true;
|
|
}
|
|
|
|
int main(int, char**) {
|
|
test();
|
|
#if TEST_STD_VER >= 23
|
|
static_assert(test());
|
|
#endif
|
|
|
|
return 0;
|
|
}
|