#include "StdAfx.h"
#include "ConnectManager.h"

// for connect manager function
#include <Connmgr.h>
#pragma comment ( lib, "Cellcore.lib" )

CConnectManager::CConnectManager(void)
	: m_hConnection ( NULL )
{
}

CConnectManager::~CConnectManager(void)
{
	ReleaseConnection ();
}

BOOL CConnectManager::GetConnMgrAvailable()
{
	HANDLE hConnMgr = ConnMgrApiReadyEvent ();
	BOOL bAvailbale = FALSE;
	DWORD dwResult = ::WaitForSingleObject ( hConnMgr, 2000 );
	if ( dwResult == WAIT_OBJECT_0 )
	{
		bAvailbale = TRUE;
	}

	// 關閉
	if ( hConnMgr ) CloseHandle ( hConnMgr );

	return bAvailbale;
}

int CConnectManager::MapURLAndGUID ( LPCTSTR lpszURL, OUT GUID &guidNetworkObject, OUT CString *pcsDesc/*=NULL*/ )
{
	if ( !lpszURL || lstrlen(lpszURL) < 1 )
		return FALSE;

	memset ( &guidNetworkObject, 0, sizeof(GUID) );
	int nIndex = 0;
	HRESULT hResult = ConnMgrMapURL ( lpszURL, &guidNetworkObject, (DWORD*)&nIndex );
	if ( FAILED(hResult) )
	{
		nIndex = -1;
		DWORD dwLastError = GetLastError ();
		AfxMessageBox ( _T("Could not map a request to a network identifier") );
	}
	else
	{
		if ( pcsDesc )
		{
			CONNMGR_DESTINATION_INFO DestInfo = {0};
			if ( SUCCEEDED(ConnMgrEnumDestinations(nIndex, &DestInfo)) )
			{
				*pcsDesc = DestInfo.szDescription;
			}
		}
	}

	return nIndex;
}

void CConnectManager::EnumNetIdentifier ( OUT CStringArray &StrAry )
{
	CONNMGR_DESTINATION_INFO networkDestInfo = {0};

	// 得到網絡列表
	for ( DWORD dwEnumIndex=0; ; dwEnumIndex++ )
	{
		memset ( &networkDestInfo, 0, sizeof(CONNMGR_DESTINATION_INFO) );
		if ( ConnMgrEnumDestinations ( dwEnumIndex, &networkDestInfo ) == E_FAIL )
		{
			break;
		}
		StrAry.Add ( networkDestInfo.szDescription );
	}
}

BOOL CConnectManager::EstablishConnection ( DWORD dwIndex )
{
	// 釋放之前的連接
	ReleaseConnection ();

	// 得到正確的連接信息
	CONNMGR_DESTINATION_INFO DestInfo = {0};
	HRESULT hResult = ConnMgrEnumDestinations(dwIndex, &DestInfo);
	BOOL bRet = FALSE;
	if(SUCCEEDED(hResult))
	{
		// 初始化連接結構
		CONNMGR_CONNECTIONINFO ConnInfo;

		ZeroMemory(&ConnInfo, sizeof(ConnInfo));
		ConnInfo.cbSize = sizeof(ConnInfo);
		ConnInfo.dwParams = CONNMGR_PARAM_GUIDDESTNET;
		ConnInfo.dwFlags = CONNMGR_FLAG_PROXY_HTTP | CONNMGR_FLAG_PROXY_WAP | CONNMGR_FLAG_PROXY_SOCKS4 | CONNMGR_FLAG_PROXY_SOCKS5;
		ConnInfo.dwPriority = CONNMGR_PRIORITY_USERINTERACTIVE;
		ConnInfo.guidDestNet = DestInfo.guid;
		ConnInfo.bExclusive	= FALSE; 
		ConnInfo.bDisabled = FALSE;

		DWORD dwStatus = 0;
		hResult = ConnMgrEstablishConnectionSync(&ConnInfo, &m_hConnection, 10*1000, &dwStatus );
		if(FAILED(hResult))
		{
			m_hConnection = NULL;
		}
		else bRet = TRUE;
	}

	return bRet;
}

BOOL CConnectManager::WaitForConnected ( int nTimeoutSec, DWORD *pdwStatus/*=NULL*/ )
{
	DWORD dwStartTime = GetTickCount ();
	BOOL bRet = FALSE;
	while ( GetTickCount ()-dwStartTime < (DWORD)nTimeoutSec * 1000 )
	{
		if ( m_hConnection )
		{
			DWORD dwStatus = 0;
			HRESULT hr = ConnMgrConnectionStatus ( m_hConnection, &dwStatus );
			if ( pdwStatus ) *pdwStatus = dwStatus;
			if ( SUCCEEDED(hr) )
			{
				if ( dwStatus == CONNMGR_STATUS_CONNECTED )
				{
					bRet = TRUE;
					break;
				}
			}
		}
		Sleep ( 100 );
	}

	return bRet;
}

void CConnectManager::ReleaseConnection ()
{
	if ( m_hConnection )
	{
		ConnMgrReleaseConnection(m_hConnection, FALSE);
		m_hConnection = NULL;
	}
}