X-Git-Url: https://git.saurik.com/apple/security.git/blobdiff_plain/80e2389990082500d76eb566d4946be3e786c3ef..d8f41ccd20de16f8ebe2ccc84d47bf1cb2b26bbb:/Security/libsecurity_utilities/lib/refcount.h

diff --git a/Security/libsecurity_utilities/lib/refcount.h b/Security/libsecurity_utilities/lib/refcount.h
new file mode 100644
index 00000000..677376c5
--- /dev/null
+++ b/Security/libsecurity_utilities/lib/refcount.h
@@ -0,0 +1,179 @@
+/*
+ * Copyright (c) 2000-2004,2011-2014 Apple Inc. All Rights Reserved.
+ * 
+ * @APPLE_LICENSE_HEADER_START@
+ * 
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
+ * 
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ * 
+ * @APPLE_LICENSE_HEADER_END@
+ */
+
+
+/*
+ */
+#ifndef _SECURITY_REFCOUNT_H_
+#define _SECURITY_REFCOUNT_H_
+
+#include <security_utilities/threading.h>
+#include <libkern/OSAtomic.h>
+
+namespace Security {
+
+
+//
+// RefCount/RefPointer - a simple reference counting facility.
+//
+// To make an object reference-counted, inherit from RefCount. To track refcounted
+// objects, use RefPointer<TheType>, where TheType must inherit from RefCount.
+//
+// RefCount is thread safe - any number of threads can hold and manipulate references
+// in parallel. It does however NOT protect the contents of your object - just the
+// reference count itself. If you need to share your object contents, you must provide
+// appropriate locking yourself.
+//
+// There is no (thread safe) way to determine whether you are the only thread holding
+// a pointer to a particular RefCount object. Thus there is no (thread safe)
+// way to "demand copy" a RefCount subclass. Trust me; it's been tried. Don't.
+//
+
+#if !defined(DEBUG_REFCOUNTS)
+# define DEBUG_REFCOUNTS 1
+#endif
+
+#if DEBUG_REFCOUNTS
+# define RCDEBUG(_kind, _args...)	SECURITY_DEBUG_REFCOUNT_##_kind((void *)this, ##_args)
+#else
+# define RCDEBUG(kind)	/* nothing */
+#endif
+
+
+//
+// Base class for reference counted objects
+//
+class RefCount {	
+public:
+	RefCount() : mRefCount(0) { RCDEBUG(CREATE); }
+
+protected:
+	template <class T> friend class RefPointer;
+
+	void ref() const
+    {
+        OSAtomicIncrement32(&mRefCount);
+        RCDEBUG(UP, mRefCount);
+    }
+	
+    unsigned int unref() const
+    {
+        RCDEBUG(DOWN, mRefCount - 1);
+        return OSAtomicDecrement32(&mRefCount);
+    }
+	
+	// if you call this for anything but debug output, you will go to hell (free handbasket included)
+	unsigned int refCountForDebuggingOnly() const { return mRefCount; }
+
+private:
+    volatile mutable int32_t mRefCount;
+};
+
+
+//
+// A pointer type supported by reference counts.
+// T must be derived from RefCount.
+//
+template <class T>
+class RefPointer {
+	template <class Sub> friend class RefPointer; // share with other instances
+public:
+	RefPointer() : ptr(0) {}			// default to NULL pointer
+	RefPointer(const RefPointer& p) { if (p) p->ref(); ptr = p.ptr; }
+	RefPointer(T *p) { if (p) p->ref(); ptr = p; }
+	
+	template <class Sub>
+	RefPointer(const RefPointer<Sub>& p) { if (p) p->ref(); ptr = p.ptr; }
+    
+	~RefPointer() { release(); }
+
+	RefPointer& operator = (const RefPointer& p)	{ setPointer(p.ptr); return *this; }
+ 	RefPointer& operator = (T * p)					{ setPointer(p); return *this; }
+
+	template <class Sub>
+	RefPointer& operator = (const RefPointer<Sub>& p) { setPointer(p.ptr); return *this; }
+
+	// dereference operations
+    T* get () const				{ _check(); return ptr; }	// mimic auto_ptr
+	operator T * () const		{ _check(); return ptr; }
+	T * operator -> () const	{ _check(); return ptr; }
+	T & operator * () const		{ _check(); return *ptr; }
+
+protected:
+	void release_internal()
+    {
+        if (ptr && ptr->unref() == 0)
+        {
+            delete ptr;
+            ptr = NULL;
+        }
+    }
+	
+    void release()
+    {
+        StLock<Mutex> mutexLock(mMutex);
+        release_internal();
+    }
+    
+    void setPointer(T *p)
+    {
+        StLock<Mutex> mutexLock(mMutex);
+        if (p)
+        {
+            p->ref();
+        }
+        
+        release_internal();
+        ptr = p;
+    }
+	
+	void _check() const { }
+
+	T *ptr;
+    Mutex mMutex;
+};
+
+template <class T>
+bool operator <(const RefPointer<T> &r1, const RefPointer<T> &r2)
+{
+	T *p1 = r1.get(), *p2 = r2.get();
+	return p1 && p2 ? *p1 < *p2 : p1 < p2;
+}
+
+template <class T>
+bool operator ==(const RefPointer<T> &r1, const RefPointer<T> &r2)
+{
+	T *p1 = r1.get(), *p2 = r2.get();
+	return p1 && p2 ? *p1 == *p2 : p1 == p2;
+}
+
+template <class T>
+bool operator !=(const RefPointer<T> &r1, const RefPointer<T> &r2)
+{
+	T *p1 = r1.get(), *p2 = r2.get();
+	return p1 && p2 ? *p1 != *p2 : p1 != p2;
+}
+
+} // end namespace Security
+
+#endif // !_SECURITY_REFCOUNT_H_