/*
* 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: 
*
*/
// RegistryImpl.cpp: implementation of the CRegistryImpl class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "RegistryImpl.h"
#include "TCErrorConstants.h"

#ifdef _DEBUG
extern BOOL gDoLogging;
#endif

//#define LOG_REGISTRY
#if defined(LOG_REGISTRY) && defined(_DEBUG)
extern char TCDebugMsg[];
extern CServerManager* gManager;
#define TCDEBUGOPEN() if (gDoLogging) { gManager->m_DebugLog->WaitForAccess(); }
#define TCDEBUGLOGS(s) if (gDoLogging) { sprintf(TCDebugMsg,"%s", s); gManager->m_DebugLog->log(TCDebugMsg); }
#define TCDEBUGLOGA1(s, a1) if (gDoLogging) { sprintf(TCDebugMsg, s, a1); gManager->m_DebugLog->log(TCDebugMsg); }
#define TCDEBUGLOGA2(s, a1, a2) if (gDoLogging) { sprintf(TCDebugMsg, s, a1, a2); gManager->m_DebugLog->log(TCDebugMsg); }
#define TCDEBUGLOGA3(s, a1, a2, a3) if (gDoLogging) { sprintf(TCDebugMsg, s, a1, a2, a3); gManager->m_DebugLog->log(TCDebugMsg); }
#define TCDEBUGCLOSE() if (gDoLogging) { gManager->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

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CRegistryImpl::CRegistryImpl()
{
	TCDEBUGOPEN();
	TCDEBUGLOGS("CRegistryImpl::CRegistryImpl\n");
	TCDEBUGCLOSE();
}

CRegistryImpl::CRegistryImpl(DWORD connectionId)
{
	TCDEBUGOPEN();
	TCDEBUGLOGA1("CRegistryImpl::CRegistryImpl connectionId = %d\n", connectionId);

	for (int i = 0; i < MAX_MESSAGE_IDS; i++)
	{
		m_Registry[i].clist = new IdClientList();
		m_Registry[i].clist->clear();
	}

	char mutexName[200];

	sprintf(mutexName, "%s%d", REGISTRY_MUTEX_BASENAME, connectionId);
	m_Mutex.Open(mutexName, REGISTRY_MUTEX_TIMEOUT);
	TCDEBUGCLOSE();
}

CRegistryImpl::~CRegistryImpl()
{
	TCDEBUGOPEN();
	TCDEBUGLOGS("CRegistryImpl::~CRegistryImpl\n");

	for (int i = 0; i < MAX_MESSAGE_IDS; i++)
	{
		if (m_Registry[i].clist != NULL)
		{
			m_Registry[i].clist->clear();
			delete m_Registry[i].clist;
		}
	}
	m_Mutex.Close();
	TCDEBUGCLOSE();
}

BOOL CRegistryImpl::AddClient(CClient* newClient, long numberIds, BYTE* ids)
{
	TCDEBUGOPEN();
	TCDEBUGLOGS("CRegistryImpl::AddClient\n");

	m_Mutex.Wait();

	for (int i = 0; i < numberIds; i++)
	{
		TCDEBUGLOGA1(" id=%x\n", ids[i]);
		m_Registry[ids[i]].clist->push_back(newClient);
	}

	DumpRegistry();
	m_Mutex.Release();
	TCDEBUGCLOSE();
	return TRUE;
}
BOOL CRegistryImpl::RemoveClient(CClient* client)
{
	TCDEBUGOPEN();
	TCDEBUGLOGS("CRegistryImpl::RemoveClient\n");

	m_Mutex.Wait();

	BOOL found = FALSE;

	for (int i = 0; i < MAX_MESSAGE_IDS; i++)
	{
		long num = m_Registry[i].clist->size();
		if (num != 0)
		{
			TCDEBUGLOGA3(" CRegistryImpl::RemoveClient client = %x i = %x num = %d\n", client, i, num);
			IdClientList::iterator iter;
			for (iter = m_Registry[i].clist->begin(); iter != m_Registry[i].clist->end(); iter++)
			{
				TCDEBUGLOGA2(" CRegistryImpl::RemoveClient iter = %x *iter = %x\n", iter, *iter);
				if (client == *iter)
				{
					m_Registry[i].clist->erase(iter);
					found = TRUE;
					break;
				}
			}
		}
	}

	m_Mutex.Release();
	TCDEBUGCLOSE();
	return found;
}

