From c440593788db19e909447b1092f3ec4d3038cb49 Mon Sep 17 00:00:00 2001
From: Connor McLaughlin <stenzek@gmail.com>
Date: Thu, 2 Dec 2021 18:40:31 +1000
Subject: [PATCH] CPU/CodeCache: Add InvalidateAll() method

---
 src/core/cpu_code_cache.cpp | 71 +++++++++++++++++++++++--------------
 src/core/cpu_code_cache.h   |  3 ++
 2 files changed, 47 insertions(+), 27 deletions(-)

diff --git a/src/core/cpu_code_cache.cpp b/src/core/cpu_code_cache.cpp
index cea715719..37a8d1264 100644
--- a/src/core/cpu_code_cache.cpp
+++ b/src/core/cpu_code_cache.cpp
@@ -836,44 +836,61 @@ void InvalidCodeFunction()
 
 #endif
 
+static void InvalidateBlock(CodeBlock* block)
+{
+  // Invalidate forces the block to be checked again.
+  Log_DebugPrintf("Invalidating block at 0x%08X", block->GetPC());
+  block->invalidated = true;
+
+  if (block->can_link)
+  {
+    const u32 frame_number = System::GetFrameNumber();
+    const u32 frame_diff = frame_number - block->invalidate_frame_number;
+    if (frame_diff <= INVALIDATE_THRESHOLD_TO_DISABLE_LINKING)
+    {
+      Log_DevPrintf("Block 0x%08X has been invalidated in %u frames, disabling linking", block->GetPC(), frame_diff);
+      block->can_link = false;
+    }
+    else
+    {
+      // It's been a while since this block was modified, so it's all good.
+      block->invalidate_frame_number = frame_number;
+    }
+  }
+
+  UnlinkBlock(block);
+
+#ifdef WITH_RECOMPILER
+  SetFastMap(block->GetPC(), FastCompileBlockFunction);
+#endif
+}
+
 void InvalidateBlocksWithPageIndex(u32 page_index)
 {
   DebugAssert(page_index < Bus::RAM_8MB_CODE_PAGE_COUNT);
   auto& blocks = m_ram_block_map[page_index];
   for (CodeBlock* block : blocks)
-  {
-    // Invalidate forces the block to be checked again.
-    Log_DebugPrintf("Invalidating block at 0x%08X", block->GetPC());
-    block->invalidated = true;
-
-    if (block->can_link)
-    {
-      const u32 frame_number = System::GetFrameNumber();
-      const u32 frame_diff = frame_number - block->invalidate_frame_number;
-      if (frame_diff <= INVALIDATE_THRESHOLD_TO_DISABLE_LINKING)
-      {
-        Log_DevPrintf("Block 0x%08X has been invalidated in %u frames, disabling linking", block->GetPC(), frame_diff);
-        block->can_link = false;
-      }
-      else
-      {
-        // It's been a while since this block was modified, so it's all good.
-        block->invalidate_frame_number = frame_number;
-      }
-    }
-
-    UnlinkBlock(block);
-
-#ifdef WITH_RECOMPILER
-    SetFastMap(block->GetPC(), FastCompileBlockFunction);
-#endif
-  }
+    InvalidateBlock(block);
 
   // Block will be re-added next execution.
   blocks.clear();
   Bus::ClearRAMCodePage(page_index);
 }
 
+void InvalidateAll()
+{
+  for (auto& it : s_blocks)
+  {
+    CodeBlock* block = it.second;
+    if (block && !block->invalidated)
+      InvalidateBlock(block);
+  }
+
+  Bus::ClearRAMCodePageFlags();
+  for (auto& it : m_ram_block_map)
+    it.clear();
+}
+
 void RemoveReferencesToBlock(CodeBlock* block)
 {
   BlockMap::iterator iter = s_blocks.find(block->key.GetPC());
diff --git a/src/core/cpu_code_cache.h b/src/core/cpu_code_cache.h
index 1b4debe5d..161c78d89 100644
--- a/src/core/cpu_code_cache.h
+++ b/src/core/cpu_code_cache.h
@@ -140,6 +140,9 @@ void Reinitialize();
 /// Invalidates all blocks which are in the range of the specified code page.
 void InvalidateBlocksWithPageIndex(u32 page_index);
 
+/// Invalidates all blocks in the cache.
+void InvalidateAll();
+
 template<PGXPMode pgxp_mode>
 void InterpretCachedBlock(const CodeBlock& block);