113 lines
4.0 KiB
C++
113 lines
4.0 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "CoreMinimal.h"
|
|
#include "Misc/CommandLine.h"
|
|
#include "Modules/ModuleManager.h"
|
|
#include "INetworkFileSystemModule.h"
|
|
#include "NetworkFileSystemLog.h"
|
|
#include "NetworkFileServer.h"
|
|
#include "Interfaces/ITargetPlatformManagerModule.h"
|
|
#include "CookOnTheFlyNetServer.h"
|
|
#include "IPAddress.h"
|
|
|
|
DEFINE_LOG_CATEGORY(LogFileServer);
|
|
|
|
|
|
/**
|
|
* Implements the NetworkFileSystem module.
|
|
*/
|
|
class FNetworkFileSystemModule
|
|
: public INetworkFileSystemModule
|
|
{
|
|
public:
|
|
|
|
// INetworkFileSystemModule interface
|
|
|
|
virtual INetworkFileServer* CreateNetworkFileServer( bool bLoadTargetPlatforms, int32 Port, FNetworkFileDelegateContainer NetworkFileDelegateContainer, const ENetworkFileServerProtocol Protocol ) const
|
|
{
|
|
FNetworkFileServerOptions FileServerOptions;
|
|
FileServerOptions.Protocol = Protocol;
|
|
FileServerOptions.Port = Port;
|
|
FileServerOptions.Delegates = MoveTemp(NetworkFileDelegateContainer);
|
|
FileServerOptions.bRestrictPackageAssetsToSandbox = false;
|
|
|
|
return CreateNetworkFileServer(MoveTemp(FileServerOptions), bLoadTargetPlatforms);
|
|
}
|
|
|
|
virtual INetworkFileServer* CreateNetworkFileServer(FNetworkFileServerOptions FileServerOptions, bool bLoadTargetPlatforms) const override
|
|
{
|
|
if (bLoadTargetPlatforms)
|
|
{
|
|
ITargetPlatformManagerModule& TPM = GetTargetPlatformManagerRef();
|
|
|
|
// if we didn't specify a target platform then use the entire target platform list (they could all be possible!)
|
|
FString Platforms;
|
|
if (FParse::Value(FCommandLine::Get(), TEXT("TARGETPLATFORM="), Platforms))
|
|
{
|
|
FileServerOptions.TargetPlatforms = TPM.GetActiveTargetPlatforms();
|
|
}
|
|
else
|
|
{
|
|
FileServerOptions.TargetPlatforms = TPM.GetTargetPlatforms();
|
|
}
|
|
}
|
|
|
|
switch (FileServerOptions.Protocol)
|
|
{
|
|
case NFSP_Tcp:
|
|
case NFSP_Platform:
|
|
UE::Cook::FCookOnTheFlyNetworkServerOptions CookOnTheFlyServerOptions;
|
|
if (FileServerOptions.Protocol == NFSP_Tcp)
|
|
{
|
|
CookOnTheFlyServerOptions.Protocol = UE::Cook::ECookOnTheFlyNetworkServerProtocol::Tcp;
|
|
}
|
|
else
|
|
{
|
|
CookOnTheFlyServerOptions.Protocol = UE::Cook::ECookOnTheFlyNetworkServerProtocol::Platform;
|
|
}
|
|
CookOnTheFlyServerOptions.TargetPlatforms = FileServerOptions.TargetPlatforms;
|
|
CookOnTheFlyServerOptions.Port = FileServerOptions.Port;
|
|
UE::Cook::ICookOnTheFlyNetworkServerModule& CookOnTheFlyNetworkServerModule = FModuleManager::LoadModuleChecked<UE::Cook::ICookOnTheFlyNetworkServerModule>(TEXT("CookOnTheFlyNetServer"));
|
|
TSharedPtr<UE::Cook::ICookOnTheFlyNetworkServer> CookOnTheFlyNetworkServer = CookOnTheFlyNetworkServerModule.CreateServer(CookOnTheFlyServerOptions);
|
|
if (!CookOnTheFlyNetworkServer)
|
|
{
|
|
return nullptr;
|
|
}
|
|
UE_LOG(LogFileServer, Display, TEXT("Unreal Network File Server starting up..."));
|
|
if (CookOnTheFlyNetworkServer->Start())
|
|
{
|
|
TArray<TSharedPtr<FInternetAddr>> ListenAddresses;
|
|
if (CookOnTheFlyNetworkServer->GetAddressList(ListenAddresses))
|
|
{
|
|
UE_LOG(LogFileServer, Display, TEXT("Unreal Network File Server is ready for client connections on %s!"), *ListenAddresses[0]->ToString(true));
|
|
}
|
|
return new FNetworkFileServer(MoveTemp(FileServerOptions), CookOnTheFlyNetworkServer.ToSharedRef());
|
|
}
|
|
break;
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
virtual INetworkFileServer* CreateNetworkFileServer(TSharedRef<UE::Cook::ICookOnTheFlyNetworkServer> CookOnTheFlyNetworkServer, FNetworkFileDelegateContainer Delegates) const override
|
|
{
|
|
FNetworkFileServerOptions FileServerOptions;
|
|
FileServerOptions.Delegates = MoveTemp(Delegates);
|
|
FileServerOptions.bRestrictPackageAssetsToSandbox = true;
|
|
ITargetPlatformManagerModule& TPM = GetTargetPlatformManagerRef();
|
|
FString Platforms;
|
|
if (FParse::Value(FCommandLine::Get(), TEXT("TARGETPLATFORM="), Platforms))
|
|
{
|
|
FileServerOptions.TargetPlatforms = TPM.GetActiveTargetPlatforms();
|
|
}
|
|
else
|
|
{
|
|
FileServerOptions.TargetPlatforms = TPM.GetTargetPlatforms();
|
|
}
|
|
return new FNetworkFileServer(MoveTemp(FileServerOptions), CookOnTheFlyNetworkServer);
|
|
}
|
|
};
|
|
|
|
|
|
IMPLEMENT_MODULE(FNetworkFileSystemModule, NetworkFileSystem);
|