fix StackOverflowError when processing JLookupStmt

bbhunter · Jan 12, 2024 · bc6a8ae · bc6a8ae
1 parent cf5831c
commit bc6a8ae
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/main/kotlin/net/bytedance/security/app/ui/TaintPathModeHtmlWriter.kt b/src/main/kotlin/net/bytedance/security/app/ui/TaintPathModeHtmlWriter.kt
@@ -224,7 +224,11 @@ class TaintPathModeHtmlWriter(
                                         if (gotoTgtLabelMap.containsKey(stmt)) {
                                             +"${gotoTgtLabelMap[stmt]}:\n"
                                         }
-                                        +"$index:  $stmt\n"
+                                        try {
+                                            +"$index:  $stmt\n"
+                                        } catch (ex: StackOverflowError) {
+                                            +"$index:   StackOverflowError: ${stmt.javaClass.canonicalName}"
+                                        }
                                     }
                                 }
                             }

diff --git a/src/main/kotlin/net/bytedance/security/app/util/TaskQueue.kt b/src/main/kotlin/net/bytedance/security/app/util/TaskQueue.kt
@@ -32,6 +32,11 @@ val oomHandler = CoroutineExceptionHandler { ctx, exception ->
         Log.logErr("${coroutineName} CoroutineException because of oom")
         exitProcess(37)
     }
+    if (exception is StackOverflowError) {
+        val coroutineName = ctx[CoroutineName]?.name
+        Log.logErr("${coroutineName} CoroutineException because of oom")
+        exitProcess(38)
+    }
     throw exception
 }