Don't log remote API calls

Equinox- · Mar 30, 2024 · afa97f5 · afa97f5
1 parent 90480d7
commit afa97f5
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 0 deletions.
diff --git a/Meds.Wrapper/MedsCoreSystem.cs b/Meds.Wrapper/MedsCoreSystem.cs
@@ -59,7 +59,11 @@ protected override void Start()
         {
             PatchHelper.PatchAlways(true);
             if (Config.Metrics.PrometheusKey != null)
+            {
                 PatchHelper.Patch(typeof(PrometheusPatch));
+                PatchHelper.Patch(typeof(RemoteServerLessLogging));
+            }
+
             if (Config.Metrics.Network)
                 TransportLayerMetrics.Register();
             if (Config.Metrics.Player)

diff --git a/Meds.Wrapper/Output/Prometheus/PrometheusPatch.cs b/Meds.Wrapper/Output/Prometheus/PrometheusPatch.cs
@@ -1,10 +1,15 @@
 using System;
+using System.Collections.Generic;
 using System.IO;
 using System.IO.Compression;
+using System.Linq;
 using System.Net;
+using System.Reflection;
+using System.Reflection.Emit;
 using HarmonyLib;
 using Meds.Metrics;
 using VRage.Dedicated.RemoteAPI;
+using VRage.Logging;
 
 namespace Meds.Wrapper.Output.Prometheus
 {
@@ -45,4 +50,35 @@ public static bool Prefix(HttpListenerContext context)
             return false;
         }
     }
+
+    [HarmonyPatch(typeof(MyRemoteServer), "Listen")]
+    public sealed class RemoteServerLessLogging
+    {
+        public static IEnumerable<CodeInstruction> Transpiler(IEnumerable<CodeInstruction> instructions)
+        {
+            var removeLogTriggers = new[] { "Received request from", "URL:", "HTTP Method", "Request processed in" };
+            var removingLog = false;
+            foreach (var instruction in instructions)
+            {
+                if (instruction.opcode == OpCodes.Ldstr && instruction.operand is string str && removeLogTriggers.Any(trigger => str.Contains(trigger)))
+                    removingLog = true;
+                if (removingLog
+                    && instruction.operand is MethodInfo { IsStatic: false } method
+                    && method.ReturnType == typeof(void)
+                    && (method.DeclaringType == typeof(MyLog) || method.DeclaringType == typeof(NamedLogger))
+                    && !(method.Name == nameof(NamedLogger.OpenBlock) || method.Name == nameof(MyLog.IncreaseIndent) ||
+                         method.Name == nameof(MyLog.DecreaseIndent)))
+                {
+                    yield return new CodeInstruction(OpCodes.Nop).WithBlocks(instruction.blocks).WithLabels(instruction.labels);
+                    var pops = 1 + method.GetParameters().Length;
+                    for (var i = 0; i < pops; i++)
+                        yield return new CodeInstruction(OpCodes.Pop);
+                    removingLog = false;
+                    continue;
+                }
+
+                yield return instruction;
+            }
+        }
+    }
 }