From 0d59561bee4cf7db10d53a8aa58952ae65e856b5 Mon Sep 17 00:00:00 2001
From: Wither OrNot
Date: Thu, 13 Feb 2025 09:49:35 -0500
Subject: Initial commit

Co-authored-by: neko <109633131+nekoppai@users.noreply.github.com>
Co-authored-by: Lyssa <75037904+thecatontheceiling@users.noreply.github.com>
Co-authored-by: abbodi1406 <33669284+abbodi1406@users.noreply.github.com>
---
 LibTSforge/Modifiers/RearmReset.cs | 45 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 LibTSforge/Modifiers/RearmReset.cs

(limited to 'LibTSforge/Modifiers/RearmReset.cs')

diff --git a/LibTSforge/Modifiers/RearmReset.cs b/LibTSforge/Modifiers/RearmReset.cs
new file mode 100644
index 0000000..be2b174
--- /dev/null
+++ b/LibTSforge/Modifiers/RearmReset.cs
@@ -0,0 +1,45 @@
+namespace LibTSforge.Modifiers
+{
+    using System;
+    using System.Collections.Generic;
+    using System.Linq;
+    using LibTSforge.PhysicalStore;
+
+    public static class RearmReset
+    {
+        public static void Reset(PSVersion version, bool production)
+        {
+            Utils.KillSPP();
+
+            Logger.WriteLine("Writing TrustedStore data...");
+
+            using (IPhysicalStore store = Utils.GetStore(version, production))
+            {
+                List<PSBlock> blocks;
+
+                if (version == PSVersion.Win7)
+                {
+                    blocks = store.FindBlocks(0xA0000).ToList();
+                }
+                else
+                {
+                    blocks = store.FindBlocks("__##USERSEP-RESERVED##__$$REARM-COUNT$$").ToList();
+                }
+
+                foreach (PSBlock block in blocks)
+                {
+                    if (version == PSVersion.Win7)
+                    {
+                        store.SetBlock(block.KeyAsStr, block.ValueAsInt, new byte[8]);
+                    }
+                    else
+                    {
+                        store.SetBlock(block.KeyAsStr, block.ValueAsStr, new byte[8]);
+                    }
+                }
+
+                Logger.WriteLine("Successfully reset all rearm counters.");
+            }
+        }
+    }
+}
-- 
cgit v1.2.3