Files
SLikeNet/Samples/AutopatcherClientGFx3.0/AutopatcherClientGFx3Impl.cpp
2025-11-24 14:19:51 +05:30

343 lines
12 KiB
C++

/*
* Original work: Copyright (c) 2014, Oculus VR, Inc.
* All rights reserved.
*
* This source code is licensed under the BSD-style license found in the
* RakNet License.txt file in the licenses directory of this source tree. An additional grant
* of patent rights can be found in the RakNet Patents.txt file in the same directory.
*
*
* Modified work: Copyright (c) 2016-2020, SLikeSoft UG (haftungsbeschränkt)
*
* This source code was modified by SLikeSoft. Modifications are licensed under the MIT-style
* license found in the license.txt file in the root directory of this source tree.
*/
// Common includes
#include <stdio.h>
#include <stdlib.h>
#include "slikenet/Kbhit.h"
#include "slikenet/GetTime.h"
#include "slikenet/peerinterface.h"
#include "slikenet/MessageIdentifiers.h"
#include "slikenet/BitStream.h"
#include "slikenet/StringCompressor.h"
#include "slikenet/PacketizedTCP.h"
// Client only includes
#include "slikenet/FileListTransferCBInterface.h"
#include "slikenet/FileListTransfer.h"
#include "AutopatcherClient.h"
#include "slikenet/AutopatcherPatchContext.h"
#include "slikenet/sleep.h"
#include "slikenet/linux_adapter.h"
#include "slikenet/osx_adapter.h"
#include "AutopatcherClientGFx3Impl.h"
using namespace SLNet;
static const char *AUTOPATCHER_LAST_UPDATE_FILE="autopatcherLastUpdate.txt";
static const char *AUTOPATCHER_RESTART_FILE="autopatcherRestart.txt";
class TestCB : public SLNet::FileListTransferCBInterface
{
public:
virtual bool OnFile(OnFileStruct *onFileStruct)
{
char buff[1024];
if (onFileStruct->context.op==PC_HASH_1_WITH_PATCH || onFileStruct->context.op==PC_HASH_2_WITH_PATCH)
strcpy_s(buff,"Patched: ");
else if (onFileStruct->context.op==PC_WRITE_FILE)
strcpy_s(buff,"Written: ");
else if (onFileStruct->context.op==PC_ERROR_FILE_WRITE_FAILURE)
strcpy_s(buff,"Write Failure: ");
else if (onFileStruct->context.op==PC_ERROR_PATCH_TARGET_MISSING)
strcpy_s(buff,"Patch target missing: ");
else if (onFileStruct->context.op==PC_ERROR_PATCH_APPLICATION_FAILURE)
strcpy_s(buff,"Patch process failure: ");
else if (onFileStruct->context.op==PC_ERROR_PATCH_RESULT_CHECKSUM_FAILURE)
strcpy_s(buff,"Patch checksum failure: ");
else if (onFileStruct->context.op==PC_NOTICE_WILL_COPY_ON_RESTART)
strcpy_s(buff,"Copy pending restart: ");
else if (onFileStruct->context.op==PC_NOTICE_FILE_DOWNLOADED)
strcpy_s(buff,"Downloaded: ");
else if (onFileStruct->context.op==PC_NOTICE_FILE_DOWNLOADED_PATCH)
strcpy_s(buff,"Downloaded Patch: ");
else
RakAssert(0);
sprintf(buff+strlen(buff), "%i. (100%%) %i/%i %s %ib / %ib\n", onFileStruct->setID, onFileStruct->fileIndex+1, onFileStruct->numberOfFilesInThisSet,
onFileStruct->fileName, onFileStruct->byteLengthOfThisFile,
onFileStruct->byteLengthOfThisSet);
FxResponseArgs<1> args;
args.Add(GFxValue(buff));
FxDelegate::Invoke2(autopatcherClient->movie, "addToPatchNotesText", args);
FxResponseArgs<5> args2;
args2.Add(GFxValue(buff));
args2.Add(GFxValue(1.0));
args2.Add(GFxValue(1.0));
args2.Add(GFxValue((double)onFileStruct->bytesDownloadedForThisSet));
args2.Add(GFxValue((double)onFileStruct->byteLengthOfThisSet));
FxDelegate::Invoke2(autopatcherClient->movie, "updateProgressBars", args2);
// Return false for the file data to be deallocated automatically
return false;
}
virtual void OnFileProgress(FileProgressStruct *fps)
{
char buff[1024];
sprintf(buff, "%s %ib / %ib\n", fps->onFileStruct->fileName,
fps->onFileStruct->bytesDownloadedForThisFile, fps->onFileStruct->byteLengthOfThisFile);
FxResponseArgs<5> args2;
float thisFileProgress,totalProgress;
thisFileProgress=(float)fps->partCount/(float)fps->partTotal;
totalProgress=(float)(fps->onFileStruct->fileIndex+1)/(float)fps->onFileStruct->numberOfFilesInThisSet;
args2.Add(GFxValue(buff));
args2.Add(GFxValue((double)fps->onFileStruct->bytesDownloadedForThisFile));
args2.Add(GFxValue((double)fps->onFileStruct->byteLengthOfThisFile));
args2.Add(GFxValue((double)fps->onFileStruct->bytesDownloadedForThisSet));
args2.Add(GFxValue((double)fps->onFileStruct->byteLengthOfThisSet));
FxDelegate::Invoke2(autopatcherClient->movie, "updateProgressBars", args2);
}
AutopatcherClientGFx3Impl *autopatcherClient;
} transferCallback;
AutopatcherClientGFx3Impl::AutopatcherClientGFx3Impl()
{
autopatcherClient=0;
fileListTransfer=0;
packetizedTCP=0;
}
AutopatcherClientGFx3Impl::~AutopatcherClientGFx3Impl()
{
Shutdown();
}
void AutopatcherClientGFx3Impl::Init(const char *_pathToThisExe, GPtr<FxDelegate> pDelegate, GPtr<GFxMovieView> pMovie)
{
pDelegate->RegisterHandler(this);
delegate=pDelegate;
movie=pMovie;
strcpy_s(pathToThisExe,_pathToThisExe);
autopatcherClient= SLNet::OP_NEW<AutopatcherClient>(_FILE_AND_LINE_);
fileListTransfer= SLNet::OP_NEW<FileListTransfer>(_FILE_AND_LINE_);
packetizedTCP= SLNet::OP_NEW<PacketizedTCP>(_FILE_AND_LINE_);
autopatcherClient->SetFileListTransferPlugin(fileListTransfer);
packetizedTCP->AttachPlugin(autopatcherClient);
packetizedTCP->AttachPlugin(fileListTransfer);
}
void AutopatcherClientGFx3Impl::Update(void)
{
SLNet::Packet *p;
SystemAddress notificationAddress;
notificationAddress=packetizedTCP->HasCompletedConnectionAttempt();
if (notificationAddress!= SLNet::UNASSIGNED_SYSTEM_ADDRESS)
{
UpdateConnectResult(true);
serverAddress=notificationAddress;
}
notificationAddress=packetizedTCP->HasFailedConnectionAttempt();
if (notificationAddress!= SLNet::UNASSIGNED_SYSTEM_ADDRESS)
{
UpdateConnectResult(false);
}
notificationAddress=packetizedTCP->HasNewIncomingConnection();
notificationAddress=packetizedTCP->HasLostConnection();
if (notificationAddress!= SLNet::UNASSIGNED_SYSTEM_ADDRESS)
{
UpdateConnectResult(false);
}
p=packetizedTCP->Receive();
while (p)
{
if (p->data[0]==ID_AUTOPATCHER_REPOSITORY_FATAL_ERROR)
{
char buff[256];
SLNet::BitStream temp(p->data, p->length, false);
temp.IgnoreBits(8);
StringCompressor::Instance()->DecodeString(buff, 256, &temp);
// Error.
FxDelegate::Invoke2(movie, "gotoCompletionMenu", FxResponseArgs<0>());
FxResponseArgs<1> args2;
args2.Add(GFxValue(buff));
FxDelegate::Invoke2(movie, "setCompletionMessage", args2);
}
else if (p->data[0]==ID_AUTOPATCHER_FINISHED)
{
FxDelegate::Invoke2(movie, "gotoCompletionMenu", FxResponseArgs<0>());
SaveLastUpdateDate();
}
else if (p->data[0]==ID_AUTOPATCHER_RESTART_APPLICATION)
{
FxDelegate::Invoke2(movie, "gotoCompletionMenu", FxResponseArgs<0>());
FxResponseArgs<1> args2;
SLNet::RakString completionMsg("Launch \"AutopatcherClientRestarter.exe %s\"\nQuit this application immediately after to unlock files.\n", AUTOPATCHER_RESTART_FILE);
args2.Add(GFxValue(completionMsg.C_String()));
FxDelegate::Invoke2(movie, "setCompletionMessage", args2);
SaveLastUpdateDate();
}
packetizedTCP->DeallocatePacket(p);
p=packetizedTCP->Receive();
}
}
void AutopatcherClientGFx3Impl::Shutdown(void)
{
if (delegate.GetPtr()!=0)
{
delegate->UnregisterHandler(this);
delegate.Clear();
}
movie.Clear();
if (packetizedTCP)
packetizedTCP->Stop();
SLNet::OP_DELETE(autopatcherClient,_FILE_AND_LINE_);
SLNet::OP_DELETE(fileListTransfer,_FILE_AND_LINE_);
SLNet::OP_DELETE(packetizedTCP,_FILE_AND_LINE_);
autopatcherClient=0;
fileListTransfer=0;
packetizedTCP=0;
}
const char* AutopatcherClientGFx3Impl::Connect(const char *ip, unsigned short port)
{
if (packetizedTCP->Start(0,1)==true)
{
packetizedTCP->Connect(ip,port,false);
return "Connecting";
}
else
return "Start call failed.";
}
void AutopatcherClientGFx3Impl::PressedPatch(const FxDelegateArgs& pparams)
{
AutopatcherClientGFx3Impl* prt = (AutopatcherClientGFx3Impl*)pparams.GetHandler();
//appNameText.text, appDirectoryText.text, fullRescanBtn.selected
const char *appName = pparams[0].GetString();
const char *appDir = pparams[1].GetString();
bool fullRescan = pparams[2].GetBool();
strcpy_s(prt->appDirectory, appDir);
char restartFile[512];
strcpy_s(restartFile, appDir);
strcat(restartFile, "/");
strcat(restartFile, AUTOPATCHER_RESTART_FILE);
double lastUpdateDate;
if (fullRescan==false)
prt->LoadLastUpdateDate(&lastUpdateDate,appDir);
else
lastUpdateDate=0;
transferCallback.autopatcherClient=prt;
if (prt->autopatcherClient->PatchApplication(appName, appDir, lastUpdateDate, prt->serverAddress, &transferCallback, restartFile, prt->pathToThisExe))
{
FxDelegate::Invoke2(prt->movie, "gotoPatchMenu", FxResponseArgs<0>());
}
else
{
prt->packetizedTCP->Stop();
//prt->UpdateConnectResult("Failed to start patching");
FxDelegate::Invoke2(prt->movie, "gotoPatchStartMenu", FxResponseArgs<0>());
}
}
void AutopatcherClientGFx3Impl::OpenSite(const FxDelegateArgs& pparams)
{
AutopatcherClientGFx3Impl* prt = (AutopatcherClientGFx3Impl*)pparams.GetHandler();
const char *siteType = pparams[0].GetString();
if (_stricmp(siteType, "help")==0)
{
ShellExecute(nullptr, "open", "http://www.jenkinssoftware.com/raknet/manual/autopatcher.html", nullptr, nullptr, SW_SHOWNORMAL);
}
else if (_stricmp(siteType, "raknet")==0)
{
ShellExecute(nullptr, "open", "http://www.jenkinssoftware.com/", nullptr, nullptr, SW_SHOWNORMAL);
}
else if (_stricmp(siteType, "scaleform")==0)
{
ShellExecute(nullptr, "open", "https://www.scaleform.com/", nullptr, nullptr, SW_SHOWNORMAL);
}
}
void AutopatcherClientGFx3Impl::PressedConnect(const FxDelegateArgs& pparams)
{
AutopatcherClientGFx3Impl* prt = (AutopatcherClientGFx3Impl*)pparams.GetHandler();
const char *result = prt->Connect(pparams[0].GetString(), atoi(pparams[1].GetString()));
}
void AutopatcherClientGFx3Impl::PressedOKBtn(const FxDelegateArgs& pparams)
{
AutopatcherClientGFx3Impl* prt = (AutopatcherClientGFx3Impl*)pparams.GetHandler();
prt->autopatcherClient->Clear();
prt->packetizedTCP->Stop();
prt->GotoMainMenu();
}
void AutopatcherClientGFx3Impl::UpdateConnectResult( bool isConnected )
{
FxResponseArgs<1> args;
args.Add(GFxValue(isConnected));
FxDelegate::Invoke2(movie, "ConnectResult", args);
}
void AutopatcherClientGFx3Impl::Accept(CallbackProcessor* cbreg)
{
cbreg->Process( "PressedOKBtn", &AutopatcherClientGFx3Impl::PressedOKBtn );
cbreg->Process( "PressedConnect", &AutopatcherClientGFx3Impl::PressedConnect );
cbreg->Process( "PressedPatch", &AutopatcherClientGFx3Impl::PressedPatch );
cbreg->Process( "openSite", &AutopatcherClientGFx3Impl::OpenSite );
}
void AutopatcherClientGFx3Impl::SaveLastUpdateDate(void)
{
char inPath[512];
double serverDate=autopatcherClient->GetServerDate();
strcpy_s(inPath, appDirectory);
strcat(inPath, "/");
strcat(inPath, AUTOPATCHER_LAST_UPDATE_FILE);
FILE *fp;
if (fopen_s(&fp,inPath,"wb")==0)
{
fwrite(&serverDate,sizeof(double),1,fp);
fclose(fp);
}
}
void AutopatcherClientGFx3Impl::LoadLastUpdateDate(double *out, const char *appDir)
{
char inPath[512];
strcpy_s(appDirectory,appDir);
strcpy_s(inPath, appDirectory);
strcat(inPath, "/");
strcat(inPath, AUTOPATCHER_LAST_UPDATE_FILE);
FILE *fp;
if (fopen_s(&fp,inPath,"rb")==0)
{
fread(out,sizeof(double),1,fp);
fclose(fp);
}
else
out[0]=0;
}
void AutopatcherClientGFx3Impl::GotoMainMenu(void)
{
FxDelegate::Invoke2(movie, "gotoMainMenu", FxResponseArgs<0>());
autopatcherClient->Clear();
packetizedTCP->Stop();
}