mirror of
https://github.com/vcmi/vcmi.git
synced 2025-02-09 13:14:02 +02:00
100 lines
2.4 KiB
C++
100 lines
2.4 KiB
C++
/*
|
|
* NetworkConnection.cpp, part of VCMI engine
|
|
*
|
|
* Authors: listed in file AUTHORS in main folder
|
|
*
|
|
* License: GNU General Public License v2.0 or later
|
|
* Full text of license available in license.txt file, in main folder
|
|
*
|
|
*/
|
|
#include "StdInc.h"
|
|
#include "NetworkConnection.h"
|
|
|
|
VCMI_LIB_NAMESPACE_BEGIN
|
|
|
|
NetworkConnection::NetworkConnection(const std::shared_ptr<NetworkSocket> & socket, INetworkConnectionListener & listener)
|
|
: socket(socket)
|
|
, listener(listener)
|
|
{
|
|
|
|
}
|
|
|
|
void NetworkConnection::start()
|
|
{
|
|
boost::asio::async_read(*socket,
|
|
readBuffer,
|
|
boost::asio::transfer_exactly(messageHeaderSize),
|
|
std::bind(&NetworkConnection::onHeaderReceived,this, _1));
|
|
}
|
|
|
|
void NetworkConnection::onHeaderReceived(const boost::system::error_code & ec)
|
|
{
|
|
if (ec)
|
|
{
|
|
listener.onDisconnected(shared_from_this());
|
|
return;
|
|
}
|
|
|
|
uint32_t messageSize = readPacketSize();
|
|
|
|
boost::asio::async_read(*socket,
|
|
readBuffer,
|
|
boost::asio::transfer_exactly(messageSize),
|
|
std::bind(&NetworkConnection::onPacketReceived,this, _1, messageSize));
|
|
}
|
|
|
|
uint32_t NetworkConnection::readPacketSize()
|
|
{
|
|
if (readBuffer.size() < messageHeaderSize)
|
|
throw std::runtime_error("Failed to read header!");
|
|
|
|
std::istream istream(&readBuffer);
|
|
|
|
uint32_t messageSize;
|
|
istream.read(reinterpret_cast<char *>(&messageSize), messageHeaderSize);
|
|
if (messageSize > messageMaxSize)
|
|
{
|
|
throw std::runtime_error("Invalid packet size!");
|
|
}
|
|
|
|
return messageSize;
|
|
}
|
|
|
|
void NetworkConnection::onPacketReceived(const boost::system::error_code & ec, uint32_t expectedPacketSize)
|
|
{
|
|
if (ec)
|
|
{
|
|
listener.onDisconnected(shared_from_this());
|
|
return;
|
|
}
|
|
|
|
if (readBuffer.size() < expectedPacketSize)
|
|
{
|
|
throw std::runtime_error("Failed to read header!");
|
|
}
|
|
|
|
std::vector<uint8_t> message;
|
|
|
|
message.resize(expectedPacketSize);
|
|
std::istream istream(&readBuffer);
|
|
istream.read(reinterpret_cast<char *>(message.data()), expectedPacketSize);
|
|
|
|
listener.onPacketReceived(shared_from_this(), message);
|
|
|
|
start();
|
|
}
|
|
|
|
void NetworkConnection::sendPacket(const std::vector<uint8_t> & message)
|
|
{
|
|
NetworkBuffer writeBuffer;
|
|
|
|
std::ostream ostream(&writeBuffer);
|
|
uint32_t messageSize = message.size();
|
|
ostream.write(reinterpret_cast<const char *>(&messageSize), messageHeaderSize);
|
|
ostream.write(reinterpret_cast<const char *>(message.data()), message.size());
|
|
|
|
boost::asio::write(*socket, writeBuffer );
|
|
}
|
|
|
|
VCMI_LIB_NAMESPACE_END
|