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