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