// Socket.cpp
// 
// Copyright (c) 2010 Accenture. All rights reserved.
// This component and the accompanying materials are made available
// under the terms of the "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:
// Accenture - Initial contribution
//

#include "Misc.h"
#include "stdafx.h"
#include "console_host.h"
#include "Socket.h"
#include "DynamicBuffer.h"


CSocket::CSocket(CWindow& aWindow)
	: iWindow(aWindow), iSocket(INVALID_SOCKET), iClosureObserver(NULL)
	{
	}

CSocket::CSocket(CWindow& aWindow, SOCKET aSocket)
	: iWindow(aWindow), iSocket(aSocket), iClosureObserver(NULL)
	{
	}

CSocket::~CSocket()
	{
	if (iSocket != INVALID_SOCKET)
		{
		closesocket(iSocket);
		}

	iWindow.RemoveMessageHandler(*this);
	}

void CSocket::SetClosureObserver(MSocketClosureObserver* aClosureObserver)
	{
	iClosureObserver = aClosureObserver;
	}

void CSocket::Construct()
	{
	if (iSocket == INVALID_SOCKET)
		{
		if ((iSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET)
			{
			throw KExceptionSocketConstructFailed;
			}
		BOOL b = 1;
		setsockopt(iSocket, IPPROTO_TCP, TCP_NODELAY, (char*)&b, sizeof(BOOL));
		}

	iWindow.AddMessageHandler(*this);
	}

void CSocket::HandleClosure()
	{
	if (iClosureObserver)
		{
		iClosureObserver->HandleSocketClosure(*this);
		}
	}

LRESULT CSocket::HandleWindowMessage(UINT aMessage, WPARAM aWParam, LPARAM aLParam)
	{
	if ((aMessage == KSocketMessage) && (aWParam == iSocket))
		{
		if (WSAGETSELECTEVENT(aLParam) == FD_CLOSE)
			{
			HandleClosure();
			}
		else
			{
			HandleSocketMessage(aLParam);
			}
		return 0;
		}
	return 1;
	}