1 files changed, 123 insertions, 0 deletions
diff --git a/zen/com_ptr.h b/zen/com_ptr.h
new file mode 100644
index 00000000..380f4536
--- /dev/null
+++ b/zen/com_ptr.h
@@ -0,0 +1,123 @@
+// **************************************************************************
+// * This file is part of the FreeFileSync project. It is distributed under *
+// * GNU General Public License: http://www.gnu.org/licenses/gpl.html       *
+// * Copyright (C) 2008-2011 ZenJu (zhnmju123 AT gmx.de)                    *
+// **************************************************************************
+
+#ifndef SMART_COM_PTR_H
+#define SMART_COM_PTR_H
+
+#include <algorithm>
+#include <Objbase.h>
+
+namespace zen
+{
+/*
+ComPtr: RAII class handling COM objects
+
+Example:
+   --------
+    ComPtr<IUPnPDeviceFinder> devFinder;
+	if (FAILED(::CoCreateInstance(CLSID_UPnPDeviceFinder,
+                          NULL,
+                          CLSCTX_ALL,
+                          IID_PPV_ARGS(devFinder.init()))))
+					  return -1;
+
+	ComPtr<IEnumUnknown> devEnum = com_dynamic_cast<IEnumUnknown>(devColl);
+	if (!devEnum)
+		return -1;
+*/
+
+template <class T>
+class ComPtr
+{
+public:
+    ComPtr() : ptr(NULL) {}
+
+    ComPtr(const ComPtr&  other) : ptr(other.ptr) { if (ptr) ptr->AddRef(); }
+    ComPtr(      ComPtr&& other) : ptr(other.release()) {}
+
+    ComPtr& operator=(const ComPtr&  other) { ComPtr(other).swap(*this); return *this; }
+    ComPtr& operator=(      ComPtr&& other) { swap(other); return *this; }
+
+    ~ComPtr() { if (ptr) ptr->Release(); }
+
+    T** init() //get pointer for use with ::CoCreateInstance()
+    {
+        ComPtr<T>().swap(*this);
+        return &ptr;
+    }
+
+    T* get() const { return ptr; }
+
+    T* release() //throw()
+    {
+        T* tmp = ptr;
+        ptr = NULL;
+        return tmp;
+    }
+
+    void swap(ComPtr& rhs) { std::swap(ptr, rhs.ptr); } //throw()
+
+    T* operator->() const { return ptr; }
+
+private:
+    T* ptr;
+
+    struct ConversionToBool { int dummy; };
+public:
+    //use member pointer as implicit conversion to bool (C++ Templates - Vandevoorde/Josuttis; chapter 20)
+    operator int ConversionToBool::* () const { return ptr != NULL ? &ConversionToBool::dummy : NULL; }
+};
+
+
+template <class S, class T>
+ComPtr<S> com_dynamic_cast(const ComPtr<T>& other); //throw()
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+//################# Inline Implementation #############################
+
+//we cannot specialize std::swap() for a class template and are not allowed to overload it => offer swap in own namespace
+template <class T> inline
+void swap(zen::ComPtr<T>& lhs, zen::ComPtr<T>& rhs)
+{
+    lhs.swap(rhs);
+}
+
+
+template <class S, class T> inline
+ComPtr<S> com_dynamic_cast(const ComPtr<T>& other) //throw()
+{
+    ComPtr<S> outPtr;
+    if (other)
+        other->QueryInterface(IID_PPV_ARGS(outPtr.init()));
+    return outPtr;
+}
+}
+
+
+#endif //SMART_COM_PTR_H
+\ No newline at end of file