如何使用ByteBuddy委托来拦截引导程序类方法

时间:2019-04-09 06:45:03

标签: java instrumentation byte-buddy javaagents

我正在使用Java代理和bytebuddy来拦截FileIOStreams中的“读取”和“写入”方法。要实现的功能是“在某些情况下调用原始方法,否则通过”。因此,我需要使用方法委托完全控制调用流程,而不是使用Advice对其进行包装。

当@Morph不存在时,方法拦截工作正常,但是当我将@Morph添加到参数时,该方法拦截不起作用。我已经用其他一些注释进行了测试:

添加@ AllArguments,@这不会阻止委托,该方法将作为我的拦截器运行

添加@ Morph,@ SuperCall将阻止委派。不会抛出异常:原始方法将按原样运行。

这是我要实现的代码:

public static void mountAgent(Instrumentation inst) {

        new AgentBuilder.Default()
                .with(AgentBuilder.RedefinitionStrategy.REDEFINITION)
                .with(AgentBuilder.InitializationStrategy.NoOp.INSTANCE)
                .with(AgentBuilder.TypeStrategy.Default.REDEFINE)
                .ignore(new AgentBuilder.RawMatcher.ForElementMatchers(nameStartsWith("net.bytebuddy.").or(isSynthetic()), any(), any()))
                .with(new AgentBuilder.Listener.Filtering(
                        new StringMatcher("java.io.FileInputStream", StringMatcher.Mode.EQUALS_FULLY)
                                .or(new StringMatcher("java.io.FileOutputStream", StringMatcher.Mode.EQUALS_FULLY)),
                        AgentBuilder.Listener.StreamWriting.toSystemOut()))
                .type(named("java.io.FileOutputStream"))
                .transform(new AgentBuilder.Transformer() {
                    @Override
                    public DynamicType.Builder<?> transform(DynamicType.Builder<?> builder,
                                                            TypeDescription typeDescription,
                                                            ClassLoader classLoader,
                                                            JavaModule module) {
                        return builder.method(named("write").and(not(isNative())).and(takesArgument(0, byte[].class)))
                                .intercept(MethodDelegation
                                        .withDefaultConfiguration()
                                        .withBinders(Morph.Binder.install(Morphing.class))
                                        .to(WriteInterceptor.class));
                    }})
                .installOn(inst);
    }

(跳过了将拦截器附加到BootstrapClassLoaderSearch的代码)

以下是我的拦截器:

public interface Morphing<T> {
        T Object invoke(Object[] agrs);
    }

@SuppressWarnings("unused")
public static class WriteInterceptor {
    @RuntimeType
    public static void write(
//change the input here
            byte[] bytes,
            @AllArguments Object[] args,
            @Morph Morphing<Void> morphing
    ) throws Exception {
        if (true) {
            morphing.invoke(args);
        }
        else {
            // do something
            throw new Exception();
        }
    }
}

如果拦截函数的输入为空或只有byte []个字节,则委派将起作用并且抛出Exception:

[Byte Buddy] IGNORE java.io.FileInputStream [null, module java.base, loaded=true]
[Byte Buddy] COMPLETE java.io.FileInputStream [null, module java.base, loaded=true]
[Byte Buddy] DISCOVERY java.io.FileOutputStream [null, module java.base, loaded=true]
[Byte Buddy] TRANSFORM java.io.FileOutputStream [null, module java.base, loaded=true]
[Byte Buddy] COMPLETE java.io.FileOutputStream [null, module java.base, loaded=true]

Exception: java.lang.Exception thrown from the UncaughtExceptionHandler in thread "main"

如果输入为

  

byte []个字节,@AllArguments对象[] args,@Morph Morphing变形

  

@AllArguments对象[]参数,@ Morph变形变形

调用内置写函数,输出为

[Byte Buddy] IGNORE java.io.FileInputStream [null, module java.base, loaded=true]
[Byte Buddy] COMPLETE java.io.FileInputStream [null, module java.base, loaded=true]
[Byte Buddy] DISCOVERY java.io.FileOutputStream [null, module java.base, loaded=true]
[Byte Buddy] TRANSFORM java.io.FileOutputStream [null, module java.base, loaded=true]
[Byte Buddy] COMPLETE java.io.FileOutputStream [null, module java.base, loaded=true]

添加@Morph后,委托不起作用的原因是什么,但是bytebuddy仍然说转换已完成?在这种情况下如何获得正确的变形?谢谢!

1 个答案:

答案 0 :(得分:0)

我认为您的重新转换已经失败。您是否尝试将侦听器添加到重新转换过程中。 从线程“ main”中的UncaughtExceptionHandler抛出的 Exception:java.lang.Exception是什么?

我注意到的一件事是您不调整模块图。 java.base 模块将无法看到您的拦截器,该拦截器很可能已加载到引导加载程序的未命名模块中。您是否尝试过在指向拦截器类的转换中添加 assureReadEdgeTo

另外,请注意Advice确实允许您跳过甚至重复执行方法。看一下enter或exit方法的javadoc。通常,在安装引导程序类时,建议往往更可靠。