[debug] Introduce SafeAgentPremain class to avoid errors during debug if initialization of AgentPremain failed.

Author: mvicsokolova

kotlinx-coroutines-core/build.gradle.kts

@@ -167,7 +167,7 @@ fun setupManifest(jar: Jar) {
     jar.manifest {
         attributes(
             mapOf(
-                "Premain-Class" to "kotlinx.coroutines.debug.internal.AgentPremain",
+                "Premain-Class" to "kotlinx.coroutines.debug.internal.SafeAgentPremain",
                 "Can-Retransform-Classes" to "true",
             )
         )

kotlinx-coroutines-core/jvm/src/debug/internal/AgentPremain.kt

@@ -69,3 +69,28 @@ internal object AgentPremain {
         }
     }
 }
+
+/**
+ * This class is loaded if and only if kotlinx-coroutines-core was used as a javaagent argument.
+ * It serves as a "safe" wrapper around [AgentPremain] that does not require any kotlin-stdlib classes to be loaded for its initialization.
+ *
+ * Initialization of [AgentPremain] may fail, e.g., because it started before the required kotlin-stdlib classes were loaded.
+ * If `kotlinx.coroutines.apply.debug.agent` is set to `false`, the coroutine debug agent will not be applied.
+ */
+@Suppress("unused")
+internal object SafeAgentPremain {
+    @JvmStatic
+    fun premain(args: String?, instrumentation: Instrumentation) {
+        // If this property is disabled, the debug agent will not be applied at all
+        val applyDebugAgent = System.getProperty("kotlinx.coroutines.apply.debug.agent", "true").toBoolean()
+        if (!applyDebugAgent) return
+        try {
+            AgentPremain.premain(args, instrumentation)
+        } catch (t: Throwable) {
+            error("Something went wrong while loading kotlinx.coroutines debug agent.\n" +
+                "Please ensure that kotlin-stdlib is present in the classpath.\n" +
+                "Alternatively, you can disable kotlinx.coroutines debug agent with this system property: `kotlinx.coroutines.apply.debug.agent=false`\n" +
+                t.cause)
+        }
+    }
+}