如何使用ASM将异常处理项添加至方法异常表末尾

ASM的方法异常表条目是按照visitTryCatchBlock的调用顺序依次添加的——如果你先调用自定义的visitTryCatchBlock，它就会排在异常表开头。要让自定义的异常处理逻辑排在末尾，核心思路是：先完整处理原方法的所有原有异常表条目，再添加自己的try-catch块。

以下分两种常用场景给出实现方案：

1. 直接使用ClassVisitor/MethodVisitor #

如果你继承ClassVisitor和MethodVisitor处理字节码，需要在自定义MethodVisitor中先委托原MethodVisitor处理所有原生的visitTryCatchBlock调用，等原方法的异常表条目全部写入后，再添加自己的try-catch逻辑。

示例代码片段：

class MyMethodVisitor extends MethodVisitor {
    private Label tryStart, tryEnd, catchStart;
    private boolean addedCustomTryCatch = false;

    public MyMethodVisitor(int api, MethodVisitor mv) {
        super(api, mv);
    }

    // 优先处理原方法的所有try-catch块
    @Override
    public void visitTryCatchBlock(Label start, Label end, Label handler, String type) {
        super.visitTryCatchBlock(start, end, handler, type);
    }

    // 在方法收尾阶段添加自定义try-catch
    @Override
    public void visitMaxs(int maxStack, int maxLocals) {
        if (!addedCustomTryCatch) {
            // 初始化标签（需提前在visitCode时将tryStart插入方法体开头）
            tryEnd = new Label();
            catchStart = new Label();

            // 将自定义try-catch添加到异常表末尾
            super.visitTryCatchBlock(tryStart, tryEnd, catchStart, "java/lang/Throwable");
            
            // 编写catch逻辑（示例为捕获后重新抛出，不干扰原方法执行）
            super.visitLabel(catchStart);
            super.visitVarInsn(ASTORE, 1);
            super.visitTypeInsn(NEW, "java/lang/RuntimeException");
            super.visitInsn(DUP);
            super.visitVarInsn(ALOAD, 1);
            super.visitMethodInsn(INVOKESPECIAL, "java/lang/RuntimeException", "<init>", "(Ljava/lang/Throwable;)V", false);
            super.visitInsn(ATHROW);

            // 在方法体末尾插入tryEnd标签
            super.visitLabel(tryEnd);
            addedCustomTryCatch = true;
        }
        super.visitMaxs(maxStack, maxLocals);
    }

    @Override
    public void visitCode() {
        tryStart = new Label();
        super.visitLabel(tryStart);
        super.visitCode();
    }
}

2. 使用ASM Tree API（ClassNode/MethodNode） #

如果用Tree API处理，逻辑会更直观：先将类结构加载到ClassNode，目标方法的tryCatchBlocks列表保存了所有原生异常表条目，直接将自定义的TryCatchBlockNode添加到该列表末尾即可。

示例代码片段：

// 加载原始类到ClassNode
ClassNode classNode = new ClassNode(ASM9);
new ClassReader(originalClassBytes).accept(classNode, 0);

// 遍历目标方法
for (MethodNode method : classNode.methods) {
    if ("yourTargetMethod".equals(method.name)) {
        // 创建自定义try-catch块的标签和节点
        LabelNode tryStart = new LabelNode();
        LabelNode tryEnd = new LabelNode();
        LabelNode catchStart = new LabelNode();
        TryCatchBlockNode customTcBlock = new TryCatchBlockNode(tryStart, tryEnd, catchStart, "java/lang/Throwable");
        
        // 添加到原异常表列表的末尾
        method.tryCatchBlocks.add(customTcBlock);

        // 调整方法体：在开头插入tryStart，结尾插入tryEnd，追加catch逻辑
        method.instructions.insertBefore(method.instructions.getFirst(), tryStart);
        method.instructions.insert(method.instructions.getLast(), tryEnd);
        method.instructions.add(catchStart);
        
        // 编写catch逻辑（示例为捕获后重新抛出）
        method.instructions.add(new VarInsnNode(ASTORE, 1));
        method.instructions.add(new TypeInsnNode(NEW, "java/lang/RuntimeException"));
        method.instructions.add(new InsnNode(DUP));
        method.instructions.add(new VarInsnNode(ALOAD, 1));
        method.instructions.add(new MethodInsnNode(INVOKESPECIAL, "java/lang/RuntimeException", "<init>", "(Ljava/lang/Throwable;)V", false));
        method.instructions.add(new InsnNode(ATHROW));
    }
}

// 生成修改后的字节码
ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);
classNode.accept(cw);
byte[] modifiedClassBytes = cw.toByteArray();

关键注意事项 #

• 确保自定义try-catch的标签范围正确包裹整个方法体，避免与原生try-catch的范围冲突。
• 若原方法有返回值，需保证try块末尾保留原生返回指令，catch逻辑不破坏原方法的正常执行流程（如捕获后重新抛出、或处理后返回合法值）。
• 使用ClassWriter.COMPUTE_FRAMES可自动计算栈帧和局部变量表，减少手动计算的错误。

内容的提问来源于stack exchange，提问作者a stand-out flamingo