blob: 380f45364622e7659170b514de08e95f02d4b188 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
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
|
