Support repeated 'invoke' calls on coroutines defined within inline lambdas

#KT-12782 Fixed

compiler/testData/codegen/box/coroutines/multipleInvokeCallsInsideInlineLambda1.kt

@@ -0,0 +1,88 @@
+class Controller {
+    var lastSuspension: Continuation<String>? = null
+    var result = "fail"
+    suspend fun suspendHere(x: Continuation<String>) {
+        lastSuspension = x
+    }
+
+    fun hasNext() = lastSuspension != null
+    fun next() {
+        val x = lastSuspension!!
+        lastSuspension = null
+        x.resume("56")
+    }
+}
+
+fun builder(coroutine c: Controller.() -> Continuation<Unit>) {
+    val controller1 = Controller()
+    val controller2 = Controller()
+
+    c(controller1).resume(Unit)
+    c(controller2).resume(Unit)
+
+    runControllers(controller1, controller2)
+}
+
+fun builder2(coroutine c: Controller.(Long, String) -> Continuation<Unit>) {
+    val controller1 = Controller()
+    val controller2 = Controller()
+
+    c(controller1, 1234567890123456789L, "Q").resume(Unit)
+    c(controller2, 1234567890123456789L, "Q").resume(Unit)
+
+    runControllers(controller1, controller2)
+}
+
+
+private fun runControllers(controller1: Controller, controller2: Controller) {
+    while (controller1.hasNext()) {
+        if (!controller2.hasNext()) throw RuntimeException("fail 1")
+
+        if (controller1.lastSuspension === controller2.lastSuspension) throw RuntimeException("equal references")
+
+        controller1.next()
+        controller2.next()
+    }
+
+    if (controller2.hasNext()) throw RuntimeException("fail 2")
+
+    if (controller1.result != "OK") throw RuntimeException("fail 3")
+    if (controller2.result != "OK") throw RuntimeException("fail 4")
+}
+
+inline fun run(b: () -> Unit) {
+    b()
+}
+
+fun box(): String {
+    // with capture and params
+
+    var x = "O"
+    var y = "K"
+
+    // inlined
+    run {
+        // no suspension
+        builder2 { a, b ->
+            if (a != 1234567890123456789L || b != "Q" ) return@builder2
+            result = x + y
+        }
+
+        // 1 suspension
+        builder2 { a, b ->
+            if (a != 1234567890123456789L || b != "Q" ) return@builder2
+            if (suspendHere() != "56") return@builder2
+            result = x + y
+        }
+
+        // 2 suspensions
+        builder2 { a, b ->
+            if (a != 1234567890123456789L || b != "Q" ) return@builder2
+            if (suspendHere() != "56") return@builder2
+            suspendHere()
+            result = x + y
+        }
+    }
+
+    return "OK"
+}

compiler/testData/codegen/box/coroutines/multipleInvokeCallsInsideInlineLambda2.kt

@@ -0,0 +1,109 @@
+class Controller {
+    var lastSuspension: Continuation<String>? = null
+    var result = "fail"
+    suspend fun suspendHere(x: Continuation<String>) {
+        lastSuspension = x
+    }
+
+    fun hasNext() = lastSuspension != null
+    fun next() {
+        val x = lastSuspension!!
+        lastSuspension = null
+        x.resume("56")
+    }
+}
+
+fun builder(coroutine c: Controller.() -> Continuation<Unit>) {
+    val controller1 = Controller()
+    val controller2 = Controller()
+
+    c(controller1).resume(Unit)
+    c(controller2).resume(Unit)
+
+    runControllers(controller1, controller2)
+}
+
+fun builder2(coroutine c: Controller.(Long, String) -> Continuation<Unit>) {
+    val controller1 = Controller()
+    val controller2 = Controller()
+
+    c(controller1, 1234567890123456789L, "Q").resume(Unit)
+    c(controller2, 1234567890123456789L, "Q").resume(Unit)
+
+    runControllers(controller1, controller2)
+}
+
+
+private fun runControllers(controller1: Controller, controller2: Controller) {
+    while (controller1.hasNext()) {
+        if (!controller2.hasNext()) throw RuntimeException("fail 1")
+
+        if (controller1.lastSuspension === controller2.lastSuspension) throw RuntimeException("equal references")
+
+        controller1.next()
+        controller2.next()
+    }
+
+    if (controller2.hasNext()) throw RuntimeException("fail 2")
+
+    if (controller1.result != "OK") throw RuntimeException("fail 3")
+    if (controller2.result != "OK") throw RuntimeException("fail 4")
+}
+
+inline fun run(b: () -> Unit) {
+    b()
+}
+
+fun box(): String {
+    // with capture and params
+    var x = "O"
+
+    // inlined
+    run {
+        var y = "K"
+
+        {
+            // no suspension
+            builder {
+                result = "OK"
+            }
+
+            // 1 suspension
+            builder {
+                if (suspendHere() != "56") return@builder
+                result = "OK"
+            }
+
+            // 2 suspensions
+            builder {
+                if (suspendHere() != "56") return@builder
+                suspendHere()
+                result = "OK"
+            }
+
+            // no suspension
+            builder2 { a, b ->
+                if (a != 1234567890123456789L || b != "Q" ) return@builder2
+                result = x + y
+            }
+
+            // 1 suspension
+            builder2 { a, b ->
+                if (a != 1234567890123456789L || b != "Q" ) return@builder2
+                if (suspendHere() != "56") return@builder2
+                result = x + y
+            }
+
+            // 2 suspensions
+            builder2 { a, b ->
+                if (a != 1234567890123456789L || b != "Q" ) return@builder2
+                if (suspendHere() != "56") return@builder2
+                suspendHere()
+                result = x + y
+            }
+
+        }()
+    }
+
+    return "OK"
+}

