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