You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
213 lines
4.9 KiB
213 lines
4.9 KiB
14 years ago
|
/*==LICENSE==*
|
||
|
|
||
|
CyanWorlds.com Engine - MMOG client, server and tools
|
||
13 years ago
|
Copyright (C) 2011 Cyan Worlds, Inc.
|
||
14 years ago
|
|
||
|
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
|
||
13 years ago
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
14 years ago
|
GNU General Public License for more details.
|
||
|
|
||
|
You should have received a copy of the GNU General Public License
|
||
13 years ago
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
||
|
Additional permissions under GNU GPL version 3 section 7
|
||
|
|
||
|
If you modify this Program, or any covered work, by linking or
|
||
|
combining it with any of RAD Game Tools Bink SDK, Autodesk 3ds Max SDK,
|
||
|
NVIDIA PhysX SDK, Microsoft DirectX SDK, OpenSSL library, Independent
|
||
|
JPEG Group JPEG library, Microsoft Windows Media SDK, or Apple QuickTime SDK
|
||
|
(or a modified version of those libraries),
|
||
|
containing parts covered by the terms of the Bink SDK EULA, 3ds Max EULA,
|
||
|
PhysX SDK EULA, DirectX SDK EULA, OpenSSL and SSLeay licenses, IJG
|
||
|
JPEG Library README, Windows Media SDK EULA, or QuickTime SDK EULA, the
|
||
|
licensors of this Program grant you additional
|
||
|
permission to convey the resulting work. Corresponding Source for a
|
||
|
non-source form of such a combination shall include the source code for
|
||
|
the parts of OpenSSL and IJG JPEG Library used as well as that of the covered
|
||
|
work.
|
||
14 years ago
|
|
||
|
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;
|
||
|
}
|
||
|
|