void CRegistryImpl::DumpRegistry()
{
	for (int i = 0; i < MAX_MESSAGE_IDS; i++)
	{
		long num = m_Registry[i].clist->size();
		if (num != 0)
		{
			TCDEBUGLOGA2(" CRegistryImpl::DumpRegistry i = %x num = %d\n", i, num);
			IdClientList::iterator iter;
			for (iter = m_Registry[i].clist->begin(); iter != m_Registry[i].clist->end(); iter++)
			{
				TCDEBUGLOGA2(" CRegistryImpl::DumpRegistry iter = %x *iter = %x\n", iter, *iter);
			}
		}
	}
}

// fullmessage includes the protocol header
// message is unwrapped from the protocol header
// this is because some clients want the full message and some clients just want the actual message
long CRegistryImpl::RouteMessage(BYTE msgId, BYTE* fullMessage, DWORD fullLen, BYTE* realMessage, DWORD realMessageLen)
{

	long err = TCAPI_ERR_NONE;

//	TCDEBUGOPEN();
//	TCDEBUGLOGS("CRegistryImpl::RouteMessage\n");
//	TCDEBUGCLOSE();
	m_Mutex.Wait();

	if (msgId >= 0 && msgId < MAX_MESSAGE_IDS)
	{
		long numClients = m_Registry[msgId&0xff].clist->size();

		TCDEBUGOPEN();
		TCDEBUGLOGA2(" CRegistryImpl::RouteMessage msgId = %x numClients = %d\n", msgId, numClients);
		TCDEBUGCLOSE();

		if (numClients > 0)
		{
			IdClientList::iterator iter;
			for (iter = m_Registry[msgId&0xff].clist->begin(); iter != m_Registry[msgId&0xff].clist->end(); iter++)
			{
				CClient* client = *iter;
				if (client && client->IsStarted())
				{
					if (client->IsStreamOpen())
					{
						CInputStream* stream = client->GetInputStream();
						// get unwrap format
						if (client->m_Options.unWrapFormat == UNWRAP_LEAVE_HEADERS)
						{
							// use full message
							err = stream->AddMessage(fullLen, fullMessage);
							// routing errors here can be input stream overflows
							//  notify this client right now (no OS error)
							if (err != TCAPI_ERR_NONE)
							{
								client->m_ErrorMonitor->PutError(err, false, 0);
								err = TCAPI_ERR_NONE;
							}
						}
						else
						{
							// use raw message
							err = stream->AddMessage(realMessageLen, realMessage);
							// routing errors here can be input stream overflows
							//  notify this client right now (no OS error)
							if (err != TCAPI_ERR_NONE)
							{
								client->m_ErrorMonitor->PutError(err, false, 0);
								err = TCAPI_ERR_NONE;
							}
						}
					}
					else if (client->IsMessageFileOpen())
					{
						CMessageFile* file = client->GetMessageFile();
						// get unwrap format
						if (client->m_Options.unWrapFormat == UNWRAP_LEAVE_HEADERS)
						{
							// use full message
							err = file->AddMessage(fullLen, fullMessage);
							// routing errors here can be input stream overflows
							//  notify this client right now (no OS error)
							if (err != TCAPI_ERR_NONE)
							{
								client->m_ErrorMonitor->PutError(err, false, 0);
								err = TCAPI_ERR_NONE;
							}
						}
						else
						{
							// use raw message
							err = file->AddMessage(realMessageLen, realMessage);
							// routing errors here can be input stream overflows
							//  notify this client right now (no OS error)
							if (err != TCAPI_ERR_NONE)
							{
								client->m_ErrorMonitor->PutError(err, false, 0);
								err = TCAPI_ERR_NONE;
							}
						}
					}
				}
			}
		}
	}

	m_Mutex.Release();
	return err;
}