|
|
- using System;
- using System.Collections.Generic;
- using System.Diagnostics;
- using System.Diagnostics.CodeAnalysis;
- using System.IO;
- using System.Linq;
- using System.Reflection;
- using System.Runtime.InteropServices;
- using System.Text.RegularExpressions;
- using System.Windows.Forms;
- using IPA.Patcher;
-
- namespace IPA
- {
- [SuppressMessage("ReSharper", "ClassNeverInstantiated.Global")]
- public class Program
- {
- [SuppressMessage("ReSharper", "InconsistentNaming")]
- public enum Architecture
- {
- x86,
- x64,
- Unknown
- }
-
- public const string FileVersion = "3.99.99.0";
-
- public static Version Version => Assembly.GetEntryAssembly().GetName().Version;
-
- public static readonly ArgumentFlag ArgHelp = new ArgumentFlag("--help", "-h") { DocString = "prints this message" };
- public static readonly ArgumentFlag ArgWaitFor = new ArgumentFlag("--waitfor", "-w") { DocString = "waits for the specified PID to exit", ValueString = "PID" };
- public static readonly ArgumentFlag ArgForce = new ArgumentFlag("--force", "-f") { DocString = "forces the operation to go through" };
- public static readonly ArgumentFlag ArgRevert = new ArgumentFlag("--revert", "-r") { DocString = "reverts the IPA installation" };
- public static readonly ArgumentFlag ArgNoRevert = new ArgumentFlag("--no-revert", "-R") { DocString = "prevents a normal installation from first reverting" };
- public static readonly ArgumentFlag ArgNoWait = new ArgumentFlag("--nowait", "-n") { DocString = "doesn't wait for user input after the operation" };
- public static readonly ArgumentFlag ArgStart = new ArgumentFlag("--start", "-s") { DocString = "uses the specified arguments to start the game after the patch/unpatch", ValueString = "ARGUMENTS" };
- public static readonly ArgumentFlag ArgLaunch = new ArgumentFlag("--launch", "-l") { DocString = "uses positional parameters as arguments to start the game after patch/unpatch" };
-
- [STAThread]
- public static void Main(string[] args)
- {
- Arguments.CmdLine.Flags(ArgHelp, ArgWaitFor, ArgForce, ArgRevert, ArgNoWait, ArgStart, ArgLaunch, ArgNoRevert).Process();
-
- if (ArgHelp)
- {
- Arguments.CmdLine.PrintHelp();
- return;
- }
-
- try
- {
- if (ArgWaitFor.HasValue)
- { // wait for process if necessary
- var pid = int.Parse(ArgWaitFor.Value);
-
- try
- { // wait for beat saber to exit (ensures we can modify the file)
- var parent = Process.GetProcessById(pid);
-
- Console.WriteLine($"Waiting for parent ({pid}) process to die...");
-
- parent.WaitForExit();
- }
- catch (Exception)
- {
- // ignored
- }
- }
-
- PatchContext context = null;
-
- Assembly AssemblyLibLoader(object source, ResolveEventArgs e)
- {
- // ReSharper disable AccessToModifiedClosure
- if (context == null) return null;
- var libsDir = context.LibsPathSrc;
- // ReSharper enable AccessToModifiedClosure
-
- var asmName = new AssemblyName(e.Name);
- var testFile = Path.Combine(libsDir, $"{asmName.Name}.{asmName.Version}.dll");
-
- if (File.Exists(testFile))
- return Assembly.LoadFile(testFile);
-
- Console.WriteLine($"Could not load library {asmName}");
-
- return null;
- }
- AppDomain.CurrentDomain.AssemblyResolve += AssemblyLibLoader;
-
- var argExeName = Arguments.CmdLine.PositionalArgs.FirstOrDefault(s => s.EndsWith(".exe"));
- if (argExeName == null)
- context = PatchContext.Create(new DirectoryInfo(Directory.GetCurrentDirectory()).GetFiles()
- .First(o => o.Extension == ".exe" && o.FullName != Assembly.GetEntryAssembly().Location)
- .FullName);
- else
- context = PatchContext.Create(argExeName);
-
- // Sanitizing
- Validate(context);
-
- if (ArgRevert || Keyboard.IsKeyDown(Keys.LMenu))
- Revert(context);
- else
- {
- Install(context);
- StartIfNeedBe(context);
- }
- }
- catch (Exception e)
- {
- Fail(e.Message);
- }
-
- WaitForEnd();
- }
-
- private static void WaitForEnd()
- {
- if (!ArgNoWait)
- {
- Console.ForegroundColor = ConsoleColor.DarkYellow;
- Console.WriteLine("[Press any key to continue]");
- Console.ResetColor();
- Console.ReadKey();
- }
- }
-
- private static void Validate(PatchContext c)
- {
- 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.");
- Console.WriteLine($"DataPath: {c.DataPathDst}");
- Console.WriteLine($"EngineFile: {c.EngineFile}");
- }
- }
-
- private static void Install(PatchContext context)
- {
- try
- {
- bool installFiles = true;
- // first, check currently installed version, if any
- if (File.Exists(Path.Combine(context.ProjectRoot, "winhttp.dll")))
- { // installed, so check version of installed assembly
- string injectorPath = Path.Combine(context.ManagedPath, "IPA.Injector.dll");
- if (File.Exists(injectorPath))
- {
- var verInfo = FileVersionInfo.GetVersionInfo(injectorPath);
- var fileVersion = new Version(verInfo.FileVersion);
-
- if (fileVersion > Version)
- installFiles = false;
- }
- }
-
- if (installFiles || ArgForce)
- {
- var backup = new BackupUnit(context);
-
- if (!ArgNoRevert)
- {
- Console.ForegroundColor = ConsoleColor.Cyan;
- Console.WriteLine("Restoring old version... ");
- if (BackupManager.HasBackup(context))
- BackupManager.Restore(context);
- }
-
- var nativePluginFolder = Path.Combine(context.DataPathDst, "Plugins");
- bool isFlat = Directory.Exists(nativePluginFolder) &&
- Directory.GetFiles(nativePluginFolder).Any(f => f.EndsWith(".dll"));
- bool force = !BackupManager.HasBackup(context) || ArgForce;
- var architecture = DetectArchitecture(context.Executable);
-
- Console.ForegroundColor = ConsoleColor.DarkCyan;
- Console.WriteLine("Installing files... ");
-
- CopyAll(new DirectoryInfo(context.DataPathSrc), new DirectoryInfo(context.DataPathDst), force,
- backup,
- (from, to) => NativePluginInterceptor(from, to, new DirectoryInfo(nativePluginFolder), isFlat,
- architecture));
- CopyAll(new DirectoryInfo(context.LibsPathSrc), new DirectoryInfo(context.LibsPathDst), force,
- backup,
- (from, to) => NativePluginInterceptor(from, to, new DirectoryInfo(nativePluginFolder), isFlat,
- architecture));
- CopyAll(new DirectoryInfo(context.IPARoot), new DirectoryInfo(context.ProjectRoot), force,
- backup,
- null, false);
- }
- else
- {
- Console.ForegroundColor = ConsoleColor.Yellow;
- Console.WriteLine("Not copying files because newer version already installed");
- }
-
- #region Create Plugin Folder
-
- if (!Directory.Exists(context.PluginsFolder))
- {
- Console.ForegroundColor = ConsoleColor.DarkYellow;
- Console.WriteLine("Creating plugins folder... ");
- Directory.CreateDirectory(context.PluginsFolder);
- Console.ResetColor();
- }
-
- #endregion
-
- }
- catch (Exception e)
- {
- Console.ForegroundColor = ConsoleColor.Red;
- Fail("Oops! This should not have happened.\n\n" + e);
- }
- Console.ResetColor();
- }
-
- private static void Revert(PatchContext context)
- {
- Console.ForegroundColor = ConsoleColor.Cyan;
-
- Console.Write("Restoring backup... ");
- if (BackupManager.Restore(context))
- {
- Console.WriteLine("Done!");
- }
- else
- {
- Console.WriteLine("Already vanilla or you removed your backups!");
- }
-
-
- if (File.Exists(context.ShortcutPath))
- {
- Console.WriteLine("Deleting shortcut...");
- File.Delete(context.ShortcutPath);
- }
-
- Console.WriteLine("");
- Console.WriteLine("--- Done reverting ---");
-
- Console.ResetColor();
- }
-
- private static void StartIfNeedBe(PatchContext context)
- {
- if (ArgStart.HasValue)
- {
- Process.Start(context.Executable, ArgStart.Value);
- }
- else
- {
- var argList = Arguments.CmdLine.PositionalArgs.ToList();
-
- argList.Remove(context.Executable);
-
- if (ArgLaunch)
- {
- 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))
- {
- var relevantBit = to.FullName.Substring(nativePluginFolder.FullName.Length + 1);
- // Goes into the plugin folder!
- bool isFileFlat = !relevantBit.StartsWith("x86");
- if (isFlat && !isFileFlat)
- {
- // Flatten structure
- bool is64Bit = relevantBit.StartsWith("x86_64");
- if (!is64Bit && preferredArchitecture == Architecture.x86)
- {
- // 32 bit
- yield return new FileInfo(Path.Combine(nativePluginFolder.FullName,
- relevantBit.Substring("x86".Length + 1)));
- }
- else if (is64Bit && (preferredArchitecture == Architecture.x64 ||
- preferredArchitecture == Architecture.Unknown))
- {
- // 64 bit
- yield return new FileInfo(Path.Combine(nativePluginFolder.FullName,
- relevantBit.Substring("x86_64".Length + 1)));
- }
- }
- else if (!isFlat && isFileFlat)
- {
- // 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));
- }
- else
- {
- yield return to;
- }
- }
- else
- {
- yield return to;
- }
- }
-
- public static void ClearLine()
- {
- if (IsConsole)
- {
- Console.SetCursorPosition(0, Console.CursorTop);
- int tpos = Console.CursorTop;
- Console.Write(new string(' ', Console.WindowWidth));
- Console.SetCursorPosition(0, tpos);
- }
- }
-
- private static IEnumerable<FileInfo> PassThroughInterceptor(FileInfo from, FileInfo to)
- {
- yield return to;
- }
-
- public static void CopyAll(DirectoryInfo source, DirectoryInfo target, bool aggressive, BackupUnit backup,
- Func<FileInfo, FileInfo, IEnumerable<FileInfo>> interceptor = null, bool recurse = true)
- {
- if (interceptor == null)
- {
- interceptor = PassThroughInterceptor;
- }
-
- // Copy each file into the new directory.
- foreach (var 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)
- continue;
-
- Debug.Assert(targetFile.Directory != null, "targetFile.Directory != null");
- targetFile.Directory?.Create();
-
- LineBack();
- ClearLine();
- Console.WriteLine(@"Copying {0}", targetFile.FullName);
- backup.Add(targetFile);
- fi.CopyTo(targetFile.FullName, true);
- }
- }
-
- // Copy each subdirectory using recursion.
- if (!recurse) return;
- foreach (var diSourceSubDir in source.GetDirectories())
- {
- var nextTargetSubDir = new DirectoryInfo(Path.Combine(target.FullName, diSourceSubDir.Name));
- CopyAll(diSourceSubDir, nextTargetSubDir, aggressive, backup, interceptor);
- }
- }
-
-
- private static void Fail(string message)
- {
- Console.Error.WriteLine("ERROR: " + message);
-
- WaitForEnd();
-
- Environment.Exit(1);
- }
-
- public static string Args(params string[] args)
- {
- return string.Join(" ", args.Select(EncodeParameterArgument).ToArray());
- }
-
- /// <summary>
- /// Encodes an argument for passing into a program
- /// </summary>
- /// <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_
- /// to come through</returns>
- public static string EncodeParameterArgument(string original)
- {
- if (string.IsNullOrEmpty(original))
- return original;
- string value = Regex.Replace(original, @"(\\*)" + "\"", @"$1\$0");
- value = Regex.Replace(value, @"^(.*\s.*?)(\\*)$", "\"$1$2$2\"");
- return value;
- }
-
- public static Architecture DetectArchitecture(string assembly)
- {
- using (var reader = new BinaryReader(File.OpenRead(assembly)))
- {
- var header = reader.ReadUInt16();
- if (header == 0x5a4d)
- {
- reader.BaseStream.Seek(60, SeekOrigin.Begin); // this location contains the offset for the PE header
- var peOffset = reader.ReadUInt32();
-
- reader.BaseStream.Seek(peOffset + 4, SeekOrigin.Begin);
- var machine = reader.ReadUInt16();
-
- if (machine == 0x8664) // IMAGE_FILE_MACHINE_AMD64
- return Architecture.x64;
- if (machine == 0x014c) // IMAGE_FILE_MACHINE_I386
- return Architecture.x86;
- if (machine == 0x0200) // IMAGE_FILE_MACHINE_IA64
- return Architecture.x64;
- return Architecture.Unknown;
- }
-
- // Not a supported binary
- return Architecture.Unknown;
- }
- }
-
- public static void ResetLine()
- {
- if (IsConsole)
- Console.CursorLeft = 0;
- else
- Console.Write("\r");
- }
-
- public static void LineBack()
- {
- if (IsConsole)
- Console.CursorTop--;
- else
- Console.Write("\x1b[1A");
- }
-
- [DllImport("kernel32.dll")]
- private static extern IntPtr GetConsoleWindow();
-
- private static bool? isConsole;
- public static bool IsConsole
- {
- get
- {
- if (isConsole == null)
- isConsole = GetConsoleWindow() != IntPtr.Zero;
- return isConsole.Value;
- }
- }
-
- public abstract class Keyboard
- {
- [Flags]
- private enum KeyStates
- {
- None = 0,
- Down = 1,
- Toggled = 2
- }
-
- [DllImport("user32.dll", CharSet = CharSet.Auto, ExactSpelling = true)]
- private static extern short GetKeyState(int keyCode);
-
- private static KeyStates KeyState(Keys key)
- {
- KeyStates state = KeyStates.None;
-
- short retVal = GetKeyState((int)key);
-
- //If the high-order bit is 1, the key is down
- //otherwise, it is up.
- if ((retVal & 0x8000) == 0x8000)
- state |= KeyStates.Down;
-
- //If the low-order bit is 1, the key is toggled.
- if ((retVal & 1) == 1)
- state |= KeyStates.Toggled;
-
- return state;
- }
-
- public static bool IsKeyDown(Keys key)
- {
- return KeyStates.Down == (KeyState(key) & KeyStates.Down);
- }
-
- public static bool IsKeyToggled(Keys key)
- {
- return KeyStates.Toggled == (KeyState(key) & KeyStates.Toggled);
- }
- }
- }
- }
|