Esempio n. 1
0
        public static void Main()
        {
            ChalkerInstance = new Chalker();
            ChalkerInstance.WriteMessage("SAMPLE: Casing");
            Sample_Casing.Run();

            Console.WriteLine("\n");
            ChalkerInstance.WriteMessage("SAMPLE: RightPadding");
            Sample_RightPadding.Run();

            Console.WriteLine("\n");
            ChalkerInstance.WriteMessage("SAMPLE: AppendSpace");
            Sample_AppendSpace.Run();

            Console.WriteLine("\n");
            ChalkerInstance.WriteMessage("SAMPLE: CustomPrefix");
            Sample_CustomPrefix.Run();
            Console.ReadKey();
        }
Esempio n. 2
0
        public static void Main(string[] args)
        {
            Parser.Default.ParseArguments <Options>(args).WithParsed((options) => Options.Instance = options).WithNotParsed((_) => Util.Exit());
            Console.WriteLine($"Universal Unity Hooks v{Version}");
            Options.Instance.GetTargetAssembly(out var targetAssemblyName, out var targetAssemblyNameClean);
            TargetAssembly      = targetAssemblyName;
            TargetAssemblyClean = targetAssemblyNameClean;
            PluginsFolder       = Options.Instance.GetInputDirectory();
            OutputFolder        = Options.Instance.GetOutputDirectory();
            Util.CreateFiles();
            Chalker.WriteMessage($"Target assembly file: {TargetAssembly}");
            Chalker.WriteMessage($"Target plugins & search directory: {PluginsFolder}");
            Console.WriteLine();
            var timer = new OperationTimer();

            AssemblyHelper = new AssemblyHelper(PluginsFolder);
            AssemblyHelper.FetchAndLoadAll();
            Console.WriteLine();

            var sum = Attributes.Sum(x => x.Value.Count);

            if (sum == 0)
            {
                Chalker.WriteError("No attributes have been loaded in. Press any key to exit the program.");
                Util.Exit();
            }
            Chalker.WriteSuccess($"Total of {sum} attribute(s) have been loaded in.");

            Console.WriteLine();
            Chalker.WriteWait("Loading target assembly..");
            try
            {
                if (File.Exists(TargetAssemblyClean))
                {
                    File.Delete(TargetAssembly);
                    File.Copy(TargetAssemblyClean, TargetAssembly);
                }
                else
                {
                    File.Copy(TargetAssembly, TargetAssemblyClean);
                }
            }
            catch (Exception ex)
            {
                Chalker.WriteError("Something went wrong while trying to get or copy the target assembly file.");
                Chalker.WriteError($"Message: {ex.Message}");
                Chalker.WriteError(ex.StackTrace);
                Util.Exit();
            }
            if (!Util.ReadAssembly(TargetAssembly, OutputFolder, out var assemblyDefinition))
            {
                Chalker.WriteError("Press any key to exit this program.");
                Util.Exit();
            }
            Chalker.WriteSuccess("Target assembly loaded in.");

            Console.WriteLine();
            var injectedCorrectly = Util.InjectAllHooks(Attributes, assemblyDefinition);

            Console.WriteLine();
            if (injectedCorrectly == 0)
            {
                Chalker.WriteError($"No methods have been injected correctly. Please read above for more information why some injections may have failed.");
            }
            else
            {
                Chalker.WriteSuccess($"Injected {(injectedCorrectly == sum ? $"all {sum}" : $"{injectedCorrectly}/{sum}")} methods. {(injectedCorrectly != sum ? "Please read above for more information why some injections may have failed." : "")}");
                Chalker.WriteWait("Writing changes to target assembly..");
                if (Cecil.WriteChanges(assemblyDefinition, TargetAssembly))
                {
                    Chalker.WriteSuccess($"Changes have been written to the file {TargetAssembly}!");
                }
            }