From 6773122d0b1598af78d4619b8af2811be11f6b95 Mon Sep 17 00:00:00 2001
From: oSumAtrIX <johan.melkonyan1@web.de>
Date: Wed, 21 Sep 2022 16:44:25 +0200
Subject: [PATCH] fix(custom-video-buffer): use correct offset for
 `getMaxBuffer` call injection

---
 .../misc/videobuffer/patch/CustomVideoBufferPatch.kt  | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/main/kotlin/app/revanced/patches/youtube/misc/videobuffer/patch/CustomVideoBufferPatch.kt b/src/main/kotlin/app/revanced/patches/youtube/misc/videobuffer/patch/CustomVideoBufferPatch.kt
index d21803dd5..ef1f37188 100644
--- a/src/main/kotlin/app/revanced/patches/youtube/misc/videobuffer/patch/CustomVideoBufferPatch.kt
+++ b/src/main/kotlin/app/revanced/patches/youtube/misc/videobuffer/patch/CustomVideoBufferPatch.kt
@@ -86,7 +86,7 @@ class CustomVideoBufferPatch : BytecodePatch(
     }
 
     private fun execMaxBuffer() {
-        val (method, result) = MaxBufferFingerprint.unwrap()
+        val (method, result) = MaxBufferFingerprint.unwrap(true, -1)
         val (index, register) = result
 
         method.addInstructions(
@@ -121,10 +121,15 @@ class CustomVideoBufferPatch : BytecodePatch(
         )
     }
 
-    private fun MethodFingerprint.unwrap(): Pair<MutableMethod, Pair<Int, Int>> {
+    private fun MethodFingerprint.unwrap(
+        forEndIndex: Boolean = false,
+        offset: Int = 0
+    ): Pair<MutableMethod, Pair<Int, Int>> {
         val result = this.result!!
         val method = result.mutableMethod
-        val index = result.scanResult.patternScanResult!!.startIndex
+        val scanResult = result.scanResult.patternScanResult!!
+        val index = (if (forEndIndex) scanResult.endIndex else scanResult.startIndex) + offset
+
         val register = (method.instruction(index) as OneRegisterInstruction).registerA
 
         return method to (index to register)