【发布时间】:2013-02-13 04:26:59
【问题描述】:
我已经使用boost::asio 实现了一个 TCP 服务器。此服务器使用basic_stream_socket::read_some 函数读取数据。我知道read_some 不保证提供的缓冲区在返回之前是满的。
在我的项目中,我发送由分隔符分隔的字符串(如果重要的话)。在客户端,我使用WinSock::send() 函数发送数据。现在我的问题出在服务器端,我无法获取从客户端发送的所有字符串。我怀疑read_some 正在接收一些数据并出于某种原因丢弃剩余数据。比在下一次调用中再次接收另一个字符串。
在 TCP/IP 中真的可以吗?
我尝试使用async_receive,但这占用了我所有的 CPU,而且由于必须通过回调函数清理缓冲区,这会导致我的程序出现严重的内存泄漏。 (我使用IoService::poll() 来调用处理程序。与async_read() 的调用率相比,该处理程序的调用速度非常慢)。
我再次尝试使用免费函数read,但这并不能解决我的目的,因为它会因我提供的缓冲区大小而阻塞太多时间。
我之前的服务器实现是使用WinSock API,我可以使用WinSock::recv() 接收所有数据。
请给我一些线索,以便我可以使用boost::asio 接收完整的数据。
这是我的服务器端线程循环
void
TCPObject::receive()
{
if (!_asyncModeEnabled)
{
std::string recvString;
if ( !_tcpSocket->receiveData( _maxBufferSize, recvString ) )
{
LOG_ERROR("Error Occurred while receiving data on socket.");
}
else
_parseAndPopulateQueue ( recvString );
}
else
{
if ( !_tcpSocket->receiveDataAsync( _maxBufferSize ) )
{
LOG_ERROR("Error Occurred while receiving data on socket.");
}
}
}
TCPSocket 中的receiveData()
bool
TCPSocket::receiveData( unsigned int bufferSize, std::string& dataString )
{
boost::system::error_code error;
char *buf = new char[bufferSize + 1];
size_t len = _tcpSocket->read_some( boost::asio::buffer((void*)buf, bufferSize), error);
if(error)
{
LOG_ERROR("Error in receiving data.");
LOG_ERROR( error.message() );
_tcpSocket->close();
delete [] buf;
return false;
}
buf[len] ='\0';
dataString.insert( 0, buf );
delete [] buf;
return true;
}
TCP Socket 中的receiveDataAsync
bool
TCPSocket::receiveDataAsync( unsigned int bufferSize )
{
char *buf = new char[bufferSize + 1];
try
{
_tcpSocket->async_read_some( boost::asio::buffer( (void*)buf, bufferSize ),
boost::bind(&TCPSocket::_handleAsyncReceive,
this,
buf,
boost::asio::placeholders::error,
boost::asio::placeholders::bytes_transferred) );
//! Asks io_service to execute callback
_ioService->poll();
}
catch (std::exception& e)
{
LOG_ERROR("Error Receiving Data Asynchronously");
LOG_ERROR( e.what() );
delete [] buf;
return false;
}
//we dont delete buf here as it will be deleted by callback _handleAsyncReceive
return true;
}
异步接收处理程序
void
TCPSocket::_handleAsyncReceive(char *buf, const boost::system::error_code& ec, size_t size)
{
if(ec)
{
LOG_ERROR ("Error occurred while sending data Asynchronously.");
LOG_ERROR ( ec.message() );
}
else if ( size > 0 )
{
buf[size] = '\0';
emit _asyncDataReceivedSignal( QString::fromLocal8Bit( buf ) );
}
delete [] buf;
}
客户端 sendData 函数。
sendData(std::string data)
{
if(!_connected)
{
return;
}
const char *pBuffer = data.c_str();
int bytes = data.length() + 1;
int i = 0,j;
while (i < bytes)
{
j = send(_connectSocket, pBuffer+i, bytes-i, 0);
if(j == SOCKET_ERROR)
{
_connected = false;
if(!_bNetworkErrNotified)
{
_bNetworkErrNotified=true;
emit networkErrorSignal(j);
}
LOG_ERROR( "Unable to send Network Packet" );
break;
}
i += j;
}
}
【问题讨论】:
-
如果您需要帮助,请发布您的代码。包括服务器代码和客户端代码,因为问题可能出在其中一个或两个上。
标签: sockets network-programming boost-asio