Support special characters in coroutine names in JSON dumps

Author: dkhalanskyjb

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

@@ -184,8 +184,8 @@ internal object DebugProbesImpl {
         val coroutinesInfoAsJson = ArrayList<String>(size)
         for (info in coroutinesInfo) {
             val context = info.context
-            val name = context[CoroutineName.Key]?.name?.toStringWithQuotes()
-            val dispatcher = context[CoroutineDispatcher.Key]?.toStringWithQuotes()
+            val name = context[CoroutineName.Key]?.name?.toStringRepr()
+            val dispatcher = context[CoroutineDispatcher.Key]?.toStringRepr()
             coroutinesInfoAsJson.add(
                 """
                 {
@@ -221,7 +221,7 @@ internal object DebugProbesImpl {
                 {
                     "declaringClass": "${element.className}",
                     "methodName": "${element.methodName}",
-                    "fileName": ${element.fileName?.toStringWithQuotes()},
+                    "fileName": ${element.fileName?.toStringRepr()},
                     "lineNumber": ${element.lineNumber}
                 }
                 """.trimIndent()
@@ -231,7 +231,7 @@ internal object DebugProbesImpl {
         return "[${stackTraceElementsInfoAsJson.joinToString()}]"
     }
 
-    private fun Any.toStringWithQuotes() = "\"$this\""
+    private fun Any.toStringRepr() = toString().repr()
 
     /*
      * Internal (JVM-public) method used by IDEA debugger as of 1.4-M3.
@@ -307,7 +307,6 @@ internal object DebugProbesImpl {
      * Internal (JVM-public) method used by IDEA debugger as of 1.4-M3.
      * It is similar to [enhanceStackTraceWithThreadDumpImpl], but uses debugger-facing [DebugCoroutineInfo] type.
      */
-    @Suppress("unused")
     fun enhanceStackTraceWithThreadDump(
         info: DebugCoroutineInfo,
         coroutineTrace: List<StackTraceElement>
@@ -592,3 +591,19 @@ internal object DebugProbesImpl {
 
     private val StackTraceElement.isInternalMethod: Boolean get() = className.startsWith("kotlinx.coroutines")
 }
+
+private fun String.repr(): String = buildString {
+    append('"')
+    for (c in this@repr) {
+        when (c) {
+            '"' -> append("\\\"")
+            '\\' -> append("\\\\")
+            '\b' -> append("\\b")
+            '\n' -> append("\\n")
+            '\r' -> append("\\r")
+            '\t' -> append("\\t")
+            else -> append(c)
+        }
+    }
+    append('"')
+}

kotlinx-coroutines-debug/test/DumpCoroutineInfoAsJsonAndReferencesTest.kt

@@ -29,6 +29,10 @@ class DumpCoroutineInfoAsJsonAndReferencesTest : DebugTestBase() {
     fun testDumpOfNamedCoroutine() =
         runTestWithNamedDeferred("Name")
 
+    @Test
+    fun testDumpOfNamedCoroutineWithSpecialCharacters() =
+        runTestWithNamedDeferred("Name with\n \"special\" characters\\/\t\b")
+
     @Test
     fun testDumpWithNoCoroutines() {
         val dumpResult = DebugProbesImpl.dumpCoroutinesInfoAsJsonAndReferences()