使用byte-buddy (System.nanoTime())截取和更改静态本机方法的返回值

Question

我想改变为任意 Java 程序调用 System.nanoTime（） 时会发生什么（我想将时间移回以帮助容器检查点/恢复用例）。每当调用System.nanoTime（）时，我都想运行原始的System.nanoTime（），然后可能返回一个不同的值。我正在尝试使用字节伙伴和 JVM 仪表代理来执行此操作（根据此处和此处）。

这是我的尝试：

public final class TimeShifter {
    private TimeShifter() {}

    private static final Class<?> INTERCEPTOR_CLASS = TimeShiftedSystem.class;

    public static void premain(String _arg, Instrumentation instrumentation) throws Exception {
        injectBootstrapClasses(instrumentation);
        new AgentBuilder.Default()
                .with(AgentBuilder.RedefinitionStrategy.RETRANSFORMATION)
                .with(AgentBuilder.InitializationStrategy.NoOp.INSTANCE)
                .with(AgentBuilder.TypeStrategy.Default.REDEFINE)
                .ignore(
                        new AgentBuilder.RawMatcher.ForElementMatchers(
                                ElementMatchers.nameStartsWith("net.bytebuddy.")
                                        .or(ElementMatchers.isSynthetic())
                                        .or(ElementMatchers.nameStartsWith("my.package.name.")),
                                ElementMatchers.any(),
                                ElementMatchers.any()))
                .with(
                        new AgentBuilder.Listener.Filtering(
                                new StringMatcher("java.lang.System", StringMatcher.Mode.EQUALS_FULLY),
                                AgentBuilder.Listener.StreamWriting.toSystemOut()))
                .type(ElementMatchers.named("java.lang.System"))
                .transform((builder, type, classLoader, module) ->
                        // https://stackoverflow.com/questions/49487148/redefine-rebase-native-method
                        builder.method(ElementMatchers.named("nanoTime"))
                        .intercept(Advice.to(TimeShiftedSystem.class)))
                .installOn(instrumentation);
    }

    private static void injectBootstrapClasses(Instrumentation instrumentation) throws IOException {
        File temp = Files.createTempDirectory("tmp").toFile();
        temp.deleteOnExit();

        ClassInjector.UsingInstrumentation.of(
                        temp, ClassInjector.UsingInstrumentation.Target.BOOTSTRAP, instrumentation)
                .inject(Collections.singletonMap(
                        new TypeDescription.ForLoadedType(INTERCEPTOR_CLASS),
                        ClassFileLocator.ForClassLoader.read(INTERCEPTOR_CLASS)));
    }
}

和

public final class TimeShiftedSystem {
    private TimeShiftedSystem() {}

    @Advice.OnMethodExit
    public static long nanoTime(@Advice.Return long originalReturnValue) {
        System.out.println("originalReturnValue = " + originalReturnValue);

        return originalReturnValue - 1000;
    }
}

但这失败了：

[Byte Buddy] DISCOVERY java.lang.System [null, module java.base, loaded=true]
[Byte Buddy] ERROR java.lang.System [null, module java.base, loaded=true]
java.lang.IllegalStateException: Cannot call super (or default) method for public static native long java.lang.System.nanoTime()
    at net.bytebuddy.implementation.SuperMethodCall$Appender.apply(SuperMethodCall.java:133)

有什么办法可以实现这一点？我试过的其他东西:

• 使用 MethodDelegation.to 而不是 Advice.to 然后在 TimeShiftedSystem#nanoTime（） 中调用 System.nanoTime（） - 这似乎是第一次调用原始 nanoTime，但立即进入堆栈溢出。
• 使用 MethodDelegation.to 而不是 Advice.to 和 @Origin 方法方法作为参数。似乎根本没有被调用（如果还使用了@RuntimeType注释，则不调用）。
• 使用 enabledNativeMethodPrefix（“native”） - 不执行任何操作。

任何帮助都非常感谢。

Answer 1

解决方案是设置

    .with(AgentBuilder.TypeStrategy.Default.REBASE)
    .enableNativeMethodPrefix("native")

在< code>AgentBuilder上，而不是< code>AgentBuilder上。type strategy . default . redefine (并将< code > ' Can-Set-Native-Method-Prefix ':true 添加到我的jar清单中)。这使我能够让< code > timeshift dsystem 处理以下建议:

@Advice.OnMethodExit
public static void nanoTime(@Advice.Return(readOnly = false) long returnValue) {
    System.out.println("Called with " + returnValue);
    returnValue = 42;
}