blob: b2c42e392812e631f0f8e00756afccfa2dbf656d (
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
|
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
// This class acts a smart pointer which calls the Release method on any object
// you place in it when the ReleaseHolder class falls out of scope. You may use it
// just like you would a standard pointer to a COM object (including if (foo),
// if (!foo), if (foo == 0), etc) except for two caveats:
// 1. This class never calls AddRef and it always calls Release when it
// goes out of scope.
// 2. You should never use & to try to get a pointer to a pointer unless
// you call Release first, or you will leak whatever this object contains
// prior to updating its internal pointer.
template<class T>
class ReleaseHolder
{
public:
ReleaseHolder()
: m_ptr(NULL)
{}
ReleaseHolder(T* ptr)
: m_ptr(ptr)
{}
~ReleaseHolder()
{
Release();
}
void operator=(T *ptr)
{
Release();
m_ptr = ptr;
}
T* operator->()
{
return m_ptr;
}
operator T*()
{
return m_ptr;
}
T** operator&()
{
return &m_ptr;
}
T* GetPtr() const
{
return m_ptr;
}
T* Detach()
{
T* pT = m_ptr;
m_ptr = NULL;
return pT;
}
void Release()
{
if (m_ptr != NULL)
{
m_ptr->Release();
m_ptr = NULL;
}
}
private:
T* m_ptr;
};
|