如何在Java ASM字节码中向每个finally块插入自定义指令？

我完全懂你遇到的坑——用ASM直接操作TryCatchBlockNode的handler后，反编译出来的代码完全变样了，这其实是因为Java字节码对finally的实现和我们写的源码逻辑不一样，编译器给finally做了语法糖处理。

问题根源 #

Java编译器处理finally块时，会把finally里的代码复制到两个地方：

1. 当try块正常执行结束后，通过GOTO指令跳转到finally代码段执行；
2. 生成一个捕获所有Throwable的catch块，在catch块里执行finally代码，然后重新抛出异常。

你之前的代码只给catch块的handler插入了指令，但没处理正常执行路径的finally代码，而且插入的位置破坏了原有的finally结构，导致反编译后出现重复的代码和奇怪的catch块。

正确的实现方式 #

我们需要找到finally代码的所有执行入口（异常路径和正常路径），然后在每个入口的finally代码开头插入自定义指令。具体步骤如下：

1. 识别出编译器为finally生成的TryCatchBlockNode：就是type为null的那个（它负责捕获所有Throwable实现异常路径的finally执行）；
2. 在这个handler的开头插入自定义指令，处理异常路径；
3. 找到try块正常结束后跳转到finally的GOTO指令，在该指令之后插入同样的自定义指令，处理正常路径。

下面是修正后的代码：

List<MethodNode> methods = (List<MethodNode>) classNode.methods;
for (MethodNode method : methods) {
    List<TryCatchBlockNode> tryCatchBlocks = method.tryCatchBlocks;
    for (TryCatchBlockNode tcb : tryCatchBlocks) {
        // 筛选出finally对应的try-catch块（type为null表示捕获所有Throwable）
        if (tcb.type == null) {
            // 构建要插入的自定义指令列表
            InsnList customInsns = new InsnList();
            customInsns.add(new FieldInsnNode(GETSTATIC, "java/lang/System", "out", "Ljava/io/PrintStream;"));
            customInsns.add(new LdcInsnNode("AAAA"));
            customInsns.add(new MethodInsnNode(INVOKEVIRTUAL, "java/io/PrintStream", "println", "(Ljava/lang/String;)V"));

            // 1. 处理异常路径：在finally代码的开头（handler标签处）插入指令
            method.instructions.insert(tcb.handler, customInsns);

            // 2. 处理正常路径：找到try块结束后跳转到finally的GOTO指令
            LabelNode tryEnd = tcb.end;
            AbstractInsnNode current = tryEnd.getNext();
            while (current != null) {
                // 寻找跳转到finally起始位置的GOTO指令
                if (current.getType() == AbstractInsnNode.JUMP_INSN) {
                    JumpInsnNode gotoInsn = (JumpInsnNode) current;
                    if (gotoInsn.label == tcb.handler) {
                        // 注意要克隆指令列表，同一个InsnList不能被添加到多个位置
                        method.instructions.insert(gotoInsn, customInsns.clone());
                        break;
                    }
                }
                current = current.getNext();
            }
        }
    }
}

关键细节说明 #

• 为什么要克隆指令列表：ASM的InsnList是有状态的，同一个实例不能被添加到指令流的多个位置，所以必须用clone()复制一份用于正常路径的插入。
• 控制流的准确性：一定要确保找到的GOTO指令确实是跳转到finally代码的入口，避免误修改其他跳转逻辑。
• 字节码层面的理解：ASM操作的是字节码，不是源码，所以必须熟悉Java编译器对语法糖的处理逻辑，才能准确修改代码结构。

这样处理后，不管是try块正常执行还是抛出异常，finally块的开头都会执行你插入的自定义指令，反编译后的代码也会和预期一致。

内容的提问来源于stack exchange，提问作者TOvidiu