轻松搞定代码插桩，AGP 8.0 ASM 插桩新姿势了解一下

2024-01-15 20:47:21

AGP 8.0 中 ASM 插桩：字节码插桩的新篇章

随着 Android Gradle 插件 (AGP) 8.0 的发布，字节码插桩世界迎来了一次重大变革。Transform 接口已成为历史，取而代之的是更强大、更灵活的 ASM 插桩。本文将深入探讨 ASM 插桩的精髓，并指导你使用 AGP 官方提供的强大 API 实现代码插桩。

ASM 插桩简介

ASM 插桩是一种字节码插桩技术，它允许你对已编译的字节码进行修改，而无需修改源代码。通过在特定点插入新代码，你可以扩展、增强或更改应用程序行为。与传统插桩方法相比，ASM 插桩提供了更高的灵活性和对细节的控制。

AGP 官方 ASM 插桩 API

AGP 团队认识到 ASM 插桩的重要性，因此提供了新的 API，使代码插桩变得前所未有的简单。该 API 包含三个关键类：

AsmClassVisitorFactory： 该工厂类让你创建自定义的 ClassVisitor，它将在字节码修改过程中访问和操作类。
AsmClassVisitor： 这是一个 ClassVisitor 的子类，它提供用于遍历和修改类的字节码的方法。
AsmMethodVisitor： 这是一个 MethodVisitor 的子类，它提供用于遍历和修改方法字节码的方法。

使用 ASM 插桩 API

实现代码插桩的过程包含三个主要步骤：

创建自定义 AsmClassVisitorFactory： 创建一个自定义工厂类并重写 create() 方法，它将创建并返回一个 AsmClassVisitor 对象。
创建自定义 AsmClassVisitor： 创建一个自定义类，继承 AsmClassVisitor，并重写你希望修改的特定字节码方法。例如，你可以修改方法的访问权限或插入新代码。
在 build.gradle 中配置 ASM 插桩： 在 build.gradle 文件中添加以下配置：

android {
    buildFeatures {
        asmEnabled true
    }
    
    asmClassVisitorFactory = "com.example.MyAsmClassVisitorFactory"
}

示例代码

以下是一个简单的示例，演示如何使用 ASM 插桩 API 在类构造函数中添加日志记录：

public class MyAsmClassVisitor extends AsmClassVisitor {
    public MyAsmClassVisitor(int api, AsmClassVisitor av) {
        super(api, av);
    }

    @Override
    public MethodVisitor visitMethod(int access, String name, String desc, String signature, String[] exceptions) {
        MethodVisitor mv = super.visitMethod(access, name, desc, signature, exceptions);
        if (name.equals("<init>")) {
            mv = new AsmMethodVisitor(mv) {
                @Override
                public void visitInsn(int opcode, Object operand) {
                    if (opcode == Opcodes.RETURN) {
                        mv.visitFieldInsn(Opcodes.GETSTATIC, "java/lang/System", "out", "Ljava/io/PrintStream;");
                        mv.visitLdcInsn("MyClass constructor called");
                        mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL, "java/io/PrintStream", "println", "(Ljava/lang/String;)V", false);
                    }
                    super.visitInsn(opcode, operand);
                }
            };
        }
        return mv;
    }
}