1*4d495c6eSApple OSS Distributions #ifndef TESTS_INTRUSIVE_SHARED_PTR_TEST_POLICY_H 2*4d495c6eSApple OSS Distributions #define TESTS_INTRUSIVE_SHARED_PTR_TEST_POLICY_H 3*4d495c6eSApple OSS Distributions 4*4d495c6eSApple OSS Distributions #include <libkern/c++/intrusive_shared_ptr.h> 5*4d495c6eSApple OSS Distributions #include <darwintest_utils.h> 6*4d495c6eSApple OSS Distributions 7*4d495c6eSApple OSS Distributions struct test_policy { 8*4d495c6eSApple OSS Distributions static inline int retain_count = 0; 9*4d495c6eSApple OSS Distributions 10*4d495c6eSApple OSS Distributions template <typename T> 11*4d495c6eSApple OSS Distributions static void retaintest_policy12*4d495c6eSApple OSS Distributions retain(T&) 13*4d495c6eSApple OSS Distributions { 14*4d495c6eSApple OSS Distributions ++retain_count; 15*4d495c6eSApple OSS Distributions } 16*4d495c6eSApple OSS Distributions template <typename T> 17*4d495c6eSApple OSS Distributions static void releasetest_policy18*4d495c6eSApple OSS Distributions release(T&) 19*4d495c6eSApple OSS Distributions { 20*4d495c6eSApple OSS Distributions --retain_count; 21*4d495c6eSApple OSS Distributions } 22*4d495c6eSApple OSS Distributions }; 23*4d495c6eSApple OSS Distributions 24*4d495c6eSApple OSS Distributions struct tracking_policy { 25*4d495c6eSApple OSS Distributions static inline int retains = 0; 26*4d495c6eSApple OSS Distributions static inline int releases = 0; 27*4d495c6eSApple OSS Distributions static inline int refcount = 0; 28*4d495c6eSApple OSS Distributions static inline bool hit_zero = false; 29*4d495c6eSApple OSS Distributions 30*4d495c6eSApple OSS Distributions static void resettracking_policy31*4d495c6eSApple OSS Distributions reset() 32*4d495c6eSApple OSS Distributions { 33*4d495c6eSApple OSS Distributions retains = 0; 34*4d495c6eSApple OSS Distributions releases = 0; 35*4d495c6eSApple OSS Distributions refcount = 0; 36*4d495c6eSApple OSS Distributions hit_zero = false; 37*4d495c6eSApple OSS Distributions } 38*4d495c6eSApple OSS Distributions 39*4d495c6eSApple OSS Distributions template <typename T> 40*4d495c6eSApple OSS Distributions static void retaintracking_policy41*4d495c6eSApple OSS Distributions retain(T&) 42*4d495c6eSApple OSS Distributions { 43*4d495c6eSApple OSS Distributions ++retains; 44*4d495c6eSApple OSS Distributions ++refcount; 45*4d495c6eSApple OSS Distributions } 46*4d495c6eSApple OSS Distributions template <typename T> 47*4d495c6eSApple OSS Distributions static void releasetracking_policy48*4d495c6eSApple OSS Distributions release(T&) 49*4d495c6eSApple OSS Distributions { 50*4d495c6eSApple OSS Distributions ++releases; 51*4d495c6eSApple OSS Distributions --refcount; 52*4d495c6eSApple OSS Distributions if (refcount == 0) { 53*4d495c6eSApple OSS Distributions hit_zero = true; 54*4d495c6eSApple OSS Distributions } 55*4d495c6eSApple OSS Distributions } 56*4d495c6eSApple OSS Distributions }; 57*4d495c6eSApple OSS Distributions 58*4d495c6eSApple OSS Distributions template <int> 59*4d495c6eSApple OSS Distributions struct dummy_policy { 60*4d495c6eSApple OSS Distributions template <typename T> 61*4d495c6eSApple OSS Distributions static void retaindummy_policy62*4d495c6eSApple OSS Distributions retain(T&) 63*4d495c6eSApple OSS Distributions { 64*4d495c6eSApple OSS Distributions } 65*4d495c6eSApple OSS Distributions template <typename T> 66*4d495c6eSApple OSS Distributions static void releasedummy_policy67*4d495c6eSApple OSS Distributions release(T&) 68*4d495c6eSApple OSS Distributions { 69*4d495c6eSApple OSS Distributions } 70*4d495c6eSApple OSS Distributions }; 71*4d495c6eSApple OSS Distributions 72*4d495c6eSApple OSS Distributions template <typename T> 73*4d495c6eSApple OSS Distributions using tracked_shared_ptr = libkern::intrusive_shared_ptr<T, tracking_policy>; 74*4d495c6eSApple OSS Distributions 75*4d495c6eSApple OSS Distributions template <typename T> 76*4d495c6eSApple OSS Distributions using test_shared_ptr = libkern::intrusive_shared_ptr<T, test_policy>; 77*4d495c6eSApple OSS Distributions 78*4d495c6eSApple OSS Distributions #define CHECK(...) T_ASSERT_TRUE((__VA_ARGS__), # __VA_ARGS__) 79*4d495c6eSApple OSS Distributions 80*4d495c6eSApple OSS Distributions #endif // !TESTS_INTRUSIVE_SHARED_PTR_TEST_POLICY_H 81