From aecb925b7b20a30e7eac8fe8b04651dc2d929a2b Mon Sep 17 00:00:00 2001
From: Denis Zharkov <denis.zharkov@jetbrains.com>
Date: Tue, 13 Jan 2015 20:28:52 +0300
Subject: [PATCH] Refined skipping last goto when inlining try/catch/finally
 block

---
 .../kotlin/codegen/ExpressionCodegen.java     |  3 ++
 .../codegen/inline/InlineCodegenUtil.java     | 13 ++++++++
 .../inline/InternalFinallyBlockInliner.java   | 31 +++----------------
 ...StoreStackBeforeInlineMethodTransformer.kt |  5 ++-
 .../src/kotlin/jvm/internal/InlineMarker.java |  4 +++
 5 files changed, 28 insertions(+), 28 deletions(-)

diff --git a/compiler/backend/src/org/jetbrains/kotlin/codegen/ExpressionCodegen.java b/compiler/backend/src/org/jetbrains/kotlin/codegen/ExpressionCodegen.java
index f295d24d353..803919fbdf9 100644
--- a/compiler/backend/src/org/jetbrains/kotlin/codegen/ExpressionCodegen.java
+++ b/compiler/backend/src/org/jetbrains/kotlin/codegen/ExpressionCodegen.java
@@ -1746,6 +1746,9 @@ public class ExpressionCodegen extends JetVisitor<StackValue, StackValue> implem
         }
 
         if (tryCatchBlockEnd != null) {
+            if (context.isInlineFunction()) {
+                InlineCodegenUtil.generateGoToTryCatchBlockEndMarker(v);
+            }
             v.goTo(tryCatchBlockEnd);
         }
 
