使用JavaAgent向类路径添加类

Question

我正在使用Java代理和Javassist向一些JDK类添加一些日志记录。实际上，当系统加载一些TLS类时，Javassist会向它们添加一些额外的字节码，以帮助我调试一些连接问题。

问题是，考虑到这个类包含在代理jar中：

package com.something.myagent;
public class MyAgentPrinter {
    public static final void sayHello() {
        System.out.println("Hello!");
    }
}

在我的代理的转换方法中，假设我试图使用javassist调用该类：

// this is only called for sun.security.ssl.Handshaker
ClassPool cp = getClassPool(classfileBuffer, className);
CtClass cc = cp.get(className);
CtMethod declaredMethod = cc.getDeclaredMethod("calculateKeys");
declaredMethod.insertAfter("com.something.myagent.MyAgentPrinter.sayHello();");
cc.freeze();
return cc.toBytecode();

你认为这行得通，但我得到了这个：

java.lang.NoClassDefFoundError: com/something/myagent/MyAgentPrinter
    at sun.security.ssl.Handshaker.printLogLine(Handshaker.java)
    at sun.security.ssl.Handshaker.calculateKeys(Handshaker.java:1160)
    at sun.security.ssl.ServerHandshaker.processMessage(ServerHandshaker.java:292)

有没有办法将该类[MyAgentPrinter]添加到应用程序的类路径中？

Answer 1

您的代理的jar文件已经添加到类路径中，如java所指定。lang.instrument软件包文档：

代理类将由系统类加载器加载（请参见ClassLoader.getSystemClassLoader）。这是类加载器，它通常加载包含应用程序main方法的类。premain方法将在与应用程序main方法相同的安全性和类加载器规则下运行。

这就是为什么Javassist在转换字节码时可以找到代理的类。

问题似乎是你正在改造一个太阳** 可能由引导加载程序或扩展加载程序加载的类。标准的类加载委托是
应用程序加载器→ 扩展装载机→ 引导加载程序，因此应用程序加载程序可用的类对扩展或引导加载程序加载的类不可用。

因此，要使它们对所有类都可用，必须将代理的类添加到引导加载程序：

public class MyAgent {
    public static void premain(String agentArgs, Instrumentation inst) throws IOException {
        JarURLConnection connection = (JarURLConnection)
            MyAgent.class.getResource("MyAgent.class").openConnection();
        inst.appendToBootstrapClassLoaderSearch(connection.getJarFile());

        // proceed
    }
}

在任何其他操作之前，即在加载要提供给仪表化代码的类之前，这样做是至关重要的。这意味着代理类本身，即包含premain方法的类不能被仪表化代码访问。代理类也不应该直接引用MyAgentPrinter，以避免意外的提前加载。

一个更可靠的方法是在Agent jar的清单中添加一个Boot-Class-Path条目，请参阅包留档的清单属性部分，以便在Agent启动之前添加该条目。但是之后，jar文件的名称不得更改。