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



我想改变为任意 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 Method method作为参数。似乎根本没有被调用(如果还使用了@RuntimeType注释,则不调用(。
  • 使用enabledNativeMethodPrefix("native")- 什么都不做。

任何帮助将不胜感激。

解决方案是设置

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

AgentBuilder而不是agentBuilder.TypeStrategy.Default.REDEFINE(并将'Can-Set-Native-Method-Prefix': true添加到我的 jar 清单中(。这使我能够让TimeShiftedSystem使用此建议:

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

相关内容

  • 没有找到相关文章

最新更新