Java Agent for Bytecode Instrumentation的问题

Question

这是我第一次实现java代理，并试图学习一些关于字节码的工具。阅读了几个介绍和教程后，我编写了一个包含两个类（夏季和应用程序）的小应用程序。现在我想通过运行方法的premain一个Java代理使用下面的代码显示的执行路径：

public class TestJavaAgent { 
    public static void premain(String agentArgument, 
           Instrumentation instrumentation){ 
     instrumentation.addTransformer(new ClassFileTransformer() { 

      @Override 
      public byte[] transform(ClassLoader classLoader, String s, Class<?> aClass, ProtectionDomain protectionDomain, byte[] bytes) throws IllegalClassFormatException { 

       ClassPool cp = ClassPool.getDefault(); 
        try { 
         CtClass cc = cp.get("Summer"); 
         CtMethod methods [] = cc.getMethods(); 

         for(CtMethod method : methods){ 
          System.out.println("Entering "+method.getName()); 
          method.addLocalVariable("elapsedTime", CtClass.longType); 
          method.insertBefore("elapsedTime = System.currentTimeMillis();"); 
          method.insertAfter("{elapsedTime = System.currentTimeMillis() - elapsedTime;" 
            + "System.out.println(\"Method Executed in ms: \" + elapsedTime);}"); 
         } 
         return cc.toBytecode(); 

        } catch (Exception ex) { 
         return bytes; 
        } 
      } 
     }); 
    } 
} 

我通过java -javaagent{Agent JAR} -jar {Application Jar}开始代理，但它没有打印插入的消息什么。调试完代码后，我意识到“ClassPool.getDefault（）”后面的所有内容都无法到达，但我不知道为什么。有人能帮我吗？

Answer 1

变换器应该将正在传递的类作为参数进行转换，而不是您喜欢的任意类。注册后，将会调用加载的所有类，包括您自己使用的类（首先是ClassPool）。所以你正在创建一个循环依赖。

您必须检查类名参数并等待直到您想要转换的类的方法被调用。对于所有其他课程，只需返回null。

public class TestJavaAgent { 
    public static void premain(String agentArgument, Instrumentation instrumentation) { 
     instrumentation.addTransformer(new ClassFileTransformer() { 

      @Override 
      public byte[] transform(ClassLoader classLoader, 
       String className, Class<?> aClass, ProtectionDomain protectionDomain, byte[] bytes) throws IllegalClassFormatException { 

       if(!className.equals("Summer")) return null; 

       ClassPool cp = ClassPool.getDefault(); 
       try { 
        // use the class bytes your received as parameter 
        cp.insertClassPath(new ByteArrayClassPath(className, bytes)); 

        CtClass cc = cp.get("Summer"); 
        CtMethod[] methods = cc.getMethods(); 

        for(CtMethod method : methods){ 
         System.out.println("Entering "+method.getName()); 
         method.addLocalVariable("elapsedTime", CtClass.longType); 
         method.insertBefore("elapsedTime = System.currentTimeMillis();"); 
         method.insertAfter("{elapsedTime = System.currentTimeMillis() - elapsedTime;" 
           + "System.out.println(\"Method Executed in ms: \" + elapsedTime);}"); 
        } 
        return cc.toBytecode(); 

       } catch (Exception ex) { 
        return null; 
       } 
      } 
     }); 
    } 
} 

注意，返回null最好返回原来的数组，如果你没有任何改变的，则JVM可以立即意识到你没有任何改变，没有寻找到数组内容。