blob: bb52b5cbec834025e1dbb84a33d9dfa01b795c44 (
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
|
// **************************************************************************
// * 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) Zenju (zenju AT gmx DOT de) - All Rights Reserved *
// **************************************************************************
#ifndef SMART_COM_PTR_H
#define SMART_COM_PTR_H
#include <algorithm>
#include "win.h" //includes "windows.h" -> always include before other headers that also might include "windows.h"!
#include <Objbase.h>
namespace zen
{
/*
ComPtr: RAII class handling COM objects
Example:
--------
ComPtr<IUPnPDeviceFinder> devFinder;
if (FAILED(::CoCreateInstance(CLSID_UPnPDeviceFinder,
nullptr,
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(nullptr) {}
ComPtr(const ComPtr& other) : ptr(other.ptr) { if (ptr) ptr->AddRef(); }
ComPtr( ComPtr&& other) : ptr(other.ptr) { other.ptr = nullptr; }
~ComPtr() { if (ptr) ptr->Release(); }
ComPtr& operator=(ComPtr other) { swap(other); return *this; } //unifying assignment: no need for r-value reference assignment!
void swap(ComPtr& rhs) { std::swap(ptr, rhs.ptr); } //throw()
T** init() //get pointer for use with ::CoCreateInstance()
{
ComPtr<T>().swap(*this);
return &ptr;
}
T* get() const { return ptr; }
T* operator->() const { return ptr; }
T& operator* () const { return *ptr; }
T* release() //throw()
{
T* tmp = ptr;
ptr = nullptr;
return tmp;
}
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 != nullptr ? &ConversionToBool::dummy : nullptr; }
};
template <class S, class T>
ComPtr<S> com_dynamic_cast(const ComPtr<T>& other); //throw()
//################# 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
|
