windows-nt/Source/XPSP1/NT/com/ole32/common/catinit.cxx

128 lines
3.6 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
#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;
}