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