UnrealEngineUWP/Engine/Source/Runtime/NetworkFile/Private/PlatformTransport.cpp

// Copyright Epic Games, Inc. All Rights Reserved.

#include "PlatformTransport.h"
#include "NetworkMessage.h"
#include "NetworkPlatformFile.h"
#include "HAL/PlatformProcess.h"
#include "GenericPlatform/GenericPlatformHostSocket.h"
#include "GenericPlatform/GenericPlatformHostCommunication.h"


namespace
{
	/**
	 * Socket abstraction required by FNFSMessageHeader that uses IPlatformHostSocket.
	 */
	class FSimpleAbstractSocket_PlatformProtocol : public FSimpleAbstractSocket
	{
	public:

		FSimpleAbstractSocket_PlatformProtocol(IPlatformHostSocketPtr InHostSocket)
			: HostSocket(InHostSocket)
		{
			check(HostSocket != nullptr);
		}

		virtual bool Receive(uint8* Results, int32 Size) const override
		{
			return (HostSocket->Receive(Results, Size) == IPlatformHostSocket::EResultNet::Ok);
		}

		virtual bool Send(const uint8* Buffer, int32 Size) const override
		{
			return (HostSocket->Send(Buffer, Size) == IPlatformHostSocket::EResultNet::Ok);
		}

		virtual uint32 GetMagic() const override
		{
			return 0x9E2B83C7;
		}

	private:
		IPlatformHostSocketPtr HostSocket;
	};
}


FPlatformTransport::FPlatformTransport(int32 InProtocolIndex, const FString& InProtocolName)
	: ProtocolIndex(InProtocolIndex)
	, ProtocolName(InProtocolName)
	, HostSocket(nullptr)
{
}


bool FPlatformTransport::Initialize(const TCHAR* InHostIp)
{
	check(HostSocket == nullptr);

	IPlatformHostCommunication& HostCommunication = FPlatformMisc::GetPlatformHostCommunication();

	if (!HostCommunication.Available())
	{
		return false;
	}

	HostSocket = HostCommunication.OpenConnection(ProtocolIndex, ProtocolName);

	if (!HostSocket)
	{
		return false;
	}

	UE_LOG(LogNetworkPlatformFile, Display, TEXT("Waiting for the server to accept the connection (custom protocol)..."));

	// We need to wait because this transport is a bit different than typical sockets.
	// In this case, it's the client (game) that enables the communication by opening the socket.
	// The server (pc) can connect only if it detects this enabled communication protocol.
	return WaitUntilConnected();
}


bool FPlatformTransport::SendPayloadAndReceiveResponse(TArray<uint8>& In, TArray<uint8>& Out)
{
	bool SendResult = false;

	SendResult = FNFSMessageHeader::WrapAndSendPayload(In, FSimpleAbstractSocket_PlatformProtocol(HostSocket));

	if (!SendResult)
		return false;

	FArrayReader Response;
	bool RetResult = false;
	RetResult = FNFSMessageHeader::ReceivePayload(Response, FSimpleAbstractSocket_PlatformProtocol(HostSocket));

	if (RetResult)
	{
		Out.Append(Response.GetData(), Response.Num());
		return true;
	}

	return false;
}


bool FPlatformTransport::ReceiveResponse(TArray<uint8>& Out)
{
	FArrayReader Response;
	bool RetResult = true;
	RetResult &= FNFSMessageHeader::ReceivePayload(Response, FSimpleAbstractSocket_PlatformProtocol(HostSocket));

	if (RetResult)
	{
		Out.Append(Response.GetData(), Response.Num());
		return true;
	}

	return false;
}


bool FPlatformTransport::WaitUntilConnected()
{
	while (true)
	{
		IPlatformHostSocket::EConnectionState ConnectionState = HostSocket->GetState();

		// We're interested in the Connected state, but break the loop in other non-default states
		// in case any error has appeared.
		if (HostSocket->GetState() != IPlatformHostSocket::EConnectionState::Created)
		{
			break;
		}

		FPlatformProcess::Sleep(0.5f);
	}

	return HostSocket->GetState() == IPlatformHostSocket::EConnectionState::Connected;
}


FPlatformTransport::~FPlatformTransport()
{
	if (HostSocket)
	{
		IPlatformHostCommunication& HostCommunication = FPlatformMisc::GetPlatformHostCommunication();

		if (HostCommunication.Available())
		{
			HostCommunication.CloseConnection(HostSocket);

			HostSocket = nullptr;
		}
	}
}