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
|
#include "main.h"
#include "./storageHandler.h"
StorageHandler::StorageHandler(LPCWSTR pszKey, ifc_omstoragehelper::HandlerProc handlerProc, UINT flags)
: ref(1), key(NULL), handler(handlerProc), flags(flags)
{
if (0 != (flagCopyKey & flags))
key = Plugin_CopyString(pszKey);
else
key = (LPWSTR)pszKey;
}
StorageHandler::~StorageHandler()
{
if (0 != (flagCopyKey & flags))
Plugin_FreeString(key);
}
HRESULT StorageHandler::CreateInstance(LPCWSTR pszKey, ifc_omstoragehelper::HandlerProc handlerProc, UINT flags, StorageHandler **instance)
{
if (NULL == instance)
return E_POINTER;
if (NULL == pszKey || L'\0' == pszKey || NULL == handlerProc)
{
*instance = NULL;
return E_INVALIDARG;
}
*instance = new StorageHandler(pszKey, handlerProc, flags);
if (NULL == *instance) return E_OUTOFMEMORY;
return S_OK;
}
size_t StorageHandler::AddRef()
{
return InterlockedIncrement((LONG*)&ref);
}
size_t StorageHandler::Release()
{
if (0 == ref)
return ref;
LONG r = InterlockedDecrement((LONG*)&ref);
if (0 == r)
delete(this);
return r;
}
int StorageHandler::QueryInterface(GUID interface_guid, void **object)
{
if (NULL == object) return E_POINTER;
if (IsEqualIID(interface_guid, IFC_OmStorageHandler))
*object = static_cast<ifc_omstoragehandler*>(this);
else
{
*object = NULL;
return E_NOINTERFACE;
}
if (NULL == *object)
return E_UNEXPECTED;
AddRef();
return S_OK;
}
HRESULT StorageHandler::GetKey(LPCWSTR *ppKey)
{
if (NULL == ppKey) return E_POINTER;
*ppKey = key;
return S_OK;
}
void StorageHandler::Invoke(ifc_omservice *service, LPCWSTR pszKey, LPCWSTR pszValue)
{
handler(service, pszKey, pszValue);
}
#define CBCLASS StorageHandler
START_DISPATCH;
CB(ADDREF, AddRef)
CB(RELEASE, Release)
CB(QUERYINTERFACE, QueryInterface)
CB(API_GETKEY, GetKey)
VCB(API_INVOKE, Invoke)
END_DISPATCH;
#undef CBCLASS
|
