diff --git a/Streaming.cpp b/Streaming.cpp index d68f08c7..7af4d291 100644 --- a/Streaming.cpp +++ b/Streaming.cpp @@ -248,7 +248,12 @@ namespace stream return 0; } - // either non-empty or we have received empty + // either non-empty or we have received something + return ConcatenatePackets (buf, len); + } + + size_t Stream::ConcatenatePackets (uint8_t * buf, size_t len) + { size_t pos = 0; while (pos < len) { @@ -269,7 +274,7 @@ namespace stream break; } return pos; - } + } bool Stream::SendPacket (Packet * packet) { diff --git a/Streaming.h b/Streaming.h index 3de8f3e2..37195ebb 100644 --- a/Streaming.h +++ b/Streaming.h @@ -97,6 +97,7 @@ namespace stream void SavePacket (Packet * packet); void ProcessPacket (Packet * packet); + size_t ConcatenatePackets (uint8_t * buf, size_t len); void UpdateCurrentRemoteLease (); @@ -207,13 +208,13 @@ namespace stream template void Stream::HandleReceiveTimer (const boost::system::error_code& ecode, const Buffer& buffer, ReceiveHandler handler) { - // TODO: + size_t received = ConcatenatePackets (boost::asio::buffer_cast(buffer), boost::asio::buffer_size(buffer)); if (ecode == boost::asio::error::operation_aborted) // timeout not expired - handler (boost::system::error_code (), 0); + handler (boost::system::error_code (), received); else // timeout expired - handler (boost::asio::error::make_error_code (boost::asio::error::timed_out), 0); + handler (boost::asio::error::make_error_code (boost::asio::error::timed_out), received); } } }