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