// cNetwork.cpp: implementation of the cNetwork class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "cIocpContext.h"
#include "cConnectorQueueElem.h"
#include "cIOCP.h"
#include "cSystem.h"
#include "cNetwork.h"
#include "../include/cSingleton.h"


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

cNetwork::cNetwork()
{
	Init();
}

cNetwork::~cNetwork()
{
	Destroy();
}

void cNetwork::Init()
{
}

void cNetwork::Destroy()
{
}

BOOL cNetwork::Startup( const uint32 maximum )
{
	WSADATA WsaData;
    if( WSAStartup(MAKEWORD( 2, 2 ), &WsaData) != 0 ) 	
	{
		Trace( _T("WSAStartup Error :: %lu\n"), WSAGetLastError() );
	   return FALSE;
    }

	m_pooler.Create(maximum);
	return TRUE;
}

//////////////////////////////////////////////////////////////////////
// operation
//////////////////////////////////////////////////////////////////////

// Socket »ý¼º
BOOL cNetwork::CreateServerSocket( uint16 port, int32 backlog, uint32& portId )
{
	if(m_serverSocketManager.AddSocket(port, backlog, portId))
	{
		cServerSocket *serverSocket = m_serverSocketManager.GetSocket( portId );
		if(serverSocket)
		{
			cIOCP* IOCP = cSingleton<cSystem>::GetInstance()->GetIOCP();
			return IOCP->AssocInstance(	reinterpret_cast<HANDLE>(serverSocket->GetSocketHandle()), 
											reinterpret_cast<DWORD>(serverSocket) );
			return TRUE;
		}
	}
	return FALSE;
}

// Accept¿äû
BOOL cNetwork::AcceptReqeust( uint32 portId )
{
	// ServerSocketÀ» ¾ò¾î³½´Ù
	cServerSocket* serverSocket = m_serverSocketManager.GetSocket( portId );
	if( !serverSocket )	
	{
		// ÀÌ °æ¿ì´Â ¹ß»ýÇÒ ¼ö ¾ø´Ù.
		Information( _T("[%lu]LS system :: accept failed - critical\n"), GetCurrentThreadId() );
		return FALSE;
	}

	// backlog¸¸Å­ Accept¿äû
	uint32 tryCount = 0;
	while( serverSocket->GetAcceptingCount() < serverSocket->GetBacklog() )
	{
		cIocpContext* iocpContext = m_pooler.Pop();
		if( !iocpContext ) 
		{
			// ¸Þ¸ð¸® Ç®ÀÌ ºñ¾îÀÖ´Ù.
			//Debug( _T("[%lu]LS system :: accept failed - socket pool is empty\n"), GetCurrentThreadId() );
			break;
		}

		iocpContext->SetPortID( portId );
		iocpContext->CleanOverlapped(E_IO_ACCEPT);

		for( tryCount = 1 ; tryCount <= 5 ; tryCount++ )
		{
			if( !serverSocket->AcceptRequest(	iocpContext->GetSockHandle(), 
												iocpContext->GetWsaBuffer(),
												iocpContext->GetOverlapped(E_IO_ACCEPT) ) )
			{
				Debug( _T("[%lu]LS system :: socket create\n"), GetCurrentThreadId() );

				iocpContext->CloseSocket();
				iocpContext->CreateSocket();
			}
			else
			{
				// successful request.
				break;
			}
		}
	}

	if( serverSocket->GetAcceptingCount() == 0 )
	{
		Debug( _T("[%lu]LS system :: empty socket\n"), GetCurrentThreadId() );
	}
	else
	{
		//Debug( _T("[%lu]LS system :: socket(%lu)\n"), GetCurrentThreadId(), serverSocket->GetAcceptingCount() );
	}

	return TRUE;
}

// Accept ¿äûÀÌ ¿Ï·áµÈ ÈÄÀÇ Ã³¸®
void cNetwork::AcceptCompleted( uint32 portId )
{
	// ServerSocketÀ» ¾ò¾î³½´Ù
	cServerSocket* serverSocket = m_serverSocketManager.GetSocket( portId );
	if( serverSocket )
	{
		serverSocket->DecreaseAcceptingCount();
	}
}

// ¸ðµç ÀÏÀ» ¸¶Ä£ Context¿¡ ´ëÇÑ Ã³¸®
void cNetwork::Finish( cIocpContext* iocpContext )
{
	m_pooler.Push(iocpContext);

	// Accept¿äûÀÌ ¾øÀ» °æ¿ì ¿äûÇϵµ·Ï ÇÔ
	const uint32 portId = iocpContext->GetPortID();
	cServerSocket* serverSocket = m_serverSocketManager.GetSocket( portId );
	if( serverSocket )
	{
		AcceptReqeust( portId );
	}
}

// ConnectorÃß°¡
BOOL cNetwork::AddConnector( const uint32 ID, const TCHAR* IP, const uint16 port )
{
	return m_connectorManager.AddConnector( ID, IP, port );
}

// Connector¸¦ ÅëÇØ Àü´Þ
void cNetwork::SendConnector( cConnectorQueueElem* command )
{
	m_connectorManager.Execute( command );
}