read reamaining data from closed stream

AddressBook.cpp

@@ -462,17 +462,18 @@ namespace client
 						{
 							if (bytes_transferred)
 								response.write ((char *)buf, bytes_transferred);
-							else
-							{	
-								if (ecode != boost::asio::error::timed_out || !stream->IsOpen ())
-									end = true;	
-							}	
+							if (ecode == boost::asio::error::timed_out || !stream->IsOpen ())
+								end = true;	
 							newDataReceived.notify_all ();
 						},
 						30); // wait for 30 seconds
 					std::unique_lock<std::mutex> l(newDataReceivedMutex);
 					newDataReceived.wait (l);
 				}
+				// process remaining buffer
+				while (size_t len = stream->ReadSome (buf, 4096))
+					response.write ((char *)buf, len);
+				
 				// parse response
 				std::string version;
 				response >> version; // HTTP version

Streaming.h

@@ -100,7 +100,8 @@ namespace stream
 			
 			template<typename Buffer, typename ReceiveHandler>
 			void AsyncReceive (const Buffer& buffer, ReceiveHandler handler, int timeout = 0);
-
+			size_t ReadSome (uint8_t * buf, size_t len) { return ConcatenatePackets (buf, len); };
+			
 			void Close ();
 			void Cancel () { m_ReceiveTimer.cancel (); };
 
@@ -225,7 +226,7 @@ namespace stream
 			else
 				// socket closed
 				handler (m_IsReset ? boost::asio::error::make_error_code (boost::asio::error::connection_reset) :
-					boost::asio::error::make_error_code (boost::asio::error::operation_aborted), 0);
+					boost::asio::error::make_error_code (boost::asio::error::operation_aborted), received);
 		}	
 		else
 			// timeout expired