基于ByteBuddy实现JavaAgent方法插装：为每个方法生成唯一前置调用入参

需求背景与核心要求 #

我们正在构建基于ByteBuddy的JavaAgent实现类方法插装，此前使用Javassist的实现与Mockito兼容性不佳，因此转用ByteBuddy。需通过DynamicType$Builder针对目标类的构造器、类初始化器、普通方法三类方法进行插装，核心要求：

• 前置拦截器beforeMethod需接收每个被插装方法的唯一参数（基于方法类名、名称、签名生成的murmur32哈希）
• 后置拦截器afterMethod为无参方法
• 每个被插装方法对应专属的前置MethodCall实例，所有被插装方法共用一个后置MethodCall实例

提供的示例代码参考：
前置/后置MethodCall生成逻辑：

private static MethodCall getInsertBefore(Method method) {
    // 基于方法类名、名称、签名生成murmur32哈希
    int murmur32Hash = ...;
    return MethodCall.invoke(named("beforeMethod"))
            .on(SomeClass.class)
            .withReference(murmur32Hash);
}
private static MethodCall getInsertAfter() {
    return MethodCall.invoke(named("afterMethod")).on(SomeClass.class); 
}

拦截器类定义：

public class SomeClass {
    @Advice.OnMethodEnter
    public static void beforeMethod(int murmur32HashCode) {
        /* murmur32HashCode为转换阶段生成，并非来自方法参数列表 */
        ...
    }
    @Advice.OnMethodExit
    public static void afterMethod() {
        ...
    }
}

问题核心 #

能否通过DynamicType$Builder遍历目标类所有声明方法，为每个方法创建对应的前置、后置MethodCall调用？此前Javassist可直观实现，但ByteBuddy现有示例多从被插装方法参数中选择拦截器参数，而非生成新值。查阅ByteBuddy源码后认为可通过MethodVisitor实现，但未理清流程；尝试自定义注解映射无法为每个方法设置不同值，推测需为每个方法添加常量并注入字节码引用。

实现方案 #

ByteBuddy完全支持该需求，无需直接操作MethodVisitor，可通过Advice结合ElementMatcher与DynamicType.Builder的方法遍历能力实现，核心思路是为每个方法生成唯一常量并在Advice中引用。

1. 实现Murmur32哈希生成工具 #

编写工具类生成方法的唯一哈希值：

public class MethodHashUtils {
    public static int generateMurmur32Hash(MethodDescription method) {
        // 拼接方法唯一标识：类名#方法名+方法描述符
        String uniqueKey = method.getDeclaringType().getName() + "#" + method.getName() + method.getDescriptor();
        // 这里使用Guava的Murmur32实现，也可自行实现哈希逻辑
        return Hashing.murmur32().hashString(uniqueKey, StandardCharsets.UTF_8).asInt();
    }
}

2. 自定义Advice类 #

修改拦截器类，通过Advice.FieldValue引用动态生成的静态常量：

public class MethodInterceptionAdvice {
    @Advice.OnMethodEnter
    public static void beforeMethod(@Advice.FieldValue(value = "$$METHOD_HASH$$", readOnly = true) int murmur32HashCode) {
        // 前置拦截逻辑
    }

    @Advice.OnMethodExit
    public static void afterMethod() {
        // 后置拦截逻辑
    }
}

3. 通过DynamicType.Builder遍历插装所有方法 #

遍历目标类的所有声明方法，为每个方法生成专属哈希常量，再插入前置/后置Advice：

public class ClassTransformer {
    public static DynamicType.Builder<?> transform(DynamicType.Builder<?> builder, TypeDescription typeDescription) {
        // 遍历当前类所有声明方法（含构造器、初始化器）
        for (MethodDescription method : typeDescription.getDeclaredMethods()) {
            // 跳过桥接方法、合成方法（可按需调整过滤规则）
            if (method.isBridge() || method.isSynthetic()) {
                continue;
            }

            // 生成当前方法的唯一哈希
            int methodHash = MethodHashUtils.generateMurmur32Hash(method);
            // 生成唯一的常量名称，避免类内冲突
            String constantName = "$$METHOD_HASH_" + method.getInternalName() + method.getDescriptor() + "$$";
            
            // 为目标类添加静态final常量存储哈希值
            builder = builder.defineField(
                    constantName,
                    int.class,
                    Visibility.PRIVATE,
                    FieldManifestation.STATIC,
                    FieldManifestation.FINAL
            ).value(methodHash);

            // 为当前方法插入Advice
            builder = builder.visit(Advice.to(MethodInterceptionAdvice.class).on(ElementMatchers.is(method)));
        }
        return builder;
    }
}

4. 在JavaAgent中应用转换器 #

在Agent的premain方法中通过ByteBuddy的AgentBuilder应用转换器：

public class MyAgent {
    public static void premain(String args, Instrumentation instrumentation) {
        new AgentBuilder.Default()
                .type(ElementMatchers.any()) // 按需调整目标类匹配规则
                .transform((builder, typeDescription, classLoader, module) ->
                        ClassTransformer.transform(builder, typeDescription))
                .installOn(instrumentation);
    }
}

关键说明 #

• 每个方法的哈希常量采用唯一命名规则，避免类内字段冲突
• 通过Advice.FieldValue直接引用静态常量，无需从方法参数中提取值
• 完全基于ByteBuddy高层API实现，无需操作底层MethodVisitor，兼容性与可维护性更强

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