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