1 files changed, 166 insertions, 0 deletions
diff --git a/src/utility/smartPtr.hpp b/src/utility/smartPtr.hpp
new file mode 100644
index 00000000..9905ae2f
--- /dev/null
+++ b/src/utility/smartPtr.hpp
@@ -0,0 +1,166 @@
+//
+// VMime library (http://vmime.sourceforge.net)
+// Copyright (C) 2002-2004 Vincent Richard <[email protected]>
+//
+// This program is free software; you can redistribute it and/or
+// modify it under the terms of the GNU General Public License as
+// published by the Free Software Foundation; either version 2 of
+// the License, or (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+// General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program; if not, write to the Free Software
+// Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+//
+
+#ifndef VMIME_UTILITY_SMARTPTR_HPP_INCLUDED
+#define VMIME_UTILITY_SMARTPTR_HPP_INCLUDED
+
+
+namespace vmime {
+namespace utility {
+
+
+/** Simple auto-delete pointer.
+  */
+
+template <class T>
+class auto_ptr
+{
+private:
+
+	T* const m_ptr;
+
+public:
+
+	auto_ptr(T* const ptr) : m_ptr(ptr) { }
+	~auto_ptr() { delete (m_ptr); }
+
+	operator T*() { return (m_ptr); }
+
+	T* const operator ->() { return (m_ptr); }
+	T& operator *() { return (*m_ptr); }
+};
+
+
+/** Smart auto-delete, referencable and copiable pointer.
+  */
+
+template <class T>
+class smart_ptr
+{
+private:
+
+	struct data
+	{
+		int refCount;
+		T* ptr;
+	};
+
+	data* m_data;
+
+
+	typedef std::map <T*, data*> MapType;
+	static MapType sm_map;
+
+public:
+
+	smart_ptr() : m_data(NULL) { }
+	smart_ptr(T* const ptr) : m_data(NULL) { if (ptr) { attach(ptr); } }
+	smart_ptr(smart_ptr& ptr) : m_data(NULL) { if (ptr.m_data) { attach(ptr); } }
+
+	~smart_ptr() { detach(); }
+
+	smart_ptr& operator=(smart_ptr& ptr)
+	{
+		attach(ptr);
+		return (*this);
+	}
+
+	smart_ptr& operator=(T* const ptr)
+	{
+		if (!ptr)
+			detach();
+		else
+			attach(ptr);
+
+		return (*this);
+	}
+
+	operator T*() { return (m_data ? m_data->ptr : NULL); }
+	operator const T*() { return (m_data ? m_data->ptr : NULL); }
+
+	T& operator *() { return (*(m_data->ptr)); }
+	T* operator ->() { return (m_data->ptr); }
+
+	const T* const ptr() const { return (m_data ? m_data->ptr : NULL); }
+	T* const ptr() { return (m_data ? m_data->ptr : NULL); }
+
+private:
+
+	void detach()
+	{
+		if (m_data)
+		{
+			if (m_data->refCount == 1)
+			{
+				typename MapType::iterator it = sm_map.find(m_data->ptr);
+				if (it != sm_map.end()) sm_map.erase(it);
+
+				delete (m_data->ptr);
+				delete (m_data);
+			}
+			else
+			{
+				m_data->refCount--;
+			}
+
+			m_data = NULL;
+		}
+	}
+
+	void attach(T* const ptr)
+	{
+		detach();
+
+		typename MapType::iterator it = sm_map.find(ptr);
+
+		if (it != sm_map.end())
+		{
+			(*it).second->refCount++;
+		}
+		else
+		{
+			m_data = new data;
+			m_data->refCount = 1;
+			m_data->ptr = ptr;
+
+			sm_map.insert(typename MapType::value_type(ptr, m_data));
+		}
+	}
+
+	void attach(smart_ptr <T>& ptr)
+	{
+		data* newData = ptr.m_data;
+		if (newData) newData->refCount++;
+
+		detach();
+
+		m_data = newData;
+	}
+};
+
+
+template <class T>
+typename smart_ptr <T>::MapType smart_ptr <T>::sm_map;
+
+
+} // utility
+} // vmime
+
+
+#endif // VMIME_UTILITY_SMARTPTR_HPP_INCLUDED