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