fix: agent deadlock.

Author: ‘niuerzhuang’

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

@@ -62,6 +62,13 @@ public static IastClassFileTransformer getInstance(Instrumentation inst, PolicyM
         return INSTANCE;
     }
 
+    public static IastClassFileTransformer getInstance() {
+        if (null != INSTANCE) {
+            return INSTANCE;
+        }
+        return null;
+    }
+
     IastClassFileTransformer(Instrumentation inst, PolicyManager policyManager) {
         this.inst = inst;
         this.isDumpClass = EngineManager.getInstance().isEnableDumpClass();
@@ -192,11 +199,9 @@ public byte[] transform(final ClassLoader loader,
                         transformMap.put(classBeingRedefined, srcByteCodeArray);
                     }
                     transformCount++;
-                    classDiagram.setLoader(null);
                     return dumpClassIfNecessary(cr.getClassName(), cw.toByteArray(), srcByteCodeArray);
                 }
             }
-            classDiagram.setLoader(null);
         } catch (Throwable throwable) {
             DongTaiLog.warn(ErrorCode.get("TRANSFORM_CLASS_FAILED"), internalClassName, throwable);
         } finally {
@@ -356,5 +361,9 @@ public void reTransform() {
     public static HashMap<Object, byte[]> getTransformMap() {
         return transformMap;
     }
+
+    public IastClassDiagram getClassDiagram() {
+        return classDiagram;
+    }
 }
 

dongtai-core/src/main/java/io/dongtai/iast/core/init/impl/TransformEngine.java

@@ -73,6 +73,9 @@ public void destroy() {
                 DongTaiLog.error(ErrorCode.get("TRANSFORM_ENGINE_DESTROY_REDEFINE_CLASSES_FAILED"), e);
             }
         }
+        if (IastClassFileTransformer.getInstance() != null) {
+            IastClassFileTransformer.getInstance().getClassDiagram().setLoader(null);
+        }
         inst = null;
         classFileTransformer = null;
     }