如何使用 ByteBuddy 的 @Advice.AllArguments 替换输入参数?

How to replace input arguments using ByteBuddy's @Advice.AllArguments?

我正在使用 ByteBuddy 的 @Advice 来转换我的 类,它工作正常,直到我尝试替换输入参数。

我有一个 FooServicejoin 方法,它只是用 space.

连接两个字符串
public class FooService {
    public String join(String message, String message1) {
        return message + " " + message1;
    }
}

我还有另一种方法,它采用 Object[] args 数组输入并更改数组中的元素。

    public static ArgsProcessor argsProcessor = args -> {
        args[0] = args[0] + "-suffix";
        args[1] = "replaced";
    };

我尝试了不同的方法来使用 argsProcessor 来操作 @Advice.OnMethodEnter 方法中的输入参数。对我来说,以下建议实现几乎是等效的,应该都有效,但不知何故只有 Advice1 有效。

    public static class Advice1 {
        @Advice.OnMethodEnter
        public static void onEnter(@Advice.AllArguments(readOnly = false, typing = Assigner.Typing.DYNAMIC) Object[] args) {
            Object[] newArgs = Arrays.copyOf(args, args.length);
            ArgsProcessor argsProcessor = Demo.argsProcessor;
            argsProcessor.process(newArgs);
            args = newArgs;
        }
    }

    public static class Advice2 {
        @Advice.OnMethodEnter
        public static void onEnter(@Advice.AllArguments(readOnly = false, typing = Assigner.Typing.DYNAMIC) Object[] args) {
            Object[] newArgs = new Object[args.length];
            ArgsProcessor argsProcessor = Demo.argsProcessor;
            argsProcessor.process(args);
            System.arraycopy(args, 0, newArgs, 0, args.length);
            args = newArgs;
        }
    }

    public static class Advice3 {
        @Advice.OnMethodEnter
        public static void onEnter(@Advice.AllArguments(readOnly = false, typing = Assigner.Typing.DYNAMIC) Object[] args) {
            Object[] newArgs = Arrays.copyOf(args, args.length);
            try {
                ArgsProcessor argsProcessor = Demo.argsProcessor;
                argsProcessor.process(newArgs);
                args = newArgs;
            } catch (Exception ex) {
                ex.printStackTrace();
            }
        }
    }

    public static class Advice4 {
        @Advice.OnMethodEnter
        public static void onEnter(@Advice.AllArguments(readOnly = false, typing = Assigner.Typing.DYNAMIC) Object[] args) {
            ArgsProcessor argsProcessor = Demo.argsProcessor;
            argsProcessor.process(args);
        }
    }

    public static class Advice5 {
        @Advice.OnMethodEnter
        public static void onEnter(@Advice.AllArguments(readOnly = false, typing = Assigner.Typing.DYNAMIC) Object[] args) {
            ArgsProcessor argsProcessor = Demo.argsProcessor;
            argsProcessor.process(args);
            args = Arrays.copyOf(args, args.length);
        }
    }

输出

Advice1 a-suffix replaced
Advice2 a b
Advice3 a b
Advice4 a b
Advice5 a b

代码片段https://gist.github.com/raptium/ab7830e5d7f7cba43bbd2c2a5c7b38e0

运行 你的代码,我得到

Advice1 a-suffix replaced 
Advice2 a b 
Advice3 a-suffix replaced 
Advice4 a b 
Advice5 a b

我期望的是什么。 Byte Buddy 使用建议方法作为模板。这段代码并没有真正执行。当您在方法中读取 args 时,Byte Buddy 每次都会创建一个新数组

因此,计算 args == args 将 return false 因为 Byte Buddy 每次都会创建一个包含所有参数的新数组!如果更改 args 数组,则必须将其写回,以便 Byte Buddy 发现相应的字节码并将其映射回赋值。