bool ImportObjImpl::GetMessage(FString & Result)
{
// get all data from client and create one long FString from it
// return the result when the client is out of pending data.
//FString Result;
uint32 DataSize = 0;
while (Client->HasPendingData(DataSize) && DataSize > 0)
{
//UE_LOG(LogM2U, Log, TEXT("pending data size %i"), DataSize);
// create data array to read from client
//FArrayReaderPtr Data = MakeShareable(new FArrayReader(true));
FArrayReader Data;
Data.SetNumUninitialized(DataSize);
int32 BytesRead = 0;
// read pending data into the Data array reader
if (Client->Recv(Data.GetData(), Data.Num(), BytesRead))
{
//UE_LOG(LogM2U, Log, TEXT("DataNum %i, BytesRead: %i"), Data.Num(), BytesRead);
// the data we receive is supposed to be ansi, but we will work with TCHAR, so we have to convert
int32 DestLen = TStringConvert<ANSICHAR, TCHAR>::ConvertedLength((char*)(Data.GetData()), Data.Num());
//UE_LOG(LogM2U, Log, TEXT("DestLen will be %i"), DestLen);
TCHAR* Dest = new TCHAR[DestLen + 1];
TStringConvert<ANSICHAR, TCHAR>::Convert(Dest, DestLen, (char*)(Data.GetData()), Data.Num());
Dest[DestLen] = '\0';
//FString Text(Dest); // FString from tchar array
//UE_LOG(LogM2U, Log, TEXT("server received %s"), *Text);
//UE_LOG(LogM2U, Log, TEXT("Server received: %s"), Dest);
//FString Result = ExecuteCommand(Dest, this);
//SendResponse(Result);
// add all the message parts to the Result
Result += Dest;
delete Dest;
}
}// while
if (!Result.IsEmpty())
return true;
else
return false;
}
bool FTcpMessageTransportConnection::ReceiveMessages()
{
uint32 PendingDataSize = 0;
// check if the socket has closed
{
int32 BytesRead;
uint8 Dummy;
if (!Socket->Recv(&Dummy, 1, BytesRead, ESocketReceiveFlags::Peek))
{
return false;
}
}
if (!bReceivedHeader)
{
if (Socket->HasPendingData(PendingDataSize) && PendingDataSize >= sizeof(FTcpMessageHeader))
{
FArrayReader HeaderData = FArrayReader(true);
HeaderData.SetNumUninitialized(sizeof(FTcpMessageHeader));
int32 BytesRead = 0;
if (!Socket->Recv(HeaderData.GetData(), sizeof(FTcpMessageHeader), BytesRead))
{
return false;
}
check(BytesRead == sizeof(FTcpMessageHeader));
TotalBytesReceived += BytesRead;
FTcpMessageHeader MessageHeader;
HeaderData << MessageHeader;
if (!MessageHeader.IsValid())
{
return false;
}
else
{
RemoteNodeId = MessageHeader.GetNodeId();
RemoteProtocolVersion = MessageHeader.GetVersion();
bReceivedHeader = true;
OpenedTime = FDateTime::UtcNow();
UpdateConnectionState(STATE_Connected);
}
}
else
{
// no header yet
return true;
}
}
// keep going until we have no data.
for(;;)
{
int32 BytesRead = 0;
// See if we're in the process of receiving a (large) message
if (RecvMessageDataRemaining == 0)
{
// no partial message. Try to receive the size of a message
if (!Socket->HasPendingData(PendingDataSize) || (PendingDataSize < sizeof(uint32)))
{
// no messages
return true;
}
FArrayReader MessagesizeData = FArrayReader(true);
MessagesizeData.SetNumUninitialized(sizeof(uint32));
// read message size from the stream
BytesRead = 0;
if (!Socket->Recv(MessagesizeData.GetData(), sizeof(uint32), BytesRead))
{
return false;
}
check(BytesRead == sizeof(uint32));
TotalBytesReceived += BytesRead;
// Setup variables to receive the message
MessagesizeData << RecvMessageDataRemaining;
RecvMessageData = MakeShareable(new FArrayReader(true));
RecvMessageData->SetNumUninitialized(RecvMessageDataRemaining);
check(RecvMessageDataRemaining > 0);
}
BytesRead = 0;
if (!Socket->Recv(RecvMessageData->GetData() + RecvMessageData->Num() - RecvMessageDataRemaining, RecvMessageDataRemaining, BytesRead))
{
return false;
}
if (BytesRead > 0)
{
check(BytesRead <= RecvMessageDataRemaining);
TotalBytesReceived += BytesRead;
RecvMessageDataRemaining -= BytesRead;
if (RecvMessageDataRemaining == 0)
{
// @todo gmp: move message deserialization into an async task
FTcpDeserializedMessage* DeserializedMessage = new FTcpDeserializedMessage(nullptr);
if (DeserializedMessage->Deserialize(RecvMessageData))
{
Inbox.Enqueue(MakeShareable(DeserializedMessage));
}
RecvMessageData.Reset();
}
}
else
{
// no data
return true;
}
}
}
