javaagent(使用Javassist编写)无法正常工作

时间:2017-07-26 15:39:38

标签: java instrumentation javassist javaagents

我正在编写一个java代理并使用javassist库来修改字节码。 我有一个代理主类(MyAgent)和一个类文件转换器类。 MyAgent.java注册转换器类MyClzTransformer

public class MyAgent
{
    private static Instrumentation instrumentation;

    public static void premain(String arg, Instrumentation inst)
    {
        inst.addTransformer(new MyClzTransformer());
    }
}

-

public class MyClzTransformer implements ClassFileTransformer
{
    public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException
    {
        if (className.startsWith("a/b/"))
        {
            ClassPool clPool = ClassPool.getDefault();
            System.out.println("clPool =======> " + clPool);
            try
            {
                CtClass ctClass = clPool.get(className);
                System.out.println("ct class: " + ctClass + ": methods = " + Arrays.deepToString(ctClass.getMethods()));
                for(CtMethod method : ctClass.getMethods())
                {
                    method.insertAfter("System.out.println(\" ============ added instrumented code ============ \");");
                }
            }
            catch (NotFoundException | CannotCompileException | IOException e)
            {
                e.printStackTrace();
            }
        }
        return classfileBuffer;
    }
}

当我将MyAgent jar作为javaagent运行时如下:

 java -javaagent:<path to myagent.jar> a.b.MyTestProgram

我没有得到任何日志(sysouts)输出。

这是一个类加载器问题吗?因为,clPool.get(className)之后的日志没有执行。我没有从try block内获得任何输出。

1 个答案:

答案 0 :(得分:1)

有一个基本错误。虽然类名用/分隔,即if (className.startsWith("a/b/")),但正在使用相同格式的calssName 加载类,这是不正确的。

修正为CtClass ctClass = clPool.get(className.replaceAll("/","."));