// ClientSocket.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 "ClientSocket.h"

#include "DynamicBuffer.h"

const int KInitialBufferLength = 1;

const int KNotifyReadComplete = 9000;

CClientSocket* CClientSocket::New(CWindow& aWindow, SOCKET aSocket)

	{

	std::auto_ptr<CClientSocket> self(new(EThrow) CClientSocket(aWindow, aSocket));

	self->Construct();

	return self.release();

	}

CClientSocket::CClientSocket(CWindow& aWindow, SOCKET aSocket)

	: CSocket(aWindow, aSocket), iReadHandler(NULL), iReadBuf(NULL), iReadLength(0)

	{

	}

void CClientSocket::Read(char* aBuf, int aLength, MSocketReadHandler& aHandler)

	{

	ASSERT(iReadHandler == NULL);

	iReadBuf = aBuf;

	iReadLength = aLength;

	iReadHandler = &aHandler;

	HandleRead(FALSE);

	}

void CClientSocket::Write(char* aBuf, int aLength)

	{

	iWriteBuf->Add(aBuf, aLength);

	HandleWrite();

	}

CClientSocket::~CClientSocket()

	{

	delete iWriteBuf;

	}

void CClientSocket::Construct()

	{

	CSocket::Construct();

	if (WSAAsyncSelect(iSocket, iWindow.Handle(), KSocketMessage, FD_READ | FD_WRITE | FD_CLOSE) == SOCKET_ERROR)

		{

		throw KExceptionSocketSelectFailed;

		}

	iWriteBuf = CDynamicWriteBuffer::New(KInitialBufferLength);

	}

void CClientSocket::HandleRead(bool aNotifySynchronously)

	{

	if (iReadLength > 0)

		{

		int ret = recv(iSocket, iReadBuf, iReadLength, 0);

		if (ret == SOCKET_ERROR)

			{

			if (WSAGetLastError() != WSAEWOULDBLOCK)

				{

				HandleClosure();

				}

			}

		else if (ret == 0)

			{

			HandleClosure();

			}

		else

			{

			iReadLength -= ret;

			ASSERT(iReadLength >= 0);

			if (iReadLength == 0)

				{

				if (aNotifySynchronously)

					{

					NotifyReadComplete();

					}

				else

					{

					PostMessage(iWindow.Handle(), KSocketMessage, iSocket, KNotifyReadComplete);

					}

				}

			else

				{

				iReadBuf += ret;

				}

			}

		}

	}

void CClientSocket::HandleWrite()

	{

	if (iWriteBuf->Length() > 0)

		{

		int ret = send(iSocket, iWriteBuf->Buffer(), iWriteBuf->Length(), 0);

		if (ret == SOCKET_ERROR)

			{

			if (WSAGetLastError() != WSAEWOULDBLOCK)

				{

				HandleClosure();

				}

			}

		else if (ret == 0)

			{

			HandleClosure();

			}

		else

			{

			iWriteBuf->Remove(ret);

			}

		}

	}

void CClientSocket::NotifyReadComplete()

	{

	ASSERT(iReadHandler);

	MSocketReadHandler* readHandler = iReadHandler;

	iReadBuf = NULL;

	iReadLength = 0;

	iReadHandler = NULL;

	readHandler->SocketReadComplete();

	}

LRESULT CClientSocket::HandleSocketMessage(LPARAM aLParam)

	{

	switch (WSAGETSELECTEVENT(aLParam))

		{

		case FD_READ:

			HandleRead(TRUE);

			break;

		case FD_WRITE:

			HandleWrite();

			break;

		case KNotifyReadComplete:

			NotifyReadComplete();

			break;

		default:

			ASSERT(FALSE);

		}

	return 0;

	}