Initial community commit
This commit is contained in:
Jef
parent
537bcbc862
commit
fc06254474
16440 changed files with 4239995 additions and 2 deletions
317
Src/omBrowser/storageIni.cpp
Normal file
317
Src/omBrowser/storageIni.cpp
Normal file
|
|
@ -0,0 +1,317 @@
|
|||
#include "main.h"
|
||||
#include "./storageIni.h"
|
||||
#include "./resource.h"
|
||||
#include "./ifc_omservice.h"
|
||||
#include "./loaderIni.h"
|
||||
#include "./enumIniFile.h"
|
||||
#include "./enumAsync.h"
|
||||
#include "./ifc_omservicehost.h"
|
||||
#include "./ifc_omservicehostext.h"
|
||||
#include "./ifc_omstoragehandlerenum.h"
|
||||
#include "./ifc_omstorageext.h"
|
||||
#include <strsafe.h>
|
||||
|
||||
OmStorageIni::OmStorageIni()
|
||||
: ref(1)
|
||||
{
|
||||
}
|
||||
|
||||
OmStorageIni::~OmStorageIni()
|
||||
{
|
||||
}
|
||||
|
||||
HRESULT OmStorageIni::CreateInstance(OmStorageIni **instance)
|
||||
{
|
||||
if (NULL == instance) return E_POINTER;
|
||||
|
||||
*instance = new OmStorageIni();
|
||||
if (NULL == *instance) return E_OUTOFMEMORY;
|
||||
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
size_t OmStorageIni::AddRef()
|
||||
{
|
||||
return InterlockedIncrement((LONG*)&ref);
|
||||
}
|
||||
|
||||
size_t OmStorageIni::Release()
|
||||
{
|
||||
if (0 == ref)
|
||||
return ref;
|
||||
|
||||
LONG r = InterlockedDecrement((LONG*)&ref);
|
||||
if (0 == r)
|
||||
delete(this);
|
||||
|
||||
return r;
|
||||
}
|
||||
|
||||
int OmStorageIni::QueryInterface(GUID interface_guid, void **object)
|
||||
{
|
||||
if (NULL == object) return E_POINTER;
|
||||
|
||||
if (IsEqualIID(interface_guid, IFC_OmStorage))
|
||||
*object = static_cast<ifc_omstorage*>(this);
|
||||
else if (IsEqualIID(interface_guid, STID_OmFileStorage))
|
||||
*object = static_cast<ifc_omfilestorage*>(this);
|
||||
else if (IsEqualIID(interface_guid, SUID_OmStorageIni))
|
||||
*object = static_cast<ifc_omstorage*>(this);
|
||||
else
|
||||
{
|
||||
*object = NULL;
|
||||
return E_NOINTERFACE;
|
||||
}
|
||||
|
||||
if (NULL == *object)
|
||||
return E_UNEXPECTED;
|
||||
|
||||
AddRef();
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
HRESULT OmStorageIni::GetId(GUID *storageUid)
|
||||
{
|
||||
if (NULL == storageUid) return E_POINTER;
|
||||
*storageUid = SUID_OmStorageIni;
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
HRESULT OmStorageIni::GetType(GUID *storageType)
|
||||
{
|
||||
if (NULL == storageType) return E_POINTER;
|
||||
*storageType = STID_OmFileStorage;
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
UINT OmStorageIni::GetCapabilities()
|
||||
{
|
||||
return capLoad | capSave | capDelete | capReload | capPublic;
|
||||
}
|
||||
|
||||
HRESULT OmStorageIni::GetDescription(LPWSTR pszBuffer, UINT cchBufferMax)
|
||||
{
|
||||
if (NULL == pszBuffer) return E_POINTER;
|
||||
Plugin_LoadString(IDS_STORAGE_INI, pszBuffer, cchBufferMax);
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
HRESULT OmStorageIni::Load(LPCWSTR pszAddress, ifc_omservicehost *host, ifc_omserviceenum **ppEnum)
|
||||
{
|
||||
if (NULL == ppEnum) return E_POINTER;
|
||||
*ppEnum = NULL;
|
||||
|
||||
if (NULL == pszAddress || L'\0' == *pszAddress)
|
||||
return E_INVALIDARG;
|
||||
|
||||
return EnumIniFile::CreateInstance(pszAddress, host, (EnumIniFile**)ppEnum);
|
||||
}
|
||||
|
||||
HRESULT OmStorageIni::Save(ifc_omservice **serviceList, ULONG listCount, UINT saveFlags, ULONG *savedCount)
|
||||
{
|
||||
if (NULL != savedCount)
|
||||
*savedCount = 0;
|
||||
|
||||
if(NULL == serviceList && 0 != listCount)
|
||||
return E_INVALIDARG;
|
||||
|
||||
HRESULT hr = S_OK;
|
||||
ULONG saved = 0;
|
||||
|
||||
LoaderIni loader;
|
||||
|
||||
for(ULONG i = 0; i < listCount; i++)
|
||||
{
|
||||
ifc_omservice *service = serviceList[i];
|
||||
if (NULL == service) continue;
|
||||
|
||||
if (FAILED(loader.Save(service, saveFlags)))
|
||||
hr = E_FAIL;
|
||||
else
|
||||
saved++;
|
||||
}
|
||||
|
||||
if (NULL != savedCount)
|
||||
*savedCount = saved++;
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT OmStorageIni::Delete(ifc_omservice **serviceList, ULONG listCount, ULONG *deletedCount)
|
||||
{
|
||||
if (NULL == serviceList)
|
||||
{
|
||||
if (NULL != deletedCount) *deletedCount = 0;
|
||||
return E_INVALIDARG;
|
||||
}
|
||||
|
||||
WCHAR szBuffer[MAX_PATH * 2] = {0};
|
||||
ULONG deleted = 0;
|
||||
|
||||
for(ULONG i = 0; i < listCount; i++)
|
||||
{
|
||||
if (SUCCEEDED(serviceList[i]->GetAddress(szBuffer, ARRAYSIZE(szBuffer))) &&
|
||||
0 != DeleteFile(szBuffer))
|
||||
{
|
||||
deleted++;
|
||||
}
|
||||
}
|
||||
|
||||
if (NULL != deletedCount)
|
||||
*deletedCount = deleted;
|
||||
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
HRESULT OmStorageIni::Reload(ifc_omservice **serviceList, ULONG listCount, ULONG *reloadedCount)
|
||||
{
|
||||
if (NULL != reloadedCount)
|
||||
*reloadedCount = 0;
|
||||
|
||||
if(NULL == serviceList && 0 != listCount)
|
||||
return E_INVALIDARG;
|
||||
|
||||
HRESULT hr = S_OK;
|
||||
ULONG loaded = 0;
|
||||
|
||||
LoaderIni loader;
|
||||
|
||||
for(ULONG i = 0; i < listCount; i++)
|
||||
{
|
||||
ifc_omservice *service = serviceList[i];
|
||||
if (NULL == service) continue;
|
||||
|
||||
loader.RegisterHandlers(NULL);
|
||||
|
||||
ifc_omservicehostext *serviceExt = NULL;
|
||||
if (service->QueryInterface(IFC_OmServiceHostExt, (void**)&serviceExt))
|
||||
{
|
||||
ifc_omservicehost *serviceHost = NULL;
|
||||
if (SUCCEEDED(serviceExt->GetHost(&serviceHost)) && serviceHost != NULL)
|
||||
{
|
||||
ifc_omstorageext *storageExt = NULL;
|
||||
if (SUCCEEDED(serviceHost->QueryInterface(IFC_OmStorageExt, (void**)&storageExt)) && storageExt != NULL)
|
||||
{
|
||||
ifc_omstoragehandlerenum *handlerEnum = NULL;
|
||||
if (SUCCEEDED(storageExt->Enumerate(&SUID_OmStorageIni, &handlerEnum)) && handlerEnum != NULL)
|
||||
{
|
||||
loader.RegisterHandlers(handlerEnum);
|
||||
handlerEnum->Release();
|
||||
}
|
||||
storageExt->Release();
|
||||
}
|
||||
serviceHost->Release();
|
||||
}
|
||||
serviceExt->Release();
|
||||
}
|
||||
|
||||
if (FAILED(loader.Reload(service)))
|
||||
hr = E_FAIL;
|
||||
else
|
||||
loaded++;
|
||||
}
|
||||
|
||||
if (NULL != reloadedCount)
|
||||
*reloadedCount = loaded++;
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT OmStorageIni::BeginLoad(LPCWSTR pszAddress, ifc_omservicehost *serviceHost, ifc_omstorageasync::AsyncCallback callback, void *data, ifc_omstorageasync **async)
|
||||
{
|
||||
if (NULL == async)
|
||||
return E_POINTER;
|
||||
|
||||
ifc_omserviceenum *enumerator = NULL;
|
||||
HRESULT hr = Load(pszAddress, serviceHost, &enumerator);
|
||||
if (SUCCEEDED(hr) && enumerator != NULL)
|
||||
{
|
||||
EnumAsyncWrapper *asyncWrapper = NULL;
|
||||
hr = EnumAsyncWrapper::CreateInstance(enumerator, &asyncWrapper);
|
||||
if (SUCCEEDED(hr) && asyncWrapper != NULL)
|
||||
{
|
||||
asyncWrapper->SetCallback(callback);
|
||||
asyncWrapper->SetData(data);
|
||||
*async = asyncWrapper;
|
||||
|
||||
hr = asyncWrapper->BeginEnumerate();
|
||||
if (FAILED(hr))
|
||||
{
|
||||
*async = NULL;
|
||||
asyncWrapper->Release();
|
||||
}
|
||||
}
|
||||
enumerator->Release();
|
||||
}
|
||||
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT OmStorageIni::EndLoad(ifc_omstorageasync *async, ifc_omserviceenum **ppEnum)
|
||||
{
|
||||
EnumAsyncWrapper *asyncWrapper = (EnumAsyncWrapper*)async;
|
||||
if (NULL == asyncWrapper) return E_INVALIDARG;
|
||||
|
||||
if (NULL != ppEnum)
|
||||
*ppEnum = NULL;
|
||||
|
||||
UINT state = 0;
|
||||
if (FAILED(async->GetState(&state)) || ifc_omstorageasync::stateCompleted != state)
|
||||
{
|
||||
HANDLE completed = NULL;
|
||||
if (FAILED(async->GetWaitHandle(&completed)))
|
||||
return E_UNEXPECTED;
|
||||
|
||||
while(WAIT_OBJECT_0 != WaitForSingleObjectEx(completed, INFINITE, TRUE));
|
||||
CloseHandle(completed);
|
||||
}
|
||||
|
||||
HRESULT hr = asyncWrapper->GetResultCode();
|
||||
if (SUCCEEDED(hr))
|
||||
{
|
||||
if (NULL != ppEnum)
|
||||
{
|
||||
hr = asyncWrapper->GetServiceList(ppEnum);
|
||||
}
|
||||
}
|
||||
return hr;
|
||||
}
|
||||
|
||||
HRESULT OmStorageIni::RequestAbort(ifc_omstorageasync *async, BOOL fDrop)
|
||||
{
|
||||
EnumAsyncWrapper *asyncWrapper = (EnumAsyncWrapper*)async;
|
||||
if (NULL == asyncWrapper) return E_INVALIDARG;
|
||||
return asyncWrapper->RequestAbort(fDrop);
|
||||
}
|
||||
|
||||
HRESULT OmStorageIni::GetFilter(LPWSTR pszBuffer, UINT cchBufferMax)
|
||||
{
|
||||
if (NULL == pszBuffer) return E_POINTER;
|
||||
return StringCchCopy(pszBuffer, cchBufferMax, L"*.ini");
|
||||
}
|
||||
|
||||
#define CBCLASS OmStorageIni
|
||||
START_MULTIPATCH;
|
||||
START_PATCH(MPIID_OMSTORAGE)
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, ADDREF, AddRef);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, RELEASE, Release);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, QUERYINTERFACE, QueryInterface);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, API_GETID, GetId);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, API_GETTYPE, GetType);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, API_GETCAPABILITIES, GetCapabilities);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, API_GETDESCRIPTION, GetDescription);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, API_LOAD, Load);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, API_SAVE, Save);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, API_DELETE, Delete);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, API_RELOAD, Reload);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, API_BEGINLOAD, BeginLoad);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, API_ENDLOAD, EndLoad);
|
||||
M_CB(MPIID_OMSTORAGE, ifc_omstorage, API_REQUESTABORT, RequestAbort);
|
||||
NEXT_PATCH(MPIID_OMFILESTORAGE)
|
||||
M_CB(MPIID_OMFILESTORAGE, ifc_omfilestorage, ADDREF, AddRef);
|
||||
M_CB(MPIID_OMFILESTORAGE, ifc_omfilestorage, RELEASE, Release);
|
||||
M_CB(MPIID_OMFILESTORAGE, ifc_omfilestorage, QUERYINTERFACE, QueryInterface);
|
||||
M_CB(MPIID_OMFILESTORAGE, ifc_omfilestorage, API_GETFILTER, GetFilter);
|
||||
END_PATCH
|
||||
END_MULTIPATCH;
|
||||
#undef CBCLASS
|
||||
Loading…
Add table
Add a link
Reference in a new issue