You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
732d31f633
#rb none #lockdown Nick.Penwarden #ROBOMERGE-OWNER: robert.manuszewski #ROBOMERGE-AUTHOR: jeanmichel.dignard #ROBOMERGE-SOURCE: CL 4784913 in //UE4/Main/... #ROBOMERGE-BOT: CORE (Main -> Dev-Core) [CL 4793830 by Robert Manuszewski in Dev-Core branch]
86 lines
2.2 KiB
C++
86 lines
2.2 KiB
C++
// Copyright 1998-2019 Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "AsyncConnection.h"
|
|
#include "Logging.h"
|
|
|
|
FAsyncConnection::FAsyncConnection(const std::string& ConnectionName, IAsyncConnectionObserver& Observer) :
|
|
Name(ConnectionName),
|
|
Observer(Observer)
|
|
{}
|
|
|
|
void FAsyncConnection::Connect(const std::string& IP, uint16_t Port)
|
|
{
|
|
SocketAddress.SetIP(IP);
|
|
SocketAddress.SetPort(Port);
|
|
|
|
verify(!Socket || Socket->GetState() == rtc::AsyncSocket::CS_CLOSED);
|
|
Socket.reset(rtc::ThreadManager::Instance()->CurrentThread()->socketserver()->CreateAsyncSocket(SocketAddress.family(), SOCK_STREAM));
|
|
|
|
Socket->SignalConnectEvent.connect(this, &FAsyncConnection::OnConnect);
|
|
Socket->SignalReadEvent.connect(this, &FAsyncConnection::OnRead);
|
|
Socket->SignalCloseEvent.connect(this, &FAsyncConnection::OnClose);
|
|
|
|
bReconnect = true;
|
|
|
|
EG_LOG(LogDefault, Log, "Connecting to %s %s:%d", Name.c_str(), IP.c_str(), Port);
|
|
if (Socket->Connect(SocketAddress) == SOCKET_ERROR)
|
|
{
|
|
OnClose(Socket.get(), SOCKET_ERROR);
|
|
}
|
|
}
|
|
|
|
void FAsyncConnection::Disconnect()
|
|
{
|
|
bReconnect = false;
|
|
Socket->Close();
|
|
}
|
|
|
|
void FAsyncConnection::OnConnect(rtc::AsyncSocket*)
|
|
{
|
|
EG_LOG(LogDefault, Log, "Connected to %s", Name.c_str());
|
|
bReportDisconnection = true;
|
|
Observer.OnConnect();
|
|
}
|
|
|
|
void FAsyncConnection::OnClose(rtc::AsyncSocket*, int Err)
|
|
{
|
|
if (bReportDisconnection)
|
|
{
|
|
EG_LOG(LogDefault, Warning, "Disconnected from %s, error %d. Reconnecting...", Name.c_str(), Err);
|
|
bReportDisconnection = false;
|
|
Observer.OnDisconnect(Err);
|
|
}
|
|
|
|
if (!bReconnect)
|
|
return;
|
|
|
|
while (Socket->Connect(SocketAddress) == SOCKET_ERROR)
|
|
{
|
|
}
|
|
}
|
|
|
|
void FAsyncConnection::OnRead(rtc::AsyncSocket*)
|
|
{
|
|
do
|
|
{
|
|
int ReceivedBytes = Socket->Recv(TmpReadBuffer, sizeof(TmpReadBuffer), nullptr);
|
|
if (ReceivedBytes <= 0)
|
|
{
|
|
break;
|
|
}
|
|
ReadBuffer.insert(ReadBuffer.end(), TmpReadBuffer, TmpReadBuffer + ReceivedBytes);
|
|
} while (true);
|
|
|
|
uint32_t Consumed = 0;
|
|
while (!ReadBuffer.empty() &&
|
|
(Consumed = Observer.OnRead(&ReadBuffer.front(), static_cast<uint32_t>(ReadBuffer.size()))) != 0)
|
|
{
|
|
ReadBuffer.erase(ReadBuffer.begin(), ReadBuffer.begin() + Consumed);
|
|
}
|
|
}
|
|
|
|
void FAsyncConnection::Send(const void* Data, uint32_t Size)
|
|
{
|
|
Socket->Send(Data, Size);
|
|
}
|