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 Distributions tests()
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