main #2
@ -21,24 +21,18 @@ namespace PatchClient
|
|||||||
{
|
{
|
||||||
Task.Run(() =>
|
Task.Run(() =>
|
||||||
{
|
{
|
||||||
FilePatcher bp = new FilePatcher()
|
PatchHelper patcher = new PatchHelper(Environment.CurrentDirectory, null, LazyOperations.PatchFolder.FromCwd());
|
||||||
{
|
|
||||||
TargetBase = Environment.CurrentDirectory,
|
|
||||||
PatchBase = LazyOperations.PatchFolder.FromCwd()
|
|
||||||
};
|
|
||||||
|
|
||||||
bp.ProgressChanged += Bp_ProgressChanged;
|
patcher.ProgressChanged += patcher_ProgressChanged;
|
||||||
|
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
if (bp.Run())
|
LazyOperations.CleanupTempDir();
|
||||||
{
|
LazyOperations.PrepTempDir();
|
||||||
MessageBox.Show("Patch completed without issues", "Patching Successful");
|
|
||||||
}
|
string message = patcher.ApplyPatches();
|
||||||
else
|
|
||||||
{
|
MessageBox.Show(message, "Patcher");
|
||||||
MessageBox.Show("Failed to patch client.", "Patching Failed", MessageBoxButton.OK, MessageBoxImage.Error);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
catch(Exception ex)
|
catch(Exception ex)
|
||||||
{
|
{
|
||||||
@ -54,7 +48,7 @@ namespace PatchClient
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
private void Bp_ProgressChanged(object Sender, int Progress, int Total, int Percent, string Message = "", params LineItem[] AdditionalLineItems)
|
private void patcher_ProgressChanged(object Sender, int Progress, int Total, int Percent, string Message = "", params LineItem[] AdditionalLineItems)
|
||||||
{
|
{
|
||||||
string additionalInfo = "";
|
string additionalInfo = "";
|
||||||
foreach (LineItem item in AdditionalLineItems)
|
foreach (LineItem item in AdditionalLineItems)
|
||||||
|
@ -6,6 +6,14 @@
|
|||||||
<UseWPF>true</UseWPF>
|
<UseWPF>true</UseWPF>
|
||||||
</PropertyGroup>
|
</PropertyGroup>
|
||||||
|
|
||||||
|
<ItemGroup>
|
||||||
|
<None Remove="Resources\xdelta3.exe" />
|
||||||
|
</ItemGroup>
|
||||||
|
|
||||||
|
<ItemGroup>
|
||||||
|
<EmbeddedResource Include="Resources\xdelta3.exe" />
|
||||||
|
</ItemGroup>
|
||||||
|
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<ProjectReference Include="..\PatcherUtils\PatcherUtils.csproj" />
|
<ProjectReference Include="..\PatcherUtils\PatcherUtils.csproj" />
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
|
7
Patcher/PatchClient/Properties/launchSettings.json
Normal file
7
Patcher/PatchClient/Properties/launchSettings.json
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
{
|
||||||
|
"profiles": {
|
||||||
|
"PatchClient": {
|
||||||
|
"commandName": "Project"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
BIN
Patcher/PatchClient/Resources/xdelta3.exe
Normal file
BIN
Patcher/PatchClient/Resources/xdelta3.exe
Normal file
Binary file not shown.
@ -120,13 +120,14 @@ namespace PatchGenerator
|
|||||||
GenProgressBar.DispatcherSetIndetermination(false);
|
GenProgressBar.DispatcherSetIndetermination(false);
|
||||||
|
|
||||||
//generate patches
|
//generate patches
|
||||||
FileCompare bc = new FileCompare(targetFolder, compareFolder, patchBase);
|
//TODO - fix these weird variable names (why did I do this?)
|
||||||
|
PatchHelper patcher = new PatchHelper(compareFolder, targetFolder, patchBase);
|
||||||
|
|
||||||
bc.ProgressChanged += Bc_ProgressChanged;
|
patcher.ProgressChanged += patcher_ProgressChanged;
|
||||||
|
|
||||||
if (!bc.CompareAll())
|
if (!patcher.GeneratePatches())
|
||||||
{
|
{
|
||||||
MessageBox.Show("Failed to generate diffs.", ":(", MessageBoxButton.OK, MessageBoxImage.Error);
|
MessageBox.Show("One of the provided folder paths doesn't exist.", "Oops :(", MessageBoxButton.OK, MessageBoxImage.Error);
|
||||||
}
|
}
|
||||||
|
|
||||||
//Copy patch client to output folder
|
//Copy patch client to output folder
|
||||||
@ -142,7 +143,7 @@ namespace PatchGenerator
|
|||||||
GenProgressMessageLabel.DispaatcherSetContent("Done");
|
GenProgressMessageLabel.DispaatcherSetContent("Done");
|
||||||
}
|
}
|
||||||
|
|
||||||
private void Bc_ProgressChanged(object Sender, int Progress, int Total, int Percent, string Message = "", params LineItem[] AdditionalLineItems)
|
private void patcher_ProgressChanged(object Sender, int Progress, int Total, int Percent, string Message = "", params LineItem[] AdditionalLineItems)
|
||||||
{
|
{
|
||||||
string additionalInfo = "";
|
string additionalInfo = "";
|
||||||
foreach (LineItem item in AdditionalLineItems)
|
foreach (LineItem item in AdditionalLineItems)
|
||||||
|
@ -11,15 +11,9 @@
|
|||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
|
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<Reference Include="Aki.ByteBanger">
|
|
||||||
<HintPath>References\Aki.ByteBanger.dll</HintPath>
|
|
||||||
</Reference>
|
|
||||||
<Reference Include="Aki.Common">
|
<Reference Include="Aki.Common">
|
||||||
<HintPath>References\Aki.Common.dll</HintPath>
|
<HintPath>References\Aki.Common.dll</HintPath>
|
||||||
</Reference>
|
</Reference>
|
||||||
<Reference Include="ComponentAce.Compression.Libs.zlib">
|
|
||||||
<HintPath>References\ComponentAce.Compression.Libs.zlib.dll</HintPath>
|
|
||||||
</Reference>
|
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
|
|
||||||
</Project>
|
</Project>
|
||||||
|
@ -1,185 +0,0 @@
|
|||||||
// NOTES:
|
|
||||||
// - redo search pattern;
|
|
||||||
// - compare both directories against eachother, not just one to the other
|
|
||||||
// - add ability to handle missing directories
|
|
||||||
|
|
||||||
using System.IO;
|
|
||||||
using Aki.Common.Utils;
|
|
||||||
using Aki.ByteBanger;
|
|
||||||
using System;
|
|
||||||
using System.Collections.Generic;
|
|
||||||
using System.Linq;
|
|
||||||
|
|
||||||
namespace PatcherUtils
|
|
||||||
{
|
|
||||||
public class FileCompare
|
|
||||||
{
|
|
||||||
public string PatchBase;
|
|
||||||
public string TargetBase;
|
|
||||||
public string CompareBase;
|
|
||||||
private int fileCount;
|
|
||||||
private int fileIt;
|
|
||||||
|
|
||||||
private int diffCount = 0;
|
|
||||||
private int newCount = 0;
|
|
||||||
private int delCount = 0;
|
|
||||||
private int matchCount = 0;
|
|
||||||
|
|
||||||
private List<FileInfo> TargetPaths;
|
|
||||||
private List<FileInfo> ComparePaths;
|
|
||||||
private List<LineItem> AdditionalInfo = new List<LineItem>();
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Provides patch generation progress changes
|
|
||||||
/// </summary>
|
|
||||||
public event ProgressChangedHandler ProgressChanged;
|
|
||||||
protected virtual void RaiseProgressChanged(int progress, int total, string Message = "", params LineItem[] AdditionalLineItems)
|
|
||||||
{
|
|
||||||
int percent = (int)Math.Floor((double)progress / total * 100);
|
|
||||||
|
|
||||||
ProgressChanged?.Invoke(this, progress, total, percent, Message, AdditionalLineItems);
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Compare a target file to an assumed compareable file.
|
|
||||||
/// </summary>
|
|
||||||
/// <param name="targetFile">The known target path</param>
|
|
||||||
/// <param name="assumedCompareFile">The assumed comparable file path</param>
|
|
||||||
/// <returns>True if a comparison was made | False if a comparison could not be made</returns>
|
|
||||||
private bool Compare(string targetFile, string assumedCompareFile)
|
|
||||||
{
|
|
||||||
string patchFilePath = targetFile.Replace(TargetBase, PatchBase);
|
|
||||||
//we know our target file exists
|
|
||||||
byte[] targetData = VFS.ReadFile(targetFile);
|
|
||||||
|
|
||||||
if(!File.Exists(assumedCompareFile))
|
|
||||||
{
|
|
||||||
//save the data we won't have in our target as new
|
|
||||||
VFS.WriteFile($"{patchFilePath}.new", Zlib.Compress(targetData, ZlibCompression.Maximum));
|
|
||||||
newCount++;
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
//now our compare file is known to exist
|
|
||||||
byte[] compareData = VFS.ReadFile(assumedCompareFile);
|
|
||||||
|
|
||||||
// get diffs
|
|
||||||
DiffResult result = PatchUtil.Diff(compareData, targetData);
|
|
||||||
|
|
||||||
switch (result.Result)
|
|
||||||
{
|
|
||||||
case DiffResultType.Success:
|
|
||||||
VFS.WriteFile($"{patchFilePath}.bpf", result.PatchInfo.ToBytes());
|
|
||||||
diffCount++;
|
|
||||||
return true;
|
|
||||||
|
|
||||||
case DiffResultType.FilesMatch:
|
|
||||||
matchCount++;
|
|
||||||
return true;
|
|
||||||
|
|
||||||
default:
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// <summary>
|
|
||||||
/// Compares the base folders and generates patch files.
|
|
||||||
/// </summary>
|
|
||||||
/// <returns>True if patches were generated successfully | False if patch generation failed</returns>
|
|
||||||
public bool CompareAll()
|
|
||||||
{
|
|
||||||
DirectoryInfo targetDirInfo = new DirectoryInfo(TargetBase);
|
|
||||||
DirectoryInfo compareDirInfo = new DirectoryInfo(CompareBase);
|
|
||||||
|
|
||||||
AdditionalInfo.Add(new LineItem("Diff Patch", "0"));
|
|
||||||
AdditionalInfo.Add(new LineItem("New Patch", "0"));
|
|
||||||
AdditionalInfo.Add(new LineItem("Del Patch", "0"));
|
|
||||||
AdditionalInfo.Add(new LineItem("Files Match", "0"));
|
|
||||||
|
|
||||||
if (!targetDirInfo.Exists || !compareDirInfo.Exists)
|
|
||||||
{
|
|
||||||
Console.WriteLine("Target or Compare folder does not exist");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
//Get all the files recursively
|
|
||||||
TargetPaths = new List<FileInfo>(targetDirInfo.GetFiles("*.*", SearchOption.AllDirectories));
|
|
||||||
ComparePaths = new List<FileInfo>(compareDirInfo.GetFiles("*.*", SearchOption.AllDirectories));
|
|
||||||
|
|
||||||
RaiseProgressChanged(0, fileCount, "Generating diffs...");
|
|
||||||
|
|
||||||
/* Comparing Target files -> Compare files
|
|
||||||
* - Exists = Diff (.bfd file)
|
|
||||||
* - Doesn't Exist = New (.new file)
|
|
||||||
*
|
|
||||||
* Once everything has been compared from one side, any remaining paths in our ComparePaths
|
|
||||||
* are things that don't exist in our target and can be deleted (.del file)
|
|
||||||
*/
|
|
||||||
|
|
||||||
for (int x = 0; x < TargetPaths.Count; x++)
|
|
||||||
{
|
|
||||||
FileInfo file = TargetPaths[x];
|
|
||||||
|
|
||||||
string assumedComparePath = file.DirectoryName.Replace(TargetBase, CompareBase);
|
|
||||||
|
|
||||||
if (!Compare(file.FullName, VFS.Combine(assumedComparePath, file.Name)))
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
//remove any existing files from our ComparePaths
|
|
||||||
FileInfo assumedFile = new FileInfo(VFS.Combine(assumedComparePath, file.Name));
|
|
||||||
if (assumedFile.Exists && ComparePaths.Exists(x => x.FullName == assumedFile.FullName))
|
|
||||||
{
|
|
||||||
ComparePaths.Remove(ComparePaths.Where(x => x.FullName == assumedFile.FullName).FirstOrDefault());
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
AdditionalInfo[0].ItemValue = diffCount.ToString();
|
|
||||||
AdditionalInfo[1].ItemValue = newCount.ToString();
|
|
||||||
AdditionalInfo[3].ItemValue = matchCount.ToString();
|
|
||||||
|
|
||||||
fileIt++;
|
|
||||||
RaiseProgressChanged(fileIt, fileCount, file.Name, AdditionalInfo.ToArray());
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
if (ComparePaths.Count == 0)
|
|
||||||
{
|
|
||||||
//if there are no files to delete, just return true
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
//progress reset for files that need to be deleted
|
|
||||||
RaiseProgressChanged(0, ComparePaths.Count, "Processing .del files...");
|
|
||||||
fileIt = 0;
|
|
||||||
fileCount = ComparePaths.Count;
|
|
||||||
|
|
||||||
//the paths remaining in ComparePaths don't exist in our target and need to be removed during patching.
|
|
||||||
foreach (FileInfo file in ComparePaths)
|
|
||||||
{
|
|
||||||
//add del files replace root dir with patch base
|
|
||||||
string patchFilePath = file.FullName.Replace(CompareBase, PatchBase);
|
|
||||||
VFS.WriteFile($"{patchFilePath}.del", new byte[0]);
|
|
||||||
|
|
||||||
delCount++;
|
|
||||||
AdditionalInfo[2].ItemValue = delCount.ToString();
|
|
||||||
|
|
||||||
fileIt++;
|
|
||||||
RaiseProgressChanged(fileIt, fileCount, "", AdditionalInfo.ToArray());
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
public FileCompare(string TargetBase, string CompareBase, string PatchBase)
|
|
||||||
{
|
|
||||||
this.TargetBase = TargetBase;
|
|
||||||
this.CompareBase = CompareBase;
|
|
||||||
this.PatchBase = PatchBase;
|
|
||||||
|
|
||||||
fileCount = VFS.GetFilesCount(TargetBase);
|
|
||||||
fileIt = 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,147 +0,0 @@
|
|||||||
using System;
|
|
||||||
using System.IO;
|
|
||||||
using Aki.Common.Utils;
|
|
||||||
using Aki.ByteBanger;
|
|
||||||
using System.Collections.Generic;
|
|
||||||
using System.Linq;
|
|
||||||
|
|
||||||
namespace PatcherUtils
|
|
||||||
{
|
|
||||||
public class FilePatcher
|
|
||||||
{
|
|
||||||
public string TargetBase;
|
|
||||||
public string PatchBase;
|
|
||||||
private int fileCount;
|
|
||||||
private int fileIt;
|
|
||||||
|
|
||||||
private int diffCount;
|
|
||||||
private int newCount;
|
|
||||||
private int delCount;
|
|
||||||
|
|
||||||
private List<LineItem> AdditionalInfo;
|
|
||||||
|
|
||||||
|
|
||||||
public event ProgressChangedHandler ProgressChanged;
|
|
||||||
|
|
||||||
protected virtual void RaiseProgressChanged(int progress, int total, string Message = "", params LineItem[] AdditionalLineItems)
|
|
||||||
{
|
|
||||||
int percent = (int)Math.Floor((double)progress / total * 100);
|
|
||||||
|
|
||||||
ProgressChanged?.Invoke(this, progress, total, percent, Message, AdditionalLineItems);
|
|
||||||
}
|
|
||||||
|
|
||||||
public bool Patch(string targetfile, string patchfile)
|
|
||||||
{
|
|
||||||
byte[] target = VFS.ReadFile(targetfile);
|
|
||||||
byte[] patch = VFS.ReadFile(patchfile);
|
|
||||||
|
|
||||||
PatchResult result = PatchUtil.Patch(target, PatchInfo.FromBytes(patch));
|
|
||||||
|
|
||||||
switch (result.Result)
|
|
||||||
{
|
|
||||||
case PatchResultType.Success:
|
|
||||||
VFS.WriteFile(targetfile, result.PatchedData);
|
|
||||||
return true;
|
|
||||||
|
|
||||||
case PatchResultType.AlreadyPatched:
|
|
||||||
case PatchResultType.InputChecksumMismatch:
|
|
||||||
case PatchResultType.InputLengthMismatch:
|
|
||||||
return true;
|
|
||||||
|
|
||||||
case PatchResultType.OutputChecksumMismatch:
|
|
||||||
default:
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private bool PatchAll(string targetpath, string patchpath)
|
|
||||||
{
|
|
||||||
DirectoryInfo di = new DirectoryInfo(patchpath);
|
|
||||||
|
|
||||||
foreach (FileInfo file in di.GetFiles())
|
|
||||||
{
|
|
||||||
FileInfo target = null;
|
|
||||||
|
|
||||||
switch (file.Extension)
|
|
||||||
{
|
|
||||||
// patch
|
|
||||||
case ".bpf":
|
|
||||||
{
|
|
||||||
target = new FileInfo(VFS.Combine(targetpath, file.Name.Replace(".bpf", "")));
|
|
||||||
|
|
||||||
if (!Patch(target.FullName, file.FullName))
|
|
||||||
{
|
|
||||||
// patch failed
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
diffCount--;
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
|
|
||||||
// add new files
|
|
||||||
case ".new":
|
|
||||||
{
|
|
||||||
target = new FileInfo(VFS.Combine(targetpath, file.Name.Replace(".new", "")));
|
|
||||||
VFS.WriteFile(target.FullName, Zlib.Decompress(VFS.ReadFile(file.FullName)));
|
|
||||||
newCount--;
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
|
|
||||||
// delete old files
|
|
||||||
case ".del":
|
|
||||||
{
|
|
||||||
target = new FileInfo(VFS.Combine(targetpath, file.Name.Replace(".del", "")));
|
|
||||||
target.IsReadOnly = false;
|
|
||||||
target.Delete();
|
|
||||||
delCount--;
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
AdditionalInfo[0].ItemValue = diffCount.ToString();
|
|
||||||
AdditionalInfo[1].ItemValue = newCount.ToString();
|
|
||||||
AdditionalInfo[2].ItemValue = delCount.ToString();
|
|
||||||
|
|
||||||
++fileIt;
|
|
||||||
RaiseProgressChanged(fileIt, fileCount, target.Name, AdditionalInfo.ToArray());
|
|
||||||
}
|
|
||||||
|
|
||||||
foreach (DirectoryInfo directory in di.GetDirectories())
|
|
||||||
{
|
|
||||||
PatchAll(VFS.Combine(targetpath, directory.Name), directory.FullName);
|
|
||||||
}
|
|
||||||
|
|
||||||
di.Refresh();
|
|
||||||
|
|
||||||
if (di.GetFiles().Length == 0 && di.GetDirectories().Length == 0)
|
|
||||||
{
|
|
||||||
// remove empty folders
|
|
||||||
di.Delete();
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
public bool Run()
|
|
||||||
{
|
|
||||||
fileCount = VFS.GetFilesCount(PatchBase);
|
|
||||||
|
|
||||||
FileInfo[] files = new DirectoryInfo(PatchBase).GetFiles("*.*", SearchOption.AllDirectories);
|
|
||||||
|
|
||||||
diffCount = files.Where(x => x.Extension == ".bpf").Count();
|
|
||||||
newCount = files.Where(x => x.Extension == ".new").Count();
|
|
||||||
delCount = files.Where(x => x.Extension == ".del").Count();
|
|
||||||
|
|
||||||
AdditionalInfo = new List<LineItem>()
|
|
||||||
{
|
|
||||||
new LineItem("Patches Remaining", diffCount.ToString()),
|
|
||||||
new LineItem("New Files to Inflate", newCount.ToString()),
|
|
||||||
new LineItem("Files to Delete", delCount.ToString())
|
|
||||||
};
|
|
||||||
|
|
||||||
fileIt = 0;
|
|
||||||
return PatchAll(TargetBase, PatchBase);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
@ -12,12 +12,13 @@ namespace PatcherUtils
|
|||||||
/// </summary>
|
/// </summary>
|
||||||
public static string TempDir = "PATCHER_TEMP".FromCwd();
|
public static string TempDir = "PATCHER_TEMP".FromCwd();
|
||||||
|
|
||||||
private static string SevenZExe = "7za.exe";
|
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// The folder that the patches will be stored in
|
/// The folder that the patches will be stored in
|
||||||
/// </summary>
|
/// </summary>
|
||||||
public static string PatchFolder = "Aki_Data\\Patcher";
|
public static string PatchFolder = "Aki_Patches";
|
||||||
|
|
||||||
|
private static string SevenZExe = "7za.exe";
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// The path to the 7za.exe file in the <see cref="TempDir"/>
|
/// The path to the 7za.exe file in the <see cref="TempDir"/>
|
||||||
@ -30,6 +31,13 @@ namespace PatcherUtils
|
|||||||
/// </summary>
|
/// </summary>
|
||||||
public static string PatcherClientPath = $"{TempDir}\\{PatcherClient}";
|
public static string PatcherClientPath = $"{TempDir}\\{PatcherClient}";
|
||||||
|
|
||||||
|
private static string XDelta3EXE = "xdelta3.exe";
|
||||||
|
|
||||||
|
/// <summary>
|
||||||
|
/// The path to the xdelta3.exe flie in the <see cref="TempDir"/>
|
||||||
|
/// </summary>
|
||||||
|
public static string XDelta3Path = $"{TempDir}\\{XDelta3EXE}";
|
||||||
|
|
||||||
/// <summary>
|
/// <summary>
|
||||||
/// Streams embedded resources out of the assembly
|
/// Streams embedded resources out of the assembly
|
||||||
/// </summary>
|
/// </summary>
|
||||||
@ -73,6 +81,11 @@ namespace PatcherUtils
|
|||||||
StreamResourceOut(resource, PatcherClientPath);
|
StreamResourceOut(resource, PatcherClientPath);
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
case string a when a.EndsWith(XDelta3EXE):
|
||||||
|
{
|
||||||
|
StreamResourceOut(resource, XDelta3Path);
|
||||||
|
break;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
278
Patcher/PatcherUtils/PatchHelper.cs
Normal file
278
Patcher/PatcherUtils/PatchHelper.cs
Normal file
@ -0,0 +1,278 @@
|
|||||||
|
using Aki.Common.Utils;
|
||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Diagnostics;
|
||||||
|
using System.IO;
|
||||||
|
using System.Linq;
|
||||||
|
using System.Text;
|
||||||
|
|
||||||
|
namespace PatcherUtils
|
||||||
|
{
|
||||||
|
public class PatchHelper
|
||||||
|
{
|
||||||
|
private string SourceFolder = "";
|
||||||
|
private string TargetFolder = "";
|
||||||
|
private string DeltaFolder = "";
|
||||||
|
|
||||||
|
private int fileCountTotal;
|
||||||
|
private int filesProcessed;
|
||||||
|
|
||||||
|
private int deltaCount;
|
||||||
|
private int newCount;
|
||||||
|
private int delCount;
|
||||||
|
|
||||||
|
private List<LineItem> AdditionalInfo = new List<LineItem>();
|
||||||
|
|
||||||
|
public event ProgressChangedHandler ProgressChanged;
|
||||||
|
|
||||||
|
protected virtual void RaiseProgressChanged(int progress, int total, string Message = "", params LineItem[] AdditionalLineItems)
|
||||||
|
{
|
||||||
|
int percent = (int)Math.Floor((double)progress / total * 100);
|
||||||
|
|
||||||
|
ProgressChanged?.Invoke(this, progress, total, percent, Message, AdditionalLineItems);
|
||||||
|
}
|
||||||
|
|
||||||
|
public PatchHelper(string SourceFolder, string TargetFolder, string DeltaFolder)
|
||||||
|
{
|
||||||
|
this.SourceFolder = SourceFolder;
|
||||||
|
this.TargetFolder = TargetFolder;
|
||||||
|
this.DeltaFolder = DeltaFolder;
|
||||||
|
}
|
||||||
|
|
||||||
|
private string GetDeltaPath(string sourceFile, string sourceFolder, string extension)
|
||||||
|
{
|
||||||
|
return Path.Join(DeltaFolder, $"{sourceFile.Replace(sourceFolder, "")}.{extension}");
|
||||||
|
}
|
||||||
|
|
||||||
|
private void ApplyDelta(string SourceFilePath, string DeltaFilePath)
|
||||||
|
{
|
||||||
|
string decodedPath = SourceFilePath + ".decoded";
|
||||||
|
|
||||||
|
Process.Start(new ProcessStartInfo
|
||||||
|
{
|
||||||
|
FileName = LazyOperations.XDelta3Path,
|
||||||
|
Arguments = $"-d -f -s \"{SourceFilePath}\" \"{DeltaFilePath}\" \"{decodedPath}\"",
|
||||||
|
CreateNoWindow = true
|
||||||
|
})
|
||||||
|
.WaitForExit();
|
||||||
|
|
||||||
|
if(File.Exists(decodedPath))
|
||||||
|
{
|
||||||
|
File.Delete(SourceFilePath);
|
||||||
|
File.Move(decodedPath, SourceFilePath);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void CreateDelta(string SourceFilePath, string TargetFilePath)
|
||||||
|
{
|
||||||
|
FileInfo sourceFileInfo = new FileInfo(SourceFilePath);
|
||||||
|
|
||||||
|
string deltaPath = GetDeltaPath(SourceFilePath, SourceFolder, "delta");
|
||||||
|
|
||||||
|
Directory.CreateDirectory(deltaPath.Replace(sourceFileInfo.Name+".delta", ""));
|
||||||
|
|
||||||
|
//TODO - don't hardcode FileName
|
||||||
|
|
||||||
|
Process.Start(new ProcessStartInfo
|
||||||
|
{
|
||||||
|
FileName = LazyOperations.XDelta3Path,
|
||||||
|
Arguments = $"-0 -e -f -s \"{SourceFilePath}\" \"{TargetFilePath}\" \"{deltaPath}\"",
|
||||||
|
CreateNoWindow = true
|
||||||
|
})
|
||||||
|
.WaitForExit();
|
||||||
|
}
|
||||||
|
|
||||||
|
private void CreateDelFile(string SourceFile)
|
||||||
|
{
|
||||||
|
FileInfo sourceFileInfo = new FileInfo(SourceFile);
|
||||||
|
|
||||||
|
string deltaPath = GetDeltaPath(SourceFile, SourceFolder, "del");
|
||||||
|
|
||||||
|
Directory.CreateDirectory(deltaPath.Replace(sourceFileInfo.Name+".del", ""));
|
||||||
|
|
||||||
|
File.Create(deltaPath);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void CreateNewFile(string TargetFile)
|
||||||
|
{
|
||||||
|
FileInfo targetSourceInfo = new FileInfo(TargetFile);
|
||||||
|
|
||||||
|
string deltaPath = GetDeltaPath(TargetFile, TargetFolder, "new");
|
||||||
|
|
||||||
|
Directory.CreateDirectory(deltaPath.Replace(targetSourceInfo.Name+".new", ""));
|
||||||
|
|
||||||
|
targetSourceInfo.CopyTo(deltaPath, true);
|
||||||
|
}
|
||||||
|
|
||||||
|
public bool GeneratePatches()
|
||||||
|
{
|
||||||
|
//get all directory information needed
|
||||||
|
DirectoryInfo sourceDir = new DirectoryInfo(SourceFolder);
|
||||||
|
DirectoryInfo targetDir = new DirectoryInfo(TargetFolder);
|
||||||
|
DirectoryInfo deltaDir = Directory.CreateDirectory(DeltaFolder);
|
||||||
|
|
||||||
|
//make sure all directories exist
|
||||||
|
if (!sourceDir.Exists || !targetDir.Exists || !deltaDir.Exists)
|
||||||
|
{
|
||||||
|
//One of the directories doesn't exist
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
List<FileInfo> SourceFiles = sourceDir.GetFiles("*", SearchOption.AllDirectories).ToList();
|
||||||
|
|
||||||
|
fileCountTotal = SourceFiles.Count;
|
||||||
|
|
||||||
|
AdditionalInfo.Clear();
|
||||||
|
AdditionalInfo.Add(new LineItem("Delta Patch", "0"));
|
||||||
|
AdditionalInfo.Add(new LineItem("New Patch", "0"));
|
||||||
|
AdditionalInfo.Add(new LineItem("Del Patch", "0"));
|
||||||
|
|
||||||
|
filesProcessed = 0;
|
||||||
|
|
||||||
|
RaiseProgressChanged(0, fileCountTotal, "Generating deltas...");
|
||||||
|
|
||||||
|
foreach (FileInfo targetFile in targetDir.GetFiles("*", SearchOption.AllDirectories))
|
||||||
|
{
|
||||||
|
//find a matching source file based on the relative path of the file
|
||||||
|
FileInfo sourceFile = SourceFiles.Find(f => f.FullName.Replace(sourceDir.FullName, "") == targetFile.FullName.Replace(targetDir.FullName, ""));
|
||||||
|
|
||||||
|
//if the target file doesn't exist in the source files, the target file needs to be added.
|
||||||
|
if (sourceFile == null)
|
||||||
|
{
|
||||||
|
CreateNewFile(targetFile.FullName);
|
||||||
|
|
||||||
|
newCount++;
|
||||||
|
filesProcessed++;
|
||||||
|
|
||||||
|
RaiseProgressChanged(filesProcessed, fileCountTotal, targetFile.Name, AdditionalInfo.ToArray());
|
||||||
|
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
//if a matching source file was found, get the delta for it.
|
||||||
|
CreateDelta(sourceFile.FullName, targetFile.FullName);
|
||||||
|
|
||||||
|
SourceFiles.Remove(sourceFile);
|
||||||
|
|
||||||
|
deltaCount++;
|
||||||
|
filesProcessed++;
|
||||||
|
|
||||||
|
AdditionalInfo[0].ItemValue = deltaCount.ToString();
|
||||||
|
AdditionalInfo[1].ItemValue = newCount.ToString();
|
||||||
|
|
||||||
|
RaiseProgressChanged(filesProcessed, fileCountTotal, targetFile.Name, AdditionalInfo.ToArray());
|
||||||
|
}
|
||||||
|
|
||||||
|
//Any remaining source files do not exist in the target folder and can be removed.
|
||||||
|
//reset progress info
|
||||||
|
RaiseProgressChanged(0, SourceFiles.Count, "Processing .del files...");
|
||||||
|
filesProcessed = 0;
|
||||||
|
fileCountTotal = SourceFiles.Count;
|
||||||
|
|
||||||
|
foreach (FileInfo delFile in SourceFiles)
|
||||||
|
{
|
||||||
|
CreateDelFile(delFile.FullName);
|
||||||
|
|
||||||
|
delCount++;
|
||||||
|
|
||||||
|
AdditionalInfo[2].ItemValue = delCount.ToString();
|
||||||
|
|
||||||
|
filesProcessed++;
|
||||||
|
RaiseProgressChanged(filesProcessed, fileCountTotal, "", AdditionalInfo.ToArray());
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
public string ApplyPatches()
|
||||||
|
{
|
||||||
|
//get needed directory information
|
||||||
|
DirectoryInfo sourceDir = new DirectoryInfo(SourceFolder);
|
||||||
|
DirectoryInfo deltaDir = new DirectoryInfo(DeltaFolder);
|
||||||
|
|
||||||
|
//check directories exist
|
||||||
|
if (!sourceDir.Exists || !deltaDir.Exists)
|
||||||
|
{
|
||||||
|
return "One of the supplied directories doesn't exist";
|
||||||
|
}
|
||||||
|
|
||||||
|
List<FileInfo> SourceFiles = sourceDir.GetFiles("*", SearchOption.AllDirectories).ToList();
|
||||||
|
|
||||||
|
List<FileInfo> deltaFiles = deltaDir.GetFiles("*", SearchOption.AllDirectories).ToList();
|
||||||
|
|
||||||
|
deltaCount = deltaFiles.Where(x => x.Extension == ".delta").Count();
|
||||||
|
newCount = deltaFiles.Where(x => x.Extension == ".new").Count();
|
||||||
|
delCount = deltaFiles.Where(x => x.Extension == ".del").Count();
|
||||||
|
|
||||||
|
|
||||||
|
AdditionalInfo = new List<LineItem>()
|
||||||
|
{
|
||||||
|
new LineItem("Patches Remaining", deltaCount.ToString()),
|
||||||
|
new LineItem("New Files to Add", newCount.ToString()),
|
||||||
|
new LineItem("Files to Delete", delCount.ToString())
|
||||||
|
};
|
||||||
|
|
||||||
|
filesProcessed = 0;
|
||||||
|
|
||||||
|
fileCountTotal = deltaFiles.Count;
|
||||||
|
|
||||||
|
foreach (FileInfo deltaFile in deltaDir.GetFiles("*", SearchOption.AllDirectories))
|
||||||
|
{
|
||||||
|
switch(deltaFile.Extension)
|
||||||
|
{
|
||||||
|
case ".delta":
|
||||||
|
{
|
||||||
|
//apply delta
|
||||||
|
FileInfo sourceFile = SourceFiles.Find(f => f.FullName.Replace(sourceDir.FullName, "") == deltaFile.FullName.Replace(deltaDir.FullName, "").Replace(".delta", ""));
|
||||||
|
|
||||||
|
if(sourceFile == null)
|
||||||
|
{
|
||||||
|
return $"Failed to find matching source file for '{deltaFile.FullName}'";
|
||||||
|
}
|
||||||
|
|
||||||
|
ApplyDelta(sourceFile.FullName, deltaFile.FullName);
|
||||||
|
|
||||||
|
deltaCount--;
|
||||||
|
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case ".new":
|
||||||
|
{
|
||||||
|
//copy new file
|
||||||
|
string destination = Path.Join(sourceDir.FullName, deltaFile.FullName.Replace(deltaDir.FullName, "").Replace(".new", ""));
|
||||||
|
|
||||||
|
File.Copy(deltaFile.FullName, destination);
|
||||||
|
|
||||||
|
newCount--;
|
||||||
|
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case ".del":
|
||||||
|
{
|
||||||
|
//remove unneeded file
|
||||||
|
string delFilePath = Path.Join(sourceDir.FullName, deltaFile.FullName.Replace(deltaDir.FullName, "").Replace(".del", ""));
|
||||||
|
|
||||||
|
File.Delete(delFilePath);
|
||||||
|
|
||||||
|
delCount--;
|
||||||
|
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
AdditionalInfo[0].ItemValue = deltaCount.ToString();
|
||||||
|
AdditionalInfo[1].ItemValue = newCount.ToString();
|
||||||
|
AdditionalInfo[2].ItemValue = delCount.ToString();
|
||||||
|
|
||||||
|
++filesProcessed;
|
||||||
|
RaiseProgressChanged(filesProcessed, fileCountTotal, deltaFile.Name, AdditionalInfo.ToArray());
|
||||||
|
}
|
||||||
|
|
||||||
|
LazyOperations.CleanupTempDir();
|
||||||
|
|
||||||
|
Directory.Delete(LazyOperations.PatchFolder, true);
|
||||||
|
|
||||||
|
return $"Patching Complete. You can delete the patcher.exe file.";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
@ -7,23 +7,19 @@
|
|||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<None Remove="Resources\7za.exe" />
|
<None Remove="Resources\7za.exe" />
|
||||||
<None Remove="Resources\PatchClient.exe" />
|
<None Remove="Resources\PatchClient.exe" />
|
||||||
|
<None Remove="Resources\xdelta3.exe" />
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
|
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<EmbeddedResource Include="Resources\7za.exe" />
|
<EmbeddedResource Include="Resources\7za.exe" />
|
||||||
<EmbeddedResource Include="Resources\PatchClient.exe" />
|
<EmbeddedResource Include="Resources\PatchClient.exe" />
|
||||||
|
<EmbeddedResource Include="Resources\xdelta3.exe" />
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
|
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<Reference Include="Aki.ByteBanger">
|
|
||||||
<HintPath>..\PatchGenerator\References\Aki.ByteBanger.dll</HintPath>
|
|
||||||
</Reference>
|
|
||||||
<Reference Include="Aki.Common">
|
<Reference Include="Aki.Common">
|
||||||
<HintPath>..\PatchGenerator\References\Aki.Common.dll</HintPath>
|
<HintPath>..\PatchGenerator\References\Aki.Common.dll</HintPath>
|
||||||
</Reference>
|
</Reference>
|
||||||
<Reference Include="ComponentAce.Compression.Libs.zlib">
|
|
||||||
<HintPath>..\PatchGenerator\References\ComponentAce.Compression.Libs.zlib.dll</HintPath>
|
|
||||||
</Reference>
|
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
|
|
||||||
</Project>
|
</Project>
|
||||||
|
Binary file not shown.
BIN
Patcher/PatcherUtils/Resources/xdelta3.exe
Normal file
BIN
Patcher/PatcherUtils/Resources/xdelta3.exe
Normal file
Binary file not shown.
Loading…
x
Reference in New Issue
Block a user