128 lines
3.6 KiB
C++
128 lines
3.6 KiB
C++
|
#include <ole2int.h>
|
||
|
|
|
||
|
|
STDAPI GetCatalogHelper ( REFIID riid, void** ppv );
|
||
|
|
|
||
|
|
// Global catalog reference used in all activation
|
||
|
|
IComCatalog *gpCatalog = NULL;
|
||
|
|
IComCatalogSCM *gpCatalogSCM = NULL;
|
||
|
|
|
||
|
|
//+-------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: InitializeCatalogIfNecessary
|
||
|
|
//
|
||
|
|
// Synopsis: Load the catalog DLL and initialize the catalog interface
|
||
|
|
// if it hasn't already been done
|
||
|
|
//
|
||
|
|
// Arguments: none
|
||
|
|
//
|
||
|
|
// Returns: S_OK - no errors or inconsistencies found
|
||
|
|
// E_OUTOFMEMORY - couldn't create catalog object
|
||
|
|
// FAILED(hr) - return from CCI() of catalog or
|
||
|
|
// from QI() of catalog for IComCatalogSCM
|
||
|
|
//
|
||
|
|
//--------------------------------------------------------------------------
|
||
|
|
|
||
|
|
HRESULT InitializeCatalogIfNecessary()
|
||
|
|
{
|
||
|
|
if (gpCatalog != NULL && gpCatalogSCM != NULL)
|
||
|
|
{
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
IComCatalog *pCatalog = NULL;
|
||
|
|
IComCatalogSCM *pCatalogSCM = NULL;
|
||
|
|
void* pICEPRet = NULL;
|
||
|
|
|
||
|
|
// get the actual catalog object - abort on failure
|
||
|
|
hr = GetCatalogHelper (IID_IComCatalog, (void**) &pCatalog);
|
||
|
|
if ( FAILED(hr) || pCatalog == NULL)
|
||
|
|
{
|
||
|
|
ComDebOut((DEB_WARN,
|
||
|
|
"CCI of Catalog failed, hr=0x%x, pCatalog=0x%x",
|
||
|
|
hr, pCatalog));
|
||
|
|
}
|
||
|
|
if ( FAILED(hr) )
|
||
|
|
{
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
if ( pCatalog == NULL )
|
||
|
|
{
|
||
|
|
return E_OUTOFMEMORY;
|
||
|
|
}
|
||
|
|
|
||
|
|
// store the catalog object. guard against races by
|
||
|
|
// releasing if we didn't find a NULL in the variable
|
||
|
|
pICEPRet = InterlockedCompareExchangePointer ( (void **) &gpCatalog, (void *) pCatalog, NULL);
|
||
|
|
if ( pICEPRet != NULL )
|
||
|
|
{
|
||
|
|
pCatalog->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
// create the SCM object -- again, abort on failure
|
||
|
|
hr = GetCatalogHelper (IID_IComCatalogSCM, (void**) &pCatalogSCM);
|
||
|
|
if ( FAILED(hr) || pCatalog == NULL)
|
||
|
|
{
|
||
|
|
ComDebOut((DEB_WARN,
|
||
|
|
"CCI of Catalog failed, hr=0x%x, pCatalog=0x%x",
|
||
|
|
hr, pCatalog));
|
||
|
|
}
|
||
|
|
if ( FAILED(hr) )
|
||
|
|
{
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
if ( pCatalogSCM == NULL )
|
||
|
|
{
|
||
|
|
return E_OUTOFMEMORY;
|
||
|
|
}
|
||
|
|
|
||
|
|
// store the SCM catalog. Again, if it was already non-NULL, the exchange
|
||
|
|
// didn't happen, and you release the thing you just created.
|
||
|
|
pICEPRet = InterlockedCompareExchangePointer ( (void **) &gpCatalogSCM, (void *) pCatalogSCM, NULL );
|
||
|
|
if ( pICEPRet != NULL )
|
||
|
|
{
|
||
|
|
pCatalogSCM->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//+-------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: UninitializeCatalog
|
||
|
|
//
|
||
|
|
// Synopsis: Load the catalog DLL and initialize the catalog interface
|
||
|
|
// if it hasn't already been done
|
||
|
|
//
|
||
|
|
// Arguments: none
|
||
|
|
//
|
||
|
|
// Returns: S_OK - no errors or inconsistencies observed here
|
||
|
|
//
|
||
|
|
// Note: This can only be called from a place like wCoUninitialize(),
|
||
|
|
// where you're sure that nobody's using the catalog or doing
|
||
|
|
// anything out in the process.
|
||
|
|
//
|
||
|
|
//--------------------------------------------------------------------------
|
||
|
|
|
||
|
|
HRESULT UninitializeCatalog()
|
||
|
|
{
|
||
|
|
IComCatalog *pCatalog = NULL;
|
||
|
|
IComCatalogSCM *pCatalogSCM = NULL;
|
||
|
|
|
||
|
|
pCatalogSCM = (IComCatalogSCM *) InterlockedExchangePointer ((PVOID*)&gpCatalogSCM, NULL);
|
||
|
|
if ( pCatalogSCM != NULL )
|
||
|
|
{
|
||
|
|
pCatalogSCM->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
pCatalog = (IComCatalog *) InterlockedExchangePointer ((PVOID*)&gpCatalog, NULL);
|
||
|
|
if ( pCatalog != NULL )
|
||
|
|
{
|
||
|
|
pCatalog->FlushCache();
|
||
|
|
pCatalog->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
return S_OK;
|
||
|
|
}
|