stiffly
126 lines
3.4 KiB
C++
126 lines
3.4 KiB
C++
#include "Network/TCPClient.h"
|
|
|
|
using namespace boost::asio::ip;
|
|
|
|
TCPClient::TCPClient()
|
|
{
|
|
}
|
|
|
|
TCPClient::~TCPClient()
|
|
{
|
|
}
|
|
|
|
bool TCPClient::Connect(std::string playerName, std::string address, int port)
|
|
{
|
|
if (m_Socket) {
|
|
if (m_IsConnected) {
|
|
Packet packet(MessageType::Connect, m_SendPacketID);
|
|
packet.WriteString(playerName);
|
|
Send(packet);
|
|
LOG_INFO("Connect message sent again!");
|
|
}
|
|
return true;
|
|
}
|
|
else if (!m_IsConnected) {
|
|
boost::system::error_code error = boost::asio::error::host_not_found;
|
|
m_Endpoint = tcp::endpoint(boost::asio::ip::address::from_string(address), port);
|
|
m_Socket = std::unique_ptr<tcp::socket>(new tcp::socket(m_IOService));
|
|
m_Socket->connect(m_Endpoint, error);
|
|
tcp::no_delay option(true);
|
|
m_Socket->set_option(option);
|
|
LOG_INFO(error.message().c_str());
|
|
if (!error) {
|
|
m_IsConnected = true;
|
|
Packet packet(MessageType::Connect, m_SendPacketID);
|
|
packet.WriteString(playerName);
|
|
Send(packet);
|
|
LOG_INFO("Connect message sent!");
|
|
return true;
|
|
}
|
|
// If error
|
|
else {
|
|
m_Socket->close();
|
|
m_Socket = nullptr;
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
void TCPClient::Disconnect()
|
|
{
|
|
if (!m_IsConnected) {
|
|
return;
|
|
}
|
|
m_Socket->shutdown(boost::asio::ip::tcp::socket::shutdown_both);
|
|
m_Socket->close();
|
|
m_Socket = nullptr;
|
|
m_IsConnected = false;
|
|
}
|
|
|
|
void TCPClient::Receive(Packet& packet)
|
|
{
|
|
size_t bytesRead = readBuffer();
|
|
if (bytesRead > 0) {
|
|
packet.ReconstructFromData(m_ReadBuffer, bytesRead);
|
|
}
|
|
}
|
|
|
|
size_t TCPClient::readBuffer()
|
|
{
|
|
if (!m_Socket) {
|
|
return 0;
|
|
}
|
|
boost::system::error_code error;
|
|
// Read size of packet
|
|
m_Socket->receive(boost
|
|
::asio::buffer((void*)m_ReadBuffer, sizeof(int)),
|
|
boost::asio::ip::tcp::socket::message_peek, error);
|
|
unsigned int sizeOfPacket = 0;
|
|
memcpy(&sizeOfPacket, m_ReadBuffer, sizeof(int));
|
|
if (sizeOfPacket > m_Socket->available()) {
|
|
LOG_WARNING("TCPClient::readBuffer(): We haven't got the whole packet yet.");
|
|
//return 0;
|
|
}
|
|
// if the buffer is to small increase the size of it
|
|
// TODO if message is huge 1 time the buffer will not decrease.
|
|
if (sizeOfPacket > m_BufferSize) {
|
|
delete[] m_ReadBuffer;
|
|
m_ReadBuffer = new char[sizeOfPacket];
|
|
m_BufferSize = sizeOfPacket;
|
|
}
|
|
size_t bytesReceived = 0;
|
|
while (sizeOfPacket > bytesReceived) {
|
|
// Read the rest of the message
|
|
bytesReceived += m_Socket->read_some(boost
|
|
::asio::buffer((void*)(m_ReadBuffer), sizeOfPacket - bytesReceived),
|
|
error);
|
|
if (error) {
|
|
//LOG_ERROR("receive: %s", error.message().c_str());
|
|
}
|
|
}
|
|
if (sizeOfPacket > 1000000)
|
|
LOG_WARNING("The packets received are bigger than 1MB");
|
|
|
|
return bytesReceived;
|
|
}
|
|
|
|
void TCPClient::Send(Packet & packet)
|
|
{
|
|
if (!m_Socket) {
|
|
LOG_WARNING("TCPClient::Send: Socket is null");
|
|
return;
|
|
}
|
|
packet.UpdateSize();
|
|
boost::system::error_code error;
|
|
m_Socket->send(boost::asio::buffer(
|
|
packet.Data(),
|
|
packet.Size()), 0, error);
|
|
}
|
|
|
|
bool TCPClient::IsSocketAvailable()
|
|
{
|
|
if (!m_Socket) {
|
|
return false;
|
|
}
|
|
return m_Socket->available();
|
|
} |