1*a1e26a70SApple OSS Distributions // 2*a1e26a70SApple OSS Distributions // Tests for 3*a1e26a70SApple OSS Distributions // explicit safe_allocation(T* data, size_t n, adopt_memory_t); 4*a1e26a70SApple OSS Distributions // 5*a1e26a70SApple OSS Distributions 6*a1e26a70SApple OSS Distributions #include <libkern/c++/safe_allocation.h> 7*a1e26a70SApple OSS Distributions #include <darwintest.h> 8*a1e26a70SApple OSS Distributions #include "test_utils.h" 9*a1e26a70SApple OSS Distributions 10*a1e26a70SApple OSS Distributions struct T { 11*a1e26a70SApple OSS Distributions int i; 12*a1e26a70SApple OSS Distributions }; 13*a1e26a70SApple OSS Distributions 14*a1e26a70SApple OSS Distributions template <typename T> 15*a1e26a70SApple OSS Distributions static void tests()16*a1e26a70SApple OSS Distributionstests() 17*a1e26a70SApple OSS Distributions { 18*a1e26a70SApple OSS Distributions { 19*a1e26a70SApple OSS Distributions T* memory = reinterpret_cast<T*>(tracking_allocator::allocate(10 * sizeof(T))); 20*a1e26a70SApple OSS Distributions tracking_allocator::reset(); 21*a1e26a70SApple OSS Distributions { 22*a1e26a70SApple OSS Distributions tracked_safe_allocation<T> array(memory, 10, libkern::adopt_memory); 23*a1e26a70SApple OSS Distributions CHECK(!tracking_allocator::did_allocate); 24*a1e26a70SApple OSS Distributions CHECK(array.data() == memory); 25*a1e26a70SApple OSS Distributions CHECK(array.size() == 10); 26*a1e26a70SApple OSS Distributions CHECK(array.begin() == array.data()); 27*a1e26a70SApple OSS Distributions CHECK(array.end() == array.data() + 10); 28*a1e26a70SApple OSS Distributions } 29*a1e26a70SApple OSS Distributions CHECK(tracking_allocator::deallocated_size == 10 * sizeof(T)); 30*a1e26a70SApple OSS Distributions } 31*a1e26a70SApple OSS Distributions } 32*a1e26a70SApple OSS Distributions 33*a1e26a70SApple OSS Distributions T_DECL(ctor_adopt, "safe_allocation.ctor.adopt", T_META_TAG_VM_PREFERRED) { 34*a1e26a70SApple OSS Distributions tests<T>(); 35*a1e26a70SApple OSS Distributions tests<T const>(); 36*a1e26a70SApple OSS Distributions } 37