/*

* Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of the License "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description: 

*

*/

// ClientManager.cpp: implementation of the CClientManager class.

//

//////////////////////////////////////////////////////////////////////

#include "stdafx.h"

#include "ClientManager.h"

#include "TCErrorConstants.h"

#include "resource.h"

#include <stdio.h>

#ifdef _PSAPI_PRESENT

	#include "psapi.h"

#endif

#ifdef _DEBUG

extern BOOL gDoLogging;

extern char TCDebugMsg[100];

#define TCDEBUGOPEN() if (gDoLogging) m_DebugLog->WaitForAccess();

#define TCDEBUGLOGS(s) if (gDoLogging) sprintf(TCDebugMsg,"%s", s); if (gDoLogging) m_DebugLog->log(TCDebugMsg);

#define TCDEBUGLOGA1(s, a1) if (gDoLogging) sprintf(TCDebugMsg, s, a1); if (gDoLogging) m_DebugLog->log(TCDebugMsg);

#define TCDEBUGLOGA2(s, a1, a2) if (gDoLogging) sprintf(TCDebugMsg, s, a1, a2); if (gDoLogging) m_DebugLog->log(TCDebugMsg);

#define TCDEBUGLOGA3(s, a1, a2, a3) if (gDoLogging) sprintf(TCDebugMsg, s, a1, a2, a3); if (gDoLogging) m_DebugLog->log(TCDebugMsg);

#define TCDEBUGCLOSE() if (gDoLogging) m_DebugLog->ReleaseAccess();

#else

#define TCDEBUGOPEN()

#define TCDEBUGLOGS(s)

#define TCDEBUGLOGA1(s, a1)

#define TCDEBUGLOGA2(s, a1, a2)

#define TCDEBUGLOGA3(s, a1, a2, a3)

#define TCDEBUGCLOSE()

#endif

#ifdef _DEBUG

static char* GetErrorText(DWORD inError);

#endif

//////////////////////////////////////////////////////////////////////

// Construction/Destruction

//////////////////////////////////////////////////////////////////////

CClientManager::CClientManager()

{

	m_Server = NULL;

	m_StreamList = NULL;

	m_ErrorMonitorList = NULL;

	m_DebugLog = NULL;

	m_DllLocation = NULL;

	m_hServer = NULL;

	m_hServerThread = NULL;

	m_Version[0] = NULL;

	m_ServerLockFile = NULL;

	m_ServerExeFile = NULL;

}

CClientManager::CClientManager(HINSTANCE hinstDLL)

{

#ifdef _DEBUG

	if (gDoLogging)

		m_DebugLog = new TCDebugLog("TCF_ClientLog", ::GetCurrentProcessId());

	else

		m_DebugLog = NULL;

#else

	m_DebugLog = NULL;

#endif

	TCDEBUGOPEN();

	TCDEBUGLOGS("CClientManager::CClientManager\n");

	m_Server = new CServerCommand();

	// lock server access (it might be running)

	m_Server->WaitforServerPipeAccess();

	m_StreamList = new InputStreamList();

	m_StreamList->clear();

	m_ErrorMonitorList = new ErrorMonitorList();

	m_ErrorMonitorList->clear();

	m_DllLocation = new char[MAX_DLLPATHNAME];

	::GetModuleFileName(hinstDLL, m_DllLocation, MAX_DLLPATHNAME);

	char exeDirectory[MAX_DLLPATHNAME] = {0};

	strncpy(exeDirectory, m_DllLocation, MAX_DLLPATHNAME);

	size_t len = strlen(exeDirectory);

	// remove file

	for (int i = len-1; i > 0; i--)

	{

		if (exeDirectory[i] == PATH_DELIMITER)

			break;

	}

	exeDirectory[i] = NULL;

	m_ServerExeFile = new char[MAX_DLLPATHNAME];

	sprintf(m_ServerExeFile, "\"%s%c%s\"", exeDirectory, PATH_DELIMITER, SERVER_PROCESS_NAME);

	m_ServerLockFile = new char[MAX_DLLPATHNAME];

	sprintf(m_ServerLockFile, "%s%c%s", exeDirectory, PATH_DELIMITER, SERVER_LOCKFILE_NAME);

#ifdef _PSAPI_PRESENT

	FindOrCreateRunningServer();

#endif

	char name[100];

	sprintf(name, "%s%ld", ERRORMONITORLIST_MUTEX_BASENAME, ::GetCurrentProcessId());

	m_ErrorMonitorListMutex.Open(name, ERRORMONITORLIST_MUTEX_TIMEOUT);

	sprintf(name, "%s%ld", INPUTSTREAMLIST_MUTEX_BASENAME, ::GetCurrentProcessId());

	m_StreamListMutex.Open(name, INPUTSTREAMLIST_MUTEX_TIMEOUT);

	m_hServer = NULL;

	m_hServerThread = NULL;

	// release server access

	m_Server->ReleaseServerPipeAccess();

	int ret = ::LoadString(hinstDLL, IDS_VERSION, m_Version, MAX_VERSION_STRING);

	TCDEBUGCLOSE();

}

