1*a1e26a70SApple OSS Distributions // 2*a1e26a70SApple OSS Distributions // Tests for 3*a1e26a70SApple OSS Distributions // explicit safe_allocation(size_t n, allocate_memory_zero_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 test_safe_allocation<T> const array(10, libkern::allocate_memory_zero); 20*a1e26a70SApple OSS Distributions CHECK(array.data() != nullptr); 21*a1e26a70SApple OSS Distributions CHECK(array.size() == 10); 22*a1e26a70SApple OSS Distributions CHECK(array.begin() == array.data()); 23*a1e26a70SApple OSS Distributions CHECK(array.end() == array.data() + 10); 24*a1e26a70SApple OSS Distributions 25*a1e26a70SApple OSS Distributions auto const byteArray = reinterpret_cast<uint8_t const*>(array.data()); 26*a1e26a70SApple OSS Distributions size_t const byteLength = array.size() * sizeof(T); 27*a1e26a70SApple OSS Distributions for (size_t i = 0; i != byteLength; ++i) { 28*a1e26a70SApple OSS Distributions CHECK(byteArray[i] == 0); 29*a1e26a70SApple OSS Distributions } 30*a1e26a70SApple OSS Distributions } 31*a1e26a70SApple OSS Distributions } 32*a1e26a70SApple OSS Distributions 33*a1e26a70SApple OSS Distributions T_DECL(ctor_allocate_zero, "safe_allocation.ctor.allocate_zero", T_META_TAG_VM_PREFERRED) { 34*a1e26a70SApple OSS Distributions tests<T>(); 35*a1e26a70SApple OSS Distributions tests<T const>(); 36*a1e26a70SApple OSS Distributions tests<int>(); 37*a1e26a70SApple OSS Distributions tests<int const>(); 38*a1e26a70SApple OSS Distributions } 39