/* 
 * Copyright (C) 2012 Yee Young Han <websearch@naver.com> (http://blog.naver.com/websearch)
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA 
 */

#include "SipPlatformDefine.h"
#include "TcpStack.h"
#include "TcpThread.h"
#include "TcpClientMap.h"
#include "Log.h"
#include "MemoryDebug.h"

CTcpClientIndex::CTcpClientIndex() : m_iThreadIndex(-1), m_iSessionIndex(-1)
{
}

CTcpClientInfo::CTcpClientInfo() : m_bTrying(true)
{
}

CTcpClientMap::CTcpClientMap() : m_pclsStack(NULL)
{
}

CTcpClientMap::~CTcpClientMap()
{
}

/**
 * @ingroup TcpStack
 * @brief CTcpStack 객체를 설정한다.
 * @param pclsStack CTcpStack 객체
 * @returns true 를 리턴한다.
 */
bool CTcpClientMap::Create( CTcpStack * pclsStack )
{
	m_pclsStack = pclsStack;

	return true;
}

/**
 * @ingroup TcpStack
 * @brief TCP 서버에 연결하여서 TCP 패킷을 전송한다. 이미 연결된 TCP 서버에는 TCP 패킷만 전송한다.
 * @param pszIp				TCP 서버 IP 주소
 * @param iPort				TCP 서버 포트 번호
 * @param pszPacket		패킷
 * @param iPacketLen	패킷 길이
 * @param bConnectIfNoSession	TCP 세션이 존재하지 않으면 새로운 TCP 세션을 연결한 후, 패킷을 전송하는가?
 * @returns 성공하면 true 를 리턴하고 실패하면 false 를 리턴한다.
 */
bool CTcpClientMap::Send( const char * pszIp, int iPort, const char * pszPacket, int iPacketLen, bool bConnectIfNoSession )
{
	bool bRes = false;
	std::string strKey;
	TCP_CLIENT_MAP::iterator itMap;
	CTcpClientIndex clsIndex;

	GetKey( pszIp, iPort, strKey );

	m_clsMutex.acquire();
	itMap = m_clsMap.find( strKey );
	if( itMap != m_clsMap.end() )
	{
		if( itMap->second.m_bTrying )
		{
			// TCP connect 진행 중이면 전송된 패킷을 자료구조에 저장한다.
			std::string * pstrPacket = new std::string();
			if( pstrPacket )
			{
				pstrPacket->append( pszPacket, iPacketLen );
				itMap->second.m_clsSendPacketList.push_back( pstrPacket );
				bRes = true;
			}
		}
		else
		{
			// TCP connected 상태이면 해당 소켓으로 전송하기 위한 정보를 설정한다.
			clsIndex = itMap->second.m_clsIndex;
			bRes = true;
		}
	}
	else if( bConnectIfNoSession )
	{
		std::string * pstrPacket = new std::string();
		if( pstrPacket )
		{
			CTcpClientInfo clsInfo;

			pstrPacket->append( pszPacket, iPacketLen );
			m_clsMap.insert( TCP_CLIENT_MAP::value_type( strKey, clsInfo ) );
			itMap = m_clsMap.find( strKey );
			itMap->second.m_clsSendPacketList.push_back( pstrPacket );

			if( StartTcpClientThread( pszIp, iPort, m_pclsStack ) == false )
			{
				delete pstrPacket;
				m_clsMap.erase( itMap );
			}
			else
			{
				bRes = true;
			}
		}
	}
	m_clsMutex.release();

	if( clsIndex.m_iThreadIndex != -1 && clsIndex.m_iSessionIndex != -1 )
	{
		m_pclsStack->Send( clsIndex.m_iThreadIndex, clsIndex.m_iSessionIndex, pszPacket, iPacketLen );
	}

	return bRes;
}

/**
 * @ingroup TcpStack
 * @brief TCP 연결 정보를 삭제한다.
 * @param pszIp TCP 서버 IP 주소
 * @param iPort TCP 서버 포트 번호
 * @returns true 를 리턴한다.
 */
bool CTcpClientMap::Delete( const char * pszIp, int iPort )
{
	SEND_PACKET_LIST clsSendPacketList;
	SEND_PACKET_LIST::iterator	itList;

	Delete( pszIp, iPort, clsSendPacketList );
	for( itList = clsSendPacketList.begin(); itList != clsSendPacketList.end(); ++itList )
	{
		//m_pclsStack->m_pclsCallBack->SendPacketError( pszIp, iPort, (*itList)->c_str(), (*itList)->length() );
		delete *itList;
	}

	return true;
}

/**
 * @ingroup TcpStack
 * @brief TCP 연결 후, 전송할 패킷 리스트를 삭제한다.
 * @param pszIp		TCP 서버 IP 주소
 * @param iPort		TCP 서버 포트 번호
 * @param clsList TCP 연결 후, 전송할 패킷 리스트 저장 변수
 * @returns 성공하면 true 를 리턴하고 그렇지 않으면 false 를 리턴한다.
 */
