Windows编程

开发平台：
Visual C++

component.cpp：源码内容
							// component.cpp
#include <iostream.h>
#include <stdio.h>
#include "Componentcomponent.h" // Generated by MIDL
#include "registry.h"  // Need this
HINSTANCE g_hInstance;
long g_cComponents = 0;
long g_cServerLocks = 0;
class CInsideCOM : public ISum
{
public:
	// IUnknown
	ULONG __stdcall AddRef();
	ULONG __stdcall Release();
	HRESULT __stdcall QueryInterface(REFIID iid, void** ppv);
	// IDispatch
	HRESULT __stdcall GetTypeInfoCount(UINT* pCountTypeInfo);
	HRESULT __stdcall GetTypeInfo(UINT iTypeInfo, LCID lcid, ITypeInfo** ppITypeInfo);
	HRESULT __stdcall GetIDsOfNames(REFIID riid, LPOLESTR* rgszNames, UINT cNames, LCID lcid, DISPID* rgDispId);
	HRESULT __stdcall Invoke(DISPID dispIdMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS* pDispParams, VARIANT* pVarResult, EXCEPINFO* pExcepInfo, UINT* puArgErr);
	// IDispatchEx
    HRESULT __stdcall GetDispID(BSTR bstrName, DWORD grfdex, DISPID* pid);
	HRESULT __stdcall InvokeEx(DISPID id, LCID lcid, WORD wFlags, DISPPARAMS* pdp, VARIANT* pvarRes, EXCEPINFO* pei, IServiceProvider* pspCaller);
	HRESULT __stdcall DeleteMemberByName(BSTR bstrName, DWORD grfdex);        
    HRESULT __stdcall DeleteMemberByDispID(DISPID id);
	HRESULT __stdcall GetMemberProperties(DISPID id, DWORD grfdexFetch, DWORD* pgrfdex);
	HRESULT __stdcall GetMemberName(DISPID id, BSTR* pbstrName);
	HRESULT __stdcall GetNextDispID(DWORD grfdex, DISPID id, DISPID* pid);
	HRESULT __stdcall GetNameSpaceParent(IUnknown** ppunk);
	// ISum
	HRESULT __stdcall Sum(int x, int y, int* retval);
	HRESULT __stdcall CreateNewSum(BSTR name);
	CInsideCOM() : m_cRef(1) { g_cComponents++; }
	~CInsideCOM() { cout << "Component: CInsideCOM::~CInsideCOM()" << endl, g_cComponents--; }
	bool Init(void);
private:
	ULONG m_cRef;
	ITypeInfo* m_pTypeInfo;
	BSTR newsum;
};
HRESULT CInsideCOM::CreateNewSum(BSTR name)
{
	wprintf(L"COMPONENT: CreateNewSum, name = %sn", name);
	newsum = SysAllocString(name);
	return S_OK;
}
HRESULT CInsideCOM::GetDispID(BSTR bstrName, DWORD grfdex, DISPID* pid)
{
	wprintf(L"COMPONENT: GetDispID, bstrName = %sn", bstrName);
	HRESULT hr = GetIDsOfNames(IID_NULL, &bstrName, 1, LOCALE_USER_DEFAULT, pid);
	if(hr == DISP_E_UNKNOWNNAME)
	{
		if(grfdex & fdexNameEnsure)
			newsum = SysAllocString(bstrName);
		if(wcscmp(bstrName, newsum) == 0)
		{
			*pid = 17;
			wprintf(L"COMPONENT: GetDispID, returning bstrName = %s, pid = %d, newsum = %dn", bstrName, *pid, newsum);
			return S_OK;
		}
	}
	return hr;
}
HRESULT CInsideCOM::InvokeEx(DISPID id, LCID lcid, WORD wFlags, DISPPARAMS* pdp, VARIANT* pvarRes, EXCEPINFO* pei, IServiceProvider* pspCaller)
{
	cout << "COMPONENT: InvokeEx" << endl;
	HRESULT hr = DispInvoke(this, m_pTypeInfo, id, wFlags, pdp, pvarRes, pei, NULL);
	if(hr == DISP_E_MEMBERNOTFOUND)
		if(id == 17)
		{
			cout << "COMPONENT: DISPID == 17" << endl;
			HRESULT hr = DispInvoke(this, m_pTypeInfo, 1, wFlags, pdp, pvarRes, pei, NULL);
			printf("HRESULT = %0xn", hr);
			return hr;
		}
		
	return hr;
}
HRESULT CInsideCOM::DeleteMemberByName(BSTR bstrName, DWORD grfdex)
{
	cout << "COMPONENT: DeleteMemberByName" << endl;
	return E_NOTIMPL;
}
HRESULT CInsideCOM::DeleteMemberByDispID(DISPID id)
{
	cout << "COMPONENT: DeleteMemberByDispID" << endl;
	return E_NOTIMPL;
}
HRESULT CInsideCOM::GetMemberProperties(DISPID id, DWORD grfdexFetch, DWORD* pgrfdex)
{
	cout << "COMPONENT: GetMemberProperties" << endl;
	return E_NOTIMPL;
}
HRESULT CInsideCOM::GetMemberName(DISPID id, BSTR* pbstrName)
{
	cout << "COMPONENT: GetMemberName" << endl;
	return E_NOTIMPL;
}
HRESULT CInsideCOM::GetNextDispID(DWORD grfdex, DISPID id, DISPID* pid)
{
	cout << "COMPONENT: GetNextDispID" << endl;
	return E_NOTIMPL;
}
HRESULT CInsideCOM::GetNameSpaceParent(IUnknown** ppunk)
{
	cout << "COMPONENT: GetNameSpaceParent" << endl;
	return E_NOTIMPL;
}
HRESULT CInsideCOM::GetTypeInfoCount(UINT* pCountTypeInfo)
{
	*pCountTypeInfo = 1;
	return S_OK;
}
HRESULT CInsideCOM::GetTypeInfo(UINT iTypeInfo, LCID lcid, ITypeInfo** ppITypeInfo)
{
	*ppITypeInfo = NULL;
	if(iTypeInfo != 0)
		return DISP_E_BADINDEX;
	m_pTypeInfo->AddRef();
	*ppITypeInfo = m_pTypeInfo;
	return S_OK;
}
HRESULT CInsideCOM::GetIDsOfNames(REFIID riid, LPOLESTR* rgszNames, UINT cNames, LCID lcid, DISPID* rgDispId)
{
	if(riid != IID_NULL)
		return DISP_E_UNKNOWNINTERFACE;
	return DispGetIDsOfNames(m_pTypeInfo, rgszNames, cNames, rgDispId);
}
HRESULT CInsideCOM::Invoke(DISPID dispIdMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS* pDispParams, VARIANT* pVarResult, EXCEPINFO* pExcepInfo, UINT* puArgErr)
{
	if(riid != IID_NULL)
		return DISP_E_UNKNOWNINTERFACE;
	return DispInvoke(this, m_pTypeInfo, dispIdMember, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr); 
}
bool CInsideCOM::Init(void)
{
	ITypeLib* pTypeLib;
	if(FAILED(LoadRegTypeLib(LIBID_Component, 1, 0, LANG_NEUTRAL, &pTypeLib)))
		return false;
	HRESULT hr = pTypeLib->GetTypeInfoOfGuid(IID_ISum, &m_pTypeInfo);
	pTypeLib->Release();
	if(FAILED(hr))
		return false;
	return true;
}
ULONG CInsideCOM::AddRef()
{
	cout << "Component: CInsideCOM::AddRef() m_cRef = " << m_cRef + 1 << endl;
	return ++m_cRef;
}
ULONG CInsideCOM::Release()
{
	cout << "Component: CInsideCOM::Release() m_cRef = " << m_cRef - 1 << endl;
	if(--m_cRef != 0)
		return m_cRef;
	m_pTypeInfo->Release();
	delete this;
	return 0;
}
HRESULT CInsideCOM::QueryInterface(REFIID riid, void** ppv)
{
	if(riid == IID_IUnknown)
		*ppv = (IUnknown*)this;
	else if(riid == IID_ISum)
		*ppv = (ISum*)this;
	else if(riid == IID_IDispatch)
		*ppv = (IDispatch*)this;
	else if(riid == IID_IDispatchEx)
	{
		cout << "COMPONENT: QueryInterface for IDispatchEx" << endl;
		*ppv = (IDispatchEx*)this;
	}
	else 
	{
		*ppv = NULL;
		return E_NOINTERFACE;
	}
	AddRef();
	return S_OK;
}
HRESULT CInsideCOM::Sum(int x, int y, int* retval)
{
	*retval = x + y;
	return S_OK;
}
class CFactory : public IClassFactory
{
public:
	// IUnknown
	ULONG __stdcall AddRef();
	ULONG __stdcall Release();
	HRESULT __stdcall QueryInterface(REFIID iid, void** ppv);
	// IClassFactory
	HRESULT __stdcall CreateInstance(IUnknown *pUnknownOuter, REFIID iid, void** ppv);
	HRESULT __stdcall LockServer(BOOL bLock);
	CFactory() : m_cRef(1) { }
	~CFactory() { }
private:
	long m_cRef;
};
ULONG CFactory::AddRef()
{
	cout << "Component: CFactory::AddRef() m_cRef = " << m_cRef + 1 << endl;
	return ++m_cRef;
}
ULONG CFactory::Release()
{
	cout << "Component: CFactory::Release() m_cRef = " << m_cRef - 1 << endl;
	if(--m_cRef != 0)
		return m_cRef;
	delete this;
	return 0;
}
HRESULT CFactory::QueryInterface(REFIID iid, void** ppv)
{
	if((iid == IID_IUnknown) || (iid == IID_IClassFactory))
	{
		cout << "Component: CFactory::QueryInteface() for IUnknown or IClassFactory " << this << endl;
		*ppv = (IClassFactory *)this;
	}
	else
	{
		*ppv = NULL;
		return E_NOINTERFACE;
	}
	AddRef();
	return S_OK;
}
HRESULT CFactory::CreateInstance(IUnknown *pUnknownOuter, REFIID iid, void** ppv)
{
	if(pUnknownOuter != NULL)
		return CLASS_E_NOAGGREGATION;
	CInsideCOM *pInsideCOM = new CInsideCOM;
	cout << "Component: CFactory::CreateInstance() " << pInsideCOM << endl;
	if(pInsideCOM == NULL)
		return E_OUTOFMEMORY;
	// Call the Init method to load the type information
	pInsideCOM->Init();
	HRESULT hr = pInsideCOM->QueryInterface(iid, ppv);
	pInsideCOM->Release();
	return hr;
}
HRESULT CFactory::LockServer(BOOL bLock)
{
	if(bLock)
		g_cServerLocks++;
	else
		g_cServerLocks--;
	return S_OK;
}
HRESULT __stdcall DllCanUnloadNow()
{
	cout << "Component: DllCanUnloadNow() " << (g_cServerLocks == 0 && g_cComponents == 0 ? "Yes" : "No") << endl;
	if(g_cServerLocks == 0 && g_cComponents == 0)
		return S_OK;
	else
		return S_FALSE;
}
HRESULT __stdcall DllGetClassObject(REFCLSID clsid, REFIID iid, void** ppv)
{
	cout << "Component: DllGetClassObject" << endl;
	
	if(clsid != CLSID_InsideCOM)
		return CLASS_E_CLASSNOTAVAILABLE;
	CFactory* pFactory = new CFactory;
	if(pFactory == NULL)
		return E_OUTOFMEMORY;
	// QueryInterface probably for IClassFactory
	HRESULT hr = pFactory->QueryInterface(iid, ppv);
	pFactory->Release();
	return hr;
}
HRESULT __stdcall DllRegisterServer()
{
	char DllPath[256];
	OLECHAR wDllPath[256];
	GetModuleFileName(g_hInstance, DllPath, 256);
	mbstowcs(wDllPath, DllPath, 256);
	ITypeLib* pTypeLib;
	HRESULT hr = LoadTypeLibEx(wDllPath, REGKIND_REGISTER, &pTypeLib);
	if(FAILED(hr))
		return hr;
	pTypeLib->Release();
	return RegisterServer("component.dll", CLSID_InsideCOM, "Inside COM+ Sample", "Component.InsideCOM", "Component.InsideCOM.1", NULL);
}
HRESULT __stdcall DllUnregisterServer()
{
	UnRegisterTypeLib(LIBID_Component, 1, 0, LANG_NEUTRAL, SYS_WIN32);
	return UnregisterServer(CLSID_InsideCOM, "Component.InsideCOM", "Component.InsideCOM.1");
}
BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, void* pv)
{
	g_hInstance = hInstance;
	return TRUE;
}

						