Browse Source

refactor to use regions instead of comments

refactor
artman41 6 years ago
parent
commit
65079683d9
1 changed files with 105 additions and 119 deletions
  1. +105
    -119
      IPA/Program.cs

+ 105
- 119
IPA/Program.cs View File

@ -10,11 +10,8 @@ using System.Text;
using System.Text.RegularExpressions; using System.Text.RegularExpressions;
using System.Windows.Forms; using System.Windows.Forms;
namespace IPA
{
public class Program
{
namespace IPA {
public class Program {
public enum Architecture { public enum Architecture {
x86, x86,
x64, x64,
@ -25,57 +22,53 @@ namespace IPA
static void Main(string[] args) { static void Main(string[] args) {
PatchContext context; PatchContext context;
if(args.Length < 1 || !args[0].EndsWith(".exe"))
{
if (args.Length < 1 || !args[0].EndsWith(".exe")) {
//Fail("Drag an (executable) file on the exe!"); //Fail("Drag an (executable) file on the exe!");
context = PatchContext.Create(new [] {new DirectoryInfo(Directory.GetCurrentDirectory()).GetFiles().First(o => o.FullName.EndsWith(".exe"))
.FullName});
context = PatchContext.Create(new[] {
new DirectoryInfo(Directory.GetCurrentDirectory()).GetFiles()
.First(o => o.FullName.EndsWith(".exe"))
.FullName
});
} }
else { else {
context = PatchContext.Create(args); context = PatchContext.Create(args);
} }
try
{
try {
bool isRevert = args.Contains("--revert") || Keyboard.IsKeyDown(Keys.LMenu); bool isRevert = args.Contains("--revert") || Keyboard.IsKeyDown(Keys.LMenu);
// Sanitizing // Sanitizing
Validate(context); Validate(context);
if (isRevert)
{
if (isRevert) {
Revert(context); Revert(context);
} }
else
{
else {
Install(context); Install(context);
StartIfNeedBe(context); StartIfNeedBe(context);
} }
} catch(Exception e)
{
}
catch (Exception e) {
Fail(e.Message); Fail(e.Message);
} }
} }
private static void Validate(PatchContext c)
{
if (!File.Exists(c.LauncherPathSrc)) Fail("Couldn't find DLLs! Make sure you extracted all contents of the release archive.");
if (!Directory.Exists(c.DataPathDst) || !File.Exists(c.EngineFile))
{
private static void Validate(PatchContext c) {
if (!File.Exists(c.LauncherPathSrc))
Fail("Couldn't find DLLs! Make sure you extracted all contents of the release archive.");
if (!Directory.Exists(c.DataPathDst) || !File.Exists(c.EngineFile)) {
Fail("Game does not seem to be a Unity project. Could not find the libraries to patch."); Fail("Game does not seem to be a Unity project. Could not find the libraries to patch.");
Console.WriteLine($"DataPath: {c.DataPathDst}"); Console.WriteLine($"DataPath: {c.DataPathDst}");
Console.WriteLine($"EngineFile: {c.EngineFile}"); Console.WriteLine($"EngineFile: {c.EngineFile}");
} }
} }
private static void Install(PatchContext context)
{
try
{
private static void Install(PatchContext context) {
try {
var backup = new BackupUnit(context); var backup = new BackupUnit(context);
//new version check
#region Patch Version Check
var patchedModule = PatchedModule.Load(context.EngineFile); var patchedModule = PatchedModule.Load(context.EngineFile);
var isCurrentNewer = Version.CompareTo(patchedModule.Data.Version) > 0; var isCurrentNewer = Version.CompareTo(patchedModule.Data.Version) > 0;
if (isCurrentNewer) { if (isCurrentNewer) {
@ -87,7 +80,8 @@ namespace IPA
Console.ResetColor(); Console.ResetColor();
// Copying
#region File Copying
Console.ForegroundColor = ConsoleColor.Magenta; Console.ForegroundColor = ConsoleColor.Magenta;
Console.WriteLine("Updating files... "); Console.WriteLine("Updating files... ");
var nativePluginFolder = Path.Combine(context.DataPathDst, "Plugins"); var nativePluginFolder = Path.Combine(context.DataPathDst, "Plugins");
@ -105,6 +99,8 @@ namespace IPA
architecture)); architecture));
Console.WriteLine("Successfully updated files!"); Console.WriteLine("Successfully updated files!");
#endregion
} }
else { else {
Console.ForegroundColor = ConsoleColor.Red; Console.ForegroundColor = ConsoleColor.Red;
@ -112,16 +108,22 @@ namespace IPA
Console.ResetColor(); Console.ResetColor();
} }
if (!Directory.Exists(context.PluginsFolder))
{
#endregion
#region Create Plugin Folder
if (!Directory.Exists(context.PluginsFolder)) {
Console.ForegroundColor = ConsoleColor.DarkYellow;
Console.WriteLine("Creating plugins folder... "); Console.WriteLine("Creating plugins folder... ");
Directory.CreateDirectory(context.PluginsFolder); Directory.CreateDirectory(context.PluginsFolder);
Console.ResetColor();
} }
Console.ResetColor();
// Patching
if (!patchedModule.Data.IsPatched || isCurrentNewer)
{
#endregion
#region Patching
if (!patchedModule.Data.IsPatched || isCurrentNewer) {
Console.ForegroundColor = ConsoleColor.Yellow; Console.ForegroundColor = ConsoleColor.Yellow;
Console.WriteLine($"Patching UnityEngine.dll with Version {Application.ProductVersion}... "); Console.WriteLine($"Patching UnityEngine.dll with Version {Application.ProductVersion}... ");
backup.Add(context.EngineFile); backup.Add(context.EngineFile);
@ -130,9 +132,11 @@ namespace IPA
Console.ResetColor(); Console.ResetColor();
} }
// Virtualizing
if (File.Exists(context.AssemblyFile))
{
#endregion
#region Virtualizing
if (File.Exists(context.AssemblyFile)) {
var virtualizedModule = VirtualizedModule.Load(context.AssemblyFile); var virtualizedModule = VirtualizedModule.Load(context.AssemblyFile);
if (!virtualizedModule.IsVirtualized) { if (!virtualizedModule.IsVirtualized) {
Console.ForegroundColor = ConsoleColor.Blue; Console.ForegroundColor = ConsoleColor.Blue;
@ -144,7 +148,9 @@ namespace IPA
} }
} }
// Creating shortcut
#endregion
#region Creating shortcut
/*if(!File.Exists(context.ShortcutPath)) /*if(!File.Exists(context.ShortcutPath))
{ {
Console.Write("Creating shortcut to IPA ({0})... ", context.IPA); Console.Write("Creating shortcut to IPA ({0})... ", context.IPA);
@ -165,9 +171,9 @@ namespace IPA
Console.Error.WriteLine("Failed to create shortcut, but game was patched!"); Console.Error.WriteLine("Failed to create shortcut, but game was patched!");
} }
}*/ }*/
#endregion
} }
catch (Exception e)
{
catch (Exception e) {
Fail("Oops! This should not have happened.\n\n" + e); Fail("Oops! This should not have happened.\n\n" + e);
} }
@ -181,19 +187,17 @@ namespace IPA
private static void Revert(PatchContext context, string[] args = null) { private static void Revert(PatchContext context, string[] args = null) {
Console.ForegroundColor = ConsoleColor.Cyan; Console.ForegroundColor = ConsoleColor.Cyan;
bool isNewVersion = (args != null && args.Contains("newVersion")); bool isNewVersion = (args != null && args.Contains("newVersion"));
Console.Write("Restoring backup... "); Console.Write("Restoring backup... ");
if(BackupManager.Restore(context))
{
if (BackupManager.Restore(context)) {
Console.WriteLine("Done!"); Console.WriteLine("Done!");
} else
{
}
else {
Console.WriteLine("Already vanilla!"); Console.WriteLine("Already vanilla!");
} }
if (File.Exists(context.ShortcutPath))
{
if (File.Exists(context.ShortcutPath)) {
Console.WriteLine("Deleting shortcut..."); Console.WriteLine("Deleting shortcut...");
File.Delete(context.ShortcutPath); File.Delete(context.ShortcutPath);
} }
@ -201,86 +205,80 @@ namespace IPA
Console.WriteLine(""); Console.WriteLine("");
Console.WriteLine("--- Done reverting ---"); Console.WriteLine("--- Done reverting ---");
if (!Environment.CommandLine.Contains("--nowait") && !isNewVersion)
{
if (!Environment.CommandLine.Contains("--nowait") && !isNewVersion) {
Console.WriteLine("\n\n[Press any key to quit]"); Console.WriteLine("\n\n[Press any key to quit]");
Console.ReadKey(); Console.ReadKey();
} }
Console.ResetColor(); Console.ResetColor();
} }
private static void StartIfNeedBe(PatchContext context)
{
private static void StartIfNeedBe(PatchContext context) {
var argList = context.Args.ToList(); var argList = context.Args.ToList();
bool launch = argList.Remove("--launch"); bool launch = argList.Remove("--launch");
argList.RemoveAt(0); argList.RemoveAt(0);
if(launch)
{
if (launch) {
Process.Start(context.Executable, Args(argList.ToArray())); Process.Start(context.Executable, Args(argList.ToArray()));
} }
} }
public static IEnumerable<FileInfo> NativePluginInterceptor(FileInfo from, FileInfo to, DirectoryInfo nativePluginFolder, bool isFlat, Architecture preferredArchitecture)
{
if (to.FullName.StartsWith(nativePluginFolder.FullName))
{
public static IEnumerable<FileInfo> NativePluginInterceptor(FileInfo from, FileInfo to,
DirectoryInfo nativePluginFolder, bool isFlat, Architecture preferredArchitecture) {
if (to.FullName.StartsWith(nativePluginFolder.FullName)) {
var relevantBit = to.FullName.Substring(nativePluginFolder.FullName.Length + 1); var relevantBit = to.FullName.Substring(nativePluginFolder.FullName.Length + 1);
// Goes into the plugin folder! // Goes into the plugin folder!
bool isFileFlat = !relevantBit.StartsWith("x86"); bool isFileFlat = !relevantBit.StartsWith("x86");
if (isFlat && !isFileFlat)
{
if (isFlat && !isFileFlat) {
// Flatten structure // Flatten structure
bool is64Bit = relevantBit.StartsWith("x86_64"); bool is64Bit = relevantBit.StartsWith("x86_64");
if (!is64Bit && preferredArchitecture == Architecture.x86)
{
if (!is64Bit && preferredArchitecture == Architecture.x86) {
// 32 bit // 32 bit
yield return new FileInfo(Path.Combine(nativePluginFolder.FullName, relevantBit.Substring("x86".Length + 1)));
yield return new FileInfo(Path.Combine(nativePluginFolder.FullName,
relevantBit.Substring("x86".Length + 1)));
} }
else if(is64Bit && (preferredArchitecture == Architecture.x64 || preferredArchitecture == Architecture.Unknown))
{
else if (is64Bit && (preferredArchitecture == Architecture.x64 ||
preferredArchitecture == Architecture.Unknown)) {
// 64 bit // 64 bit
yield return new FileInfo(Path.Combine(nativePluginFolder.FullName, relevantBit.Substring("x86_64".Length + 1)));
} else {
yield return new FileInfo(Path.Combine(nativePluginFolder.FullName,
relevantBit.Substring("x86_64".Length + 1)));
}
else {
// Throw away // Throw away
yield break; yield break;
} }
} }
else if (!isFlat && isFileFlat)
{
else if (!isFlat && isFileFlat) {
// Deepen structure // Deepen structure
yield return new FileInfo(Path.Combine(Path.Combine(nativePluginFolder.FullName, "x86"), relevantBit));
yield return new FileInfo(Path.Combine(Path.Combine(nativePluginFolder.FullName, "x86_64"), relevantBit));
yield return new FileInfo(Path.Combine(Path.Combine(nativePluginFolder.FullName, "x86"),
relevantBit));
yield return new FileInfo(Path.Combine(Path.Combine(nativePluginFolder.FullName, "x86_64"),
relevantBit));
} }
else
{
else {
yield return to; yield return to;
} }
} }
else
{
else {
yield return to; yield return to;
} }
} }
private static IEnumerable<FileInfo> PassThroughInterceptor(FileInfo from, FileInfo to)
{
private static IEnumerable<FileInfo> PassThroughInterceptor(FileInfo from, FileInfo to) {
yield return to; yield return to;
} }
public static void CopyAll(DirectoryInfo source, DirectoryInfo target, bool aggressive, BackupUnit backup, Func<FileInfo, FileInfo, IEnumerable<FileInfo>> interceptor = null)
{
if(interceptor == null)
{
public static void CopyAll(DirectoryInfo source, DirectoryInfo target, bool aggressive, BackupUnit backup,
Func<FileInfo, FileInfo, IEnumerable<FileInfo>> interceptor = null) {
if (interceptor == null) {
interceptor = PassThroughInterceptor; interceptor = PassThroughInterceptor;
} }
// Copy each file into the new directory. // Copy each file into the new directory.
foreach (FileInfo fi in source.GetFiles())
{
foreach(var targetFile in interceptor(fi, new FileInfo(Path.Combine(target.FullName, fi.Name)))) {
if (!targetFile.Exists || targetFile.LastWriteTimeUtc < fi.LastWriteTimeUtc || aggressive)
{
foreach (FileInfo fi in source.GetFiles()) {
foreach (var targetFile in interceptor(fi, new FileInfo(Path.Combine(target.FullName, fi.Name)))) {
if (!targetFile.Exists || targetFile.LastWriteTimeUtc < fi.LastWriteTimeUtc || aggressive) {
targetFile.Directory.Create(); targetFile.Directory.Create();
Console.WriteLine(@"Copying {0}", targetFile.FullName); Console.WriteLine(@"Copying {0}", targetFile.FullName);
@ -291,27 +289,24 @@ namespace IPA
} }
// Copy each subdirectory using recursion. // Copy each subdirectory using recursion.
foreach (DirectoryInfo diSourceSubDir in source.GetDirectories())
{
foreach (DirectoryInfo diSourceSubDir in source.GetDirectories()) {
DirectoryInfo nextTargetSubDir = new DirectoryInfo(Path.Combine(target.FullName, diSourceSubDir.Name)); DirectoryInfo nextTargetSubDir = new DirectoryInfo(Path.Combine(target.FullName, diSourceSubDir.Name));
CopyAll(diSourceSubDir, nextTargetSubDir, aggressive, backup, interceptor); CopyAll(diSourceSubDir, nextTargetSubDir, aggressive, backup, interceptor);
} }
} }
static void Fail(string message)
{
static void Fail(string message) {
Console.Error.Write("ERROR: " + message); Console.Error.Write("ERROR: " + message);
if (!Environment.CommandLine.Contains("--nowait"))
{
if (!Environment.CommandLine.Contains("--nowait")) {
Console.WriteLine("\n\n[Press any key to quit]"); Console.WriteLine("\n\n[Press any key to quit]");
Console.ReadKey(); Console.ReadKey();
} }
Environment.Exit(1); Environment.Exit(1);
} }
public static string Args(params string[] args)
{
public static string Args(params string[] args) {
return string.Join(" ", args.Select(EncodeParameterArgument).ToArray()); return string.Join(" ", args.Select(EncodeParameterArgument).ToArray());
} }
@ -321,8 +316,7 @@ namespace IPA
/// <param name="original">The value that should be received by the program</param> /// <param name="original">The value that should be received by the program</param>
/// <returns>The value which needs to be passed to the program for the original value /// <returns>The value which needs to be passed to the program for the original value
/// to come through</returns> /// to come through</returns>
public static string EncodeParameterArgument(string original)
{
public static string EncodeParameterArgument(string original) {
if (string.IsNullOrEmpty(original)) if (string.IsNullOrEmpty(original))
return original; return original;
string value = Regex.Replace(original, @"(\\*)" + "\"", @"$1\$0"); string value = Regex.Replace(original, @"(\\*)" + "\"", @"$1\$0");
@ -330,13 +324,10 @@ namespace IPA
return value; return value;
} }
public static Architecture DetectArchitecture(string assembly)
{
using (var reader = new BinaryReader(File.OpenRead(assembly)))
{
public static Architecture DetectArchitecture(string assembly) {
using (var reader = new BinaryReader(File.OpenRead(assembly))) {
var header = reader.ReadUInt16(); var header = reader.ReadUInt16();
if(header == 0x5a4d)
{
if (header == 0x5a4d) {
reader.BaseStream.Seek(60, SeekOrigin.Begin); // this location contains the offset for the PE header reader.BaseStream.Seek(60, SeekOrigin.Begin); // this location contains the offset for the PE header
var peOffset = reader.ReadUInt32(); var peOffset = reader.ReadUInt32();
@ -351,19 +342,17 @@ namespace IPA
return Architecture.x64; return Architecture.x64;
else else
return Architecture.Unknown; return Architecture.Unknown;
} else
{
}
else {
// Not a supported binary // Not a supported binary
return Architecture.Unknown; return Architecture.Unknown;
} }
} }
} }
public abstract class Keyboard
{
public abstract class Keyboard {
[Flags] [Flags]
private enum KeyStates
{
private enum KeyStates {
None = 0, None = 0,
Down = 1, Down = 1,
Toggled = 2 Toggled = 2
@ -372,11 +361,10 @@ namespace IPA
[DllImport("user32.dll", CharSet = CharSet.Auto, ExactSpelling = true)] [DllImport("user32.dll", CharSet = CharSet.Auto, ExactSpelling = true)]
private static extern short GetKeyState(int keyCode); private static extern short GetKeyState(int keyCode);
private static KeyStates GetKeyState(Keys key)
{
private static KeyStates GetKeyState(Keys key) {
KeyStates state = KeyStates.None; KeyStates state = KeyStates.None;
short retVal = GetKeyState((int)key);
short retVal = GetKeyState((int) key);
//If the high-order bit is 1, the key is down //If the high-order bit is 1, the key is down
//otherwise, it is up. //otherwise, it is up.
@ -390,15 +378,13 @@ namespace IPA
return state; return state;
} }
public static bool IsKeyDown(Keys key)
{
public static bool IsKeyDown(Keys key) {
return KeyStates.Down == (GetKeyState(key) & KeyStates.Down); return KeyStates.Down == (GetKeyState(key) & KeyStates.Down);
} }
public static bool IsKeyToggled(Keys key)
{
public static bool IsKeyToggled(Keys key) {
return KeyStates.Toggled == (GetKeyState(key) & KeyStates.Toggled); return KeyStates.Toggled == (GetKeyState(key) & KeyStates.Toggled);
} }
} }
} }
}
}

Loading…
Cancel
Save