bool CTcpClientMap::DeleteSendPacketList( const char * pszIp, int iPort, SEND_PACKET_LIST & clsList )
{
	bool bRes = false;
	std::string strKey;
	TCP_CLIENT_MAP::iterator itMap;

	clsList.clear();
	GetKey( pszIp, iPort, strKey );

	m_clsMutex.acquire();
	itMap = m_clsMap.find( strKey );
	if( itMap != m_clsMap.end() )
	{
		clsList = itMap->second.m_clsSendPacketList;
		itMap->second.m_clsSendPacketList.clear();
		bRes = true;
	}
	m_clsMutex.release();

	return bRes;
}

/**
 * @ingroup TcpStack
 * @brief TCP 연결됨으로 설정한다.
 * @param pszIp					TCP 서버 IP 주소
 * @param iPort					TCP 서버 포트 번호
 * @param iThreadIndex	TCP 쓰레드 번호
 * @param iSessionIndex TCP 세션 번호 ( TCP 쓰레드 내의 세션 번호 )
 * @returns 성공하면 true 를 리턴하고 그렇지 않으면 false 를 리턴한다.
 */
bool CTcpClientMap::SetConnected( const char * pszIp, int iPort, int iThreadIndex, int iSessionIndex )
{
	bool bRes = false;
	std::string strKey;
	TCP_CLIENT_MAP::iterator itMap;

	GetKey( pszIp, iPort, strKey );

	m_clsMutex.acquire();
	itMap = m_clsMap.find( strKey );
	if( itMap != m_clsMap.end() )
	{
		itMap->second.m_clsIndex.m_iThreadIndex = iThreadIndex;
		itMap->second.m_clsIndex.m_iSessionIndex = iSessionIndex;
		itMap->second.m_bTrying = false;
		bRes = true;
	}
	m_clsMutex.release();

	return bRes;
}

/**
 * @ingroup TcpStack
 * @brief TCP 연결을 저장한다.
 * @param pszIp					TCP 서버 IP 주소
 * @param iPort					TCP 서버 포트 번호
 * @param iThreadIndex	TCP 쓰레드 번호
 * @param iSessionIndex TCP 세션 번호 ( TCP 쓰레드 내의 세션 번호 )
 * @returns 성공하면 true 를 리턴하고 그렇지 않으면 false 를 리턴한다.
 */
bool CTcpClientMap::Insert( const char * pszIp, int iPort, int iThreadIndex, int iSessionIndex )
{
	bool bRes = false;
	std::string strKey;
	CTcpClientInfo clsInfo;
	TCP_CLIENT_MAP::iterator itMap;

	GetKey( pszIp, iPort, strKey );

	clsInfo.m_bTrying = false;
	clsInfo.m_clsIndex.m_iThreadIndex = iThreadIndex;
	clsInfo.m_clsIndex.m_iSessionIndex = iSessionIndex;

	m_clsMutex.acquire();
	itMap = m_clsMap.find( strKey );
	if( itMap == m_clsMap.end() )
	{
		m_clsMap.insert( TCP_CLIENT_MAP::value_type( strKey, clsInfo ) );
		bRes = true;
	}
	m_clsMutex.release();

	return bRes;
}

/**
 * @ingroup TcpStack
 * @brief TCP 연결 정보를 삭제한다.
 * @param pszIp TCP 서버 IP 주소
 * @param iPort TCP 서버 포트 번호
 * @param clsList TCP 연결 후, 전송할 패킷 리스트 저장용 변수
 * @returns 성공하면 true 를 리턴하고 그렇지 않으면 false 를 리턴한다.
 */
bool CTcpClientMap::Delete( const char * pszIp, int iPort, SEND_PACKET_LIST & clsList )
{
	bool bRes = false;
	std::string strKey;
	TCP_CLIENT_MAP::iterator itMap;

	GetKey( pszIp, iPort, strKey );

	m_clsMutex.acquire();
	itMap = m_clsMap.find( strKey );
	if( itMap != m_clsMap.end() )
	{
		clsList = itMap->second.m_clsSendPacketList;
		bRes = true;
		m_clsMap.erase( itMap );
	}
	m_clsMutex.release();

	return bRes;
}

/**
 * @ingroup TcpStack
 * @brief 자료구조의 키를 생성한다.
 * @param pszIp TCP 서버 IP 주소
 * @param iPort TCP 서버 포트 번호
 * @param strKey 자료구조의 키를 저장할 변수
 */
void CTcpClientMap::GetKey( const char * pszIp, int iPort, std::string & strKey )
{
	char szPort[6];

	snprintf( szPort, sizeof(szPort), "%d", iPort );

	strKey = pszIp;
	strKey.append( ":" );
	strKey.append( szPort );
}