1 files changed, 69 insertions, 0 deletions
diff --git a/common/cotaskmem.h b/common/cotaskmem.h
new file mode 100644
index 0000000..bb141c4
--- /dev/null
+++ b/common/cotaskmem.h
@@ -0,0 +1,69 @@
+// cotaskmem.h
+//
+// WinDirStat - Directory Statistics
+// Copyright (C) 2003 Bernhard Seifert
+//
+// 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+//
+// This general purpose header is published under GPL with
+// the friendly permission of D'accord (www.daccord.net).
+//
+// 08.09.1998 - JR
+
+#ifndef _COTASKMEM_H
+#define _COTASKMEM_H
+
+// return type for 'CCoTaskMem<char *>::operator ->' is not a UDT or reference to a UDT.  Will produce errors if applied using infix notation
+#pragma warning(disable: 4284) 
+
+//
+// CCoTaskMem<>. Some Windows APIs return memory which must be freed with CoTaskMemFree().
+// This template does that in its destructor.
+//
+template<class T>
+class CCoTaskMem
+{
+// construction
+public:
+	CCoTaskMem( T lp = 0 )
+	{ p = lp; }
+	CCoTaskMem(const CCoTaskMem<T>&) // operator not allowed for CCoTaskMem 
+	{ _ASSERTE( 0 ); p = 0; }
+	~CCoTaskMem()
+	{ if( p ) CoTaskMemFree( p ); }
+
+	operator T() { return p; }
+	T& operator*() { _ASSERTE( p != NULL ); return p; }
+	//The assert on operator& usually indicates a bug.  If this is really
+	//what is needed, however, take the address of the p member explicitly.
+	T* operator&() 
+	{ _ASSERTE( p == NULL ); return &p; }
+	T operator->()
+	{ _ASSERTE( p != NULL ); return p; }
+	T operator = ( T lp ) 
+	{ if( p != NULL ) CoTaskMemFree( p ); p = lp; return p;}
+	T operator=( const CCoTaskMem<T>& lp ) // operator not allowed for CCoTaskMem 
+	{ _ASSERTE( 0 ); return p;}
+
+#if _MSC_VER>1020
+	bool operator!() { return (p == NULL); }
+#else
+	BOOL operator!() { return (p == NULL) ? TRUE : FALSE; }
+#endif
+	
+	T p;
+};
+
+#endif