added CreateMemoryPatch

roflmuffin · Dec 12, 2023 · 964b9b4 · 964b9b4
1 parent 6138806
commit 964b9b4
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/managed/CounterStrikeSharp.API/Core/BasePlugin.cs b/managed/CounterStrikeSharp.API/Core/BasePlugin.cs
@@ -30,6 +30,7 @@
 using CounterStrikeSharp.API.Modules.Entities;
 using Microsoft.Extensions.Localization;
 using Microsoft.Extensions.Logging;
+using CounterStrikeSharp.API.Modules.Memory;
 
 namespace CounterStrikeSharp.API.Core
 {
@@ -126,6 +127,9 @@ public void Dispose()
         internal readonly Dictionary<Delegate, EntityIO.EntityOutputCallback> EntitySingleOutputHooks =
             new Dictionary<Delegate, EntityIO.EntityOutputCallback>();
 
+        internal readonly Dictionary<string, MemoryPatch> MemoryPatches =
+            new Dictionary<string, MemoryPatch>();
+
         public readonly List<Timer> Timers = new List<Timer>();
 
         public delegate HookResult GameEventHandler<T>(T @event, GameEventInfo info) where T : GameEvent;
@@ -388,6 +392,18 @@ public Timer AddTimer(float interval, Action callback, TimerFlags? flags = null)
             return timer;
         }
 
+        public MemoryPatch? CreateMemoryPatch(string patchName, string patchSignature)
+        {
+            MemoryPatch patch = new MemoryPatch(patchName, patchSignature);
+
+            if (patch.PerformPatch())
+            {
+                MemoryPatches.Add(patchName, patch);
+                return patch;
+            }
+
+            return null;
+        }
 
         public void RegisterAllAttributes(object instance)
         {
@@ -585,6 +601,11 @@ protected virtual void Dispose(bool disposing)
                 timer.Kill();
             }
 
+            foreach (var patch in MemoryPatches)
+            {
+                patch.Value.UndoPatch();
+            }
+
             _disposed = true;
         }
     }