1*a325d9c4SApple OSS Distributions // 2*a325d9c4SApple OSS Distributions // Tests for 3*a325d9c4SApple OSS Distributions // explicit safe_allocation(size_t n, allocate_memory_t); 4*a325d9c4SApple OSS Distributions // 5*a325d9c4SApple OSS Distributions 6*a325d9c4SApple OSS Distributions #include <libkern/c++/safe_allocation.h> 7*a325d9c4SApple OSS Distributions #include <darwintest.h> 8*a325d9c4SApple OSS Distributions #include "test_utils.h" 9*a325d9c4SApple OSS Distributions #include <cstddef> 10*a325d9c4SApple OSS Distributions #include <limits> 11*a325d9c4SApple OSS Distributions 12*a325d9c4SApple OSS Distributions struct T { 13*a325d9c4SApple OSS Distributions int i; 14*a325d9c4SApple OSS Distributions }; 15*a325d9c4SApple OSS Distributions 16*a325d9c4SApple OSS Distributions struct TrackInit { 17*a325d9c4SApple OSS Distributions bool initialized; TrackInitTrackInit18*a325d9c4SApple OSS Distributions TrackInit() : initialized(true) 19*a325d9c4SApple OSS Distributions { 20*a325d9c4SApple OSS Distributions } 21*a325d9c4SApple OSS Distributions }; 22*a325d9c4SApple OSS Distributions 23*a325d9c4SApple OSS Distributions template <typename T> 24*a325d9c4SApple OSS Distributions static void tests()25*a325d9c4SApple OSS Distributionstests() 26*a325d9c4SApple OSS Distributions { 27*a325d9c4SApple OSS Distributions { 28*a325d9c4SApple OSS Distributions tracking_allocator::reset(); 29*a325d9c4SApple OSS Distributions { 30*a325d9c4SApple OSS Distributions tracked_safe_allocation<T> array(10, libkern::allocate_memory); 31*a325d9c4SApple OSS Distributions CHECK(tracking_allocator::allocated_size == 10 * sizeof(T)); 32*a325d9c4SApple OSS Distributions CHECK(array.data() != nullptr); 33*a325d9c4SApple OSS Distributions CHECK(array.size() == 10); 34*a325d9c4SApple OSS Distributions CHECK(array.begin() == array.data()); 35*a325d9c4SApple OSS Distributions CHECK(array.end() == array.data() + 10); 36*a325d9c4SApple OSS Distributions } 37*a325d9c4SApple OSS Distributions CHECK(tracking_allocator::deallocated_size == 10 * sizeof(T)); 38*a325d9c4SApple OSS Distributions } 39*a325d9c4SApple OSS Distributions 40*a325d9c4SApple OSS Distributions // Check with a huge number of elements that will overflow size_t 41*a325d9c4SApple OSS Distributions { 42*a325d9c4SApple OSS Distributions std::size_t max_n = std::numeric_limits<std::size_t>::max() / sizeof(T); 43*a325d9c4SApple OSS Distributions tracking_allocator::reset(); 44*a325d9c4SApple OSS Distributions 45*a325d9c4SApple OSS Distributions { 46*a325d9c4SApple OSS Distributions tracked_safe_allocation<T> array(max_n + 1, libkern::allocate_memory); 47*a325d9c4SApple OSS Distributions CHECK(array.data() == nullptr); 48*a325d9c4SApple OSS Distributions CHECK(array.size() == 0); 49*a325d9c4SApple OSS Distributions CHECK(array.begin() == array.end()); 50*a325d9c4SApple OSS Distributions CHECK(!tracking_allocator::did_allocate); 51*a325d9c4SApple OSS Distributions } 52*a325d9c4SApple OSS Distributions CHECK(!tracking_allocator::did_deallocate); 53*a325d9c4SApple OSS Distributions } 54*a325d9c4SApple OSS Distributions } 55*a325d9c4SApple OSS Distributions 56*a325d9c4SApple OSS Distributions T_DECL(ctor_allocate, "safe_allocation.ctor.allocate") { 57*a325d9c4SApple OSS Distributions tests<T>(); 58*a325d9c4SApple OSS Distributions tests<T const>(); 59*a325d9c4SApple OSS Distributions 60*a325d9c4SApple OSS Distributions // Make sure we value-initialize elements 61*a325d9c4SApple OSS Distributions { 62*a325d9c4SApple OSS Distributions tracked_safe_allocation<TrackInit> array(10, libkern::allocate_memory); 63*a325d9c4SApple OSS Distributions for (int i = 0; i != 10; ++i) { 64*a325d9c4SApple OSS Distributions CHECK(array[i].initialized == true); 65*a325d9c4SApple OSS Distributions } 66*a325d9c4SApple OSS Distributions } 67*a325d9c4SApple OSS Distributions } 68