136 lines
3.2 KiB
C++
136 lines
3.2 KiB
C++
#include "shellprv.h"
|
|
#include "ftascstr.h"
|
|
#include "ftassoc.h" //for now, until use CoCreateInstance
|
|
#include "ftenum.h" //for now, until use CoCreateInstance
|
|
|
|
HRESULT CFTAssocStore::_hresAccess = -1;
|
|
|
|
HRESULT CFTAssocStore::EnumAssocInfo(ASENUM asenumFlags, LPTSTR pszStr,
|
|
AIINIT aiinitFlags, IEnumAssocInfo** ppEnum)
|
|
{
|
|
//for now
|
|
*ppEnum = new CFTEnumAssocInfo();
|
|
|
|
if (*ppEnum)
|
|
{
|
|
(*ppEnum)->Init(asenumFlags, pszStr, aiinitFlags);
|
|
}
|
|
|
|
return (*ppEnum) ? S_OK : E_OUTOFMEMORY;
|
|
}
|
|
|
|
HRESULT CFTAssocStore::GetAssocInfo(LPTSTR pszStr, AIINIT aiinitFlags, IAssocInfo** ppAI)
|
|
{
|
|
HRESULT hres = E_FAIL;
|
|
|
|
*ppAI = new CFTAssocInfo();
|
|
|
|
if (*ppAI)
|
|
hres = (*ppAI)->Init(aiinitFlags, pszStr);
|
|
else
|
|
hres = E_OUTOFMEMORY;
|
|
|
|
return hres;
|
|
}
|
|
|
|
HRESULT CFTAssocStore::GetComplexAssocInfo(LPTSTR pszStr1, AIINIT aiinitFlags1,
|
|
LPTSTR pszStr2, AIINIT aiinitFlags2, IAssocInfo** ppAI)
|
|
{
|
|
HRESULT hres = E_FAIL;
|
|
|
|
*ppAI = new CFTAssocInfo();
|
|
|
|
if (*ppAI)
|
|
hres = (*ppAI)->InitComplex(aiinitFlags1, pszStr1, aiinitFlags2, pszStr2);
|
|
else
|
|
hres = E_OUTOFMEMORY;
|
|
|
|
return hres;
|
|
}
|
|
|
|
HRESULT CFTAssocStore::CheckAccess()
|
|
{
|
|
if (-1 == _hresAccess)
|
|
{
|
|
TCHAR szGUID[] = TEXT("{A4BFEC7C-F821-11d2-86BE-0000F8757D7E}");
|
|
DWORD dwDisp = 0;
|
|
int cTry = 0;
|
|
HKEY hkey;
|
|
|
|
_hresAccess = S_FALSE;
|
|
|
|
// we want to try this only two times
|
|
while ((S_FALSE == _hresAccess) && (cTry <= 1))
|
|
{
|
|
++cTry;
|
|
|
|
// we try to write a GUID to HKCR and delete it
|
|
if (ERROR_SUCCESS == RegCreateKeyEx(HKEY_CLASSES_ROOT, szGUID, 0, NULL, REG_OPTION_NON_VOLATILE,
|
|
MAXIMUM_ALLOWED, NULL, &hkey, &dwDisp))
|
|
{
|
|
// Did we really created a new key?
|
|
if (REG_CREATED_NEW_KEY == dwDisp)
|
|
{
|
|
// yes
|
|
RegCloseKey(hkey);
|
|
|
|
if (ERROR_SUCCESS == RegDeleteKey(HKEY_CLASSES_ROOT, szGUID))
|
|
{
|
|
_hresAccess = S_OK;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// No, there was already one, maybe we crashed right in the middle of this fct
|
|
// some other time in the past
|
|
|
|
// delete the key and try again
|
|
RegDeleteKey(HKEY_CLASSES_ROOT, szGUID);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return _hresAccess;
|
|
}
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
CFTAssocStore::CFTAssocStore()
|
|
{
|
|
_hresCoInit = SHCoInitialize();
|
|
|
|
//DLLAddRef();
|
|
}
|
|
|
|
CFTAssocStore::~CFTAssocStore()
|
|
{
|
|
//DLLRelease();
|
|
SHCoUninitialize(_hresCoInit);
|
|
}
|
|
|
|
//IUnknown methods
|
|
HRESULT CFTAssocStore::QueryInterface(REFIID riid, PVOID* ppv)
|
|
{
|
|
if (IsEqualIID(riid, IID_IUnknown))
|
|
*ppv = static_cast<IUnknown*>(this);
|
|
else
|
|
*ppv = static_cast<IAssocStore*>(this);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
ULONG CFTAssocStore::AddRef()
|
|
{
|
|
return InterlockedIncrement(&_cRef);
|
|
}
|
|
|
|
ULONG CFTAssocStore::Release()
|
|
{
|
|
ASSERT( 0 != _cRef );
|
|
ULONG cRef = InterlockedDecrement(&_cRef);
|
|
if ( 0 == cRef )
|
|
{
|
|
delete this;
|
|
}
|
|
return cRef;
|
|
} |