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