From ea1b4b3ee533146763245ecbbb4a31e3ed791498 Mon Sep 17 00:00:00 2001 From: Tatsuhiro Tsujikawa Date: Sun, 16 Jan 2011 16:55:41 +0900 Subject: Eliminated SocketCore::peekData() form HTTP/FTP downloads. We introduced SocketRecvBuffer which buffers received bytes. Since HTTP response header and response body are divided with \r\n, we have to buffer up several bytes to find this delimiter. We use SocketRecvBuffer to hold these bytes and only consumes header and passes SocketRecvBuffer, which may contain head of response body, to next Command. Since FTPConnection doesn't use SocketCore::peekData(), we left it as is. --- src/SocketRecvBuffer.cc | 77 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 77 insertions(+) create mode 100644 src/SocketRecvBuffer.cc (limited to 'src/SocketRecvBuffer.cc') diff --git a/src/SocketRecvBuffer.cc b/src/SocketRecvBuffer.cc new file mode 100644 index 00000000..2c5a1cdf --- /dev/null +++ b/src/SocketRecvBuffer.cc @@ -0,0 +1,77 @@ +/* */ +#include "SocketRecvBuffer.h" + +#include + +#include "SocketCore.h" +#include "LogFactory.h" + +namespace aria2 { + +SocketRecvBuffer::SocketRecvBuffer +(const SharedHandle& socket, + size_t capacity) + : socket_(socket), + capacity_(capacity), + buf_(new unsigned char[capacity_]), + bufLen_(0) +{} + +SocketRecvBuffer::~SocketRecvBuffer() +{ + delete [] buf_; +} + +ssize_t SocketRecvBuffer::recv() +{ + size_t len = capacity_-bufLen_; + if(len > 0) { + socket_->readData(buf_+bufLen_, len); + bufLen_ += len; + } else { + A2_LOG_DEBUG("Buffer full"); + } + return len; +} + +void SocketRecvBuffer::shiftBuffer(size_t offset) +{ + assert(offset <= bufLen_); + memmove(buf_, buf_+offset, bufLen_-offset); + bufLen_ -= offset; +} + +} // namespace aria2 -- cgit