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.
109 lines
2.7 KiB
109 lines
2.7 KiB
/*==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 "plBufferedSocketWriter.h" |
|
#include "plTcpSocket.h" |
|
|
|
plBufferedSocketWriter::plBufferedSocketWriter(int size, int bytesPerFlush, bool blockOnSend, int flushPoint) |
|
: plRingBuffer(size) |
|
, fFlushPoint(flushPoint) |
|
, fBlockOnSend(blockOnSend) |
|
, fBytesPerFlush(bytesPerFlush) |
|
{} |
|
|
|
|
|
int plBufferedSocketWriter::WriteBlock(const char * data, int len, plTcpSocket & sck) |
|
{ |
|
int ans = kSuccessNoDataSent; |
|
|
|
if(len > BufferAvailable()) |
|
ans = Flush(sck); |
|
|
|
if(ans >= 0) |
|
ans = WriteBlock(data,len); |
|
|
|
if(ans >= 0 && fFlushPoint >= 0 && fFlushPoint < AmountBuffered()) |
|
ans = Flush(sck); |
|
|
|
return ans; |
|
} |
|
|
|
int plBufferedSocketWriter::WriteBlock(const char * data, int len) |
|
{ |
|
int ans = kSuccessNoDataSent; |
|
if(!Put(data,len)) |
|
ans = kFailedNoBufferSpace; |
|
return ans; |
|
} |
|
|
|
int plBufferedSocketWriter::Flush(plTcpSocket & sck) // this is where things get ugly. |
|
{ |
|
int ans = kSuccessNoDataSent; |
|
|
|
int writeSize = MIN(FastAmountBuffered(),fBytesPerFlush); |
|
|
|
if(writeSize > 0) |
|
{ |
|
int nBytesWritten = 0; |
|
// int nBytesWritten = sck.SendData(FastGetBufferStart(), writeSize); |
|
|
|
// if (nBytesWritten<0 && fBlockOnSend |
|
// && plNet::GetError()==kBlockingError) |
|
{ |
|
bool wasBlockingOrNot = sck.IsBlocking(); |
|
sck.SetBlocking(fBlockOnSend); |
|
nBytesWritten = sck.SendData(FastGetBufferStart(), writeSize); |
|
sck.SetBlocking(wasBlockingOrNot); |
|
} |
|
|
|
if (nBytesWritten > 0) |
|
{ |
|
fStartPos += nBytesWritten; |
|
FullCompress(); |
|
ans = kSuccessDataSent; |
|
} |
|
else if (nBytesWritten < 0) |
|
{ |
|
if (plNet::GetError()!=kBlockingError) |
|
ans = kFailedWriteError; |
|
} |
|
else |
|
{ |
|
ans = kFailedSocketClosed; |
|
} |
|
} |
|
return ans; |
|
} |
|
|
|
bool plBufferedSocketWriter::IsEmpty() |
|
{ |
|
return !(FastAmountBuffered() > 0); |
|
} |
|
|
|
void plBufferedSocketWriter::Reset() |
|
{ |
|
plRingBuffer::Reset(); |
|
} |
|
|
|
|