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