1*e7776783SApple OSS Distributions // 2*e7776783SApple OSS Distributions // Tests for 3*e7776783SApple OSS Distributions // explicit safe_allocation(size_t n, allocate_memory_zero_t); 4*e7776783SApple OSS Distributions // 5*e7776783SApple OSS Distributions 6*e7776783SApple OSS Distributions #include <libkern/c++/safe_allocation.h> 7*e7776783SApple OSS Distributions #include <darwintest.h> 8*e7776783SApple OSS Distributions #include "test_utils.h" 9*e7776783SApple OSS Distributions #include <algorithm> 10*e7776783SApple OSS Distributions 11*e7776783SApple OSS Distributions struct T { 12*e7776783SApple OSS Distributions int i; 13*e7776783SApple OSS Distributions }; 14*e7776783SApple OSS Distributions 15*e7776783SApple OSS Distributions template <typename T> 16*e7776783SApple OSS Distributions static void tests()17*e7776783SApple OSS Distributionstests() 18*e7776783SApple OSS Distributions { 19*e7776783SApple OSS Distributions { 20*e7776783SApple OSS Distributions test_safe_allocation<T> const array(10, libkern::allocate_memory_zero); 21*e7776783SApple OSS Distributions CHECK(array.data() != nullptr); 22*e7776783SApple OSS Distributions CHECK(array.size() == 10); 23*e7776783SApple OSS Distributions CHECK(array.begin() == array.data()); 24*e7776783SApple OSS Distributions CHECK(array.end() == array.data() + 10); 25*e7776783SApple OSS Distributions 26*e7776783SApple OSS Distributions auto const byteArray = reinterpret_cast<uint8_t const*>(array.data()); 27*e7776783SApple OSS Distributions size_t const byteLength = array.size() * sizeof(T); 28*e7776783SApple OSS Distributions CHECK(std::all_of(byteArray, byteArray + byteLength, [](const auto& elem){ 29*e7776783SApple OSS Distributions return elem == 0; 30*e7776783SApple OSS Distributions })); 31*e7776783SApple OSS Distributions } 32*e7776783SApple OSS Distributions } 33*e7776783SApple OSS Distributions 34*e7776783SApple OSS Distributions T_DECL(ctor_allocate_zero, "safe_allocation.ctor.allocate_zero") { 35*e7776783SApple OSS Distributions tests<T>(); 36*e7776783SApple OSS Distributions tests<T const>(); 37*e7776783SApple OSS Distributions tests<int>(); 38*e7776783SApple OSS Distributions tests<int const>(); 39*e7776783SApple OSS Distributions } 40