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