1*8d741a5dSApple OSS Distributions // 2*8d741a5dSApple OSS Distributions // Make sure `safe_allocation` works nicely with the range-based for-loop. 3*8d741a5dSApple OSS Distributions // 4*8d741a5dSApple OSS Distributions 5*8d741a5dSApple OSS Distributions #include <libkern/c++/safe_allocation.h> 6*8d741a5dSApple OSS Distributions #include <darwintest.h> 7*8d741a5dSApple OSS Distributions #include "test_utils.h" 8*8d741a5dSApple OSS Distributions 9*8d741a5dSApple OSS Distributions struct T { 10*8d741a5dSApple OSS Distributions int i; 11*8d741a5dSApple OSS Distributions }; 12*8d741a5dSApple OSS Distributions 13*8d741a5dSApple OSS Distributions template <typename T> 14*8d741a5dSApple OSS Distributions static void tests()15*8d741a5dSApple OSS Distributionstests() 16*8d741a5dSApple OSS Distributions { 17*8d741a5dSApple OSS Distributions test_safe_allocation<T> array(10, libkern::allocate_memory); 18*8d741a5dSApple OSS Distributions for (T& element : array) { 19*8d741a5dSApple OSS Distributions element = T{3}; 20*8d741a5dSApple OSS Distributions } 21*8d741a5dSApple OSS Distributions 22*8d741a5dSApple OSS Distributions for (T const& element : array) { 23*8d741a5dSApple OSS Distributions CHECK(element.i == 3); 24*8d741a5dSApple OSS Distributions } 25*8d741a5dSApple OSS Distributions } 26*8d741a5dSApple OSS Distributions 27*8d741a5dSApple OSS Distributions T_DECL(usage_for_loop, "safe_allocation.usage.for_loop", T_META_TAG_VM_PREFERRED) { 28*8d741a5dSApple OSS Distributions tests<T>(); 29*8d741a5dSApple OSS Distributions } 30