Java Agent技能实现实战：从字节码操作到动态加载

1次阅读

共计 1908 个字符，预计需要花费 5 分钟才能阅读完成。

Java Agent 技术最吸引人的地方在于它能无侵入地监控和修改运行中的 Java 程序。通过直接操作字节码，我们可以在不修改源代码的情况下实现性能监控、动态增强等功能。这比传统 AOP（Aspect-Oriented Programming）更底层，也更灵活。

传统 AOP 在面对一些复杂场景时显得力不从心。比如，它很难对第三方库的代码进行增强，也无法在运行时动态添加功能。热部署也是个头疼的问题，每次修改都需要重启应用，这在生产环境是不可接受的。此外，很多性能监控工具对 JVM 内部的细节无能为力，形成监控盲区。

Java Agent 支持两种加载方式：

Premain 模式 ：在 JVM 启动时通过-javaagent 参数加载
Agentmain 模式：通过 Attach API 在运行时动态加载

Premain 模式更稳定，但需要重启应用；Agentmain 模式更灵活，但对 JVM 版本有要求。

这里以 ASM 为例，展示如何修改方法字节码。假设我们要在所有方法入口处添加日志：

public class LogTransformer implements ClassFileTransformer {
    @Override
    public byte[] transform(ClassLoader loader, String className, 
            Class<?> classBeingRedefined,
            ProtectionDomain protectionDomain, byte[] classfileBuffer) {if (!className.startsWith("com/example")) {return null; // 只处理特定包}

        ClassReader reader = new ClassReader(classfileBuffer);
        ClassWriter writer = new ClassWriter(reader, ClassWriter.COMPUTE_MAXS);

        ClassVisitor visitor = new ClassVisitor(Opcodes.ASM7, writer) {
            @Override
            public MethodVisitor visitMethod(int access, String name, 
                    String descriptor, String signature, String[] exceptions) {MethodVisitor mv = super.visitMethod(access, name, descriptor, signature, exceptions);
                return new MethodVisitor(Opcodes.ASM7, mv) {
                    @Override
                    public void visitCode() {
                        // 在方法开始处插入日志代码
                        mv.visitLdcInsn("Method" + name + "called");
                        mv.visitMethodInsn(Opcodes.INVOKESTATIC, 
                                "java/lang/System", "out", "println", false);
                        super.visitCode();}
                };
            }
        };

        reader.accept(visitor, ClassReader.EXPAND_FRAMES);
        return writer.toByteArray();}
}

为了避免类冲突，建议：