xref: /xnu-11417.101.15/tests/safe_allocation_src/usage.two_dimensions.cpp (revision e3723e1f17661b24996789d8afc084c0c3303b26)
1*e3723e1fSApple OSS Distributions //
2*e3723e1fSApple OSS Distributions // Make sure `safe_allocation` can be used to create a two-dimensional array.
3*e3723e1fSApple OSS Distributions //
4*e3723e1fSApple OSS Distributions // Note that we don't really recommend using that representation for two
5*e3723e1fSApple OSS Distributions // dimensional arrays because other representations are better, but it
6*e3723e1fSApple OSS Distributions // should at least work.
7*e3723e1fSApple OSS Distributions //
8*e3723e1fSApple OSS Distributions 
9*e3723e1fSApple OSS Distributions #include <libkern/c++/safe_allocation.h>
10*e3723e1fSApple OSS Distributions #include <darwintest.h>
11*e3723e1fSApple OSS Distributions #include "test_utils.h"
12*e3723e1fSApple OSS Distributions 
13*e3723e1fSApple OSS Distributions struct T {
14*e3723e1fSApple OSS Distributions 	int i;
15*e3723e1fSApple OSS Distributions };
16*e3723e1fSApple OSS Distributions 
17*e3723e1fSApple OSS Distributions template <typename T>
18*e3723e1fSApple OSS Distributions static void
tests()19*e3723e1fSApple OSS Distributions tests()
20*e3723e1fSApple OSS Distributions {
21*e3723e1fSApple OSS Distributions 	test_safe_allocation<test_safe_allocation<int> > array(10, libkern::allocate_memory);
22*e3723e1fSApple OSS Distributions 
23*e3723e1fSApple OSS Distributions 	for (int i = 0; i < 10; i++) {
24*e3723e1fSApple OSS Distributions 		array[i] = test_safe_allocation<int>(10, libkern::allocate_memory);
25*e3723e1fSApple OSS Distributions 		for (int j = 0; j < 10; ++j) {
26*e3723e1fSApple OSS Distributions 			array[i][j] = i + j;
27*e3723e1fSApple OSS Distributions 		}
28*e3723e1fSApple OSS Distributions 	}
29*e3723e1fSApple OSS Distributions 
30*e3723e1fSApple OSS Distributions 	for (int i = 0; i < 10; i++) {
31*e3723e1fSApple OSS Distributions 		for (int j = 0; j < 10; ++j) {
32*e3723e1fSApple OSS Distributions 			CHECK(array[i][j] == i + j);
33*e3723e1fSApple OSS Distributions 		}
34*e3723e1fSApple OSS Distributions 	}
35*e3723e1fSApple OSS Distributions }
36*e3723e1fSApple OSS Distributions 
37*e3723e1fSApple OSS Distributions T_DECL(usage_two_dimensions, "safe_allocation.usage.two_dimensions", T_META_TAG_VM_PREFERRED) {
38*e3723e1fSApple OSS Distributions 	tests<T>();
39*e3723e1fSApple OSS Distributions }
40