CWE-ou-minkata/MOULOpenSourceClientPlugin/Plasma20/Sources/Plasma/PubUtilLib/plSockets/plBufferedSocketReader.cpp

/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

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, see <http://www.gnu.org/licenses/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/
#include "plBufferedSocketReader.h"
#include "plTcpSocket.h"
#include <string.h>


plBufferedSocketReader::plBufferedSocketReader(int size)
:	plRingBuffer(size)
{    
}

int plBufferedSocketReader::ReadBlock(char * buf, int buflen, plTcpSocket & sck)
{
	if(GetBlock(buf, buflen))
		return kSuccessWithData;
	
	int ans = ReadFrom(sck);
	if(ans<=0)
		return ans;
	
	if(GetBlock(buf, buflen))
		return kSuccessWithData;
	
	return kSuccessNoData;
}

int plBufferedSocketReader::ReadString(char * buf, int buflen, char * termChars, plTcpSocket & sck)
{
	if(GetString(buf, buflen, termChars))
		return kSuccessWithData;
	
	int ans = kSuccessNoData;

	while ( ans>=0 )
	{
		ans = ReadFrom(sck);
		if(ans>0)
		{
			if ( GetString(buf, buflen, termChars) )
				return kSuccessWithData;
		}
	}
	
	return ans;
}

int plBufferedSocketReader::ReadStringInPlace(char ** buf, char * termChars, plTcpSocket & sck)
{
	if(GetStringInPlace(buf, termChars))
		return kSuccessWithData;

	int ans = kSuccessNoData;

	while ( ans>=0 )
	{
		ans = ReadFrom(sck);
		if(ans>0)
		{
			if ( GetStringInPlace(buf, termChars) )
				return kSuccessWithData;
		}
	}
	
	return ans;
}

void plBufferedSocketReader::Reset() 
{
	plRingBuffer::Reset();
}

int plBufferedSocketReader::ReadFrom(plTcpSocket & sck)	// this is where things get ugly.
{        
	int ans = kSuccessNoData;
	int readSize = BufferAvailable();
	
	if(readSize < 1)
	{
		Compress();
		readSize = BufferAvailable();
	}
	
	if(readSize > 0)
	{
		char * dst = GetBufferOpen();
		int nBytesRead = sck.RecvData(dst, readSize);
		if(nBytesRead < 0)
		{
			int err = plNet::GetError(); 
			if(err != kBlockingError)
			{
				ans = kFailedReadError;
			}
		}
		else if(nBytesRead > 0)
		{
			fEndPos += nBytesRead;
			ans = kSuccessWithData;
		}
		else
		{
			ans = kFailedSocketClosed;
		}
	}        
	else
	{
		ans = kFailedNoBufferSpace;
	}
	return ans;
}


bool plBufferedSocketReader::GetBlock(char * buf, int buflen)
{
	int dataAvailable = FastAmountBuffered();
	int maxRead = buflen;
	if(maxRead > dataAvailable)
		maxRead = dataAvailable;

	if (maxRead==0)
		return false;
	
	char * wrk = FastGetBufferStart();
	memcpy(buf,FastGetBufferStart(),maxRead);

	return true;
}

bool plBufferedSocketReader::GetString(char * buf, int buflen, char * termChars)
{
	bool ans = false;
	int dataAvailable = FastAmountBuffered();
	int maxRead = buflen;
	if(maxRead > dataAvailable)
		maxRead = dataAvailable;
	
	char * wrk = FastGetBufferStart();
	for(int i=0; i<maxRead; i++)
	{
		if(strchr(termChars,wrk[i])!=0)
		{
			memcpy(buf,wrk,i);
			buf[i] = '\0';
			fStartPos += i+1;                
			Compress();
			ans = true;
			break;
		}            
	}
	return ans;
}

bool plBufferedSocketReader::GetStringInPlace(char ** buf, char * termChars)
{
	bool ans = false;
	int dataAvailable = FastAmountBuffered();
	
	*buf = FastGetBufferStart();
	for(int i=0; i<dataAvailable; i++)
	{
		if(strchr(termChars,(*buf)[i])!=0)
		{
			(*buf)[i] = '\0';
			fStartPos += i+1;                
			Compress();
			ans = true;
			break;
		}            
	}
	return ans;
}