fix: agent deadlock.

Author: ‘niuerzhuang’

dongtai-core/src/main/java/io/dongtai/iast/core/bytecode/IastClassFileTransformer.java

@@ -112,29 +112,39 @@ public byte[] transform(final ClassLoader loader,
                             final Class<?> classBeingRedefined,
                             final ProtectionDomain protectionDomain,
                             final byte[] srcByteCodeArray) {
+        String threadName = Thread.currentThread().getName();
+        if (threadName.startsWith("DongTai-IAST-Core")
+                || threadName.startsWith("DongTai-IAST-AgentStateMonitor")
+                || threadName.startsWith("DongTai-IAST-ConfigMonitor")
+                || threadName.startsWith("DongTai-IAST-FallbackConfigMonitor")
+                || threadName.startsWith("DongTai-IAST-HearBeatMonitor")
+                || threadName.startsWith("DongTai-IAST-PerformanceMonitor")) {
+            return null;
+        }
+
+        if (internalClassName == null
+                || internalClassName.startsWith("io/dongtai/")
+                || internalClassName.startsWith("com/secnium/iast/")
+                || internalClassName.startsWith("java/lang/iast/")
+                || internalClassName.startsWith("cn/huoxian/iast/")
+                || internalClassName.startsWith("META-INF/")
+                || "module-info".equals(internalClassName)) {
+            return null;
+        }
+
+        if (null != loader && loader.toString().toLowerCase().contains("rasp")) {
+            return null;
+        }
+
         try {
             ScopeManager.SCOPE_TRACKER.getPolicyScope().enterAgent();
 
-            if (internalClassName == null
-                    || internalClassName.startsWith("io/dongtai/")
-                    || internalClassName.startsWith("com/secnium/iast/")
-                    || internalClassName.startsWith("java/lang/iast/")
-                    || internalClassName.startsWith("cn/huoxian/iast/")
-                    || internalClassName.startsWith("META-INF/")
-                    || "module-info".equals(internalClassName)) {
-                return null;
-            }
-
             if (" com/alibaba/fastjson/JSON".substring(1).equals(internalClassName)) {
                 FastjsonCheck.setJsonClassLoader(loader);
             } else if (" com/alibaba/fastjson/parser/ParserConfig".substring(1).equals(internalClassName)) {
                 FastjsonCheck.setParseConfigClassLoader(loader);
             }
 
-            if (null != loader && loader.toString().toLowerCase().contains("rasp")) {
-                return null;
-            }
-
             if (loader != null && protectionDomain != null) {
                 final CodeSource codeSource = protectionDomain.getCodeSource();
                 if (codeSource == null) {
@@ -156,7 +166,6 @@ public byte[] transform(final ClassLoader loader,
 
             ClassContext classContext = new ClassContext(cr, loader);
             if (Modifier.isInterface(classContext.getModifier())) {
-                sourceCodeBak = null;
                 return null;
             }
             final String className = classContext.getClassName();
@@ -183,14 +192,14 @@ public byte[] transform(final ClassLoader loader,
                         transformMap.put(classBeingRedefined, srcByteCodeArray);
                     }
                     transformCount++;
+                    classDiagram.setLoader(null);
                     return dumpClassIfNecessary(cr.getClassName(), cw.toByteArray(), srcByteCodeArray);
                 }
             }
-            sourceCodeBak = null;
+            classDiagram.setLoader(null);
         } catch (Throwable throwable) {
             DongTaiLog.warn(ErrorCode.get("TRANSFORM_CLASS_FAILED"), internalClassName, throwable);
         } finally {
-            classDiagram.setLoader(null);
             ScopeManager.SCOPE_TRACKER.getPolicyScope().leaveAgent();
         }