Java ASM 字节码注入实战：正确修改方法并加载类-java教程-PHP中文网

Java ASM 字节码注入实战：正确修改方法并加载类

本文详解如何使用 asm 正确实现字节码注入——包括拦截目标方法、插入新逻辑、定义辅助方法，并通过 methodhandles.lookup().defineclass() 安全加载修改后的类，避免类加载冲突与重复方法错误。

本文详解如何使用 asm 正确实现字节码注入——包括拦截目标方法、插入新逻辑、定义辅助方法，并通过 methodhandles.lookup().defineclass() 安全加载修改后的类，避免类加载冲突与重复方法错误。

在 Java 中使用 ASM 进行动态字节码操作（如方法增强、日志注入、性能监控等）是一项强大但易出错的技术。初学者常陷入两个典型陷阱：类加载器跳过自定义 findClass 导致原始类被父加载器抢先加载，以及未正确覆盖目标方法而引发 ClassFormatError: Duplicate method name。本文将带你构建一个健壮、可复用的 ASM 注入方案。

✅ 正确的注入流程：四步关键实践

不依赖自定义 ClassLoader 的 findClass
使用 new ClassLoader() 默认以系统类加载器为父，loadClass() 会优先委托给父加载器——这意味着即使你重写了 findClass，原始 MyClass 仍可能已被系统类加载器加载，导致你的字节码被忽略。更可靠的方式是绕过类加载器委托机制，直接在当前运行时上下文中定义类：
```
Class> myClass = MethodHandles.lookup().defineClass(modifiedClass);
```
✅ 优势：无需继承 ClassLoader，避免双亲委派干扰；生成的类与当前模块共享同一运行时包和访问权限；支持反射及直接引用（如 MyClass obj = new MyClass();）。

Alibabacloud Sdk Client Initialization For Java
在 Java 中初始化和管理阿里云 SDK客户端。包括单例模式、线程安全、endpoint 与 region 配置、VPC 终端节点、同步与异步等。

下载

在 ClassVisitor.visitMethod 中精准拦截并替换方法
错误做法（如原代码中调用 cv.visitMethod(...) 后单独获取 MethodVisitor）会导致 ASM 仍保留原始 myMethod，再额外添加一个同名方法，触发 Duplicate method name 异常。正确方式是在 visitMethod 回调中判断目标方法，并返回 null 阻止其向下传递，同时立即注入新逻辑：

@Override
public MethodVisitor visitMethod(int access, String name, String descriptor,
                                 String signature, String[] exceptions) {
    MethodVisitor mv = super.visitMethod(access, name, descriptor, signature, exceptions);
    if ("myMethod".equals(name) && "()V".equals(descriptor)) {
        instrument(mv); // 替换方法体
        return null; // 阻止原始方法被写入
    }
    return mv;
}

在 visitEnd 中添加新方法（如 newMethod），确保其属于目标类
原代码试图在 main 中调用 cv.visitMethod(...) 添加 newMethod，但此时 cv 已脱离 ASM 访问链，该调用无效。应统一在 ClassVisitor.visitEnd() 中完成新增方法定义，并注意目标类名必须匹配（如 MyClass.newMethod，而非 MyClassInjector.newMethod）：

@Override
public void visitEnd() {
    // 在目标类中定义静态辅助方法
    MethodVisitor mv = cv.visitMethod(
        Opcodes.ACC_PUBLIC | Opcodes.ACC_STATIC,
        "newMethod", "()V", null, null);
    mv.visitCode();
    mv.visitFieldInsn(Opcodes.GETSTATIC, "java/lang/System", "out",
                      "Ljava/io/PrintStream;");
    mv.visitLdcInsn("Injected code");
    mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "java/io/PrintStream", "println",
                     "(Ljava/lang/String;)V", false);
    mv.visitInsn(Opcodes.RETURN);
    mv.visitMaxs(2, 0); // 注意：visitMaxs 参数需合理（此处为 maxStack=2, maxLocals=0）
    mv.visitEnd();
    super.visitEnd();
}

