[apple/xnu.git] / tests / intrusive_shared_ptr_src / test_policy.h

#ifndef TESTS_INTRUSIVE_SHARED_PTR_TEST_POLICY_H
#define TESTS_INTRUSIVE_SHARED_PTR_TEST_POLICY_H

#include <libkern/c++/intrusive_shared_ptr.h>
#include <darwintest_utils.h>

struct test_policy {
	static inline int retain_count = 0;

	template <typename T>
	static void
	retain(T&)
	{
		++retain_count;
	}
	template <typename T>
	static void
	release(T&)
	{
		--retain_count;
	}
};

struct tracking_policy {
	static inline int retains = 0;
	static inline int releases = 0;
	static inline int refcount = 0;
	static inline bool hit_zero = false;

	static void
	reset()
	{
		retains = 0;
		releases = 0;
		refcount = 0;
		hit_zero = false;
	}

	template <typename T>
	static void
	retain(T&)
	{
		++retains;
		++refcount;
	}
	template <typename T>
	static void
	release(T&)
	{
		++releases;
		--refcount;
		if (refcount == 0) {
			hit_zero = true;
		}
	}
};

template <int>
struct dummy_policy {
	template <typename T>
	static void
	retain(T&)
	{
	}
	template <typename T>
	static void
	release(T&)
	{
	}
};

template <typename T>
using tracked_shared_ptr = libkern::intrusive_shared_ptr<T, tracking_policy>;

template <typename T>
using test_shared_ptr = libkern::intrusive_shared_ptr<T, test_policy>;

#define CHECK(...) T_ASSERT_TRUE((__VA_ARGS__), # __VA_ARGS__)

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