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