安全调用注入后的方法
使用现代反射 API 替代已弃用的 Class.newInstance()：

Object myObject = myClass.getConstructor().newInstance();
myClass.getMethod("myMethod").invoke(myObject);
// 输出：
// Injected code
// Hello, world! （原逻辑仍执行，取决于注入位置）

⚠️ 注意事项与最佳实践

ASM 版本兼容性：示例使用 Opcodes.ASM5，请根据 JDK 版本选用对应 ASM（如 JDK 17+ 推荐 ASM 9+），并在 ClassWriter 构造中启用 ClassWriter.COMPUTE_FRAMES（比 COMPUTE_MAXS 更鲁棒，尤其涉及异常表或局部变量时）。
visitMaxs 参数修正：visitMaxs(0, 0) 是危险的占位写法。ASM 无法自动推导栈深度时会抛出 AnalyzerException。建议使用 COMPUTE_FRAMES 并移除手动 visitMaxs，或依据实际指令计算（如上述 println 调用需栈深 ≥2）。
方法签名严格匹配：descriptor 必须完全一致（如 ()V 不等于 ()Ljava/lang/Object;），推荐用 Type.getMethodDescriptor 生成，避免手写错误。
生产环境进阶方案：若类已被加载（如 Spring 应用中 MyClass 启动即初始化），defineClass 将失败。此时必须采用 Java Agent + Instrumentation.retransformClasses()，通过 JVM TI 实现运行时重定义（需 premain 或 agentmain 支持）。

✅ 完整可运行示例（精简版）

import java.lang.invoke.MethodHandles;
import org.objectweb.asm.*;

public class MyClassInjector {
    public static void main(String[] args) throws Exception {
        ClassReader cr = new ClassReader("MyClass");
        ClassWriter cw = new ClassWriter(cr, ClassWriter.COMPUTE_FRAMES);
        cr.accept(new MyClassVisitor(cw), 0);
        byte[] modified = cw.toByteArray();

        Class> clazz = MethodHandles.lookup().defineClass(modified);
        Object instance = clazz.getDeclaredConstructor().newInstance();
        clazz.getMethod("myMethod").invoke(instance);
    }
}

class MyClassVisitor extends ClassVisitor {
    public MyClassVisitor(ClassVisitor cv) {
        super(Opcodes.ASM9, cv);
    }

    @Override
    public MethodVisitor visitMethod(int access, String name, String desc,
                                     String sig, String[] ex) {
        MethodVisitor mv = super.visitMethod(access, name, desc, sig, ex);
        if ("myMethod".equals(name) && "()V".equals(desc)) {
            // 注入前置逻辑：调用 MyClass.newMethod()
            mv.visitCode();
            mv.visitMethodInsn(Opcodes.INVOKESTATIC, "MyClass", "newMethod", "()V", false);
            // 保留原方法逻辑（可选：调用原始字节码或跳过）
            mv.visitFieldInsn(Opcodes.GETSTATIC, "java/lang/System", "out",
                              "Ljava/io/PrintStream;");
            mv.visitLdcInsn("Hello, world!");
            mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "java/io/PrintStream", "println",
                             "(Ljava/lang/String;)V", false);
            mv.visitInsn(Opcodes.RETURN);
            mv.visitMaxs(2, 1); // 栈深2，局部变量1（this）
            mv.visitEnd();
            return null;
        }
        return mv;
    }

    @Override
    public void visitEnd() {
        // 添加辅助方法到 MyClass
        MethodVisitor mv = cv.visitMethod(
            Opcodes.ACC_PUBLIC | Opcodes.ACC_STATIC,
            "newMethod", "()V", null, null);
        mv.visitCode();
        mv.visitFieldInsn(Opcodes.GETSTATIC, "java/lang/System", "out",
                          "Ljava/io/PrintStream;");
        mv.visitLdcInsn("Injected code");
        mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "java/io/PrintStream", "println",
                         "(Ljava/lang/String;)V", false);
        mv.visitInsn(Opcodes.RETURN);
        mv.visitMaxs(2, 0);
        mv.visitEnd();
        super.visitEnd();
    }
}