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