diff --git a/compiler/backend/src/org/jetbrains/kotlin/codegen/inline/InlineCodegenUtil.java b/compiler/backend/src/org/jetbrains/kotlin/codegen/inline/InlineCodegenUtil.java
index e31858f1050..88f61b39bcd 100644
--- a/compiler/backend/src/org/jetbrains/kotlin/codegen/inline/InlineCodegenUtil.java
+++ b/compiler/backend/src/org/jetbrains/kotlin/codegen/inline/InlineCodegenUtil.java
@@ -75,6 +75,7 @@ public class InlineCodegenUtil {
     public static final String INLINE_MARKER_CLASS_NAME = "kotlin/jvm/internal/InlineMarker";
     public static final String INLINE_MARKER_BEFORE_METHOD_NAME = "beforeInlineCall";
     public static final String INLINE_MARKER_AFTER_METHOD_NAME = "afterInlineCall";
+    public static final String INLINE_MARKER_GOTO_TRY_CATCH_BLOCK_END = "goToTryCatchBlockEnd";
 
     @Nullable
     public static MethodNode getMethodNode(
@@ -391,6 +392,18 @@ public class InlineCodegenUtil {
         }
     }
 
+    public static void generateGoToTryCatchBlockEndMarker(@NotNull InstructionAdapter v) {
+        v.invokestatic(INLINE_MARKER_CLASS_NAME, INLINE_MARKER_GOTO_TRY_CATCH_BLOCK_END, "()V", false);
+    }
+
+    public static boolean isGoToTryCatchBlockEnd(@NotNull AbstractInsnNode node) {
+        if (!(node.getPrevious() instanceof MethodInsnNode)) return false;
+        MethodInsnNode previous = (MethodInsnNode) node.getPrevious();
+        return node.getOpcode() == Opcodes.GOTO &&
+               INLINE_MARKER_CLASS_NAME.equals(previous.owner) &&
+               INLINE_MARKER_GOTO_TRY_CATCH_BLOCK_END.equals(previous.name);
+    }
+
     public static class LabelTextifier extends Textifier {
 
         public LabelTextifier() {
diff --git a/compiler/backend/src/org/jetbrains/kotlin/codegen/inline/InternalFinallyBlockInliner.java b/compiler/backend/src/org/jetbrains/kotlin/codegen/inline/InternalFinallyBlockInliner.java
index f03c016b0b1..93b148e2a5f 100644
--- a/compiler/backend/src/org/jetbrains/kotlin/codegen/inline/InternalFinallyBlockInliner.java
+++ b/compiler/backend/src/org/jetbrains/kotlin/codegen/inline/InternalFinallyBlockInliner.java
@@ -470,8 +470,7 @@ public class InternalFinallyBlockInliner extends CoveringTryCatchNodeProcessor<T
 
         TryCatchBlockNodeInfo nextIntervalWithSameDefaultHandler = sameDefaultHandler.get(1);
         AbstractInsnNode startFinallyChain = tryCatchBlock.getNode().end;
-        AbstractInsnNode endFinallyChainExclusive = skipLastGotoIfNeeded(nextIntervalWithSameDefaultHandler.getNode().handler,
-                                                        nextIntervalWithSameDefaultHandler.getNode().start);
+        AbstractInsnNode endFinallyChainExclusive = skipLastGotoIfNeeded(nextIntervalWithSameDefaultHandler.getNode().start);
 
         FinallyBlockInfo finallyInfo = new FinallyBlockInfo(startFinallyChain.getNext(), endFinallyChainExclusive);
 
@@ -483,37 +482,15 @@ public class InternalFinallyBlockInliner extends CoveringTryCatchNodeProcessor<T
     }
 
     @NotNull
-    private AbstractInsnNode skipLastGotoIfNeeded(
-            @NotNull LabelNode defaultHandlerStartLabel,
+    private static AbstractInsnNode skipLastGotoIfNeeded(
             @NotNull AbstractInsnNode lastFinallyInsExclusive
     ) {
 
         AbstractInsnNode prevLast = getPrevNoLineNumberOrLabel(lastFinallyInsExclusive, true);
         assert prevLast != null : "Empty finally block: " + lastFinallyInsExclusive;
 
-        if (prevLast.getOpcode() == Opcodes.GOTO) {
-            //There we should understand whether goto is jump over catches or last break/continue command inside finally.
-            //If it's a jump over catches so next is true:
-            //      1. jump label should go after default catch handler start label
-            //          AND
-            //      2. it shouldn't be present in default catch block, otherwise it break/continue
-            LabelNode targetJump = ((JumpInsnNode) prevLast).label;
-
-            InsnList instructions = inlineFun.instructions;
-            if (instructions.indexOf(defaultHandlerStartLabel) < instructions.indexOf(targetJump)) { //1 condition
-                AbstractInsnNode cur = defaultHandlerStartLabel;
-                while (cur != targetJump) {
-                    if (cur.getOpcode() == Opcodes.GOTO) {
-                        //noinspection ConstantConditions
-                        if (((JumpInsnNode) cur).label == targetJump) { //fail of 2 condition
-                            return lastFinallyInsExclusive;
-                        }
-                    }
-                    cur = cur.getNext();
-                }
-
-                return prevLast;
-            }
+        if (InlineCodegenUtil.isGoToTryCatchBlockEnd(prevLast)) {
+            return prevLast.getPrevious();
         }
         return lastFinallyInsExclusive;
     }
diff --git a/compiler/backend/src/org/jetbrains/kotlin/codegen/optimization/StoreStackBeforeInlineMethodTransformer.kt b/compiler/backend/src/org/jetbrains/kotlin/codegen/optimization/StoreStackBeforeInlineMethodTransformer.kt
index 13f56120358..e3bae6ebae2 100644
--- a/compiler/backend/src/org/jetbrains/kotlin/codegen/optimization/StoreStackBeforeInlineMethodTransformer.kt
+++ b/compiler/backend/src/org/jetbrains/kotlin/codegen/optimization/StoreStackBeforeInlineMethodTransformer.kt
@@ -75,7 +75,10 @@ private fun isInlineMarker(insn: AbstractInsnNode, markerName: String? = null):
     return insn.getOpcode() == Opcodes.INVOKESTATIC &&
            insn is MethodInsnNode &&
            insn.owner == InlineCodegenUtil.INLINE_MARKER_CLASS_NAME &&
-           if (markerName != null) markerName == insn.name else true
+           if (markerName != null) markerName == insn.name else (
+               insn.name == InlineCodegenUtil.INLINE_MARKER_BEFORE_METHOD_NAME ||
+               insn.name == InlineCodegenUtil.INLINE_MARKER_AFTER_METHOD_NAME
+           )
 }
 
 private fun process(methodNode: MethodNode, frames: Array<Frame<BasicValue>?>) {
diff --git a/core/runtime.jvm/src/kotlin/jvm/internal/InlineMarker.java b/core/runtime.jvm/src/kotlin/jvm/internal/InlineMarker.java
index 5b750f54779..2623dce7f13 100644
--- a/core/runtime.jvm/src/kotlin/jvm/internal/InlineMarker.java
+++ b/core/runtime.jvm/src/kotlin/jvm/internal/InlineMarker.java
@@ -25,4 +25,8 @@ public class InlineMarker {
     public static void afterInlineCall() {
 
     }
+
+    public static void goToTryCatchBlockEnd() {
+
+    }
 }
\ No newline at end of file