CClientManager::~CClientManager()

{

	TCDEBUGOPEN();

	TCDEBUGLOGS("CClientManager::~CClientManager\n");

	pServerProcessData pData = m_Server->GetProcessPtr();

	if (m_Server)

	{

		delete m_Server;

		m_Server = NULL;

	}

	WaitForStreamListAccess();

	TCDEBUGLOGA1("CClientManager::~CClientManager: erasing stream list size=%d\n", InputStreamListSize());

	if (m_StreamList)

	{

		InputStreamList::iterator iter;

		for (iter = m_StreamList->begin(); iter != m_StreamList->end(); iter++)

		{

			TCDEBUGLOGS("CClientManager::~CClientManager: erasing stream list next 1\n");

//			m_StreamList->erase(iter);

			delete *iter;

//			TCDEBUGLOGS("CClientManager::~CClientManager: erasing stream list next 2\n");

		}

		m_StreamList->clear();

		TCDEBUGLOGS("CClientManager::~CClientManager: erasing stream list done 1\n");

		delete m_StreamList;

		TCDEBUGLOGS("CClientManager::~CClientManager: erasing stream list done 2\n");

	}

	ReleaseStreamListAccess();

	m_StreamListMutex.Close();

	WaitForErrorMonitorListAccess();

	TCDEBUGLOGA1("CClientManager::~CClientManager: erasing monitor list size=%d\n", ErrorMonitorListSize());

	if (m_ErrorMonitorList)

	{

		ErrorMonitorList::iterator iter;

		for (iter = m_ErrorMonitorList->begin(); iter != m_ErrorMonitorList->end(); iter++)

		{

			TCDEBUGLOGS("CClientManager::~CClientManager: erasing monitor list next 1\n");

//			m_ErrorMonitorList->erase(iter);

			delete *iter;

//			TCDEBUGLOGS("CClientManager::~CClientManager: erasing monitor list next 1\n");

		}

		m_ErrorMonitorList->clear();

		TCDEBUGLOGS("CClientManager::~CClientManager: erasing monitor list done 1\n");

		delete m_ErrorMonitorList;

		TCDEBUGLOGS("CClientManager::~CClientManager: erasing monitor list done 2\n");

	}

	ReleaseErrorMonitorListAccess();

	m_ErrorMonitorListMutex.Close();

	if (m_DllLocation)

	{

		delete[] m_DllLocation;

		m_DllLocation = NULL;

	}

	if (m_ServerLockFile)

	{

		delete[] m_ServerLockFile;

		m_ServerLockFile = NULL;

	}

	if (m_ServerExeFile)

	{

		delete[] m_ServerExeFile;

		m_ServerExeFile = NULL;

	}

	TCDEBUGLOGS("CClientManager::~CClientManager: closing log\n");

	TCDEBUGCLOSE();

	if (m_DebugLog)

	{

		delete m_DebugLog;

		m_DebugLog = NULL;

	}

}

#ifdef _PSAPI_PRESENT

void

CClientManager::FindOrCreateRunningServer()

