From 146456902a6893d67848c331e0e2797f7154667b Mon Sep 17 00:00:00 2001
From: rushiiMachine <33725716+rushiiMachine@users.noreply.github.com>
Date: Sat, 23 Mar 2024 23:35:54 -0700
Subject: [PATCH] feat: add `entireMethod` option to signature matching

---
 Osu.Stubs/Color.cs                |  3 +--
 Osu.Stubs/Opcode/LazyMethod.cs    |  5 +++--
 Osu.Stubs/Opcode/OpCodeMatcher.cs | 14 +++++++++-----
 3 files changed, 13 insertions(+), 9 deletions(-)

diff --git a/Osu.Stubs/Color.cs b/Osu.Stubs/Color.cs
index 1116abf..c3cc923 100644
--- a/Osu.Stubs/Color.cs
+++ b/Osu.Stubs/Color.cs
@@ -1,9 +1,8 @@
-using System;
 using Osu.Stubs.Opcode;
 
 namespace Osu.Stubs;
 
-using XnaColor = Object;
+using XnaColor = object;
 
 /// <summary>
 ///     Original: <c>Microsoft.Xna.Framework.Graphics.Color</c>
diff --git a/Osu.Stubs/Opcode/LazyMethod.cs b/Osu.Stubs/Opcode/LazyMethod.cs
index 96cd5ff..9625a48 100644
--- a/Osu.Stubs/Opcode/LazyMethod.cs
+++ b/Osu.Stubs/Opcode/LazyMethod.cs
@@ -18,10 +18,11 @@ public class LazyMethod
     /// </summary>
     /// <param name="name"><c>Class#Method</c> name of what this signature is matching.</param>
     /// <param name="signature">Sequential opcodes to search the target method with.</param>
-    internal LazyMethod(string name, IReadOnlyList<OpCode> signature)
+    /// <param name="entireMethod">Whether the signature is the entire method.</param>
+    internal LazyMethod(string name, IReadOnlyList<OpCode> signature, bool entireMethod = false)
     {
         _name = name;
-        _lazy = new Lazy<MethodBase?>(() => OpCodeMatcher.FindMethodBySignature(signature));
+        _lazy = new Lazy<MethodBase?>(() => OpCodeMatcher.FindMethodBySignature(signature, entireMethod));
     }
 
     /// <summary>
diff --git a/Osu.Stubs/Opcode/OpCodeMatcher.cs b/Osu.Stubs/Opcode/OpCodeMatcher.cs
index 40620ca..67f1a4b 100644
--- a/Osu.Stubs/Opcode/OpCodeMatcher.cs
+++ b/Osu.Stubs/Opcode/OpCodeMatcher.cs
@@ -10,8 +10,9 @@ public static class OpCodeMatcher
     ///     Search for a method inside the osu! assembly by an IL OpCode signature.
     /// </summary>
     /// <param name="signature">A set of sequential OpCodes to match.</param>
+    /// <param name="entireMethod">Whether the signature is the entire method to search for.</param>
     /// <returns>The found method or null if none found.</returns>
-    public static MethodInfo? FindMethodBySignature(IReadOnlyList<OpCode> signature)
+    public static MethodInfo? FindMethodBySignature(IReadOnlyList<OpCode> signature, bool entireMethod = false)
     {
         if (signature.Count <= 0) return null;
 
@@ -24,7 +25,7 @@ public static class OpCodeMatcher
             var instructions = method.GetMethodBody()?.GetILAsByteArray();
             if (instructions == null) continue;
 
-            if (InstructionsMatchesSignature(instructions, signature))
+            if (InstructionsMatchesSignature(instructions, signature, entireMethod))
                 return method;
         }
 
@@ -49,7 +50,7 @@ public static class OpCodeMatcher
             var instructions = method.GetMethodBody()?.GetILAsByteArray();
             if (instructions == null) continue;
 
-            if (InstructionsMatchesSignature(instructions, signature))
+            if (InstructionsMatchesSignature(instructions, signature, false))
                 return method;
         }
 
@@ -64,18 +65,21 @@ public static class OpCodeMatcher
     ///     for example.
     /// </param>
     /// <param name="signature">A set of sequential OpCodes to search for in instructions.</param>
+    /// <param name="entireMethod">Whether the signature should be the entire method.</param>
     /// <returns></returns>
     private static bool InstructionsMatchesSignature(
         IReadOnlyList<byte> ilInstructions,
-        IReadOnlyList<OpCode> signature)
+        IReadOnlyList<OpCode> signature,
+        bool entireMethod)
     {
         var sequentialMatching = 0;
         foreach (var instruction in new OpCodeReader(ilInstructions).GetOpCodes())
         {
             if (instruction == signature[sequentialMatching])
                 sequentialMatching++;
-            else
+            else if (!entireMethod)
                 sequentialMatching = 0;
+            else return false;
 
             if (sequentialMatching == signature.Count)
                 return true;