CFG: break and continue are now handled in the same (safe) way EA-85060 Fixed

(cherry picked from commit eec4b1b)

compiler/frontend/src/org/jetbrains/kotlin/cfg/ControlFlowBuilder.kt

@@ -40,7 +40,7 @@ interface ControlFlowBuilder {
     fun exitBlockScope(block: KtElement)
 
     fun getSubroutineExitPoint(labelElement: KtElement): Label?
-    fun getLoopConditionEntryPoint(loop: KtLoopExpression): Label
+    fun getLoopConditionEntryPoint(loop: KtLoopExpression): Label?
     fun getLoopExitPoint(loop: KtLoopExpression): Label?
 
     // Declarations

compiler/frontend/src/org/jetbrains/kotlin/cfg/ControlFlowBuilderAdapter.kt

@@ -127,7 +127,7 @@ abstract class ControlFlowBuilderAdapter : ControlFlowBuilder {
         return delegateBuilder.getSubroutineExitPoint(labelElement)
     }
 
-    override fun getLoopConditionEntryPoint(loop: KtLoopExpression): Label {
+    override fun getLoopConditionEntryPoint(loop: KtLoopExpression): Label? {
         return delegateBuilder.getLoopConditionEntryPoint(loop)
     }
 

compiler/frontend/src/org/jetbrains/kotlin/cfg/ControlFlowProcessor.kt

@@ -806,7 +806,7 @@ class ControlFlowProcessor(private val trace: BindingTrace) {
             val loop = getCorrespondingLoop(expression)
             if (loop != null) {
                 if (jumpDoesNotCrossFunctionBoundary(expression, loop)) {
-                    builder.jump(builder.getLoopConditionEntryPoint(loop), expression)
+                    builder.getLoopConditionEntryPoint(loop)?.let { builder.jump(it, expression) }
                 }
             }
         }

compiler/frontend/src/org/jetbrains/kotlin/cfg/pseudocode/ControlFlowInstructionsGenerator.kt

@@ -166,9 +166,8 @@ class ControlFlowInstructionsGenerator : ControlFlowBuilderAdapter() {
         override val currentSubroutine: KtElement
             get() = pseudocode.correspondingElement
 
-        override fun getLoopConditionEntryPoint(loop: KtLoopExpression): Label {
-            val blockInfo = elementToLoopInfo[loop] ?: error("expected LoopInfo for ${loop.text}")
-            return blockInfo.conditionEntryPoint
+        override fun getLoopConditionEntryPoint(loop: KtLoopExpression): Label? {
+            return elementToLoopInfo[loop]?.conditionEntryPoint
         }
 
         override fun getLoopExitPoint(loop: KtLoopExpression): Label? {