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