Add EXACTLY_ONCE contract to suspendCoroutine* functions

Update line numbers in the affected test.

js/js.translator/testData/lineNumbers/coroutine.kt

@@ -14,4 +14,4 @@ suspend fun bar(): Unit {
     println(a + b)
 }
 
-// LINES: 38 4 4 4 7 5 5 44 44 5 92 44 5 5 6 4 4 4 9 15 9 9 9 * 9 15 10 10 11 11 11 11 11 * 11 12 12 13 13 13 13 13 13 13 14 14 * 9 15 9 9 9 9
+// LINES: 39 4 4 4 7 5 5 45 45 5 92 45 5 5 6 4 4 4 9 15 9 9 9 * 9 15 10 10 11 11 11 11 11 * 11 12 12 13 13 13 13 13 13 13 14 14 * 9 15 9 9 9 9

libraries/stdlib/jvm/test/coroutines/CoroutinesTest.kt

@@ -39,11 +39,17 @@ class DispatcherSwitcher(
     private val contextDispatcher: TestDispatcher,
     private val resumeDispatcher: TestDispatcher
 ) {
-    suspend fun run(): Int = suspendCoroutine { cont ->
+    suspend fun run(): Int {
-        contextDispatcher.assertThread()
+        val sideEffect: Int
-        resumeDispatcher.executor.execute {
+        val runResult = suspendCoroutine<Int> { cont ->
-            resumeDispatcher.assertThread()
+            contextDispatcher.assertThread()
-            cont.resume(42)
+            resumeDispatcher.executor.execute {
+                resumeDispatcher.assertThread()
+                cont.resume(42)
+            }
+            sideEffect = 21
         }
+        assertEquals(21, sideEffect)
+        return runResult
     }
 }

libraries/stdlib/src/kotlin/coroutines/Continuation.kt

@@ -1,10 +1,11 @@
 /*
- * Copyright 2010-2018 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
  * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
  */
 
 package kotlin.coroutines
 
+import kotlin.contracts.*
 import kotlin.coroutines.intrinsics.*
 import kotlin.internal.InlineOnly
 
@@ -138,12 +139,14 @@ public fun <R, T> (suspend R.() -> T).startCoroutine(
  */
 @SinceKotlin("1.3")
 @InlineOnly
-public suspend inline fun <T> suspendCoroutine(crossinline block: (Continuation<T>) -> Unit): T =
+public suspend inline fun <T> suspendCoroutine(crossinline block: (Continuation<T>) -> Unit): T {
-    suspendCoroutineUninterceptedOrReturn { c: Continuation<T> ->
+    contract { callsInPlace(block, InvocationKind.EXACTLY_ONCE) }
+    return suspendCoroutineUninterceptedOrReturn { c: Continuation<T> ->
         val safe = SafeContinuation(c.intercepted())
         block(safe)
         safe.getOrThrow()
     }
+}
 
 /**
  * Returns the context of the current coroutine.

libraries/stdlib/src/kotlin/coroutines/intrinsics/Intrinsics.kt

@@ -1,5 +1,5 @@
 /*
- * Copyright 2010-2018 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
  * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
  */
 
@@ -8,6 +8,7 @@
 
 package kotlin.coroutines.intrinsics
 
+import kotlin.contracts.*
 import kotlin.coroutines.*
 import kotlin.internal.InlineOnly
 
@@ -37,8 +38,10 @@ import kotlin.internal.InlineOnly
 @SinceKotlin("1.3")
 @InlineOnly
 @Suppress("UNUSED_PARAMETER", "RedundantSuspendModifier")
-public suspend inline fun <T> suspendCoroutineUninterceptedOrReturn(crossinline block: (Continuation<T>) -> Any?): T =
+public suspend inline fun <T> suspendCoroutineUninterceptedOrReturn(crossinline block: (Continuation<T>) -> Any?): T {
+    contract { callsInPlace(block, InvocationKind.EXACTLY_ONCE) }
     throw NotImplementedError("Implementation of suspendCoroutineUninterceptedOrReturn is intrinsic")
+}
 
 /**
  * This value is used as a return value of [suspendCoroutineUninterceptedOrReturn] `block` argument to state that