{

	TCDEBUGOPEN();

	TCDEBUGLOGS("CClientManager::FindOrCreateRunningServer: searching for existing TCFServer.exe\n");

	// Directory where TCFServer.exe lives

	char exeDirectory[MAX_DLLPATHNAME] = {0};

	BOOL foundRunningServer = FALSE;

	// Get the list of process identifiers.

	DWORD aProcesses[1024], cbNeeded, cProcesses;

	unsigned int i;

	if ( EnumProcesses( aProcesses, sizeof(aProcesses), &cbNeeded ) )

	{

		// Calculate how many process identifiers were returned.

		cProcesses = cbNeeded / sizeof(DWORD);

		// Find existing TCFServer

		for ( i = 0; i < cProcesses; i++ )

		{

			if( aProcesses[i] != 0 )

			{

				TCHAR szProcessName[MAX_PATH] = TEXT("<unknown>");

				// Get a handle to the process.

				HANDLE hProcess = OpenProcess( PROCESS_QUERY_INFORMATION |

											   PROCESS_VM_READ,

											   FALSE, aProcesses[i] );

				// Get the process name.

				if (NULL != hProcess )

				{

					HMODULE hMod;

					DWORD cbNeeded;

					if ( EnumProcessModules( hProcess, &hMod, sizeof(hMod),

						 &cbNeeded) )

					{

						if (GetModuleBaseNameA( hProcess, hMod, szProcessName,

										   sizeof(szProcessName)/sizeof(char) ))

						{

							if (strcmp(SERVER_PROCESS_NAME, szProcessName) == 0)

							{

								// get the full path

								GetModuleFileNameExA(hProcess, hMod, exeDirectory,

											sizeof(exeDirectory)/sizeof(char) );

								TCDEBUGLOGA1("Found TCFServer at %s\n", exeDirectory);

								CloseHandle( hProcess );

								foundRunningServer = TRUE;

								break;

							}

						}

					}

				}

				CloseHandle( hProcess );

			}

		}

	}

	if (!foundRunningServer)

	{

		// start the one next to the DLL

		strncpy(exeDirectory, m_DllLocation, strlen(m_DllLocation) + 1);

	}

	size_t len = strlen(exeDirectory);

	// remove file

	int j;

	for (j = len-1; j > 0; j--)

	{

		if (exeDirectory[j] == PATH_DELIMITER)

			break;

	}

	exeDirectory[j] = NULL;

	m_ServerExeFile = new char[MAX_DLLPATHNAME];

	sprintf(m_ServerExeFile, "\"%s%c%s\"", exeDirectory, PATH_DELIMITER, SERVER_PROCESS_NAME);

	TCDEBUGLOGA1("exeFile at %s\n", m_ServerExeFile);

	m_ServerLockFile = new char[MAX_DLLPATHNAME];

	sprintf(m_ServerLockFile, "%s%c%s", exeDirectory, PATH_DELIMITER, SERVER_LOCKFILE_NAME);

	TCDEBUGLOGA1("lockFile at %s\n", m_ServerLockFile);

	TCDEBUGCLOSE();

}

#endif

CErrorMonitor*

CClientManager::FindErrorMonitor(long inClientId)

{

	CErrorMonitor* errorMonitor = NULL;

	ErrorMonitorList::iterator iter;

	for (iter = m_ErrorMonitorList->begin(); iter != m_ErrorMonitorList->end(); iter++)

	{

		if ((*iter)->IsThisClient(inClientId))

		{

			errorMonitor = *iter;

			break;

		}

	}

	return errorMonitor;

}

long CClientManager::ErrorMonitorListSize()

{

	long size = m_ErrorMonitorList->size();

	return size;

}

void CClientManager::AddErrorMonitor(CErrorMonitor* monitor)

{

	m_ErrorMonitorList->push_back(monitor);

}

void CClientManager::RemoveErrorMonitor(CErrorMonitor* monitor)

{

	ErrorMonitorList::iterator iter;

	for (iter = m_ErrorMonitorList->begin(); iter != m_ErrorMonitorList->end(); iter++)

	{

		if ((*iter)->IsThisClient(monitor->GetClientId()))

		{

			m_ErrorMonitorList->erase(iter);

			break;

		}

	}

}

CInputStream*

CClientManager::FindInputStream(long inClientId)

{

	CInputStream* inputStream = NULL;

	InputStreamList::iterator iter;

	for (iter = m_StreamList->begin(); iter != m_StreamList->end(); iter++)

	{

		if ((*iter)->IsThisClient(inClientId))

//		if ((*iter).IsThisClient(inClientId))

		{

			inputStream = *iter;

//			inputStream = iter;

			break;

		}

	}

	return inputStream;

}

long CClientManager::InputStreamListSize()

{

	long size = m_StreamList->size();

	return size;

}

void CClientManager::AddInputStream(CInputStream* stream)

{

	m_StreamList->push_back(stream);

//	m_StreamList->push_back(*stream);

}

void CClientManager::RemoveInputStream(CInputStream* stream)

