From 76afccd9db62392e19e481cd29c3efec34cf4c7a Mon Sep 17 00:00:00 2001
From: zznty <94796179+zznty@users.noreply.github.com>
Date: Tue, 1 Feb 2022 15:11:04 +0700
Subject: [PATCH] shitty gc calls removal

---
 Torch/FodyWeavers.xml                         |  1 +
 Torch/FodyWeavers.xsd                         |  1 +
 .../MSIL/MsilInstructionExtensions.cs         | 21 ++++++++
 Torch/Patches/GcCollectPatch.cs               | 52 +++++++++++++++++++
 Torch/Torch.csproj                            |  1 +
 5 files changed, 76 insertions(+)
 create mode 100644 Torch/Patches/GcCollectPatch.cs

diff --git a/Torch/FodyWeavers.xml b/Torch/FodyWeavers.xml
index d5abfed..e2c5835 100644
--- a/Torch/FodyWeavers.xml
+++ b/Torch/FodyWeavers.xml
@@ -1,3 +1,4 @@
 <Weavers xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="FodyWeavers.xsd">
   <PropertyChanged />
+  <InfoOf />
 </Weavers>
\ No newline at end of file
diff --git a/Torch/FodyWeavers.xsd b/Torch/FodyWeavers.xsd
index 69dbe48..de66ede 100644
--- a/Torch/FodyWeavers.xsd
+++ b/Torch/FodyWeavers.xsd
@@ -53,6 +53,7 @@
             </xs:attribute>
           </xs:complexType>
         </xs:element>
+        <xs:element name="InfoOf" minOccurs="0" maxOccurs="1" type="xs:anyType" />
       </xs:all>
       <xs:attribute name="VerifyAssembly" type="xs:boolean">
         <xs:annotation>
diff --git a/Torch/Managers/PatchManager/MSIL/MsilInstructionExtensions.cs b/Torch/Managers/PatchManager/MSIL/MsilInstructionExtensions.cs
index a69ada4..b0b8532 100644
--- a/Torch/Managers/PatchManager/MSIL/MsilInstructionExtensions.cs
+++ b/Torch/Managers/PatchManager/MSIL/MsilInstructionExtensions.cs
@@ -221,6 +221,27 @@ namespace Torch.Managers.PatchManager.MSIL
         {
             return new MsilInstruction(argument.Position < 0xFF ? OpCodes.Ldarga_S : OpCodes.Ldarga).InlineValue(argument);
         }
+
+        /// <summary>
+        /// Copy Instruction with null operand and new opcode.
+        /// </summary>
+        /// <param name="instruction">Instruction to copy from.</param>
+        /// <param name="opCode">New opCode to assign.</param>
+        /// <returns>Copy of <see cref="instruction"/>.</returns>
+        public static MsilInstruction CopyWithoutOperand(this MsilInstruction instruction, OpCode opCode)
+        {
+            var copied = new MsilInstruction(opCode);
+            foreach (var label in instruction.Labels)
+            {
+                copied.Labels.Add(label);
+            }
+            foreach (var operation in instruction.TryCatchOperations)
+            {
+                copied.TryCatchOperations.Add(operation);
+            }
+
+            return copied;
+        }
         #endregion
 
         #region Constant Utils
diff --git a/Torch/Patches/GcCollectPatch.cs b/Torch/Patches/GcCollectPatch.cs
new file mode 100644
index 0000000..1d4e04c
--- /dev/null
+++ b/Torch/Patches/GcCollectPatch.cs
@@ -0,0 +1,52 @@
+using System;
+using System.Collections.Generic;
+using System.Reflection;
+using System.Reflection.Emit;
+using Havok;
+using Sandbox.Engine.Voxels.Planet;
+using Sandbox.Game.World;
+using Torch.Managers.PatchManager;
+using Torch.Managers.PatchManager.MSIL;
+using Torch.Utils;
+using VRage;
+
+namespace Torch.Patches;
+
+[PatchShim]
+internal static class GcCollectPatch
+{
+    // FUCK YO KEEN
+    // every call results in freeze for seconds
+    private static readonly MethodBase[] _targets =
+    {
+        Info.OfMethod<MyPlanetTextureMapProvider>(nameof(MyPlanetTextureMapProvider.GetHeightmap)),
+        Info.OfMethod<MyPlanetTextureMapProvider>(nameof(MyPlanetTextureMapProvider.GetDetailMap)),
+        Info.OfMethod<MyPlanetTextureMapProvider>(nameof(MyPlanetTextureMapProvider.GetMaps)),
+        Info.OfMethod<MySession>(nameof(MySession.Unload)),
+        Info.OfMethod("HavokWrapper", "Havok.HkBaseSystem", nameof(HkBaseSystem.Quit)),
+        Info.OfMethod<MySimpleProfiler>(nameof(MySimpleProfiler.LogPerformanceTestResults)),
+        Info.OfConstructor<MySession>("MySyncLayer,Boolean")
+    };
+
+    public static void Patch(PatchContext context)
+    {
+        foreach (var target in _targets)
+        {
+            context.GetPattern(target).AddTranspiler(nameof(CollectRemovalTranspiler));
+        }
+    }
+
+    private static IEnumerable<MsilInstruction> CollectRemovalTranspiler(IEnumerable<MsilInstruction> instructions)
+    {
+        foreach (var instruction in instructions)
+        {
+            if (instruction.Operand is MsilOperandInline<MethodInfo> operand &&
+                operand.Value.DeclaringType == typeof(GC))
+            {
+                yield return instruction.CopyWithoutOperand(OpCodes.Nop);
+                continue;
+            }
+            yield return instruction;
+        }
+    }
+}
\ No newline at end of file
diff --git a/Torch/Torch.csproj b/Torch/Torch.csproj
index ec593d2..8ded7bc 100644
--- a/Torch/Torch.csproj
+++ b/Torch/Torch.csproj
@@ -20,6 +20,7 @@
 <!--   <Import Project="$(SolutionDir)\TransformOnBuild.targets" /> -->
   <ItemGroup>
     <PackageReference Include="ControlzEx" Version="5.0.1" />
+    <PackageReference Include="InfoOf.Fody" Version="2.1.0" />
     <PackageReference Include="MahApps.Metro" Version="2.4.9" />
     <PackageReference Include="MonoMod.RuntimeDetour" Version="22.1.4.3" />
     <PackageReference Include="NLog" Version="4.7.13" />