131 lines
2.8 KiB
C++
131 lines
2.8 KiB
C++
|
#include "DataStoreCOM.h"
|
||
|
|
|
||
|
|
enum
|
||
|
|
{
|
||
|
|
DISP_DATASTORE_STORE,
|
||
|
|
DISP_DATASTORE_RETRIEVE,
|
||
|
|
DISP_DATASTORE_GENERATE_GUID,
|
||
|
|
};
|
||
|
|
|
||
|
|
#define CHECK_ID(str, id)\
|
||
|
|
if (CSTR_EQUAL == CompareStringW(lcid, NORM_IGNORECASE, rgszNames[i], -1, L##str, -1))\
|
||
|
|
{ rgdispid[i] = id; continue; }
|
||
|
|
|
||
|
|
HRESULT DataStoreCOM::GetIDsOfNames(REFIID riid, OLECHAR FAR* FAR* rgszNames, unsigned int cNames, LCID lcid, DISPID FAR* rgdispid)
|
||
|
|
{
|
||
|
|
bool unknowns = false;
|
||
|
|
for (unsigned int i = 0;i != cNames;i++)
|
||
|
|
{
|
||
|
|
CHECK_ID("Store", DISP_DATASTORE_STORE);
|
||
|
|
CHECK_ID("Retrieve", DISP_DATASTORE_RETRIEVE);
|
||
|
|
CHECK_ID("GenerateGUID", DISP_DATASTORE_GENERATE_GUID);
|
||
|
|
|
||
|
|
rgdispid[i] = DISPID_UNKNOWN;
|
||
|
|
unknowns = true;
|
||
|
|
|
||
|
|
}
|
||
|
|
if (unknowns)
|
||
|
|
return DISP_E_UNKNOWNNAME;
|
||
|
|
else
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT DataStoreCOM::GetTypeInfo(unsigned int itinfo, LCID lcid, ITypeInfo FAR* FAR* pptinfo)
|
||
|
|
{
|
||
|
|
return E_NOTIMPL;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT DataStoreCOM::GetTypeInfoCount(unsigned int FAR * pctinfo)
|
||
|
|
{
|
||
|
|
return E_NOTIMPL;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT DataStoreCOM::Invoke(DISPID dispid, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, EXCEPINFO FAR * pexecinfo, unsigned int FAR *puArgErr)
|
||
|
|
{
|
||
|
|
switch (dispid)
|
||
|
|
{
|
||
|
|
case DISP_DATASTORE_GENERATE_GUID:
|
||
|
|
{
|
||
|
|
GUID guid;
|
||
|
|
UuidCreate(&guid);
|
||
|
|
wchar_t *guid_string;
|
||
|
|
UuidToStringW(&guid, (unsigned short **)&guid_string);
|
||
|
|
BSTR tag = SysAllocString(guid_string);
|
||
|
|
RpcStringFreeW((unsigned short **)&guid_string);
|
||
|
|
VariantInit(pvarResult);
|
||
|
|
V_VT(pvarResult) = VT_BSTR;
|
||
|
|
V_BSTR(pvarResult) = tag;
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
case DISP_DATASTORE_STORE:
|
||
|
|
if (pdispparams->cArgs == 2)
|
||
|
|
{
|
||
|
|
GUID guid;
|
||
|
|
UuidFromStringW((unsigned short *)pdispparams->rgvarg[1].bstrVal, &guid);
|
||
|
|
|
||
|
|
SAFEARRAY *bufferArray=pdispparams->rgvarg[0].parray;
|
||
|
|
SAFEARRAY *store;
|
||
|
|
HRESULT hr = SafeArrayCopy(bufferArray, &store);
|
||
|
|
dataStore[guid]=store;
|
||
|
|
return hr;
|
||
|
|
|
||
|
|
}
|
||
|
|
else
|
||
|
|
return DISP_E_BADPARAMCOUNT;
|
||
|
|
case DISP_DATASTORE_RETRIEVE:
|
||
|
|
if (pdispparams->cArgs == 1)
|
||
|
|
{
|
||
|
|
GUID guid;
|
||
|
|
UuidFromStringW((unsigned short *)pdispparams->rgvarg[0].bstrVal, &guid);
|
||
|
|
|
||
|
|
SAFEARRAY *store = dataStore[guid];
|
||
|
|
if (store)
|
||
|
|
{
|
||
|
|
SAFEARRAY *bufferArray;
|
||
|
|
SafeArrayCopy(store, &bufferArray);
|
||
|
|
VariantInit(pvarResult);
|
||
|
|
V_VT(pvarResult) = VT_ARRAY|VT_UI1;
|
||
|
|
V_ARRAY(pvarResult) = bufferArray;
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
return E_FAIL;
|
||
|
|
|
||
|
|
}
|
||
|
|
else
|
||
|
|
return DISP_E_BADPARAMCOUNT;
|
||
|
|
}
|
||
|
|
return DISP_E_MEMBERNOTFOUND;
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHODIMP DataStoreCOM::QueryInterface(REFIID riid, PVOID *ppvObject)
|
||
|
|
{
|
||
|
|
if (!ppvObject)
|
||
|
|
return E_POINTER;
|
||
|
|
|
||
|
|
else if (IsEqualIID(riid, IID_IDispatch))
|
||
|
|
*ppvObject = (IDispatch *)this;
|
||
|
|
else if (IsEqualIID(riid, IID_IUnknown))
|
||
|
|
*ppvObject = this;
|
||
|
|
else
|
||
|
|
{
|
||
|
|
*ppvObject = NULL;
|
||
|
|
return E_NOINTERFACE;
|
||
|
|
}
|
||
|
|
|
||
|
|
AddRef();
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
ULONG DataStoreCOM::AddRef(void)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
ULONG DataStoreCOM::Release(void)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|