{

	InputStreamList::iterator iter;

	for (iter = m_StreamList->begin(); iter != m_StreamList->end(); iter++)

	{

		if ((*iter)->IsThisClient(stream->GetClientId()))

//		if ((*iter).IsThisClient(stream->GetClientId()))

		{

			m_StreamList->erase(iter);

			break;

		}

	}

}

BOOL CClientManager::StartServer(pServerProcessData pData)

{

	TCDEBUGLOGA1("CClientManager::StartServer numRefs = %d\n",pData->numRefs);

	BOOL serverStarted = FALSE;

	// server is ref counted

	// refcount = 0 => server is not running

	// refcount > 0 => server already started by some other process

	if (pData->numRefs == 0)

	{

		// server not running

		// get exe location

		char exeLocation[MAX_DLLPATHNAME] = {0};

		strncpy(exeLocation, m_DllLocation, MAX_DLLPATHNAME);

		size_t len = strlen(exeLocation);

		// remove file

		for (int i = len-1; i > 0; i--)

		{

			if (exeLocation[i] == PATH_DELIMITER)

				break;

		}

		exeLocation[i] = NULL;

		char quotedLocation[MAX_DLLPATHNAME] = {0};

		sprintf(quotedLocation, "\"%s%c%s\"", exeLocation, PATH_DELIMITER, SERVER_PROCESS_NAME);

		TCDEBUGLOGA1("  exeLocation=%s\n", quotedLocation);

		// create process

		STARTUPINFO si;

		memset(&si,0,sizeof(si));

		si.cb = sizeof(si);

		memset(&pData->serverProcess, 0, sizeof(pData->serverProcess));

		pData->serverProcess.hProcess = NULL;

		if (!::CreateProcess(

			NULL,			// module location

			quotedLocation,	// command line

			NULL,			// process attributes

			NULL,			// thread attributes

			FALSE,			// inherit our handles

			CREATE_NO_WINDOW,	// no window

			NULL,			// use our environment

			NULL,			// user our current directory

			&si,			// startup info

			&pData->serverProcess)) // process info

		{

			// TODO: error creating process

		}

		else

		{

			// we are the creator so save handles for later

			m_hServer = pData->serverProcess.hProcess;

			m_hServerThread = pData->serverProcess.hThread;

			// add a refcount

			pData->numRefs++;

			serverStarted = TRUE;

		}

	}

	else

	{

		// already running

		// add a refcount and open our process handle to it

		pData->numRefs++;

		m_hServer = ::OpenProcess(0, FALSE, pData->serverProcess.dwProcessId);

		serverStarted = TRUE;

	}

	TCDEBUGLOGA1("CClientManager::StartServer serverStarted=%d\n", serverStarted);

	return serverStarted;

}

BOOL CClientManager::StopServer(pServerProcessData pData)

{

	TCDEBUGLOGS("CClientManager::StopServer\n");

	BOOL serverStopped = FALSE;

	if (pData->serverProcess.hProcess == NULL || pData->numRefs <= 0)

	{

		serverStopped = TRUE;

	}

	else

	{

		TCDEBUGLOGA1(" numRefs = %d\n",pData->numRefs); 

		// substract ref count

		pData->numRefs--;

		// if refcount == 0 then really stop the server process

		if (pData->numRefs <= 0)

		{

			// last client process is closing

			// tell server to exit

			ServerCommandData cmdrsp;

			cmdrsp.command = eCmdExit;

		TCDEBUGLOGS(" SendCommand eCmdExit\n");

			m_Server->SendCommand(&cmdrsp);

		TCDEBUGLOGS(" GetResponse eExit\n");

			m_Server->GetResponse(&cmdrsp);

			// wait for process to exit

		TCDEBUGLOGS(" WaitForSingleObject start\n");

			WaitForSingleObject(m_hServer, 10000L /*INFINITE*/);

		TCDEBUGLOGS(" WaitForSingleObject found\n");

			if (m_hServer != NULL)

				CloseHandle(m_hServer);

			if (m_hServerThread != NULL)

				CloseHandle(m_hServerThread);

		}

		else

		{

			// just close our handle to server process

			if (m_hServer != NULL)

				CloseHandle(m_hServer);

			if (m_hServerThread != NULL)

				CloseHandle(m_hServerThread);

		}

	}

	TCDEBUGLOGS("CClientManager::StopServer end\n");

	return TRUE;

}

long CClientManager::StartServer()

{

	long ret = TCAPI_ERR_NONE;

	pServerProcessData pData = m_Server->GetProcessPtr();