xref: /xnu-11417.121.6/tests/safe_allocation_src/size.cpp (revision a1e26a70f38d1d7daa7b49b258e2f8538ad81650)
1*a1e26a70SApple OSS Distributions //
2*a1e26a70SApple OSS Distributions // Tests for
3*a1e26a70SApple OSS Distributions //      size_t size() const;
4*a1e26a70SApple OSS Distributions //
5*a1e26a70SApple OSS Distributions 
6*a1e26a70SApple OSS Distributions #include <libkern/c++/safe_allocation.h>
7*a1e26a70SApple OSS Distributions #include <darwintest.h>
8*a1e26a70SApple OSS Distributions #include "test_utils.h"
9*a1e26a70SApple OSS Distributions #include <cstddef>
10*a1e26a70SApple OSS Distributions #include <type_traits>
11*a1e26a70SApple OSS Distributions #include <utility>
12*a1e26a70SApple OSS Distributions 
13*a1e26a70SApple OSS Distributions struct T {
14*a1e26a70SApple OSS Distributions 	int i;
15*a1e26a70SApple OSS Distributions };
16*a1e26a70SApple OSS Distributions 
17*a1e26a70SApple OSS Distributions template <typename T>
18*a1e26a70SApple OSS Distributions static void
tests()19*a1e26a70SApple OSS Distributions tests()
20*a1e26a70SApple OSS Distributions {
21*a1e26a70SApple OSS Distributions 	{
22*a1e26a70SApple OSS Distributions 		test_safe_allocation<T> const array(10, libkern::allocate_memory);
23*a1e26a70SApple OSS Distributions 		CHECK(array.size() == 10);
24*a1e26a70SApple OSS Distributions 	}
25*a1e26a70SApple OSS Distributions 	{
26*a1e26a70SApple OSS Distributions 		T* memory = reinterpret_cast<T*>(malloc_allocator::allocate(10 * sizeof(T)));
27*a1e26a70SApple OSS Distributions 		test_safe_allocation<T> const array(memory, 10, libkern::adopt_memory);
28*a1e26a70SApple OSS Distributions 		CHECK(array.size() == 10);
29*a1e26a70SApple OSS Distributions 	}
30*a1e26a70SApple OSS Distributions 	{
31*a1e26a70SApple OSS Distributions 		test_safe_allocation<T> const array(nullptr, 0, libkern::adopt_memory);
32*a1e26a70SApple OSS Distributions 		CHECK(array.size() == 0);
33*a1e26a70SApple OSS Distributions 	}
34*a1e26a70SApple OSS Distributions 	{
35*a1e26a70SApple OSS Distributions 		test_safe_allocation<T> const array;
36*a1e26a70SApple OSS Distributions 		CHECK(array.size() == 0);
37*a1e26a70SApple OSS Distributions 	}
38*a1e26a70SApple OSS Distributions 
39*a1e26a70SApple OSS Distributions 	{
40*a1e26a70SApple OSS Distributions 		using Size = decltype(std::declval<test_safe_allocation<T> const&>().size());
41*a1e26a70SApple OSS Distributions 		static_assert(std::is_same_v<Size, std::size_t>);
42*a1e26a70SApple OSS Distributions 	}
43*a1e26a70SApple OSS Distributions }
44*a1e26a70SApple OSS Distributions 
45*a1e26a70SApple OSS Distributions T_DECL(size, "safe_allocation.size", T_META_TAG_VM_PREFERRED) {
46*a1e26a70SApple OSS Distributions 	tests<T>();
47*a1e26a70SApple OSS Distributions 	tests<T const>();
48*a1e26a70SApple OSS Distributions }
49