compiler/testData/codegen/box/coroutines/multipleInvokeCallsInsideInlineLambda3.kt

@@ -0,0 +1,107 @@
+class Controller {
+    var lastSuspension: Continuation<String>? = null
+    var result = "fail"
+    suspend fun suspendHere(x: Continuation<String>) {
+        lastSuspension = x
+    }
+
+    fun hasNext() = lastSuspension != null
+    fun next() {
+        val x = lastSuspension!!
+        lastSuspension = null
+        x.resume("56")
+    }
+}
+
+fun builder(coroutine c: Controller.() -> Continuation<Unit>) {
+    val controller1 = Controller()
+    val controller2 = Controller()
+
+    c(controller1).resume(Unit)
+    c(controller2).resume(Unit)
+
+    runControllers(controller1, controller2)
+}
+
+fun builder2(coroutine c: Controller.(Long, String) -> Continuation<Unit>) {
+    val controller1 = Controller()
+    val controller2 = Controller()
+
+    c(controller1, 1234567890123456789L, "Q").resume(Unit)
+    c(controller2, 1234567890123456789L, "Q").resume(Unit)
+
+    runControllers(controller1, controller2)
+}
+
+
+private fun runControllers(controller1: Controller, controller2: Controller) {
+    while (controller1.hasNext()) {
+        if (!controller2.hasNext()) throw RuntimeException("fail 1")
+
+        if (controller1.lastSuspension === controller2.lastSuspension) throw RuntimeException("equal references")
+
+        controller1.next()
+        controller2.next()
+    }
+
+    if (controller2.hasNext()) throw RuntimeException("fail 2")
+
+    if (controller1.result != "OK") throw RuntimeException("fail 3")
+    if (controller2.result != "OK") throw RuntimeException("fail 4")
+}
+
+inline fun run(b: () -> Unit) {
+    b()
+}
+
+fun box(): String {
+    var x = "O"
+
+    {
+        var y = "K"
+        // inlined
+        run {
+            // no suspension
+            builder {
+                result = "OK"
+            }
+
+            // 1 suspension
+            builder {
+                if (suspendHere() != "56") return@builder
+                result = "OK"
+            }
+
+            // 2 suspensions
+            builder {
+                if (suspendHere() != "56") return@builder
+                suspendHere()
+                result = "OK"
+            }
+
+
+            // no suspension
+            builder2 { a, b ->
+                if (a != 1234567890123456789L || b != "Q" ) return@builder2
+                result = x + y
+            }
+
+            // 1 suspension
+            builder2 { a, b ->
+                if (a != 1234567890123456789L || b != "Q" ) return@builder2
+                if (suspendHere() != "56") return@builder2
+                result = x + y
+            }
+
+            // 2 suspensions
+            builder2 { a, b ->
+                if (a != 1234567890123456789L || b != "Q" ) return@builder2
+                if (suspendHere() != "56") return@builder2
+                suspendHere()
+                result = x + y
+            }
+        }
+    } ()
+
+    return "OK"
+}