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