1*fdd8201dSApple OSS Distributions // 2*fdd8201dSApple OSS Distributions // Tests for 3*fdd8201dSApple OSS Distributions // ~safe_allocation(); 4*fdd8201dSApple OSS Distributions // 5*fdd8201dSApple OSS Distributions 6*fdd8201dSApple OSS Distributions #include <libkern/c++/safe_allocation.h> 7*fdd8201dSApple OSS Distributions #include <darwintest.h> 8*fdd8201dSApple OSS Distributions #include "test_utils.h" 9*fdd8201dSApple OSS Distributions 10*fdd8201dSApple OSS Distributions struct TriviallyDestructible { 11*fdd8201dSApple OSS Distributions int i; 12*fdd8201dSApple OSS Distributions }; 13*fdd8201dSApple OSS Distributions 14*fdd8201dSApple OSS Distributions struct NonTriviallyDestructible { 15*fdd8201dSApple OSS Distributions int i; ~NonTriviallyDestructibleNonTriviallyDestructible16*fdd8201dSApple OSS Distributions ~NonTriviallyDestructible() 17*fdd8201dSApple OSS Distributions { 18*fdd8201dSApple OSS Distributions } 19*fdd8201dSApple OSS Distributions }; 20*fdd8201dSApple OSS Distributions 21*fdd8201dSApple OSS Distributions template <typename T> 22*fdd8201dSApple OSS Distributions static void tests()23*fdd8201dSApple OSS Distributionstests() 24*fdd8201dSApple OSS Distributions { 25*fdd8201dSApple OSS Distributions // Destroy a non-null allocation 26*fdd8201dSApple OSS Distributions { 27*fdd8201dSApple OSS Distributions { 28*fdd8201dSApple OSS Distributions tracked_safe_allocation<T> array(10, libkern::allocate_memory); 29*fdd8201dSApple OSS Distributions tracking_allocator::reset(); 30*fdd8201dSApple OSS Distributions } 31*fdd8201dSApple OSS Distributions CHECK(tracking_allocator::deallocated_size == 10 * sizeof(T)); 32*fdd8201dSApple OSS Distributions } 33*fdd8201dSApple OSS Distributions 34*fdd8201dSApple OSS Distributions // Destroy a null allocation 35*fdd8201dSApple OSS Distributions { 36*fdd8201dSApple OSS Distributions { 37*fdd8201dSApple OSS Distributions tracked_safe_allocation<T> array = nullptr; 38*fdd8201dSApple OSS Distributions tracking_allocator::reset(); 39*fdd8201dSApple OSS Distributions } 40*fdd8201dSApple OSS Distributions CHECK(!tracking_allocator::did_deallocate); 41*fdd8201dSApple OSS Distributions } 42*fdd8201dSApple OSS Distributions } 43*fdd8201dSApple OSS Distributions 44*fdd8201dSApple OSS Distributions T_DECL(dtor, "safe_allocation.dtor") { 45*fdd8201dSApple OSS Distributions tests<TriviallyDestructible>(); 46*fdd8201dSApple OSS Distributions tests<TriviallyDestructible const>(); 47*fdd8201dSApple OSS Distributions 48*fdd8201dSApple OSS Distributions tests<NonTriviallyDestructible>(); 49*fdd8201dSApple OSS Distributions tests<NonTriviallyDestructible const>(); 50*fdd8201dSApple OSS Distributions } 51