xref: /xnu-12377.1.9/tests/safe_allocation_src/data.cpp (revision f6217f891ac0bb64f3d375211650a4c1ff8ca1ea)
1*f6217f89SApple OSS Distributions //
2*f6217f89SApple OSS Distributions // Tests for
3*f6217f89SApple OSS Distributions //      T* data();
4*f6217f89SApple OSS Distributions //      T const* data() const;
5*f6217f89SApple OSS Distributions //
6*f6217f89SApple OSS Distributions 
7*f6217f89SApple OSS Distributions #include <libkern/c++/safe_allocation.h>
8*f6217f89SApple OSS Distributions #include <darwintest.h>
9*f6217f89SApple OSS Distributions #include "test_utils.h"
10*f6217f89SApple OSS Distributions 
11*f6217f89SApple OSS Distributions struct T {
12*f6217f89SApple OSS Distributions 	int i;
13*f6217f89SApple OSS Distributions };
14*f6217f89SApple OSS Distributions 
15*f6217f89SApple OSS Distributions template <typename T>
16*f6217f89SApple OSS Distributions static void
tests()17*f6217f89SApple OSS Distributions tests()
18*f6217f89SApple OSS Distributions {
19*f6217f89SApple OSS Distributions 	{
20*f6217f89SApple OSS Distributions 		test_safe_allocation<T> array(10, libkern::allocate_memory);
21*f6217f89SApple OSS Distributions 		CHECK(array.data() != nullptr);
22*f6217f89SApple OSS Distributions 	}
23*f6217f89SApple OSS Distributions 	{
24*f6217f89SApple OSS Distributions 		T* memory = reinterpret_cast<T*>(malloc_allocator::allocate(10 * sizeof(T)));
25*f6217f89SApple OSS Distributions 		test_safe_allocation<T> array(memory, 10, libkern::adopt_memory);
26*f6217f89SApple OSS Distributions 		T* data = array.data();
27*f6217f89SApple OSS Distributions 		CHECK(data == memory);
28*f6217f89SApple OSS Distributions 	}
29*f6217f89SApple OSS Distributions 	{
30*f6217f89SApple OSS Distributions 		T* memory = reinterpret_cast<T*>(malloc_allocator::allocate(10 * sizeof(T)));
31*f6217f89SApple OSS Distributions 		test_safe_allocation<T> const array(memory, 10, libkern::adopt_memory);
32*f6217f89SApple OSS Distributions 		T const* data = array.data();
33*f6217f89SApple OSS Distributions 		CHECK(data == memory);
34*f6217f89SApple OSS Distributions 	}
35*f6217f89SApple OSS Distributions }
36*f6217f89SApple OSS Distributions 
37*f6217f89SApple OSS Distributions T_DECL(data, "safe_allocation.data", T_META_TAG_VM_PREFERRED) {
38*f6217f89SApple OSS Distributions 	tests<T>();
39*f6217f89SApple OSS Distributions 	tests<T const>();
40*f6217f89SApple OSS Distributions }
41