blob: 63127d4df5a327be0124eaeb577bcd1b380d657e (
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
124
125
126
127
128
129
130
131
132
133
134
135
|
#pragma once
class ContextMap : public IUnknown
{
public:
ContextMap() :
m_cRef(1)
{
}
~ContextMap()
{
CComCritSecLock<CComAutoCriticalSection> Lock(m_CriticalSection);
IUnknown* pUnk = NULL;
POSITION elementPosition = NULL;
elementPosition = m_Map.GetStartPosition();
while(elementPosition != NULL)
{
pUnk = m_Map.GetNextValue(elementPosition);
if(pUnk != NULL)
{
pUnk->Release();
}
}
}
public: // IUnknown
ULONG __stdcall AddRef()
{
InterlockedIncrement((long*) &m_cRef);
return m_cRef;
}
_At_(this, __drv_freesMem(Mem))
ULONG __stdcall Release()
{
ULONG ulRefCount = m_cRef - 1;
if (InterlockedDecrement((long*) &m_cRef) == 0)
{
delete this;
return 0;
}
return ulRefCount;
}
HRESULT __stdcall QueryInterface(
REFIID riid,
void** ppv)
{
HRESULT hr = S_OK;
if(riid == IID_IUnknown)
{
*ppv = static_cast<IUnknown*>(this);
AddRef();
}
else
{
*ppv = NULL;
hr = E_NOINTERFACE;
}
return hr;
}
public: // Context accessor methods
// If successfull, this method AddRef's the context
HRESULT Add(
_In_ const CAtlStringW& key,
_In_ IUnknown* pContext)
{
CComCritSecLock<CComAutoCriticalSection> Lock(m_CriticalSection);
HRESULT hr = S_OK;
// Insert this into the map
POSITION elementPosition = m_Map.SetAt(key, pContext);
if(elementPosition != NULL)
{
// AddRef since we are holding onto it
pContext->AddRef();
}
else
{
hr = E_OUTOFMEMORY;
}
return hr;
}
void Remove(
_In_ const CAtlStringW& key)
{
CComCritSecLock<CComAutoCriticalSection> Lock(m_CriticalSection);
// Get the element
IUnknown* pContext = NULL;
if (m_Map.Lookup(key, pContext) == true)
{
// Remove the entry for it
m_Map.RemoveKey(key);
// Release it
pContext->Release();
}
}
// Returns the context pointer. If not found, return value is NULL.
// If non-NULL, caller is responsible for Releasing when it is done,
// since this method will AddRef the context.
IUnknown* GetContext(
_In_ const CAtlStringW& key)
{
CComCritSecLock<CComAutoCriticalSection> Lock(m_CriticalSection);
// Get the element
IUnknown* pContext = NULL;
if (m_Map.Lookup(key, pContext) == true)
{
// AddRef
pContext->AddRef();
}
return pContext;
}
private:
CComAutoCriticalSection m_CriticalSection;
CAtlMap<CAtlStringW, IUnknown*> m_Map;
DWORD m_cRef;
};
|
