#define _WIN32_WINNT 0x0400
#include <windows.h>
#include <iostream>

#include "..\TestComObject1\TestComObject1_i.c"
#include "..\TestComObject1\TestComObject1.h"

DWORD WINAPI ThreadProc(LPVOID lpv)
{
	//HRESULT hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
	HRESULT hr = CoInitializeEx(NULL, COINIT_MULTITHREADED);

	if ( FAILED(hr) )
	{
		std::cout << "CoinitializeEx failed!" << std::endl;
		return 0;
	}

	IStream *pStream = (IStream*)lpv;
	
	ITestInterface1 *pTest = NULL;

	 hr = ::CoGetInterfaceAndReleaseStream(pStream,
											IID_ITestInterface1,
											(void**)&pTest);
	if ( FAILED(hr) )
	{
		std::cout << "CoGetInterfaceAndReleaseStream failed!" << std::endl;
		return 0;
	}


	hr = pTest->TestFunc1();

	if ( FAILED(hr) )
	{
		std::cout << "TestFunc1 failed!" << std::endl;
		return 0;
	}
	

	pTest->Release();

	::CoUninitialize();

	return 0;
}

DWORD ApartMentMsgWaitForMultipleObject(HANDLE *hHandle,DWORD dwWaitCout, DWORD dwMilliseconds)
{
	
	BOOL bQuit = FALSE;
	DWORD dwRet;
	
	while(!bQuit)
	{
		int rc;
		rc = ::MsgWaitForMultipleObjects
			(
			dwWaitCout,          // 需要等待的對象數量
			hHandle,			// 對象樹組
			FALSE,				//等待所有的對象
			(DWORD)dwMilliseconds,  // 等待的時間
			(DWORD)(QS_ALLINPUT | QS_ALLPOSTMESSAGE)     // 事件類型       
			);
		
		if( rc ==  WAIT_OBJECT_0 )
		{			
			dwRet = rc;	
			bQuit = TRUE;
			
		}
		else if( rc == WAIT_OBJECT_0 + dwWaitCout )			
		{
			MSG msg;
			
			while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE))
			{
				TranslateMessage (&msg);
				DispatchMessage(&msg);
			}			
		} 
	}
	
	return dwRet;
}



int main(int argc, char* argv[])
{
	//HRESULT hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
	HRESULT hr = CoInitializeEx(NULL, COINIT_MULTITHREADED);

	if ( FAILED(hr) )
	{
		std::cout << "CoinitializeEx failed!" << std::endl;
		return 0;
	}

	ITestInterface1 *pTest = NULL;

	hr = ::CoCreateInstance(CLSID_TestInterface1,
							0,
							CLSCTX_INPROC,
							IID_ITestInterface1,
							(void**)&pTest);

	if ( FAILED(hr) )
	{
		std::cout << "CoCreateInstance failed!" << std::endl;
		return 0;
	}

	hr = pTest->TestFunc1();

	if ( FAILED(hr) )
	{
		std::cout << "TestFunc1 failed!" << std::endl;
		return 0;
	}

	IStream *pStream = NULL;

	hr = ::CoMarshalInterThreadInterfaceInStream(IID_ITestInterface1,
												pTest,
												&pStream);

	if ( FAILED(hr) )
	{
		std::cout << "CoMarshalInterThreadInterfaceInStream failed!" << std::endl;
		return 0;
	}


	DWORD threadID;
	HANDLE hThreads[1];
    hThreads[0]  =   ::CreateThread(NULL,			//創建一個進程
									0,
									ThreadProc,
									(LPVOID)pStream,  //將pStream作為一個參數傳入新線程
									0,
									&threadID);




	ApartMentMsgWaitForMultipleObject(hThreads,1,INFINITE);

	

	::CloseHandle(hThreads);				//關閉線程句柄

	pTest->Release();

	::CoUninitialize();

	system("pause");

	
	return 0;
}