1 // SocketCommandHandler.cpp

     2 // 

     3 // Copyright (c) 2010 Accenture. All rights reserved.

     4 // This component and the accompanying materials are made available

     5 // under the terms of the "Eclipse Public License v1.0"

     6 // which accompanies this distribution, and is available

     7 // at the URL "http://www.eclipse.org/legal/epl-v10.html".

     8 // 

     9 // Initial Contributors:

    10 // Accenture - Initial contribution

    11 //

    12 

    13 #include "Misc.h"

    14 #include "stdafx.h"

    15 #include "console_host.h"

    16 #include "SocketCommandHandler.h"

    17 #include "DynamicBuffer.h"

    18 

    19 const int KInitialBufferLength = 1;

    20 

    21 

    22 CSocketCommandReader* CSocketCommandReader::New(CClientSocket& aSocket, MSocketCommandHandler& aHandler)

    23 	{

    24 	std::auto_ptr<CSocketCommandReader> self(new(EThrow) CSocketCommandReader(aSocket, aHandler));

    25 	self->Construct();

    26 	return self.release();

    27 	}

    28 

    29 CSocketCommandReader::~CSocketCommandReader()

    30 	{

    31 	delete iSocket;

    32 	delete iReadBuf;

    33 	}

    34 

    35 void CSocketCommandReader::ChangeHandler(MSocketCommandHandler& aHandler)

    36 	{

    37 	iHandler = &aHandler;

    38 	}

    39 

    40 void CSocketCommandReader::ReadCommand()

    41 	{

    42 	ASSERT(iState == EIdle);

    43 	iState = EReadingHeader;

    44 	QueueRead();

    45 	}

    46 

    47 void CSocketCommandReader::SendResponse(char* aMessage, int aLength)

    48 	{

    49 	iSocket->Write(aMessage, aLength);

    50 	}

    51 

    52 CClientSocket& CSocketCommandReader::ReleaseSocket()

    53 	{

    54 	CClientSocket* socket = iSocket;

    55 	iSocket = NULL;

    56 	return *socket;

    57 	}

    58 

    59 CClientSocket& CSocketCommandReader::Socket() const

    60 	{

    61 	ASSERT(iSocket);

    62 	return *iSocket;

    63 	}

    64 

    65 CSocketCommandReader::CSocketCommandReader(CClientSocket& aSocket, MSocketCommandHandler& aHandler)

    66 	: iSocket(&aSocket), iHandler(&aHandler), iState(EIdle)

    67 	{

    68 	}

    69 

    70 void CSocketCommandReader::Construct()

    71 	{

    72 	iReadBuf = CDynamicReadBuffer::New(KInitialBufferLength);

    73 	iSocket->SetClosureObserver(this);

    74 	}

    75 

    76 void CSocketCommandReader::QueueRead()

    77 	{

    78 	switch (iState)

    79 		{

    80 		case EReadingHeader:

    81 			{

    82 			iSocket->Read((char*)&iPacketHeader, sizeof(TPacketHeader), *this);

    83 			break;

    84 			}

    85 		case EReadingBody:

    86 			{

    87 			iSocket->Read(iReadBuf->Buffer(iPacketHeader.iSize + 2), iPacketHeader.iSize, *this); // + 2 for string terminator.

    88 			break;

    89 			}

    90 		default:

    91 			{

    92 			ASSERT(FALSE);

    93 			}

    94 		}

    95 	}

    96 

    97 void CSocketCommandReader::SocketReadComplete()

    98 	{

    99 	switch (iState)

   100 		{

   101 		case EReadingHeader:

   102 			{

   103 			if (iPacketHeader.iSize > 0)

   104 				{

   105 				iState = EReadingBody;

   106 				QueueRead();

   107 				}

   108 			else

   109 				{

   110 				// Empty body.

   111 				iState = EIdle;

   112 				iHandler->HandleSocketCommand(iPacketHeader.iType, NULL, 0, *this);

   113 				}

   114 			break;

   115 			}

   116 		case EReadingBody:

   117 			{

   118 			iState = EIdle;

   119 			*(TCHAR*)((char*)iReadBuf->Buffer() + iPacketHeader.iSize) = 0; // Add terminator.

   120 			iHandler->HandleSocketCommand(iPacketHeader.iType, iReadBuf->Buffer(), iPacketHeader.iSize, *this);

   121 			break;

   122 			}

   123 		default:

   124 			{

   125 			ASSERT(FALSE);

   126 			}

   127 		}

   128 	}

   129 

   130 void CSocketCommandReader::HandleSocketClosure(CSocket& /*aSocket*/)

   131 	{

   132 	iHandler->HandleSocketClosure(*this);

   133 	}