DllMain.cpp

/****************************** Module Header ******************************\
Module Name:  DllMain.cpp
Project:      CppDllCOMServer
Copyright (c) Microsoft Corporation.



This source is subject to the Microsoft Public License.
See http://www.microsoft.com/en-us/openness/resources/licenses.aspx#MPL.
All other rights reserved.

THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND,
EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE IMPLIED
WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A PARTICULAR PURPOSE.
\***************************************************************************/

#pragma region Includes
#include <windows.h>
#include "Reg.h"
#include "CFDispatchedUiAutomation.h"      // For the class factory
#include "DispatchedUiAutomation_i.c"      // For component GUIDs
#pragma endregion


HINSTANCE   g_hInst = NULL;
long        g_cDllRef = 0;


BOOL APIENTRY DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
{
	switch (dwReason)
	{
	case DLL_PROCESS_ATTACH:
		// Hold the instance of this DLL module, we will use it to get the 
		// path of the DLL to register the component.
		g_hInst = hInstance;
		DisableThreadLibraryCalls(hInstance);
		break;
	case DLL_THREAD_ATTACH:
	case DLL_THREAD_DETACH:
	case DLL_PROCESS_DETACH:
		break;
	}

	return TRUE;
}


//
//   FUNCTION: DllGetClassObject(REFCLSID, REFIID, void **)
//
//   PURPOSE: Create the class factory and query to the specific interface.
//
//   PARAMETERS:
//   * rclsid - The CLSID that will associate the correct data and code.
//   * riid - A reference to the identifier of the interface that the caller 
//     is to use to communicate with the class object.
//   * ppv - The address of a pointer variable that receives the interface 
//     pointer requested in riid. Upon successful return, *ppv contains the 
//     requested interface pointer. If an error occurs, the interface pointer 
//     is NULL. 
//
STDAPI ProxyDllGetClassObject(REFCLSID rclsid, REFIID riid, void **ppv)
{
	HRESULT hr = CLASS_E_CLASSNOTAVAILABLE;

	if (IsEqualCLSID(CLSID_UiAutomation, rclsid))
	{
		hr = E_OUTOFMEMORY;

		CFDispatchedUiAutomation *pClassFactory = new CFDispatchedUiAutomation();
		if (pClassFactory)
		{
			hr = pClassFactory->QueryInterface(riid, ppv);
			pClassFactory->Release();
		}
	}
	/*
	else if (IsEqualCLSID(CLSID_UiAutomationElement, rclsid))
	{
		hr = E_OUTOFMEMORY;

		CFDispatchedUiAutomationElement *pClassFactory = new CFDispatchedUiAutomationElement();
		if (pClassFactory)
		{
			hr = pClassFactory->QueryInterface(riid, ppv);
			pClassFactory->Release();
		}
	}
	*/

	return hr;
}


//
//   FUNCTION: DllCanUnloadNow(void)
//
//   PURPOSE: Check if we can unload the component from the memory.
//
//   NOTE: The component can be unloaded from the memory when its reference 
//   count is zero (i.e. nobody is still using the component).
// 
STDAPI ProxyDllCanUnloadNow(void)
{
	return g_cDllRef > 0 ? S_FALSE : S_OK;
}

//
//   FUNCTION: DllRegisterServer(void)
//
//   PURPOSE: Register the COM server.
// 
STDAPI DllRegisterServer(void)
{
	HRESULT hr;
	//MessageBox(NULL, L"Stop me here", L"A caption", MB_OK);
	wchar_t szModule[MAX_PATH];
	if (GetModuleFileName(g_hInst, szModule, ARRAYSIZE(szModule)) == 0)
	{
		hr = HRESULT_FROM_WIN32(GetLastError());
		return hr;
	}

	// Register the components
	hr = RegisterInprocServer(szModule,
		CLSID_UiAutomation,
		L"DispatchedUiAutomation.UiAutomation Class",
		L"Apartment",
		LIBID_DispatchedUiAutomation,
		L"DispatchedUiAutomation.UiAutomation.1",
		L"DispatchedUiAutomation.UiAutomation");
	/*
	hr2 = RegisterInprocServer(szModule,
		CLSID_UiAutomationElement,
		L"DispatchedUiAutomation.UiAutomationElement Class",
		L"Apartment",
		LIBID_DispatchedUiAutomation,
		L"DispatchedUiAutomation.UiAutomationElement.1",
		L"DispatchedUiAutomation.UiAutomationElement");

	hr3 = RegisterInprocServer(szModule,
		CLSID_UiAutomationPropertyCondition,
		L"DispatchedUiAutomation.UiAutomationPropertyCondition Class",
		L"Apartment",
		LIBID_DispatchedUiAutomation,
		L"DispatchedUiAutomation.UiAutomationPropertyCondition.1",
		L"DispatchedUiAutomation.UiAutomationPropertyCondition");

	hr4 = RegisterInprocServer(szModule,
		CLSID_UiAndCondition,
		L"DispatchedUiAutomation.UiAndCondition Class",
		L"Apartment",
		LIBID_DispatchedUiAutomation,
		L"DispatchedUiAutomation.UiAndCondition.1",
		L"DispatchedUiAutomation.UiAndCondition");

	hr = S_FALSE;

	if (SUCCEEDED(hr1) && SUCCEEDED(hr2) && SUCCEEDED(hr3) && SUCCEEDED(hr4))*/
	if (SUCCEEDED(hr))
	{
		// Register the type library.
		hr = RegisterTypeLib(szModule);
	}

	return hr;
}


//
//   FUNCTION: DllUnregisterServer(void)
//
//   PURPOSE: Unregister the COM server.
// 
STDAPI DllUnregisterServer(void)
{
	HRESULT hr = S_OK;

	wchar_t szModule[MAX_PATH];
	if (GetModuleFileName(g_hInst, szModule, ARRAYSIZE(szModule)) == 0)
	{
		hr = HRESULT_FROM_WIN32(GetLastError());
		return hr;
	}

	// Unregister the components
	hr = UnregisterInprocServer(CLSID_UiAutomation,
		L"DispatchedUiAutomation.UiAutomation.1",
		L"DispatchedUiAutomation.UiAutomation");
	/*
	hr2 = UnregisterInprocServer(CLSID_UiAutomationElement,
		L"DispatchedUiAutomation.UiAutomationElement.1",
		L"DispatchedUiAutomation.UiAutomationElement");

	hr3 = UnregisterInprocServer(CLSID_UiAutomationPropertyCondition,
		L"DispatchedUiAutomation.UiAutomationPropertyCondition.1",
		L"DispatchedUiAutomation.UiAutomationPropertyCondition");

	hr4 = UnregisterInprocServer(CLSID_UiAndCondition,
		L"DispatchedUiAutomation.UiAndCondition.1",
		L"DispatchedUiAutomation.UiAndCondition");

	hr = S_FALSE;

	if (SUCCEEDED(hr1) && SUCCEEDED(hr2) && SUCCEEDED(hr3))*/
	if (SUCCEEDED(hr))
	{
		// Unregister the type library.
		hr = UnregisterTypeLib(szModule);
	}

	return hr;
}