/*========================================================================== * * 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; }