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