mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
346 lines
8.5 KiB
346 lines
8.5 KiB
/*==========================================================================
|
|
*
|
|
* Copyright (C) 2000 Microsoft Corporation. All Rights Reserved.
|
|
*
|
|
* File: comutil.cpp
|
|
* Content: Contains implementation of COM helper functions for DPLAY8 project.
|
|
*@@BEGIN_MSINTERNAL
|
|
* History:
|
|
* Date By Reason
|
|
* ==== == ======
|
|
* 06/07/00 rmt Created
|
|
* 06/15/2000 rmt Fixed small bug in COM_CoCreateInstance which was causing AV
|
|
* 06/27/00 rmt Added abstraction for COM_Co(Un)Initialize
|
|
* 07/06/00 rmt Modified to match updated creg usage
|
|
* 08/08/2000 rmt Bug #41736 - AV in call to lstrcpy by COM_GetDllName
|
|
* 01/11/2001 rmt MANBUG #48487 - DPLAY: Crashes if CoCreate() isn't called.
|
|
* 03/14/2001 rmt WINBUG #342420 - Restore COM emulation layer to operation.
|
|
*@@END_MSINTERNAL
|
|
*
|
|
***************************************************************************/
|
|
|
|
#include "dncmni.h"
|
|
#include "comutil.h"
|
|
|
|
#undef DPF_SUBCOMP
|
|
#define DPF_SUBCOMP DN_SUBCOMP_COMMON
|
|
|
|
|
|
typedef HRESULT (WINAPI *PFNDLLGETCLASSOBJECT)(REFCLSID rclsid,REFIID riid,LPVOID *ppvObj );
|
|
typedef HRESULT (WINAPI *PFNDLLCANUNLOADNOW)(void);
|
|
|
|
CBilink g_blComEntriesGlobal;
|
|
DNCRITICAL_SECTION csComEntriesLock;
|
|
|
|
typedef struct _COMDLL_ENTRY
|
|
{
|
|
HMODULE hDLL;
|
|
CHAR szFileName[_MAX_PATH];
|
|
GUID clsid;
|
|
PFNDLLGETCLASSOBJECT pfnGetClassObject;
|
|
PFNDLLCANUNLOADNOW pfnCanUnloadNow;
|
|
CBilink blComEntries;
|
|
} COMDLL_ENTRY, *PCOMDLL_ENTRY;
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "COM_Init"
|
|
HRESULT COM_Init()
|
|
{
|
|
g_blComEntriesGlobal.Initialize();
|
|
if (DNInitializeCriticalSection( &csComEntriesLock ) == FALSE)
|
|
{
|
|
return DPNERR_OUTOFMEMORY;
|
|
}
|
|
return DPN_OK;
|
|
}
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "COM_Free"
|
|
HRESULT COM_Free()
|
|
{
|
|
CBilink *pblSearch;
|
|
PCOMDLL_ENTRY pEntry;
|
|
|
|
pblSearch = g_blComEntriesGlobal.GetNext();
|
|
|
|
while( pblSearch != &g_blComEntriesGlobal )
|
|
{
|
|
pEntry = CONTAINING_RECORD( pblSearch, COMDLL_ENTRY, blComEntries );
|
|
pblSearch = pblSearch->GetNext();
|
|
|
|
FreeLibrary( pEntry->hDLL );
|
|
delete pEntry;
|
|
}
|
|
|
|
DNDeleteCriticalSection( &csComEntriesLock );
|
|
return DPN_OK;
|
|
}
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "COM_CoInitialize"
|
|
HRESULT COM_CoInitialize( void * pvParam )
|
|
{
|
|
return CoInitializeEx( pvParam, COINIT_MULTITHREADED );
|
|
}
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "COM_CoUninitialize"
|
|
void COM_CoUninitialize()
|
|
{
|
|
CoUninitialize();
|
|
}
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "COM_GetEntry"
|
|
HRESULT COM_GetEntry( GUID clsid, PCOMDLL_ENTRY *ppEntry )
|
|
{
|
|
CBilink *pblSearch;
|
|
PCOMDLL_ENTRY pEntry;
|
|
HRESULT hr;
|
|
DWORD dwSize;
|
|
|
|
DNEnterCriticalSection( &csComEntriesLock );
|
|
|
|
pblSearch = g_blComEntriesGlobal.GetNext();
|
|
|
|
while( pblSearch != &g_blComEntriesGlobal )
|
|
{
|
|
pEntry = CONTAINING_RECORD( pblSearch, COMDLL_ENTRY, blComEntries );
|
|
|
|
// This should never happen, but makes prefix happy
|
|
if( !pEntry )
|
|
{
|
|
DNASSERT( FALSE );
|
|
DNLeaveCriticalSection( &csComEntriesLock );
|
|
return DPNERR_GENERIC;
|
|
}
|
|
|
|
if( pEntry->clsid == clsid )
|
|
{
|
|
*ppEntry = pEntry;
|
|
DNLeaveCriticalSection( &csComEntriesLock );
|
|
return DPN_OK;
|
|
}
|
|
|
|
pblSearch = pblSearch->GetNext();
|
|
}
|
|
|
|
pEntry = new COMDLL_ENTRY;
|
|
if (pEntry == NULL)
|
|
{
|
|
DPFERR( "Error allocating COM entry" );
|
|
hr = DPNERR_OUTOFMEMORY;
|
|
goto LOAD_FAILED;
|
|
}
|
|
memset( pEntry, 0x00, sizeof( COMDLL_ENTRY ) );
|
|
|
|
pEntry->clsid = clsid;
|
|
pEntry->blComEntries.Initialize();
|
|
|
|
dwSize = _MAX_PATH;
|
|
|
|
hr = COM_GetDLLName( clsid, pEntry->szFileName, &dwSize );
|
|
|
|
if( FAILED( hr ) )
|
|
{
|
|
DPFERR( "Unable to find DLL name for COM object" );
|
|
goto LOAD_FAILED;
|
|
}
|
|
|
|
pEntry->hDLL = LoadLibraryA( pEntry->szFileName );
|
|
|
|
if( !pEntry->hDLL )
|
|
{
|
|
hr = GetLastError();
|
|
DPFX(DPFPREP, 0, "Unable to load libary err=0x%x", hr );
|
|
goto LOAD_FAILED;
|
|
}
|
|
pEntry->pfnGetClassObject = (PFNDLLGETCLASSOBJECT) GetProcAddress( pEntry->hDLL, "DllGetClassObject" );
|
|
pEntry->pfnCanUnloadNow = (PFNDLLCANUNLOADNOW) GetProcAddress( pEntry->hDLL, "DllCanUnloadNow" );
|
|
|
|
pEntry->blComEntries.InsertBefore( &g_blComEntriesGlobal );
|
|
|
|
DNLeaveCriticalSection( &csComEntriesLock );
|
|
|
|
*ppEntry = pEntry;
|
|
|
|
return DPN_OK;
|
|
|
|
LOAD_FAILED:
|
|
|
|
if( pEntry != NULL )
|
|
{
|
|
if( pEntry->hDLL != NULL )
|
|
{
|
|
FreeLibrary( pEntry->hDLL );
|
|
}
|
|
|
|
delete pEntry;
|
|
}
|
|
|
|
DNLeaveCriticalSection( &csComEntriesLock );
|
|
|
|
return hr;
|
|
}
|
|
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "COM_GetDLLName"
|
|
HRESULT COM_GetDLLName( GUID guidCLSID, CHAR *szPath, DWORD *pdwSize )
|
|
{
|
|
CRegistry cregRoot;
|
|
CRegistry cregCLSID;
|
|
CRegistry cregInProc;
|
|
|
|
HRESULT hr = DPN_OK;
|
|
BOOL fSuccess;
|
|
WCHAR *wszTmpPath = NULL;
|
|
DWORD dwTmpSize = 0;
|
|
|
|
fSuccess = cregRoot.Open( HKEY_CLASSES_ROOT, L"CLSID", TRUE, FALSE );
|
|
|
|
if( !fSuccess )
|
|
{
|
|
DPFX(DPFPREP, 0, "Error opening HKEY_CLASSES_ROOT\\CLSID" );
|
|
hr = E_FAIL;
|
|
goto COM_GETDLLNAME_ERROR;
|
|
}
|
|
|
|
fSuccess = cregCLSID.Open( cregRoot, &guidCLSID, TRUE, FALSE );
|
|
|
|
if( !fSuccess )
|
|
{
|
|
DPFX(DPFPREP, 0, "Error opening specified CLSID" );
|
|
hr = E_FAIL;
|
|
goto COM_GETDLLNAME_ERROR;
|
|
}
|
|
|
|
fSuccess = cregInProc.Open( cregCLSID, L"InprocServer32", TRUE, FALSE );
|
|
|
|
if( !fSuccess )
|
|
{
|
|
DPFX(DPFPREP, 0, "Error opening inprocserver key" );
|
|
hr = E_FAIL;
|
|
goto COM_GETDLLNAME_ERROR;
|
|
}
|
|
|
|
cregCLSID.Close();
|
|
cregRoot.Close();
|
|
|
|
fSuccess = cregInProc.ReadString( L"", wszTmpPath, &dwTmpSize );
|
|
|
|
if( !dwTmpSize )
|
|
{
|
|
DPFX(DPFPREP, 0, "Error opening default key" );
|
|
hr = E_FAIL;
|
|
goto COM_GETDLLNAME_ERROR;
|
|
}
|
|
|
|
if( dwTmpSize > *pdwSize )
|
|
{
|
|
DPFX(DPFPREP, 0, "Buffer too small" );
|
|
hr = DPNERR_BUFFERTOOSMALL;
|
|
*pdwSize = dwTmpSize;
|
|
goto COM_GETDLLNAME_ERROR;
|
|
}
|
|
|
|
*pdwSize = dwTmpSize;
|
|
|
|
wszTmpPath = new WCHAR[dwTmpSize];
|
|
|
|
if( !wszTmpPath )
|
|
{
|
|
DPFX(DPFPREP, 0, "Error allocating memory" );
|
|
hr = DPNERR_OUTOFMEMORY;
|
|
goto COM_GETDLLNAME_ERROR;
|
|
}
|
|
|
|
fSuccess = cregInProc.ReadString( L"", wszTmpPath, &dwTmpSize );
|
|
|
|
if( !fSuccess )
|
|
{
|
|
DPFX(DPFPREP, 0, "Error opening default key" );
|
|
hr = E_FAIL;
|
|
goto COM_GETDLLNAME_ERROR;
|
|
}
|
|
|
|
if( FAILED( hr = STR_jkWideToAnsi(szPath,wszTmpPath, *pdwSize ) ) )
|
|
{
|
|
DPFX(DPFPREP, 0, "Error converting path to DLL to ANSI hr=0x%x", hr );
|
|
hr = E_FAIL;
|
|
}
|
|
|
|
|
|
|
|
delete [] wszTmpPath;
|
|
|
|
return hr;
|
|
|
|
COM_GETDLLNAME_ERROR:
|
|
|
|
if( wszTmpPath )
|
|
delete [] wszTmpPath;
|
|
|
|
return hr;
|
|
|
|
}
|
|
|
|
|
|
// DP_CoCreateInstance
|
|
//
|
|
// This CoCreateInstance can be used instead of CoCreateInstance and will manually perform the
|
|
// steps neccessary to do a CoCreateInstance if COM has not been initialized.
|
|
//
|
|
#undef DPF_MODNAME
|
|
#define DPF_MODNAME "COM_CoCreateInstance"
|
|
STDAPI COM_CoCreateInstance( REFCLSID rclsid, LPUNKNOWN pUnkOuter, DWORD dwClsContext, REFIID riid, LPVOID *ppv, BOOL fWarnUser )
|
|
{
|
|
HRESULT hr;
|
|
CHAR *szFileName = NULL;
|
|
DWORD dwFileNameSize = 0;
|
|
PCOMDLL_ENTRY pEntry;
|
|
IClassFactory *pClassFactory;
|
|
|
|
hr = CoCreateInstance( rclsid, pUnkOuter, dwClsContext, riid, ppv );
|
|
|
|
if( hr == CO_E_NOTINITIALIZED )
|
|
{
|
|
if( fWarnUser )
|
|
{
|
|
DPFX(DPFPREP, 0, "=====================================================================================" );
|
|
DPFX(DPFPREP, 0, "" );
|
|
DPFX(DPFPREP, 0, "The DirectPlay8/Voice create functions are no longer supported. It is recommended" );
|
|
DPFX(DPFPREP, 0, "that your application be updated to use CoCreateInstance instead." );
|
|
DPFX(DPFPREP, 0, "" );
|
|
DPFX(DPFPREP, 0, "=====================================================================================" );
|
|
}
|
|
|
|
hr = COM_GetEntry( rclsid, &pEntry );
|
|
|
|
if( FAILED( hr ) )
|
|
return hr;
|
|
|
|
hr = (*pEntry->pfnGetClassObject)( rclsid, IID_IClassFactory, (void **) &pClassFactory );
|
|
|
|
if( FAILED( hr ) )
|
|
{
|
|
DPFX(DPFPREP, 0, "Failed getting class object on dynamic entry hr=0x%x", hr );
|
|
return hr;
|
|
}
|
|
|
|
|
|
hr = pClassFactory->lpVtbl->CreateInstance( pClassFactory, pUnkOuter, riid, ppv );
|
|
|
|
if( FAILED( hr ) )
|
|
{
|
|
DPFX(DPFPREP, 0, "Class factory returned an error hr=0x%x", hr );
|
|
}
|
|
|
|
pClassFactory->lpVtbl->Release(pClassFactory);
|
|
|
|
return hr;
|
|
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|