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